executor: add setup_ext_test extension point

The extension point allows to setup the test process in a custom way
without overwriting any of the existing files.

1 files changed, 29 insertions, 0 deletions

diff --git a/executor/common_ext_test.go b/executor/common_ext_test.go
index 45ba3a3cf..2815b095a 100644
--- a/executor/common_ext_test.go
+++ b/executor/common_ext_test.go
@@ -10,6 +10,8 @@ import (
 	"time"
 
 	"github.com/google/syzkaller/pkg/csource"
+	"github.com/google/syzkaller/pkg/ipc"
+	"github.com/google/syzkaller/pkg/ipc/ipcconfig"
 	"github.com/google/syzkaller/pkg/osutil"
 	"github.com/google/syzkaller/prog"
 )
@@ -31,4 +33,31 @@ func TestCommonExt(t *testing.T) {
 	if !bytes.Contains(out, []byte("example setup_ext called")) {
 		t.Fatalf("setup_ext wasn't called:\n%s", out)
 	}
+
+	// The example setup_ext_test does:
+	// *(uint64*)(SYZ_DATA_OFFSET + 0x1234) = 0xbadc0ffee;
+	// The following program tests that that value is present at 0x1234.
+	test := `syz_compare(&(0x7f0000001234)="", 0x8, &(0x7f0000000000)=@blob="eeffc0ad0b000000", AUTO)`
+	p, err := target.Deserialize([]byte(test), prog.Strict)
+	if err != nil {
+		t.Fatal(err)
+	}
+	cfg, opts, err := ipcconfig.Default(target)
+	if err != nil {
+		t.Fatal(err)
+	}
+	cfg.Executor = bin
+	cfg.Flags |= ipc.FlagDebug
+	env, err := ipc.MakeEnv(cfg, 0)
+	if err != nil {
+		t.Fatalf("failed to create env: %v", err)
+	}
+	defer env.Close()
+	_, info, _, err := env.Exec(opts, p)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if call := info.Calls[0]; (call.Flags&ipc.CallFinished) == 0 || call.Errno != 0 {
+		t.Fatalf("bad call result: flags=%x errno=%v", call.Flags, call.Errno)
+	}
 }