103 lines
2.2 KiB
Go
103 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func contains(s []string, str string) bool {
|
|
for _, v := range s {
|
|
if v == str {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func TestGetASNRepresentationDNS(t *testing.T) {
|
|
checkNetwork(t)
|
|
|
|
setting.dnsInterface = "asn.cymru.com"
|
|
setting.whoisServer = ""
|
|
cache := make(ASNCache)
|
|
result := cache.lookup("6939")
|
|
if !strings.Contains(result, "HURRICANE") {
|
|
t.Errorf("Lookup AS6939 failed, got %s", result)
|
|
}
|
|
}
|
|
|
|
func TestGetASNRepresentationWhois(t *testing.T) {
|
|
checkNetwork(t)
|
|
|
|
setting.dnsInterface = ""
|
|
setting.whoisServer = "whois.arin.net"
|
|
cache := make(ASNCache)
|
|
result := cache.lookup("6939")
|
|
if !strings.Contains(result, "HURRICANE") {
|
|
t.Errorf("Lookup AS6939 failed, got %s", result)
|
|
}
|
|
}
|
|
|
|
func TestGetASNRepresentationFallback(t *testing.T) {
|
|
setting.dnsInterface = ""
|
|
setting.whoisServer = ""
|
|
cache := make(ASNCache)
|
|
result := cache.lookup("6939")
|
|
if result != "AS6939" {
|
|
t.Errorf("Lookup AS6939 failed, got %s", result)
|
|
}
|
|
}
|
|
|
|
// Broken due to random order of attributes
|
|
func TestBirdRouteToGraphviz(t *testing.T) {
|
|
setting.dnsInterface = ""
|
|
|
|
// Don't change formatting of the following strings!
|
|
|
|
fakeResult := `192.168.0.1/32 unicast [alpha 2021-01-14 from 192.168.0.2] * (100) [AS12345i]
|
|
via 192.168.0.2 on eth0
|
|
Type: BGP univ
|
|
BGP.origin: IGP
|
|
BGP.as_path: 4242422601
|
|
BGP.next_hop: 172.18.0.2`
|
|
|
|
expectedLinesInResult := []string{
|
|
`"AS4242422601" [`,
|
|
`"AS4242422601" -> "Target: 192.168.0.1" [`,
|
|
`"Target: 192.168.0.1" [`,
|
|
`"alpha" [`,
|
|
`"alpha" -> "AS4242422601" [`,
|
|
}
|
|
|
|
result := birdRouteToGraphviz([]string{
|
|
"alpha",
|
|
}, []string{
|
|
fakeResult,
|
|
}, "192.168.0.1")
|
|
|
|
for _, line := range expectedLinesInResult {
|
|
if !strings.Contains(result, line) {
|
|
t.Errorf("Expected line in result not found: %s", line)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestBirdRouteToGraphvizXSS(t *testing.T) {
|
|
setting.dnsInterface = ""
|
|
|
|
// Don't change formatting of the following strings!
|
|
|
|
fakeResult := `<script>alert("evil!")</script>`
|
|
|
|
result := birdRouteToGraphviz([]string{
|
|
"alpha",
|
|
}, []string{
|
|
fakeResult,
|
|
}, fakeResult)
|
|
|
|
if strings.Contains(result, "<script>") {
|
|
t.Errorf("XSS injection succeeded: %s", result)
|
|
}
|
|
}
|