Merge pull request #37 from msoap/windows_env_bug

Fixed pass PATH environment variable on Windows
msoap · Jan 21, 2018 · b378587 · b378587
2 parents bb167f0 + fae7026
commit b378587
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/shell2http.go b/shell2http.go
@@ -627,6 +627,10 @@ func isMultipartFormData(headers http.Header) bool {
 func proxySystemEnv(cmd *exec.Cmd, appConfig Config) {
 	varsNames := []string{"PATH", "HOME", "LANG", "USER", "TMPDIR"}
 
+	if runtime.GOOS == "windows" {
+		varsNames = append(varsNames, "USERNAME", "USERPROFILE", "HOMEDRIVE", "HOMEPATH", "TEMP", "TMP", "PATHEXT", "COMSPEC", "OS")
+	}
+
 	if appConfig.exportVars != "" {
 		varsNames = append(varsNames, strings.Split(appConfig.exportVars, ",")...)
 	}
@@ -638,7 +642,7 @@ func proxySystemEnv(cmd *exec.Cmd, appConfig Config) {
 				cmd.Env = append(cmd.Env, envRaw)
 			} else {
 				for _, envVarName := range varsNames {
-					if env[0] == envVarName {
+					if strings.ToUpper(env[0]) == envVarName {
 						cmd.Env = append(cmd.Env, envRaw)
 					}
 				}