feat templater: Add IP lookup function
This introduces support for looking up IP addresses using local DNS resolver. Function will return a list of all IP addresses associated with hostname. Further processing can be achieved using supported list template functions.
This commit is contained in:
parent
b20bc5f57a
commit
68e2f99062
2 changed files with 35 additions and 0 deletions
34
templater/dns.go
Normal file
34
templater/dns.go
Normal file
|
@ -0,0 +1,34 @@
|
||||||
|
// This file contains the implementation of a template function for retrieving IP addresses from DNS
|
||||||
|
package templater
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/polydawn/meep"
|
||||||
|
"net"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
type DNSError struct {
|
||||||
|
meep.TraitAutodescribing
|
||||||
|
meep.TraitCausable
|
||||||
|
Output string
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetIPsFromDNS(host string) ([]interface{}, error) {
|
||||||
|
fmt.Fprintf(os.Stderr, "Attempting to look up IP for %s in DNS\n", host)
|
||||||
|
ips, err := net.LookupIP(host)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, meep.New(
|
||||||
|
&DNSError{Output: "IP address lookup failed"},
|
||||||
|
meep.Cause(err),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
var result []interface{} = make([]interface{}, len(ips))
|
||||||
|
for i, ip := range ips {
|
||||||
|
result[i] = ip
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
|
@ -180,6 +180,7 @@ func templateFuncs() template.FuncMap {
|
||||||
return string(b)
|
return string(b)
|
||||||
}
|
}
|
||||||
m["passLookup"] = GetFromPass
|
m["passLookup"] = GetFromPass
|
||||||
|
m["lookupIPAddr"] = GetIPsFromDNS
|
||||||
|
|
||||||
return m
|
return m
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue