标签：127.0 string 0.1 list 代理 tcp var flag Go

nginx默认只能代理http，而且只能监听80端口，这个就不能忍了，自己撸一个

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("-l=0.0.0.0:9897 -d=127.0.0.1:1789")
        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)
        if err != nil {
            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)
        if err != nil {
            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
}

标签：127.0,string,0.1,list,代理,tcp,var,flag,Go
来源： https://www.cnblogs.com/mignet/p/go_transfer.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。