1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
// +build windows
package input
import (
"os"
"syscall"
)
// LineSep is the separator for windows or unix systems
const LineSep = "\r\n"
// Magic constant from MSDN to control whether characters read are
// repeated back on the console.
//
// http://msdn.microsoft.com/en-us/library/windows/desktop/ms686033(v=vs.85).aspx
const ENABLE_ECHO_INPUT = 0x0004
// rawRead reads file with raw mode (without prompting to terminal).
//
// For this windows version of rawRead(). I referred the codes on
// hashicorp/vault/helper and cloudfoundry/cli/terminal
func (i *UI) rawRead(f *os.File) (string, error) {
// In windows, Handle can be used to examine or modify the system resource.
// https://msdn.microsoft.com/en-us/library/windows/desktop/ms724457(v=vs.85).aspx
handle := syscall.Handle(f.Fd())
resetFunc, err := makeRaw(handle)
if err != nil {
return "", err
}
defer resetFunc()
return i.rawReadline(f)
}
func makeRaw(console syscall.Handle) (func(), error) {
// Get old mode so that we can recover later
var oldMode uint32
if err := syscall.GetConsoleMode(console, &oldMode); err != nil {
return nil, err
}
var newMode uint32 = uint32(int(oldMode) & ^ENABLE_ECHO_INPUT)
if err := setConsoleMode(console, newMode); err != nil {
return nil, err
}
return func() {
setConsoleMode(console, oldMode)
}, nil
}
func setConsoleMode(console syscall.Handle, mode uint32) error {
// MustLoadDLL is like LoadDLL but panics if load operation fails.
// LoadDLL loads DLL file into memory
//
// KERNEL32.DLL exposes to applications most of the Win32 base APIs,
// such as memory management, input/output (I/O) operations,
// process and thread creation, and synchronization functions.
kernel32 := syscall.MustLoadDLL("kernel32")
// Sets the input mode of a console's input buffer or
// the output mode of a console screen buffer.
// https://msdn.microsoft.com/en-us/library/windows/desktop/ms686033(v=vs.85).aspx
proc := kernel32.MustFindProc("SetConsoleMode")
r, _, err := proc.Call(uintptr(console), uintptr(mode))
if r == 0 {
return err
}
return nil
}
|
