119 lines
2.6 KiB
Go
119 lines
2.6 KiB
Go
// +build windows
|
|
// Contributed by Ross Smith II (smithii.com)
|
|
|
|
package vmware
|
|
|
|
import (
|
|
"errors"
|
|
"io/ioutil"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
// Interface to help find the host IP that is available from within
|
|
// the VMware virtual machines.
|
|
type HostIPFinder interface {
|
|
HostIP() (string, error)
|
|
}
|
|
|
|
// IfconfigIPFinder finds the host IP based on the output of `ifconfig`.
|
|
type IfconfigIPFinder struct {
|
|
Device string
|
|
}
|
|
|
|
func (f *IfconfigIPFinder) HostIP() (string, error) {
|
|
ift, err := net.Interfaces()
|
|
if err != nil {
|
|
return "", errors.New("No network interfaces found")
|
|
}
|
|
|
|
vmwareMac, err := getVMWareMAC()
|
|
if err != nil {
|
|
log.Print(err)
|
|
}
|
|
|
|
log.Printf("Searching for MAC %s", vmwareMac)
|
|
re := regexp.MustCompile("(?i)^" + vmwareMac)
|
|
|
|
ip := ""
|
|
|
|
for _, ifi := range ift {
|
|
mac := ifi.HardwareAddr.String()
|
|
log.Printf("Found MAC %s", mac)
|
|
|
|
matches := re.FindStringSubmatch(mac)
|
|
|
|
if matches == nil {
|
|
continue
|
|
}
|
|
|
|
addrs, err := ifi.Addrs()
|
|
if err != nil {
|
|
log.Printf("No IP addresses found for MAC %s", mac)
|
|
continue
|
|
}
|
|
|
|
for _, address := range addrs {
|
|
ip = address.String()
|
|
log.Printf("Found IP address %s for MAC %s", ip, mac)
|
|
}
|
|
|
|
// continue looping as VMNet8 comes after VMNet1 (at least on my system)
|
|
}
|
|
|
|
if ip == "" {
|
|
return "", errors.New("No MACs found matching " + vmwareMac)
|
|
}
|
|
|
|
log.Printf("Returning IP address %s", ip)
|
|
|
|
return ip, nil
|
|
}
|
|
|
|
func getVMWareMAC() (string, error) {
|
|
// return the first three tuples, if the actual MAC cannot be found
|
|
const defaultMacRe = "00:50:56"
|
|
|
|
programData := os.Getenv("ProgramData")
|
|
programData = strings.Replace(programData, "\\", "/", -1)
|
|
vmnetnat := programData + "/VMware/vmnetnat.conf"
|
|
if _, err := os.Stat(vmnetnat); os.IsNotExist(err) {
|
|
log.Printf("File not found: '%s' (found '%s' in %%ProgramData%%)", vmnetnat, programData)
|
|
return defaultMacRe, err
|
|
}
|
|
|
|
log.Printf("Searching for key hostMAC in '%s'", vmnetnat)
|
|
|
|
fh, err := os.Open(vmnetnat)
|
|
if err != nil {
|
|
return defaultMacRe, err
|
|
}
|
|
defer fh.Close()
|
|
|
|
bytes, err := ioutil.ReadAll(fh)
|
|
if err != nil {
|
|
return defaultMacRe, err
|
|
}
|
|
|
|
hostMacRe := regexp.MustCompile(`(?i)^\s*hostMAC\s*=\s*(.+)\s*$`)
|
|
|
|
for _, line := range strings.Split(string(bytes), "\n") {
|
|
// Need to trim off CR character when running in windows
|
|
line = strings.TrimRight(line, "\r")
|
|
|
|
matches := hostMacRe.FindStringSubmatch(line)
|
|
if matches != nil {
|
|
log.Printf("Found MAC '%s' in '%s'", matches[1], vmnetnat)
|
|
return matches[1], nil
|
|
}
|
|
}
|
|
|
|
log.Printf("Did not find key hostMAC in '%s', using %s instead", vmnetnat, defaultMacRe)
|
|
|
|
return defaultMacRe, nil
|
|
|
|
}
|