bug fix

README.md

@@ -1,7 +1,7 @@
 lanproxy
 --------
 
-lanproxy是一个将局域网个人电脑、服务器代理到公网的工具，目前仅支持tcp流量转发，可支持任何tcp上层协议（ssh访问、web服务器访问、远程桌面...）。目前市面上提供类似服务的有花生壳、TeamView、GoToMyCloud等等，但天下没有免费的午餐，要使用第三方的公网服务器就必须为第三方付费，并且这些服务器都有各种各样的限制，此外，由于数据包会流经第三方，因此对数据安全也是一大隐患。
+lanproxy是一个将局域网个人电脑、服务器代理到公网的内网穿透工具，目前仅支持tcp流量转发，可支持任何tcp上层协议（ssh访问、web服务器访问、远程桌面...）。目前市面上提供类似服务的有花生壳、TeamView、GoToMyCloud等等，但天下没有免费的午餐，要使用第三方的公网服务器就必须为第三方付费，并且这些服务器都有各种各样的限制，此外，由于数据包会流经第三方，因此对数据安全也是一大隐患。
 
 ### 实现方案
 

proxy-client/src/main/java/org/fengfei/lanproxy/client/ClientChannelMannager.java

@@ -2,12 +2,15 @@ package org.fengfei.lanproxy.client;
 
 import java.util.Iterator;
 import java.util.Map;
+import java.util.Map.Entry;
 import java.util.concurrent.ConcurrentHashMap;
 
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import io.netty.buffer.Unpooled;
 import io.netty.channel.Channel;
+import io.netty.channel.ChannelFutureListener;
 import io.netty.channel.ChannelOption;
 import io.netty.util.AttributeKey;
 
@@ -61,8 +64,16 @@ public class ClientChannelMannager {
         return realServerChannels.remove(userId);
     }
 
+    public static boolean isRealServerReadable(Channel realServerChannel) {
+        return realServerChannel.attr(CLIENT_CHANNEL_WRITEABLE).get()
+                && realServerChannel.attr(USER_CHANNEL_WRITEABLE).get();
+    }
+
     public static void setRealServerChannelReadability(Channel realServerChannel, Boolean client, Boolean user) {
-        logger.info("update real server channel readability, {} {}", client, user);
+        logger.debug("update real server channel readability, {} {} {}", realServerChannel, client, user);
+        if (realServerChannel == null) {
+            return;
+        }
 
         if (client != null) {
             realServerChannel.attr(CLIENT_CHANNEL_WRITEABLE).set(client);
@@ -78,26 +89,25 @@ public class ClientChannelMannager {
         } else {
             realServerChannel.config().setOption(ChannelOption.AUTO_READ, false);
         }
-
     }
 
     public static void notifyChannelWritabilityChanged(Channel channel) {
-        logger.info("channel writability changed, {}", channel.isWritable());
-
-        Iterator<String> ite = realServerChannels.keySet().iterator();
-        while (ite.hasNext()) {
-            Channel realServerChannel = realServerChannels.get(ite.next());
-            setRealServerChannelReadability(realServerChannel, channel.isWritable(), null);
+        logger.debug("channel writability changed, {}", channel.isWritable());
+        Iterator<Entry<String, Channel>> entryIte = realServerChannels.entrySet().iterator();
+        while (entryIte.hasNext()) {
+            setRealServerChannelReadability(entryIte.next().getValue(), channel.isWritable(), null);
         }
     }
 
     public static void clearRealServerChannels() {
         logger.warn("channel closed, clear real server channels");
 
-        Iterator<String> ite = realServerChannels.keySet().iterator();
+        Iterator<Entry<String, Channel>> ite = realServerChannels.entrySet().iterator();
         while (ite.hasNext()) {
-            Channel realServerChannel = realServerChannels.get(ite.next());
-            realServerChannel.close();
+            Channel realServerChannel = ite.next().getValue();
+            if (realServerChannel.isActive()) {
+                realServerChannel.writeAndFlush(Unpooled.EMPTY_BUFFER).addListener(ChannelFutureListener.CLOSE);
+            }
         }
 
         realServerChannels.clear();

proxy-client/src/main/java/org/fengfei/lanproxy/client/handlers/ClientChannelHandler.java

@@ -8,6 +8,7 @@ import org.slf4j.LoggerFactory;
 
 import io.netty.bootstrap.Bootstrap;
 import io.netty.buffer.ByteBuf;
+import io.netty.buffer.Unpooled;
 import io.netty.channel.Channel;
 import io.netty.channel.ChannelFuture;
 import io.netty.channel.ChannelFutureListener;
@@ -36,20 +37,20 @@ public class ClientChannelHandler extends SimpleChannelInboundHandler<ProxyMessa
     protected void channelRead0(ChannelHandlerContext ctx, ProxyMessage proxyMessage) throws Exception {
         logger.debug("recieved proxy message, type is {}", proxyMessage.getType());
         switch (proxyMessage.getType()) {
-            case ProxyMessage.TYPE_CONNECT:
-                handleConnectMessage(ctx, proxyMessage);
-                break;
-            case ProxyMessage.TYPE_DISCONNECT:
-                handleDisconnectMessage(ctx, proxyMessage);
-                break;
-            case ProxyMessage.TYPE_TRANSFER:
-                handleTransferMessage(ctx, proxyMessage);
-                break;
-            case ProxyMessage.TYPE_WRITE_CONTROL:
-                handleWriteControlMessage(ctx, proxyMessage);
-                break;
-            default:
-                break;
+        case ProxyMessage.TYPE_CONNECT:
+            handleConnectMessage(ctx, proxyMessage);
+            break;
+        case ProxyMessage.TYPE_DISCONNECT:
+            handleDisconnectMessage(ctx, proxyMessage);
+            break;
+        case ProxyMessage.TYPE_TRANSFER:
+            handleTransferMessage(ctx, proxyMessage);
+            break;
+        case ProxyMessage.TYPE_WRITE_CONTROL:
+            handleWriteControlMessage(ctx, proxyMessage);
+            break;
+        default:
+            break;
         }
     }
 
@@ -75,9 +76,10 @@ public class ClientChannelHandler extends SimpleChannelInboundHandler<ProxyMessa
 
     private void handleDisconnectMessage(ChannelHandlerContext ctx, ProxyMessage proxyMessage) {
         String userId = proxyMessage.getUri();
-        Channel userChannel = ClientChannelMannager.removeRealServerChannel(userId);
-        if (userChannel != null) {
-            userChannel.close();
+        Channel realServerChannel = ClientChannelMannager.removeRealServerChannel(userId);
+        logger.debug("handleDisconnectMessage, {} {}", userId, realServerChannel);
+        if (realServerChannel != null) {
+            realServerChannel.writeAndFlush(Unpooled.EMPTY_BUFFER).addListener(ChannelFutureListener.CLOSE);
         }
     }
 
@@ -93,8 +95,10 @@ public class ClientChannelHandler extends SimpleChannelInboundHandler<ProxyMessa
             public void operationComplete(ChannelFuture future) throws Exception {
                 if (future.isSuccess()) {
                     Channel realServerChannel = future.channel();
-                    ClientChannelMannager.setRealServerChannelReadability(realServerChannel, true,
+                    logger.debug("connect realserver success, {}, clientChannelWriteable {}", realServerChannel,
                             channel.isWritable());
+                    ClientChannelMannager.setRealServerChannelReadability(realServerChannel, channel.isWritable(),
+                            true);
                     ClientChannelMannager.addRealServerChannel(userId, realServerChannel);
                     ClientChannelMannager.setRealServerChannelUserId(realServerChannel, userId);
                     ProxyMessage proxyMessage = new ProxyMessage();

proxy-client/src/main/java/org/fengfei/lanproxy/client/handlers/RealServerChannelHandler.java

@@ -49,6 +49,7 @@ public class RealServerChannelHandler extends SimpleChannelInboundHandler<ByteBu
         ClientChannelMannager.removeRealServerChannel(userId);
         Channel channel = ClientChannelMannager.getChannel();
         if (channel != null) {
+            logger.debug("channelInactive, {}", realServerChannel);
             ProxyMessage proxyMessage = new ProxyMessage();
             proxyMessage.setType(ProxyMessage.TYPE_DISCONNECT);
             proxyMessage.setUri(userId);

proxy-client/src/test/resources/config.properties

@@ -1,9 +1,9 @@
 client.key=client
-ssl.enable=true
+ssl.enable=false
 ssl.jksPath=test.jks
 ssl.keyStorePassword=123456
 
 server.host=127.0.0.1
 
 #default ssl port is 4993
-server.port=4993
+server.port=4900

proxy-client/src/test/resources/log4j.properties

@@ -1,7 +1,9 @@
-log4j.rootLogger=info,stdout
+log4j.rootLogger=error,stdout
 
 log4j.appender.stdout=org.apache.log4j.ConsoleAppender
 log4j.appender.stdout.layout=org.apache.log4j.PatternLayout
 log4j.appender.stdout.layout.ConversionPattern=%d %p [%c] - <%m>%n
 
 log4j.logger.io.netty=warn
+log4j.logger.org.fengfei.lanproxy.client.handlers.ClientChannelHandler=info
+log4j.logger.org.fengfei.lanproxy.protocol.IdleCheckHandler=warn

proxy-server/src/main/java/org/fengfei/lanproxy/server/handlers/ServerChannelHandler.java

@@ -9,6 +9,7 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 import io.netty.buffer.ByteBuf;
+import io.netty.buffer.Unpooled;
 import io.netty.channel.Channel;
 import io.netty.channel.ChannelFutureListener;
 import io.netty.channel.ChannelHandlerContext;
@@ -76,8 +77,7 @@ public class ServerChannelHandler extends SimpleChannelInboundHandler<ProxyMessa
         Channel userChannel = ProxyChannelManager.removeUserChannel(ctx.channel(), userId);
         if (userChannel != null) {
             // 数据发送完成后再关闭连接，解决http1.0数据传输问题
-            ByteBuf buf = ctx.alloc().buffer(0);
-            userChannel.writeAndFlush(buf).addListener(ChannelFutureListener.CLOSE);
+            userChannel.writeAndFlush(Unpooled.EMPTY_BUFFER).addListener(ChannelFutureListener.CLOSE);
         }
     }