syz-hub: add program

syz-hub is used to exchange programs between syz-managers.

2 files changed, 128 insertions, 0 deletions

diff --git a/prog/encoding.go b/prog/encoding.go
index 7afe218be..11c86fb44 100644
--- a/prog/encoding.go
+++ b/prog/encoding.go
@@ -516,3 +516,36 @@ func (p *parser) Ident() string {
 func (p *parser) failf(msg string, args ...interface{}) {
 	p.e = fmt.Errorf("%v\nline #%v: %v", fmt.Sprintf(msg, args...), p.l, p.s)
 }
+
+// CallSet returns a set of all calls in the program.
+// It does very conservative parsing and is intended to parse paste/future serialization formats.
+func CallSet(data []byte) (map[string]struct{}, error) {
+	calls := make(map[string]struct{})
+	s := bufio.NewScanner(bytes.NewReader(data))
+	for s.Scan() {
+		ln := s.Bytes()
+		if len(ln) == 0 || ln[0] == '#' {
+			continue
+		}
+		bracket := bytes.IndexByte(ln, '(')
+		if bracket == -1 {
+			return nil, fmt.Errorf("line does not contain opening bracket")
+		}
+		call := ln[:bracket]
+		if eq := bytes.IndexByte(call, '='); eq != -1 {
+			eq++
+			for eq < len(call) && call[eq] == ' ' {
+				eq++
+			}
+			call = call[eq:]
+		}
+		if len(call) == 0 {
+			return nil, fmt.Errorf("call name is empty")
+		}
+		calls[string(call)] = struct{}{}
+	}
+	if len(calls) == 0 {
+		return nil, fmt.Errorf("program does not contain any calls")
+	}
+	return calls, nil
+}
diff --git a/prog/encoding_test.go b/prog/encoding_test.go
new file mode 100644
index 000000000..f23a92cc7
--- /dev/null
+++ b/prog/encoding_test.go
@@ -0,0 +1,95 @@
+// 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 prog
+
+import (
+	"fmt"
+	"reflect"
+	"sort"
+	"testing"
+)
+
+func setToArray(s map[string]struct{}) []string {
+	a := make([]string, 0, len(s))
+	for c := range s {
+		a = append(a, c)
+	}
+	sort.Strings(a)
+	return a
+}
+
+func TestCallSet(t *testing.T) {
+	tests := []struct {
+		prog  string
+		ok    bool
+		calls []string
+	}{
+		{
+			"",
+			false,
+			[]string{},
+		},
+		{
+			"r0 =  (foo)",
+			false,
+			[]string{},
+		},
+		{
+			"getpid()",
+			true,
+			[]string{"getpid"},
+		},
+		{
+			"r11 =  getpid()",
+			true,
+			[]string{"getpid"},
+		},
+		{
+			"getpid()\n" +
+				"open(0x1, something that this package may not understand)\n" +
+				"getpid()\n" +
+				"#read()\n" +
+				"\n" +
+				"close$foo(&(0x0000) = {})\n",
+			true,
+			[]string{"getpid", "open", "close$foo"},
+		},
+	}
+	for i, test := range tests {
+		t.Run(fmt.Sprint(i), func(t *testing.T) {
+			calls, err := CallSet([]byte(test.prog))
+			if err != nil && test.ok {
+				t.Fatalf("parsing failed: %v", err)
+			}
+			if err == nil && !test.ok {
+				t.Fatalf("parsing did not fail")
+			}
+			callArray := setToArray(calls)
+			sort.Strings(test.calls)
+			if !reflect.DeepEqual(callArray, test.calls) {
+				t.Fatalf("got call set %+v, expect %+v", callArray, test.calls)
+			}
+		})
+	}
+}
+
+func TestCallSetRandom(t *testing.T) {
+	rs, iters := initTest(t)
+	for i := 0; i < iters; i++ {
+		p := Generate(rs, 10, nil)
+		calls0 := make(map[string]struct{})
+		for _, c := range p.Calls {
+			calls0[c.Meta.Name] = struct{}{}
+		}
+		calls1, err := CallSet(p.Serialize())
+		if err != nil {
+			t.Fatalf("CallSet failed: %v", err)
+		}
+		callArray0 := setToArray(calls0)
+		callArray1 := setToArray(calls1)
+		if !reflect.DeepEqual(callArray0, callArray1) {
+			t.Fatalf("got call set:\n%+v\nexpect:\n%+v", callArray1, callArray0)
+		}
+	}
+}