Go로 TCP 프록시 만들기

구성요소

type Server struct {}

func (s *Server) Start(listenAddr *net.TCPAddr) error {
    lt, err := net.ListenTCP("tcp", listenAddr)
    if err != nil {
        return err
    }
    defer lt.Close()

    for {
        conn, err := lt.AcceptTCP()
        if err != nil {
            return err
        }
        go s.handleConn(conn)
    }
}

func (s *Server) handleConn(c *net.TCPConn) {
    ...
}

func (s *Server) handleConn(c *net.TCPConn) {
    defer c.Close()

    // 何らかの方法で本物のサーバーのアドレスを取得しておく
    serverAddr := ....

    // 本物のサーバーへ接続
    serverConn, err := net.DialTCP("tcp", nil, serverAddr)
    if err != nil {
        return err
    }
    defer serverConn.Close()

    // 双方向のリレーを作る
    p := &Proxy{}
    p.Start(c, serverConn)
}

// プロキシ本体
type Proxy struct{}

// プロキシの開始（双方向リレー開始）
func (p *Proxy) Start(clientConn, serverConn) error {
    defer clientConn.Close()
    defer serverConn.Close()

    var eg errGroup.Group

    eg.Go(func() error { return p.relay(&eg, clientConn, serverConn) })
    eg.Go(func() error { return p.relay(&eg, serverConn, clientConn) })

    // wait for stop
    return eg.Wait()
}

const (
    BUFFER_SIZE = 0xFFFF
)

// fromConn -> toConn へ通信内容をリレーする
func (p *Proxy) relay(eg *errGroup.Group, fromConn, toConn *net.TCPConn) error {
    buff := make([]byte, BUFFER_SIZE)
    for {
        n, err := fromConn.Read(buff)
        if err != nil {
            return err
        }
        b := buff[:n]

        // ここでログなどを取ることができる

        n, err = toConn.Write(b)
        if err != nil {
            return err
        }
    }
}

요약