diff options
Diffstat (limited to 'internal/readpassword')
-rw-r--r-- | internal/readpassword/extpass_test.go | 8 | ||||
-rw-r--r-- | internal/readpassword/read.go | 19 | ||||
-rw-r--r-- | internal/readpassword/stdin_test.go | 4 |
3 files changed, 17 insertions, 14 deletions
diff --git a/internal/readpassword/extpass_test.go b/internal/readpassword/extpass_test.go index 4af775a..cdfea4e 100644 --- a/internal/readpassword/extpass_test.go +++ b/internal/readpassword/extpass_test.go @@ -8,6 +8,8 @@ import ( "github.com/rfjakob/gocryptfs/internal/tlog" ) +var testPw = []byte("test") + func TestMain(m *testing.M) { // Shut up info output tlog.Info.Enabled = false @@ -16,7 +18,7 @@ func TestMain(m *testing.M) { func TestExtpass(t *testing.T) { p1 := "ads2q4tw41reg52" - p2 := readPasswordExtpass("echo " + p1) + p2 := string(readPasswordExtpass("echo " + p1)) if p1 != p2 { t.Errorf("p1=%q != p2=%q", p1, p2) } @@ -24,7 +26,7 @@ func TestExtpass(t *testing.T) { func TestOnceExtpass(t *testing.T) { p1 := "lkadsf0923rdfi48rqwhdsf" - p2 := Once("echo " + p1) + p2 := string(Once("echo " + p1)) if p1 != p2 { t.Errorf("p1=%q != p2=%q", p1, p2) } @@ -32,7 +34,7 @@ func TestOnceExtpass(t *testing.T) { func TestTwiceExtpass(t *testing.T) { p1 := "w5w44t3wfe45srz434" - p2 := Once("echo " + p1) + p2 := string(Once("echo " + p1)) if p1 != p2 { t.Errorf("p1=%q != p2=%q", p1, p2) } diff --git a/internal/readpassword/read.go b/internal/readpassword/read.go index 481911b..047eba9 100644 --- a/internal/readpassword/read.go +++ b/internal/readpassword/read.go @@ -2,6 +2,7 @@ package readpassword import ( + "bytes" "fmt" "io" "os" @@ -23,7 +24,7 @@ const ( // Once tries to get a password from the user, either from the terminal, extpass // or stdin. -func Once(extpass string) string { +func Once(extpass string) []byte { if extpass != "" { return readPasswordExtpass(extpass) } @@ -35,7 +36,7 @@ func Once(extpass string) string { // Twice is the same as Once but will prompt twice if we get the password from // the terminal. -func Twice(extpass string) string { +func Twice(extpass string) []byte { if extpass != "" { return readPasswordExtpass(extpass) } @@ -44,7 +45,7 @@ func Twice(extpass string) string { } p1 := readPasswordTerminal("Password: ") p2 := readPasswordTerminal("Repeat: ") - if p1 != p2 { + if !bytes.Equal(p1, p2) { tlog.Fatal.Println("Passwords do not match") os.Exit(exitcodes.ReadPassword) } @@ -53,7 +54,7 @@ func Twice(extpass string) string { // readPasswordTerminal reads a line from the terminal. // Exits on read error or empty result. -func readPasswordTerminal(prompt string) string { +func readPasswordTerminal(prompt string) []byte { fd := int(os.Stdin.Fd()) fmt.Fprintf(os.Stderr, prompt) // terminal.ReadPassword removes the trailing newline @@ -67,12 +68,12 @@ func readPasswordTerminal(prompt string) string { tlog.Fatal.Println("Password is empty") os.Exit(exitcodes.PasswordEmpty) } - return string(p) + return p } // readPasswordStdin reads a line from stdin. // It exits with a fatal error on read error or empty result. -func readPasswordStdin() string { +func readPasswordStdin() []byte { tlog.Info.Println("Reading password from stdin") p := readLineUnbuffered(os.Stdin) if len(p) == 0 { @@ -85,7 +86,7 @@ func readPasswordStdin() string { // readPasswordExtpass executes the "extpass" program and returns the first line // of the output. // Exits on read error or empty result. -func readPasswordExtpass(extpass string) string { +func readPasswordExtpass(extpass string) []byte { tlog.Info.Println("Reading password from extpass program") var parts []string // The option "-passfile=FILE" gets transformed to @@ -125,7 +126,7 @@ func readPasswordExtpass(extpass string) string { // readLineUnbuffered reads single bytes from "r" util it gets "\n" or EOF. // The returned string does NOT contain the trailing "\n". -func readLineUnbuffered(r io.Reader) (l string) { +func readLineUnbuffered(r io.Reader) (l []byte) { b := make([]byte, 1) for { if len(l) > maxPasswordLen { @@ -146,7 +147,7 @@ func readLineUnbuffered(r io.Reader) (l string) { if b[0] == '\n' { return l } - l = l + string(b) + l = append(l, b...) } } diff --git a/internal/readpassword/stdin_test.go b/internal/readpassword/stdin_test.go index 2d9f93f..8cf9954 100644 --- a/internal/readpassword/stdin_test.go +++ b/internal/readpassword/stdin_test.go @@ -11,7 +11,7 @@ import ( func TestStdin(t *testing.T) { p1 := "g55434t55wef" if os.Getenv("TEST_SLAVE") == "1" { - p2 := readPasswordStdin() + p2 := string(readPasswordStdin()) if p1 != p2 { fmt.Fprintf(os.Stderr, "%q != %q", p1, p2) os.Exit(1) @@ -44,7 +44,7 @@ func TestStdin(t *testing.T) { func TestStdinEof(t *testing.T) { p1 := "asd45as5f4a36" if os.Getenv("TEST_SLAVE") == "1" { - p2 := readPasswordStdin() + p2 := string(readPasswordStdin()) if p1 != p2 { fmt.Fprintf(os.Stderr, "%q != %q", p1, p2) os.Exit(1) |