mirror of https://github.com/liamg/aminal.git
80 lines
1.5 KiB
Go
80 lines
1.5 KiB
Go
package pty
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"syscall"
|
|
"unsafe"
|
|
)
|
|
|
|
// todo: port these for darwin/windows:
|
|
func open() (*os.File, *os.File, error) {
|
|
|
|
pty, err := getpt()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
ptsName, err := ptsname(pty)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
err = grantpt(pty)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
err = unlockpt(pty)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
tty, err := os.OpenFile(ptsName, os.O_RDWR, 0)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
return pty, tty, nil
|
|
|
|
}
|
|
|
|
func getpt() (file *os.File, err error) {
|
|
f, err := syscall.Open("/dev/ptmx", syscall.O_RDWR|syscall.O_CLOEXEC, 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return os.NewFile(uintptr(f), "/dev/ptmx"), nil
|
|
}
|
|
|
|
func ptsname(file *os.File) (name string, err error) {
|
|
|
|
n := make([]byte, (syscall.TIOCPTYGNAME>>16)&((1<<13)-1))
|
|
_, err = ioctl(file, syscall.TIOCPTYGNAME, uintptr(unsafe.Pointer(&n[0])))
|
|
for i, c := range n {
|
|
if c == 0 {
|
|
return string(n[:i]), nil
|
|
}
|
|
}
|
|
return "", errors.New("TIOCPTYGNAME string not NUL-terminated")
|
|
}
|
|
|
|
func ioctl(file *os.File, command uint, arg uintptr) (uintptr, error) {
|
|
_, _, err := syscall.Syscall(syscall.SYS_IOCTL, uintptr(file.Fd()),
|
|
uintptr(command), uintptr(unsafe.Pointer(&arg)))
|
|
if err != 0 {
|
|
return 0, fmt.Errorf("Error no %d", err)
|
|
}
|
|
return arg, nil
|
|
}
|
|
|
|
func grantpt(f *os.File) error {
|
|
_, err := ioctl(f, syscall.TIOCPTYGRANT, 0)
|
|
return err
|
|
}
|
|
|
|
func unlockpt(f *os.File) error {
|
|
_, err := ioctl(f, syscall.TIOCPTYUNLK, 0)
|
|
return err
|
|
}
|