55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
|
package common
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
|
||
|
|
"github.com/hashicorp/packer/helper/multistep"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Step to discover the http ip
|
||
|
|
// which guests use to reach the vm host
|
||
|
|
// To make sure the IP is set before boot command and http server steps
|
||
|
|
type StepHTTPIPDiscover struct {
|
||
|
|
HTTPIP string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *StepHTTPIPDiscover) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
||
|
|
ip, err := getHostIP(s.HTTPIP)
|
||
|
|
if err != nil {
|
||
|
|
state.Put("error", err)
|
||
|
|
return multistep.ActionHalt
|
||
|
|
}
|
||
|
|
state.Put("http_ip", ip)
|
||
|
|
|
||
|
|
return multistep.ActionContinue
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *StepHTTPIPDiscover) Cleanup(state multistep.StateBag) {}
|
||
|
|
|
||
|
|
func getHostIP(s string) (string, error) {
|
||
|
|
if s != "" {
|
||
|
|
if net.ParseIP(s) != nil {
|
||
|
|
return s, nil
|
||
|
|
} else {
|
||
|
|
return "", fmt.Errorf("invalid IP address")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
addrs, err := net.InterfaceAddrs()
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, a := range addrs {
|
||
|
|
ipnet, ok := a.(*net.IPNet)
|
||
|
|
if ok && !ipnet.IP.IsLoopback() {
|
||
|
|
if ipnet.IP.To4() != nil {
|
||
|
|
return ipnet.IP.String(), nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return "", fmt.Errorf("IP not found")
|
||
|
|
}
|