diff options
-rw-r--r-- | tests/cli/cli_test.go | 71 |
1 files changed, 63 insertions, 8 deletions
diff --git a/tests/cli/cli_test.go b/tests/cli/cli_test.go index 8f85393..c6e86da 100644 --- a/tests/cli/cli_test.go +++ b/tests/cli/cli_test.go @@ -167,32 +167,85 @@ func TestPasswd(t *testing.T) { } } +// cp copies file at `src` to `dst`, overwriting +// `dst` if it already exists. Calls t.Fatal on failure. +func cp(t *testing.T, src string, dst string) { + conf, err := ioutil.ReadFile(src) + if err != nil { + t.Fatal(err) + } + syscall.Unlink(dst) + err = ioutil.WriteFile(dst, conf, 0600) + if err != nil { + t.Fatal(err) + } +} + // Test -passwd with -masterkey func TestPasswdMasterkey(t *testing.T) { // Create FS dir := test_helpers.InitFS(t) // Overwrite with config with known master key - conf, err := ioutil.ReadFile("gocryptfs.conf.b9e5ba23") + cp(t, "gocryptfs.conf.b9e5ba23", dir+"/gocryptfs.conf") + // Add content + mnt := dir + ".mnt" + test_helpers.MountOrFatal(t, dir, mnt, "-extpass", "echo test") + file1 := mnt + "/file1" + err := ioutil.WriteFile(file1, []byte("somecontent"), 0600) if err != nil { t.Fatal(err) } - syscall.Unlink(dir + "/gocryptfs.conf") - err = ioutil.WriteFile(dir+"/gocryptfs.conf", conf, 0600) + test_helpers.UnmountPanic(mnt) + // Change password using stdin + args := []string{"-q", "-passwd", "-masterkey", + "b9e5ba23-981a22b8-c8d790d8-627add29-f680513f-b7b7035f-d203fb83-21d82205"} + args = append(args, dir) + cmd := exec.Command(test_helpers.GocryptfsBinary, args...) + cmd.Stdout = os.Stdout + cmd.Stderr = os.Stderr + p, err := cmd.StdinPipe() if err != nil { t.Fatal(err) } + err = cmd.Start() + if err != nil { + t.Error(err) + } + // New password + p.Write([]byte("newpasswd\n")) + p.Close() + err = cmd.Wait() + if err != nil { + t.Error(err) + } + // Mount and verify + test_helpers.MountOrFatal(t, dir, mnt, "-extpass", "echo newpasswd") + content, err := ioutil.ReadFile(file1) + if err != nil { + t.Error(err) + } else if string(content) != "somecontent" { + t.Errorf("wrong content: %q", string(content)) + } + test_helpers.UnmountPanic(mnt) +} + +// Test -passwd with -masterkey=stdin +func TestPasswdMasterkeyStdin(t *testing.T) { + // Create FS + dir := test_helpers.InitFS(t) + // Overwrite with config with known master key + cp(t, "gocryptfs.conf.b9e5ba23", dir+"/gocryptfs.conf") // Add content mnt := dir + ".mnt" test_helpers.MountOrFatal(t, dir, mnt, "-extpass", "echo test") file1 := mnt + "/file1" - err = ioutil.WriteFile(file1, []byte("somecontent"), 0600) + err := ioutil.WriteFile(file1, []byte("somecontent"), 0600) if err != nil { t.Fatal(err) } test_helpers.UnmountPanic(mnt) // Change password using stdin - args := []string{"-q", "-passwd", "-masterkey", - "b9e5ba23-981a22b8-c8d790d8-627add29-f680513f-b7b7035f-d203fb83-21d82205"} + args := []string{"-q", "-passwd", "-masterkey=stdin"} args = append(args, dir) cmd := exec.Command(test_helpers.GocryptfsBinary, args...) cmd.Stdout = os.Stdout @@ -205,18 +258,20 @@ func TestPasswdMasterkey(t *testing.T) { if err != nil { t.Error(err) } + // Masterkey + p.Write([]byte("b9e5ba23-981a22b8-c8d790d8-627add29-f680513f-b7b7035f-d203fb83-21d82205\n")) // New password p.Write([]byte("newpasswd\n")) p.Close() err = cmd.Wait() if err != nil { - t.Error(err) + t.Fatal(err) } // Mount and verify test_helpers.MountOrFatal(t, dir, mnt, "-extpass", "echo newpasswd") content, err := ioutil.ReadFile(file1) if err != nil { - t.Error(err) + t.Fatal(err) } else if string(content) != "somecontent" { t.Errorf("wrong content: %q", string(content)) } |