X-Git-Url: https://git.pterodactylus.net/?p=fms.git;a=blobdiff_plain;f=src%2Ffreenet%2Ffreenetmasterthread.cpp;h=57541a4207914da5c12a911bb6baebd9516a17ce;hp=a6445addcadd883401c8b8ae27634145f5b4454c;hb=HEAD;hpb=44f964d9b2b2d55a5b5672e9297717bd25fa8ee2 diff --git a/src/freenet/freenetmasterthread.cpp b/src/freenet/freenetmasterthread.cpp index a6445ad..57541a4 100644 --- a/src/freenet/freenetmasterthread.cpp +++ b/src/freenet/freenetmasterthread.cpp @@ -1,10 +1,10 @@ #include "../../include/freenet/freenetmasterthread.h" #include "../../include/option.h" -#include "../../include/uuidgenerator.h" #include "../../include/stringfunctions.h" #include "../../include/freenet/unkeyedidcreator.h" #include "../../include/freenet/identityinserter.h" #include "../../include/freenet/identityrequester.h" +#include "../../include/freenet/unknownidentityrequester.h" #include "../../include/freenet/introductionpuzzleinserter.h" #include "../../include/freenet/identityintroductionrequester.h" #include "../../include/freenet/introductionpuzzlerequester.h" @@ -12,15 +12,24 @@ #include "../../include/freenet/trustlistinserter.h" #include "../../include/freenet/trustlistrequester.h" #include "../../include/freenet/messagelistrequester.h" +#include "../../include/freenet/inactivemessagelistrequester.h" +#include "../../include/freenet/messagelistinserter.h" #include "../../include/freenet/messagerequester.h" #include "../../include/freenet/messageinserter.h" -#include "../../include/freenet/messagelistinserter.h" -#include "../../include/freenet/periodicdbmaintenance.h" #include "../../include/freenet/boardlistinserter.h" #include "../../include/freenet/boardlistrequester.h" #include "../../include/freenet/siteinserter.h" #include "../../include/freenet/fileinserter.h" -#include "../../include/pthreadwrapper/thread.h" +#include "../../include/freenet/fmsversionrequester.h" +#ifdef FROST_SUPPORT + #include "../../include/freenet/frostmessagerequester.h" +#endif + +#include +#include +#include +#include +#include #include @@ -28,29 +37,8 @@ #include #endif -FreenetMasterThread::FreenetMasterThread() +FreenetMasterThread::FreenetMasterThread():m_receivednodehello(false) { - std::string fcpport; - - if(Option::Instance()->Get("FCPHost",m_fcphost)==false) - { - m_fcphost="localhost"; - Option::Instance()->Set("FCPHost",m_fcphost); - } - if(Option::Instance()->Get("FCPPort",fcpport)==false) - { - fcpport="9481"; - Option::Instance()->Set("FCPPort",fcpport); - } - - // convert fcp port to long, and make sure it's within the valid port range - if(StringFunctions::Convert(fcpport,m_fcpport)==false) - { - m_fcpport=9481; - Option::Instance()->Set("FCPPort","9481"); - } - - m_receivednodehello=false; } @@ -71,16 +59,27 @@ const bool FreenetMasterThread::FCPConnect() m_receivednodehello=false; } - m_log->WriteLog(LogFile::LOGLEVEL_INFO,"FreenetMasterThread::FCPConnect trying to connect to node "+m_fcphost); + m_log->information("FreenetMasterThread::FCPConnect trying to connect to node "+m_fcphost); if(m_fcp.Connect(m_fcphost.c_str(),m_fcpport)==true) { - UUIDGenerator uuid; - std::string clientname="FMSClient-"+uuid.Generate(); + Poco::UUIDGenerator uuidgen; + Poco::UUID uuid; + + try + { + uuid=uuidgen.createRandom(); + } + catch(...) + { + m_log->fatal("FreenetMasterThread::FCPConnect could not generate UUID"); + } + + std::string clientname="FMSClient-"+uuid.toString(); // send ClientHello message to node - m_fcp.SendMessage("ClientHello",2,"Name",clientname.c_str(),"ExpectedVersion","2.0"); + m_fcp.Send(FCPv2::Message("ClientHello",2,"Name",clientname.c_str(),"ExpectedVersion","2.0")); - m_log->WriteLog(LogFile::LOGLEVEL_INFO,"FreenetMasterThread::FCPConnect connected to node"); + m_log->information("FreenetMasterThread::FCPConnect connected to node"); return true; } @@ -91,7 +90,7 @@ const bool FreenetMasterThread::FCPConnect() } -const bool FreenetMasterThread::HandleMessage(FCPMessage &message) +const bool FreenetMasterThread::HandleMessage(FCPv2::Message &message) { if(message.GetName()=="NodeHello") { @@ -108,6 +107,7 @@ const bool FreenetMasterThread::HandleMessage(FCPMessage &message) if(m_receivednodehello==true) { bool handled=false; + std::vector::iterator i=m_fcpmessagehandlers.begin(); while(handled==false && i!=m_fcpmessagehandlers.end()) { @@ -118,26 +118,22 @@ const bool FreenetMasterThread::HandleMessage(FCPMessage &message) if(handled==false) { std::string info(""); - for(std::map::iterator mi=message.begin(); mi!=message.end(); mi++) + for(std::map::iterator mi=message.GetFields().begin(); mi!=message.GetFields().end(); mi++) { info+="\t\t\t\t"+(*mi).first+"="+(*mi).second+"\r\n"; } - m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"FreenetMasterThread::HandleMessage received unhandled "+message.GetName()+" message. Message content :\r\n"+info); + m_log->debug("FreenetMasterThread::HandleMessage received unhandled "+message.GetName()+" message. Message content :\r\n"+info); // if unhandled message was alldata - we must retrieve the data if(message.GetName()=="AllData") { long length; StringFunctions::Convert(message["DataLength"],length); - while(m_fcp.Connected() && m_fcp.ReceiveBufferSize()0) { - char *data=new char[length]; - m_fcp.ReceiveRaw(data,length); - delete [] data; + m_fcp.ReceiveIgnore(length); } } } @@ -147,7 +143,7 @@ const bool FreenetMasterThread::HandleMessage(FCPMessage &message) } else { - m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"FreenetMasterThread::HandleMessage received "+message.GetName()+" message before NodeHello"); + m_log->error("FreenetMasterThread::HandleMessage received "+message.GetName()+" message before NodeHello"); } return false; @@ -168,88 +164,102 @@ void FreenetMasterThread::RegisterPeriodicProcessor(IPeriodicProcessor *obj) m_processors.push_back(obj); } -void FreenetMasterThread::Run() +void FreenetMasterThread::run() { - DateTime lastreceivedmessage; - DateTime lastconnected; - DateTime now; - FCPMessage message; + Poco::DateTime lastreceivedmessage; + Poco::DateTime lastconnected; + Poco::DateTime now; + FCPv2::Message message; bool done=false; - lastconnected.SetToGMTime(); - lastconnected.Add(0,-1); + lastconnected-=Poco::Timespan(0,0,1,0,0); + + m_log->debug("FreenetMasterThread::run thread started."); + + LoadDatabase(); + - m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"FreenetMasterThread::run thread started."); Setup(); do { - if(m_fcp.Connected()==false) + try { - // wait at least 1 minute since last successful connect - now.SetToGMTime(); - if(lastconnected<=(now-(1.0/1440.0))) + if(m_fcp.IsConnected()==false) { - if(FCPConnect()==false) + // wait at least 1 minute since last successful connect + now=Poco::Timestamp(); + if(lastconnected<=(now-Poco::Timespan(0,0,1,0,0))) { + if(FCPConnect()==false) + { - m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"FreenetMasterThread::run could not connect to node. Waiting 60 seconds."); + m_log->error("FreenetMasterThread::run could not connect to node. Waiting 60 seconds."); - for(int i=0; i<60 && !IsCancelled(); i++) + for(int i=0; i<60 && !IsCancelled(); i++) + { + Poco::Thread::sleep(1000); + } + } + else { - Sleep(1000); + lastreceivedmessage=Poco::Timestamp(); + lastconnected=Poco::Timestamp(); } } else { - lastreceivedmessage.SetToGMTime(); - lastconnected.SetToGMTime(); + Poco::Thread::sleep(1000); } } + // fcp is connected else { - Sleep(1000); - } - } - // fcp is connected - else - { - m_fcp.Update(1); + m_fcp.Update(1000); - // check for message on receive buffer and handle it - if(m_fcp.ReceiveBufferSize()>0) - { - message.Reset(); - message=m_fcp.ReceiveMessage(); + // check for message on receive buffer and handle it + if(m_fcp.MessageReady()==true) + { + message.Clear(); + m_fcp.Receive(message); - if(message.GetName()!="") + if(message.GetName()!="") + { + HandleMessage(message); + lastreceivedmessage=Poco::Timestamp(); + } + } + + // let objects do their processing + for(std::vector::iterator i=m_processors.begin(); i!=m_processors.end(); i++) { - HandleMessage(message); - lastreceivedmessage.SetToGMTime(); + (*i)->Process(); } - } - // let objects do their processing - for(std::vector::iterator i=m_processors.begin(); i!=m_processors.end(); i++) - { - (*i)->Process(); - } + // if we haven't received any messages from the node in 10 minutes, something is wrong + now=Poco::Timestamp(); + if(lastreceivedmessage<(now-Poco::Timespan(0,0,10,0,0))) + { + m_log->error("FreenetMasterThread::Run The Freenet node has not responded in 10 minutes. Trying to reconnect."); + m_fcp.Disconnect(); + } - // if we haven't received any messages from the node in 10 minutes, something is wrong - now.SetToGMTime(); - if(lastreceivedmessage<(now-(1.0/144.0))) - { - m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"FreenetMasterThread::Run The Freenet node has not responded in 10 minutes. Trying to reconnect."); - m_fcp.Disconnect(); - } + if(m_fcp.IsConnected()==false) + { + m_log->information("FreenetMasterThread::Run Disconnected from Freenet node."); + } - if(m_fcp.Connected()==false) - { - m_log->WriteLog(LogFile::LOGLEVEL_INFO,"FreenetMasterThread::Run Disconnected from Freenet node."); } - + } + catch(Poco::Exception &e) + { + m_log->error("FreenetMasterThread::run caught exception : "+e.displayText()); + } + catch(...) + { + m_log->error("FreenetMasterThread::run caught unknown exception"); } }while(!IsCancelled() && done==false); @@ -257,34 +267,65 @@ void FreenetMasterThread::Run() Shutdown(); - m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"FreenetMasterThread::run thread exiting."); + m_log->debug("FreenetMasterThread::run thread exiting."); } void FreenetMasterThread::Setup() { + std::string temp=""; + Option option(m_db); + if(option.Get("FCPHost",m_fcphost)==false) + { + m_fcphost="localhost"; + option.Set("FCPHost",m_fcphost); + } + if(option.GetInt("FCPPort",m_fcpport)==false) + { + m_fcpport=9481; + option.Set("FCPPort",m_fcpport); + } + else + { + if(m_fcpport<1 || m_fcpport>65535) + { + m_fcpport=9481; + option.Set("FCPPort",m_fcpport); + } + } + // seed random number generator srand(time(NULL)); - m_registrables.push_back(new UnkeyedIDCreator(&m_fcp)); - m_registrables.push_back(new IdentityInserter(&m_fcp)); - m_registrables.push_back(new IdentityRequester(&m_fcp)); - m_registrables.push_back(new IntroductionPuzzleInserter(&m_fcp)); - m_registrables.push_back(new IdentityIntroductionRequester(&m_fcp)); - m_registrables.push_back(new IntroductionPuzzleRequester(&m_fcp)); - m_registrables.push_back(new IdentityIntroductionInserter(&m_fcp)); - m_registrables.push_back(new TrustListInserter(&m_fcp)); - m_registrables.push_back(new TrustListRequester(&m_fcp)); - m_registrables.push_back(new MessageListInserter(&m_fcp)); - m_registrables.push_back(new MessageListRequester(&m_fcp)); - m_registrables.push_back(new MessageRequester(&m_fcp)); - m_registrables.push_back(new MessageInserter(&m_fcp)); - m_registrables.push_back(new BoardListInserter(&m_fcp)); - m_registrables.push_back(new BoardListRequester(&m_fcp)); - m_registrables.push_back(new SiteInserter(&m_fcp)); - m_registrables.push_back(new FileInserter(&m_fcp)); - m_registrables.push_back(new PeriodicDBMaintenance()); + m_registrables.push_back(new UnkeyedIDCreator(m_db,&m_fcp)); + m_registrables.push_back(new IdentityInserter(m_db,&m_fcp)); + m_registrables.push_back(new IdentityRequester(m_db,&m_fcp)); + m_registrables.push_back(new UnknownIdentityRequester(m_db,&m_fcp)); + m_registrables.push_back(new IntroductionPuzzleInserter(m_db,&m_fcp)); + m_registrables.push_back(new IdentityIntroductionRequester(m_db,&m_fcp)); + m_registrables.push_back(new IntroductionPuzzleRequester(m_db,&m_fcp)); + m_registrables.push_back(new IdentityIntroductionInserter(m_db,&m_fcp)); + m_registrables.push_back(new TrustListInserter(m_db,&m_fcp)); + m_registrables.push_back(new TrustListRequester(m_db,&m_fcp)); + m_registrables.push_back(new MessageListInserter(m_db,&m_fcp)); + m_registrables.push_back(new MessageListRequester(m_db,&m_fcp)); + m_registrables.push_back(new InactiveMessageListRequester(m_db,&m_fcp)); + m_registrables.push_back(new MessageInserter(m_db,&m_fcp)); + m_registrables.push_back(new MessageRequester(m_db,&m_fcp)); + m_registrables.push_back(new BoardListInserter(m_db,&m_fcp)); + m_registrables.push_back(new BoardListRequester(m_db,&m_fcp)); + m_registrables.push_back(new SiteInserter(m_db,&m_fcp)); + m_registrables.push_back(new FileInserter(m_db,&m_fcp)); + m_registrables.push_back(new FMSVersionRequester(m_db,&m_fcp)); +#ifdef FROST_SUPPORT + temp=""; + option.Get("DownloadFrostMessages",temp); + if(temp=="true") + { + m_registrables.push_back(new FrostMessageRequester(m_db,&m_fcp)); + } +#endif for(std::vector::iterator i=m_registrables.begin(); i!=m_registrables.end(); i++) {