Skip to content

Commit

Permalink
remove x/sys dependency for zos
Browse files Browse the repository at this point in the history
  • Loading branch information
MacMalainey committed Oct 31, 2024
1 parent 13c571d commit 7c00df3
Show file tree
Hide file tree
Showing 4 changed files with 117 additions and 19 deletions.
14 changes: 14 additions & 0 deletions fd_helper_other_test.go
Original file line number Diff line number Diff line change
@@ -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 changes: 20 additions & 0 deletions fd_helper_zos_test.go
Original file line number Diff line number Diff line change
@@ -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
}
11 changes: 1 addition & 10 deletions io_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,6 @@ import (
"syscall"
"testing"
"time"

"golang.org/x/sys/unix"
)

const (
Expand Down Expand Up @@ -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)
Expand Down
91 changes: 82 additions & 9 deletions pty_zos.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,24 +5,44 @@ package pty

import (
"os"
"runtime"
"syscall"
"unsafe"
)

const (
SYS_UNLOCKPT = 0x37B
SYS_GRANTPT = 0x37A
SYS_POSIX_OPENPT = 0xC66
SYS_FCNTL = 0x18C
SYS___PTSNAME_A = 0x718

SETCVTON = 1

"golang.org/x/sys/unix"
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
Expand All @@ -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
}

Expand All @@ -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
}

Expand All @@ -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])
}

0 comments on commit 7c00df3

Please sign in to comment.