1 #include "../../include/freenet/freenetmasterthread.h"
\r
2 #include "../../include/option.h"
\r
3 #include "../../include/stringfunctions.h"
\r
4 #include "../../include/freenet/unkeyedidcreator.h"
\r
5 #include "../../include/freenet/identityinserter.h"
\r
6 #include "../../include/freenet/identityrequester.h"
\r
7 #include "../../include/freenet/unknownidentityrequester.h"
\r
8 #include "../../include/freenet/introductionpuzzleinserter.h"
\r
9 #include "../../include/freenet/identityintroductionrequester.h"
\r
10 #include "../../include/freenet/introductionpuzzlerequester.h"
\r
11 #include "../../include/freenet/identityintroductioninserter.h"
\r
12 #include "../../include/freenet/trustlistinserter.h"
\r
13 #include "../../include/freenet/trustlistrequester.h"
\r
14 #include "../../include/freenet/messagelistrequester.h"
\r
15 #include "../../include/freenet/inactivemessagelistrequester.h"
\r
16 #include "../../include/freenet/messagelistinserter.h"
\r
17 #include "../../include/freenet/messagerequester.h"
\r
18 #include "../../include/freenet/messageinserter.h"
\r
19 #include "../../include/freenet/boardlistinserter.h"
\r
20 #include "../../include/freenet/boardlistrequester.h"
\r
21 #include "../../include/freenet/siteinserter.h"
\r
22 #include "../../include/freenet/fileinserter.h"
\r
23 #include "../../include/freenet/fmsversionrequester.h"
\r
24 #ifdef FROST_SUPPORT
\r
25 #include "../../include/freenet/frostmessagerequester.h"
\r
28 #include <Poco/UUID.h>
\r
29 #include <Poco/UUIDGenerator.h>
\r
30 #include <Poco/DateTime.h>
\r
31 #include <Poco/Timespan.h>
\r
32 #include <Poco/Thread.h>
\r
40 FreenetMasterThread::FreenetMasterThread():m_receivednodehello(false)
\r
45 FreenetMasterThread::~FreenetMasterThread()
\r
50 const bool FreenetMasterThread::FCPConnect()
\r
52 // we were previosly connected, send FCPDisconnect to objects
\r
53 if(m_receivednodehello==true)
\r
55 for(std::vector<IFCPConnected *>::iterator i=m_fcpconnected.begin(); i!=m_fcpconnected.end(); i++)
\r
57 (*i)->FCPDisconnected();
\r
59 m_receivednodehello=false;
\r
62 m_log->information("FreenetMasterThread::FCPConnect trying to connect to node "+m_fcphost);
\r
64 if(m_fcp.Connect(m_fcphost.c_str(),m_fcpport)==true)
\r
66 Poco::UUIDGenerator uuidgen;
\r
71 uuid=uuidgen.createRandom();
\r
75 m_log->fatal("FreenetMasterThread::FCPConnect could not generate UUID");
\r
78 std::string clientname="FMSClient-"+uuid.toString();
\r
79 // send ClientHello message to node
\r
80 m_fcp.Send(FCPv2::Message("ClientHello",2,"Name",clientname.c_str(),"ExpectedVersion","2.0"));
\r
82 m_log->information("FreenetMasterThread::FCPConnect connected to node");
\r
93 const bool FreenetMasterThread::HandleMessage(FCPv2::Message &message)
\r
95 if(message.GetName()=="NodeHello")
\r
97 m_receivednodehello=true;
\r
99 // send connected message to all objects, must do this AFTER we received the NodeHello message
\r
100 for(std::vector<IFCPConnected *>::iterator i=m_fcpconnected.begin(); i!=m_fcpconnected.end(); i++)
\r
102 (*i)->FCPConnected();
\r
107 if(m_receivednodehello==true)
\r
109 bool handled=false;
\r
111 std::vector<IFCPMessageHandler *>::iterator i=m_fcpmessagehandlers.begin();
\r
112 while(handled==false && i!=m_fcpmessagehandlers.end())
\r
114 handled=(*i)->HandleMessage(message);
\r
120 std::string info("");
\r
121 for(std::map<std::string,std::string>::iterator mi=message.GetFields().begin(); mi!=message.GetFields().end(); mi++)
\r
123 info+="\t\t\t\t"+(*mi).first+"="+(*mi).second+"\r\n";
\r
125 m_log->debug("FreenetMasterThread::HandleMessage received unhandled "+message.GetName()+" message. Message content :\r\n"+info);
\r
127 // if unhandled message was alldata - we must retrieve the data
\r
128 if(message.GetName()=="AllData")
\r
131 StringFunctions::Convert(message["DataLength"],length);
\r
132 m_fcp.WaitForBytes(1000,length);
\r
134 if(m_fcp.IsConnected() && length>0)
\r
136 m_fcp.ReceiveIgnore(length);
\r
146 m_log->error("FreenetMasterThread::HandleMessage received "+message.GetName()+" message before NodeHello");
\r
152 void FreenetMasterThread::RegisterFCPConnected(IFCPConnected *obj)
\r
154 m_fcpconnected.push_back(obj);
\r
157 void FreenetMasterThread::RegisterFCPMessageHandler(IFCPMessageHandler *obj)
\r
159 m_fcpmessagehandlers.push_back(obj);
\r
162 void FreenetMasterThread::RegisterPeriodicProcessor(IPeriodicProcessor *obj)
\r
164 m_processors.push_back(obj);
\r
167 void FreenetMasterThread::run()
\r
170 Poco::DateTime lastreceivedmessage;
\r
171 Poco::DateTime lastconnected;
\r
172 Poco::DateTime now;
\r
173 FCPv2::Message message;
\r
176 lastconnected-=Poco::Timespan(0,0,1,0,0);
\r
178 m_log->debug("FreenetMasterThread::run thread started.");
\r
190 if(m_fcp.IsConnected()==false)
\r
192 // wait at least 1 minute since last successful connect
\r
193 now=Poco::Timestamp();
\r
194 if(lastconnected<=(now-Poco::Timespan(0,0,1,0,0)))
\r
196 if(FCPConnect()==false)
\r
199 m_log->error("FreenetMasterThread::run could not connect to node. Waiting 60 seconds.");
\r
201 for(int i=0; i<60 && !IsCancelled(); i++)
\r
203 Poco::Thread::sleep(1000);
\r
208 lastreceivedmessage=Poco::Timestamp();
\r
209 lastconnected=Poco::Timestamp();
\r
214 Poco::Thread::sleep(1000);
\r
217 // fcp is connected
\r
220 m_fcp.Update(1000);
\r
222 // check for message on receive buffer and handle it
\r
223 if(m_fcp.MessageReady()==true)
\r
226 m_fcp.Receive(message);
\r
228 if(message.GetName()!="")
\r
230 HandleMessage(message);
\r
231 lastreceivedmessage=Poco::Timestamp();
\r
235 // let objects do their processing
\r
236 for(std::vector<IPeriodicProcessor *>::iterator i=m_processors.begin(); i!=m_processors.end(); i++)
\r
241 // if we haven't received any messages from the node in 10 minutes, something is wrong
\r
242 now=Poco::Timestamp();
\r
243 if(lastreceivedmessage<(now-Poco::Timespan(0,0,10,0,0)))
\r
245 m_log->error("FreenetMasterThread::Run The Freenet node has not responded in 10 minutes. Trying to reconnect.");
\r
246 m_fcp.Disconnect();
\r
249 if(m_fcp.IsConnected()==false)
\r
251 m_log->information("FreenetMasterThread::Run Disconnected from Freenet node.");
\r
256 catch(Poco::Exception &e)
\r
258 m_log->error("FreenetMasterThread::run caught exception : "+e.displayText());
\r
262 m_log->error("FreenetMasterThread::run caught unknown exception");
\r
264 }while(!IsCancelled() && done==false);
\r
266 m_fcp.Disconnect();
\r
270 m_log->debug("FreenetMasterThread::run thread exiting.");
\r
274 void FreenetMasterThread::Setup()
\r
277 std::string temp="";
\r
278 Option option(m_db);
\r
279 if(option.Get("FCPHost",m_fcphost)==false)
\r
281 m_fcphost="localhost";
\r
282 option.Set("FCPHost",m_fcphost);
\r
284 if(option.GetInt("FCPPort",m_fcpport)==false)
\r
287 option.Set("FCPPort",m_fcpport);
\r
291 if(m_fcpport<1 || m_fcpport>65535)
\r
294 option.Set("FCPPort",m_fcpport);
\r
298 // seed random number generator
\r
301 m_registrables.push_back(new UnkeyedIDCreator(m_db,&m_fcp));
\r
302 m_registrables.push_back(new IdentityInserter(m_db,&m_fcp));
\r
303 m_registrables.push_back(new IdentityRequester(m_db,&m_fcp));
\r
304 m_registrables.push_back(new UnknownIdentityRequester(m_db,&m_fcp));
\r
305 m_registrables.push_back(new IntroductionPuzzleInserter(m_db,&m_fcp));
\r
306 m_registrables.push_back(new IdentityIntroductionRequester(m_db,&m_fcp));
\r
307 m_registrables.push_back(new IntroductionPuzzleRequester(m_db,&m_fcp));
\r
308 m_registrables.push_back(new IdentityIntroductionInserter(m_db,&m_fcp));
\r
309 m_registrables.push_back(new TrustListInserter(m_db,&m_fcp));
\r
310 m_registrables.push_back(new TrustListRequester(m_db,&m_fcp));
\r
311 m_registrables.push_back(new MessageListInserter(m_db,&m_fcp));
\r
312 m_registrables.push_back(new MessageListRequester(m_db,&m_fcp));
\r
313 m_registrables.push_back(new InactiveMessageListRequester(m_db,&m_fcp));
\r
314 m_registrables.push_back(new MessageInserter(m_db,&m_fcp));
\r
315 m_registrables.push_back(new MessageRequester(m_db,&m_fcp));
\r
316 m_registrables.push_back(new BoardListInserter(m_db,&m_fcp));
\r
317 m_registrables.push_back(new BoardListRequester(m_db,&m_fcp));
\r
318 m_registrables.push_back(new SiteInserter(m_db,&m_fcp));
\r
319 m_registrables.push_back(new FileInserter(m_db,&m_fcp));
\r
320 m_registrables.push_back(new FMSVersionRequester(m_db,&m_fcp));
\r
321 #ifdef FROST_SUPPORT
\r
323 option.Get("DownloadFrostMessages",temp);
\r
326 m_registrables.push_back(new FrostMessageRequester(m_db,&m_fcp));
\r
330 for(std::vector<IFreenetRegistrable *>::iterator i=m_registrables.begin(); i!=m_registrables.end(); i++)
\r
332 (*i)->RegisterWithThread(this);
\r
337 void FreenetMasterThread::Shutdown()
\r
339 // delete each registrable object
\r
340 for(std::vector<IFreenetRegistrable *>::iterator i=m_registrables.begin(); i!=m_registrables.end(); i++)
\r