mirror of
https://github.com/reactos/syzkaller.git
synced 2025-02-25 14:11:14 +00:00
c1641491e4
This is needed for both tools/syz-db and tools/syz-trace2syz. Also, remove code to resolve SHA1 collisions. Also, don't set db version as we actually want to minimize and smash these programs like anything else (not minimizing nor smashing them is only useful during tool testing).
120 lines
2.7 KiB
Go
120 lines
2.7 KiB
Go
// 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 (
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/google/syzkaller/pkg/db"
|
|
"github.com/google/syzkaller/pkg/hash"
|
|
"github.com/google/syzkaller/pkg/osutil"
|
|
"github.com/google/syzkaller/prog"
|
|
_ "github.com/google/syzkaller/sys"
|
|
)
|
|
|
|
func main() {
|
|
var (
|
|
flagVersion = flag.Uint64("version", 0, "database version")
|
|
flagOS = flag.String("os", "", "target OS")
|
|
flagArch = flag.String("arch", "", "target arch")
|
|
)
|
|
flag.Parse()
|
|
args := flag.Args()
|
|
if len(args) != 3 {
|
|
usage()
|
|
}
|
|
var target *prog.Target
|
|
if *flagOS != "" || *flagArch != "" {
|
|
var err error
|
|
target, err = prog.GetTarget(*flagOS, *flagArch)
|
|
if err != nil {
|
|
failf("failed to find target: %v", err)
|
|
}
|
|
}
|
|
switch args[0] {
|
|
case "pack":
|
|
pack(args[1], args[2], target, *flagVersion)
|
|
case "unpack":
|
|
unpack(args[1], args[2])
|
|
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, target *prog.Target, version uint64) {
|
|
files, err := ioutil.ReadDir(dir)
|
|
if err != nil {
|
|
failf("failed to read dir: %v", err)
|
|
}
|
|
var records []db.Record
|
|
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 {
|
|
if target != nil {
|
|
p, err := target.Deserialize(data)
|
|
if err != nil {
|
|
failf("failed to deserialize %v: %v", file.Name(), err)
|
|
}
|
|
data = p.Serialize()
|
|
sig = hash.String(data)
|
|
}
|
|
fmt.Fprintf(os.Stderr, "fixing hash %v -> %v\n", key, sig)
|
|
key = sig
|
|
}
|
|
records = append(records, db.Record{
|
|
Val: data,
|
|
Seq: seq,
|
|
})
|
|
}
|
|
if err := db.Create(file, version, records); err != nil {
|
|
failf("%v", err)
|
|
}
|
|
}
|
|
|
|
func unpack(file, dir string) {
|
|
db, err := db.Open(file)
|
|
if err != nil {
|
|
failf("failed to open database: %v", err)
|
|
}
|
|
osutil.MkdirAll(dir)
|
|
for key, rec := range db.Records {
|
|
fname := filepath.Join(dir, key)
|
|
if rec.Seq != 0 {
|
|
fname += fmt.Sprintf("-%v", rec.Seq)
|
|
}
|
|
if err := osutil.WriteFile(fname, rec.Val); 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)
|
|
}
|