60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"golang.org/x/crypto/ssh/terminal"
|
|
)
|
|
|
|
func readPasswordTwice(extpass string) string {
|
|
fmt.Printf("Password: ")
|
|
p1 := readPassword(extpass)
|
|
fmt.Printf("Repeat: ")
|
|
p2 := readPassword(extpass)
|
|
if p1 != p2 {
|
|
fmt.Printf("Passwords do not match\n")
|
|
os.Exit(ERREXIT_PASSWORD)
|
|
}
|
|
return p1
|
|
}
|
|
|
|
// readPassword - get password from terminal
|
|
// or from the "extpass" program
|
|
func readPassword(extpass string) string {
|
|
var password string
|
|
var err error
|
|
var output []byte
|
|
if extpass != "" {
|
|
parts := strings.Split(extpass, " ")
|
|
cmd := exec.Command(parts[0], parts[1:]...)
|
|
cmd.Stderr = os.Stderr
|
|
output, err = cmd.Output()
|
|
if err != nil {
|
|
fmt.Printf("extpass program returned error: %v\n", err)
|
|
os.Exit(ERREXIT_PASSWORD)
|
|
}
|
|
fmt.Printf("(extpass)\n")
|
|
// Trim trailing newline like terminal.ReadPassword() does
|
|
if output[len(output)-1] == '\n' {
|
|
output = output[:len(output)-1]
|
|
}
|
|
} else {
|
|
fd := int(os.Stdin.Fd())
|
|
output, err = terminal.ReadPassword(fd)
|
|
if err != nil {
|
|
fmt.Printf("Error: Could not read password from terminal: %v\n", err)
|
|
os.Exit(ERREXIT_PASSWORD)
|
|
}
|
|
fmt.Printf("\n")
|
|
}
|
|
password = string(output)
|
|
if password == "" {
|
|
fmt.Printf("Error: password is empty\n")
|
|
os.Exit(ERREXIT_PASSWORD)
|
|
}
|
|
return password
|
|
}
|