SpoofDPI/proxy/manager.go

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
        }
    }
}