Skip to content

Commit

Permalink
Rename wrappedPacketConn to just packetConn.
Browse files Browse the repository at this point in the history
  • Loading branch information
sbruens committed Nov 21, 2024
1 parent c0c0e9f commit 8d95309
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 8 deletions.
14 changes: 7 additions & 7 deletions service/udp.go
Original file line number Diff line number Diff line change
Expand Up @@ -150,13 +150,13 @@ func PacketServe(clientConn net.PacketConn, handle PacketHandleFunc) {
continue
}
pkt := buffer[:n]
conn := &wrappedPacketConn{
conn := &packetConn{
PacketConn: clientConn,
readCh: make(chan []byte, 1),
raddr: addr,
}

func(conn *wrappedPacketConn) {
func(conn *packetConn) {
defer func() {
if r := recover(); r != nil {
slog.Error("Panic in UDP loop. Continuing to listen.", "err", r)
Expand All @@ -169,24 +169,24 @@ func PacketServe(clientConn net.PacketConn, handle PacketHandleFunc) {
}
}

type wrappedPacketConn struct {
type packetConn struct {
net.PacketConn
readCh chan []byte
raddr net.Addr
}

var _ net.Conn = (*wrappedPacketConn)(nil)
var _ net.Conn = (*packetConn)(nil)

func (pc *wrappedPacketConn) Read(p []byte) (int, error) {
func (pc *packetConn) Read(p []byte) (int, error) {
data := <-pc.readCh
return copy(p, data), nil
}

func (pc *wrappedPacketConn) Write(b []byte) (n int, err error) {
func (pc *packetConn) Write(b []byte) (n int, err error) {
return pc.PacketConn.WriteTo(b, pc.raddr)
}

func (pc *wrappedPacketConn) RemoteAddr() net.Addr {
func (pc *packetConn) RemoteAddr() net.Addr {
return pc.raddr
}

Expand Down
2 changes: 1 addition & 1 deletion service/udp_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -216,7 +216,7 @@ func setupNAT() (*fakePacketConn, *fakePacketConn, *natconn) {
nat := newNATmap(timeout, &natTestMetrics{}, noopLogger())
clientConn := makePacketConn()
targetConn := makePacketConn()
nat.Add(&wrappedPacketConn{PacketConn: clientConn, raddr: &clientAddr}, targetConn, natCryptoKey, "key id")
nat.Add(&packetConn{PacketConn: clientConn, raddr: &clientAddr}, targetConn, natCryptoKey, "key id")
entry := nat.Get(clientAddr.String())
return clientConn, targetConn, entry
}
Expand Down

0 comments on commit 8d95309

Please sign in to comment.