Add support of port forwarding

Signed-off-by: Anton Litvinov <jnashicq@gmail.com>
mysteriumnetwork · Jan 22, 2024 · 20a6e96 · 20a6e96
1 parent 5cc40f0
commit 20a6e96
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 7 deletions.
diff --git a/controller/native/node.go b/controller/native/node.go
@@ -37,7 +37,7 @@ func NewController() *Controller {
 }
 
 func (c *Controller) GetCaps() int {
-	return 0
+	return 1
 }
 
 func (c *Controller) SetApp(a model_.AppState) {

diff --git a/controller/native/runner.go b/controller/native/runner.go
@@ -121,21 +121,31 @@ func KillPreviousLauncher() {
 func (r *NodeRunner) startNode() error {
 	log.Println("!startNode")
 	fullExePath := getNodeExePath()
+	c := r.mod.Config
+	log.Println("!startNode", c.EnablePortForwarding)
 
+	portsArg := ""
+	if c.EnablePortForwarding {
+		portsArg = fmt.Sprintf("--udp.ports=%d:%d", c.PortRangeBegin, c.PortRangeEnd)
+	}
+	userspaceArg := "--userspace"
 	versionArg := fmt.Sprintf("--launcher.ver=%s", r.mod.GetProductVersionString())
 	configDirArg := fmt.Sprintf("--config-dir=%s", r.configpath)
 	dataDirArg := fmt.Sprintf("--data-dir=%s", r.configpath)
 	logDirArg := fmt.Sprintf("--log-dir=%s", r.configpath)
 	nodeuiDirArg := fmt.Sprintf("--node-ui-dir=%s", path.Join(r.configpath, "nodeui"))
 
-	userspaceArg := "--userspace"
-
 	args2 := make([]string, 0)
 	if r.mod.NodeFlags != "" {
 		args2 = strings.Split(r.mod.NodeFlags, " ")
 	}
 
-	args := []string{userspaceArg, versionArg}
+	args := []string{}
+	if portsArg != "" {
+		args = append(args, portsArg)
+	}
+	args = append(args, userspaceArg, versionArg)
+
 	if len(args2) > 0 {
 		args = append(args, args2...)
 	} else {

diff --git a/gui-win32/networking_dlg.go b/gui-win32/networking_dlg.go
@@ -138,9 +138,11 @@ func (g *Gui) NetworkingDlg() {
 									walk.MsgBoxTopMost|walk.MsgBoxOK|walk.MsgBoxIconExclamation)
 								return
 							}
-							g.model.GetConfig().EnablePortForwarding = manualPortForwarding.Checked()
-							g.model.GetConfig().PortRangeBegin = portRangeBegin
-							g.model.GetConfig().PortRangeEnd = portRangeLen
+							cfg := g.model.GetConfig()
+							cfg.EnablePortForwarding = manualPortForwarding.Checked()
+							cfg.PortRangeBegin = portRangeBegin
+							cfg.PortRangeEnd = portRangeLen
+							cfg.Save()
 
 							dialog.Accept()
 							g.model.App.TriggerAction("restart")

diff --git a/model/state.go b/model/state.go
@@ -160,6 +160,7 @@ func (c *Config) Read() {
 
 	json.NewDecoder(file).Decode(&c) // for version check
 	c.getDefaultValues(false)
+	file.Seek(0, 0)
 	json.NewDecoder(file).Decode(&c)
 }