1//go:build windows
 2// +build windows
 3
 4package term
 5
 6import (
 7	"os"
 8
 9	"golang.org/x/sys/windows"
10)
11
12type state struct {
13	Mode uint32
14}
15
16func isTerminal(fd uintptr) bool {
17	var st uint32
18	err := windows.GetConsoleMode(windows.Handle(fd), &st)
19	return err == nil
20}
21
22func makeRaw(fd uintptr) (*State, error) {
23	var st uint32
24	if err := windows.GetConsoleMode(windows.Handle(fd), &st); err != nil {
25		return nil, err
26	}
27	raw := st &^ (windows.ENABLE_ECHO_INPUT | windows.ENABLE_PROCESSED_INPUT | windows.ENABLE_LINE_INPUT | windows.ENABLE_PROCESSED_OUTPUT)
28	raw |= windows.ENABLE_VIRTUAL_TERMINAL_INPUT
29	if err := windows.SetConsoleMode(windows.Handle(fd), raw); err != nil {
30		return nil, err
31	}
32	return &State{state{st}}, nil
33}
34
35func setState(fd uintptr, state *State) error {
36	var mode uint32
37	if state != nil {
38		mode = state.Mode
39	}
40	return windows.SetConsoleMode(windows.Handle(fd), mode)
41}
42
43func getState(fd uintptr) (*State, error) {
44	var st uint32
45	if err := windows.GetConsoleMode(windows.Handle(fd), &st); err != nil {
46		return nil, err
47	}
48	return &State{state{st}}, nil
49}
50
51func restore(fd uintptr, state *State) error {
52	return windows.SetConsoleMode(windows.Handle(fd), state.Mode)
53}
54
55func getSize(fd uintptr) (width, height int, err error) {
56	var info windows.ConsoleScreenBufferInfo
57	if err := windows.GetConsoleScreenBufferInfo(windows.Handle(fd), &info); err != nil {
58		return 0, 0, err
59	}
60	return int(info.Window.Right - info.Window.Left + 1), int(info.Window.Bottom - info.Window.Top + 1), nil
61}
62
63func readPassword(fd uintptr) ([]byte, error) {
64	var st uint32
65	if err := windows.GetConsoleMode(windows.Handle(fd), &st); err != nil {
66		return nil, err
67	}
68	old := st
69
70	st &^= (windows.ENABLE_ECHO_INPUT | windows.ENABLE_LINE_INPUT)
71	st |= (windows.ENABLE_PROCESSED_OUTPUT | windows.ENABLE_PROCESSED_INPUT)
72	if err := windows.SetConsoleMode(windows.Handle(fd), st); err != nil {
73		return nil, err
74	}
75
76	defer windows.SetConsoleMode(windows.Handle(fd), old)
77
78	var h windows.Handle
79	p, _ := windows.GetCurrentProcess()
80	if err := windows.DuplicateHandle(p, windows.Handle(fd), p, &h, 0, false, windows.DUPLICATE_SAME_ACCESS); err != nil {
81		return nil, err
82	}
83
84	f := os.NewFile(uintptr(h), "stdin")
85	defer f.Close()
86	return readPasswordLine(f)
87}