mirror of
https://github.com/reactos/syzkaller.git
synced 2024-11-27 13:20:34 +00:00
8af91f61b4
Currently hub allows managers to exchange programs from corpus. But reproducers are not exchanged and we don't know if a crash happens on other managers as well or not. Allow hub to exchange reproducers. Reproducers are stored in a separate db file with own sequence numbers. This allows to throttle distribution of reproducers to managers, so that they are not overloaded with reproducers and don't lose them on restarts. Based on patch by Andrey Konovalov: https://github.com/google/syzkaller/pull/325 Fixes #282
115 lines
3.2 KiB
Go
115 lines
3.2 KiB
Go
// Copyright 2016 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 state
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"testing"
|
|
)
|
|
|
|
func TestState(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "syz-hub-state-test")
|
|
if err != nil {
|
|
t.Fatalf("failed to create temp dir: %v", err)
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
st, err := Make(dir)
|
|
if err != nil {
|
|
t.Fatalf("failed to make state: %v", err)
|
|
}
|
|
_, _, err = st.Sync("foo", nil, nil)
|
|
if err == nil {
|
|
t.Fatalf("synced with unconnected manager")
|
|
}
|
|
calls := []string{"read", "write"}
|
|
if err := st.Connect("foo", false, calls, nil); err != nil {
|
|
t.Fatalf("Connect failed: %v", err)
|
|
}
|
|
_, _, err = st.Sync("foo", nil, nil)
|
|
if err != nil {
|
|
t.Fatalf("Sync failed: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestRepro(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "syz-hub-state-test")
|
|
if err != nil {
|
|
t.Fatalf("failed to create temp dir: %v", err)
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
st, err := Make(dir)
|
|
if err != nil {
|
|
t.Fatalf("failed to make state: %v", err)
|
|
}
|
|
|
|
if err := st.Connect("foo", false, []string{"open", "read", "write"}, nil); err != nil {
|
|
t.Fatalf("Connect failed: %v", err)
|
|
}
|
|
if err := st.Connect("bar", false, []string{"open", "read", "close"}, nil); err != nil {
|
|
t.Fatalf("Connect failed: %v", err)
|
|
}
|
|
checkPendingRepro(t, st, "foo", "")
|
|
checkPendingRepro(t, st, "bar", "")
|
|
|
|
if err := st.AddRepro("foo", []byte("open()")); err != nil {
|
|
t.Fatalf("AddRepro failed: %v", err)
|
|
}
|
|
checkPendingRepro(t, st, "foo", "")
|
|
checkPendingRepro(t, st, "bar", "open()")
|
|
checkPendingRepro(t, st, "bar", "")
|
|
|
|
// This repro is already present.
|
|
if err := st.AddRepro("bar", []byte("open()")); err != nil {
|
|
t.Fatalf("AddRepro failed: %v", err)
|
|
}
|
|
if err := st.AddRepro("bar", []byte("read()")); err != nil {
|
|
t.Fatalf("AddRepro failed: %v", err)
|
|
}
|
|
if err := st.AddRepro("bar", []byte("open()\nread()")); err != nil {
|
|
t.Fatalf("AddRepro failed: %v", err)
|
|
}
|
|
// This does not satisfy foo's call set.
|
|
if err := st.AddRepro("bar", []byte("close()")); err != nil {
|
|
t.Fatalf("AddRepro failed: %v", err)
|
|
}
|
|
checkPendingRepro(t, st, "bar", "")
|
|
|
|
// Check how persistence works.
|
|
st, err = Make(dir)
|
|
if err != nil {
|
|
t.Fatalf("failed to make state: %v", err)
|
|
}
|
|
if err := st.Connect("foo", false, []string{"open", "read", "write"}, nil); err != nil {
|
|
t.Fatalf("Connect failed: %v", err)
|
|
}
|
|
if err := st.Connect("bar", false, []string{"open", "read", "close"}, nil); err != nil {
|
|
t.Fatalf("Connect failed: %v", err)
|
|
}
|
|
checkPendingRepro(t, st, "bar", "")
|
|
checkPendingRepro(t, st, "foo", "read()")
|
|
checkPendingRepro(t, st, "foo", "open()\nread()")
|
|
checkPendingRepro(t, st, "foo", "")
|
|
}
|
|
|
|
func checkPendingRepro(t *testing.T, st *State, name, result string) {
|
|
repro, err := st.PendingRepro(name)
|
|
if err != nil {
|
|
t.Fatalf("\n%v: PendingRepro failed: %v", caller(1), err)
|
|
}
|
|
if string(repro) != result {
|
|
t.Fatalf("\n%v: PendingRepro returned %q, want %q", caller(1), string(repro), result)
|
|
}
|
|
}
|
|
|
|
func caller(skip int) string {
|
|
_, file, line, _ := runtime.Caller(skip + 1)
|
|
return fmt.Sprintf("%v:%v", filepath.Base(file), line)
|
|
}
|