1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
// Tests and benchmarks performed with default settings only.
package defaults
import (
"fmt"
"io"
"io/ioutil"
"os"
"testing"
"github.com/rfjakob/gocryptfs/tests/test_helpers"
)
// Benchmarks
func BenchmarkStreamWrite(t *testing.B) {
buf := make([]byte, 1024*1024)
t.SetBytes(int64(len(buf)))
file, err := os.Create(test_helpers.DefaultPlainDir + "/BenchmarkWrite")
if err != nil {
t.Fatal(err)
}
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.Fatal(err)
}
}
file.Close()
}
func BenchmarkStreamRead(t *testing.B) {
buf := make([]byte, 1024*1024)
t.SetBytes(int64(len(buf)))
fn := test_helpers.DefaultPlainDir + "/BenchmarkWrite"
fi, err := os.Stat(fn)
if err != nil {
t.Fatal(err)
}
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)
var f2 *os.File
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()
}
file, err := os.Open(fn)
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.Println("Test file too small")
t.SkipNow()
} else if err != nil {
fmt.Println(err)
t.FailNow()
}
}
file.Close()
}
// createFiles - create "count" files of size "size" bytes each
func createFiles(t *testing.B, count int, size int) {
dir := fmt.Sprintf("%s/createFiles_%d_%d", test_helpers.DefaultPlainDir, count, size)
err := os.Mkdir(dir, 0777)
if err != nil {
t.Fatal(err)
}
buf := make([]byte, size)
t.SetBytes(int64(len(buf)))
t.ResetTimer()
var i int
for i = 0; i < count; i++ {
file := fmt.Sprintf("%s/%d", dir, i)
if size > 0 {
err = ioutil.WriteFile(file, buf, 0666)
} else {
var fh *os.File
fh, err = os.Create(file)
fh.Close()
}
if err != nil {
t.Fatal(err)
}
}
t.StopTimer()
os.RemoveAll(dir)
}
func BenchmarkCreate0B(t *testing.B) {
createFiles(t, t.N, 0)
}
func BenchmarkCreate1B(t *testing.B) {
createFiles(t, t.N, 1)
}
func BenchmarkCreate100B(t *testing.B) {
createFiles(t, t.N, 100)
}
func BenchmarkCreate4kB(t *testing.B) {
createFiles(t, t.N, 4*1024)
}
func BenchmarkCreate10kB(t *testing.B) {
createFiles(t, t.N, 10*1024)
}
|
