rancher/rke
1
0
Fork 0
mirror of https://github.com/rancher/rke.git synced 2025-08-07 09:54:18 +00:00
Code Issues Releases Wiki Activity

Fall back to HOMEDRIVE+HOMEPATH or USERPROFILE

Browse Source
This commit is contained in:
Gert Palok 2018-05-23 12:34:57 +03:00 committed by Alena Prokharchyk
parent d909253640
commit 70fca52c33
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
hosts/tunnel.go
View File

@ -114,8 +114,20 @@ func getSSHConfig(username, sshPrivateKeyString string, useAgentAuth bool) (*ssh
func privateKeyPath(sshKeyPath string) string { func privateKeyPath(sshKeyPath string) string {
if sshKeyPath[:2] == "~/" { if sshKeyPath[:2] == "~/" {
sshKeyPath = filepath.Join(os.Getenv("HOME"), sshKeyPath[2:]) sshKeyPath = filepath.Join(userHome(), sshKeyPath[2:])
} }
buff, _ := ioutil.ReadFile(sshKeyPath) buff, _ := ioutil.ReadFile(sshKeyPath)
return string(buff) return string(buff)
} }
func userHome() string {
if home := os.Getenv("HOME"); home != "" {
return home
}
homeDrive := os.Getenv("HOMEDRIVE")
homePath := os.Getenv("HOMEPATH")
if homeDrive != "" && homePath != "" {
return homeDrive + homePath
}
return os.Getenv("USERPROFILE")
}