2
golang 实现tcp转发代理
source link: https://studygolang.com/articles/7261?fr=sidebar
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
golang 实现tcp转发代理
fyxichen · 2016-06-08 20:00:00 · 11020 次点击 · 预计阅读时间 2 分钟 · 大约8小时之前 开始浏览这是一个创建于 2016-06-08 20:00:00 的文章,其中的信息可能已经有所发展或是发生改变。
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", ":9897", "-l=0.0.0.0:9897 指定服务监听的端口") 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 }
有疑问加站长微信联系(非本文作者)
Recommend
About Joyk
Aggregate valuable and interesting links.
Joyk means Joy of geeK