update ftcp

This commit is contained in:
ginuerzh
2022-03-18 21:19:12 +08:00
parent 0f1f7790c5
commit 6a6367b8d1
7 changed files with 178 additions and 132 deletions

View File

@ -1,21 +1,44 @@
package ftcp
import "net"
import (
"net"
"time"
)
type fakeTCPConn struct {
raddr net.Addr
net.PacketConn
pc net.PacketConn
}
func (c *fakeTCPConn) Read(b []byte) (n int, err error) {
n, _, err = c.ReadFrom(b)
n, _, err = c.pc.ReadFrom(b)
return
}
func (c *fakeTCPConn) Write(b []byte) (n int, err error) {
return c.WriteTo(b, c.raddr)
return c.pc.WriteTo(b, c.raddr)
}
func (c *fakeTCPConn) RemoteAddr() net.Addr {
return c.raddr
}
func (c *fakeTCPConn) LocalAddr() net.Addr {
return c.pc.LocalAddr()
}
func (c *fakeTCPConn) SetDeadline(t time.Time) error {
return c.pc.SetDeadline(t)
}
func (c *fakeTCPConn) SetReadDeadline(t time.Time) error {
return c.pc.SetReadDeadline(t)
}
func (c *fakeTCPConn) SetWriteDeadline(t time.Time) error {
return c.pc.SetWriteDeadline(t)
}
func (c *fakeTCPConn) Close() error {
return c.pc.Close()
}

View File

@ -45,7 +45,7 @@ func (d *ftcpDialer) Dial(ctx context.Context, addr string, opts ...dialer.DialO
return
}
return &fakeTCPConn{
raddr: raddr,
PacketConn: c,
raddr: raddr,
pc: c,
}, nil
}