byreqz/probehost2
1
0
Fork 0
mirror of https://github.com/byReqz/probehost2.git synced 2024-11-24 15:34:54 +00:00
Code Issues Projects Releases Wiki Activity

add mtr func

Browse Source
This commit is contained in:
Nils 2021-12-24 02:10:42 +01:00
parent 051671036c
commit ac8aac1f60
Signed by: byreqz
GPG Key ID: 396A62D7D436749E
1 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
main.go
View File

@ -28,6 +28,10 @@ func init() {
}
func runner(remoteip string, command string, args... string) string{
logfile.WithFields(log.Fields{
"remote_ip": remoteip,
"command": fmt.Sprint(command, args),
}).Info("request initiated:")
cmd, err := exec.Command(command, args...).Output()
if err != nil {
if ! strings.Contains(err.Error(), "1") { // dont exit if error code is 1
@ -35,7 +39,7 @@ func runner(remoteip string, command string, args... string) string{
"remote_ip": remoteip,
"command": fmt.Sprint(command, args),
"error": err.Error(),
}).Warn("the following request failed:")
}).Warn("request failed:")
logfile.WithFields(log.Fields{
"remote_ip": remoteip,
"command": fmt.Sprint(command, args),
@ -58,16 +62,28 @@ func showhelp(w http.ResponseWriter, req *http.Request) {
func ping(w http.ResponseWriter, req *http.Request) {
geturl := strings.Split(req.URL.String(), "/")
target := geturl[2]
pingres := runner(req.RemoteAddr, "ping", "-c5", target)
if pingres == "" {
res := runner(req.RemoteAddr, "ping", "-c10", target)
if res == "" {
fmt.Fprintln(w, http.StatusInternalServerError)
} else {
fmt.Fprintln(w, pingres)
fmt.Fprint(w, res)
}
}
func mtr(w http.ResponseWriter, req *http.Request) {
geturl := strings.Split(req.URL.String(), "/")
target := geturl[2]
res := runner(req.RemoteAddr, "mtr", "-c10", "-w", target)
if res == "" {
fmt.Fprintln(w, http.StatusInternalServerError)
} else {
fmt.Fprint(w, res)
}
}
func main() {
http.HandleFunc("/ping/", ping)
http.HandleFunc("/mtr/", mtr)
http.HandleFunc("/", showhelp)
fmt.Println("Serving on :8000")
http.ListenAndServe(":8000", nil)