aboutsummaryrefslogtreecommitdiff
path: root/integration_tests
diff options
context:
space:
mode:
Diffstat (limited to 'integration_tests')
-rw-r--r--integration_tests/helpers.go94
-rw-r--r--integration_tests/main_test.go308
-rw-r--r--integration_tests/performance_test.go88
3 files changed, 490 insertions, 0 deletions
diff --git a/integration_tests/helpers.go b/integration_tests/helpers.go
new file mode 100644
index 0000000..9bfd9fc
--- /dev/null
+++ b/integration_tests/helpers.go
@@ -0,0 +1,94 @@
+package integration_tests
+
+import (
+ "os"
+ "os/exec"
+ "fmt"
+ "io/ioutil"
+ "crypto/md5"
+ "encoding/hex"
+ "testing"
+)
+
+const tmpDir = "/tmp/gocryptfs_main_test/"
+// Mountpoint
+const plainDir = tmpDir + "plain/"
+const cipherDir = tmpDir + "cipher/"
+
+func resetTmpDir() {
+ fu := exec.Command("fusermount", "-z", "-u", plainDir)
+ fu.Run()
+
+ os.RemoveAll(tmpDir)
+
+ err := os.MkdirAll(plainDir, 0777)
+ if err != nil {
+ panic("Could not create plainDir")
+ }
+
+ err = os.MkdirAll(cipherDir, 0777)
+ if err != nil {
+ panic("Could not create cipherDir")
+ }
+}
+
+func mount(extraArgs ...string) {
+ var args []string
+ args = append(args, extraArgs...)
+ //args = append(args, "--fusedebug")
+ args = append(args, cipherDir)
+ args = append(args, plainDir)
+ c := exec.Command("../gocryptfs", args...)
+ // Warning messages clutter the test output. Uncomment if you want to debug
+ // failures.
+ //c.Stdout = os.Stdout
+ //c.Stderr = os.Stderr
+ err := c.Run()
+ if err != nil {
+ fmt.Println(err)
+ os.Exit(1)
+ }
+}
+
+func unmount() error {
+ fu := exec.Command("fusermount", "-z", "-u", plainDir)
+ fu.Stdout = os.Stdout
+ fu.Stderr = os.Stderr
+ return fu.Run()
+}
+
+// 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)
+ }
+}
diff --git a/integration_tests/main_test.go b/integration_tests/main_test.go
new file mode 100644
index 0000000..222c192
--- /dev/null
+++ b/integration_tests/main_test.go
@@ -0,0 +1,308 @@
+package integration_tests
+
+import (
+ "flag"
+ "bytes"
+ "crypto/md5"
+ "encoding/hex"
+ "fmt"
+ "io/ioutil"
+ "os"
+ "runtime"
+ "sync"
+ "testing"
+)
+
+var plaintextNames bool
+
+// This is the entry point for the tests
+func TestMain(m *testing.M) {
+ var defaultonly bool
+ flag.BoolVar(&defaultonly, "defaultonly", false, "Only test default configuration (openssl=true, plaintextnames=false)")
+ flag.Parse()
+
+ if testing.Verbose() {
+ fmt.Printf("***** Testing with OpenSSL\n")
+ }
+ resetTmpDir()
+ mount("--zerokey")
+ r := m.Run()
+ unmount()
+
+ if defaultonly {
+ os.Exit(r)
+ }
+
+
+ if testing.Verbose() {
+ fmt.Printf("***** Testing with native Go crypto\n")
+ }
+ resetTmpDir()
+ mount("--zerokey", "--openssl=false")
+ r = m.Run()
+ unmount()
+
+ if testing.Verbose() {
+ fmt.Printf("***** Testing \"--plaintextnames\"\n")
+ }
+ resetTmpDir()
+ mount("--zerokey", "--plaintextnames")
+ plaintextNames = true
+ r = m.Run()
+ unmount()
+
+ os.Exit(r)
+}
+
+// Write "n" zero bytes to filename "fn", read again, compare hash
+func testWriteN(t *testing.T, fn string, n int) string {
+ file, err := os.Create(plainDir + fn)
+ if err != nil {
+ t.FailNow()
+ }
+
+ d := make([]byte, n)
+ written, err := file.Write(d)
+ if err != nil || written != len(d) {
+ fmt.Printf("err=\"%s\", written=%d\n", err, written)
+ t.Fail()
+ }
+ file.Close()
+
+ verifySize(t, plainDir+fn, n)
+
+ bin := md5.Sum(d)
+ hashWant := hex.EncodeToString(bin[:])
+
+ hashActual := md5fn(plainDir + fn)
+
+ if hashActual != hashWant {
+ t.Errorf("Wrong content, hashWant=%s hashActual=%s\n", hashWant, hashActual)
+ }
+
+ return hashActual
+}
+
+func TestWrite10(t *testing.T) {
+ testWriteN(t, "10", 10)
+}
+
+func TestWrite100(t *testing.T) {
+ testWriteN(t, "100", 100)
+}
+
+func TestWrite1M(t *testing.T) {
+ testWriteN(t, "1M", 1024*1024)
+}
+
+func TestWrite1Mx100(t *testing.T) {
+ hashWant := testWriteN(t, "1Mx100", 1024*1024)
+ // Read and check 100 times to catch race conditions
+ var i int
+ for i = 0; i < 100; i++ {
+ hashActual := md5fn(plainDir + "1M")
+ if hashActual != hashWant {
+ fmt.Printf("Read corruption in loop # %d\n", i)
+ t.FailNow()
+ } else {
+ //fmt.Print(".")
+ }
+ }
+}
+
+func TestTruncate(t *testing.T) {
+ fn := plainDir + "truncate"
+ file, err := os.Create(fn)
+ if err != nil {
+ t.FailNow()
+ }
+ // Grow to two blocks
+ file.Truncate(7000)
+ verifySize(t, fn, 7000)
+ if md5fn(fn) != "95d4ec7038e3e4fdbd5f15c34c3f0b34" {
+ t.Errorf("wrong content")
+ }
+ // Shrink - needs RMW
+ file.Truncate(6999)
+ verifySize(t, fn, 6999)
+ if md5fn(fn) != "35fd15873ec6c35380064a41b9b9683b" {
+ t.Errorf("wrong content")
+ }
+ // Shrink to one partial block
+ file.Truncate(465)
+ verifySize(t, fn, 465)
+ if md5fn(fn) != "a1534d6e98a6b21386456a8f66c55260" {
+ t.Errorf("wrong content")
+ }
+ // Grow to exactly one block
+ file.Truncate(4096)
+ verifySize(t, fn, 4096)
+ if md5fn(fn) != "620f0b67a91f7f74151bc5be745b7110" {
+ t.Errorf("wrong content")
+ }
+}
+
+func TestAppend(t *testing.T) {
+ fn := plainDir + "append"
+ file, err := os.Create(fn)
+ if err != nil {
+ t.FailNow()
+ }
+ data := []byte("testdata123456789") // length 17
+ var buf bytes.Buffer
+ var hashWant string
+ for i := 0; i <= 500; i++ {
+ file.Write(data)
+ buf.Write(data)
+ bin := md5.Sum(buf.Bytes())
+ hashWant = hex.EncodeToString(bin[:])
+ hashActual := md5fn(fn)
+ if hashWant != hashActual {
+ t.FailNow()
+ }
+ }
+
+ // Overwrite with the same data
+ // Hash must stay the same
+ file.Seek(0, 0)
+ for i := 0; i <= 500; i++ {
+ file.Write(data)
+ hashActual := md5fn(fn)
+ if hashWant != hashActual {
+ t.FailNow()
+ }
+ }
+}
+
+// Create a file with holes by writing to offset 0 (block #0) and
+// offset 4096 (block #1).
+func TestFileHoles(t *testing.T) {
+ fn := plainDir + "fileholes"
+ file, err := os.Create(fn)
+ if err != nil {
+ t.Errorf("file create failed")
+ }
+ foo := []byte("foo")
+ file.Write(foo)
+ file.WriteAt(foo, 4096)
+ _, err = ioutil.ReadFile(fn)
+ if err != nil {
+ t.Error(err)
+ }
+}
+
+func sContains(haystack []string, needle string) bool {
+ for _, element := range haystack {
+ if element == needle {
+ return true
+ }
+ }
+ return false
+}
+
+func TestRmwRace(t *testing.T) {
+
+ runtime.GOMAXPROCS(10)
+
+ fn := plainDir + "rmwrace"
+ f1, err := os.Create(fn)
+ if err != nil {
+ t.Fatalf("file create failed")
+ }
+ f2, err := os.Create(fn)
+ if err != nil {
+ t.Fatalf("file create failed")
+ }
+
+ oldBlock := bytes.Repeat([]byte("o"), 4096)
+
+ newBlock := bytes.Repeat([]byte("n"), 4096)
+
+ shortBlock := bytes.Repeat([]byte("s"), 16)
+
+ mergedBlock := make([]byte, 4096)
+ copy(mergedBlock, newBlock)
+ copy(mergedBlock[4080:], shortBlock)
+
+ goodMd5 := make(map[string]int)
+
+ for i := 0; i < 1000; i++ {
+ // Reset to [ooooooooo]
+ _, err = f1.WriteAt(oldBlock, 0)
+ if err != nil {
+ t.Fatalf("Write failed")
+ }
+
+ var wg sync.WaitGroup
+ wg.Add(2)
+
+ // Write to the end of the file, [....ssss]
+ go func() {
+ f1.WriteAt(shortBlock, 4080)
+ wg.Done()
+ }()
+
+ // Overwrite to [nnnnnnn]
+ go func() {
+ f2.WriteAt(newBlock, 0)
+ wg.Done()
+ }()
+
+ wg.Wait()
+
+ // The file should be either:
+ // [nnnnnnnnnn] (md5: 6c1660fdabccd448d1359f27b3db3c99) or
+ // [nnnnnnssss] (md5: da885006a6a284530a427c73ce1e5c32)
+ // but it must not be
+ // [oooooossss]
+
+ buf, _ := ioutil.ReadFile(fn)
+ m := md5hex(buf)
+ goodMd5[m] = goodMd5[m] + 1
+
+ /*
+ if m == "6c1660fdabccd448d1359f27b3db3c99" {
+ fmt.Println(hex.Dump(buf))
+ t.FailNow()
+ }
+ */
+ }
+ if testing.Verbose() {
+ fmt.Println(goodMd5)
+ }
+}
+
+// With "--plaintextnames", the name "/gocryptfs.conf" is reserved.
+// Otherwise there should be no restrictions.
+func TestFiltered(t *testing.T) {
+ filteredFile := plainDir + "gocryptfs.conf"
+ file, err := os.Create(filteredFile)
+ if plaintextNames == true && err == nil {
+ fmt.Errorf("should have failed but didn't")
+ } else if plaintextNames == false && err != nil {
+ t.Error(err)
+ }
+ file.Close()
+
+ err = os.Remove(filteredFile)
+ if plaintextNames == true && err == nil {
+ fmt.Errorf("should have failed but didn't")
+ } else if plaintextNames == false && err != nil {
+ t.Error(err)
+ }
+}
+
+func TestFilenameEncryption(t *testing.T) {
+ file, err := os.Create(plainDir + "TestFilenameEncryption.txt")
+ file.Close()
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ _, err = os.Stat(cipherDir + "TestFilenameEncryption.txt")
+ if plaintextNames == true && err != nil {
+ t.Errorf("plaintextnames not working: %v", err)
+ } else if plaintextNames == false && err == nil {
+ t.Errorf("file name encryption not working")
+ }
+}
diff --git a/integration_tests/performance_test.go b/integration_tests/performance_test.go
new file mode 100644
index 0000000..e4a618a
--- /dev/null
+++ b/integration_tests/performance_test.go
@@ -0,0 +1,88 @@
+package integration_tests
+
+import (
+ "os"
+ "os/exec"
+ "fmt"
+ "io"
+ "testing"
+)
+
+func BenchmarkStreamWrite(t *testing.B) {
+ buf := make([]byte, 1024*1024)
+ t.SetBytes(int64(len(buf)))
+
+ file, err := os.Create(plainDir + "BenchmarkWrite")
+ if err != nil {
+ t.FailNow()
+ }
+
+ t.ResetTimer()
+ var i int
+ for i = 0; i < t.N; i++ {
+ written, err := file.Write(buf)
+ if err != nil {
+ fmt.Printf("err=\"%s\", written=%d\n", err.Error(), written)
+ t.FailNow()
+ }
+ }
+ file.Close()
+}
+
+func BenchmarkStreamRead(t *testing.B) {
+ buf := make([]byte, 1024*1024)
+ t.SetBytes(int64(len(buf)))
+
+ fn := plainDir + "BenchmarkWrite"
+ fi, _ := os.Stat(fn)
+ mb := int(fi.Size() / 1024 / 1024)
+
+ if t.N > mb {
+ // Grow file so we can satisfy the test
+ //fmt.Printf("Growing file to %d MB... ", t.N)
+ f2, err := os.OpenFile(fn, os.O_WRONLY|os.O_APPEND, 0666)
+ if err != nil {
+ fmt.Println(err)
+ t.FailNow()
+ }
+ for h := 0; h < t.N-mb; h++ {
+ _, err = f2.Write(buf)
+ if err != nil {
+ fmt.Println(err)
+ t.FailNow()
+ }
+ }
+ f2.Close()
+ //fmt.Printf("done\n")
+ }
+
+ file, err := os.Open(plainDir + "BenchmarkWrite")
+ if err != nil {
+ t.FailNow()
+ }
+ t.ResetTimer()
+ var i int
+ for i = 0; i < t.N; i++ {
+ _, err := file.Read(buf)
+ if err == io.EOF {
+ fmt.Printf("Test file too small\n")
+ t.SkipNow()
+ } else if err != nil {
+ fmt.Println(err)
+ t.FailNow()
+ }
+ }
+ file.Close()
+}
+
+func BenchmarkUntar(t *testing.B) {
+ t.SetBytes(422229778)
+ c := exec.Command("tar", "xf", "/tmp/linux-3.0.tar.gz", "-C", plainDir)
+ c.Stdout = os.Stdout
+ c.Stderr = os.Stderr
+ t.ResetTimer()
+ err := c.Run()
+ if err != nil {
+ t.Fatal(err)
+ }
+}