Explicitly use localhost on server socket.

raphw · Jun 17, 2020 · c259f6e · c259f6e
1 parent 01573df
commit c259f6e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/byte-buddy-agent/src/main/java/net/bytebuddy/agent/VirtualMachine.java b/byte-buddy-agent/src/main/java/net/bytebuddy/agent/VirtualMachine.java
@@ -23,6 +23,7 @@
 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
 
 import java.io.*;
+import java.net.InetAddress;
 import java.net.ServerSocket;
 import java.net.Socket;
 import java.nio.ByteBuffer;
@@ -1673,7 +1674,7 @@ public static VirtualMachine attach(String processId, int timeout, Dispatcher di
                     if (target == null) {
                         throw new IllegalStateException("Could not locate target process info in " + directory);
                     }
-                    ServerSocket serverSocket = new ServerSocket(0);
+                    ServerSocket serverSocket = new ServerSocket(0, 1, InetAddress.getLocalHost());
                     try {
                         serverSocket.setSoTimeout(timeout);
                         File receiver = new File(directory, target.getProperty("vmId"));

diff --git a/byte-buddy-agent/src/test/java/net/bytebuddy/agent/VirtualMachineForOpenJ9Test.java b/byte-buddy-agent/src/test/java/net/bytebuddy/agent/VirtualMachineForOpenJ9Test.java
@@ -8,6 +8,7 @@
 
 import java.io.*;
 import java.lang.reflect.Method;
+import java.net.InetAddress;
 import java.net.InetSocketAddress;
 import java.net.Socket;
 import java.util.Arrays;
@@ -89,7 +90,7 @@ public void run() {
                                         }
                                         Socket socket = new Socket();
                                         try {
-                                            socket.connect(new InetSocketAddress(port), 5000);
+                                            socket.connect(new InetSocketAddress(InetAddress.getLocalHost(), port), 5000);
                                             socket.getOutputStream().write((' ' + key + ' ').getBytes("UTF-8"));
                                             socket.getOutputStream().write(0);
                                             socket.getOutputStream().flush();