文档章节

golang tcp 转发

明月惊鹊
 明月惊鹊
发布于 2016/10/01 17:15
字数 302
阅读 115
收藏 2
package main

import (
    "os"
    "io"
    "fmt"
    "net"
    "strings"
    "strconv"
    "syscall"
    "encoding/binary"
)

type CSPair struct {
    clientaddr net.Addr
    serveraddr net.Addr
    clientconn *net.TCPConn
    serverconn *net.TCPConn
}

const (
    SO_ORIGINAL_DST = 80
)

var (
    connection_count = 0
)

func main() {
    laddr := &net.TCPAddr{}
    laddr.Port = 8838
    ln, err := net.ListenTCP("tcp4", laddr)
    handle_error(err)
    fmt.Printf("listen on %d\n", laddr.Port)
    defer ln.Close()

    for {
        conn, err := ln.AcceptTCP()
        handle_error(err)
        pair := construct_connection(conn)
        handle_data(pair)
    }
}

func handle_data(pair *CSPair) {
    go handle_cs(pair)
    go handle_sc(pair)
}

func handle_cs(pair *CSPair) {
    defer pair.clientconn.Close()
    if strings.Index(pair.serveraddr.String(), ":843") != -1 {
        fmt.Println(":843 connection.")
        io.Copy(pair.serverconn, pair.clientconn)
        return
    }

    var remain_data []byte
    for {
        bs, err := readPacket(pair.clientconn)
        handle_error(err)
        remain_data = append(remain_data, bs...)
        packet_len := int(binary.LittleEndian.Uint32(remain_data))
        packet_len += 4 //fixed len.
        fmt.Printf("remain_data: 0x%x, packet_len: 0x%x\n", len(remain_data), packet_len)
        if packet_len > len(remain_data) {
            continue
        }
        packet_data := remain_data[:packet_len]
        remain_data = remain_data[packet_len:]
        //packet_data = append(packet_data, 0)
        fmt.Printf("receive 0x%x:%s\n", packet_len, string(packet_data))
        n, err := pair.serverconn.Write(packet_data)
        handle_error(err)
        fmt.Printf("handle_cs write 0x%x bytes\n", n)
    }
}

func handle_sc(pair *CSPair) {
    defer pair.serverconn.Close()
    io.Copy(pair.clientconn, pair.serverconn)
    fmt.Println("handle_sc close pair.serverconn")
    /*
    bs, err := readPacket(pair.serverconn)
    handle_error(err)
    fmt.Printf("handle_sc:%s\n", string(bs))
    pair.clientconn.Write(bs)
    */
}

func construct_connection(c *net.TCPConn) *CSPair {
    var pair = &CSPair{}
    pair.clientconn = c
    pair.clientaddr = (*c).RemoteAddr()
    f, err := c.File()
    handle_error(err)

    addr, err :=  syscall.GetsockoptIPv6Mreq(int(f.Fd()), syscall.IPPROTO_IP, SO_ORIGINAL_DST)
    handle_error(err)

    ipv4 := strconv.Itoa(int(addr.Multiaddr[4])) + "." +
            strconv.Itoa(int(addr.Multiaddr[5])) + "." +
            strconv.Itoa(int(addr.Multiaddr[6])) + "." +
            strconv.Itoa(int(addr.Multiaddr[7]))

    port := uint16(addr.Multiaddr[2]) << 8 + uint16(addr.Multiaddr[3])
    origin_ipv4 := ipv4
    origin_port := port

    sa, err := net.ResolveTCPAddr("tcp4", fmt.Sprintf("%s:%d", ipv4, port))
    handle_error(err)
    pair.serveraddr = sa
    pair.serverconn, err = net.DialTCP("tcp4", nil, sa)
    handle_error(err)

    connection_count++
    fmt.Printf("accept %d, %s and create a new connection to server %s(%s:%d)\n",
        connection_count, pair.clientaddr.String(), pair.serveraddr.String(), origin_ipv4, origin_port)
    return pair
}

func handle_error(err error) {
    if err != nil {
        fmt.Println(err)
        os.Exit(1)
    }
}

 

© 著作权归作者所有

共有 人打赏支持
下一篇: airodump
明月惊鹊
粉丝 25
博文 67
码字总数 14682
作品 0
广州
高级程序员
私信 提问
开源 | Service Mesh 数据平面 SOFAMosn 深层揭秘

小蚂蚁说: 本文是基于作者在 Service Mesh Meetup #2 北京的主题分享《蚂蚁金服 Service Mesh 数据平面 SOFAMosn 深层解密》部分内容所整理,完整内容见文末的直播回放。 本文作者:朵晓东,...

兔子酱
08/02
0
0
Proxy-Go v4.0 发布,内网穿透使用多路复用重构

Proxy-Go v4.0 发布了。Proxy是golang实现的高性能http,https,websocket,tcp,udp,socks5代理服务器,支持正向代理、内网穿透、TCP/UDP端口转发、SSH中转。 更新内容: 内网穿透三端重构了一个...

狂奔的蜗牛.
2017/12/04
1K
25
Proxy-Go v5.3 发布,优化了 SOCKS5 的 UDP 功能

Proxy-Go v5.3发布了。Proxy是golang实现的高性能http,https,websocket,tcp,udp,socks5代理服务器,支持正向代理、反向代理、透明代理、内网穿透、TCP/UDP端口映射、SSH中转、TLS加密传输、协...

狂奔的蜗牛.
08/06
0
0
Proxy-Go v5.0 发布:DNS 污染?不存在的!

Proxy-Go v5.0发布了。Proxy是golang实现的高性能http,https,websocket,tcp,udp,socks5代理服务器,支持正向代理、反向代理、透明代理、内网穿透、TCP/UDP端口映射、SSH中转、TLS加密传输、协...

狂奔的蜗牛.
06/19
0
0
高性能代理服务器 - Proxy-Go

Proxy是golang实现的高性能http,https,websocket,tcp,udp,socks5代理服务器,支持正向代理、反向代理、透明代理、内网穿透、TCP/UDP端口映射、SSH中转,TLS加密传输,协议转换。 Features 链式...

狂奔的蜗牛.
2017/09/13
0
8

没有更多内容

加载失败,请刷新页面

加载更多

kubernetes notes

Error from server (ServerTimeout): error when creating "mysql.yaml": No API token found for service account "default", retry after the token is automatically created and added t......

kut
36分钟前
2
0
代理模式---静态代理

package com.atguigu.java; //接口的应用:代理模式---静态代理。 public class ProxyTest { public static void main(String[] args) { NetWork work = new RealServer(); ProxyServer pro......

architect刘源源
52分钟前
2
1
工作中的一些Git使用

git stash list 查看所有stash git stash apply 0 应用序号为0那个stash并且不删除它 git stash pop 0 应用序号为0那个stash并且删除删除删除它 git stash save "env" stash当前的改动并且取...

-___-
今天
1
0
C# webservice 配置总结

为了方便下次查看,把总结配置文件放这里,看的方便。 <?xml version="1.0" encoding="utf-8"?><!-- 有关如何配置 ASP.NET 应用程序的详细消息,请访问 http://go.microsoft.com/f...

轻轻的往前走
今天
0
0
SpringCloud之Eureka

Eureka简介 什么是Eureka? Eureka是一种基于rest提供服务注册和发现的产品: Eureka-Server: 用于定位服务,以实现中间层服务器的负载平衡和故障转移。 Eureka-client:用于服务间的交互,内...

lc_fly1
今天
4
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部