fix build

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

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/pom.xml b/pom.xml
@@ -6,7 +6,7 @@
 
     <groupId>com.ghostchu.peerbanhelper</groupId>
     <artifactId>BiglyBT-PBH-Plugin</artifactId>
-    <version>1.2.5</version>
+    <version>1.2.6</version>
     <!--    <packaging>takari-jar</packaging>-->
     <packaging>jar</packaging>
 

diff --git a/src/main/java/com/ghostchu/peerbanhelper/downloaderplug/biglybt/Plugin.java b/src/main/java/com/ghostchu/peerbanhelper/downloaderplug/biglybt/Plugin.java
@@ -151,7 +151,7 @@ public void unload() throws PluginException {
         if (webContainer != null) {
             webContainer.stop();
         }
-        if(clientIDGeneratorOriginal != null){
+        if (clientIDGeneratorOriginal != null) {
             ClientIDManagerImpl.getSingleton().setGenerator(clientIDGeneratorOriginal, true);
         }
     }
@@ -281,9 +281,9 @@ private void handleBanListApplied(Context context) throws IPFilterException {
     }
 
     public ConnectorData getConnectorData() {
-        if(useClientIdModifier) {
+        if (useClientIdModifier) {
             return connectorData;
-        }else{
+        } else {
             return null;
         }
     }
@@ -510,6 +510,11 @@ private PeerRecord getPeerRecord(Peer peer) {
         }
         if (peer.getIp().endsWith(".i2p") || peer.getIp().endsWith(".onion") || peer.getIp().endsWith(".tor"))
             return null;
+        com.biglybt.pif.messaging.Message[] messages = new Message[0];
+        try {
+            messages = peer.getSupportedMessages();
+        } catch (NullPointerException ignored) {
+        }
         return new PeerRecord(
                 peer.isMyPeer(),
                 peer.getState(),
@@ -537,7 +542,7 @@ private PeerRecord getPeerRecord(Peer peer) {
                 peer.supportsMessaging(),
                 peer.isPriorityConnection(),
                 peer.getHandshakeReservedBytes(),
-                Arrays.stream(peer.getSupportedMessages()).map(Message::getID).collect(Collectors.toList())
+                Arrays.stream(messages).map(Message::getID).collect(Collectors.toList())
         );
     }