Merge pull request #131 from slingamn/hostmask_fix

fix hostmask issue
This commit is contained in:
Daniel Oaks 2017-09-06 02:01:47 +10:00 committed by GitHub
commit b975c6f182
1 changed files with 10 additions and 7 deletions

View File

@ -28,16 +28,19 @@ func AddrLookupHostname(addr net.Addr) string {
// LookupHostname returns the hostname for `addr` if it has one. Otherwise, just returns `addr`. // LookupHostname returns the hostname for `addr` if it has one. Otherwise, just returns `addr`.
func LookupHostname(addr string) string { func LookupHostname(addr string) string {
names, err := net.LookupAddr(addr) names, err := net.LookupAddr(addr)
if err != nil || len(names) < 1 || !IsHostname(names[0]) { if err == nil && len(names) > 0 {
candidate := strings.TrimSuffix(names[0], ".")
if IsHostname(candidate) {
return candidate
}
}
// return original address if no hostname found // return original address if no hostname found
if len(addr) > 0 && addr[0] == ':' { if len(addr) > 0 && addr[0] == ':' {
// fix for IPv6 hostnames (so they don't start with a colon), same as all other IRCds // fix for IPv6 hostnames (so they don't start with a colon), same as all other IRCds
addr = "0" + addr addr = "0" + addr
} }
return addr return addr
}
return names[0]
} }
var allowedHostnameChars = "abcdefghijklmnopqrstuvwxyz1234567890-." var allowedHostnameChars = "abcdefghijklmnopqrstuvwxyz1234567890-."