soypat/lneto
1
0
Fork 0
mirror of https://github.com/soypat/lneto.git synced 2026-09-10 16:49:37 +00:00
Code Issues Packages Projects Releases 1 Wiki Activity

smol changes

Browse Source
This commit is contained in:
soypat
2025-04-06 01:13:35 -03:00
parent 70a3bbbe2f
commit fb9c0a831e
2 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internet/basicstack.go
+4
View File
@@ -33,6 +33,10 @@ func (sb *StackBasic) SetAddr(addr netip.Addr) {
sb.ip = addr.As4() sb.ip = addr.As4()
} }
func (sb *StackBasic) Addr() netip.Addr {
return netip.AddrFrom4(sb.ip)
}
func (sb *StackBasic) Recv(frame []byte) error { func (sb *StackBasic) Recv(frame []byte) error {
ifrm, err := ipv4.NewFrame(frame) ifrm, err := ipv4.NewFrame(frame)
if err != nil { if err != nil {
internet/stackbasic_test.go
+12 -7
View File
@@ -12,15 +12,20 @@ func TestBasicStack(t *testing.T) {
var connCl, connSv TCPConn var connCl, connSv TCPConn
setupClientServer(t, rng, &sbCl, &sbSv, &connCl, &connSv) setupClientServer(t, rng, &sbCl, &sbSv, &connCl, &connSv)
var buf [2048]byte var buf [2048]byte
n, err := sbCl.Handle(buf[:]) expectExchange(t, &sbCl, &sbSv, buf[:])
if err != nil {
t.Fatal(err)
} else if n == 0 {
t.Fatal("expected data exchange")
} }
err = sbSv.Recv(buf[:n])
func expectExchange(t *testing.T, from, to *StackBasic, buf []byte) {
n, err := from.Handle(buf)
if err != nil { if err != nil {
t.Fatal(err) t.Error(err)
} else if n == 0 {
t.Error("expected data exchange")
}
err = to.Recv(buf[:n])
if err != nil {
t.Error(err)
} }
} }