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