tools/syz-db: add new utility

The utility allows to pack/unpack corpus database to/from seprate files.

1 files changed, 94 insertions, 0 deletions

diff --git a/tools/syz-db/syz-db.go b/tools/syz-db/syz-db.go
new file mode 100644
index 000000000..d11e73102
--- /dev/null
+++ b/tools/syz-db/syz-db.go
@@ -0,0 +1,94 @@
+// Copyright 2017 syzkaller project authors. All rights reserved.
+// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
+
+package main
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"strconv"
+	"strings"
+
+	"github.com/google/syzkaller/db"
+	"github.com/google/syzkaller/hash"
+)
+
+func main() {
+	if len(os.Args) != 4 {
+		usage()
+	}
+	switch os.Args[1] {
+	case "pack":
+		pack(os.Args[2], os.Args[3])
+	case "unpack":
+		unpack(os.Args[2], os.Args[3])
+	default:
+		usage()
+	}
+}
+
+func usage() {
+	fmt.Fprintf(os.Stderr, "usage:\n")
+	fmt.Fprintf(os.Stderr, "  syz-db pack dir corpus.db\n")
+	fmt.Fprintf(os.Stderr, "  syz-db unpack corpus.db dir\n")
+	os.Exit(1)
+}
+
+func pack(dir, file string) {
+	files, err := ioutil.ReadDir(dir)
+	if err != nil {
+		failf("failed to read dir: %v", err)
+	}
+	os.Remove(file)
+	db, err := db.Open(file)
+	if err != nil {
+		failf("failed to open database file: %v", err)
+	}
+	for _, file := range files {
+		data, err := ioutil.ReadFile(filepath.Join(dir, file.Name()))
+		if err != nil {
+			failf("failed to read file %v: %v", file.Name(), err)
+		}
+		var seq uint64
+		key := file.Name()
+		if parts := strings.Split(file.Name(), "-"); len(parts) == 2 {
+			var err error
+
+			if seq, err = strconv.ParseUint(parts[1], 10, 64); err == nil {
+				key = parts[0]
+			}
+		}
+		if sig := hash.String(data); key != sig {
+			fmt.Fprintf(os.Stderr, "fixing hash %v -> %v\n", key, sig)
+			key = sig
+		}
+		db.Save(key, data, seq)
+	}
+	if err := db.Flush(); err != nil {
+		failf("failed to save database file: %v", err)
+	}
+}
+
+func unpack(file, dir string) {
+	db, err := db.Open(file)
+	if err != nil {
+		failf("failed to open database: %v", err)
+	}
+	os.Mkdir(dir, 0750)
+	for key, rec := range db.Records {
+		fname := filepath.Join(dir, key)
+		if rec.Seq != 0 {
+			fname += fmt.Sprintf("-%v", rec.Seq)
+		}
+		if err := ioutil.WriteFile(fname, rec.Val, 0640); err != nil {
+			failf("failed to output file: %v", err)
+		}
+	}
+}
+
+func failf(msg string, args ...interface{}) {
+	fmt.Fprintf(os.Stderr, msg+"\n", args...)
+	os.Exit(1)
+}