X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ffreenet%2Fboardlistrequester.cpp;fp=src%2Ffreenet%2Fboardlistrequester.cpp;h=6b52a4603d364544451c0e8943f1a61f10707304;hb=df316253862dc50e8e5a790d9634ef90be37badb;hp=0000000000000000000000000000000000000000;hpb=f60495a029c54358f82956482fe203fe2b7b5b23;p=fms.git

diff --git a/src/freenet/boardlistrequester.cpp b/src/freenet/boardlistrequester.cpp
new file mode 100644
index 0000000..6b52a46
--- /dev/null
+++ b/src/freenet/boardlistrequester.cpp
@@ -0,0 +1,243 @@
+#include "../../include/freenet/boardlistrequester.h"
+#include "../../include/freenet/boardlistxml.h"
+
+#ifdef XMEM
+	#include <xmem.h>
+#endif
+
+BoardListRequester::BoardListRequester()
+{
+	Initialize();
+}
+
+BoardListRequester::BoardListRequester(FCPv2 *fcp):IIndexRequester<long>(fcp)
+{
+	Initialize();
+}
+
+const bool BoardListRequester::HandleAllData(FCPMessage &message)
+{	
+	DateTime now;
+	SQLite3DB::Statement st;
+	std::vector<std::string> idparts;
+	long datalength;
+	std::vector<char> data;
+	BoardListXML xml;
+	long identityid;
+	long index;
+
+	now.SetToGMTime();
+	StringFunctions::Split(message["Identifier"],"|",idparts);
+	StringFunctions::Convert(message["DataLength"],datalength);
+	StringFunctions::Convert(idparts[1],identityid);
+	StringFunctions::Convert(idparts[2],index);
+
+	// wait for all data to be received from connection
+	while(m_fcp->Connected() && m_fcp->ReceiveBufferSize()<datalength)
+	{
+		m_fcp->Update(1);
+	}
+
+	// if we got disconnected- return immediately
+	if(m_fcp->Connected()==false)
+	{
+		return false;
+	}
+
+	// receive the file
+	data.resize(datalength);
+	m_fcp->ReceiveRaw(&data[0],datalength);
+
+	// parse file into xml and update the database
+	if(xml.ParseXML(std::string(data.begin(),data.end()))==true)
+	{
+
+		SQLite3DB::Statement brd=m_db->Prepare("SELECT BoardID,BoardName,BoardDescription FROM tblBoard WHERE BoardName=?;");
+		SQLite3DB::Statement ins=m_db->Prepare("INSERT INTO tblBoard(BoardName,BoardDescription,DateAdded) VALUES(?,?,?);");
+		SQLite3DB::Statement upd=m_db->Prepare("UPDATE tblBoard SET BoardDescription=? WHERE BoardID=?;");
+		for(long i=0; i<xml.GetCount(); i++)
+		{
+			int boardid;
+			std::string name="";
+			std::string description="";
+
+			brd.Bind(0,xml.GetName(i));
+			brd.Step();
+			
+			if(brd.RowReturned())
+			{
+				brd.ResultInt(0,boardid);
+				brd.ResultText(2,description);
+				if(description=="" && xml.GetDescription(i)!="")
+				{
+					upd.Bind(0,xml.GetDescription(i));
+					upd.Step();
+					upd.Reset();
+				}
+			}
+			else
+			{
+				ins.Bind(0,xml.GetName(i));
+				ins.Bind(1,xml.GetDescription(i));
+				ins.Bind(2,now.Format("%Y-%m-%d %H:%M:%S"));
+				ins.Step();
+				ins.Reset();
+			}
+			brd.Reset();
+		}
+
+		st=m_db->Prepare("INSERT INTO tblBoardListRequests(IdentityID,Day,RequestIndex,Found) VALUES(?,?,?,'true');");
+		st.Bind(0,identityid);
+		st.Bind(1,idparts[4]);
+		st.Bind(2,index);
+		st.Step();
+		st.Finalize();
+
+		m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"BoardListRequester::HandleAllData parsed BoardList XML file : "+message["Identifier"]);
+	}
+	else
+	{
+		// bad data - mark index
+		st=m_db->Prepare("INSERT INTO tblBoardListRequests(IdentityID,Day,RequestIndex,Found) VALUES(?,?,?,'false');");
+		st.Bind(0,identityid);
+		st.Bind(1,idparts[4]);
+		st.Bind(2,index);
+		st.Step();
+		st.Finalize();
+
+		m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"BoardListRequester::HandleAllData error parsing BoardList XML file : "+message["Identifier"]);
+	}
+
+	// remove this identityid from request list
+	RemoveFromRequestList(identityid);
+
+	return true;
+
+}
+
+const bool BoardListRequester::HandleGetFailed(FCPMessage &message)
+{
+	DateTime now;
+	SQLite3DB::Statement st;
+	std::vector<std::string> idparts;
+	long identityid;
+	long index;
+
+	now.SetToGMTime();
+	StringFunctions::Split(message["Identifier"],"|",idparts);
+	StringFunctions::Convert(idparts[1],identityid);
+	StringFunctions::Convert(idparts[2],index);	
+
+	// if this is a fatal error - insert index into database so we won't try to download this index again
+	if(message["Fatal"]=="true")
+	{
+		st=m_db->Prepare("INSERT INTO tblBoardListRequests(IdentityID,Day,RequestIndex,Found) VALUES(?,?,?,'false');");
+		st.Bind(0,identityid);
+		st.Bind(1,idparts[4]);
+		st.Bind(2,index);
+		st.Step();
+		st.Finalize();
+
+		m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"BoardListRequester::HandleGetFailed fatal error requesting "+message["Identifier"]);
+	}
+
+	// remove this identityid from request list
+	RemoveFromRequestList(identityid);
+
+	return true;
+}
+
+void BoardListRequester::Initialize()
+{
+	std::string tempval="";
+
+	m_fcpuniquename="BoardListRequester";
+	m_maxrequests=0;
+
+	Option::Instance()->Get("MaxBoardListRequests",tempval);
+	StringFunctions::Convert(tempval,m_maxrequests);
+	if(m_maxrequests<0)
+	{
+		m_maxrequests=0;
+		m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"Option MaxBoardListRequests is currently set at "+tempval+".  It must be 0 or greater.");
+	}
+	if(m_maxrequests>100)
+	{
+		m_log->WriteLog(LogFile::LOGLEVEL_WARNING,"Option MaxBoardListRequests is currently set at "+tempval+".  This value might be incorrectly configured.");
+	}
+}
+
+void BoardListRequester::PopulateIDList()
+{
+	int id;
+	DateTime today;
+	today.SetToGMTime();
+
+	SQLite3DB::Statement st=m_db->Prepare("SELECT IdentityID FROM tblIdentity WHERE PublicKey IS NOT NULL AND PublicKey <> '' AND LastSeen>='"+today.Format("%Y-%m-%d")+"' AND LocalMessageTrust>=(SELECT OptionValue FROM tblOption WHERE Option='MinLocalMessageTrust') AND PublishBoardList='true' ORDER BY LocalMessageTrust+LocalTrustListTrust DESC, LastSeen;");
+	st.Step();
+
+	m_ids.clear();
+
+	while(st.RowReturned())
+	{
+		st.ResultInt(0,id);
+		m_ids[id]=false;
+		st.Step();
+	}
+
+}
+
+void BoardListRequester::StartRequest(const long &identityid)
+{
+	DateTime now;
+	FCPMessage message;
+	std::string publickey;
+	std::string indexstr;
+	int index;
+	std::string identityidstr;
+
+	SQLite3DB::Statement st=m_db->Prepare("SELECT PublicKey FROM tblIdentity WHERE identityid=?;");
+	st.Bind(0,identityid);
+	st.Step();
+
+	if(st.RowReturned())
+	{
+		st.ResultText(0,publickey);
+
+		now.SetToGMTime();
+
+		SQLite3DB::Statement st2=m_db->Prepare("SELECT MAX(RequestIndex) FROM tblBoardListRequests WHERE Day=? AND IdentityID=?;");
+		st2.Bind(0,now.Format("%Y-%m-%d"));
+		st2.Bind(1,identityid);
+		st2.Step();
+
+		index=0;
+		if(st2.RowReturned())
+		{
+			if(st2.ResultNull(0)==false)
+			{
+				st2.ResultInt(0,index);
+				index++;
+			}
+		}
+		st2.Finalize();
+
+		StringFunctions::Convert(index,indexstr);
+		StringFunctions::Convert(identityid,identityidstr);
+
+		message.SetName("ClientGet");
+		message["URI"]=publickey+m_messagebase+"|"+now.Format("%Y-%m-%d")+"|BoardList|"+indexstr+".xml";
+		message["Identifier"]=m_fcpuniquename+"|"+identityidstr+"|"+indexstr+"|"+message["URI"];
+		message["ReturnType"]="direct";
+		message["MaxSize"]="100000";			// 100 KB
+
+		m_fcp->SendMessage(message);
+
+		m_requesting.push_back(identityid);
+
+	}
+	st.Finalize();
+
+	m_ids[identityid]=true;
+
+}