pkg/host: split files into syscalls/features

pkg/host does 2 things: detects supported syscalls and supported features.
There is enough code for each for a separate file.

1 files changed, 151 insertions, 0 deletions

diff --git a/pkg/host/syscalls_linux_test.go b/pkg/host/syscalls_linux_test.go
new file mode 100644
index 000000000..2064a10e3
--- /dev/null
+++ b/pkg/host/syscalls_linux_test.go
@@ -0,0 +1,151 @@
+// Copyright 2015 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.
+
+// +build linux
+
+package host
+
+import (
+	"runtime"
+	"syscall"
+	"testing"
+
+	"github.com/google/syzkaller/prog"
+)
+
+func TestSupportedSyscalls(t *testing.T) {
+	t.Parallel()
+	target, err := prog.GetTarget("linux", runtime.GOARCH)
+	if err != nil {
+		t.Fatal(err)
+	}
+	supp, _, err := DetectSupportedSyscalls(target, "none")
+	if err != nil {
+		t.Skipf("skipping: %v", err)
+	}
+	// These are safe to execute with invalid arguments.
+	safe := []string{
+		"memfd_create",
+		"sendfile",
+		"bpf$MAP_CREATE",
+		"open",
+		"openat",
+		"read",
+		"write",
+		"stat",
+	}
+	for _, name := range safe {
+		c := target.SyscallMap[name]
+		if c == nil {
+			t.Fatalf("can't find syscall '%v'", name)
+		}
+		a := ^uintptr(0) - 4097 // hopefully invalid
+		_, _, err := syscall.Syscall6(uintptr(c.NR), a, a, a, a, a, a)
+		if err == 0 {
+			t.Fatalf("%v did not fail", name)
+		}
+		if ok := err != syscall.ENOSYS; ok != supp[c] {
+			t.Fatalf("syscall %v: perse=%v kallsyms=%v", name, ok, supp[c])
+		}
+	}
+}
+
+func TestKallsymsParse(t *testing.T) {
+	tests := []struct {
+		Arch              string
+		Kallsyms          []byte
+		ParsedSyscalls    []string
+		SupportedSyscalls []string
+	}{
+		{
+			"amd64",
+			[]byte(`
+ffffffff817cdcc0 T __sys_bind
+ffffffff817cdda0 T __x64_sys_bind
+ffffffff817cddc0 T __ia32_sys_bind
+ffffffff817cdde0 T __sys_listen
+ffffffff817cde80 T __x64_sys_listen
+ffffffff817cde90 T __ia32_sys_listen
+ffffffff817cdea0 T __sys_accept4
+ffffffff817ce080 T __x64_sys_accept4
+ffffffff817ce0a0 T __ia32_sys_accept4
+			`),
+			[]string{"bind", "listen", "accept4"},
+			[]string{"bind", "listen", "accept4"},
+		},
+		{
+			"arm64",
+			[]byte(`
+ffff000010a3ddf8 T __sys_bind
+ffff000010a3def8 T __arm64_sys_bind
+ffff000010a3df20 T __sys_listen
+ffff000010a3dfd8 T __arm64_sys_listen
+ffff000010a3e000 T __sys_accept4
+ffff000010a3e1f0 T __arm64_sys_accept4
+			`),
+			[]string{"bind", "listen", "accept4"},
+			[]string{"bind", "listen", "accept4"},
+		},
+		{
+			"ppc64le",
+			[]byte(`
+c0000000011ec810 T __sys_bind
+c0000000011eca10 T sys_bind
+c0000000011eca10 T __se_sys_bind
+c0000000011eca70 T __sys_listen
+c0000000011ecc10 T sys_listen
+c0000000011ecc10 T __se_sys_listen
+c0000000011ecc70 T __sys_accept4
+c0000000011ed050 T sys_accept4
+c0000000011ed050 T __se_sys_accept4
+			`),
+			[]string{"bind", "listen", "accept4"},
+			[]string{"bind", "listen", "accept4"},
+		},
+		{
+			"arm",
+			[]byte(`
+c037c67c T __se_sys_setfsuid
+c037c694 T __sys_setfsgid
+c037c790 T sys_setfsgid
+c037c790 T __se_sys_setfsgid
+c037c7a8 T sys_getpid
+c037c7d0 T sys_gettid
+c037c7f8 T sys_getppid
+			`),
+			[]string{"setfsgid", "getpid", "gettid", "getppid"},
+			[]string{"setfsgid", "getpid", "gettid", "getppid"},
+		},
+		// Test kallsymsRenameMap
+		{
+			"ppc64le",
+			[]byte(`
+c00000000037eb00 T sys_newstat
+			`),
+			[]string{"newstat"},
+			[]string{"stat"},
+		},
+	}
+
+	for _, test := range tests {
+		syscallSet := parseKallsyms(test.Kallsyms, test.Arch)
+		if len(syscallSet) != len(test.ParsedSyscalls) {
+			t.Fatalf("wrong number of parse syscalls, expected: %v, got: %v",
+				len(test.ParsedSyscalls), len(syscallSet))
+		}
+		for _, syscall := range test.ParsedSyscalls {
+			if _, ok := syscallSet[syscall]; !ok {
+				t.Fatalf("syscall %v not found in parsed syscall list", syscall)
+			}
+		}
+		for _, syscall := range test.SupportedSyscalls {
+			if newname := kallsymsRenameMap[syscall]; newname != "" {
+				syscall = newname
+			}
+
+			if _, ok := syscallSet[syscall]; !ok {
+				t.Fatalf("syscall %v not found in supported syscall list", syscall)
+			}
+		}
+	}
+}