Przeglądaj źródła

Fix term.RestoreTerminal behavior

Guillaume J. Charmes 11 lat temu
rodzic
commit
aa68656cd3
1 zmienionych plików z 12 dodań i 1 usunięć
  1. 12 1
      term/term.go

+ 12 - 1
term/term.go

@@ -1,12 +1,17 @@
 package term
 
 import (
+	"errors"
 	"os"
 	"os/signal"
 	"syscall"
 	"unsafe"
 )
 
+var (
+	ErrInvalidState = errors.New("Invlide terminal state")
+)
+
 type State struct {
 	termios Termios
 }
@@ -47,8 +52,14 @@ func IsTerminal(fd uintptr) bool {
 // Restore restores the terminal connected to the given file descriptor to a
 // previous state.
 func RestoreTerminal(fd uintptr, state *State) error {
+	if state == nil {
+		return ErrInvalidState
+	}
 	_, _, err := syscall.Syscall(syscall.SYS_IOCTL, fd, uintptr(setTermios), uintptr(unsafe.Pointer(&state.termios)))
-	return err
+	if err != 0 {
+		return err
+	}
+	return nil
 }
 
 func SaveState(fd uintptr) (*State, error) {