From c7e0d50a76c1970d8f326a49ef24fc5d19ee93c1 Mon Sep 17 00:00:00 2001 From: Dmitry Vyukov Date: Tue, 27 Mar 2018 09:39:02 +0200 Subject: tools/syz-db: allow to specify db version in pack --- tools/syz-db/syz-db.go | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) (limited to 'tools') diff --git a/tools/syz-db/syz-db.go b/tools/syz-db/syz-db.go index 0ea8a9c10..d777c37a6 100644 --- a/tools/syz-db/syz-db.go +++ b/tools/syz-db/syz-db.go @@ -4,6 +4,7 @@ package main import ( + "flag" "fmt" "io/ioutil" "os" @@ -17,14 +18,19 @@ import ( ) func main() { - if len(os.Args) != 4 { + var ( + flagVersion = flag.Uint64("version", 0, "database version") + ) + flag.Parse() + args := flag.Args() + if len(args) != 3 { usage() } - switch os.Args[1] { + switch args[0] { case "pack": - pack(os.Args[2], os.Args[3]) + pack(args[1], args[2], *flagVersion) case "unpack": - unpack(os.Args[2], os.Args[3]) + unpack(args[1], args[2]) default: usage() } @@ -37,7 +43,7 @@ func usage() { os.Exit(1) } -func pack(dir, file string) { +func pack(dir, file string, version uint64) { files, err := ioutil.ReadDir(dir) if err != nil { failf("failed to read dir: %v", err) @@ -47,6 +53,9 @@ func pack(dir, file string) { if err != nil { failf("failed to open database file: %v", err) } + if err := db.BumpVersion(version); err != nil { + failf("failed to bump database version: %v", err) + } for _, file := range files { data, err := ioutil.ReadFile(filepath.Join(dir, file.Name())) if err != nil { -- cgit mrf-deployment