Changeset 1172 for trunk/libtorrent/src/protocol/peer_connection_base.cc

Timestamp:

08/25/10 03:14:19 (21 months ago)

Author:

rakshasa

Message:

* Some debugging tools added.

Files:

• trunk/libtorrent/src/protocol/peer_connection_base.cc (modified) (4 diffs)

trunk/libtorrent/src/protocol/peer_connection_base.cc

@@ -38 +38 @@
 
 #include <cstdio>
+#include <fcntl.h>
 #include <rak/error_number.h>
+#include <rak/string_manip.h>
 
 #include "torrent/exceptions.h"
@@ -277 +279 @@
 }
 
+inline static void
+log_upload_chunk_mincore(Chunk* chunk, const Piece& piece, bool new_index) {
+#ifdef LT_LOG_MINCORE_FILE
+  static int mincore_fd = -1;
+  static int32_t ticker = rak::timer::current().seconds() / 10 * 10;
+
+  static int counter_incore = 0;
+  static int counter_not_incore = 0;
+  static int counter_incore_new = 0;
+  static int counter_not_incore_new = 0;
+
+  if (rak::timer::current().seconds() >= ticker + 10) {
+    char buffer[256];
+
+    if (mincore_fd == -1) {
+      snprintf(buffer, 256, "%s.%u", LT_LOG_MINCORE_FILE, getpid());
+    
+      if ((mincore_fd = open(buffer, O_WRONLY | O_CREAT | O_TRUNC)) == -1)
+        throw internal_error("Could not open mincore log file.");
+    }
+
+    // Log the result of mincore for every piece uploaded to a file.
+    unsigned int buf_lenght = snprintf(buffer, 256, "%i %u %u %u %u\n",
+                                       ticker, counter_incore, counter_incore_new, counter_not_incore, counter_not_incore_new);
+
+    write(mincore_fd, buffer, buf_lenght);
+    
+    ticker = rak::timer::current().seconds() / 10 * 10;
+
+    counter_incore = 0;
+    counter_not_incore = 0;
+    counter_incore_new = 0;
+    counter_not_incore_new = 0;
+  }
+
+  bool is_incore = (chunk->incore_length(piece.offset(), piece.length()) == piece.length());
+
+  counter_incore += !new_index && is_incore;
+  counter_incore_new += new_index && is_incore;
+  counter_not_incore += !new_index && !is_incore;
+  counter_not_incore_new += new_index && !is_incore;
+#endif
+}
+
 void
 PeerConnectionBase::load_up_chunk() {
@@ -282 +328 @@
     // Better checking needed.
     //     m_upChunk.chunk()->preload(m_upPiece.offset(), m_upChunk.chunk()->size());
+    log_upload_chunk_mincore(m_upChunk.chunk(), m_upPiece, false);
     return;
   }
@@ -296 +343 @@
     m_encryptBuffer->reset();
   }
+
+  log_upload_chunk_mincore(m_upChunk.chunk(), m_upPiece, true);
 
   // Also check if we've already preloaded in the recent past, even