1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
|
package system
import (
"context"
"net"
"strings"
"time"
"github.com/goss-org/goss/util"
)
type Addr interface {
Address() string
Exists() (bool, error)
Reachable() (bool, error)
}
type DefAddr struct {
address string
LocalAddress string
Timeout int
}
func NewDefAddr(_ context.Context, address string, system *System, config util.Config) Addr {
addr := normalizeAddress(address)
return &DefAddr{
address: addr,
LocalAddress: config.LocalAddress,
Timeout: config.TimeOutMilliSeconds(),
}
}
func (a *DefAddr) ID() string {
return a.address
}
func (a *DefAddr) Address() string {
return a.address
}
func (a *DefAddr) Exists() (bool, error) { return a.Reachable() }
func (a *DefAddr) Reachable() (bool, error) {
network, address := splitAddress(a.address)
var localAddr net.Addr
if network == "udp" {
localAddr = &net.UDPAddr{IP: net.ParseIP(a.LocalAddress)}
} else {
localAddr = &net.TCPAddr{IP: net.ParseIP(a.LocalAddress)}
}
d := net.Dialer{LocalAddr: localAddr, Timeout: time.Duration(a.Timeout) * time.Millisecond}
conn, err := d.Dial(network, address)
if err != nil {
return false, nil
}
conn.Close()
return true, nil
}
func splitAddress(fulladdress string) (network, address string) {
split := strings.SplitN(fulladdress, "://", 2)
if len(split) == 2 {
return split[0], split[1]
}
return "tcp", fulladdress
}
func normalizeAddress(fulladdress string) string {
net, addr := splitAddress(fulladdress)
return net + "://" + addr
}
|