feat: port forwarding cli args

README.md

@@ -80,6 +80,10 @@
 
 + `debug-dump`: 是否开启调试，一般不需要加此参数
 
++ `tcp-port-forwarding`: TCP 端口转发，格式为 `本地地址-远程地址,本地地址-远程地址,...`，例如 `127.0.0.1:9898-10.10.98.98:80,0.0.0.0:9899-10.10.98.98:80`。多个转发用 `,` 分隔
+
++ `udp-port-forwarding`: UDP 端口转发，格式为 `本地地址-远程地址,本地地址-远程地址,...`，例如 `127.0.0.1:53-10.10.0.21:53`。多个转发用 `,` 分隔
+
 + `twf-id`: twfID 登录，调试用途，一般不需要加此参数
 
 + `config`: 指定配置文件，内容参考 `config.toml.example`。启用配置文件时其他参数无效

main.go

@@ -4,6 +4,7 @@ import (
  "flag"
  "fmt"
  "os"
+ "strings"
 
  "github.com/BurntSushi/toml"
  "github.com/mythologyli/zju-connect/core"
@@ -46,7 +47,10 @@ func getTomlVal[T int | uint64 | string | bool](valPointer *T, defaultVal T) T {
 
 func main() {
  // CLI args
- host, port, username, password, disableServerConfig, disableZjuConfig, disableZjuDns, disableMultiLine, twfId, configFile := "", 0, "", "", false, false, false, false, "", ""
+ host, port, username, password := "", 0, "", ""
+ disableServerConfig, disableZjuConfig, disableZjuDns, disableMultiLine := false, false, false, false
+ twfId, configFile, tcpPortForwarding, udpPortForwarding := "", "", "", ""
+
  flag.StringVar(&host, "server", "rvpn.zju.edu.cn", "EasyConnect server address")
  flag.IntVar(&port, "port", 443, "EasyConnect port address")
  flag.StringVar(&username, "username", "", "Your username")
@@ -62,6 +66,8 @@ func main() {
  flag.StringVar(&core.HttpBind, "http-bind", ":1081", "The address HTTP server listens on (e.g. 127.0.0.1:1081)")
  flag.Uint64Var(&core.DnsTTL, "dns-ttl", 3600, "DNS record time to live, unit is second")
  flag.BoolVar(&core.DebugDump, "debug-dump", false, "Enable traffic debug dump (only for debug usage)")
+ flag.StringVar(&tcpPortForwarding, "tcp-port-forwarding", "", "TCP port forwarding (e.g. 0.0.0.0:9898-10.10.98.98:80,127.0.0.1:9899-10.10.98.98:80)")
+ flag.StringVar(&udpPortForwarding, "udp-port-forwarding", "", "UDP port forwarding (e.g. 127.0.0.1:53-10.10.0.21:53)")
  flag.StringVar(&twfId, "twf-id", "", "Login using twfID captured (mostly for debug usage)")
  flag.StringVar(&configFile, "config", "", "Config file")
 
@@ -138,6 +144,40 @@ func main() {
  core.UseZjuDns = !disableZjuDns
  core.TestMultiLine = !disableMultiLine
 
+ if tcpPortForwarding != "" {
+ forwardingStringList := strings.Split(tcpPortForwarding, ",")
+ for _, forwardingString := range forwardingStringList {
+ addressStringList := strings.Split(forwardingString, "-")
+ if len(addressStringList) != 2 {
+ fmt.Println("ZJU Connect: wrong tcp port forwarding format")
+ return
+ }
+
+ core.ForwardingList = append(core.ForwardingList, core.Forwarding{
+ NetworkType: "tcp",
+ BindAddress: addressStringList[0],
+ RemoteAddress: addressStringList[1],
+ })
+ }
+ }
+
+ if udpPortForwarding != "" {
+ forwardingStringList := strings.Split(udpPortForwarding, ",")
+ for _, forwardingString := range forwardingStringList {
+ addressStringList := strings.Split(forwardingString, "-")
+ if len(addressStringList) != 2 {
+ fmt.Println("ZJU Connect: wrong udp port forwarding format")
+ return
+ }
+
+ core.ForwardingList = append(core.ForwardingList, core.Forwarding{
+ NetworkType: "udp",
+ BindAddress: addressStringList[0],
+ RemoteAddress: addressStringList[1],
+ })
+ }
+ }
+
  if host == "" || ((username == "" || password == "") && twfId == "") {
  fmt.Println("ZJU Connect")
  fmt.Println("Please see: https://github.com/mythologyli/zju-connect")