config.go 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package config
  2. import (
  3. "flag"
  4. "fmt"
  5. "net"
  6. )
  7. const ServerVersion = "2026.05.14.1353"
  8. type Config struct {
  9. HTTPHost string
  10. HTTPPort int
  11. UDPHost string
  12. UDPPort int
  13. MaintenanceIP string
  14. MaintenanceCIDR string
  15. AdminPassword string
  16. ServerVersion string
  17. DeviceIDFallback string
  18. }
  19. func Load(args []string) Config {
  20. cfg := Config{
  21. HTTPHost: "0.0.0.0",
  22. HTTPPort: 48888,
  23. UDPHost: "0.0.0.0",
  24. UDPPort: 50000,
  25. MaintenanceIP: "",
  26. MaintenanceCIDR: "",
  27. AdminPassword: "Dieteng2026",
  28. ServerVersion: ServerVersion,
  29. DeviceIDFallback: "networktool-device",
  30. }
  31. fs := flag.NewFlagSet("networktool-server", flag.ContinueOnError)
  32. fs.StringVar(&cfg.MaintenanceIP, "ip", cfg.MaintenanceIP, "maintenance IPv4 address, must be in 169.254.0.0/16 when set")
  33. fs.IntVar(&cfg.HTTPPort, "port", cfg.HTTPPort, "HTTP listen port")
  34. fs.StringVar(&cfg.AdminPassword, "password", cfg.AdminPassword, "admin password")
  35. _ = fs.Parse(args)
  36. if cfg.MaintenanceIP != "" {
  37. parsed := net.ParseIP(cfg.MaintenanceIP)
  38. ipv4 := parsed.To4()
  39. if parsed == nil || ipv4 == nil || ipv4[0] != 169 || ipv4[1] != 254 {
  40. panic(fmt.Sprintf("invalid maintenance ip: %s", cfg.MaintenanceIP))
  41. }
  42. cfg.MaintenanceCIDR = fmt.Sprintf("%s/16", cfg.MaintenanceIP)
  43. }
  44. return cfg
  45. }