zishuo/go-stun
1
0
Fork 0
mirror of https://github.com/ccding/go-stun.git synced 2025-10-05 08:46:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

clean up main (#44)

Browse Source
This commit is contained in:
Cong Ding
2023-03-29 21:43:21 -05:00
committed by GitHub
parent 44e89cab78
commit b32a538098
1 changed files with 27 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
main.go
View File

@@ -17,36 +17,44 @@ package main
import ( import (
"flag" "flag"
"fmt" "fmt"
"os"
"github.com/ccding/go-stun/stun" "github.com/ccding/go-stun/stun"
) )
func main() { func main() {
var serverAddr = flag.String("s", stun.DefaultServerAddr, "STUN server address") var serverAddr = flag.String("s", stun.DefaultServerAddr, "STUN server address")
var b = flag.Bool("b", false, "NAT behavior test mode") var behaviorTestMode = flag.Bool("b", false, "Enable NAT behavior test mode")
var v = flag.Bool("v", false, "verbose mode") var verboseLevel = flag.Int("v", 0, "Verbose level (0: none, 1: verbose, 2: double verbose, 3: triple verbose)")
var vv = flag.Bool("vv", false, "double verbose mode (includes -v)")
var vvv = flag.Bool("vvv", false, "triple verbose mode (includes -v and -vv)")
flag.Parse() flag.Parse()
// Creates a STUN client. NewClientWithConnection can also be used if you want to handle the UDP listener by yourself. // Validate verbose level
client := stun.NewClient() if *verboseLevel < 0 || *verboseLevel > 3 {
// The default addr (stun.DefaultServerAddr) will be used unless we call SetServerAddr. fmt.Fprintln(os.Stderr, "Error: Invalid verbose level. Use -v with values 0, 1, 2, or 3.")
client.SetServerAddr(*serverAddr) os.Exit(1)
// Non verbose mode will be used by default unless we call SetVerbose(true) or SetVVerbose(true). }
client.SetVerbose(*v || *vv || *vvv)
client.SetVVerbose(*vv || *vvv)
if *b { // Create a STUN client
behaviorTest(client) client := stun.NewClient()
client.SetServerAddr(*serverAddr)
client.SetVerbose(*verboseLevel >= 1)
client.SetVVerbose(*verboseLevel >= 2)
// Run behavior test if specified
if *behaviorTestMode {
err := runBehaviorTest(client)
if err != nil {
fmt.Fprintln(os.Stderr, "Error:", err)
os.Exit(1)
}
return return
} }
// Discover the NAT and return the result. // Discover the NAT
nat, host, err := client.Discover() nat, host, err := client.Discover()
if err != nil { if err != nil {
fmt.Println(err) fmt.Fprintln(os.Stderr, "Error:", err)
return os.Exit(1)
} }
fmt.Println("NAT Type:", nat) fmt.Println("NAT Type:", nat)
@@ -57,10 +65,10 @@ func main() {
} }
} }
func behaviorTest(c *stun.Client) { func runBehaviorTest(c *stun.Client) error {
natBehavior, err := c.BehaviorTest() natBehavior, err := c.BehaviorTest()
if err != nil { if err != nil {
fmt.Println(err) return err
} }
if natBehavior != nil { if natBehavior != nil {
@@ -68,4 +76,5 @@ func behaviorTest(c *stun.Client) {
fmt.Println("Filtering Behavior:", natBehavior.FilteringType) fmt.Println("Filtering Behavior:", natBehavior.FilteringType)
fmt.Println(" Normal NAT Type:", natBehavior.NormalType()) fmt.Println(" Normal NAT Type:", natBehavior.NormalType())
} }
return nil
} }