feat: 模拟暴力攻击单元测试

Author: meethigher

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

@@ -1,6 +1,7 @@
 package top.meethigher.proxy.tcp.tunnel;
 
 import io.vertx.core.Vertx;
+import io.vertx.core.buffer.Buffer;
 import io.vertx.core.net.NetClient;
 import io.vertx.core.net.NetSocket;
 import org.junit.Before;
@@ -34,6 +35,24 @@ public void setUp() throws Exception {
      */
     @Test
     public void testErrorMsg() {
+//        netClient.connect(44444, "127.0.0.1")
+//                .onComplete(ar -> {
+//                    if (ar.succeeded()) {
+//                        NetSocket socket = ar.result();
+//                        socket.pause();
+//                        socket.closeHandler(v -> {
+//                            log.error("连接被关闭");
+//                            System.exit(1);
+//                        });
+//                        socket.write("hello world");
+//                        socket.resume();
+//                    } else {
+//                        ar.cause().printStackTrace();
+//                        System.exit(1);
+//                    }
+//                });
+
+        // 模拟故意制造错误的请求头长度。服务端可以配置闲时超时参数解决。注意要大于客户端心跳的频率
         netClient.connect(44444, "127.0.0.1")
                 .onComplete(ar -> {
                     if (ar.succeeded()) {
@@ -43,7 +62,34 @@ public void testErrorMsg() {
                             log.error("连接被关闭");
                             System.exit(1);
                         });
-                        socket.write("hello world");
+
+                        // 模拟暴力攻击
+                        vertx.setPeriodic(500, id -> {
+                            byte[] byteArray = TunnelMessage.OpenDataPort.newBuilder()
+                                    .setDataProxyName("ssh-proxy")
+                                    .setDataProxyPort(2222)
+                                    .setSecret("0123456789")
+                                    .build().toByteArray();
+                            int totalLength = 4 + 2 + byteArray.length;
+                            // 模拟恶意制造消息长度
+                            totalLength += 100;
+                            Buffer buffer = Buffer.buffer();
+                            buffer.appendInt(totalLength);
+                            buffer.appendShort(TunnelMessageType.OPEN_DATA_PORT.code());
+                            buffer.appendBytes(byteArray);
+                            socket.write(buffer);
+                        });
+
+//                        socket.write(buffer).onComplete(ar1 -> {
+//                            // 发送一条正常消息
+//                            socket.write(new TunnelMessageCodec().encode(TunnelMessageType.OPEN_DATA_PORT.code(),
+//                                    TunnelMessage.OpenDataPort.newBuilder()
+//                                            .setSecret("0123456789")
+//                                            .setDataProxyName("ssh-proxy")
+//                                            .setDataProxyPort(2222)
+//                                            .build().toByteArray()
+//                            ));
+//                        });
                         socket.resume();
                     } else {
                         ar.cause().printStackTrace();
@@ -83,7 +129,6 @@ public void client() {
         ReverseTcpProxyTunnelClient.create(vertx, netClient).connect("127.0.0.1", 44444);
 
 
-
         LockSupport.park();
     }
 }

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

@@ -1,18 +1,23 @@
 package top.meethigher.proxy.tcp.tunnel;
 
 import io.vertx.core.Vertx;
+import io.vertx.core.VertxOptions;
 import io.vertx.core.net.NetServer;
+import io.vertx.core.net.NetServerOptions;
 import org.junit.Test;
 
+import java.time.Duration;
+import java.util.concurrent.TimeUnit;
 import java.util.concurrent.locks.LockSupport;
 
 public class ReverseTcpProxyTunnelServerTest {
 
 
     @Test
     public void server() {
-        Vertx vertx = Vertx.vertx();
-        NetServer netServer = vertx.createNetServer();
+        Vertx vertx = Vertx.vertx(new VertxOptions().setMaxEventLoopExecuteTime(Duration.ofDays(1).toNanos()));
+        // 设置空闲超时，注意该超时参数，应该大于客户端的心跳时间
+        NetServer netServer = vertx.createNetServer(new NetServerOptions().setIdleTimeout(10).setIdleTimeoutUnit(TimeUnit.SECONDS));
         ReverseTcpProxyTunnelServer.create(vertx, netServer)
                 .start();
         LockSupport.park();