feat: 添加单元测试。一个TunnelServer支持控制多个TunnelClient

Author: meethigher

src/main/java/top/meethigher/proxy/tcp/tunnel/ReverseTcpProxyTunnelServer.java

@@ -107,17 +107,28 @@ protected boolean doHandle(Vertx vertx, NetSocket netSocket, TunnelMessageType t
                     TunnelMessage.OpenDataPortAck.Builder builder = TunnelMessage.OpenDataPortAck
                             .newBuilder();
                     if (secret.equals(parsed.getSecret())) {
-                        final DataProxyServer dataProxyServer = new DataProxyServer(vertx, parsed.getDataProxyName(), parsed.getDataProxyPort(), netSocket);
-                        if (dataProxyServer.startSync()) {
-                            result = true;
-                            builder.setSuccess(result).setMessage("success");
-                            netSocket.write(encode(TunnelMessageType.OPEN_DATA_PORT_ACK,
-                                    builder.build().toByteArray()));
-                            authedSockets.put(netSocket, dataProxyServer);
-                        } else {
-                            builder.setSuccess(result).setMessage("fail to open data port " + parsed.getDataProxyPort());
-                            netSocket.write(encode(TunnelMessageType.OPEN_DATA_PORT_ACK,
-                                    builder.build().toByteArray())).onComplete(ar -> netSocket.close());
+                        synchronized (ReverseTcpProxyTunnelServer.class) {
+                            // 判断dataProxyName是否唯一
+                            for (DataProxyServer server : authedSockets.values()) {
+                                if (server.name.equals(parsed.getDataProxyName())) {
+                                    builder.setSuccess(result).setMessage(server.name + " already started");
+                                    netSocket.write(encode(TunnelMessageType.OPEN_DATA_PORT_ACK,
+                                            builder.build().toByteArray())).onComplete(ar -> netSocket.close());
+                                    return result;
+                                }
+                            }
+                            final DataProxyServer dataProxyServer = new DataProxyServer(vertx, parsed.getDataProxyName(), parsed.getDataProxyPort(), netSocket);
+                            if (dataProxyServer.startSync()) {
+                                result = true;
+                                builder.setSuccess(result).setMessage("success");
+                                netSocket.write(encode(TunnelMessageType.OPEN_DATA_PORT_ACK,
+                                        builder.build().toByteArray()));
+                                authedSockets.put(netSocket, dataProxyServer);
+                            } else {
+                                builder.setSuccess(result).setMessage("fail to open data port " + parsed.getDataProxyPort());
+                                netSocket.write(encode(TunnelMessageType.OPEN_DATA_PORT_ACK,
+                                        builder.build().toByteArray())).onComplete(ar -> netSocket.close());
+                            }
                         }
 
                     } else {

src/test/java/top/meethigher/proxy/tcp/tunnel/ReverseTcpProxyTunnelClientTest.java

@@ -131,9 +131,22 @@ public void client() {
         Vertx vertx = Vertx.vertx(new VertxOptions().setAddressResolverOptions(
                 new AddressResolverOptions().setQueryTimeout(2000)
         ));
-        ReverseTcpProxyTunnelClient.create(ReverseTcpProxyTunnelClientTest.vertx, vertx.createNetClient()).connect("127.0.0.1", 44444);
-
-
+        // http内网穿透
+        ReverseTcpProxyTunnelClient.create(ReverseTcpProxyTunnelClientTest.vertx, vertx.createNetClient())
+                .backendHost("reqres.in")
+                .backendPort(80)
+                .dataProxyName("http-proxy")
+                .dataProxyHost("127.0.0.1")
+                .dataProxyPort(80)
+                .connect("127.0.0.1", 44444);
+        // ssh内网穿透
+        ReverseTcpProxyTunnelClient.create(ReverseTcpProxyTunnelClientTest.vertx, vertx.createNetClient())
+                .backendHost("meethigher.top")
+                .backendPort(22)
+                .dataProxyName("ssh-proxy")
+                .dataProxyHost("127.0.0.1")
+                .dataProxyPort(22)
+                .connect("127.0.0.1", 44444);
         LockSupport.park();
     }
 }