mirror of
https://github.com/sorenisanerd/gotty.git
synced 2024-11-09 23:34:26 +00:00
66 lines
1.2 KiB
Go
66 lines
1.2 KiB
Go
package pty
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"syscall"
|
|
"unsafe"
|
|
)
|
|
|
|
func open() (pty, tty *os.File, err error) {
|
|
pFD, err := syscall.Open("/dev/ptmx", syscall.O_RDWR|syscall.O_CLOEXEC, 0)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
p := os.NewFile(uintptr(pFD), "/dev/ptmx")
|
|
// In case of error after this point, make sure we close the ptmx fd.
|
|
defer func() {
|
|
if err != nil {
|
|
_ = p.Close() // Best effort.
|
|
}
|
|
}()
|
|
|
|
sname, err := ptsname(p)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
if err := grantpt(p); err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
if err := unlockpt(p); err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
t, err := os.OpenFile(sname, os.O_RDWR, 0)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
return p, t, nil
|
|
}
|
|
|
|
func ptsname(f *os.File) (string, error) {
|
|
n := make([]byte, _IOC_PARM_LEN(syscall.TIOCPTYGNAME))
|
|
|
|
err := ioctl(f.Fd(), syscall.TIOCPTYGNAME, uintptr(unsafe.Pointer(&n[0])))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
for i, c := range n {
|
|
if c == 0 {
|
|
return string(n[:i]), nil
|
|
}
|
|
}
|
|
return "", errors.New("TIOCPTYGNAME string not NUL-terminated")
|
|
}
|
|
|
|
func grantpt(f *os.File) error {
|
|
return ioctl(f.Fd(), syscall.TIOCPTYGRANT, 0)
|
|
}
|
|
|
|
func unlockpt(f *os.File) error {
|
|
return ioctl(f.Fd(), syscall.TIOCPTYUNLK, 0)
|
|
}
|