blob: 4f4259069d1f739ee570110bc94125f6bc0fc1c4 [file] [log] [blame]
// Copyright 2023 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
//go:build wasip1
package net
import (
"syscall"
"testing"
)
// The tests in this file intend to validate the ability for net.FileConn and
// net.FileListener to handle both TCP and UDP sockets. Ideally we would test
// the public interface by constructing an *os.File from a file descriptor
// opened on a socket, but the WASI preview 1 specification is too limited to
// support this approach for UDP sockets. Instead, we test the internals that
// make it possible for WASI host runtimes and guest programs to integrate
// socket extensions with the net package using net.FileConn/net.FileListener.
//
// Note that the creation of net.Conn and net.Listener values for TCP sockets
// has an end-to-end test in src/runtime/internal/wasitest, here we are only
// verifying the code paths specific to UDP, and error handling for invalid use
// of the functions.
func TestWasip1FileConnNet(t *testing.T) {
tests := []struct {
filetype syscall.Filetype
network string
error error
}{
{syscall.FILETYPE_SOCKET_STREAM, "tcp", nil},
{syscall.FILETYPE_SOCKET_DGRAM, "udp", nil},
{syscall.FILETYPE_BLOCK_DEVICE, "", syscall.ENOTSOCK},
{syscall.FILETYPE_CHARACTER_DEVICE, "", syscall.ENOTSOCK},
{syscall.FILETYPE_DIRECTORY, "", syscall.ENOTSOCK},
{syscall.FILETYPE_REGULAR_FILE, "", syscall.ENOTSOCK},
{syscall.FILETYPE_SYMBOLIC_LINK, "", syscall.ENOTSOCK},
{syscall.FILETYPE_UNKNOWN, "", syscall.ENOTSOCK},
}
for _, test := range tests {
net, err := fileConnNet(test.filetype)
if net != test.network {
t.Errorf("fileConnNet: network mismatch: want=%q got=%q", test.network, net)
}
if err != test.error {
t.Errorf("fileConnNet: error mismatch: want=%v got=%v", test.error, err)
}
}
}
func TestWasip1FileListenNet(t *testing.T) {
tests := []struct {
filetype syscall.Filetype
network string
error error
}{
{syscall.FILETYPE_SOCKET_STREAM, "tcp", nil},
{syscall.FILETYPE_SOCKET_DGRAM, "", syscall.EOPNOTSUPP},
{syscall.FILETYPE_BLOCK_DEVICE, "", syscall.ENOTSOCK},
{syscall.FILETYPE_CHARACTER_DEVICE, "", syscall.ENOTSOCK},
{syscall.FILETYPE_DIRECTORY, "", syscall.ENOTSOCK},
{syscall.FILETYPE_REGULAR_FILE, "", syscall.ENOTSOCK},
{syscall.FILETYPE_SYMBOLIC_LINK, "", syscall.ENOTSOCK},
{syscall.FILETYPE_UNKNOWN, "", syscall.ENOTSOCK},
}
for _, test := range tests {
net, err := fileListenNet(test.filetype)
if net != test.network {
t.Errorf("fileListenNet: network mismatch: want=%q got=%q", test.network, net)
}
if err != test.error {
t.Errorf("fileListenNet: error mismatch: want=%v got=%v", test.error, err)
}
}
}
func TestWasip1NewFileListener(t *testing.T) {
if l, ok := newFileListener(newFD("tcp", -1)).(*TCPListener); !ok {
t.Errorf("newFileListener: tcp listener type mismatch: %T", l)
} else {
testIsTCPAddr(t, "Addr", l.Addr())
}
}
func TestWasip1NewFileConn(t *testing.T) {
if c, ok := newFileConn(newFD("tcp", -1)).(*TCPConn); !ok {
t.Errorf("newFileConn: tcp conn type mismatch: %T", c)
} else {
testIsTCPAddr(t, "LocalAddr", c.LocalAddr())
testIsTCPAddr(t, "RemoteAddr", c.RemoteAddr())
}
if c, ok := newFileConn(newFD("udp", -1)).(*UDPConn); !ok {
t.Errorf("newFileConn: udp conn type mismatch: %T", c)
} else {
testIsUDPAddr(t, "LocalAddr", c.LocalAddr())
testIsUDPAddr(t, "RemoteAddr", c.RemoteAddr())
}
}
func testIsTCPAddr(t *testing.T, method string, addr Addr) {
if _, ok := addr.(*TCPAddr); !ok {
t.Errorf("%s: returned address is not a *TCPAddr: %T", method, addr)
}
}
func testIsUDPAddr(t *testing.T, method string, addr Addr) {
if _, ok := addr.(*UDPAddr); !ok {
t.Errorf("%s: returned address is not a *UDPAddr: %T", method, addr)
}
}