speed test

nm-proxy/manager/manager.go

@@ -25,6 +25,7 @@ TODO:-
 			-> start remote conn after endpoint is updated
 		-->
 */
+var sent bool
 
 type ProxyAction string
 
@@ -76,6 +77,7 @@ func StartProxyManager(manageChan chan *ManagerAction) {
 			log.Printf("-------> PROXY-MANAGER: %+v\n", mI)
 			switch mI.Action {
 			case AddInterface:
+
 				common.IsRelay = mI.Payload.IsRelay
 				if mI.Payload.IsRelay {
 					mI.RelayPeers()

nm-proxy/proxy/proxy.go

@@ -160,7 +160,7 @@ func StartSniffer(ctx context.Context, ifaceName, extClientAddr string, port int
 					if ip.DstIP.String() == extClientAddr || ip.SrcIP.String() == extClientAddr {
 						if ifacePeers, ok := common.PeerAddrMap[ifaceName]; ok {
 							if peerConf, ok := ifacePeers[ip.DstIP.String()]; ok {
-								log.Println("-----> Fowarding PKT From ExtClient: ", extClientAddr, " to: ", peerConf.Config.RemoteProxyIP)
+								log.Println("-----> Fowarding PKT From ExtClient: ", extClientAddr, " to: ", peerConf)
 								//server.NmProxyServer.Server.WriteTo(packet.Data(),  )
 							}
 

nm-proxy/proxy/wireguard.go

@@ -25,6 +25,7 @@ func NewProxy(config Config) *Proxy {
 // proxyToRemote proxies everything from Wireguard to the RemoteKey peer
 func (p *Proxy) ProxyToRemote() {
 	buf := make([]byte, 1500)
+	peers := common.WgIFaceMap[p.Config.WgInterface.Name]
 	go func() {
 		<-p.Ctx.Done()
 		log.Println("Closing connection for: ", p.LocalConn.LocalAddr().String())
@@ -58,7 +59,8 @@ func (p *Proxy) ProxyToRemote() {
 				log.Println("ERRR READ: ", err)
 				continue
 			}
-			peers := common.WgIFaceMap[p.Config.WgInterface.Name]
+			go func() {
+
 				if peerI, ok := peers[p.Config.RemoteKey]; ok {
 					var srcPeerKeyHash, dstPeerKeyHash string
 					buf, n, srcPeerKeyHash, dstPeerKeyHash = packet.ProcessPacketBeforeSending(buf, n, peerI.Config.LocalKey, peerI.Config.Key)
@@ -70,7 +72,7 @@ func (p *Proxy) ProxyToRemote() {
 				} else {
 					log.Printf("Peer: %s not found in config\n", p.Config.RemoteKey)
 					p.Cancel()
-				continue
+					return
 				}
 				//test(n, buf)
 
@@ -78,6 +80,8 @@ func (p *Proxy) ProxyToRemote() {
 				if err != nil {
 					log.Println("Failed to send to remote: ", err)
 				}
+			}()
+
 		}
 	}
 }

nm-proxy/server/server.go

@@ -59,6 +59,8 @@ func (p *ProxyServer) Listen(ctx context.Context) {
 				log.Println("RECV ERROR: ", err)
 				continue
 			}
+			go func() {
+
 				var srcPeerKeyHash, dstPeerKeyHash string
 				n, srcPeerKeyHash, dstPeerKeyHash = packet.ExtractInfo(buffer, n)
 				//log.Printf("--------> RECV PKT [DSTPORT: %d], [SRCKEYHASH: %s], SourceIP: [%s] \n", localWgPort, srcPeerKeyHash, source.IP.String())
@@ -90,7 +92,7 @@ func (p *ProxyServer) Listen(ctx context.Context) {
 								if err != nil {
 									log.Println("Failed to send to remote: ", err)
 								}
-							continue
+								return
 							}
 						} else {
 							if remoteMap, ok := common.RelayPeerMap[dstPeerKeyHash]; ok {
@@ -101,7 +103,7 @@ func (p *ProxyServer) Listen(ctx context.Context) {
 									if err != nil {
 										log.Println("Failed to send to remote: ", err)
 									}
-								continue
+									return
 								}
 							}
 
@@ -120,12 +122,12 @@ func (p *ProxyServer) Listen(ctx context.Context) {
 							_, err = peerI.Proxy.LocalConn.Write(buffer[:n])
 							if err != nil {
 								log.Println("Failed to proxy to Wg local interface: ", err)
-							continue
+								return
 							}
 
 						}
 					}
-				continue
+					return
 
 				}
 				// // forward to all interfaces
@@ -141,6 +143,7 @@ func (p *ProxyServer) Listen(ctx context.Context) {
 				// 	}
 
 				// }
+			}()
 		}
 
 	}