From f4a76fa1f2b277241f373ad8cc26de74cbd0d323 Mon Sep 17 00:00:00 2001 From: mzack Date: Mon, 11 Oct 2021 07:42:03 +0200 Subject: [PATCH] adding custom drive support --- hostsfile/hostsfile.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/hostsfile/hostsfile.go b/hostsfile/hostsfile.go index c4a1031..2a3946f 100644 --- 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" }