Add opt-in go-fuzz modules support

dvyukov · Oct 7, 2019 · 703c4f1 · 703c4f1
1 parent 3901820
commit 703c4f1
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/go-fuzz-build/main.go b/go-fuzz-build/main.go
@@ -60,7 +60,13 @@ func makeTags() string {
 // that clients can then modify and use for calls to go/packages.
 func basePackagesConfig() *packages.Config {
 	cfg := new(packages.Config)
-	cfg.Env = append(os.Environ(), "GO111MODULE=off")
+	goFuzzModule, isGoFuzzModuleSet := os.LookupEnv("GOFUZZ111MODULE")
+
+	if isGoFuzzModuleSet {
+		cfg.Env = append(os.Environ(), "GO111MODULE=" + goFuzzModule)
+	} else {
+		cfg.Env = append(os.Environ(), "GO111MODULE=off")
+	}
 	return cfg
 }