Multiple UI and functionality fixes
This commit is contained in:
Lan Tian
parent
5d91c9d6d1
commit
19fd44c28e
@ -1,58 +1,61 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"net/url"
|
||||
"io/ioutil"
|
||||
"strconv"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type channelData struct {
|
||||
id int
|
||||
data string
|
||||
id int
|
||||
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))
|
||||
// Channel and array for storing responses
|
||||
var ch chan channelData = make(chan channelData)
|
||||
var responseArray []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 {
|
||||
isValidServer = true
|
||||
break
|
||||
}
|
||||
}
|
||||
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 {
|
||||
isValidServer = true
|
||||
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)
|
||||
if err != nil {
|
||||
ch <- channelData{i, "request failed: " + err.Error() + "\n"}
|
||||
return
|
||||
}
|
||||
text, _ := ioutil.ReadAll(response.Body)
|
||||
ch <- channelData{i, string(text)}
|
||||
} (url, i)
|
||||
}
|
||||
}
|
||||
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)
|
||||
if err != nil {
|
||||
ch <- channelData{i, "request failed: " + err.Error() + "\n"}
|
||||
return
|
||||
}
|
||||
text, _ := ioutil.ReadAll(response.Body)
|
||||
if len(text) == 0 {
|
||||
text = []byte("node returned empty response, please refresh to try again.")
|
||||
}
|
||||
ch <- channelData{i, string(text)}
|
||||
}(url, i)
|
||||
}
|
||||
}
|
||||
|
||||
// Sort the responses by their ids, to return data in order
|
||||
for range servers {
|
||||
var output channelData = <-ch
|
||||
response_array[output.id] = output.data
|
||||
}
|
||||
// Sort the responses by their ids, to return data in order
|
||||
for range servers {
|
||||
var output channelData = <-ch
|
||||
responseArray[output.id] = output.data
|
||||
}
|
||||
|
||||
return response_array
|
||||
return responseArray
|
||||
}
|
||||
|
||||
@ -1,144 +1,106 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net"
|
||||
"net/http"
|
||||
"strings"
|
||||
"strconv"
|
||||
"net"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Helper to check if the IP is valid
|
||||
func isIP(s string) bool {
|
||||
return nil != net.ParseIP(s)
|
||||
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
|
||||
_, 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, "/")
|
||||
path := r.URL.Path[1:]
|
||||
split := strings.Split(path, "/")
|
||||
|
||||
// Mark if the URL is for a whois query
|
||||
var isWhois bool = false
|
||||
if len(split) >= 2 && split[1] == "whois" {
|
||||
isWhois = true
|
||||
}
|
||||
// Mark if the URL is for a whois query
|
||||
var isWhois bool = split[0] == "whois"
|
||||
var whoisTarget string = strings.Join(split[1:], "/")
|
||||
|
||||
// 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 += "/"
|
||||
}
|
||||
// Use a default URL if the request URL is too short
|
||||
// The URL is for return to IPv4 summary page
|
||||
if len(split) < 3 {
|
||||
path = "ipv4/summary/" + strings.Join(settingServers[:], "+") + "/"
|
||||
} else if len(split) == 3 {
|
||||
path += "/"
|
||||
}
|
||||
|
||||
// Compose URLs for link in navbar
|
||||
split = strings.Split(path, "/")
|
||||
split[1] = "ipv4"
|
||||
ipv4_url := strings.Join(split, "/")
|
||||
split = strings.Split(path, "/")
|
||||
|
||||
split = strings.Split(path, "/")
|
||||
split[1] = "ipv6"
|
||||
ipv6_url := strings.Join(split, "/")
|
||||
// Compose URLs for link in navbar
|
||||
ipv4URL := "/" + strings.Join([]string{"ipv4", split[1], split[2], strings.Join(split[3:], "/")}, "/")
|
||||
ipv6URL := "/" + strings.Join([]string{"ipv6", split[1], split[2], strings.Join(split[3:], "/")}, "/")
|
||||
allURL := "/" + strings.Join([]string{split[0], split[1], strings.Join(settingServers[:], "+"), strings.Join(split[3:], "/")}, "/")
|
||||
|
||||
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
|
||||
var serverAllActive bool = strings.ToLower(split[2]) == strings.ToLower(strings.Join(settingServers[:], "+"))
|
||||
|
||||
// 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 = `
|
||||
<li class="nav-item">
|
||||
<a class="nav-link" href="` + ipv4_url + `"> IPv4 </a>
|
||||
</li>
|
||||
<li class="nav-item">
|
||||
<a class="nav-link" href="` + ipv6_url + `"> IPv6 </a>
|
||||
</li>
|
||||
// Print the IPv4, IPv6, All Servers link in navbar
|
||||
var serverNavigation string = `
|
||||
<li class="nav-item"><a class="nav-link" href="` + ipv4URL + `"` + (map[bool]string{true: " active"})[strings.ToLower(split[0]) == "ipv4"] + `> IPv4 </a></li>
|
||||
<li class="nav-item"><a class="nav-link" href="` + ipv6URL + `"` + (map[bool]string{true: " active"})[strings.ToLower(split[0]) == "ipv6"] + `> IPv6 </a></li>
|
||||
<span class="navbar-text">|</span>
|
||||
<li class="nav-item">
|
||||
<a class="nav-link` + serverAllActive + `" href="` + all_url + `"> All Servers </a>
|
||||
</li>
|
||||
`
|
||||
<a class="nav-link` + (map[bool]string{true: " active"})[serverAllActive] + `" href="` + allURL + `"> All Servers </a>
|
||||
</li>`
|
||||
|
||||
// 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, "/")
|
||||
// Add a link for each of the servers
|
||||
for _, server := range settingServers {
|
||||
var serverActive string
|
||||
if split[2] == server {
|
||||
serverActive = " active"
|
||||
}
|
||||
serverURL := "/" + strings.Join([]string{split[0], split[1], server, strings.Join(split[3:], "/")}, "/")
|
||||
|
||||
serverNavigation += `
|
||||
serverNavigation += `
|
||||
<li class="nav-item">
|
||||
<a class="nav-link` + serverActive + `" href="` + server_url + `">` + server + `</a>
|
||||
</li>
|
||||
`
|
||||
}
|
||||
<a class="nav-link` + serverActive + `" href="` + serverURL + `">` + server + `</a>
|
||||
</li>`
|
||||
}
|
||||
|
||||
// Add the options in navbar form, and check if they are active
|
||||
var options string
|
||||
split = strings.Split(path, "/")
|
||||
if split[2] == "summary" {
|
||||
options += `<option value="summary" selected>show protocol</option>`
|
||||
} else {
|
||||
options += `<option value="summary">show protocol</option>`
|
||||
}
|
||||
if split[2] == "route" {
|
||||
options += `<option value="route" selected>show route for ...</option>`
|
||||
} else {
|
||||
options += `<option value="route">show route for ...</option>`
|
||||
}
|
||||
if split[2] == "route_all" {
|
||||
options += `<option value="route_all" selected>show route for ... all</option>`
|
||||
} else {
|
||||
options += `<option value="route_all">show route for ... all</option>`
|
||||
}
|
||||
if split[2] == "route_where" {
|
||||
options += `<option value="route_where" selected>show route where net ~ [ ... ]</option>`
|
||||
} else {
|
||||
options += `<option value="route_where">show route where net ~ [ ... ]</option>`
|
||||
}
|
||||
if split[2] == "route_where_all" {
|
||||
options += `<option value="route_where_all" selected>show route where net ~ [ ... ] all</option>`
|
||||
} else {
|
||||
options += `<option value="route_where_all">show route where net ~ [ ... ] all</option>`
|
||||
}
|
||||
if isWhois {
|
||||
options += `<option value="whois" selected>whois ...</option>`
|
||||
} else {
|
||||
options += `<option value="whois">whois ...</option>`
|
||||
}
|
||||
if split[2] == "traceroute" {
|
||||
options += `<option value="traceroute" selected>traceroute ...</option>`
|
||||
} else {
|
||||
options += `<option value="traceroute">traceroute ...</option>`
|
||||
}
|
||||
// Add the options in navbar form, and check if they are active
|
||||
var optionKeys = []string{"summary", "detail", "route", "route_all", "route_where", "route_where_all", "whois", "traceroute"}
|
||||
var optionDisplays = []string{
|
||||
"show protocol",
|
||||
"show protocol all",
|
||||
"show route for ...",
|
||||
"show route for ... all",
|
||||
"show route where net ~ [ ... ]",
|
||||
"show route where net ~ [ ... ] all",
|
||||
"whois ...",
|
||||
"traceroute ...",
|
||||
}
|
||||
|
||||
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 = strings.Join(whoisSplit[2:], "/")
|
||||
} else if len(split) >= 5 {
|
||||
// This is a normal request, just extract the target
|
||||
target = strings.Join(split[4:], "/")
|
||||
}
|
||||
var options string
|
||||
for optionKeyID, optionKey := range optionKeys {
|
||||
options += "<option value=\"" + optionKey + "\""
|
||||
if (optionKey == "whois" && isWhois) || optionKey == split[1] {
|
||||
options += " selected"
|
||||
}
|
||||
options += ">" + optionDisplays[optionKeyID] + "</option>"
|
||||
}
|
||||
|
||||
w.Write([]byte(`
|
||||
var target string
|
||||
if isWhois {
|
||||
// This is a whois request, use original path URL instead of the modified one
|
||||
// and extract the target
|
||||
target = whoisTarget
|
||||
} else if len(split) >= 4 {
|
||||
// This is a normal request, just extract the target
|
||||
target = strings.Join(split[3:], "/")
|
||||
}
|
||||
|
||||
w.Write([]byte(`
|
||||
<!DOCTYPE html>
|
||||
<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="zh-CN" lang="zh-CN" class="no-js">
|
||||
<head>
|
||||
@ -162,8 +124,8 @@ func templateHeader(w http.ResponseWriter, r *http.Request, title string) {
|
||||
<form class="form-inline" action="/redir" method="GET">
|
||||
<div class="input-group">
|
||||
<select name="action" class="form-control">` + options + `</select>
|
||||
<input name="proto" class="d-none" value="` + split[1] + `">
|
||||
<input name="server" class="d-none" value="` + split[3] + `">
|
||||
<input name="proto" class="d-none" value="` + split[0] + `">
|
||||
<input name="server" class="d-none" value="` + split[2] + `">
|
||||
<input name="target" class="form-control" placeholder="Target" aria-label="Target" value="` + target + `">
|
||||
<div class="input-group-append">
|
||||
<button class="btn btn-outline-success" type="submit">»</button>
|
||||
@ -179,7 +141,7 @@ func templateHeader(w http.ResponseWriter, r *http.Request, title string) {
|
||||
|
||||
// Print HTML footer to http response
|
||||
func templateFooter(w http.ResponseWriter) {
|
||||
w.Write([]byte(`
|
||||
w.Write([]byte(`
|
||||
</div>
|
||||
</body>
|
||||
</html>
|
||||
@ -189,120 +151,112 @@ func templateFooter(w http.ResponseWriter) {
|
||||
// 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("<pre>"))
|
||||
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(" "))
|
||||
}
|
||||
w.Write([]byte("<pre>"))
|
||||
for _, line := range strings.Split(s, "\n") {
|
||||
var isASes bool = false
|
||||
|
||||
if isIP(word) {
|
||||
// Add whois link to the IP, handles IPv4 and IPv6
|
||||
w.Write([]byte("<a href=\"/whois/" + word + "\">" + word + "</a>"))
|
||||
} 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("<a href=\"/whois/" + strings.Split(word, "%")[0] + "\">" + strings.Split(word, "%")[0] + "</a>"))
|
||||
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("<a href=\"/whois/" + strings.Split(word, "/")[0] + "\">" + strings.Split(word, "/")[0] + "</a>"))
|
||||
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("<a href=\"/whois/AS" + word + "\">" + word + "</a>"))
|
||||
} else {
|
||||
// Just an ordinary word, print it and done
|
||||
w.Write([]byte(word))
|
||||
}
|
||||
}
|
||||
}
|
||||
w.Write([]byte("\n"))
|
||||
}
|
||||
w.Write([]byte("</pre>"))
|
||||
var lineFormatted string
|
||||
words := strings.Split(line, " ")
|
||||
|
||||
for wordID, word := range words {
|
||||
if len(word) == 0 {
|
||||
continue
|
||||
}
|
||||
if wordID > 0 && (len(words[wordID-1]) == 0 || words[wordID-1][len(words[wordID-1])-1] == ':') {
|
||||
// Insert TAB if there are multiple spaces before this word
|
||||
lineFormatted += "\t"
|
||||
} else {
|
||||
lineFormatted += " "
|
||||
}
|
||||
|
||||
if isIP(word) {
|
||||
// Add whois link to the IP, handles IPv4 and IPv6
|
||||
lineFormatted += "<a href=\"/whois/" + word + "\">" + word + "</a>"
|
||||
} 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
|
||||
lineFormatted += "<a href=\"/whois/" + strings.Split(word, "%")[0] + "\">" + strings.Split(word, "%")[0] + "</a>"
|
||||
lineFormatted += "%" + 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
|
||||
lineFormatted += "<a href=\"/whois/" + strings.Split(word, "/")[0] + "\">" + strings.Split(word, "/")[0] + "</a>"
|
||||
lineFormatted += "/" + strings.Split(word, "/")[1]
|
||||
} else if word == "AS:" || word == "\tBGP.as_path:" {
|
||||
// Bird will output ASNs later
|
||||
isASes = true
|
||||
lineFormatted += word
|
||||
} else if isASes && isNumber(word) {
|
||||
// Bird is outputing ASNs, ass whois for them
|
||||
lineFormatted += "<a href=\"/whois/AS" + word + "\">" + word + "</a>"
|
||||
} else {
|
||||
// Just an ordinary word, print it and done
|
||||
lineFormatted += word
|
||||
}
|
||||
}
|
||||
lineFormatted += "\n"
|
||||
w.Write([]byte(lineFormatted))
|
||||
}
|
||||
w.Write([]byte("</pre>"))
|
||||
}
|
||||
|
||||
// Output a table for the summary page
|
||||
func summaryTable(w http.ResponseWriter, isIPv6 bool, data string, serverName string) {
|
||||
w.Write([]byte("<table class=\"table table-striped table-bordered table-sm\">"))
|
||||
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
|
||||
}
|
||||
}
|
||||
// w.Write([]byte("<pre>" + data + "</pre>"))
|
||||
w.Write([]byte("<table class=\"table table-striped table-bordered table-sm\">"))
|
||||
for lineID, line := range strings.Split(data, "\n") {
|
||||
var row [6]string
|
||||
var rowIndex int = 0
|
||||
|
||||
// Ignore empty lines
|
||||
if len(row[0]) == 0 {
|
||||
continue
|
||||
}
|
||||
words := strings.Split(line, " ")
|
||||
for wordID, word := range words {
|
||||
if len(word) == 0 {
|
||||
continue
|
||||
}
|
||||
if rowIndex < 4 {
|
||||
row[rowIndex] += word
|
||||
rowIndex++
|
||||
} else if len(words[wordID-1]) == 0 && rowIndex < len(row)-1 {
|
||||
if len(row[rowIndex]) > 0 {
|
||||
rowIndex++
|
||||
}
|
||||
row[rowIndex] += word
|
||||
} else {
|
||||
row[rowIndex] += " " + word
|
||||
}
|
||||
}
|
||||
|
||||
if lineId == 0 {
|
||||
// Draw the table head
|
||||
w.Write([]byte("<thead>"))
|
||||
for i := 0; i < 6; i++ {
|
||||
w.Write([]byte("<th scope=\"col\">" + row[i] + "</th>"))
|
||||
}
|
||||
w.Write([]byte("</thead><tbody>"))
|
||||
} else {
|
||||
// Draw the row in red if the link isn't up
|
||||
if row[3] == "up" {
|
||||
w.Write([]byte("<tr>"))
|
||||
} else if lineId != 0 {
|
||||
w.Write([]byte("<tr class=\"table-danger\">"))
|
||||
}
|
||||
// Add link to detail for first column
|
||||
if isIPv6 {
|
||||
w.Write([]byte("<td><a href=\"/ipv6/detail/" + serverName + "/" + row[0] + "\">" + row[0] + "</a></td>"))
|
||||
} else {
|
||||
w.Write([]byte("<td><a href=\"/ipv4/detail/" + serverName + "/" + row[0] + "\">" + row[0] + "</a></td>"))
|
||||
}
|
||||
// Draw the other cells
|
||||
for i := 1; i < 6; i++ {
|
||||
w.Write([]byte("<td>" + row[i] + "</td>"))
|
||||
}
|
||||
w.Write([]byte("</tr>"))
|
||||
}
|
||||
}
|
||||
w.Write([]byte("</tbody></table>"))
|
||||
// Ignore empty lines
|
||||
if len(row[0]) == 0 {
|
||||
continue
|
||||
}
|
||||
|
||||
if lineID == 0 {
|
||||
// Draw the table head
|
||||
w.Write([]byte("<thead>"))
|
||||
for i := 0; i < 6; i++ {
|
||||
w.Write([]byte("<th scope=\"col\">" + row[i] + "</th>"))
|
||||
}
|
||||
w.Write([]byte("</thead><tbody>"))
|
||||
} else {
|
||||
// Draw the row in red if the link isn't up
|
||||
w.Write([]byte("<tr class=\"" + (map[string]string{
|
||||
"up": "table-success",
|
||||
"down": "table-danger",
|
||||
"start": "table-danger",
|
||||
})[row[3]] + "\">"))
|
||||
// Add link to detail for first column
|
||||
if isIPv6 {
|
||||
w.Write([]byte("<td><a href=\"/ipv6/detail/" + serverName + "/" + row[0] + "\">" + row[0] + "</a></td>"))
|
||||
} else {
|
||||
w.Write([]byte("<td><a href=\"/ipv4/detail/" + serverName + "/" + row[0] + "\">" + row[0] + "</a></td>"))
|
||||
}
|
||||
// Draw the other cells
|
||||
for i := 1; i < 6; i++ {
|
||||
w.Write([]byte("<td>" + row[i] + "</td>"))
|
||||
}
|
||||
w.Write([]byte("</tr>"))
|
||||
}
|
||||
}
|
||||
w.Write([]byte("</tbody></table>"))
|
||||
}
|
||||
|
||||
@ -1,144 +1,87 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"strings"
|
||||
"html"
|
||||
"html"
|
||||
"net/http"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func webDispatcherIPv4Summary(w http.ResponseWriter, r *http.Request) {
|
||||
split := strings.Split(r.URL.Path[len("/ipv4/summary/"):], "/")
|
||||
webHandler(w, r, "bird", split[0], "show protocols")
|
||||
func webHandlerWhois(w http.ResponseWriter, r *http.Request) {
|
||||
var target string = r.URL.Path[len("/whois/"):]
|
||||
|
||||
templateHeader(w, r, "Bird-lg Go - whois "+html.EscapeString(target))
|
||||
|
||||
w.Write([]byte("<h2>whois " + html.EscapeString(target) + "</h2>"))
|
||||
smartWriter(w, whois(target))
|
||||
|
||||
templateFooter(w)
|
||||
}
|
||||
|
||||
func webDispatcherIPv6Summary(w http.ResponseWriter, r *http.Request) {
|
||||
split := strings.Split(r.URL.Path[len("/ipv6/summary/"):], "/")
|
||||
webHandler(w, r, "bird6", split[0], "show protocols")
|
||||
func webBackendCommunicator(w http.ResponseWriter, r *http.Request, endpoint string, command string) {
|
||||
split := strings.Split(r.URL.Path[1:], "/")
|
||||
urlCommands := strings.Join(split[3:], "/")
|
||||
|
||||
command = (map[string]string{
|
||||
"summary": "show protocols",
|
||||
"detail": "show protocols all " + urlCommands,
|
||||
"route": "show route for " + urlCommands,
|
||||
"route_all": "show route for " + urlCommands + " all",
|
||||
"route_where": "show route where net ~ [ " + urlCommands + " ]",
|
||||
"route_where_all": "show route where net ~ [ " + urlCommands + " ] all",
|
||||
"traceroute": urlCommands,
|
||||
})[command]
|
||||
|
||||
templateHeader(w, r, "Bird-lg Go - "+html.EscapeString(endpoint+" "+command))
|
||||
|
||||
var servers []string = strings.Split(split[2], "+")
|
||||
|
||||
var responses []string = batchRequest(servers, endpoint, command)
|
||||
for i, response := range responses {
|
||||
w.Write([]byte("<h2>" + html.EscapeString(servers[i]) + ": " + html.EscapeString(command) + "</h2>"))
|
||||
if (endpoint == "bird" || endpoint == "bird6") && command == "show protocols" && len(response) > 4 && strings.ToLower(response[0:4]) == "name" {
|
||||
var isIPv6 bool = endpoint[len(endpoint)-1] == '6'
|
||||
summaryTable(w, isIPv6, response, servers[i])
|
||||
} else {
|
||||
smartWriter(w, response)
|
||||
}
|
||||
}
|
||||
|
||||
templateFooter(w)
|
||||
}
|
||||
|
||||
func webDispatcherIPv4Detail(w http.ResponseWriter, r *http.Request) {
|
||||
split := strings.Split(r.URL.Path[len("/ipv4/detail/"):], "/")
|
||||
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, "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, "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, "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, "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, "bird6", split[0], "show route for " + strings.Join(split[1:], "/") + " all")
|
||||
}
|
||||
|
||||
func webDispatcherIPv4RouteWhere(w http.ResponseWriter, r *http.Request) {
|
||||
split := strings.Split(r.URL.Path[len("/ipv4/route_where/"):], "/")
|
||||
webHandler(w, r, "bird", split[0], "show route where net ~ [ " + strings.Join(split[1:], "/") + " ]")
|
||||
}
|
||||
|
||||
func webDispatcherIPv6RouteWhere(w http.ResponseWriter, r *http.Request) {
|
||||
split := strings.Split(r.URL.Path[len("/ipv6/route_where/"):], "/")
|
||||
webHandler(w, r, "bird6", split[0], "show route where net ~ [ " + strings.Join(split[1:], "/") + " ]")
|
||||
}
|
||||
|
||||
func webDispatcherIPv4RouteWhereAll(w http.ResponseWriter, r *http.Request) {
|
||||
split := strings.Split(r.URL.Path[len("/ipv4/route_where_all/"):], "/")
|
||||
webHandler(w, r, "bird", split[0], "show route where net ~ [ " + strings.Join(split[1:], "/") + " ] all")
|
||||
}
|
||||
|
||||
func webDispatcherIPv6RouteWhereAll(w http.ResponseWriter, r *http.Request) {
|
||||
split := strings.Split(r.URL.Path[len("/ipv6/route_where_all/"):], "/")
|
||||
webHandler(w, r, "bird6", split[0], "show route where net ~ [ " + 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))
|
||||
|
||||
w.Write([]byte("<h2>whois " + html.EscapeString(target) + "</h2>"))
|
||||
smartWriter(w, whois(target))
|
||||
|
||||
templateFooter(w)
|
||||
}
|
||||
|
||||
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 = batchRequest(servers, endpoint, command)
|
||||
for i, response := range responses {
|
||||
w.Write([]byte("<h2>" + html.EscapeString(servers[i]) + ": " + html.EscapeString(command) + "</h2>"))
|
||||
if (endpoint == "bird" || endpoint == "bird6") && command == "show protocols" && strings.ToLower(response[0:4]) == "name" {
|
||||
var isIPv6 bool = endpoint[len(endpoint) - 1] == '6'
|
||||
summaryTable(w, isIPv6, response, servers[i])
|
||||
} else {
|
||||
smartWriter(w, response)
|
||||
}
|
||||
}
|
||||
|
||||
templateFooter(w)
|
||||
}
|
||||
|
||||
func defaultRedirect(w http.ResponseWriter, r *http.Request) {
|
||||
http.Redirect(w, r, "/ipv4/summary/" + strings.Join(settingServers[:], "+"), 302)
|
||||
}
|
||||
|
||||
func navbarFormRedirect(w http.ResponseWriter, r *http.Request) {
|
||||
query := r.URL.Query()
|
||||
if query.Get("action") == "whois" {
|
||||
http.Redirect(w, r, "/" + query.Get("action") + "/" + query.Get("target"), 302)
|
||||
} else if query.Get("action") == "summary" {
|
||||
http.Redirect(w, r, "/" + query.Get("proto") + "/" + query.Get("action") + "/" + query.Get("server"), 302)
|
||||
} else {
|
||||
http.Redirect(w, r, "/" + query.Get("proto") + "/" + query.Get("action") + "/" + query.Get("server") + "/" + query.Get("target"), 302)
|
||||
}
|
||||
func webHandlerNavbarFormRedirect(w http.ResponseWriter, r *http.Request) {
|
||||
query := r.URL.Query()
|
||||
if query.Get("action") == "whois" {
|
||||
http.Redirect(w, r, "/"+query.Get("action")+"/"+query.Get("target"), 302)
|
||||
} else if query.Get("action") == "summary" {
|
||||
http.Redirect(w, r, "/"+query.Get("proto")+"/"+query.Get("action")+"/"+query.Get("server"), 302)
|
||||
} else {
|
||||
http.Redirect(w, r, "/"+query.Get("proto")+"/"+query.Get("action")+"/"+query.Get("server")+"/"+query.Get("target"), 302)
|
||||
}
|
||||
}
|
||||
|
||||
func webServerStart() {
|
||||
// Start HTTP server
|
||||
http.HandleFunc("/", defaultRedirect)
|
||||
http.HandleFunc("/ipv4/summary/", webDispatcherIPv4Summary)
|
||||
http.HandleFunc("/ipv6/summary/", webDispatcherIPv6Summary)
|
||||
http.HandleFunc("/ipv4/detail/", webDispatcherIPv4Detail)
|
||||
http.HandleFunc("/ipv6/detail/", webDispatcherIPv6Detail)
|
||||
http.HandleFunc("/ipv4/route/", webDispatcherIPv4Route)
|
||||
http.HandleFunc("/ipv6/route/", webDispatcherIPv6Route)
|
||||
http.HandleFunc("/ipv4/route_all/", webDispatcherIPv4RouteAll)
|
||||
http.HandleFunc("/ipv6/route_all/", webDispatcherIPv6RouteAll)
|
||||
http.HandleFunc("/ipv4/route_where/", webDispatcherIPv4RouteWhere)
|
||||
http.HandleFunc("/ipv6/route_where/", webDispatcherIPv6RouteWhere)
|
||||
http.HandleFunc("/ipv4/route_where_all/", webDispatcherIPv4RouteWhereAll)
|
||||
http.HandleFunc("/ipv6/route_where_all/", webDispatcherIPv6RouteWhereAll)
|
||||
http.HandleFunc("/ipv4/traceroute/", webDispatcherIPv4Traceroute)
|
||||
http.HandleFunc("/ipv6/traceroute/", webDispatcherIPv6Traceroute)
|
||||
http.HandleFunc("/whois/", webDispatcherWhois)
|
||||
http.HandleFunc("/redir/", navbarFormRedirect)
|
||||
http.ListenAndServe(settingListen, nil)
|
||||
// Start HTTP server
|
||||
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||||
http.Redirect(w, r, "/ipv4/summary/"+strings.Join(settingServers[:], "+"), 302)
|
||||
})
|
||||
http.HandleFunc("/ipv4/summary/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "bird", "summary") })
|
||||
http.HandleFunc("/ipv6/summary/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "bird6", "summary") })
|
||||
http.HandleFunc("/ipv4/detail/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "bird", "detail") })
|
||||
http.HandleFunc("/ipv6/detail/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "bird6", "detail") })
|
||||
http.HandleFunc("/ipv4/route/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "bird", "route") })
|
||||
http.HandleFunc("/ipv6/route/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "bird6", "route") })
|
||||
http.HandleFunc("/ipv4/route_all/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "bird", "route_all") })
|
||||
http.HandleFunc("/ipv6/route_all/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "bird6", "route_all") })
|
||||
http.HandleFunc("/ipv4/route_where/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "bird", "route_where") })
|
||||
http.HandleFunc("/ipv6/route_where/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "bird6", "route_where") })
|
||||
http.HandleFunc("/ipv4/route_where_all/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "bird", "route_where_all") })
|
||||
http.HandleFunc("/ipv6/route_where_all/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "bird6", "route_where_all") })
|
||||
http.HandleFunc("/ipv4/traceroute/", func(w http.ResponseWriter, r *http.Request) { webBackendCommunicator(w, r, "traceroute", "traceroute") })
|
||||
http.HandleFunc("/ipv6/traceroute/", func(w http.ResponseWriter, r *http.Request) {
|
||||
webBackendCommunicator(w, r, "traceroute6", "traceroute")
|
||||
})
|
||||
http.HandleFunc("/whois/", webHandlerWhois)
|
||||
http.HandleFunc("/redir/", webHandlerNavbarFormRedirect)
|
||||
http.ListenAndServe(settingListen, nil)
|
||||
}
|
||||
|
||||
144
proxy/bird.go
144
proxy/bird.go
@ -1,10 +1,10 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"io"
|
||||
"net"
|
||||
"net/http"
|
||||
"sync"
|
||||
"io"
|
||||
"net"
|
||||
"net/http"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// BIRDv4 connection & mutex lock
|
||||
@ -18,83 +18,93 @@ var bird6Mutex = &sync.Mutex{}
|
||||
// Read a line from bird socket, removing preceding status number, output it.
|
||||
// Returns if there are more lines.
|
||||
func birdReadln(bird io.Reader, w io.Writer) bool {
|
||||
// Read from socket byte by byte, until reaching newline character
|
||||
c := make([]byte, 1024, 1024)
|
||||
pos := 0
|
||||
for {
|
||||
if pos >= 1024 { break }
|
||||
_, err := bird.Read(c[pos:pos+1])
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
if c[pos] == byte('\n') {
|
||||
break
|
||||
}
|
||||
pos++
|
||||
}
|
||||
// Read from socket byte by byte, until reaching newline character
|
||||
c := make([]byte, 1024, 1024)
|
||||
pos := 0
|
||||
for {
|
||||
if pos >= 1024 {
|
||||
break
|
||||
}
|
||||
_, err := bird.Read(c[pos : pos+1])
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
if c[pos] == byte('\n') {
|
||||
break
|
||||
}
|
||||
pos++
|
||||
}
|
||||
|
||||
c = c[:pos+1]
|
||||
// print(string(c[:]))
|
||||
c = c[:pos+1]
|
||||
// print(string(c[:]))
|
||||
|
||||
// Remove preceding status number, different situations
|
||||
if pos < 4 {
|
||||
// Line is too short to have a status number
|
||||
if w != nil {
|
||||
pos = 0
|
||||
for c[pos] == byte(' ') { pos++ }
|
||||
w.Write(c[pos:])
|
||||
}
|
||||
return true
|
||||
} else if isNumeric(c[0]) && isNumeric(c[1]) && isNumeric(c[2]) && isNumeric(c[3]) {
|
||||
// There is a status number at beginning, remove first 5 bytes
|
||||
if w != nil && pos > 6 {
|
||||
pos = 5
|
||||
for c[pos] == byte(' ') { pos++ }
|
||||
w.Write(c[pos:])
|
||||
}
|
||||
return c[0] != byte('0') && c[0] != byte('8') && c[0] != byte('9')
|
||||
} else {
|
||||
// There is no status number, only remove preceding spaces
|
||||
if w != nil {
|
||||
pos = 0
|
||||
for c[pos] == byte(' ') { pos++ }
|
||||
w.Write(c[pos:])
|
||||
}
|
||||
return true
|
||||
}
|
||||
// Remove preceding status number, different situations
|
||||
if pos < 4 {
|
||||
// Line is too short to have a status number
|
||||
if w != nil {
|
||||
pos = 0
|
||||
for c[pos] == byte(' ') {
|
||||
pos++
|
||||
}
|
||||
w.Write(c[pos:])
|
||||
}
|
||||
return true
|
||||
} else if isNumeric(c[0]) && isNumeric(c[1]) && isNumeric(c[2]) && isNumeric(c[3]) {
|
||||
// There is a status number at beginning, remove first 5 bytes
|
||||
if w != nil && pos > 6 {
|
||||
pos = 5
|
||||
for c[pos] == byte(' ') {
|
||||
pos++
|
||||
}
|
||||
w.Write(c[pos:])
|
||||
}
|
||||
return c[0] != byte('0') && c[0] != byte('8') && c[0] != byte('9')
|
||||
} else {
|
||||
// There is no status number, only remove preceding spaces
|
||||
if w != nil {
|
||||
pos = 0
|
||||
for c[pos] == byte(' ') {
|
||||
pos++
|
||||
}
|
||||
w.Write(c[pos:])
|
||||
}
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
// Write a command to a bird socket
|
||||
func birdWriteln(bird io.Writer, s string) {
|
||||
bird.Write([]byte(s + "\n"))
|
||||
bird.Write([]byte(s + "\n"))
|
||||
}
|
||||
|
||||
// Handles BIRDv4 queries
|
||||
func birdHandler(httpW http.ResponseWriter, httpR *http.Request) {
|
||||
query := string(httpR.URL.Query().Get("q"))
|
||||
if query == "" {
|
||||
invalidHandler(httpW, httpR)
|
||||
} else {
|
||||
birdMutex.Lock()
|
||||
defer birdMutex.Unlock()
|
||||
query := string(httpR.URL.Query().Get("q"))
|
||||
if query == "" {
|
||||
invalidHandler(httpW, httpR)
|
||||
} else {
|
||||
birdMutex.Lock()
|
||||
defer birdMutex.Unlock()
|
||||
|
||||
println(query)
|
||||
birdWriteln(bird, query)
|
||||
for birdReadln(bird, httpW) {}
|
||||
}
|
||||
println(query)
|
||||
birdWriteln(bird, query)
|
||||
for birdReadln(bird, httpW) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Handles BIRDv6 queries
|
||||
func bird6Handler(httpW http.ResponseWriter, httpR *http.Request) {
|
||||
query := string(httpR.URL.Query().Get("q"))
|
||||
if query == "" {
|
||||
invalidHandler(httpW, httpR)
|
||||
} else {
|
||||
bird6Mutex.Lock()
|
||||
defer bird6Mutex.Unlock()
|
||||
query := string(httpR.URL.Query().Get("q"))
|
||||
if query == "" {
|
||||
invalidHandler(httpW, httpR)
|
||||
} else {
|
||||
bird6Mutex.Lock()
|
||||
defer bird6Mutex.Unlock()
|
||||
|
||||
println(query)
|
||||
birdWriteln(bird6, query)
|
||||
for birdReadln(bird6, httpW) {}
|
||||
}
|
||||
println(query)
|
||||
birdWriteln(bird6, query)
|
||||
for birdReadln(bird6, httpW) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
102
proxy/main.go
102
proxy/main.go
@ -1,75 +1,75 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net"
|
||||
"net/http"
|
||||
"flag"
|
||||
"os"
|
||||
"flag"
|
||||
"net"
|
||||
"net/http"
|
||||
"os"
|
||||
)
|
||||
|
||||
// Check if a byte is character for number
|
||||
func isNumeric(b byte) bool {
|
||||
return b >= byte('0') && b <= byte('9')
|
||||
return b >= byte('0') && b <= byte('9')
|
||||
}
|
||||
|
||||
// Default handler, returns 500 Internal Server Error
|
||||
func invalidHandler(httpW http.ResponseWriter, httpR *http.Request) {
|
||||
httpW.WriteHeader(http.StatusInternalServerError)
|
||||
httpW.Write([]byte("Invalid Request\n"))
|
||||
httpW.WriteHeader(http.StatusInternalServerError)
|
||||
httpW.Write([]byte("Invalid Request\n"))
|
||||
}
|
||||
|
||||
// Wrapper of tracer
|
||||
func main() {
|
||||
var err error
|
||||
var err error
|
||||
|
||||
// Prepare default socket paths, use environment variable if possible
|
||||
birdSocketDefault := "/var/run/bird/bird.ctl"
|
||||
bird6SocketDefault := "/var/run/bird/bird6.ctl"
|
||||
listenDefault := ":8000"
|
||||
// Prepare default socket paths, use environment variable if possible
|
||||
birdSocketDefault := "/var/run/bird/bird.ctl"
|
||||
bird6SocketDefault := "/var/run/bird/bird6.ctl"
|
||||
listenDefault := ":8000"
|
||||
|
||||
if birdSocketEnv := os.Getenv("BIRD_SOCKET"); birdSocketEnv != "" {
|
||||
birdSocketDefault = birdSocketEnv
|
||||
}
|
||||
if bird6SocketEnv := os.Getenv("BIRD6_SOCKET"); bird6SocketEnv != "" {
|
||||
bird6SocketDefault = bird6SocketEnv
|
||||
}
|
||||
if listenEnv := os.Getenv("BIRDLG_LISTEN"); listenEnv != "" {
|
||||
listenDefault = listenEnv
|
||||
}
|
||||
if birdSocketEnv := os.Getenv("BIRD_SOCKET"); birdSocketEnv != "" {
|
||||
birdSocketDefault = birdSocketEnv
|
||||
}
|
||||
if bird6SocketEnv := os.Getenv("BIRD6_SOCKET"); bird6SocketEnv != "" {
|
||||
bird6SocketDefault = bird6SocketEnv
|
||||
}
|
||||
if listenEnv := os.Getenv("BIRDLG_LISTEN"); listenEnv != "" {
|
||||
listenDefault = listenEnv
|
||||
}
|
||||
|
||||
// Allow parameters to override environment variables
|
||||
birdParam := flag.String("bird", birdSocketDefault, "socket file for bird, set either in parameter or environment variable BIRD_SOCKET")
|
||||
bird6Param := flag.String("bird6", bird6SocketDefault, "socket file for bird6, set either in parameter or environment variable BIRD6_SOCKET")
|
||||
listenParam := flag.String("listen", listenDefault, "listen address, set either in parameter or environment variable BIRDLG_LISTEN")
|
||||
flag.Parse()
|
||||
// Allow parameters to override environment variables
|
||||
birdParam := flag.String("bird", birdSocketDefault, "socket file for bird, set either in parameter or environment variable BIRD_SOCKET")
|
||||
bird6Param := flag.String("bird6", bird6SocketDefault, "socket file for bird6, set either in parameter or environment variable BIRD6_SOCKET")
|
||||
listenParam := flag.String("listen", listenDefault, "listen address, set either in parameter or environment variable BIRDLG_LISTEN")
|
||||
flag.Parse()
|
||||
|
||||
// Initialize BIRDv4 socket
|
||||
bird, err = net.Dial("unix", *birdParam)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
defer bird.Close()
|
||||
// Initialize BIRDv4 socket
|
||||
bird, err = net.Dial("unix", *birdParam)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
defer bird.Close()
|
||||
|
||||
birdReadln(bird, nil)
|
||||
birdWriteln(bird, "restrict")
|
||||
birdReadln(bird, nil)
|
||||
birdReadln(bird, nil)
|
||||
birdWriteln(bird, "restrict")
|
||||
birdReadln(bird, nil)
|
||||
|
||||
// Initialize BIRDv6 socket
|
||||
bird6, err = net.Dial("unix", *bird6Param)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
defer bird6.Close()
|
||||
// Initialize BIRDv6 socket
|
||||
bird6, err = net.Dial("unix", *bird6Param)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
defer bird6.Close()
|
||||
|
||||
birdReadln(bird6, nil)
|
||||
birdWriteln(bird6, "restrict")
|
||||
birdReadln(bird6, nil)
|
||||
birdReadln(bird6, nil)
|
||||
birdWriteln(bird6, "restrict")
|
||||
birdReadln(bird6, nil)
|
||||
|
||||
// Start HTTP server
|
||||
http.HandleFunc("/", invalidHandler)
|
||||
http.HandleFunc("/bird", birdHandler)
|
||||
http.HandleFunc("/bird6", bird6Handler)
|
||||
http.HandleFunc("/traceroute", tracerouteIPv4Wrapper)
|
||||
http.HandleFunc("/traceroute6", tracerouteIPv6Wrapper)
|
||||
http.ListenAndServe(*listenParam, nil)
|
||||
// Start HTTP server
|
||||
http.HandleFunc("/", invalidHandler)
|
||||
http.HandleFunc("/bird", birdHandler)
|
||||
http.HandleFunc("/bird6", bird6Handler)
|
||||
http.HandleFunc("/traceroute", tracerouteIPv4Wrapper)
|
||||
http.HandleFunc("/traceroute6", tracerouteIPv6Wrapper)
|
||||
http.ListenAndServe(*listenParam, nil)
|
||||
}
|
||||
|
||||
@ -1,67 +1,75 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"runtime"
|
||||
"os/exec"
|
||||
"net/http"
|
||||
"os/exec"
|
||||
"runtime"
|
||||
)
|
||||
|
||||
// Wrapper of traceroute, IPv4
|
||||
func tracerouteIPv4Wrapper(httpW http.ResponseWriter, httpR *http.Request) {
|
||||
tracerouteRealHandler(false, httpW, httpR)
|
||||
tracerouteRealHandler(false, httpW, httpR)
|
||||
}
|
||||
|
||||
// Wrapper of traceroute, IPv6
|
||||
func tracerouteIPv6Wrapper(httpW http.ResponseWriter, httpR *http.Request) {
|
||||
tracerouteRealHandler(true, httpW, httpR)
|
||||
tracerouteRealHandler(true, httpW, httpR)
|
||||
}
|
||||
|
||||
// Real handler of traceroute requests
|
||||
func tracerouteRealHandler(useIPv6 bool, httpW http.ResponseWriter, httpR *http.Request) {
|
||||
query := string(httpR.URL.Query().Get("q"))
|
||||
if query == "" {
|
||||
invalidHandler(httpW, httpR)
|
||||
} else {
|
||||
var cmd string
|
||||
var args []string
|
||||
if runtime.GOOS == "freebsd" || runtime.GOOS == "netbsd" {
|
||||
if useIPv6 { cmd = "traceroute6" } else { cmd = "traceroute" }
|
||||
args = []string{"-a", "-q1", "-w1", "-m15", query}
|
||||
} else if runtime.GOOS == "openbsd" {
|
||||
if useIPv6 { cmd = "traceroute6" } else { cmd = "traceroute" }
|
||||
args = []string{"-A", "-q1", "-w1", "-m15", query}
|
||||
} else if runtime.GOOS == "linux" {
|
||||
cmd = "traceroute"
|
||||
if useIPv6 {
|
||||
args = []string{"-6", "-A", "-q1", "-N32", "-w1", "-m15", query}
|
||||
} else {
|
||||
args = []string{"-4", "-A", "-q1", "-N32", "-w1", "-m15", query}
|
||||
}
|
||||
} else {
|
||||
httpW.WriteHeader(http.StatusInternalServerError)
|
||||
httpW.Write([]byte("Traceroute Not Supported\n"))
|
||||
return
|
||||
}
|
||||
instance := exec.Command(cmd, args...)
|
||||
output, err := instance.Output()
|
||||
if err != nil && runtime.GOOS == "linux" {
|
||||
// Standard traceroute utility failed, maybe system using busybox
|
||||
// Run with less parameters
|
||||
cmd = "traceroute"
|
||||
if useIPv6 {
|
||||
args = []string{"-6", "-q1", "-w1", "-m15", query}
|
||||
} else {
|
||||
args = []string{"-4", "-q1", "-w1", "-m15", query}
|
||||
}
|
||||
instance = exec.Command(cmd, args...)
|
||||
output, err = instance.Output()
|
||||
}
|
||||
if err != nil {
|
||||
httpW.WriteHeader(http.StatusInternalServerError)
|
||||
httpW.Write([]byte("Traceroute Execution Error: "))
|
||||
httpW.Write([]byte(err.Error() + "\n"))
|
||||
return
|
||||
}
|
||||
httpW.Write(output)
|
||||
}
|
||||
query := string(httpR.URL.Query().Get("q"))
|
||||
if query == "" {
|
||||
invalidHandler(httpW, httpR)
|
||||
} else {
|
||||
var cmd string
|
||||
var args []string
|
||||
if runtime.GOOS == "freebsd" || runtime.GOOS == "netbsd" {
|
||||
if useIPv6 {
|
||||
cmd = "traceroute6"
|
||||
} else {
|
||||
cmd = "traceroute"
|
||||
}
|
||||
args = []string{"-a", "-q1", "-w1", "-m15", query}
|
||||
} else if runtime.GOOS == "openbsd" {
|
||||
if useIPv6 {
|
||||
cmd = "traceroute6"
|
||||
} else {
|
||||
cmd = "traceroute"
|
||||
}
|
||||
args = []string{"-A", "-q1", "-w1", "-m15", query}
|
||||
} else if runtime.GOOS == "linux" {
|
||||
cmd = "traceroute"
|
||||
if useIPv6 {
|
||||
args = []string{"-6", "-A", "-q1", "-N32", "-w1", "-m15", query}
|
||||
} else {
|
||||
args = []string{"-4", "-A", "-q1", "-N32", "-w1", "-m15", query}
|
||||
}
|
||||
} else {
|
||||
httpW.WriteHeader(http.StatusInternalServerError)
|
||||
httpW.Write([]byte("Traceroute Not Supported\n"))
|
||||
return
|
||||
}
|
||||
instance := exec.Command(cmd, args...)
|
||||
output, err := instance.Output()
|
||||
if err != nil && runtime.GOOS == "linux" {
|
||||
// Standard traceroute utility failed, maybe system using busybox
|
||||
// Run with less parameters
|
||||
cmd = "traceroute"
|
||||
if useIPv6 {
|
||||
args = []string{"-6", "-q1", "-w1", "-m15", query}
|
||||
} else {
|
||||
args = []string{"-4", "-q1", "-w1", "-m15", query}
|
||||
}
|
||||
instance = exec.Command(cmd, args...)
|
||||
output, err = instance.Output()
|
||||
}
|
||||
if err != nil {
|
||||
httpW.WriteHeader(http.StatusInternalServerError)
|
||||
httpW.Write([]byte("Traceroute Execution Error: "))
|
||||
httpW.Write([]byte(err.Error() + "\n"))
|
||||
return
|
||||
}
|
||||
httpW.Write(output)
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user