src/freenet/messageinserter.cpp

   1 #include "../../include/freenet/messageinserter.h"\r
   2 #include "../../include/freenet/messagexml.h"\r
   3 \r
   4 MessageInserter::MessageInserter()\r
   5 {\r
   6         Initialize();\r
   7 }\r
   8 \r
   9 MessageInserter::MessageInserter(FCPv2 *fcp):IIndexInserter<std::string>(fcp)\r
  10 {\r
  11         Initialize();\r
  12 }\r
  13 \r
  14 void MessageInserter::CheckForNeededInsert()\r
  15 {\r
  16         DateTime now;\r
  17         now.SetToGMTime();\r
  18         bool didinsert=false;\r
  19         // only do 1 insert at a time\r
  20         if(m_inserting.size()==0)\r
  21         {\r
  22                 SQLite3DB::Statement st=m_db->Prepare("SELECT MessageUUID FROM tblMessageInserts INNER JOIN tblLocalIdentity ON tblMessageInserts.LocalIdentityID=tblLocalIdentity.LocalIdentityID WHERE tblLocalIdentity.PrivateKey IS NOT NULL AND tblLocalIdentity.PrivateKey <> '' AND tblMessageInserts.Inserted='false' AND tblMessageInserts.SendDate<=?;");\r
  23                 st.Bind(0,now.Format("%Y-%m-%d %H:%M:%S"));\r
  24                 st.Step();\r
  25 \r
  26                 while(st.RowReturned() && m_inserting.size()==0)\r
  27                 {\r
  28                         std::string messageuuid="";\r
  29                         st.ResultText(0,messageuuid);\r
  30 \r
  31                         // make sure there are no uninserted files attached to this message\r
  32                         SQLite3DB::Statement st2=m_db->Prepare("SELECT FileInsertID FROM tblFileInserts WHERE Key IS NULL AND MessageUUID=?;");\r
  33                         st2.Bind(0,messageuuid);\r
  34                         st2.Step();\r
  35 \r
  36                         if(st2.RowReturned()==false)\r
  37                         {\r
  38                                 StartInsert(messageuuid);\r
  39                         }\r
  40 \r
  41                         st.Step();\r
  42                 }\r
  43         }\r
  44 }\r
  45 \r
  46 const bool MessageInserter::HandlePutFailed(FCPMessage &message)\r
  47 {\r
  48         int index;\r
  49         int localidentityid;\r
  50         std::vector<std::string> idparts;\r
  51         StringFunctions::Split(message["Identifier"],"|",idparts);\r
  52         StringFunctions::Convert(idparts[2],localidentityid);\r
  53         StringFunctions::Convert(idparts[3],index);\r
  54 \r
  55         // fatal put - or data exists - insert bogus index into database so we'll try to insert this message again\r
  56         if(message["Fatal"]=="true" || message["Code"]=="9")\r
  57         {\r
  58                 SQLite3DB::Statement st=m_db->Prepare("INSERT INTO tblMessageInserts(LocalIdentityID,Day,InsertIndex,Inserted) VALUES(?,?,?,'true');");\r
  59                 st.Bind(0,localidentityid);\r
  60                 st.Bind(1,idparts[6]);\r
  61                 st.Bind(2,index);\r
  62                 st.Step();\r
  63         }\r
  64 \r
  65         RemoveFromInsertList(idparts[1]);\r
  66 \r
  67         return true;\r
  68 }\r
  69 \r
  70 const bool MessageInserter::HandlePutSuccessful(FCPMessage &message)\r
  71 {\r
  72         MessageXML xml;\r
  73         DateTime date;\r
  74         int localidentityid;\r
  75         int index;\r
  76         std::vector<std::string> idparts;\r
  77 \r
  78         StringFunctions::Split(message["Identifier"],"|",idparts);\r
  79         StringFunctions::Convert(idparts[3],index);\r
  80         StringFunctions::Convert(idparts[2],localidentityid);\r
  81 \r
  82         SQLite3DB::Statement st=m_db->Prepare("UPDATE tblMessageInserts SET Day=?, InsertIndex=?, Inserted='true' WHERE MessageUUID=?;");\r
  83         st.Bind(0,idparts[6]);\r
  84         st.Bind(1,index);\r
  85         st.Bind(2,idparts[1]);\r
  86         st.Step();\r
  87 \r
  88         // insert record into temp table so MessageList will be inserted ASAP\r
  89         date.SetToGMTime();\r
  90         st=m_db->Prepare("INSERT INTO tmpMessageListInsert(LocalIdentityID,Date) VALUES(?,?);");\r
  91         st.Bind(0,localidentityid);\r
  92         st.Bind(1,date.Format("%Y-%m-%d"));\r
  93         st.Step();\r
  94 \r
  95         // update the messageuuid to the real messageuuid\r
  96         st=m_db->Prepare("SELECT MessageXML FROM tblMessageInserts WHERE MessageUUID=?;");\r
  97         st.Bind(0,idparts[1]);\r
  98         st.Step();\r
  99         if(st.RowReturned())\r
 100         {\r
 101                 std::string xmldata="";\r
 102                 st.ResultText(0,xmldata);\r
 103                 xml.ParseXML(xmldata);\r
 104                 xml.SetMessageID(idparts[4]);\r
 105 \r
 106                 SQLite3DB::Statement st2=m_db->Prepare("UPDATE tblMessageInserts SET MessageUUID=?, MessageXML=? WHERE MessageUUID=?;");\r
 107                 st2.Bind(0,idparts[4]);\r
 108                 st2.Bind(1,xml.GetXML());\r
 109                 st2.Bind(2,idparts[1]);\r
 110                 st2.Step();\r
 111 \r
 112                 //update file insert MessageUUID as well\r
 113                 st2=m_db->Prepare("UPDATE tblFileInserts SET MessageUUID=? WHERE MessageUUID=?;");\r
 114                 st2.Bind(0,idparts[4]);\r
 115                 st2.Bind(1,idparts[1]);\r
 116                 st2.Step();\r
 117         }\r
 118 \r
 119         RemoveFromInsertList(idparts[1]);\r
 120 \r
 121         m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"MessageInserter::HandlePutSuccessful successfully inserted message "+message["Identifier"]);\r
 122 \r
 123         return true;\r
 124 }\r
 125 \r
 126 void MessageInserter::Initialize()\r
 127 {\r
 128         m_fcpuniquename="MessageInserter";\r
 129 }\r
 130 \r
 131 const bool MessageInserter::StartInsert(const std::string &messageuuid)\r
 132 {\r
 133         MessageXML xmlfile;\r
 134         DateTime now;\r
 135         now.SetToGMTime();\r
 136         SQLite3DB::Statement st=m_db->Prepare("SELECT MessageXML,PrivateKey,tblLocalIdentity.LocalIdentityID,PublicKey FROM tblMessageInserts INNER JOIN tblLocalIdentity ON tblMessageInserts.LocalIdentityID=tblLocalIdentity.LocalIdentityID WHERE MessageUUID=?;");\r
 137         st.Bind(0,messageuuid);\r
 138         st.Step();\r
 139 \r
 140         if(st.RowReturned())\r
 141         {\r
 142                 int localidentityid;\r
 143                 std::string idstr;\r
 144                 std::string xml;\r
 145                 std::string xmlsizestr;\r
 146                 std::string privatekey;\r
 147                 std::string publickey;\r
 148                 FCPMessage message;\r
 149                 std::string indexstr;\r
 150                 int index=0;\r
 151                 \r
 152                 st.ResultText(0,xml);\r
 153                 st.ResultText(1,privatekey);\r
 154                 st.ResultInt(2,localidentityid);\r
 155                 st.ResultText(3,publickey);\r
 156                 StringFunctions::Convert(localidentityid,idstr);\r
 157 \r
 158                 st=m_db->Prepare("SELECT MAX(InsertIndex) FROM tblMessageInserts WHERE Day=? AND LocalIdentityID=?;");\r
 159                 st.Bind(0,now.Format("%Y-%m-%d"));\r
 160                 st.Bind(1,localidentityid);\r
 161                 st.Step();\r
 162 \r
 163                 if(st.ResultNull(0)==false)\r
 164                 {\r
 165                         st.ResultInt(0,index);\r
 166                         index++;\r
 167                 }\r
 168                 StringFunctions::Convert(index,indexstr);\r
 169 \r
 170                 xmlfile.ParseXML(xml);\r
 171 \r
 172                 // add file attachments to xml - must do this before we change UUID\r
 173                 st=m_db->Prepare("SELECT Key, Size FROM tblFileInserts WHERE MessageUUID=?;");\r
 174                 st.Bind(0,xmlfile.GetMessageID());\r
 175                 st.Step();\r
 176                 while(st.RowReturned())\r
 177                 {\r
 178                         std::string key="";\r
 179                         int size;\r
 180                         \r
 181                         st.ResultText(0,key);\r
 182                         st.ResultInt(1,size);\r
 183 \r
 184                         xmlfile.AddFileAttachment(key,size);\r
 185 \r
 186                         st.Step();\r
 187                 }\r
 188 \r
 189                 // recreate messageuuid in xml - UUID of message will not match entry in MessageInserts table until we successfully insert it\r
 190                 // see HandlePutSuccessful\r
 191                 // if we don't already have an @sskpart - add it\r
 192                 if(xmlfile.GetMessageID().find("@")==std::string::npos)\r
 193                 {\r
 194                         // remove - and ~ from publickey part\r
 195                         std::string publickeypart=StringFunctions::Replace(StringFunctions::Replace(publickey.substr(4,43),"-",""),"~","");\r
 196                         xmlfile.SetMessageID(xmlfile.GetMessageID()+"@"+publickeypart);\r
 197                 }\r
 198                 xml=xmlfile.GetXML();\r
 199 \r
 200                 StringFunctions::Convert(xml.size(),xmlsizestr);\r
 201 \r
 202                 message.SetName("ClientPut");\r
 203                 message["URI"]=privatekey+m_messagebase+"|"+now.Format("%Y-%m-%d")+"|Message|"+indexstr+".xml";\r
 204                 message["Identifier"]=m_fcpuniquename+"|"+messageuuid+"|"+idstr+"|"+indexstr+"|"+xmlfile.GetMessageID()+"|"+message["URI"];\r
 205                 message["UploadFrom"]="direct";\r
 206                 message["DataLength"]=xmlsizestr;\r
 207                 m_fcp->SendMessage(message);\r
 208                 m_fcp->SendRaw(xml.c_str(),xml.size());\r
 209 \r
 210                 m_inserting.push_back(messageuuid);\r
 211 \r
 212                 m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"MessageInserter::StartInsert started message insert "+message["URI"]);\r
 213         \r
 214                 return true;\r
 215         }\r
 216         else\r
 217         {\r
 218                 return false;\r
 219         }\r
 220 \r
 221 }\r