Initial commit
This commit is contained in:
Lan Tian
commit
a9a07d4de4
51
frontend/lgproxy.go
Normal file
51
frontend/lgproxy.go
Normal file
@ -0,0 +1,51 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"net/url"
|
||||
"io/ioutil"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type channelData struct {
|
||||
id int
|
||||
data string
|
||||
}
|
||||
|
||||
func batchRequest(servers []string, endpoint string, command string) []string {
|
||||
var ch chan channelData = make(chan channelData)
|
||||
var response_array []string = make([]string, len(servers))
|
||||
|
||||
for i, server := range servers {
|
||||
var isValidServer bool = false
|
||||
for _, validServer := range settingServers {
|
||||
if validServer == server {
|
||||
isValidServer = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !isValidServer {
|
||||
go func (i int) {
|
||||
ch <- channelData{i, "request failed: invalid server\n"}
|
||||
} (i)
|
||||
} else {
|
||||
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)
|
||||
}
|
||||
}
|
||||
|
||||
for range servers {
|
||||
var output channelData = <-ch
|
||||
response_array[output.id] = output.data
|
||||
}
|
||||
|
||||
return response_array
|
||||
}
|
||||
25
frontend/main.go
Normal file
25
frontend/main.go
Normal file
@ -0,0 +1,25 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"strings"
|
||||
)
|
||||
|
||||
var settingServers []string
|
||||
var settingServersDomain string
|
||||
var settingServersPort int
|
||||
var settingWhoisServer 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")
|
||||
flag.Parse()
|
||||
|
||||
settingServers = strings.Split(*serversPtr, ",")
|
||||
settingServersDomain = *domainPtr
|
||||
settingServersPort = *portPtr
|
||||
settingWhoisServer = *whoisPtr
|
||||
webServerStart()
|
||||
}
|
||||
243
frontend/template.go
Normal file
243
frontend/template.go
Normal file
@ -0,0 +1,243 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net"
|
||||
"net/http"
|
||||
"strings"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
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(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, "/")
|
||||
|
||||
split = strings.Split(path, "/")
|
||||
var serverAllActive string
|
||||
if split[3] == strings.Join(settingServers[:], "+") {
|
||||
serverAllActive = " active"
|
||||
}
|
||||
|
||||
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>
|
||||
<span class="navbar-text">|</span>
|
||||
<li class="nav-item">
|
||||
<a class="nav-link` + serverAllActive + `" href="` + all_url + `"> All Servers </a>
|
||||
</li>
|
||||
`
|
||||
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 += `
|
||||
<li class="nav-item">
|
||||
<a class="nav-link` + serverActive + `" href="` + server_url + `">` + server + `</a>
|
||||
</li>
|
||||
`
|
||||
}
|
||||
|
||||
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] == "whois" {
|
||||
options += `<option value="whois" selected>whois ...</option>`
|
||||
} else {
|
||||
options += `<option value="whois">whois ...</option>`
|
||||
}
|
||||
|
||||
var target string
|
||||
if len(split) >= 5 {
|
||||
target = split[4]
|
||||
}
|
||||
|
||||
w.Write([]byte(`
|
||||
<!DOCTYPE html>
|
||||
<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="zh-CN" lang="zh-CN" class="no-js">
|
||||
<head>
|
||||
<meta http-equiv="Content-Type" content="text/html;charset=UTF-8" />
|
||||
<meta http-equiv="X-UA-Compatible" content="IE=edge"/>
|
||||
<meta name="viewport" content="width=device-width,initial-scale=1,shrink-to-fit=no"/>
|
||||
<meta name="renderer" content="webkit"/>
|
||||
<title>` + title + `</title>
|
||||
<link href="https://cdn.jsdelivr.net/npm/bootstrap@4.2.1/dist/css/bootstrap.min.css" rel="stylesheet">
|
||||
</head>
|
||||
<body>
|
||||
|
||||
<nav class="navbar navbar-expand-lg navbar-light bg-light">
|
||||
<a class="navbar-brand" href="/">Bird-lg Go</a>
|
||||
<button class="navbar-toggler" type="button" data-toggle="collapse" data-target="#navbarSupportedContent" aria-controls="navbarSupportedContent" aria-expanded="false" aria-label="Toggle navigation">
|
||||
<span class="navbar-toggler-icon"></span>
|
||||
</button>
|
||||
|
||||
<div class="collapse navbar-collapse" id="navbarSupportedContent">
|
||||
<ul class="navbar-nav mr-auto">` + serverNavigation + `</ul>
|
||||
<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="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>
|
||||
</div>
|
||||
</div>
|
||||
</form>
|
||||
</div>
|
||||
</nav>
|
||||
|
||||
<div class="container">
|
||||
`))
|
||||
}
|
||||
|
||||
func templateFooter(w http.ResponseWriter) {
|
||||
w.Write([]byte(`
|
||||
</div>
|
||||
<!--<script data-no-instant src="https://cdn.jsdelivr.net/npm/jquery@3.3.1/dist/jquery.min.js"></script>
|
||||
<script data-no-instant src="https://cdn.jsdelivr.net/npm/bootstrap@4.2.1/dist/js/bootstrap.bundle.min.js"></script>-->
|
||||
</body>
|
||||
</html>
|
||||
`))
|
||||
}
|
||||
|
||||
func isIP(s string) bool {
|
||||
return nil != net.ParseIP(s)
|
||||
}
|
||||
|
||||
func isNumber(s string) bool {
|
||||
_, err := strconv.Atoi(s)
|
||||
return nil == err
|
||||
}
|
||||
|
||||
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, " ") {
|
||||
if len(word) == 0 {
|
||||
tabPending = true
|
||||
} else {
|
||||
if isFirstWord {
|
||||
isFirstWord = false
|
||||
} else if tabPending {
|
||||
w.Write([]byte("\t"))
|
||||
tabPending = false
|
||||
} else {
|
||||
w.Write([]byte(" "))
|
||||
}
|
||||
|
||||
if isIP(word) {
|
||||
w.Write([]byte("<a href=\"/whois/" + word + "\">" + word + "</a>"))
|
||||
} else if len(strings.Split(word, "%")) == 2 && isIP(strings.Split(word, "%")[0]) {
|
||||
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]) {
|
||||
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:" {
|
||||
isASes = true
|
||||
w.Write([]byte(word))
|
||||
} else if isASes && isNumber(word) {
|
||||
w.Write([]byte("<a href=\"/whois/AS" + word + "\">" + word + "</a>"))
|
||||
} else {
|
||||
w.Write([]byte(word))
|
||||
}
|
||||
}
|
||||
}
|
||||
w.Write([]byte("\n"))
|
||||
}
|
||||
w.Write([]byte("</pre>"))
|
||||
}
|
||||
|
||||
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 {
|
||||
if tableCells < 5 {
|
||||
tableCells++
|
||||
} else {
|
||||
row[tableCells] += " "
|
||||
}
|
||||
tabPending = false
|
||||
} else {
|
||||
row[tableCells] += " "
|
||||
}
|
||||
row[tableCells] += word
|
||||
}
|
||||
}
|
||||
|
||||
if len(row[0]) == 0 {
|
||||
continue
|
||||
}
|
||||
if lineId == 0 {
|
||||
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 {
|
||||
if row[3] == "up" {
|
||||
w.Write([]byte("<tr>"))
|
||||
} else if lineId != 0 {
|
||||
w.Write([]byte("<tr class=\"table-danger\">"))
|
||||
}
|
||||
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>"))
|
||||
}
|
||||
for i := 1; i < 6; i++ {
|
||||
w.Write([]byte("<td>" + row[i] + "</td>"))
|
||||
}
|
||||
w.Write([]byte("</tr>"))
|
||||
}
|
||||
}
|
||||
w.Write([]byte("</tbody></table>"))
|
||||
}
|
||||
110
frontend/webserver.go
Normal file
110
frontend/webserver.go
Normal file
@ -0,0 +1,110 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"strings"
|
||||
"html"
|
||||
)
|
||||
|
||||
func webDispatcherIPv4Summary(w http.ResponseWriter, r *http.Request) {
|
||||
webHandler(w, r, false, r.URL.Path[len("/ipv4/summary/"):], "show protocols")
|
||||
}
|
||||
|
||||
func webDispatcherIPv6Summary(w http.ResponseWriter, r *http.Request) {
|
||||
webHandler(w, r, true, r.URL.Path[len("/ipv6/summary/"):], "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])
|
||||
}
|
||||
|
||||
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])
|
||||
}
|
||||
|
||||
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:], "/"))
|
||||
}
|
||||
|
||||
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:], "/"))
|
||||
}
|
||||
|
||||
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")
|
||||
}
|
||||
|
||||
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")
|
||||
}
|
||||
|
||||
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 webHandler(w http.ResponseWriter, r *http.Request, isIPv6 bool, serverQuery string, command string) {
|
||||
templateHeader(w, r, "Bird-lg Go - " + html.EscapeString(command))
|
||||
|
||||
var servers []string = strings.Split(serverQuery, "+")
|
||||
|
||||
var responses []string
|
||||
if isIPv6 {
|
||||
responses = batchRequest(servers, "bird6", command)
|
||||
} else {
|
||||
responses = batchRequest(servers, "bird", command)
|
||||
}
|
||||
for i, response := range responses {
|
||||
w.Write([]byte("<h2>" + html.EscapeString(servers[i]) + ": " + html.EscapeString(command) + "</h2>"))
|
||||
if command == "show protocols" && response[0:4] == "name" {
|
||||
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 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("/whois/", webDispatcherWhois)
|
||||
http.HandleFunc("/redir/", navbarFormRedirect)
|
||||
http.ListenAndServe(":5000", nil)
|
||||
}
|
||||
21
frontend/whois.go
Normal file
21
frontend/whois.go
Normal file
@ -0,0 +1,21 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net"
|
||||
"io/ioutil"
|
||||
)
|
||||
|
||||
func whois(s string) string {
|
||||
conn, err := net.Dial("tcp", settingWhoisServer + ":43")
|
||||
if err != nil {
|
||||
return err.Error()
|
||||
}
|
||||
defer conn.Close()
|
||||
|
||||
conn.Write([]byte(s + "\r\n"))
|
||||
result, err := ioutil.ReadAll(conn)
|
||||
if err != nil {
|
||||
return err.Error()
|
||||
}
|
||||
return string(result)
|
||||
}
|
||||
200
proxy/main.go
Normal file
200
proxy/main.go
Normal file
@ -0,0 +1,200 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"io"
|
||||
"net"
|
||||
"net/http"
|
||||
"sync"
|
||||
"runtime"
|
||||
"os/exec"
|
||||
"flag"
|
||||
)
|
||||
|
||||
// BIRDv4 connection & mutex lock
|
||||
var bird net.Conn
|
||||
var birdMutex = &sync.Mutex{}
|
||||
|
||||
// BIRDv6 connection & mutex lock
|
||||
var bird6 net.Conn
|
||||
var bird6Mutex = &sync.Mutex{}
|
||||
|
||||
// Check if a byte is character for number
|
||||
func isNumeric(b byte) bool {
|
||||
return b >= byte('0') && b <= byte('9')
|
||||
}
|
||||
|
||||
// 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++
|
||||
}
|
||||
|
||||
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
|
||||
}
|
||||
}
|
||||
|
||||
// Write a command to a bird socket
|
||||
func birdWriteln(bird io.Writer, s string) {
|
||||
bird.Write([]byte(s + "\n"))
|
||||
}
|
||||
|
||||
// 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"))
|
||||
}
|
||||
|
||||
// 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()
|
||||
|
||||
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()
|
||||
|
||||
println(query)
|
||||
birdWriteln(bird6, query)
|
||||
for birdReadln(bird6, httpW) {}
|
||||
}
|
||||
}
|
||||
|
||||
// Wrapper of traceroute, IPv4
|
||||
func tracerouteIPv4Wrapper(httpW http.ResponseWriter, httpR *http.Request) {
|
||||
tracerouteRealHandler(false, httpW, httpR)
|
||||
}
|
||||
|
||||
// Wrapper of traceroute, IPv6
|
||||
func tracerouteIPv6Wrapper(httpW http.ResponseWriter, httpR *http.Request) {
|
||||
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 {
|
||||
httpW.WriteHeader(http.StatusInternalServerError)
|
||||
httpW.Write([]byte("Traceroute Execution Error: "))
|
||||
httpW.Write([]byte(err.Error() + "\n"))
|
||||
return
|
||||
}
|
||||
httpW.Write(output)
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
var err error
|
||||
|
||||
birdPtr := flag.String("bird", "/var/run/bird/bird.ctl", "socket file for bird")
|
||||
bird6Ptr := flag.String("bird6", "/var/run/bird/bird6.ctl", "socket file for bird6")
|
||||
flag.Parse()
|
||||
|
||||
// Initialize BIRDv4 socket
|
||||
bird, err = net.Dial("unix", *birdPtr)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
defer bird.Close()
|
||||
|
||||
birdReadln(bird, nil)
|
||||
birdWriteln(bird, "restrict")
|
||||
birdReadln(bird, nil)
|
||||
|
||||
// Initialize BIRDv6 socket
|
||||
bird6, err = net.Dial("unix", *bird6Ptr)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
defer bird6.Close()
|
||||
|
||||
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(":8000", nil)
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user