mirror of
https://github.com/xvzc/SpoofDPI.git
synced 2024-12-31 07:04:39 +00:00
refactor: embed net.TcpConn struct in custom connection struct to avoid manual method implementations
This commit is contained in:
parent
8908d2d051
commit
d987a08549
47
net/conn.go
47
net/conn.go
@ -14,50 +14,7 @@ import (
|
|||||||
const BUF_SIZE = 1024
|
const BUF_SIZE = 1024
|
||||||
|
|
||||||
type Conn struct {
|
type Conn struct {
|
||||||
conn *net.TCPConn
|
net.TCPConn
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Conn) CloseWrite() {
|
|
||||||
c.conn.CloseWrite()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Conn) CloseRead() {
|
|
||||||
c.conn.CloseRead()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Conn) Close() {
|
|
||||||
c.conn.Close()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Conn) RemoteAddr() net.Addr {
|
|
||||||
return c.conn.RemoteAddr()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Conn) LocalAddr() net.Addr {
|
|
||||||
return c.conn.LocalAddr()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Conn) Read(b []byte) (n int, err error) {
|
|
||||||
return c.conn.Read(b)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Conn) Write(b []byte) (n int, err error) {
|
|
||||||
return c.conn.Write(b)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Conn) SetReadDeadline(t time.Time) error {
|
|
||||||
c.conn.SetReadDeadline(t)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Conn) SetDeadLine(t time.Time) error {
|
|
||||||
c.conn.SetDeadline(t)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Conn) SetKeepAlive(b bool) error {
|
|
||||||
c.conn.SetKeepAlive(b)
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (conn *Conn) WriteChunks(c [][]byte) (n int, err error) {
|
func (conn *Conn) WriteChunks(c [][]byte) (n int, err error) {
|
||||||
@ -219,7 +176,7 @@ func (from *Conn) Serve(to *Conn, proto string, fd string, td string) {
|
|||||||
proto += " "
|
proto += " "
|
||||||
|
|
||||||
for {
|
for {
|
||||||
from.conn.SetReadDeadline(time.Now().Add(2000 * time.Millisecond))
|
from.SetReadDeadline(time.Now().Add(2000 * time.Millisecond))
|
||||||
buf, err := from.ReadBytes()
|
buf, err := from.ReadBytes()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if err == io.EOF {
|
if err == io.EOF {
|
||||||
|
@ -27,5 +27,5 @@ func DialTCP(network string, ip string, port string) (*Conn, error) {
|
|||||||
return &Conn{}, err
|
return &Conn{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return &Conn{conn: conn}, nil
|
return &Conn{*conn}, nil
|
||||||
}
|
}
|
||||||
|
@ -14,5 +14,5 @@ func (l *Listener) Accept() (*Conn, error) {
|
|||||||
return &Conn{}, err
|
return &Conn{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return &Conn{conn: conn}, nil
|
return &Conn{*conn}, nil
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user