src/freenet/identityinserter.cpp

   1 #include "../../include/freenet/identityinserter.h"\r
   2 #include "../../include/freenet/identityxml.h"\r
   3 #include "../../include/stringfunctions.h"\r
   4 #include "../../include/option.h"\r
   5 \r
   6 #ifdef XMEM\r
   7         #include <xmem.h>\r
   8 #endif\r
   9 \r
  10 IdentityInserter::IdentityInserter()\r
  11 {\r
  12         Initialize();\r
  13 }\r
  14 \r
  15 IdentityInserter::IdentityInserter(FCPv2 *fcp):IFCPConnected(fcp)\r
  16 {\r
  17         Initialize();\r
  18 }\r
  19 \r
  20 void IdentityInserter::CheckForNeededInsert()\r
  21 {\r
  22         DateTime now;\r
  23         DateTime date;\r
  24         now.SetToGMTime();\r
  25         date.SetToGMTime();\r
  26         // set date to 1 hour back\r
  27         date.Add(0,0,-1);\r
  28 \r
  29         // Because of importance of Identity.xml, if we are now at the next day we immediately want to insert identities so change the date back to 12:00 AM so we find all identities not inserted yet today\r
  30         if(date.GetDay()!=now.GetDay())\r
  31         {\r
  32                 date=now;\r
  33                 date.SetHour(0);\r
  34                 date.SetMinute(0);\r
  35                 date.SetSecond(0);\r
  36         }\r
  37 \r
  38         SQLite3DB::Recordset rs=m_db->Query("SELECT LocalIdentityID FROM tblLocalIdentity WHERE PrivateKey IS NOT NULL AND PrivateKey <> '' AND InsertingIdentity='false' AND (LastInsertedIdentity<'"+date.Format("%Y-%m-%d %H:%M:%S")+"' OR LastInsertedIdentity IS NULL) ORDER BY LastInsertedIdentity;");\r
  39         \r
  40         if(rs.Empty()==false)\r
  41         {\r
  42                 StartInsert(rs.GetInt(0));\r
  43         }\r
  44 \r
  45 }\r
  46 \r
  47 void IdentityInserter::FCPConnected()\r
  48 {\r
  49         m_db->Execute("UPDATE tblLocalIdentity SET InsertingIdentity='false';");\r
  50 }\r
  51 \r
  52 \r
  53 void IdentityInserter::FCPDisconnected()\r
  54 {\r
  55         \r
  56 }\r
  57 \r
  58 const bool IdentityInserter::HandleMessage(FCPMessage &message)\r
  59 {\r
  60 \r
  61         if(message["Identifier"].find("IdentityInserter")==0)\r
  62         {\r
  63                 DateTime now;\r
  64                 std::vector<std::string> idparts;\r
  65 \r
  66                 now.SetToGMTime();\r
  67                 StringFunctions::Split(message["Identifier"],"|",idparts);\r
  68 \r
  69                 // no action for URIGenerated\r
  70                 if(message.GetName()=="URIGenerated")\r
  71                 {\r
  72                         return true;\r
  73                 }\r
  74 \r
  75                 // no action for IdentifierCollision\r
  76                 if(message.GetName()=="IdentifierCollision")\r
  77                 {\r
  78                         return true;\r
  79                 }\r
  80 \r
  81                 if(message.GetName()=="PutSuccessful")\r
  82                 {\r
  83                         m_db->Execute("UPDATE tblLocalIdentity SET InsertingIdentity='false', LastInsertedIdentity='"+now.Format("%Y-%m-%d %H:%M:%S")+"' WHERE LocalIdentityID="+idparts[1]+";");\r
  84                         m_db->Execute("INSERT INTO tblLocalIdentityInserts(LocalIdentityID,Day,InsertIndex) VALUES("+idparts[1]+",'"+idparts[4]+"',"+idparts[2]+");");\r
  85                         m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"IdentityInserter::HandleMessage inserted Identity xml");\r
  86                         return true;\r
  87                 }\r
  88 \r
  89                 if(message.GetName()=="PutFailed")\r
  90                 {\r
  91                         m_db->Execute("UPDATE tblLocalIdentity SET InsertingIdentity='false' WHERE LocalIdentityID="+idparts[1]+";");\r
  92                         m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"IdentityInserter::HandleMessage failure inserting Identity xml.  Code="+message["Code"]+" Description="+message["CodeDescription"]);\r
  93                         \r
  94                         // if code 9 (collision), then insert index into inserted table\r
  95                         if(message["Code"]=="9")\r
  96                         {\r
  97                                 m_db->Execute("INSERT INTO tblLocalIdentityInserts(LocalIdentityID,Day,InsertIndex) VALUES("+idparts[1]+",'"+idparts[4]+"',"+idparts[2]+");");\r
  98                         }\r
  99                         \r
 100                         return true;\r
 101                 }\r
 102 \r
 103         }\r
 104 \r
 105         return false;\r
 106 \r
 107 }\r
 108 \r
 109 void IdentityInserter::Initialize()\r
 110 {\r
 111         m_lastchecked.SetToGMTime();\r
 112 }\r
 113 \r
 114 void IdentityInserter::Process()\r
 115 {\r
 116         DateTime now;\r
 117         now.SetToGMTime();\r
 118 \r
 119         if(m_lastchecked<(now-(1.0/1440.0)))\r
 120         {\r
 121                 CheckForNeededInsert();\r
 122                 m_lastchecked=now;\r
 123         }\r
 124 \r
 125 }\r
 126 \r
 127 void IdentityInserter::RegisterWithThread(FreenetMasterThread *thread)\r
 128 {\r
 129         thread->RegisterFCPConnected(this);\r
 130         thread->RegisterFCPMessageHandler(this);\r
 131         thread->RegisterPeriodicProcessor(this);\r
 132 }\r
 133 \r
 134 void IdentityInserter::StartInsert(const long localidentityid)\r
 135 {\r
 136         DateTime date;\r
 137         std::string idstring;\r
 138 \r
 139         StringFunctions::Convert(localidentityid,idstring);\r
 140         date.SetToGMTime();\r
 141 \r
 142         SQLite3DB::Recordset rs=m_db->Query("SELECT Name,PrivateKey,SingleUse,PublishTrustList,PublishBoardList FROM tblLocalIdentity WHERE LocalIdentityID="+idstring+";");\r
 143 \r
 144         if(rs.Empty()==false)\r
 145         {\r
 146                 IdentityXML idxml;\r
 147                 FCPMessage mess;\r
 148                 DateTime now;\r
 149                 std::string messagebase;\r
 150                 std::string data;\r
 151                 std::string datasizestr;\r
 152                 std::string privatekey;\r
 153                 long index=0;\r
 154                 std::string indexstr;\r
 155                 std::string singleuse="false";\r
 156                 std::string publishtrustlist="false";\r
 157                 std::string publishboardlist="false";\r
 158 \r
 159                 now.SetToGMTime();\r
 160 \r
 161                 SQLite3DB::Recordset rs2=m_db->Query("SELECT MAX(InsertIndex) FROM tblLocalIdentityInserts WHERE LocalIdentityID="+idstring+" AND Day='"+now.Format("%Y-%m-%d")+"';");\r
 162                 if(rs2.Empty()==false)\r
 163                 {\r
 164                         if(rs2.GetField(0)==NULL)\r
 165                         {\r
 166                                 index=0;\r
 167                         }\r
 168                         else\r
 169                         {\r
 170                                 index=rs2.GetInt(0)+1;\r
 171                         }\r
 172                 }\r
 173                 StringFunctions::Convert(index,indexstr);\r
 174 \r
 175                 Option::instance()->Get("MessageBase",messagebase);\r
 176 \r
 177                 if(rs.GetField(0))\r
 178                 {\r
 179                         idxml.SetName(rs.GetField(0));\r
 180                 }\r
 181 \r
 182                 if(rs.GetField(1))\r
 183                 {\r
 184                         privatekey=rs.GetField(1);\r
 185                 }\r
 186 \r
 187                 if(rs.GetField(2))\r
 188                 {\r
 189                         singleuse=rs.GetField(2);\r
 190                 }\r
 191                 singleuse=="true" ? idxml.SetSingleUse(true) : idxml.SetSingleUse(false);\r
 192 \r
 193                 if(rs.GetField(3))\r
 194                 {\r
 195                         publishtrustlist=rs.GetField(3);\r
 196                 }\r
 197                 publishtrustlist=="true" ? idxml.SetPublishTrustList(true) : idxml.SetPublishTrustList(false);\r
 198 \r
 199                 if(rs.GetField(4))\r
 200                 {\r
 201                         publishboardlist=rs.GetField(4);\r
 202                 }\r
 203                 publishboardlist=="true" ? idxml.SetPublishBoardList(true) : idxml.SetPublishBoardList(false);\r
 204 \r
 205                 data=idxml.GetXML();\r
 206                 StringFunctions::Convert(data.size(),datasizestr);\r
 207 \r
 208                 mess.SetName("ClientPut");\r
 209                 mess["URI"]=privatekey+messagebase+"|"+now.Format("%Y-%m-%d")+"|Identity|"+indexstr+".xml";\r
 210                 mess["Identifier"]="IdentityInserter|"+idstring+"|"+indexstr+"|"+mess["URI"];\r
 211                 mess["UploadFrom"]="direct";\r
 212                 mess["DataLength"]=datasizestr;\r
 213                 m_fcp->SendMessage(mess);\r
 214                 m_fcp->SendRaw(data.c_str(),data.size());\r
 215 \r
 216                 m_db->Execute("UPDATE tblLocalIdentity SET InsertingIdentity='true' WHERE LocalIdentityID="+idstring+";");\r
 217 \r
 218         }\r
 219 }\r