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 #include "../../include/freenet/fileinserter.h"\r
  23 \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 lastconnected;\r
 175         DateTime now;\r
 176         FCPMessage message;\r
 177         bool done=false;\r
 178 \r
 179         lastconnected.SetToGMTime();\r
 180         lastconnected.Add(0,-1);\r
 181 \r
 182         m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"FreenetMasterThread::run thread started.");\r
 183 \r
 184         Setup();\r
 185 \r
 186         do\r
 187         {\r
 188                 if(m_fcp.Connected()==false)\r
 189                 {\r
 190                         // wait at least 1 minute since last successful connect\r
 191                         now.SetToGMTime();\r
 192                         if(lastconnected<=(now-(1.0/1440.0)))\r
 193                         {\r
 194                                 if(FCPConnect()==false)\r
 195                                 {\r
 196 \r
 197                                         m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"FreenetMasterThread::run could not connect to node.  Waiting 60 seconds.");\r
 198 \r
 199                                         for(int i=0; i<60 && !IsCancelled(); i++)\r
 200                                         {\r
 201                                                 Sleep(1000);\r
 202                                         }\r
 203                                 }\r
 204                                 else\r
 205                                 {\r
 206                                         lastreceivedmessage.SetToGMTime();\r
 207                                         lastconnected.SetToGMTime();\r
 208                                 }\r
 209                         }\r
 210                         else\r
 211                         {\r
 212                                 Sleep(1000);\r
 213                         }\r
 214                 }\r
 215                 // fcp is connected\r
 216                 else\r
 217                 {\r
 218                         m_fcp.Update(1);\r
 219 \r
 220                         // check for message on receive buffer and handle it\r
 221                         if(m_fcp.ReceiveBufferSize()>0)\r
 222                         {\r
 223                                 message.Reset();\r
 224                                 message=m_fcp.ReceiveMessage();\r
 225 \r
 226                                 if(message.GetName()!="")\r
 227                                 {\r
 228                                         HandleMessage(message);\r
 229                                         lastreceivedmessage.SetToGMTime();\r
 230                                 }\r
 231                         }\r
 232 \r
 233                         // let objects do their processing\r
 234                         for(std::vector<IPeriodicProcessor *>::iterator i=m_processors.begin(); i!=m_processors.end(); i++)\r
 235                         {\r
 236                                 (*i)->Process();\r
 237                         }\r
 238 \r
 239                         // if we haven't received any messages from the node in 10 minutes, something is wrong\r
 240                         now.SetToGMTime();\r
 241                         if(lastreceivedmessage<(now-(1.0/144.0)))\r
 242                         {\r
 243                                 m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"FreenetMasterThread::Run The Freenet node has not responded in 10 minutes.  Trying to reconnect.");\r
 244                                 m_fcp.Disconnect();\r
 245                         }\r
 246 \r
 247                         if(m_fcp.Connected()==false)\r
 248                         {\r
 249                                 m_log->WriteLog(LogFile::LOGLEVEL_INFO,"FreenetMasterThread::Run Disconnected from Freenet node.");\r
 250                         }\r
 251 \r
 252                 }\r
 253         }while(!IsCancelled() && done==false);\r
 254 \r
 255         m_fcp.Disconnect();\r
 256 \r
 257         Shutdown();\r
 258 \r
 259         m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"FreenetMasterThread::run thread exiting.");\r
 260 \r
 261 }\r
 262 \r
 263 void FreenetMasterThread::Setup()\r
 264 {\r
 265 \r
 266         // seed random number generator\r
 267         srand(time(NULL));\r
 268 \r
 269         m_registrables.push_back(new UnkeyedIDCreator(&m_fcp));\r
 270         m_registrables.push_back(new IdentityInserter(&m_fcp));\r
 271         m_registrables.push_back(new IdentityRequester(&m_fcp));\r
 272         m_registrables.push_back(new IntroductionPuzzleInserter(&m_fcp));\r
 273         m_registrables.push_back(new IdentityIntroductionRequester(&m_fcp));\r
 274         m_registrables.push_back(new IntroductionPuzzleRequester(&m_fcp));\r
 275         m_registrables.push_back(new IdentityIntroductionInserter(&m_fcp));\r
 276         m_registrables.push_back(new TrustListInserter(&m_fcp));\r
 277         m_registrables.push_back(new TrustListRequester(&m_fcp));\r
 278         m_registrables.push_back(new MessageListInserter(&m_fcp));\r
 279         m_registrables.push_back(new MessageListRequester(&m_fcp));\r
 280         m_registrables.push_back(new MessageRequester(&m_fcp));\r
 281         m_registrables.push_back(new MessageInserter(&m_fcp));\r
 282         m_registrables.push_back(new BoardListInserter(&m_fcp));\r
 283         m_registrables.push_back(new BoardListRequester(&m_fcp));\r
 284         m_registrables.push_back(new SiteInserter(&m_fcp));\r
 285         m_registrables.push_back(new FileInserter(&m_fcp));\r
 286         m_registrables.push_back(new PeriodicDBMaintenance());\r
 287 \r
 288         for(std::vector<IFreenetRegistrable *>::iterator i=m_registrables.begin(); i!=m_registrables.end(); i++)\r
 289         {\r
 290                 (*i)->RegisterWithThread(this);\r
 291         }\r
 292 \r
 293 }\r
 294 \r
 295 void FreenetMasterThread::Shutdown()\r
 296 {\r
 297         // delete each registrable object\r
 298         for(std::vector<IFreenetRegistrable *>::iterator i=m_registrables.begin(); i!=m_registrables.end(); i++)\r
 299         {\r
 300                 delete (*i);\r
 301         }\r
 302 }\r