76 lines
2.0 KiB
Go
76 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"net/http"
|
|
"os/exec"
|
|
"runtime"
|
|
)
|
|
|
|
// 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 && 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)
|
|
}
|
|
}
|