burble.dn42/bird-lg-go
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Merge pull request #27 from miegl/master

Browse Source 
frontend: optional ASN resolution using whois
This commit is contained in:
Yuhui Xu 2021-07-31 17:02:25 +08:00 committed by GitHub
commit 2e0cb131ca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
frontend/bgpmap.go
View File

@ -8,17 +8,33 @@ import (
) )
func getASNRepresentation(asn string) string { func getASNRepresentation(asn string) string {
records, err := net.LookupTXT(fmt.Sprintf("AS%s.%s", asn, setting.dnsInterface)) if setting.dnsInterface != "" {
if err != nil { // get ASN representation using DNS
// DNS query failed, only use ASN as output records, err := net.LookupTXT(fmt.Sprintf("AS%s.%s", asn, setting.dnsInterface))
return fmt.Sprintf("AS%s", asn) if err != nil {
} // DNS query failed, only use ASN as output
return fmt.Sprintf("AS%s", asn)
}
result := strings.Join(records, " ") result := strings.Join(records, " ")
if resultSplit := strings.Split(result, " | "); len(resultSplit) > 1 { if resultSplit := strings.Split(result, " | "); len(resultSplit) > 1 {
result = strings.Join(resultSplit[1:], "\\n") result = strings.Join(resultSplit[1:], "\\n")
}
return fmt.Sprintf("AS%s\\n%s", asn, result)
} else {
// get ASN representation using WHOIS
records := whois(fmt.Sprintf("AS%s", asn))
recordsSplit := strings.Split(records, "\n")
result := "\\n"
for _, line := range recordsSplit {
if strings.Contains(line, "as-name:") || strings.Contains(line, "ASName:") {
result = result + strings.TrimSpace(strings.SplitN(line, ":", 2)[1])
} else if strings.Contains(line, "descr:") {
result = result + "\\n" + strings.TrimSpace(strings.SplitN(line, ":", 2)[1])
}
}
return fmt.Sprintf("AS%s%s", asn, result)
} }
return fmt.Sprintf("AS%s\\n%s", asn, result)
} }
func birdRouteToGraphviz(servers []string, responses []string, target string) string { func birdRouteToGraphviz(servers []string, responses []string, target string) string {