115 lines
2.4 KiB
Go
115 lines
2.4 KiB
Go
package integration_tests
|
|
|
|
import (
|
|
"crypto/md5"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"testing"
|
|
)
|
|
|
|
// Note: the code assumes that all have a trailing slash
|
|
const tmpDir = "/tmp/gocryptfs_main_test/"
|
|
const defaultPlainDir = tmpDir + "plain/"
|
|
const defaultCipherDir = tmpDir + "cipher/"
|
|
|
|
const gocryptfsBinary = "../gocryptfs"
|
|
|
|
func resetTmpDir() {
|
|
fu := exec.Command("fusermount", "-z", "-u", defaultPlainDir)
|
|
fu.Run()
|
|
|
|
err := os.RemoveAll(tmpDir)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
err = os.MkdirAll(defaultPlainDir, 0777)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
err = os.MkdirAll(defaultCipherDir, 0777)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
dirIV := make([]byte, 16)
|
|
err = ioutil.WriteFile(defaultCipherDir+"gocryptfs.diriv", dirIV, 0444)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
// mount CIPHERDIR "c" on PLAINDIR "p"
|
|
func mount(c string, p string, extraArgs ...string) {
|
|
var args []string
|
|
args = append(args, extraArgs...)
|
|
//args = append(args, "--fusedebug")
|
|
args = append(args, c)
|
|
args = append(args, p)
|
|
cmd := exec.Command(gocryptfsBinary, args...)
|
|
if testing.Verbose() {
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
}
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
// unmount PLAINDIR "p"
|
|
func unmount(p string) error {
|
|
fu := exec.Command("fusermount", "-u", "-z", p)
|
|
fu.Stdout = os.Stdout
|
|
fu.Stderr = os.Stderr
|
|
err := fu.Run()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
return err
|
|
}
|
|
|
|
// Return md5 string for file "filename"
|
|
func md5fn(filename string) string {
|
|
buf, err := ioutil.ReadFile(filename)
|
|
if err != nil {
|
|
fmt.Printf("ReadFile: %v\n", err)
|
|
return ""
|
|
}
|
|
return md5hex(buf)
|
|
}
|
|
|
|
// Return md5 string for "buf"
|
|
func md5hex(buf []byte) string {
|
|
rawHash := md5.Sum(buf)
|
|
hash := hex.EncodeToString(rawHash[:])
|
|
return hash
|
|
}
|
|
|
|
// Verify that the file size equals "want". This checks:
|
|
// 1) Size reported by Stat()
|
|
// 2) Number of bytes returned when reading the whole file
|
|
func verifySize(t *testing.T, path string, want int) {
|
|
buf, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
t.Errorf("ReadFile failed: %v", err)
|
|
} else if len(buf) != want {
|
|
t.Errorf("wrong read size: got=%d want=%d", len(buf), want)
|
|
}
|
|
|
|
fi, err := os.Stat(path)
|
|
if err != nil {
|
|
t.Errorf("Stat failed: %v", err)
|
|
} else if fi.Size() != int64(want) {
|
|
t.Errorf("wrong stat file size, got=%d want=%d", fi.Size(), want)
|
|
}
|
|
}
|