添加保留字节和支持消息类型交换

PBH-BTN · Oct 18, 2024 · 69edf66 · 69edf66
1 parent 4d45b6b
commit 69edf66
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/main/java/com/ghostchu/peerbanhelper/downloaderplug/biglybt/Plugin.java b/src/main/java/com/ghostchu/peerbanhelper/downloaderplug/biglybt/Plugin.java
@@ -13,6 +13,7 @@
 import com.biglybt.pif.ipfilter.IPBanned;
 import com.biglybt.pif.ipfilter.IPFilter;
 import com.biglybt.pif.ipfilter.IPFilterException;
+import com.biglybt.pif.messaging.Message;
 import com.biglybt.pif.peers.*;
 import com.biglybt.pif.tag.Tag;
 import com.biglybt.pif.torrent.Torrent;
@@ -534,7 +535,9 @@ private PeerRecord getPeerRecord(Peer peer) {
                 client,
                 peer.isOptimisticUnchoke(),
                 peer.supportsMessaging(),
-                peer.isPriorityConnection()
+                peer.isPriorityConnection(),
+                peer.getHandshakeReservedBytes(),
+                Arrays.stream(peer.getSupportedMessages()).map(Message::getID).collect(Collectors.toList())
         );
     }
 

diff --git a/...in/java/com/ghostchu/peerbanhelper/downloaderplug/biglybt/network/wrapper/PeerRecord.java b/...in/java/com/ghostchu/peerbanhelper/downloaderplug/biglybt/network/wrapper/PeerRecord.java
@@ -5,6 +5,7 @@
 import lombok.NoArgsConstructor;
 
 import java.util.List;
+
 /*
 com.biglybt.pif.peers.Peer.java
  */
@@ -37,4 +38,6 @@ public class PeerRecord {
     private boolean optimisticUnchoke;
     private boolean supportsMessaging;
     private boolean priorityConnection;
+    private byte[] handshakeReservedBytes;
+    private List<String> peerSupportedMessages;
 }