pnx/pinger
1
0
Fork 0
mirror of https://github.com/pnx/pinger.git synced 2026-06-16 03:24:54 +02:00
Code Releases Activity

main.go: Adding cli options.

Browse source
This commit is contained in:
Henrik Hautakoski 2023-03-12 16:12:36 +01:00
parent 26babe4ae2
commit 8cec3f855d
3 changed files with 33 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

1
go.mod
View file

@ -6,6 +6,7 @@ require github.com/prometheus-community/pro-bing v0.1.0
require (
github.com/google/uuid v1.3.0 // indirect
github.com/pborman/getopt/v2 v2.1.0 // indirect
golang.org/x/net v0.5.0 // indirect
golang.org/x/sync v0.0.0-20220601150217-0de741cfad7f // indirect
golang.org/x/sys v0.4.0 // indirect

2
go.sum
View file

@ -1,5 +1,7 @@
github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/pborman/getopt/v2 v2.1.0 h1:eNfR+r+dWLdWmV8g5OlpyrTYHkhVNxHBdN2cCrJmOEA=
github.com/pborman/getopt/v2 v2.1.0/go.mod h1:4NtW75ny4eBw9fO1bhtNdYTlZKYX5/tBLtsOpwKIKd0=
github.com/prometheus-community/pro-bing v0.1.0 h1:zjzLGhfNPP0bP1OlzGB+SJcguOViw7df12LPg2vUJh8=
github.com/prometheus-community/pro-bing v0.1.0/go.mod h1:BpWlHurD9flHtzq8wrh8QGWYz9ka9z9ZJAyOel8ej58=
golang.org/x/net v0.5.0 h1:GyT4nK/YDHSqa1c4753ouYCDajOYKTja9Xb/OHtgvSw=

36
main.go
View file

@ -4,10 +4,10 @@ import (
"fmt"
"os"
"os/signal"
"path/filepath"
"syscall"
"time"
"github.com/pborman/getopt/v2"
"github.com/prometheus-community/pro-bing"
)
@ -44,20 +44,44 @@ func eventLoop(pinger *probing.Pinger) {
}
func main() {
if len(os.Args) < 2 {
fmt.Printf("Usage: %s <ip>\n", filepath.Base(os.Args[0]))
// Cli options.
help := getopt.BoolLong("help", 'h', "Show this help text")
version := getopt.BoolLong("version", 'v', "Show program version")
source := getopt.StringLong("source", 's', "", "Source IP address to send packages from.")
count := getopt.IntLong("count", 'c', 0, "Stop after this many packages has been sent (and received). If this option is not specified, pinger will operate until interrupted.")
timeout := getopt.DurationLong("timeout", 't', 0, "Exit the program after this time is reached, regardless of how many packets have been received.")
interval := getopt.DurationLong("interval", 'i', time.Second*2, "Wait time between each packet send")
proto_udp := getopt.BoolLong("udp", 0, "Send UDP ping instead of a raw IMCP ping, IMCP required super-user privileges")
record_rtt := getopt.BoolLong("rtt", 0, "Keep a record of rtts of all received packets.")
getopt.SetParameters("<ip>")
getopt.Parse()
if *version {
fmt.Println("Version 0.0.1")
os.Exit(0)
}
if *help || len(getopt.Args()) < 1 {
getopt.Usage()
os.Exit(1)
}
// Setup pinger.
pinger, err := probing.NewPinger(os.Args[1])
pinger, err := probing.NewPinger(getopt.Arg(0))
if err != nil {
fmt.Println("Error:", err)
return
}
pinger.SetPrivileged(true)
pinger.Interval = time.Second * 2
pinger.SetPrivileged(!*proto_udp)
pinger.Interval = *interval
pinger.Source = *source
pinger.Count = *count
pinger.RecordRtts = *record_rtt
if timeout != nil && *timeout > 0 {
pinger.Timeout = *timeout
}
// Enter event loop in another goroutine.
go eventLoop(pinger)