ip_exit_enum/main.go at main · sinnet3000/ip_exit_enum · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package main

import (
	"context"
	"flag"
	"fmt"
	"os"
	"strings"
	"time"

	"ip_exit_enum/internal/discovery"
	"ip_exit_enum/internal/update"
	"ip_exit_enum/internal/version"
)

func main() {
	verbose := flag.Bool("v", false, "Verbose output")
	showVersion := flag.Bool("version", false, "Show version and exit")
	doUpdate := flag.Bool("update", false, "Update to the latest version")
	flag.Parse()

	if *showVersion {
		fmt.Printf("ip_exit_enum %s\n", version.Version)
		return
	}

	if *doUpdate {
		runUpdate()
		return
	}

	// Configuration - Mirroring Python service list
	httpServices := []discovery.ServiceConfig{
		// Primary IPv4
		{Name: "ipify", URL: "https://api.ipify.org", Protocol: "HTTP", Timeout: 5 * time.Second},
		{Name: "httpbin", URL: "https://httpbin.org/ip", Protocol: "HTTP", ExtractMethod: "json", ExtractField: "origin", Timeout: 5 * time.Second},
		{Name: "icanhazip", URL: "https://icanhazip.com", Protocol: "HTTP", Timeout: 5 * time.Second},
		{Name: "jsonip", URL: "https://jsonip.com", Protocol: "HTTP", ExtractMethod: "json", ExtractField: "ip", Timeout: 5 * time.Second},
		{Name: "ipecho", URL: "http://ipecho.net/plain", Protocol: "HTTP", Timeout: 5 * time.Second},
		{Name: "myip", URL: "https://api.myip.com", Protocol: "HTTP", ExtractMethod: "json", ExtractField: "ip", Timeout: 5 * time.Second},

		// IPv4 Specific
		{Name: "icanhazip-ipv4", URL: "https://ipv4.icanhazip.com", Protocol: "HTTP", Timeout: 5 * time.Second},
		{Name: "seeip-ipv4", URL: "https://ipv4.seeip.org", Protocol: "HTTP", Timeout: 5 * time.Second},

		// IPv6 Specific
		{Name: "ipify-v6", URL: "https://api6.ipify.org", Protocol: "HTTP", Timeout: 5 * time.Second},
		{Name: "icanhazip-ipv6", URL: "https://ipv6.icanhazip.com", Protocol: "HTTP", Timeout: 5 * time.Second},
		{Name: "seeip-ipv6", URL: "https://ipv6.seeip.org", Protocol: "HTTP", Timeout: 5 * time.Second},
	}

	udpServices := []discovery.ServiceConfig{
		// IPv4 STUN
		{Name: "stun-google-v4", URL: "stun.l.google.com:19302", Protocol: "UDP-STUN", Timeout: 5 * time.Second},
		{Name: "stun-cloudflare-v4", URL: "stun.cloudflare.com:3478", Protocol: "UDP-STUN", Timeout: 5 * time.Second},

		// IPv6 STUN
		// Note: pion/stun will resolve to AAAA records if available, but to force v6 we rely on system routing
		// or specific listeners in our generic engine.
		{Name: "stun-google-v6", URL: "stun.l.google.com:19302", Protocol: "UDP-STUN6", Timeout: 5 * time.Second},
		{Name: "stun-cloudflare-v6", URL: "stun.cloudflare.com:3478", Protocol: "UDP-STUN6", Timeout: 5 * time.Second},
	}

	engine := discovery.NewEngine(httpServices, udpServices)
	engine.Run(context.Background(), *verbose)
}

func runUpdate() {
	fmt.Println("Checking for updates...")

	info, err := update.CheckForUpdate()
	if err != nil {
		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
		os.Exit(1)
	}

	if info == nil {
		fmt.Printf("Already running latest version (%s)\n", version.Version)
		return
	}

	fmt.Printf("\n  Current version: %s\n", info.CurrentVersion)
	fmt.Printf("  Latest version:  %s\n", info.LatestVersion)
	fmt.Printf("\n  Download: %s\n", info.DownloadURL)
	fmt.Printf("  Size:     %s\n", update.FormatSize(info.Size))

	fmt.Print("\nProceed with update? [y/N] ")
	var response string
	fmt.Scanln(&response)
	if strings.ToLower(response) != "y" && strings.ToLower(response) != "yes" {
		fmt.Println("Update cancelled")
		return
	}

	fmt.Println()
	if err := update.PerformUpdate(info); err != nil {
		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
		os.Exit(1)
	}

	fmt.Printf("Updated to %s\n", info.LatestVersion)
}