gravitl/netmaker
1
1
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-10-03 02:14:24 +08:00
Code Issues Projects Releases Packages Wiki Activity

speed test

Browse source
This commit is contained in:
Abhishek Kondur 2022-11-16 10:02:54 +05:30
parent 8b81a4b1ec
commit 1434147279
2 changed files with 85 additions and 85 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
nm-proxy/proxy/wireguard.go
View file

@ -31,6 +31,7 @@ func (p *Proxy) ProxyToRemote() {
log.Println("Closing connection for: ", p.LocalConn.LocalAddr().String())
p.LocalConn.Close()
}()
buf := make([]byte, 65535)
for {
select {
case <-p.Ctx.Done():
@ -53,22 +54,22 @@ func (p *Proxy) ProxyToRemote() {
return
default:
buf := make([]byte, 1500)
n, err := p.LocalConn.Read(buf)
if err != nil {
log.Println("ERRR READ: ", err)
continue
}
go func(buf []byte, n int) {
// go func(buf []byte, n int) {
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)
//var srcPeerKeyHash, dstPeerKeyHash string
buf, n, _, _ = packet.ProcessPacketBeforeSending(buf, n, peerI.Config.LocalKey, peerI.Config.Key)
if err != nil {
log.Println("failed to process pkt before sending: ", err)
}
log.Printf("PROXING TO REMOTE!!!---> %s >>>>> %s >>>>> %s [[ SrcPeerHash: %s, DstPeerHash: %s ]]\n",
p.LocalConn.LocalAddr(), server.NmProxyServer.Server.LocalAddr().String(), p.RemoteConn.String(), srcPeerKeyHash, dstPeerKeyHash)
// log.Printf("PROXING TO REMOTE!!!---> %s >>>>> %s >>>>> %s [[ SrcPeerHash: %s, DstPeerHash: %s ]]\n",
// p.LocalConn.LocalAddr(), server.NmProxyServer.Server.LocalAddr().String(), p.RemoteConn.String(), srcPeerKeyHash, dstPeerKeyHash)
} else {
log.Printf("Peer: %s not found in config\n", p.Config.RemoteKey)
p.Cancel()
@ -80,7 +81,7 @@ func (p *Proxy) ProxyToRemote() {
if err != nil {
log.Println("Failed to send to remote: ", err)
}
}(buf, n)
// }(buf, n)
}
}

15
nm-proxy/server/server.go
View file

@ -2,7 +2,6 @@ package server
import (
"context"
"fmt"
"log"
"net"
"time"
@ -36,7 +35,7 @@ type ProxyServer struct {
func (p *ProxyServer) Listen(ctx context.Context) {
// Buffer with indicated body size
buffer := make([]byte, 65535+32)
for {
select {
@ -54,13 +53,13 @@ func (p *ProxyServer) Listen(ctx context.Context) {
return
default:
// Read Packet
buffer := make([]byte, 1532)
n, source, err := p.Server.ReadFromUDP(buffer)
if err != nil { // in future log errors?
log.Println("RECV ERROR: ", err)
continue
}
go func(buffer []byte, source *net.UDPAddr, n int) {
// go func(buffer []byte, source *net.UDPAddr, n int) {
var srcPeerKeyHash, dstPeerKeyHash string
n, srcPeerKeyHash, dstPeerKeyHash = packet.ExtractInfo(buffer, n)
@ -117,9 +116,9 @@ func (p *ProxyServer) Listen(ctx context.Context) {
if peerInfo, ok := common.PeerKeyHashMap[srcPeerKeyHash]; ok {
if peers, ok := common.WgIFaceMap[peerInfo.Interface]; ok {
if peerI, ok := peers[peerInfo.PeerKey]; ok {
log.Printf("PROXING TO LOCAL!!!---> %s <<<< %s <<<<<<<< %s [[ RECV PKT [SRCKEYHASH: %s], [DSTKEYHASH: %s], SourceIP: [%s] ]]\n",
peerI.Proxy.LocalConn.RemoteAddr(), peerI.Proxy.LocalConn.LocalAddr(),
fmt.Sprintf("%s:%d", source.IP.String(), source.Port), srcPeerKeyHash, dstPeerKeyHash, source.IP.String())
// log.Printf("PROXING TO LOCAL!!!---> %s <<<< %s <<<<<<<< %s [[ RECV PKT [SRCKEYHASH: %s], [DSTKEYHASH: %s], SourceIP: [%s] ]]\n",
// peerI.Proxy.LocalConn.RemoteAddr(), peerI.Proxy.LocalConn.LocalAddr(),
// fmt.Sprintf("%s:%d", source.IP.String(), source.Port), srcPeerKeyHash, dstPeerKeyHash, source.IP.String())
_, err = peerI.Proxy.LocalConn.Write(buffer[:n])
if err != nil {
log.Println("Failed to proxy to Wg local interface: ", err)
@ -144,7 +143,7 @@ func (p *ProxyServer) Listen(ctx context.Context) {
// }
// }
}(buffer, source, n)
// }(buffer, source, n)
}
}