diff --git a/frontend/lgproxy.go b/frontend/lgproxy.go index c10389c..9c820ee 100644 --- a/frontend/lgproxy.go +++ b/frontend/lgproxy.go @@ -12,11 +12,14 @@ type channelData struct { data string } +// Send commands to lgproxy instances in parallel, and retrieve their responses func batchRequest(servers []string, endpoint string, command string) []string { + // Channel and array for storing responses var ch chan channelData = make(chan channelData) var response_array []string = make([]string, len(servers)) for i, server := range servers { + // Check if the server is in the valid server list passed at startup var isValidServer bool = false for _, validServer := range settingServers { if validServer == server { @@ -24,11 +27,14 @@ func batchRequest(servers []string, endpoint string, command string) []string { break } } + if !isValidServer { + // If the server is not valid, create a dummy goroutine to return a failure go func (i int) { ch <- channelData{i, "request failed: invalid server\n"} } (i) } else { + // Compose URL and send the request url := "http://" + server + "." + settingServersDomain + ":" + strconv.Itoa(settingServersPort) + "/" + url.PathEscape(endpoint) + "?q=" + url.QueryEscape(command) go func (url string, i int){ response, err := http.Get(url) @@ -42,6 +48,7 @@ func batchRequest(servers []string, endpoint string, command string) []string { } } + // Sort the responses by their ids, to return data in order for range servers { var output channelData = <-ch response_array[output.id] = output.data diff --git a/frontend/main.go b/frontend/main.go index 4850475..3382169 100644 --- a/frontend/main.go +++ b/frontend/main.go @@ -9,17 +9,20 @@ var settingServers []string var settingServersDomain string var settingServersPort int var settingWhoisServer string +var settingListen string func main() { serversPtr := flag.String("servers", "", "server name prefixes, separated by comma") domainPtr := flag.String("domain", "", "server name domain suffixes") portPtr := flag.Int("port", 8000, "port bird-lgproxy is running on") whoisPtr := flag.String("whois", "whois.verisign-grs.com", "whois server for queries") + listenPortPtr := flag.String("listen", ":5000", "address bird-lg is listening on") flag.Parse() settingServers = strings.Split(*serversPtr, ",") settingServersDomain = *domainPtr settingServersPort = *portPtr settingWhoisServer = *whoisPtr + settingListen = *listenPortPtr webServerStart() } diff --git a/frontend/template.go b/frontend/template.go index a125524..a709beb 100644 --- a/frontend/template.go +++ b/frontend/template.go @@ -7,13 +7,38 @@ import ( "strconv" ) +// Helper to check if the IP is valid +func isIP(s string) bool { + return nil != net.ParseIP(s) +} + +// Helper to check if the number is valid +func isNumber(s string) bool { + _, err := strconv.Atoi(s) + return nil == err +} + +// Print HTML header to the given http response func templateHeader(w http.ResponseWriter, r *http.Request, title string) { path := r.URL.Path - if len(strings.Split(r.URL.Path, "/")) < 4 { - path = "/ipv4/summary/" + strings.Join(settingServers[:], "+") + "/" + split := strings.Split(r.URL.Path, "/") + + // Mark if the URL is for a whois query + var isWhois bool = false + if len(split) >= 2 && split[1] == "whois" { + isWhois = true } - split := strings.Split(path, "/") + // Use a default URL if the request URL is too short + // The URL is for return to IPv4 summary page + if len(split) < 4 { + path = "/ipv4/summary/" + strings.Join(settingServers[:], "+") + "/" + } else if len(split) == 4 { + path += "/" + } + + // Compose URLs for link in navbar + split = strings.Split(path, "/") split[1] = "ipv4" ipv4_url := strings.Join(split, "/") @@ -25,12 +50,14 @@ func templateHeader(w http.ResponseWriter, r *http.Request, title string) { split[3] = strings.Join(settingServers[:], "+") all_url := strings.Join(split, "/") + // Check if the "All Server" link should be marked as active split = strings.Split(path, "/") var serverAllActive string if split[3] == strings.Join(settingServers[:], "+") { serverAllActive = " active" } + // Print the IPv4, IPv6, All Servers link in navbar var serverNavigation string = `
  • IPv4 @@ -43,6 +70,8 @@ func templateHeader(w http.ResponseWriter, r *http.Request, title string) { All Servers
    • ` + + // Add a link for each of the servers for _, server := range settingServers { split = strings.Split(path, "/") var serverActive string @@ -59,6 +88,7 @@ func templateHeader(w http.ResponseWriter, r *http.Request, title string) { ` } + // Add the options in navbar form, and check if they are active var options string split = strings.Split(path, "/") if split[2] == "summary" { @@ -76,14 +106,25 @@ func templateHeader(w http.ResponseWriter, r *http.Request, title string) { } else { options += `` } - if split[2] == "whois" { + if isWhois { options += `` } else { options += `` } + if split[2] == "traceroute" { + options += `` + } else { + options += `` + } var target string - if len(split) >= 5 { + if isWhois { + // This is a whois request, use original path URL instead of the modified one + // and extract the target + whoisSplit := strings.Split(r.URL.Path, "/") + target = whoisSplit[2] + } else if len(split) >= 5 { + // This is a normal request, just extract the target target = split[4] } @@ -126,25 +167,17 @@ func templateHeader(w http.ResponseWriter, r *http.Request, title string) { `)) } +// Print HTML footer to http response func templateFooter(w http.ResponseWriter) { w.Write([]byte(` - `)) } -func isIP(s string) bool { - return nil != net.ParseIP(s) -} - -func isNumber(s string) bool { - _, err := strconv.Atoi(s) - return nil == err -} - +// Write the given text to http response, and add whois links for +// ASNs and IP addresses func smartWriter(w http.ResponseWriter, s string) { w.Write([]byte("
    "))
     for _, line := range strings.Split(s, "\n") {
@@ -152,32 +185,46 @@ func smartWriter(w http.ResponseWriter, s string) {
         var isFirstWord bool = true
         var isASes bool = false
         for _, word := range strings.Split(line, " ") {
+            // Process each word
             if len(word) == 0 {
+                // Indicates that two spaces are connected together
+                // Replace this with a tab later
                 tabPending = true
             } else {
                 if isFirstWord {
+                    // Do not add space before the first word
                     isFirstWord = false
                 } else if tabPending {
+                    // A tab should be added; add it
                     w.Write([]byte("\t"))
                     tabPending = false
                 } else {
+                    // Two words separated by a space, just print the space
                     w.Write([]byte(" "))
                 }
 
                 if isIP(word) {
+                    // Add whois link to the IP, handles IPv4 and IPv6
                     w.Write([]byte("" + word + ""))
                 } else if len(strings.Split(word, "%")) == 2 && isIP(strings.Split(word, "%")[0]) {
+                    // IPv6 link-local with interface name, like fd00::1%eth0
+                    // Add whois link to address part
                     w.Write([]byte("" + strings.Split(word, "%")[0] + ""))
                     w.Write([]byte("%" + strings.Split(word, "%")[1]))
                 } else if len(strings.Split(word, "/")) == 2 && isIP(strings.Split(word, "/")[0]) {
+                    // IP with a CIDR range, like 192.168.0.1/24
+                    // Add whois link to first part
                     w.Write([]byte("" + strings.Split(word, "/")[0] + ""))
                     w.Write([]byte("/" + strings.Split(word, "/")[1]))
                 } else if word == "AS:" || word == "\tBGP.as_path:" {
+                    // Bird will output ASNs later
                     isASes = true
                     w.Write([]byte(word))
                 } else if isASes && isNumber(word) {
+                    // Bird is outputing ASNs, ass whois for them
                     w.Write([]byte("" + word + ""))
                 } else {
+                    // Just an ordinary word, print it and done
                     w.Write([]byte(word))
                 }
             }
@@ -187,6 +234,7 @@ func smartWriter(w http.ResponseWriter, s string) {
     w.Write([]byte("
    ")) } +// Output a table for the summary page func summaryTable(w http.ResponseWriter, isIPv6 bool, data string, serverName string) { w.Write([]byte("")) for lineId, line := range strings.Split(data, "\n") { @@ -200,6 +248,7 @@ func summaryTable(w http.ResponseWriter, isIPv6 bool, data string, serverName st if i == 0 { tabPending = true } else if tabPending { + // Allow up to 6 columns in the table, any more is ignored if tableCells < 5 { tableCells++ } else { @@ -213,26 +262,32 @@ func summaryTable(w http.ResponseWriter, isIPv6 bool, data string, serverName st } } + // Ignore empty lines if len(row[0]) == 0 { continue } + if lineId == 0 { + // Draw the table head w.Write([]byte("")) for i := 0; i < 6; i++ { w.Write([]byte("")) } w.Write([]byte("")) } else { + // Draw the row in red if the link isn't up if row[3] == "up" { w.Write([]byte("")) } else if lineId != 0 { w.Write([]byte("")) } + // Add link to detail for first column if isIPv6 { w.Write([]byte("")) } else { w.Write([]byte("")) } + // Draw the other cells for i := 1; i < 6; i++ { w.Write([]byte("")) } diff --git a/frontend/webserver.go b/frontend/webserver.go index 24250ef..304be0f 100644 --- a/frontend/webserver.go +++ b/frontend/webserver.go @@ -7,47 +7,49 @@ import ( ) func webDispatcherIPv4Summary(w http.ResponseWriter, r *http.Request) { - webHandler(w, r, false, r.URL.Path[len("/ipv4/summary/"):], "show protocols") + split := strings.Split(r.URL.Path[len("/ipv4/summary/"):], "/") + webHandler(w, r, "bird", split[0], "show protocols") } func webDispatcherIPv6Summary(w http.ResponseWriter, r *http.Request) { - webHandler(w, r, true, r.URL.Path[len("/ipv6/summary/"):], "show protocols") + split := strings.Split(r.URL.Path[len("/ipv6/summary/"):], "/") + webHandler(w, r, "bird6", split[0], "show protocols") } func webDispatcherIPv4Detail(w http.ResponseWriter, r *http.Request) { split := strings.Split(r.URL.Path[len("/ipv4/detail/"):], "/") - webHandler(w, r, false, split[0], "show protocols all " + split[1]) + webHandler(w, r, "bird", split[0], "show protocols all " + split[1]) } func webDispatcherIPv6Detail(w http.ResponseWriter, r *http.Request) { split := strings.Split(r.URL.Path[len("/ipv6/detail/"):], "/") - webHandler(w, r, true, split[0], "show protocols all " + split[1]) + webHandler(w, r, "bird6", split[0], "show protocols all " + split[1]) } func webDispatcherIPv4Route(w http.ResponseWriter, r *http.Request) { split := strings.Split(r.URL.Path[len("/ipv4/route/"):], "/") - webHandler(w, r, false, split[0], "show route for " + strings.Join(split[1:], "/")) + webHandler(w, r, "bird", split[0], "show route for " + strings.Join(split[1:], "/")) } func webDispatcherIPv6Route(w http.ResponseWriter, r *http.Request) { split := strings.Split(r.URL.Path[len("/ipv6/route/"):], "/") - webHandler(w, r, true, split[0], "show route for " + strings.Join(split[1:], "/")) + webHandler(w, r, "bird6", split[0], "show route for " + strings.Join(split[1:], "/")) } func webDispatcherIPv4RouteAll(w http.ResponseWriter, r *http.Request) { split := strings.Split(r.URL.Path[len("/ipv4/route_all/"):], "/") - webHandler(w, r, false, split[0], "show route for " + strings.Join(split[1:], "/") + " all") + webHandler(w, r, "bird", split[0], "show route for " + strings.Join(split[1:], "/") + " all") } func webDispatcherIPv6RouteAll(w http.ResponseWriter, r *http.Request) { split := strings.Split(r.URL.Path[len("/ipv6/route_all/"):], "/") - webHandler(w, r, true, split[0], "show route for " + strings.Join(split[1:], "/") + " all") + webHandler(w, r, "bird6", split[0], "show route for " + strings.Join(split[1:], "/") + " all") } func webDispatcherWhois(w http.ResponseWriter, r *http.Request) { var target string = r.URL.Path[len("/whois/"):] - templateHeader(w, r, "Bird-lg Go - WHOIS " + html.EscapeString(target)) + templateHeader(w, r, "Bird-lg Go - whois " + html.EscapeString(target)) w.Write([]byte("

    whois " + html.EscapeString(target) + "

    ")) smartWriter(w, whois(target)) @@ -55,20 +57,26 @@ func webDispatcherWhois(w http.ResponseWriter, r *http.Request) { templateFooter(w) } -func webHandler(w http.ResponseWriter, r *http.Request, isIPv6 bool, serverQuery string, command string) { - templateHeader(w, r, "Bird-lg Go - " + html.EscapeString(command)) +func webDispatcherIPv4Traceroute(w http.ResponseWriter, r *http.Request) { + split := strings.Split(r.URL.Path[len("/ipv4/traceroute/"):], "/") + webHandler(w, r, "traceroute", split[0], strings.Join(split[1:], "/")) +} + +func webDispatcherIPv6Traceroute(w http.ResponseWriter, r *http.Request) { + split := strings.Split(r.URL.Path[len("/ipv6/traceroute/"):], "/") + webHandler(w, r, "traceroute6", split[0], strings.Join(split[1:], "/")) +} + +func webHandler(w http.ResponseWriter, r *http.Request, endpoint string, serverQuery string, command string) { + templateHeader(w, r, "Bird-lg Go - " + html.EscapeString(endpoint + " " + command)) var servers []string = strings.Split(serverQuery, "+") - var responses []string - if isIPv6 { - responses = batchRequest(servers, "bird6", command) - } else { - responses = batchRequest(servers, "bird", command) - } + var responses []string = batchRequest(servers, endpoint, command) for i, response := range responses { w.Write([]byte("

    " + html.EscapeString(servers[i]) + ": " + html.EscapeString(command) + "

    ")) - if command == "show protocols" && response[0:4] == "name" { + if (endpoint == "bird" || endpoint == "bird6") && command == "show protocols" && response[0:4] == "name" { + var isIPv6 bool = endpoint[len(endpoint) - 1] == '6' summaryTable(w, isIPv6, response, servers[i]) } else { smartWriter(w, response) @@ -104,7 +112,9 @@ func webServerStart() { http.HandleFunc("/ipv6/route/", webDispatcherIPv6Route) http.HandleFunc("/ipv4/route_all/", webDispatcherIPv4RouteAll) http.HandleFunc("/ipv6/route_all/", webDispatcherIPv6RouteAll) + http.HandleFunc("/ipv4/traceroute/", webDispatcherIPv4Traceroute) + http.HandleFunc("/ipv6/traceroute/", webDispatcherIPv6Traceroute) http.HandleFunc("/whois/", webDispatcherWhois) http.HandleFunc("/redir/", navbarFormRedirect) - http.ListenAndServe(":5000", nil) + http.ListenAndServe(settingListen, nil) } diff --git a/frontend/whois.go b/frontend/whois.go index 327d220..bc8f0ba 100644 --- a/frontend/whois.go +++ b/frontend/whois.go @@ -5,6 +5,7 @@ import ( "io/ioutil" ) +// Send a whois request func whois(s string) string { conn, err := net.Dial("tcp", settingWhoisServer + ":43") if err != nil {
    " + row[i] + "
    " + row[0] + "" + row[0] + "" + row[i] + "