API touch ups

This commit is contained in:
Patricio Whittingslow
2026-05-14 12:59:57 -03:00
parent 9fcb7e9b52
commit d010e6a7e9
6 changed files with 12 additions and 12 deletions
+3 -3
View File
@@ -489,7 +489,7 @@ func (s *StackAsync) ListenTCP4(conn *tcp.Conn, localPort uint16) (err error) {
return nil
}
func (s *StackAsync) RegisterListener(listener *tcp.Listener) (err error) {
func (s *StackAsync) RegisterListenerTCP(listener *tcp.Listener) (err error) {
// TODO(pato): Possible to forward both IPv4 and IPv6 packets to the listener and have it selectively mux out correctly?
// Can try changing listener to inspect carrierData on demux and get the IPversion to know which tcp.Conns match the IP version.
s.mu.Lock()
@@ -500,7 +500,7 @@ func (s *StackAsync) RegisterListener(listener *tcp.Listener) (err error) {
// RegisterUDP4 registers a StackNode on a UDP port with the given remote address and port.
// The StackUDPPort wrapping is handled internally. The number of user-registered UDP ports
// is limited by [StackConfig.MaxUDPConns].
func (s *StackAsync) RegisterUDP4(node lneto.StackNode, remoteAddr []byte, remotePort uint16) error {
func (s *StackAsync) RegisterUDP4(node lneto.StackNode, remoteAddr [4]byte, remotePort uint16) error {
s.mu.Lock()
defer s.mu.Unlock()
idx := len(s.userUDPs)
@@ -508,7 +508,7 @@ func (s *StackAsync) RegisterUDP4(node lneto.StackNode, remoteAddr []byte, remot
return lneto.ErrExhausted
}
s.userUDPs = s.userUDPs[:idx+1]
s.userUDPs[idx].SetStackNode(node, remoteAddr, remotePort)
s.userUDPs[idx].SetStackNode(node, remoteAddr[:], remotePort)
return s.udps.RegisterMACFiltered(&s.userUDPs[idx], nil)
}
+1 -1
View File
@@ -187,7 +187,7 @@ func (s StackGo) SocketNetip(ctx context.Context, network string, family, sotype
if err != nil {
return nil, err
}
err = s.blk.async.RegisterListener(&l.l)
err = s.blk.async.RegisterListenerTCP(&l.l)
if err != nil {
return nil, err
}
+1 -1
View File
@@ -63,7 +63,7 @@ func TestTCPListener_ConcurrentEcho(t *testing.T) {
if err != nil {
t.Fatal(err)
}
err = serverStack.RegisterListener(&listener)
err = serverStack.RegisterListenerTCP(&listener)
if err != nil {
t.Fatal(err)
}
+2 -2
View File
@@ -72,7 +72,7 @@ func TestStackAsyncListener_SingleConnection(t *testing.T) {
if err != nil {
t.Fatal(err)
}
err = sv.RegisterListener(&listener)
err = sv.RegisterListenerTCP(&listener)
if err != nil {
t.Fatal(err)
}
@@ -160,7 +160,7 @@ func TestStackAsyncListener_MultiSequentialConn(t *testing.T) {
if err != nil {
t.Fatal(err)
}
err = sv.RegisterListener(&listener)
err = sv.RegisterListenerTCP(&listener)
if err != nil {
t.Fatal(err)
}
+4 -4
View File
@@ -38,7 +38,7 @@ func TestMDNS_QueryResponse(t *testing.T) {
responderMAC := [6]byte{0xAA, 0xBB, 0xCC, 0xDD, 0xEE, 0x01}
querierAddr := [4]byte{192, 168, 1, 100}
querierMAC := [6]byte{0xAA, 0xBB, 0xCC, 0xDD, 0xEE, 0x02}
mcastAddr := []byte{224, 0, 0, 251}
mcastAddr := [4]byte{224, 0, 0, 251}
// Setup responder stack with mDNS service.
responderStack := new(StackAsync)
@@ -60,7 +60,7 @@ func TestMDNS_QueryResponse(t *testing.T) {
err = responderClient.Configure(mdns.ClientConfig{
LocalPort: mdns.Port,
Services: []mdns.Service{svc},
MulticastAddr: mcastAddr,
MulticastAddr: mcastAddr[:],
})
if err != nil {
t.Fatal("responder configure:", err)
@@ -89,7 +89,7 @@ func TestMDNS_QueryResponse(t *testing.T) {
var querierClient mdns.Client
err = querierClient.Configure(mdns.ClientConfig{
LocalPort: mdns.Port,
MulticastAddr: mcastAddr,
MulticastAddr: mcastAddr[:],
})
if err != nil {
t.Fatal("querier configure:", err)
@@ -297,7 +297,7 @@ func newMDNSStack(t *testing.T, hostname string, seed int64,
t.Fatal(hostname, "mdns configure:", err)
}
err = stack.RegisterUDP4(&client, mdnsCfg.MulticastAddr, mdns.Port)
err = stack.RegisterUDP4(&client, addr.As4(), mdns.Port)
if err != nil {
t.Fatal(hostname, "register udp:", err)
}
+1 -1
View File
@@ -81,7 +81,7 @@ func TestStackAsync_ListenerSynAckAddressedToClient(t *testing.T) {
if err = listener.Reset(svPort, pool); err != nil {
t.Fatal(err)
}
if err = sv.RegisterListener(&listener); err != nil {
if err = sv.RegisterListenerTCP(&listener); err != nil {
t.Fatal(err)
}