67 lines
1.3 KiB
Go
67 lines
1.3 KiB
Go
package vmware
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
// VMnetNatConfIPFinder finds the IP address of the host machine by
|
|
// retrieving the IP from the vmnetnat.conf. This isn't a full proof
|
|
// technique but so far it has not failed.
|
|
type VMnetNatConfIPFinder struct{}
|
|
|
|
func (*VMnetNatConfIPFinder) HostIP() (string, error) {
|
|
programData := os.Getenv("ProgramData")
|
|
if programData == "" {
|
|
return "", errors.New("ProgramData directory not found.")
|
|
}
|
|
|
|
programData = strings.Replace(programData, "\\", "/", -1)
|
|
vmnetnat := filepath.Join(programData, "/VMware/vmnetnat.conf")
|
|
if _, err := os.Stat(vmnetnat); err != nil {
|
|
return "", fmt.Errorf("Error with vmnetnat.conf: %s", err)
|
|
}
|
|
|
|
f, err := os.Open(vmnetnat)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer f.Close()
|
|
|
|
ipRe := regexp.MustCompile(`^\s*ip\s*=\s*(.+?)\s*$`)
|
|
|
|
r := bufio.NewReader(f)
|
|
for {
|
|
line, err := r.ReadString('\n')
|
|
if line != "" {
|
|
matches := ipRe.FindStringSubmatch(line)
|
|
if matches != nil {
|
|
ip := matches[1]
|
|
dotIndex := strings.LastIndex(ip, ".")
|
|
if dotIndex == -1 {
|
|
continue
|
|
}
|
|
|
|
ip = ip[0:dotIndex] + ".1"
|
|
return ip, nil
|
|
}
|
|
}
|
|
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
|
|
return "", errors.New("host IP not found in NAT config")
|
|
}
|