原创

用golang写一个proxy

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/u010278923/article/details/78889075

我复杂的网络环境中,proxy是个很有用的工具,我们可以通过proxy代理帮我们完成网络流量的转发。
这个proxy得先能接收请求,所以这里先启动一个tcp的监听,获取请求


func (s *Server) Start() {
    var err error
    s.listener, err = net.Listen("tcp", s.addr)
    if err != nil {
        glog.Fatal(err)
    }


    glog.Infof("proxy listen in %s, waiting for connection...\n", s.addr)

    for {
        conn, err := s.listener.Accept()
        if err != nil {
            glog.Error(err)
            continue
        }
        go newConn(s,conn).serve()
    }
}

当获取的请求连接conn,交个serve方法去处理。

func (c *conn) serve() {
    defer c.netConn.Close()
    rawHttpRequestHeader, remote, isHttps, err := c.getTunnelInfo()
    if err != nil {
        glog.Error(err)
        return
    }


    glog.Info("connecting to " + remote)
    remoteConn, err := net.Dial("tcp", remote)
    if err != nil {
        glog.Error(err)
        return
    }

    if isHttps {
        glog.V(10).Infof("get https req remote %s \n",remote)
        // if https, should sent 200 to client
        _, err = c.netConn.Write([]byte("HTTP/1.1 200 Connection established\r\n\r\n"))
        if err != nil {
            glog.Error(err)
            return
        }
    } else {
        // if not https, should sent the request header to remote
        _, err = rawHttpRequestHeader.WriteTo(remoteConn)
        if err != nil {
            glog.Error(err)
            return
        }
    }

    // build bidirectional-streams
    glog.Info("begin tunnel", c.netConn.RemoteAddr(), "<->", remote)
    c.tunnel(remoteConn)
    glog.Info("stop tunnel", c.netConn.RemoteAddr(), "<->", remote)
}

getTunnelInfo是获取连接信息的,譬如访问host、port、协议和请求头之类的东西。
然后就可以建立远程连接,通过net.Dial(“tcp”, remote)建立。
最后就是简单复制流量c.tunnel(remoteConn),具体代码如下

_, err := io.Copy(c.netConn, remoteConn)

这样就可以通过它转发流量
下面是例子
启动服务

 ./goproxy --addr :9999 --log_dir=/tmp/goproxy --v=10

到另外的机器上面执行
http:
curl --proxy 10.39.0.45:9999 http://10.39.0.45:9090

https:
curl -k --proxy 10.39.0.45:9999 https://10.0.0.43:443

完整代码如下
https://github.com/timchenxiaoyu/goproxy

这里写图片描述

文章最后发布于: 2017-12-25 09:03:40
展开阅读全文
0 个人打赏

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 1024 设计师: 上身试试

分享到微信朋友圈

×

扫一扫,手机浏览