Changeset 1058 for trunk/libtorrent/src/protocol/handshake.cc

Timestamp:

05/05/08 15:16:26 (4 years ago)

Author:

rakshasa

Message:

* Initial seeding support added. Patch by Josef Drexler.

Files:

• trunk/libtorrent/src/protocol/handshake.cc (modified) (2 diffs)

trunk/libtorrent/src/protocol/handshake.cc

@@ -527 +527 @@
   // The download is just starting so we're not sending any
   // bitfield. Pretend we wrote it already.
-  if (m_download->file_list()->bitfield()->is_all_unset())
+  if (m_download->file_list()->bitfield()->is_all_unset() || m_download->initial_seeding() != NULL) {
     m_writePos = m_download->file_list()->bitfield()->size_bytes();
-  else
+    m_writeBuffer.write_32(0);
+
+    if (m_encryption.info()->is_encrypted())
+      m_encryption.info()->encrypt(m_writeBuffer.end() - 4, 4);
+
+  } else {
     prepare_bitfield();
+  }
 
   m_state = READ_MESSAGE;
@@ -622 +628 @@
   // bitfield to send, we need to send a keep-alive now.
   if (m_writePos == m_download->file_list()->bitfield()->size_bytes())
-    prepare_post_handshake(m_download->file_list()->bitfield()->is_all_unset());
+    prepare_post_handshake(m_download->file_list()->bitfield()->is_all_unset() || m_download->initial_seeding() != NULL);
 
   if (m_writeDone)