remove x/sys dependency for zos

This commit is contained in:
Mac Malainey
2024-10-31 10:42:45 -04:00
parent 13c571dda3
commit 7c00df3597
4 changed files with 118 additions and 20 deletions

14
fd_helper_other_test.go Normal file
View File

@@ -0,0 +1,14 @@
//go:build !zos
// +build !zos
package pty
import (
"os"
"testing"
)
func getNonBlockingFile(t *testing.T, file *os.File, path string) *os.File {
t.Helper()
return file
}

20
fd_helper_zos_test.go Normal file
View File

@@ -0,0 +1,20 @@
//go:build zos
// +build zos
package pty
import (
"os"
"testing"
)
func getNonBlockingFile(t *testing.T, file *os.File, path string) *os.File {
t.Helper()
// z/OS doesn't open a pollable FD - fix that here
if _, err := fcntl(uintptr(file.Fd()), F_SETFL, O_NONBLOCK); err != nil {
t.Fatalf("Error: zos-nonblock: %s.\n", err)
}
nf := os.NewFile(file.Fd(), path)
t.Cleanup(func() { _ = nf.Close() })
return nf
}

View File

@@ -12,8 +12,6 @@ import (
"syscall"
"testing"
"time"
"golang.org/x/sys/unix"
)
const (
@@ -112,14 +110,7 @@ func prepare(t *testing.T) (ptmx *os.File, done func()) {
t.Cleanup(func() { _ = _ptmx.Close() })
t.Cleanup(func() { _ = pts.Close() })
// z/OS doesn't open a pollable FD - fix that here
if runtime.GOOS == "zos" {
if _, err = unix.Fcntl(uintptr(ptmx.Fd()), unix.F_SETFL, unix.O_NONBLOCK); err != nil {
t.Fatalf("Error: zos-nonblock: %s.\n", err)
}
ptmx = os.NewFile(ptmx.Fd(), "/dev/ptmx")
t.Cleanup(func() { _ = ptmx.Close() })
}
ptmx = getNonBlockingFile(t, ptmx, "/dev/ptmx")
ctx, done := context.WithCancel(context.Background())
t.Cleanup(done)

View File

@@ -5,24 +5,44 @@ package pty
import (
"os"
"runtime"
"syscall"
"golang.org/x/sys/unix"
"unsafe"
)
const (
SYS_UNLOCKPT = 0x37B
SYS_GRANTPT = 0x37A
SYS_POSIX_OPENPT = 0xC66
SYS_FCNTL = 0x18C
SYS___PTSNAME_A = 0x718
SETCVTON = 1
O_NONBLOCK = 0x04
F_SETFL = 4
F_CONTROL_CVT = 13
)
type f_cnvrt struct {
Cvtcmd int32
Pccsid int16
Fccsid int16
}
func open() (pty, tty *os.File, err error) {
ptmxfd, err := unix.Posix_openpt(os.O_RDWR|syscall.O_NOCTTY)
ptmxfd, err := openpt(os.O_RDWR | syscall.O_NOCTTY)
if err != nil {
return nil, nil, err
}
// Needed for z/OS so that the characters are not garbled if ptyp* is untagged
cvtreq := unix.F_cnvrt{Cvtcmd: unix.SETCVTON, Pccsid: 0, Fccsid: 1047}
if _, err = unix.Fcntl(uintptr(ptmxfd), unix.F_CONTROL_CVT, &cvtreq); err != nil {
cvtreq := f_cnvrt{Cvtcmd: SETCVTON, Pccsid: 0, Fccsid: 1047}
if _, err = fcntl(uintptr(ptmxfd), F_CONTROL_CVT, uintptr(unsafe.Pointer(&cvtreq))); err != nil {
return nil, nil, err
}
p := os.NewFile(uintptr(ptmxfd), "/dev/ptmx")
if p == nil {
return nil, nil, err
@@ -35,17 +55,17 @@ func open() (pty, tty *os.File, err error) {
}
}()
sname, err := unix.Ptsname(ptmxfd)
sname, err := ptsname(ptmxfd)
if err != nil {
return nil, nil, err
}
_, err = unix.Grantpt(ptmxfd)
_, err = grantpt(ptmxfd)
if err != nil {
return nil, nil, err
}
if _, err = unix.Unlockpt(ptmxfd); err != nil {
if _, err = unlockpt(ptmxfd); err != nil {
return nil, nil, err
}
@@ -54,7 +74,7 @@ func open() (pty, tty *os.File, err error) {
return nil, nil, err
}
if _, err = unix.Fcntl(uintptr(ptsfd), unix.F_CONTROL_CVT, &cvtreq); err != nil {
if _, err = fcntl(uintptr(ptsfd), F_CONTROL_CVT, uintptr(unsafe.Pointer(&cvtreq))); err != nil {
return nil, nil, err
}
@@ -66,3 +86,56 @@ func open() (pty, tty *os.File, err error) {
return p, t, nil
}
func openpt(oflag int) (fd int, err error) {
r0, _, e1 := runtime.CallLeFuncWithErr(runtime.GetZosLibVec()+SYS_POSIX_OPENPT<<4, uintptr(oflag))
fd = int(r0)
if e1 != 0 {
err = syscall.Errno(e1)
}
return
}
func fcntl(fd uintptr, cmd int, arg uintptr) (val int, err error) {
r0, _, e1 := runtime.CallLeFuncWithErr(runtime.GetZosLibVec()+SYS_FCNTL<<4, uintptr(fd), uintptr(cmd), arg)
val = int(r0)
if e1 != 0 {
err = syscall.Errno(e1)
}
return
}
func ptsname(fd int) (name string, err error) {
r0, _, e1 := runtime.CallLeFuncWithPtrReturn(runtime.GetZosLibVec()+SYS___PTSNAME_A<<4, uintptr(fd))
name = u2s(unsafe.Pointer(r0))
if e1 != 0 {
err = syscall.Errno(e1)
}
return
}
func grantpt(fildes int) (rc int, err error) {
r0, _, e1 := runtime.CallLeFuncWithErr(runtime.GetZosLibVec()+SYS_GRANTPT<<4, uintptr(fildes))
rc = int(r0)
if e1 != 0 {
err = syscall.Errno(e1)
}
return
}
func unlockpt(fildes int) (rc int, err error) {
r0, _, e1 := runtime.CallLeFuncWithErr(runtime.GetZosLibVec()+SYS_UNLOCKPT<<4, uintptr(fildes))
rc = int(r0)
if e1 != 0 {
err = syscall.Errno(e1)
}
return
}
func u2s(cstr unsafe.Pointer) string {
str := (*[1024]uint8)(cstr)
i := 0
for str[i] != 0 {
i++
}
return string(str[:i])
}