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