1// Copyright 2018 The Go Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style
3// license that can be found in the LICENSE file.
4
5// Test use of raw connections.
6//
7//go:build !plan9 && !js && !wasip1
8
9package os_test
10
11import (
12	"os"
13	"syscall"
14	"testing"
15)
16
17func TestRawConnReadWrite(t *testing.T) {
18	t.Parallel()
19
20	r, w, err := os.Pipe()
21	if err != nil {
22		t.Fatal(err)
23	}
24	defer r.Close()
25	defer w.Close()
26
27	rconn, err := r.SyscallConn()
28	if err != nil {
29		t.Fatal(err)
30	}
31	wconn, err := w.SyscallConn()
32	if err != nil {
33		t.Fatal(err)
34	}
35
36	var operr error
37	err = wconn.Write(func(s uintptr) bool {
38		_, operr = syscall.Write(syscallDescriptor(s), []byte{'b'})
39		return operr != syscall.EAGAIN
40	})
41	if err != nil {
42		t.Fatal(err)
43	}
44	if operr != nil {
45		t.Fatal(err)
46	}
47
48	var n int
49	buf := make([]byte, 1)
50	err = rconn.Read(func(s uintptr) bool {
51		n, operr = syscall.Read(syscallDescriptor(s), buf)
52		return operr != syscall.EAGAIN
53	})
54	if err != nil {
55		t.Fatal(err)
56	}
57	if operr != nil {
58		t.Fatal(operr)
59	}
60	if n != 1 {
61		t.Errorf("read %d bytes, expected 1", n)
62	}
63	if buf[0] != 'b' {
64		t.Errorf("read %q, expected %q", buf, "b")
65	}
66}
67