mirror of
https://github.com/reactos/syzkaller.git
synced 2024-10-08 09:43:21 +00:00
tools/syz-db: add new utility
The utility allows to pack/unpack corpus database to/from seprate files.
This commit is contained in:
parent
c91904333f
commit
5d3f1d6614
94
tools/syz-db/syz-db.go
Normal file
94
tools/syz-db/syz-db.go
Normal file
@ -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)
|
||||
}
|
Loading…
Reference in New Issue
Block a user