src/freenet/freenetmasterthread.cpp

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