脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Golang - golang 实现tcp转发代理的方法

golang 实现tcp转发代理的方法

2020-05-27 10:25JieLinDee Golang

今天小编就为大家分享一篇golang 实现tcp转发代理的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

我就废话不多说了,直接上代码吧:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
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
}

以上这篇golang 实现tcp转发代理的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。

原文链接:https://blog.csdn.net/fyxichen/article/details/51505542

延伸 · 阅读

精彩推荐
  • Golang深入浅析Go中三个点(...)用法

    深入浅析Go中三个点(...)用法

    这篇文章主要介绍了深入浅析Go中三个点(...)用法,需要的朋友可以参考下...

    踏雪无痕SS6472021-11-17
  • GolangGo语言range关键字循环时的坑

    Go语言range关键字循环时的坑

    今天小编就为大家分享一篇关于Go语言range关键字循环时的坑,小编觉得内容挺不错的,现在分享给大家,具有很好的参考价值,需要的朋友一起跟随小编来...

    benben_20154202020-05-23
  • GolangGo语言基础单元测试与性能测试示例详解

    Go语言基础单元测试与性能测试示例详解

    这篇文章主要为大家介绍了Go语言基础单元测试与性能测试示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助祝大家多多进步...

    枫少文7812021-12-05
  • GolangGo语言实现自动填写古诗词实例代码

    Go语言实现自动填写古诗词实例代码

    这篇文章主要给大家介绍了关于Go语言实现自动填写古诗词的相关资料,这是最近在项目中遇到的一个需求,文中通过示例代码介绍的非常详细,需要的朋...

    FengY5862020-05-14
  • GolangGolang 语言极简类型转换库cast的使用详解

    Golang 语言极简类型转换库cast的使用详解

    本文我们通过 cast.ToString() 函数的使用,简单介绍了cast 的使用方法,除此之外,它还支持很多其他类型,在这没有多多介绍,对Golang 类型转换库 cast相关知...

    Golang语言开发栈6112021-12-02
  • Golanggo语言获取系统盘符的方法

    go语言获取系统盘符的方法

    这篇文章主要介绍了go语言获取系统盘符的方法,涉及Go语言调用winapi获取系统硬件信息的技巧,具有一定参考借鉴价值,需要的朋友可以参考下 ...

    无尽海3862020-04-24
  • GolangGolang实现四种负载均衡的算法(随机,轮询等)

    Golang实现四种负载均衡的算法(随机,轮询等)

    本文介绍了示例介绍了Golang 负载均衡的四种实现,主要包括了随机,轮询,加权轮询负载,一致性hash,感兴趣的小伙伴们可以参考一下...

    Gundy_8442021-08-09
  • GolangGO语言字符串处理Strings包的函数使用示例讲解

    GO语言字符串处理Strings包的函数使用示例讲解

    这篇文章主要为大家介绍了GO语言字符串处理Strings包的函数使用示例讲解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步早日升职加...

    Jeff的技术栈6882022-04-14