67 lines
2.0 KiB
Go
67 lines
2.0 KiB
Go
// Copyright 2020 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.
|
|
|
|
package socket
|
|
|
|
import (
|
|
"net"
|
|
"syscall"
|
|
"unsafe"
|
|
)
|
|
|
|
func syscall_syscall(trap, a1, a2, a3 uintptr) (r1, r2 uintptr, err syscall.Errno)
|
|
func syscall_syscall6(trap, a1, a2, a3, a4, a5, a6 uintptr) (r1, r2 uintptr, err syscall.Errno)
|
|
|
|
func probeProtocolStack() int {
|
|
return 4 // sizeof(int) on GOOS=zos GOARCH=s390x
|
|
}
|
|
|
|
func getsockopt(s uintptr, level, name int, b []byte) (int, error) {
|
|
l := uint32(len(b))
|
|
_, _, errno := syscall_syscall6(syscall.SYS_GETSOCKOPT, s, uintptr(level), uintptr(name), uintptr(unsafe.Pointer(&b[0])), uintptr(unsafe.Pointer(&l)), 0)
|
|
return int(l), errnoErr(errno)
|
|
}
|
|
|
|
func setsockopt(s uintptr, level, name int, b []byte) error {
|
|
_, _, errno := syscall_syscall6(syscall.SYS_SETSOCKOPT, s, uintptr(level), uintptr(name), uintptr(unsafe.Pointer(&b[0])), uintptr(len(b)), 0)
|
|
return errnoErr(errno)
|
|
}
|
|
|
|
func recvmsg(s uintptr, buffers [][]byte, oob []byte, flags int, network string) (n, oobn int, recvflags int, from net.Addr, err error) {
|
|
var h msghdr
|
|
vs := make([]iovec, len(buffers))
|
|
var sa []byte
|
|
if network != "tcp" {
|
|
sa = make([]byte, sizeofSockaddrInet6)
|
|
}
|
|
h.pack(vs, buffers, oob, sa)
|
|
sn, _, errno := syscall_syscall(syscall.SYS___RECVMSG_A, s, uintptr(unsafe.Pointer(&h)), uintptr(flags))
|
|
n = int(sn)
|
|
oobn = h.controllen()
|
|
recvflags = h.flags()
|
|
err = errnoErr(errno)
|
|
if network != "tcp" {
|
|
var err2 error
|
|
from, err2 = parseInetAddr(sa, network)
|
|
if err2 != nil && err == nil {
|
|
err = err2
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func sendmsg(s uintptr, buffers [][]byte, oob []byte, to net.Addr, flags int) (int, error) {
|
|
var h msghdr
|
|
vs := make([]iovec, len(buffers))
|
|
var sa []byte
|
|
if to != nil {
|
|
var a [sizeofSockaddrInet6]byte
|
|
n := marshalInetAddr(to, a[:])
|
|
sa = a[:n]
|
|
}
|
|
h.pack(vs, buffers, oob, sa)
|
|
n, _, errno := syscall_syscall(syscall.SYS___SENDMSG_A, s, uintptr(unsafe.Pointer(&h)), uintptr(flags))
|
|
return int(n), errnoErr(errno)
|
|
}
|