monthly stats, subversion, better dump

bitcoin.cpp

@@ -81,7 +81,7 @@ class CNode {
   }
  
   void GotVersion() {
-    // printf("%s: version %i\n", ToString(you).c_str(), nVersion);
+    // printf("\n%s: version %i\n", ToString(you).c_str(), nVersion);
     BeginMessage("getaddr");
     EndMessage();
     doneAfter = time(NULL) + 10;
@@ -256,9 +256,13 @@ public:
   int GetClientVersion() {
     return nVersion;
   }
+  
+  std::string GetClientSubVersion() {
+    return strSubVer;
+  }
 };
 
-bool TestNode(const CIPPort &cip, int &ban, int &clientV, vector<CAddress>& vAddr) {
+bool TestNode(const CIPPort &cip, int &ban, int &clientV, std::string &clientSV, vector<CAddress>& vAddr) {
   try {
     CNode node(cip, vAddr);
     bool ret = node.Run();
@@ -268,6 +272,7 @@ bool TestNode(const CIPPort &cip, int &ban, int &clientV, vector<CAddress>& vAdd
       ban = 0;
     }
     clientV = node.GetClientVersion();
+    clientSV = node.GetClientSubVersion();
 //  printf("%s: %s!!!\n", cip.ToString().c_str(), ret ? "GOOD" : "BAD");
     return ret;
   } catch(std::ios_base::failure& e) {

bitcoin.h

@@ -3,6 +3,6 @@
 
 #include "protocol.h"
 
-bool TestNode(const CIPPort &cip, int &ban, int &client, std::vector<CAddress>& vAddr);
+bool TestNode(const CIPPort &cip, int &ban, int &client, std::string &clientSV, std::vector<CAddress>& vAddr);
 
 #endif

db.cpp

@@ -16,6 +16,7 @@ void CAddrInfo::Update(bool good) {
   stat8H.Update(good, age, 3600*8);
   stat1D.Update(good, age, 3600*24);
   stat1W.Update(good, age, 3600*24*7);
+  stat1M.Update(good, age, 3600*24*30);
   int ign = GetIgnoreTime();
   if (ign && (ignoreTill==0 || ignoreTill < ign+now)) ignoreTill = ign+now;
 //  printf("%s: got %s result: success=%i/%i; 2H:%.2f%%-%.2f%%(%.2f) 8H:%.2f%%-%.2f%%(%.2f) 1D:%.2f%%-%.2f%%(%.2f) 1W:%.2f%%-%.2f%%(%.2f) \n", ToString(ip).c_str(), good ? "good" : "bad", success, total, 
@@ -83,13 +84,14 @@ int CAddrDb::Lookup_(const CIPPort &ip) {
   return -1;
 }
 
-void CAddrDb::Good_(const CIPPort &addr, int clientV) {
+void CAddrDb::Good_(const CIPPort &addr, int clientV, std::string clientSV) {
   int id = Lookup_(addr);
   if (id == -1) return;
   unkId.erase(id);
   banned.erase(addr);
   CAddrInfo &info = idToInfo[id];
   info.clientVersion = clientV;
+  info.clientSubVersion = clientSV;
   info.Update(true);
   if (info.IsGood() && goodId.count(id)==0) {
     goodId.insert(id);

db.h

@@ -46,6 +46,8 @@ class CAddrReport {
 public:
   CIPPort ip;
   int clientVersion;
+  double uptime[5];
+  std::string clientSubVersion;
 };
 
 
@@ -60,9 +62,11 @@ private:
   CAddrStat stat8H;
   CAddrStat stat1D;
   CAddrStat stat1W;
+  CAddrStat stat1M;
   int clientVersion;
   int total;
   int success;
+  std::string clientSubVersion;
 public:
   CAddrInfo() : services(0), lastTry(0), ourLastTry(0), ignoreTill(0), clientVersion(0), total(0), success(0) {}
   
@@ -70,6 +74,12 @@ public:
     CAddrReport ret;
     ret.ip = ip;
     ret.clientVersion = clientVersion;
+    ret.clientSubVersion = clientSubVersion;
+    ret.uptime[0] = stat2H.reliability;
+    ret.uptime[1] = stat8H.reliability;
+    ret.uptime[2] = stat1D.reliability;
+    ret.uptime[3] = stat1W.reliability;
+    ret.uptime[4] = stat1M.reliability;
     return ret;
   }
   
@@ -109,7 +119,7 @@ public:
   friend class CAddrDb;
   
   IMPLEMENT_SERIALIZE (
-    unsigned char version = 0;
+    unsigned char version = 2;
     READWRITE(version);
     READWRITE(ip);
     READWRITE(services);
@@ -123,9 +133,16 @@ public:
       READWRITE(stat8H);
       READWRITE(stat1D);
       READWRITE(stat1W);
+      if (version >= 1)
+          READWRITE(stat1M);
+      else
+          if (!fWrite)
+              *((CAddrStat*)(&stat1M)) = stat1W;
       READWRITE(total);
       READWRITE(success);
       READWRITE(clientVersion);
+      if (version >= 2)
+          READWRITE(clientSubVersion);
     }
   )
 };
@@ -157,20 +174,20 @@ private:
   std::deque<int> ourId; // sequence of tried nodes, in order we have tried connecting to them (c,d)
   std::set<int> unkId; // set of nodes not yet tried (b)
   std::set<int> goodId; // set of good nodes  (d, good e)
-  std::map<CIPPort, time_t> banned; // nodes that are banned, with their unban time (a)
   int nDirty;
   
 protected:
   // internal routines that assume proper locks are acquired
   void Add_(const CAddress &addr, bool force);   // add an address
   bool Get_(CIPPort &ip, int& wait);      // get an IP to test (must call Good_, Bad_, or Skipped_ on result afterwards)
-  void Good_(const CIPPort &ip, int clientV); // mark an IP as good (must have been returned by Get_)
+  void Good_(const CIPPort &ip, int clientV, std::string clientSV); // mark an IP as good (must have been returned by Get_)
   void Bad_(const CIPPort &ip, int ban);  // mark an IP as bad (and optionally ban it) (must have been returned by Get_)
   void Skipped_(const CIPPort &ip);       // mark an IP as skipped (must have been returned by Get_)
   int Lookup_(const CIPPort &ip);         // look up id of an IP
   void GetIPs_(std::set<CIP>& ips, int max, bool fOnlyIPv4); // get a random set of IPs (shared lock only)
 
 public:
+  std::map<CIPPort, time_t> banned; // nodes that are banned, with their unban time (a)
 
   void GetStats(CAddrDbStats &stats) {
     SHARED_CRITICAL_BLOCK(cs) {
@@ -255,9 +272,9 @@ public:
       for (int i=0; i<vAddr.size(); i++)
         Add_(vAddr[i], fForce);
   }
-  void Good(const CIPPort &addr, int clientVersion) {
+  void Good(const CIPPort &addr, int clientVersion, std::string clientSubVersion) {
     CRITICAL_BLOCK(cs)
-      Good_(addr, clientVersion);
+      Good_(addr, clientVersion, clientSubVersion);
   }
   void Skipped(const CIPPort &addr) {
     CRITICAL_BLOCK(cs)

main.cpp

@@ -1,3 +1,5 @@
+#include <algorithm>
+
 #include <pthread.h>
 #include <stdio.h>
 #include <stdlib.h>
@@ -105,10 +107,11 @@ extern "C" void* ThreadCrawler(void* data) {
     int ban = 0;
     vector<CAddress> addr;
     int clientV = 0;
-    bool ret = TestNode(ip,ban,clientV,addr);
+    std::string clientSV;
+    bool ret = TestNode(ip,ban,clientV,clientSV,addr);
     db.Add(addr);
     if (ret) {
-      db.Good(ip, clientV);
+      db.Good(ip, clientV, clientSV);
     } else {
       db.Bad(ip, ban);
     }
@@ -150,6 +153,18 @@ extern "C" void* ThreadDNS(void* arg) {
   dnsserver(&dns_opt);
 }
 
+int StatCompare(const CAddrReport& a, const CAddrReport& b) {
+  if (a.uptime[4] == b.uptime[4]) {
+    if (a.uptime[3] == b.uptime[3]) {
+      return a.clientVersion > b.clientVersion;
+    } else {
+      return a.uptime[3] > b.uptime[3];
+    }
+  } else {
+    return a.uptime[4] > b.uptime[4];
+  }
+}
+
 extern "C" void* ThreadDumper(void*) {
   do {
     Sleep(100000);
@@ -159,16 +174,28 @@ extern "C" void* ThreadDumper(void*) {
         CAutoFile cf(f);
         cf << db;
       }
+      FILE *d = fopen("dnsseed.dump", "w");
+      vector<CAddrReport> v = db.GetAll();
+      sort(v.begin(), v.end(), StatCompare);
+      for (vector<CAddrReport>::const_iterator it = v.begin(); it < v.end(); it++) {
+        CAddrReport rep = *it;
+        fprintf(d, "%s\t%.2f%%\t%.2f%%\t%.2f%%\t%.2f%%\t%.2f%%\t%i \"%s\"\n", rep.ip.ToString().c_str(), 100.0*rep.uptime[0], 100.0*rep.uptime[1], 100.0*rep.uptime[2], 100.0*rep.uptime[3], 100.0*rep.uptime[4], rep.clientVersion, rep.clientSubVersion.c_str());
+      }
+      fclose(d);
     }
   } while(1);
 }
 
 extern "C" void* ThreadStats(void*) {
   do {
+    char c[256];
+    time_t tim = time(NULL);
+    struct tm *tmp = localtime(&tim);
+    strftime(c, 256, "[%y-%m-%d %H:%M:%S]", tmp);
     CAddrDbStats stats;
     db.GetStats(stats);
     printf("\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b");
-    printf("*** %i/%i available (%i tried in %is, %i new, %i active), %i banned; %llu DNS requests", stats.nGood, stats.nAvail, stats.nTracked, stats.nAge, stats.nNew, stats.nAvail - stats.nTracked - stats.nNew, stats.nBanned, (unsigned long long)dns_opt.nRequests);
+    printf("%s %i/%i available (%i tried in %is, %i new, %i active), %i banned; %llu DNS requests", c, stats.nGood, stats.nAvail, stats.nTracked, stats.nAge, stats.nNew, stats.nAvail - stats.nTracked - stats.nNew, stats.nBanned, (unsigned long long)dns_opt.nRequests);
     Sleep(1000);
   } while(1);
 }
@@ -205,13 +232,7 @@ int main(int argc, char **argv) {
     printf("Loading dnsseed.dat...");
     CAutoFile cf(f);
     cf >> db;
-//    FILE *d = fopen("dnsseed.dump", "w");
-//    vector<CAddrReport> v = db.GetAll();
-//    for (vector<CAddrReport>::const_iterator it = v.begin(); it < v.end(); it++) {
-//      CAddrReport rep = *it;
-//      fprintf(d, "%s %i\n", rep.ip.ToString().c_str(), rep.clientVersion);
-//    }
-//     fclose(d);
+//    db.banned.clear();
     printf("done\n");
   }
   pthread_t threadDns, threadSeed, threadDump, threadStats;