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

add options to the requests

Browse Source
This commit is contained in:
Nils 2021-12-30 02:58:33 +01:00
parent 55dfb5405a
commit 48a74f8dde
Signed by: byreqz
GPG Key ID: 396A62D7D436749E
1 changed files with 60 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

84
main.go
View File

@ -43,18 +43,16 @@ func runner(remoteip string, command string, args... string) string{
}).Info("request initiated:") }).Info("request initiated:")
cmd, err := exec.Command(command, args...).Output() cmd, err := exec.Command(command, args...).Output()
if err != nil { if err != nil {
if ! strings.Contains(err.Error(), "1") { // dont exit if error code is 1 logstdout.WithFields(log.Fields{
logstdout.WithFields(log.Fields{ "remote_ip": remoteip,
"remote_ip": remoteip, "command": fmt.Sprint(command, args),
"command": fmt.Sprint(command, args), "error": err.Error(),
"error": err.Error(), }).Warn("request failed:")
}).Warn("request failed:") logfile.WithFields(log.Fields{
logfile.WithFields(log.Fields{ "remote_ip": remoteip,
"remote_ip": remoteip, "command": fmt.Sprint(command, args),
"command": fmt.Sprint(command, args), "error": err.Error(),
"error": err.Error(), }).Warn("request failed:")
}).Warn("request failed:")
}
} else { } else {
logfile.WithFields(log.Fields{ logfile.WithFields(log.Fields{
"remote_ip": remoteip, "remote_ip": remoteip,
@ -76,15 +74,33 @@ func validatehosts(hosts []string) []string{
return valid return valid
} }
func parseopts(options []string, cmdopts map[string]string) []string{
var opts []string
for _, opt := range options {
opts = append(opts, cmdopts[opt])
}
return opts
}
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(), "/")
targets := strings.Split(geturl[2], ",") targets := strings.Split(geturl[2], ",")
hosts := validatehosts(targets) hosts := validatehosts(targets)
var res string cmdopts := map[string]string{"4": "-4", "6": "-6", "d": "-D", "n": "-n", "v": "-v", "c1": "-c1", "c5": "-c5", "c10": "-c10"}
for _, host := range hosts { var opts []string
res = fmt.Sprint(res, runner(req.RemoteAddr, "ping", "-c10", host), "\n") opts = append(opts, "-c10") // add default options
if len(geturl) > 3 {
options := strings.Split(geturl[3], ",")
opts = append(opts, parseopts(options, cmdopts)...)
} }
if res == "" { var res string
var args []string
for _, host := range hosts {
args = append(args, opts...)
args = append(args, host)
res = fmt.Sprint(res, runner(req.RemoteAddr, "ping", args...), "\n")
}
if strings.TrimSpace(res) == "" {
fmt.Fprintln(w, http.StatusInternalServerError) fmt.Fprintln(w, http.StatusInternalServerError)
} else { } else {
fmt.Fprint(w, strings.TrimSpace(res), "\n") fmt.Fprint(w, strings.TrimSpace(res), "\n")
@ -95,11 +111,21 @@ func mtr(w http.ResponseWriter, req *http.Request) {
geturl := strings.Split(req.URL.String(), "/") geturl := strings.Split(req.URL.String(), "/")
targets := strings.Split(geturl[2], ",") targets := strings.Split(geturl[2], ",")
hosts := validatehosts(targets) hosts := validatehosts(targets)
var res string cmdopts := map[string]string{"4": "-4", "6": "-6", "u": "-u", "t": "-T", "e": "-e", "x": "-x", "n": "-n", "b": "-b", "z": "-z", "c1": "-c1", "c5": "-c5", "c10": "-c10"}
for _, host := range hosts { var opts []string
res = fmt.Sprint(res, runner(req.RemoteAddr, "mtr", "-c10", "-w", host), "\n") opts = append(opts, "-r", "-w", "-c10") // add default options
if len(geturl) > 3 {
options := strings.Split(geturl[3], ",")
opts = append(opts, parseopts(options, cmdopts)...)
} }
if res == "" { var res string
var args []string
for _, host := range hosts {
args = append(args, opts...)
args = append(args, host)
res = fmt.Sprint(res, runner(req.RemoteAddr, "mtr", args...), "\n")
}
if strings.TrimSpace(res) == "" {
fmt.Fprintln(w, http.StatusInternalServerError) fmt.Fprintln(w, http.StatusInternalServerError)
} else { } else {
fmt.Fprint(w, strings.TrimSpace(res), "\n") fmt.Fprint(w, strings.TrimSpace(res), "\n")
@ -110,11 +136,21 @@ func traceroute(w http.ResponseWriter, req *http.Request) {
geturl := strings.Split(req.URL.String(), "/") geturl := strings.Split(req.URL.String(), "/")
targets := strings.Split(geturl[2], ",") targets := strings.Split(geturl[2], ",")
hosts := validatehosts(targets) hosts := validatehosts(targets)
var res string cmdopts := map[string]string{"4": "-4", "6": "-6", "dnf": "-F", "i": "-I", "t": "-T", "n": "-n", "u": "-U", "ul": "-UL", "d": "-D", "b": "--back"}
for _, host := range hosts { var opts []string
res = fmt.Sprint(res, runner(req.RemoteAddr, "traceroute", host), "\n") //opts = append(opts, "-c10") // no default options for traceroute
if len(geturl) > 3 {
options := strings.Split(geturl[3], ",")
opts = append(opts, parseopts(options, cmdopts)...)
} }
if res == "" { var res string
var args []string
for _, host := range hosts {
args = append(args, opts...)
args = append(args, host)
res = fmt.Sprint(res, runner(req.RemoteAddr, "traceroute", args...), "\n")
}
if strings.TrimSpace(res) == "" {
fmt.Fprintln(w, http.StatusInternalServerError) fmt.Fprintln(w, http.StatusInternalServerError)
} else { } else {
fmt.Fprint(w, strings.TrimSpace(res), "\n") fmt.Fprint(w, strings.TrimSpace(res), "\n")