1// Copyright 2017 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
5package net
6
7import (
8	"internal/poll"
9	"runtime"
10	"syscall"
11)
12
13// BUG(tmm1): On Windows, the Write method of syscall.RawConn
14// does not integrate with the runtime's network poller. It cannot
15// wait for the connection to become writeable, and does not respect
16// deadlines. If the user-provided callback returns false, the Write
17// method will fail immediately.
18
19// BUG(mikio): On JS and Plan 9, the Control, Read and Write
20// methods of syscall.RawConn are not implemented.
21
22type rawConn struct {
23	fd *netFD
24}
25
26func (c *rawConn) ok() bool { return c != nil && c.fd != nil }
27
28func (c *rawConn) Control(f func(uintptr)) error {
29	if !c.ok() {
30		return syscall.EINVAL
31	}
32	err := c.fd.pfd.RawControl(f)
33	runtime.KeepAlive(c.fd)
34	if err != nil {
35		err = &OpError{Op: "raw-control", Net: c.fd.net, Source: nil, Addr: c.fd.laddr, Err: err}
36	}
37	return err
38}
39
40func (c *rawConn) Read(f func(uintptr) bool) error {
41	if !c.ok() {
42		return syscall.EINVAL
43	}
44	err := c.fd.pfd.RawRead(f)
45	runtime.KeepAlive(c.fd)
46	if err != nil {
47		err = &OpError{Op: "raw-read", Net: c.fd.net, Source: c.fd.laddr, Addr: c.fd.raddr, Err: err}
48	}
49	return err
50}
51
52func (c *rawConn) Write(f func(uintptr) bool) error {
53	if !c.ok() {
54		return syscall.EINVAL
55	}
56	err := c.fd.pfd.RawWrite(f)
57	runtime.KeepAlive(c.fd)
58	if err != nil {
59		err = &OpError{Op: "raw-write", Net: c.fd.net, Source: c.fd.laddr, Addr: c.fd.raddr, Err: err}
60	}
61	return err
62}
63
64// PollFD returns the poll.FD of the underlying connection.
65//
66// Other packages in std that also import [internal/poll] (such as os)
67// can use a type assertion to access this extension method so that
68// they can pass the *poll.FD to functions like poll.Splice.
69//
70// PollFD is not intended for use outside the standard library.
71func (c *rawConn) PollFD() *poll.FD {
72	if !c.ok() {
73		return nil
74	}
75	return &c.fd.pfd
76}
77
78func newRawConn(fd *netFD) *rawConn {
79	return &rawConn{fd: fd}
80}
81
82// Network returns the network type of the underlying connection.
83//
84// Other packages in std that import internal/poll and are unable to
85// import net (such as os) can use a type assertion to access this
86// extension method so that they can distinguish different socket types.
87//
88// Network is not intended for use outside the standard library.
89func (c *rawConn) Network() poll.String {
90	return poll.String(c.fd.net)
91}
92
93type rawListener struct {
94	rawConn
95}
96
97func (l *rawListener) Read(func(uintptr) bool) error {
98	return syscall.EINVAL
99}
100
101func (l *rawListener) Write(func(uintptr) bool) error {
102	return syscall.EINVAL
103}
104
105func newRawListener(fd *netFD) *rawListener {
106	return &rawListener{rawConn{fd: fd}}
107}
108