2014-10-23 23:44:57 +00:00
|
|
|
// +build windows
|
2015-07-01 20:16:16 +00:00
|
|
|
|
2014-10-23 23:44:57 +00:00
|
|
|
package term
|
|
|
|
|
2015-03-06 00:41:48 +00:00
|
|
|
import (
|
|
|
|
"io"
|
2015-03-23 18:38:43 +00:00
|
|
|
"os"
|
2015-04-16 21:01:37 +00:00
|
|
|
"os/signal"
|
2015-10-15 18:40:14 +00:00
|
|
|
"syscall"
|
2015-03-22 16:55:21 +00:00
|
|
|
|
2015-04-16 21:01:37 +00:00
|
|
|
"github.com/Azure/go-ansiterm/winterm"
|
|
|
|
"github.com/docker/docker/pkg/term/windows"
|
2015-03-06 00:41:48 +00:00
|
|
|
)
|
|
|
|
|
2015-01-24 01:33:49 +00:00
|
|
|
// State holds the console mode for the terminal.
|
2014-10-23 23:44:57 +00:00
|
|
|
type State struct {
|
2016-06-22 23:34:01 +00:00
|
|
|
mode uint32
|
2014-10-23 23:44:57 +00:00
|
|
|
}
|
|
|
|
|
2015-01-24 01:33:49 +00:00
|
|
|
// Winsize is used for window size.
|
2014-10-23 23:44:57 +00:00
|
|
|
type Winsize struct {
|
|
|
|
Height uint16
|
|
|
|
Width uint16
|
|
|
|
}
|
|
|
|
|
2015-11-11 23:04:40 +00:00
|
|
|
const (
|
|
|
|
// https://msdn.microsoft.com/en-us/library/windows/desktop/ms683167(v=vs.85).aspx
|
|
|
|
enableVirtualTerminalInput = 0x0200
|
|
|
|
enableVirtualTerminalProcessing = 0x0004
|
2016-03-24 22:37:47 +00:00
|
|
|
disableNewlineAutoReturn = 0x0008
|
2015-11-11 23:04:40 +00:00
|
|
|
)
|
|
|
|
|
2016-06-22 23:34:01 +00:00
|
|
|
// vtInputSupported is true if enableVirtualTerminalInput is supported by the console
|
|
|
|
var vtInputSupported bool
|
2015-11-11 23:04:40 +00:00
|
|
|
|
2015-07-25 08:35:07 +00:00
|
|
|
// StdStreams returns the standard streams (stdin, stdout, stedrr).
|
2015-04-02 16:35:13 +00:00
|
|
|
func StdStreams() (stdIn io.ReadCloser, stdOut, stdErr io.Writer) {
|
2016-06-22 23:34:01 +00:00
|
|
|
// Turn on VT handling on all std handles, if possible. This might
|
|
|
|
// fail, in which case we will fall back to terminal emulation.
|
|
|
|
var emulateStdin, emulateStdout, emulateStderr bool
|
|
|
|
fd := os.Stdin.Fd()
|
|
|
|
if mode, err := winterm.GetConsoleMode(fd); err == nil {
|
|
|
|
// Validate that enableVirtualTerminalInput is supported, but do not set it.
|
|
|
|
if err = winterm.SetConsoleMode(fd, mode|enableVirtualTerminalInput); err != nil {
|
|
|
|
emulateStdin = true
|
|
|
|
} else {
|
|
|
|
vtInputSupported = true
|
2015-10-15 18:40:14 +00:00
|
|
|
}
|
2016-07-23 01:34:14 +00:00
|
|
|
// Unconditionally set the console mode back even on failure because SetConsoleMode
|
|
|
|
// remembers invalid bits on input handles.
|
|
|
|
winterm.SetConsoleMode(fd, mode)
|
2015-11-11 23:04:40 +00:00
|
|
|
}
|
|
|
|
|
2016-06-22 23:34:01 +00:00
|
|
|
fd = os.Stdout.Fd()
|
|
|
|
if mode, err := winterm.GetConsoleMode(fd); err == nil {
|
|
|
|
// Validate disableNewlineAutoReturn is supported, but do not set it.
|
|
|
|
if err = winterm.SetConsoleMode(fd, mode|enableVirtualTerminalProcessing|disableNewlineAutoReturn); err != nil {
|
|
|
|
emulateStdout = true
|
|
|
|
} else {
|
|
|
|
winterm.SetConsoleMode(fd, mode|enableVirtualTerminalProcessing)
|
2015-10-15 18:40:14 +00:00
|
|
|
}
|
2016-05-21 02:19:26 +00:00
|
|
|
}
|
|
|
|
|
2016-06-22 23:34:01 +00:00
|
|
|
fd = os.Stderr.Fd()
|
|
|
|
if mode, err := winterm.GetConsoleMode(fd); err == nil {
|
|
|
|
// Validate disableNewlineAutoReturn is supported, but do not set it.
|
|
|
|
if err = winterm.SetConsoleMode(fd, mode|enableVirtualTerminalProcessing|disableNewlineAutoReturn); err != nil {
|
|
|
|
emulateStderr = true
|
|
|
|
} else {
|
|
|
|
winterm.SetConsoleMode(fd, mode|enableVirtualTerminalProcessing)
|
|
|
|
}
|
2015-10-15 18:40:14 +00:00
|
|
|
}
|
|
|
|
|
2016-10-05 18:24:58 +00:00
|
|
|
if os.Getenv("ConEmuANSI") == "ON" || os.Getenv("ConsoleZVersion") != "" {
|
|
|
|
// The ConEmu and ConsoleZ terminals emulate ANSI on output streams well.
|
2016-08-09 03:59:52 +00:00
|
|
|
emulateStdin = true
|
2016-06-22 23:34:01 +00:00
|
|
|
emulateStdout = false
|
|
|
|
emulateStderr = false
|
2015-10-15 18:40:14 +00:00
|
|
|
}
|
|
|
|
|
2016-06-22 23:34:01 +00:00
|
|
|
if emulateStdin {
|
|
|
|
stdIn = windows.NewAnsiReader(syscall.STD_INPUT_HANDLE)
|
|
|
|
} else {
|
|
|
|
stdIn = os.Stdin
|
2015-11-11 23:04:40 +00:00
|
|
|
}
|
|
|
|
|
2016-06-22 23:34:01 +00:00
|
|
|
if emulateStdout {
|
|
|
|
stdOut = windows.NewAnsiWriter(syscall.STD_OUTPUT_HANDLE)
|
|
|
|
} else {
|
|
|
|
stdOut = os.Stdout
|
2015-11-11 23:04:40 +00:00
|
|
|
}
|
|
|
|
|
2016-06-22 23:34:01 +00:00
|
|
|
if emulateStderr {
|
|
|
|
stdErr = windows.NewAnsiWriter(syscall.STD_ERROR_HANDLE)
|
|
|
|
} else {
|
|
|
|
stdErr = os.Stderr
|
2015-11-11 23:04:40 +00:00
|
|
|
}
|
|
|
|
|
2016-06-22 23:34:01 +00:00
|
|
|
return
|
2015-10-15 18:40:14 +00:00
|
|
|
}
|
|
|
|
|
2015-07-25 08:35:07 +00:00
|
|
|
// GetFdInfo returns the file descriptor for an os.File and indicates whether the file represents a terminal.
|
2015-04-02 16:35:13 +00:00
|
|
|
func GetFdInfo(in interface{}) (uintptr, bool) {
|
2015-04-16 21:01:37 +00:00
|
|
|
return windows.GetHandleInfo(in)
|
2015-04-02 16:35:13 +00:00
|
|
|
}
|
|
|
|
|
2015-07-25 08:35:07 +00:00
|
|
|
// GetWinsize returns the window size based on the specified file descriptor.
|
2014-10-23 23:44:57 +00:00
|
|
|
func GetWinsize(fd uintptr) (*Winsize, error) {
|
2015-04-16 21:01:37 +00:00
|
|
|
info, err := winterm.GetConsoleScreenBufferInfo(fd)
|
2014-10-23 23:44:57 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2015-01-24 01:33:49 +00:00
|
|
|
|
2015-04-16 21:01:37 +00:00
|
|
|
winsize := &Winsize{
|
2015-04-06 21:31:42 +00:00
|
|
|
Width: uint16(info.Window.Right - info.Window.Left + 1),
|
|
|
|
Height: uint16(info.Window.Bottom - info.Window.Top + 1),
|
2016-03-31 17:53:21 +00:00
|
|
|
}
|
2015-04-16 21:01:37 +00:00
|
|
|
|
|
|
|
return winsize, nil
|
2014-10-23 23:44:57 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// IsTerminal returns true if the given file descriptor is a terminal.
|
|
|
|
func IsTerminal(fd uintptr) bool {
|
2015-04-16 21:01:37 +00:00
|
|
|
return windows.IsConsole(fd)
|
2014-10-23 23:44:57 +00:00
|
|
|
}
|
|
|
|
|
2015-07-25 08:35:07 +00:00
|
|
|
// RestoreTerminal restores the terminal connected to the given file descriptor
|
|
|
|
// to a previous state.
|
2014-10-23 23:44:57 +00:00
|
|
|
func RestoreTerminal(fd uintptr, state *State) error {
|
2016-06-22 23:34:01 +00:00
|
|
|
return winterm.SetConsoleMode(fd, state.mode)
|
2014-10-23 23:44:57 +00:00
|
|
|
}
|
|
|
|
|
2015-04-02 16:35:13 +00:00
|
|
|
// SaveState saves the state of the terminal connected to the given file descriptor.
|
2014-10-23 23:44:57 +00:00
|
|
|
func SaveState(fd uintptr) (*State, error) {
|
2015-04-16 21:01:37 +00:00
|
|
|
mode, e := winterm.GetConsoleMode(fd)
|
2014-10-23 23:44:57 +00:00
|
|
|
if e != nil {
|
|
|
|
return nil, e
|
|
|
|
}
|
2015-11-11 23:04:40 +00:00
|
|
|
|
2016-06-22 23:34:01 +00:00
|
|
|
return &State{mode: mode}, nil
|
2014-10-23 23:44:57 +00:00
|
|
|
}
|
|
|
|
|
2015-04-02 16:35:13 +00:00
|
|
|
// DisableEcho disables echo for the terminal connected to the given file descriptor.
|
2015-04-16 21:01:37 +00:00
|
|
|
// -- See https://msdn.microsoft.com/en-us/library/windows/desktop/ms683462(v=vs.85).aspx
|
2014-10-23 23:44:57 +00:00
|
|
|
func DisableEcho(fd uintptr, state *State) error {
|
2016-06-22 23:34:01 +00:00
|
|
|
mode := state.mode
|
2015-04-16 21:01:37 +00:00
|
|
|
mode &^= winterm.ENABLE_ECHO_INPUT
|
|
|
|
mode |= winterm.ENABLE_PROCESSED_INPUT | winterm.ENABLE_LINE_INPUT
|
|
|
|
err := winterm.SetConsoleMode(fd, mode)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register an interrupt handler to catch and restore prior state
|
|
|
|
restoreAtInterrupt(fd, state)
|
|
|
|
return nil
|
2014-10-23 23:44:57 +00:00
|
|
|
}
|
|
|
|
|
2016-06-22 23:34:01 +00:00
|
|
|
// SetRawTerminal puts the terminal connected to the given file descriptor into
|
|
|
|
// raw mode and returns the previous state. On UNIX, this puts both the input
|
|
|
|
// and output into raw mode. On Windows, it only puts the input into raw mode.
|
2014-10-23 23:44:57 +00:00
|
|
|
func SetRawTerminal(fd uintptr) (*State, error) {
|
2015-04-02 16:35:13 +00:00
|
|
|
state, err := MakeRaw(fd)
|
2014-10-23 23:44:57 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2015-04-16 21:01:37 +00:00
|
|
|
|
|
|
|
// Register an interrupt handler to catch and restore prior state
|
|
|
|
restoreAtInterrupt(fd, state)
|
2015-04-02 16:35:13 +00:00
|
|
|
return state, err
|
2014-10-23 23:44:57 +00:00
|
|
|
}
|
|
|
|
|
2016-06-22 23:34:01 +00:00
|
|
|
// SetRawTerminalOutput puts the output of terminal connected to the given file
|
|
|
|
// descriptor into raw mode. On UNIX, this does nothing and returns nil for the
|
|
|
|
// state. On Windows, it disables LF -> CRLF translation.
|
|
|
|
func SetRawTerminalOutput(fd uintptr) (*State, error) {
|
|
|
|
state, err := SaveState(fd)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ignore failures, since disableNewlineAutoReturn might not be supported on this
|
|
|
|
// version of Windows.
|
|
|
|
winterm.SetConsoleMode(fd, state.mode|disableNewlineAutoReturn)
|
|
|
|
return state, err
|
|
|
|
}
|
|
|
|
|
2015-04-16 21:01:37 +00:00
|
|
|
// MakeRaw puts the terminal (Windows Console) connected to the given file descriptor into raw
|
|
|
|
// mode and returns the previous state of the terminal so that it can be restored.
|
2014-10-23 23:44:57 +00:00
|
|
|
func MakeRaw(fd uintptr) (*State, error) {
|
|
|
|
state, err := SaveState(fd)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2016-06-22 23:34:01 +00:00
|
|
|
mode := state.mode
|
2015-11-11 23:04:40 +00:00
|
|
|
|
2015-04-02 16:35:13 +00:00
|
|
|
// See
|
|
|
|
// -- https://msdn.microsoft.com/en-us/library/windows/desktop/ms686033(v=vs.85).aspx
|
|
|
|
// -- https://msdn.microsoft.com/en-us/library/windows/desktop/ms683462(v=vs.85).aspx
|
|
|
|
|
|
|
|
// Disable these modes
|
2015-04-16 21:01:37 +00:00
|
|
|
mode &^= winterm.ENABLE_ECHO_INPUT
|
|
|
|
mode &^= winterm.ENABLE_LINE_INPUT
|
|
|
|
mode &^= winterm.ENABLE_MOUSE_INPUT
|
|
|
|
mode &^= winterm.ENABLE_WINDOW_INPUT
|
|
|
|
mode &^= winterm.ENABLE_PROCESSED_INPUT
|
2015-04-02 16:35:13 +00:00
|
|
|
|
|
|
|
// Enable these modes
|
2015-04-16 21:01:37 +00:00
|
|
|
mode |= winterm.ENABLE_EXTENDED_FLAGS
|
|
|
|
mode |= winterm.ENABLE_INSERT_MODE
|
|
|
|
mode |= winterm.ENABLE_QUICK_EDIT_MODE
|
2016-06-22 23:34:01 +00:00
|
|
|
if vtInputSupported {
|
|
|
|
mode |= enableVirtualTerminalInput
|
|
|
|
}
|
2015-04-02 16:35:13 +00:00
|
|
|
|
2015-04-16 21:01:37 +00:00
|
|
|
err = winterm.SetConsoleMode(fd, mode)
|
2014-10-23 23:44:57 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return state, nil
|
|
|
|
}
|
2015-04-16 21:01:37 +00:00
|
|
|
|
|
|
|
func restoreAtInterrupt(fd uintptr, state *State) {
|
|
|
|
sigchan := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(sigchan, os.Interrupt)
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
_ = <-sigchan
|
|
|
|
RestoreTerminal(fd, state)
|
|
|
|
os.Exit(0)
|
|
|
|
}()
|
|
|
|
}
|