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

add validate function and multiple input

Browse Source
This commit is contained in:
Nils 2021-12-24 14:41:14 +01:00
parent b3f8063735
commit 8faf66401d
Signed by: byreqz
GPG Key ID: 396A62D7D436749E
1 changed files with 27 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
main.go
View File

@ -5,6 +5,7 @@ import (
"os/exec" "os/exec"
"strings" "strings"
"net/http" "net/http"
"net"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
) )
@ -59,25 +60,45 @@ func showhelp(w http.ResponseWriter, req *http.Request) {
fmt.Fprintln(w, "placeholder") fmt.Fprintln(w, "placeholder")
} }
func validatehosts(hosts []string) []string{
var valid []string
for _, host := range hosts {
if net.ParseIP(host) != nil {
valid = append(valid, host)
} else if _, err := net.LookupIP(host); err == nil {
valid = append(valid, host)
}
}
return valid
}
func ping(w http.ResponseWriter, req *http.Request) { func ping(w http.ResponseWriter, req *http.Request) {
geturl := strings.Split(req.URL.String(), "/") geturl := strings.Split(req.URL.String(), "/")
target := geturl[2] targets := strings.Split(geturl[2], ",")
res := runner(req.RemoteAddr, "ping", "-c10", target) hosts := validatehosts(targets)
var res string
for _, host := range hosts {
res = fmt.Sprint(res, runner(req.RemoteAddr, "ping", "-c10", host), "\n")
}
if res == "" { if res == "" {
fmt.Fprintln(w, http.StatusInternalServerError) fmt.Fprintln(w, http.StatusInternalServerError)
} else { } else {
fmt.Fprint(w, res) fmt.Fprint(w, strings.TrimSpace(res), "\n")
} }
} }
func mtr(w http.ResponseWriter, req *http.Request) { func mtr(w http.ResponseWriter, req *http.Request) {
geturl := strings.Split(req.URL.String(), "/") geturl := strings.Split(req.URL.String(), "/")
target := geturl[2] targets := strings.Split(geturl[2], ",")
res := runner(req.RemoteAddr, "mtr", "-c10", "-w", target) hosts := validatehosts(targets)
var res string
for _, host := range hosts {
res = fmt.Sprint(res, runner(req.RemoteAddr, "mtr", "-c10", "-w", host), "\n")
}
if res == "" { if res == "" {
fmt.Fprintln(w, http.StatusInternalServerError) fmt.Fprintln(w, http.StatusInternalServerError)
} else { } else {
fmt.Fprint(w, res) fmt.Fprint(w, strings.TrimSpace(res), "\n")
} }
} }