executor: don't try to open tun if it's not enabled

2 files changed, 16 insertions, 6 deletions

diff --git a/csource/common.go b/csource/common.go
index 0cedc9920..17aa4b0ef 100644
--- a/csource/common.go
+++ b/csource/common.go
@@ -34,6 +34,7 @@ var commonHeader = `
 #include <setjmp.h>
 #include <signal.h>
 #include <stdarg.h>
+#include <stdbool.h>
 #include <stddef.h>
 #include <stdint.h>
 #include <stdio.h>
@@ -161,7 +162,7 @@ static void execute_command(const char* format, ...)
 	va_end(args);
 }
 
-int tunfd;
+int tunfd = -1;
 
 #define ADDR_MAX_LEN 32
 
@@ -222,6 +223,9 @@ static void initialize_tun(uint64_t pid)
 
 static uintptr_t syz_emit_ethernet(uintptr_t a0, uintptr_t a1)
 {
+	if (tunfd < 0)
+		return (uintptr_t)-1;
+
 	int64_t length = a0;
 	char* data = (char*)a1;
 	return write(tunfd, data, length);
@@ -351,7 +355,7 @@ static uintptr_t execute_syscall(int nr, uintptr_t a0, uintptr_t a1, uintptr_t a
 	}
 }
 
-static void setup_main_process(uint64_t pid)
+static void setup_main_process(uint64_t pid, bool enable_tun)
 {
 	struct sigaction sa;
 	memset(&sa, 0, sizeof(sa));
@@ -361,7 +365,8 @@ static void setup_main_process(uint64_t pid)
 	install_segv_handler();
 
 #ifdef __NR_syz_emit_ethernet
-	initialize_tun(pid);
+	if (enable_tun)
+		initialize_tun(pid);
 #endif
 
 	char tmpdir_template[] = "./syzkaller.XXXXXX";
diff --git a/csource/csource.go b/csource/csource.go
index d717ce153..8f3f99145 100644
--- a/csource/csource.go
+++ b/csource/csource.go
@@ -45,6 +45,11 @@ func Write(p *prog.Prog, opts Options) ([]byte, error) {
 	}
 	fmt.Fprintf(w, "\n")
 
+	enableTun := "false"
+	if _, ok := handled["syz_emit_ethernet"]; ok {
+		enableTun = "true"
+	}
+
 	hdr, err := preprocessCommonHeader(opts, handled)
 	if err != nil {
 		return nil, err
@@ -59,7 +64,7 @@ func Write(p *prog.Prog, opts Options) ([]byte, error) {
 		generateTestFunc(w, opts, calls, "loop")
 
 		fmt.Fprint(w, "int main()\n{\n")
-		fmt.Fprint(w, "\tsetup_main_process(0);\n")
+		fmt.Fprintf(w, "\tsetup_main_process(0, %v);\n", enableTun)
 		fmt.Fprintf(w, "\tint pid = do_sandbox_%v();\n", opts.Sandbox)
 		fmt.Fprint(w, "\tint status = 0;\n")
 		fmt.Fprint(w, "\twhile (waitpid(pid, &status, __WALL) != pid) {}\n")
@@ -68,7 +73,7 @@ func Write(p *prog.Prog, opts Options) ([]byte, error) {
 		generateTestFunc(w, opts, calls, "test")
 		if opts.Procs <= 1 {
 			fmt.Fprint(w, "int main()\n{\n")
-			fmt.Fprint(w, "\tsetup_main_process(0);\n")
+			fmt.Fprintf(w, "\tsetup_main_process(0, %v);\n", enableTun)
 			fmt.Fprintf(w, "\tint pid = do_sandbox_%v();\n", opts.Sandbox)
 			fmt.Fprint(w, "\tint status = 0;\n")
 			fmt.Fprint(w, "\twhile (waitpid(pid, &status, __WALL) != pid) {}\n")
@@ -78,7 +83,7 @@ func Write(p *prog.Prog, opts Options) ([]byte, error) {
 			fmt.Fprint(w, "\tint i;")
 			fmt.Fprintf(w, "\tfor (i = 0; i < %v; i++) {\n", opts.Procs)
 			fmt.Fprint(w, "\t\tif (fork() == 0) {\n")
-			fmt.Fprint(w, "\t\t\tsetup_main_process(i);\n")
+			fmt.Fprintf(w, "\t\t\tsetup_main_process(i, %v);\n", enableTun)
 			fmt.Fprintf(w, "\t\t\tdo_sandbox_%v();\n", opts.Sandbox)
 			fmt.Fprint(w, "\t\t}\n")
 			fmt.Fprint(w, "\t}\n")