src/freenet/freenetmasterthread.cpp

   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
  26 #endif\r
  27 \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
  33 \r
  34 #include <cstdlib>\r
  35 \r
  36 #ifdef XMEM\r
  37         #include <xmem.h>\r
  38 #endif\r
  39 \r
  40 FreenetMasterThread::FreenetMasterThread():m_receivednodehello(false)\r
  41 {\r
  42 \r
  43 }\r
  44 \r
  45 FreenetMasterThread::~FreenetMasterThread()\r
  46 {\r
  47 \r
  48 }\r
  49 \r
  50 const bool FreenetMasterThread::FCPConnect()\r
  51 {\r
  52         // we were previosly connected, send FCPDisconnect to objects\r
  53         if(m_receivednodehello==true)\r
  54         {\r
  55                 for(std::vector<IFCPConnected *>::iterator i=m_fcpconnected.begin(); i!=m_fcpconnected.end(); i++)\r
  56                 {\r
  57                         (*i)->FCPDisconnected();\r
  58                 }\r
  59                 m_receivednodehello=false;\r
  60         }\r
  61 \r
  62         m_log->information("FreenetMasterThread::FCPConnect trying to connect to node "+m_fcphost);\r
  63 \r
  64         if(m_fcp.Connect(m_fcphost.c_str(),m_fcpport)==true)\r
  65         {\r
  66                 Poco::UUIDGenerator uuidgen;\r
  67                 Poco::UUID uuid;\r
  68 \r
  69                 try\r
  70                 {\r
  71                         uuid=uuidgen.createRandom();\r
  72                 }\r
  73                 catch(...)\r
  74                 {\r
  75                         m_log->fatal("FreenetMasterThread::FCPConnect could not generate UUID");\r
  76                 }\r
  77 \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
  81 \r
  82                 m_log->information("FreenetMasterThread::FCPConnect connected to node");\r
  83 \r
  84                 return true;\r
  85         }\r
  86         else\r
  87         {\r
  88                 return false;\r
  89         }\r
  90 \r
  91 }\r
  92 \r
  93 const bool FreenetMasterThread::HandleMessage(FCPv2::Message &message)\r
  94 {\r
  95         if(message.GetName()=="NodeHello")\r
  96         {\r
  97                 m_receivednodehello=true;\r
  98 \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
 101                 {\r
 102                         (*i)->FCPConnected();\r
 103                 }\r
 104 \r
 105                 return true;\r
 106         }\r
 107         if(m_receivednodehello==true)\r
 108         {\r
 109                 bool handled=false;\r
 110 \r
 111                 std::vector<IFCPMessageHandler *>::iterator i=m_fcpmessagehandlers.begin();\r
 112                 while(handled==false && i!=m_fcpmessagehandlers.end())\r
 113                 {\r
 114                         handled=(*i)->HandleMessage(message);\r
 115                         i++;\r
 116                 }\r
 117 \r
 118                 if(handled==false)\r
 119                 {\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
 122                         {\r
 123                                 info+="\t\t\t\t"+(*mi).first+"="+(*mi).second+"\r\n";\r
 124                         }\r
 125                         m_log->debug("FreenetMasterThread::HandleMessage received unhandled "+message.GetName()+" message.  Message content :\r\n"+info);\r
 126 \r
 127                         // if unhandled message was alldata - we must retrieve the data\r
 128                         if(message.GetName()=="AllData")\r
 129                         {\r
 130                                 long length;\r
 131                                 StringFunctions::Convert(message["DataLength"],length);\r
 132                                 m_fcp.WaitForBytes(1000,length);\r
 133 \r
 134                                 if(m_fcp.IsConnected() && length>0)\r
 135                                 {\r
 136                                         m_fcp.ReceiveIgnore(length);\r
 137                                 }\r
 138                         }\r
 139                 }\r
 140 \r
 141                 return handled;\r
 142 \r
 143         }\r
 144         else\r
 145         {\r
 146                 m_log->error("FreenetMasterThread::HandleMessage received "+message.GetName()+" message before NodeHello");\r
 147         }\r
 148 \r
 149         return false;\r
 150 }\r
 151 \r
 152 void FreenetMasterThread::RegisterFCPConnected(IFCPConnected *obj)\r
 153 {\r
 154         m_fcpconnected.push_back(obj);\r
 155 }\r
 156 \r
 157 void FreenetMasterThread::RegisterFCPMessageHandler(IFCPMessageHandler *obj)\r
 158 {\r
 159         m_fcpmessagehandlers.push_back(obj);\r
 160 }\r
 161 \r
 162 void FreenetMasterThread::RegisterPeriodicProcessor(IPeriodicProcessor *obj)\r
 163 {\r
 164         m_processors.push_back(obj);\r
 165 }\r
 166 \r
 167 void FreenetMasterThread::run()\r
 168 {\r
 169 \r
 170         Poco::DateTime lastreceivedmessage;\r
 171         Poco::DateTime lastconnected;\r
 172         Poco::DateTime now;\r
 173         FCPv2::Message message;\r
 174         bool done=false;\r
 175 \r
 176         lastconnected-=Poco::Timespan(0,0,1,0,0);\r
 177 \r
 178         m_log->debug("FreenetMasterThread::run thread started.");\r
 179 \r
 180         LoadDatabase();\r
 181 \r
 182 \r
 183 \r
 184         Setup();\r
 185 \r
 186         do\r
 187         {\r
 188                 try\r
 189                 {\r
 190                         if(m_fcp.IsConnected()==false)\r
 191                         {\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
 195                                 {\r
 196                                         if(FCPConnect()==false)\r
 197                                         {\r
 198 \r
 199                                                 m_log->error("FreenetMasterThread::run could not connect to node.  Waiting 60 seconds.");\r
 200 \r
 201                                                 for(int i=0; i<60 && !IsCancelled(); i++)\r
 202                                                 {\r
 203                                                         Poco::Thread::sleep(1000);\r
 204                                                 }\r
 205                                         }\r
 206                                         else\r
 207                                         {\r
 208                                                 lastreceivedmessage=Poco::Timestamp();\r
 209                                                 lastconnected=Poco::Timestamp();\r
 210                                         }\r
 211                                 }\r
 212                                 else\r
 213                                 {\r
 214                                         Poco::Thread::sleep(1000);\r
 215                                 }\r
 216                         }\r
 217                         // fcp is connected\r
 218                         else\r
 219                         {\r
 220                                 m_fcp.Update(1000);\r
 221 \r
 222                                 // check for message on receive buffer and handle it\r
 223                                 if(m_fcp.MessageReady()==true)\r
 224                                 {\r
 225                                         message.Clear();\r
 226                                         m_fcp.Receive(message);\r
 227 \r
 228                                         if(message.GetName()!="")\r
 229                                         {\r
 230                                                 HandleMessage(message);\r
 231                                                 lastreceivedmessage=Poco::Timestamp();\r
 232                                         }\r
 233                                 }\r
 234 \r
 235                                 // let objects do their processing\r
 236                                 for(std::vector<IPeriodicProcessor *>::iterator i=m_processors.begin(); i!=m_processors.end(); i++)\r
 237                                 {\r
 238                                         (*i)->Process();\r
 239                                 }\r
 240 \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
 244                                 {\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
 247                                 }\r
 248 \r
 249                                 if(m_fcp.IsConnected()==false)\r
 250                                 {\r
 251                                         m_log->information("FreenetMasterThread::Run Disconnected from Freenet node.");\r
 252                                 }\r
 253 \r
 254                         }\r
 255                 }\r
 256                 catch(Poco::Exception &e)\r
 257                 {\r
 258                         m_log->error("FreenetMasterThread::run caught exception : "+e.displayText());\r
 259                 }\r
 260                 catch(...)\r
 261                 {\r
 262                         m_log->error("FreenetMasterThread::run caught unknown exception");\r
 263                 }\r
 264         }while(!IsCancelled() && done==false);\r
 265 \r
 266         m_fcp.Disconnect();\r
 267 \r
 268         Shutdown();\r
 269 \r
 270         m_log->debug("FreenetMasterThread::run thread exiting.");\r
 271 \r
 272 }\r
 273 \r
 274 void FreenetMasterThread::Setup()\r
 275 {\r
 276 \r
 277         std::string temp="";\r
 278         Option option(m_db);\r
 279         if(option.Get("FCPHost",m_fcphost)==false)\r
 280         {\r
 281                 m_fcphost="localhost";\r
 282                 option.Set("FCPHost",m_fcphost);\r
 283         }\r
 284         if(option.GetInt("FCPPort",m_fcpport)==false)\r
 285         {\r
 286                 m_fcpport=9481;\r
 287                 option.Set("FCPPort",m_fcpport);\r
 288         }\r
 289         else\r
 290         {\r
 291                 if(m_fcpport<1 || m_fcpport>65535)\r
 292                 {\r
 293                         m_fcpport=9481;\r
 294                         option.Set("FCPPort",m_fcpport);\r
 295                 }\r
 296         }\r
 297 \r
 298         // seed random number generator\r
 299         srand(time(NULL));\r
 300 \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
 322         temp="";\r
 323         option.Get("DownloadFrostMessages",temp);\r
 324         if(temp=="true")\r
 325         {\r
 326                 m_registrables.push_back(new FrostMessageRequester(m_db,&m_fcp));\r
 327         }\r
 328 #endif\r
 329 \r
 330         for(std::vector<IFreenetRegistrable *>::iterator i=m_registrables.begin(); i!=m_registrables.end(); i++)\r
 331         {\r
 332                 (*i)->RegisterWithThread(this);\r
 333         }\r
 334 \r
 335 }\r
 336 \r
 337 void FreenetMasterThread::Shutdown()\r
 338 {\r
 339         // delete each registrable object\r
 340         for(std::vector<IFreenetRegistrable *>::iterator i=m_registrables.begin(); i!=m_registrables.end(); i++)\r
 341         {\r
 342                 delete (*i);\r
 343         }\r
 344 }\r