46 lines
1.0 KiB
Go
46 lines
1.0 KiB
Go
|
package uring
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
qt "github.com/frankban/quicktest"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestUDPSendRecv(t *testing.T) {
|
||
|
|
if !Available() {
|
||
|
|
t.Skip("io_uring not available")
|
||
|
|
}
|
||
|
|
c := qt.New(t)
|
||
|
|
|
||
|
|
listen, err := net.ListenUDP("udp4", &net.UDPAddr{Port: 9999})
|
||
|
|
t.Cleanup(func() { listen.Close() })
|
||
|
|
c.Assert(err, qt.IsNil)
|
||
|
|
|
||
|
|
conn, err := NewUDPConn(listen)
|
||
|
|
t.Cleanup(func() { conn.Close() })
|
||
|
|
if err != nil {
|
||
|
|
t.Skipf("io_uring not available: %v", err)
|
||
|
|
}
|
||
|
|
addr := listen.LocalAddr()
|
||
|
|
sendBuf := make([]byte, 200)
|
||
|
|
for i := range sendBuf {
|
||
|
|
sendBuf[i] = byte(i)
|
||
|
|
}
|
||
|
|
recvBuf := make([]byte, 200)
|
||
|
|
|
||
|
|
// Write one direction.
|
||
|
|
_, err = conn.WriteTo(sendBuf, addr)
|
||
|
|
c.Assert(err, qt.IsNil)
|
||
|
|
n, ipp, err := conn.ReadFromNetaddr(recvBuf)
|
||
|
|
c.Assert(err, qt.IsNil)
|
||
|
|
c.Assert(recvBuf[:n], qt.DeepEquals, sendBuf)
|
||
|
|
|
||
|
|
// Write the other direction, to check that ipp is correct.
|
||
|
|
_, err = conn.WriteTo(sendBuf, ipp.UDPAddr())
|
||
|
|
c.Assert(err, qt.IsNil)
|
||
|
|
n, _, err = conn.ReadFromNetaddr(recvBuf)
|
||
|
|
c.Assert(err, qt.IsNil)
|
||
|
|
c.Assert(recvBuf[:n], qt.DeepEquals, sendBuf)
|
||
|
|
}
|