Go tcp 端口转发

package main
import(
	"fmt"
	"net"
	"io"
	"sync"
	"time"
	//"context"
)

func main(){
	listener, err := net.Listen("tcp","0.0.0.0:5000")
	if err != nil{
		fmt.Println(err.Error())
		return
	}
	defer listener.Close()

	for{
		conn,err := listener.Accept()
		if err != nil{
			fmt.Println(err.Error())
			continue
		}
		conn.SetDeadline(time.Now().Add(time.Duration(10)*time.Second))
		//conn.SetReadDeadline(time.Now().Add(time.Duration(10)*time.Second))
		//conn.SetWriteDeadline(time.Now().Add(time.Duration(10)*time.Second))
		go doServer(conn)
	}
}

func doServer(conn net.Conn){
	defer conn.Close()

	//conn2,err := net.DialTimeout("tcp","0.0.0.0:6000",time.Duration(10)*time.Second)
	conn2,err := net.Dial("tcp","0.0.0.0:6000")
	if err != nil{
		fmt.Println(err)
		return
	}

	defer conn2.Close()

	conn2.SetDeadline(time.Now().Add(time.Duration(10)*time.Second))
	//conn2.SetReadDeadline(time.Now().Add(time.Duration(10)*time.Second))
	//conn2.SetWriteDeadline(time.Now().Add(time.Duration(10)*time.Second))
	wg := &sync.WaitGroup{}
	wg.Add(2)

	
	go func(){
		_ ,err := io.Copy(conn2,conn)
		if err != nil{
			fmt.Println(err)
		}
		wg.Done()
	}()

	go func(){
		_ ,err := io.Copy(conn,conn2)
		if err != nil{
			fmt.Println(err)
		}
		wg.Done()
	}()
	wg.Wait()
}

 

 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值