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/introductionpuzzleinserter.h"\r
   8 #include "../../include/freenet/identityintroductionrequester.h"\r
   9 #include "../../include/freenet/introductionpuzzlerequester.h"\r
  10 #include "../../include/freenet/identityintroductioninserter.h"\r
  11 #include "../../include/freenet/trustlistinserter.h"\r
  12 #include "../../include/freenet/trustlistrequester.h"\r
  13 #include "../../include/freenet/messagelistrequester.h"\r
  14 #include "../../include/freenet/messagelistinserter.h"\r
  15 #include "../../include/freenet/messagerequester.h"\r
  16 #include "../../include/freenet/messageinserter.h"\r
  17 #include "../../include/freenet/boardlistinserter.h"\r
  18 #include "../../include/freenet/boardlistrequester.h"\r
  19 #include "../../include/freenet/siteinserter.h"\r
  20 #include "../../include/freenet/fileinserter.h"\r
  21 #include "../../include/freenet/fmsversionrequester.h"\r
  22 \r
  23 #include <Poco/UUID.h>\r
  24 #include <Poco/UUIDGenerator.h>\r
  25 #include <Poco/DateTime.h>\r
  26 #include <Poco/Timespan.h>\r
  27 #include <Poco/Thread.h>\r
  28 \r
  29 #include <cstdlib>\r
  30 \r
  31 #ifdef XMEM\r
  32         #include <xmem.h>\r
  33 #endif\r
  34 \r
  35 FreenetMasterThread::FreenetMasterThread()\r
  36 {\r
  37         std::string fcpport;\r
  38 \r
  39         if(Option::Instance()->Get("FCPHost",m_fcphost)==false)\r
  40         {\r
  41                 m_fcphost="localhost";\r
  42                 Option::Instance()->Set("FCPHost",m_fcphost);\r
  43         }\r
  44         if(Option::Instance()->Get("FCPPort",fcpport)==false)\r
  45         {\r
  46                 fcpport="9481";\r
  47                 Option::Instance()->Set("FCPPort",fcpport);\r
  48         }\r
  49 \r
  50         // convert fcp port to long, and make sure it's within the valid port range\r
  51         if(StringFunctions::Convert(fcpport,m_fcpport)==false)\r
  52         {\r
  53                 m_fcpport=9481;\r
  54                 Option::Instance()->Set("FCPPort","9481");\r
  55         }\r
  56 \r
  57         m_receivednodehello=false;\r
  58 \r
  59 }\r
  60 \r
  61 FreenetMasterThread::~FreenetMasterThread()\r
  62 {\r
  63 \r
  64 }\r
  65 \r
  66 const bool FreenetMasterThread::FCPConnect()\r
  67 {\r
  68         // we were previosly connected, send FCPDisconnect to objects\r
  69         if(m_receivednodehello==true)\r
  70         {\r
  71                 for(std::vector<IFCPConnected *>::iterator i=m_fcpconnected.begin(); i!=m_fcpconnected.end(); i++)\r
  72                 {\r
  73                         (*i)->FCPDisconnected();\r
  74                 }\r
  75                 m_receivednodehello=false;\r
  76         }\r
  77 \r
  78         m_log->information("FreenetMasterThread::FCPConnect trying to connect to node "+m_fcphost);\r
  79 \r
  80         if(m_fcp.Connect(m_fcphost.c_str(),m_fcpport)==true)\r
  81         {\r
  82                 Poco::UUIDGenerator uuidgen;\r
  83                 Poco::UUID uuid;\r
  84 \r
  85                 try\r
  86                 {\r
  87                         uuid=uuidgen.createRandom();\r
  88                 }\r
  89                 catch(...)\r
  90                 {\r
  91                         m_log->fatal("FreenetMasterThread::FCPConnect could not generate UUID");\r
  92                 }\r
  93 \r
  94                 std::string clientname="FMSClient-"+uuid.toString();\r
  95                 // send ClientHello message to node\r
  96                 m_fcp.SendMessage("ClientHello",2,"Name",clientname.c_str(),"ExpectedVersion","2.0");\r
  97 \r
  98                 m_log->information("FreenetMasterThread::FCPConnect connected to node");\r
  99 \r
 100                 return true;\r
 101         }\r
 102         else\r
 103         {\r
 104                 return false;\r
 105         }\r
 106 \r
 107 }\r
 108 \r
 109 const bool FreenetMasterThread::HandleMessage(FCPMessage &message)\r
 110 {\r
 111         if(message.GetName()=="NodeHello")\r
 112         {\r
 113                 m_receivednodehello=true;\r
 114 \r
 115                 // send connected message to all objects, must do this AFTER we received the NodeHello message\r
 116                 for(std::vector<IFCPConnected *>::iterator i=m_fcpconnected.begin(); i!=m_fcpconnected.end(); i++)\r
 117                 {\r
 118                         (*i)->FCPConnected();\r
 119                 }\r
 120 \r
 121                 return true;\r
 122         }\r
 123         if(m_receivednodehello==true)\r
 124         {\r
 125                 bool handled=false;\r
 126                 std::vector<IFCPMessageHandler *>::iterator i=m_fcpmessagehandlers.begin();\r
 127                 while(handled==false && i!=m_fcpmessagehandlers.end())\r
 128                 {\r
 129                         handled=(*i)->HandleMessage(message);\r
 130                         i++;\r
 131                 }\r
 132 \r
 133                 if(handled==false)\r
 134                 {\r
 135                         std::string info("");\r
 136                         for(std::map<std::string,std::string>::iterator mi=message.begin(); mi!=message.end(); mi++)\r
 137                         {\r
 138                                 info+="\t\t\t\t"+(*mi).first+"="+(*mi).second+"\r\n";\r
 139                         }\r
 140                         m_log->debug("FreenetMasterThread::HandleMessage received unhandled "+message.GetName()+" message.  Message content :\r\n"+info);\r
 141 \r
 142                         // if unhandled message was alldata - we must retrieve the data\r
 143                         if(message.GetName()=="AllData")\r
 144                         {\r
 145                                 long length;\r
 146                                 StringFunctions::Convert(message["DataLength"],length);\r
 147                                 while(m_fcp.Connected() && m_fcp.ReceiveBufferSize()<length)\r
 148                                 {\r
 149                                         m_fcp.Update(1);\r
 150                                 }\r
 151                                 if(m_fcp.Connected())\r
 152                                 {\r
 153                                         char *data=new char[length];\r
 154                                         m_fcp.ReceiveRaw(data,length);\r
 155                                         delete [] data;\r
 156                                 }\r
 157                         }\r
 158                 }\r
 159 \r
 160                 return handled;\r
 161 \r
 162         }\r
 163         else\r
 164         {\r
 165                 m_log->error("FreenetMasterThread::HandleMessage received "+message.GetName()+" message before NodeHello");\r
 166         }\r
 167 \r
 168         return false;\r
 169 }\r
 170 \r
 171 void FreenetMasterThread::RegisterFCPConnected(IFCPConnected *obj)\r
 172 {\r
 173         m_fcpconnected.push_back(obj);\r
 174 }\r
 175 \r
 176 void FreenetMasterThread::RegisterFCPMessageHandler(IFCPMessageHandler *obj)\r
 177 {\r
 178         m_fcpmessagehandlers.push_back(obj);\r
 179 }\r
 180 \r
 181 void FreenetMasterThread::RegisterPeriodicProcessor(IPeriodicProcessor *obj)\r
 182 {\r
 183         m_processors.push_back(obj);\r
 184 }\r
 185 \r
 186 void FreenetMasterThread::run()\r
 187 {\r
 188 \r
 189         Poco::DateTime lastreceivedmessage;\r
 190         Poco::DateTime lastconnected;\r
 191         Poco::DateTime now;\r
 192         FCPMessage message;\r
 193         bool done=false;\r
 194 \r
 195         lastconnected-=Poco::Timespan(0,0,1,0,0);\r
 196 \r
 197         m_log->debug("FreenetMasterThread::run thread started.");\r
 198 \r
 199         Setup();\r
 200 \r
 201         do\r
 202         {\r
 203                 if(m_fcp.Connected()==false)\r
 204                 {\r
 205                         // wait at least 1 minute since last successful connect\r
 206                         now=Poco::Timestamp();\r
 207                         if(lastconnected<=(now-Poco::Timespan(0,0,1,0,0)))\r
 208                         {\r
 209                                 if(FCPConnect()==false)\r
 210                                 {\r
 211 \r
 212                                         m_log->error("FreenetMasterThread::run could not connect to node.  Waiting 60 seconds.");\r
 213 \r
 214                                         for(int i=0; i<60 && !IsCancelled(); i++)\r
 215                                         {\r
 216                                                 Poco::Thread::sleep(1000);\r
 217                                         }\r
 218                                 }\r
 219                                 else\r
 220                                 {\r
 221                                         lastreceivedmessage=Poco::Timestamp();\r
 222                                         lastconnected=Poco::Timestamp();\r
 223                                 }\r
 224                         }\r
 225                         else\r
 226                         {\r
 227                                 Poco::Thread::sleep(1000);\r
 228                         }\r
 229                 }\r
 230                 // fcp is connected\r
 231                 else\r
 232                 {\r
 233                         m_fcp.Update(1);\r
 234 \r
 235                         // check for message on receive buffer and handle it\r
 236                         if(m_fcp.ReceiveBufferSize()>0)\r
 237                         {\r
 238                                 message.Reset();\r
 239                                 message=m_fcp.ReceiveMessage();\r
 240 \r
 241                                 if(message.GetName()!="")\r
 242                                 {\r
 243                                         HandleMessage(message);\r
 244                                         lastreceivedmessage=Poco::Timestamp();\r
 245                                 }\r
 246                         }\r
 247 \r
 248                         // let objects do their processing\r
 249                         for(std::vector<IPeriodicProcessor *>::iterator i=m_processors.begin(); i!=m_processors.end(); i++)\r
 250                         {\r
 251                                 (*i)->Process();\r
 252                         }\r
 253 \r
 254                         // if we haven't received any messages from the node in 10 minutes, something is wrong\r
 255                         now=Poco::Timestamp();\r
 256                         if(lastreceivedmessage<(now-Poco::Timespan(0,0,10,0,0)))\r
 257                         {\r
 258                                 m_log->error("FreenetMasterThread::Run The Freenet node has not responded in 10 minutes.  Trying to reconnect.");\r
 259                                 m_fcp.Disconnect();\r
 260                         }\r
 261 \r
 262                         if(m_fcp.Connected()==false)\r
 263                         {\r
 264                                 m_log->information("FreenetMasterThread::Run Disconnected from Freenet node.");\r
 265                         }\r
 266 \r
 267                 }\r
 268         }while(!IsCancelled() && done==false);\r
 269 \r
 270         m_fcp.Disconnect();\r
 271 \r
 272         Shutdown();\r
 273 \r
 274         m_log->debug("FreenetMasterThread::run thread exiting.");\r
 275 \r
 276 }\r
 277 \r
 278 void FreenetMasterThread::Setup()\r
 279 {\r
 280 \r
 281         // seed random number generator\r
 282         srand(time(NULL));\r
 283 \r
 284         m_registrables.push_back(new UnkeyedIDCreator(&m_fcp));\r
 285         m_registrables.push_back(new IdentityInserter(&m_fcp));\r
 286         m_registrables.push_back(new IdentityRequester(&m_fcp));\r
 287         m_registrables.push_back(new IntroductionPuzzleInserter(&m_fcp));\r
 288         m_registrables.push_back(new IdentityIntroductionRequester(&m_fcp));\r
 289         m_registrables.push_back(new IntroductionPuzzleRequester(&m_fcp));\r
 290         m_registrables.push_back(new IdentityIntroductionInserter(&m_fcp));\r
 291         m_registrables.push_back(new TrustListInserter(&m_fcp));\r
 292         m_registrables.push_back(new TrustListRequester(&m_fcp));\r
 293         m_registrables.push_back(new MessageListInserter(&m_fcp));\r
 294         m_registrables.push_back(new MessageListRequester(&m_fcp));\r
 295         m_registrables.push_back(new MessageInserter(&m_fcp));\r
 296         m_registrables.push_back(new MessageRequester(&m_fcp));\r
 297         m_registrables.push_back(new BoardListInserter(&m_fcp));\r
 298         m_registrables.push_back(new BoardListRequester(&m_fcp));\r
 299         m_registrables.push_back(new SiteInserter(&m_fcp));\r
 300         m_registrables.push_back(new FileInserter(&m_fcp));\r
 301         m_registrables.push_back(new FMSVersionRequester(&m_fcp));\r
 302 \r
 303         for(std::vector<IFreenetRegistrable *>::iterator i=m_registrables.begin(); i!=m_registrables.end(); i++)\r
 304         {\r
 305                 (*i)->RegisterWithThread(this);\r
 306         }\r
 307 \r
 308 }\r
 309 \r
 310 void FreenetMasterThread::Shutdown()\r
 311 {\r
 312         // delete each registrable object\r
 313         for(std::vector<IFreenetRegistrable *>::iterator i=m_registrables.begin(); i!=m_registrables.end(); i++)\r
 314         {\r
 315                 delete (*i);\r
 316         }\r
 317 }\r