2 files changed, 51 insertions, 0 deletions
diff --git a/pkg/db/db.go b/pkg/db/db.go
index 17723b224..7961c0797 100644
--- a/pkg/db/db.go
+++ b/pkg/db/db.go
@@ -17,6 +17,7 @@ import (
 	"io"
 	"io/ioutil"
 	"os"
+	"path/filepath"
 
 	"github.com/google/syzkaller/pkg/hash"
 	"github.com/google/syzkaller/pkg/log"
@@ -68,6 +69,41 @@ func (db *DB) Save(key string, val []byte, seq uint64) {
 	db.uncompacted++
 }
 
+// Load the test progs in the given directory store them inside db.
+func (db *DB) LoadTestAsSeed(target *prog.Target, seedsDir string) {
+	var files []string
+	err := filepath.Walk(seedsDir, func(path string, info os.FileInfo, err error) error {
+		if !info.IsDir() {
+			files = append(files, path)
+		}
+		return nil
+	})
+	if err != nil {
+		panic(err)
+	}
+	progEntries := loadPrograms(target, files)
+	if len(progEntries) == 0 {
+		return
+	}
+	for _, progEntry := range progEntries {
+		prog := progEntry.P.Serialize()
+		sig := hash.String(prog)
+		db.Save(sig, prog, 0)
+	}
+}
+
+func loadPrograms(target *prog.Target, files []string) []*prog.LogEntry {
+	var entries []*prog.LogEntry
+	for _, fn := range files {
+		data, err := ioutil.ReadFile(fn)
+		if err != nil {
+			log.Fatalf("failed to read log file: %v", err)
+		}
+		entries = append(entries, target.ParseLog(data)...)
+	}
+	return entries
+}
+
 func (db *DB) Delete(key string) {
 	if _, ok := db.Records[key]; !ok {
 		return
diff --git a/pkg/osutil/osutil.go b/pkg/osutil/osutil.go
index 47fcc8b7b..1eafafcb9 100644
--- a/pkg/osutil/osutil.go
+++ b/pkg/osutil/osutil.go
@@ -222,6 +222,21 @@ func LinkFiles(srcDir, dstDir string, files map[string]bool) error {
 	return nil
 }
 
+// LinkDir creates soft links from dstDir to srcDir.
+// All other files in dstDir are removed.
+func LinkDir(srcDir, dstDir string) error {
+	if err := os.RemoveAll(dstDir); err != nil {
+		return err
+	}
+	if err := MkdirAll(dstDir); err != nil {
+		return err
+	}
+	if err := os.Symlink(srcDir, dstDir); err != nil {
+		return err
+	}
+	return nil
+}
+
 func MkdirAll(dir string) error {
 	return os.MkdirAll(dir, DefaultDirPerm)
 }