golang 實現tcp轉送 Proxy

這是一個建立於 的文章，其中的資訊可能已經有所發展或是發生改變。

package mainimport ("flag""fmt""io""net""os""strings""sync")var lock sync.Mutexvar trueList []stringvar ip stringvar list stringfunc 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)<-ExitChandconn.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}