adding custom drive support

projectdiscovery · Oct 11, 2021 · f4a76fa · f4a76fa
1 parent db15799
commit f4a76fa
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/hostsfile/hostsfile.go b/hostsfile/hostsfile.go
@@ -2,6 +2,8 @@ package hostsfile
 
 import (
 	"errors"
+	"fmt"
+	"os"
 	"runtime"
 	"strings"
 
@@ -11,7 +13,7 @@ import (
 
 func Path() string {
 	if runtime.GOOS == "windows" {
-		return `C:\Windows\System32\Drivers\etc\hosts`
+		return fmt.Sprintf(`%s\System32\Drivers\etc\hosts`, os.Getenv("SystemRoot"))
 	}
 	return "/etc/hosts"
 }