use a constructor for PacketConnWrapper

This commit is contained in:
Shelikhoo 2024-09-11 13:39:16 +01:00 committed by WofWca
parent 85b716bb40
commit f10dc3d619
2 changed files with 10 additions and 5 deletions

View file

@ -9,6 +9,14 @@ import (
var errENOSYS = errors.New("not implemented")
func newPacketConnWrapper(localAddr, remoteAddr net.Addr, rw io.ReadWriter) net.PacketConn {
return &packetConnWrapper{
ReadWriter: rw,
remoteAddr: remoteAddr,
localAddr: localAddr,
}
}
type packetConnWrapper struct {
io.ReadWriter
remoteAddr net.Addr

View file

@ -344,11 +344,8 @@ func newSession(snowflakes SnowflakeCollector, clientIDCandid turbotunnel.Client
return nil, errors.New("handler: Received invalid Snowflake")
}
log.Println("---- Handler: snowflake assigned ----")
packetConnWrapper := &packetConnWrapper{
ReadWriter: conn,
remoteAddr: dummyAddr{},
localAddr: dummyAddr{},
}
packetConnWrapper := newPacketConnWrapper(dummyAddr{}, dummyAddr{}, conn)
return packetConnWrapper, nil
}
pconn := turbotunnel.NewRedialPacketConn(dummyAddr{}, dummyAddr{}, dialContext)