version 0.3.9

[fms.git] / src / freenet / trustlistrequester.cpp

diff --git a/src/freenet/trustlistrequester.cpp b/src/freenet/trustlistrequester.cpp
index 668cdb9..d7dc27b 100644 (file)
--- a/src/freenet/trustlistrequester.cpp
+++ b/src/freenet/trustlistrequester.cpp
@@ -30,6 +30,8 @@ const bool TrustListRequester::HandleAllData(FCPMessage &message)
        TrustListXML xml;\r
        long identityid;\r
        long index;\r
+       int insertcount=0;\r
+       int dayinsertcount=0;\r
 \r
        StringFunctions::Split(message["Identifier"],"|",idparts);\r
        StringFunctions::Convert(message["DataLength"],datalength);\r
@@ -52,6 +54,22 @@ const bool TrustListRequester::HandleAllData(FCPMessage &message)
        data.resize(datalength);\r
        m_fcp->ReceiveRaw(&data[0],datalength);\r
 \r
+       // get count of identities added in last 24 hours\r
+       st=m_db->Prepare("SELECT COUNT(*) FROM tblIdentity WHERE DateAdded>=?;");\r
+       st.Bind(0,Poco::DateTimeFormatter::format(now,"%Y-%m-%d %H:%M:%S"));\r
+       st.Step();\r
+       if(st.RowReturned())\r
+       {\r
+               if(st.ResultNull(0)==false)\r
+               {\r
+                       st.ResultInt(0,dayinsertcount);\r
+               }\r
+       }\r
+       else\r
+       {\r
+               m_log->error("TrustListRequester::HandleAllData couldn't get count of identities added in last 24 hours");\r
+       }\r
+\r
        // parse file into xml and update the database\r
        if(xml.ParseXML(std::string(data.begin(),data.end()))==true)\r
        {\r
@@ -98,13 +116,17 @@ const bool TrustListRequester::HandleAllData(FCPMessage &message)
                        st.Step();\r
                        if(st.RowReturned()==false)\r
                        {\r
-                               idinsert.Bind(0,identity);\r
-                               idinsert.Bind(1,Poco::DateTimeFormatter::format(now,"%Y-%m-%d %H:%M:%S"));\r
-                               idinsert.Bind(2,"trust list of "+publisherid);\r
-                               idinsert.Step(true);\r
-                               id=idinsert.GetLastInsertRowID();\r
-                               idinsert.Reset();\r
-                       //      m_db->ExecuteInsert("INSERT INTO tblIdentity(PublicKey,DateAdded,AddedMethod) VALUES('"+identity+"','"+now.Format("%Y-%m-%d %H:%M:%S")+"');",(long &)id);\r
+                               if(insertcount<50 && dayinsertcount<100)\r
+                               {\r
+                                       idinsert.Bind(0,identity);\r
+                                       idinsert.Bind(1,Poco::DateTimeFormatter::format(now,"%Y-%m-%d %H:%M:%S"));\r
+                                       idinsert.Bind(2,"trust list of "+publisherid);\r
+                                       idinsert.Step(true);\r
+                                       id=idinsert.GetLastInsertRowID();\r
+                                       idinsert.Reset();\r
+                               }\r
+                               insertcount++;\r
+                               dayinsertcount++;\r
                        }\r
                        else\r
                        {\r
@@ -151,6 +173,15 @@ const bool TrustListRequester::HandleAllData(FCPMessage &message)
                trustst.Finalize();\r
                st.Finalize();\r
 \r
+               if(insertcount>=50)\r
+               {\r
+                       m_log->warning("TrustListRequester::HandleAllData TrustList contained more than 50 new identities : "+message["Identifier"]);\r
+               }\r
+               if(dayinsertcount>=100)\r
+               {\r
+                       m_log->warning("TrustListRequester::HandleAllData TrustList would have inserted more than 100 new identities in the last 24 hours : "+message["Identifier"]);\r
+               }\r
+\r
                st=m_db->Prepare("INSERT INTO tblTrustListRequests(IdentityID,Day,RequestIndex,Found) VALUES(?,?,?,'true');");\r
                st.Bind(0,identityid);\r
                st.Bind(1,idparts[4]);\r