mirror of
https://github.com/creack/pty.git
synced 2026-04-01 02:57:06 +09:00
remove x/sys dependency for zos
This commit is contained in:
14
fd_helper_other_test.go
Normal file
14
fd_helper_other_test.go
Normal 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
20
fd_helper_zos_test.go
Normal 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
|
||||
}
|
||||
11
io_test.go
11
io_test.go
@@ -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)
|
||||
|
||||
93
pty_zos.go
93
pty_zos.go
@@ -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])
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user