update refactor

proxy/conn.go

@@ -0,0 +1,48 @@
+package proxy
+
+import (
+	"net"
+
+	"github.com/xvzc/SpoofDPI/util"
+)
+
+const BUF_SIZE = 1024
+
+func ReadBytes(conn net.Conn) ([]byte, error) {
+	ret := make([]byte, 0)
+	buf := make([]byte, BUF_SIZE)
+
+	for {
+		n, err := conn.Read(buf)
+		if err != nil {
+			return nil, err
+		}
+		ret = append(ret, buf[:n]...)
+
+		if n < BUF_SIZE {
+			break
+		}
+	}
+
+	return ret, nil
+}
+
+func Serve(from net.Conn, to net.Conn, proto string) {
+	for {
+		buf, err := ReadBytes(from)
+		if err != nil {
+			util.Debug("["+proto+"]"+"Error reading from ", from.RemoteAddr())
+			util.Debug(err, " Closing the connection.. ")
+			break
+		}
+
+		util.Debug(from.RemoteAddr(), "sent data", len(buf))
+
+		_, write_err := to.Write(buf)
+		if write_err != nil {
+			util.Debug("["+proto+"]"+"Error reading from ", to.RemoteAddr())
+			util.Debug(err, " Closing the connection.. ")
+			break
+		}
+	}
+}

proxy/manager.go

@@ -1,49 +0,0 @@
-package proxy
-
-import(
-    "net"
-	"github.com/xvzc/SpoofDPI/util"
-)
-
-const BUF_SIZE = 1024
-
-func ReadBytes(conn net.Conn)([]byte, error) {
-    buf := make([]byte, 0) // big buffer
-    tmp := make([]byte, BUF_SIZE)     // using small tmo buffer for demonstrating
-
-    for {
-        n, err := conn.Read(tmp)
-        if err != nil {
-            return nil, err
-        }
-        buf = append(buf, tmp[:n]...)
-
-        if n < BUF_SIZE {
-            break
-        }
-    }
-
-    return buf, nil
-}
-
-
-func Serve(from net.Conn, to net.Conn, proto string) {
-    for {
-        buf, err := ReadBytes(from)
-        if err != nil {
-            util.Debug("["+ proto +"]" + "Error reading from ", from.RemoteAddr())
-            util.Debug(err, " Closing the connection.. ")
-            break
-        }
-
-        util.Debug(from.RemoteAddr(), "sent data", len(buf))
-
-        _, write_err := to.Write(buf)
-        if write_err != nil {
-            util.Debug("["+ proto +"]" + "Error reading from ", to.RemoteAddr())
-            util.Debug(err, " Closing the connection.. ")
-            break
-        }
-    }
-}
-

proxy/proxy.go

@@ -5,8 +5,8 @@ import (
 	"net"
 	"os"
 
-	"github.com/xvzc/SpoofDPI/util"
 	"github.com/xvzc/SpoofDPI/config"
+	"github.com/xvzc/SpoofDPI/util"
 )
 
 func Start() {