package main
import (
"net"
"net/http"
"strings"
"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
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
}
// 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, "/")
split = strings.Split(path, "/")
split[1] = "ipv6"
ipv6_url := strings.Join(split, "/")
split = strings.Split(path, "/")
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 = `
`
// Add a link for each of the servers
for _, server := range settingServers {
split = strings.Split(path, "/")
var serverActive string
if split[3] == server {
serverActive = " active"
}
split[3] = server
server_url := strings.Join(split, "/")
serverNavigation += `
`
}
// Add the options in navbar form, and check if they are active
var options string
split = strings.Split(path, "/")
if split[2] == "summary" {
options += ``
} else {
options += ``
}
if split[2] == "route" {
options += ``
} else {
options += ``
}
if split[2] == "route_all" {
options += ``
} else {
options += ``
}
if isWhois {
options += ``
} else {
options += ``
}
if split[2] == "traceroute" {
options += ``
} else {
options += ``
}
var target string
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]
}
w.Write([]byte(`
` + title + `
`))
}
// Print HTML footer to http response
func templateFooter(w http.ResponseWriter) {
w.Write([]byte(`
`))
}
// 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") {
var tabPending bool = false
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))
}
}
}
w.Write([]byte("\n"))
}
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") {
var tabPending bool = false
var tableCells int = 0
var row [6]string
for i, word := range strings.Split(line, " ") {
if len(word) == 0 {
tabPending = true
} else {
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 {
row[tableCells] += " "
}
tabPending = false
} else {
row[tableCells] += " "
}
row[tableCells] += word
}
}
// 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("
" + row[i] + "
"))
}
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("