tcp udp proxy

服务目的

首先如下图所示:
tcp udp proxy_第1张图片
作为一个内外网的通信,必须使用tcp 和 udp 的proxy 把内网和外网打通,比如中间是一个有两个网卡的路由器,打通以后,由proxy 发送数据到服务端,服务端按照上图处于外网。

服务端测试

使用python 来写服务端测试永远是一个好主意

使用python写一个服务端,也可以使用py写一个tcp服务端,同样的道理

import socket
host = "192.168.0.109"
port = 7000
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock.bind((host, port))
print("udp server start on port:", port)
while True:
    data, addr = sock.recvfrom(1024)
    print("Receive data = %s" % data)
    if data == b"exit":
        sock.sendto(b"Goodbye!\n", addr)
        continue
    sock.sendto(b"Hello %s!\n" % data, addr)

以上简单实现了一个udp服务端,下面使用go 写一个udp 客户端

udp 客户端


使用go写一个udp客户端,也 可以使用go写一个tcp 客户端。同样的道理
```go
package main

import (
	"fmt"
	"net"
)

func main() {
	conn, err := net.Dial("udp", "127.0.0.1:7000")
	if err != nil {
		fmt.Println("net.dial err:", err)
		return
	}
	defer conn.Close()
	conn.Write([]byte("are you ready?"))
	buf := make([]byte, 4096)
	n, err := conn.Read(buf)
	if err != nil {
		fmt.Println("conn.Read err:", err)
		return
	}
	fmt.Println("server back:", string(buf[:n])) 
}

转发服务

下面使用go写一个tcp 转发服务


package main

import (
	"flag"
	"fmt"
	"io"
	"net"
	"os"
	"strings"
	"sync"
)

var lock sync.Mutex
var trueList []string
var ip string
var list string

func main() {
	flag.StringVar(&ip, "l", ":6000", "-l=0.0.0.0:6000 指定服务监听的端口")
	flag.StringVar(&list, "d", "127.0.0.1:1789,127.0.0.1:1788", "-d=127.0.0.1:1789,127.0.0.1:1788 指定后端的IP和端口,多个用','隔开")
	flag.Parse()
	trueList = strings.Split(list, ",")
	if len(trueList) <= 0 {
		fmt.Println("后端IP和端口不能空,或者无效")
		os.Exit(1)
	}
	server()
}

func server() {
	lis, err := net.Listen("tcp", ip)
	if err != nil {
		fmt.Println(err)
		return
	}
	defer lis.Close()
	for {
		conn, err := lis.Accept()
		if err != nil {
			fmt.Println("建立连接错误:%v\n", err)
			continue
		}
		fmt.Println(conn.RemoteAddr(), conn.LocalAddr())
		go handle(conn)
	}
}

func handle(sconn net.Conn) {
	defer sconn.Close()
	ip, ok := getIP()
	if !ok {
		return
	}
	dconn, err := net.Dial("tcp", ip)
	if err != nil {
		fmt.Printf("连接%v失败:%v\n", ip, err)
		return
	}
	ExitChan := make(chan bool, 1)
	go func(sconn net.Conn, dconn net.Conn, Exit chan bool) {
		_, err := io.Copy(dconn, sconn)
		fmt.Printf("往%v发送数据失败:%v\n", ip, err)
		ExitChan <- true
	}(sconn, dconn, ExitChan)
	go func(sconn net.Conn, dconn net.Conn, Exit chan bool) {
		_, err := io.Copy(sconn, dconn)
		fmt.Printf("从%v接收数据失败:%v\n", ip, err)
		ExitChan <- true
	}(sconn, dconn, ExitChan)
	<-ExitChan
	dconn.Close()
}

func getIP() (string, bool) {
	lock.Lock()
	defer lock.Unlock()

	if len(trueList) < 1 {
		return "", false
	}
	ip := trueList[0]
	trueList = append(trueList[1:], ip)
	return ip, true
}

go 语言本身实现了协程,以上程序并没有实现产品化,只是一个代码示例,读者需要自己补充写好。

你可能感兴趣的:(go,udp,tcp/ip,网络,proxy,端口转发)