update

app/proxy.go

@@ -0,0 +1,72 @@
+package app
+
+import (
+	"fmt"
+	"git.makemake.in/test/mycommon/mylog"
+	"io"
+	"net"
+)
+
+func Run() {
+	for _, item := range ProxyMap {
+		go listenTcp(item)
+	}
+
+	select {}
+}
+
+func listenTcp(item ProxyItem) {
+	localAddr := fmt.Sprintf(":%d", item.LocalPort)
+	listen, err := net.Listen("tcp", localAddr)
+	if err != nil {
+		mylog.Errorf("listen err: %s", localAddr, err)
+		return
+	}
+
+	mylog.Infof("listen %s", localAddr)
+
+	for {
+		conn, err := listen.Accept()
+		if err != nil {
+			mylog.Errorf("conn accept err: %s", localAddr, err)
+			return
+		}
+
+		go proxyTcp(conn, item)
+	}
+}
+
+func proxyTcp(conn net.Conn, item ProxyItem) {
+	//defer conn.Close()
+
+	for _, addr := range item.RemoteAddr {
+		remoteConn, err := net.Dial("tcp", addr)
+		if err != nil {
+			mylog.Errorf("connect remote err: %s\n", err)
+			return::
+		}
+
+		mylog.Infof("connect remote %s", addr)
+
+		go copyConn(conn, remoteConn)
+		go copyConn(remoteConn, conn)
+	}
+
+}
+
+func copyConn(src, dst net.Conn) {
+	defer func() {
+		src.Close()
+		dst.Close()
+
+		mylog.Warn("close conn")
+	}()
+
+	_, err := io.Copy(src, dst)
+	if err != nil {
+		mylog.Error(err)
+		return
+	}
+
+	//fmt.Println(written)
+}