实现http代理服务器转发请求

http代理一般用于突破网络限制和加速网络用,其原理就是代转发请求和回应。
http流量,未加密,可以直接从请求头中获得要访问目标的地址,代理服创建一条访问目标的tcp连接,将http流量相互转发即可
https流量,由于加密,无法读取内容,所以多出一个CONNECT请求(我们最经常接触的是GET,POST),客户端会先向代理服发送CONNECT请求,CONNECT请求会带上要访问目标的地址,代理服创建一条访问目标的tcp连接,相互转发流量即可

golang实现具体代码如下

package main

import (
	"fmt"
	"io"
	"net"
	"net/http"
	"net/url"
	"time"
)

type Pxy struct {
	// The transport used to send proxy requests to actual server.
	// If nil, http.DefaultTransport is used.
	Transport  http.RoundTripper
	Credential string
}

// NewProxy returns a new Pxy object
func NewProxy() *Pxy {
	return &Pxy{}
}

func (p *Pxy) handleTunnel(rw http.ResponseWriter, req *http.Request, localPort int) {
	host := req.URL.Host

	hij, ok := rw.(http.Hijacker)
	if !ok {
		panic("HTTP Server does not support hijacking")
	}
	client, _, err := hij.Hijack()
	if err != nil {
		return
	}
	// Connect to the proxy server
	proxyConn, err := net.Dial("tcp", fmt.Sprintf("127.0.0.1:%d", localPort))
	if err != nil {
		return
	}
	defer proxyConn.Close()

	// Send a CONNECT request to the proxy server
	connectRequest := fmt.Sprintf("CONNECT %s HTTP/1.1\r\nHost: %s\r\n\r\n", host, host)
	_, err = proxyConn.Write([]byte(connectRequest))
	if err != nil {
		return
	}

	go io.Copy(proxyConn, client)
	io.Copy(client, proxyConn)
}

// ServeHTTP is the main handler for all requests.
func (p *Pxy) ServeHTTP(rw http.ResponseWriter, req *http.Request) {

	localPort := 2080
	if req.Method == "CONNECT" {
		p.handleTunnel(rw, req, localPort)
		return
	}

	transport := &http.Transport{
		Proxy: func(req *http.Request) (*url.URL, error) {
			return url.Parse("http://127.0.0.1:2080")
		},
		ForceAttemptHTTP2:     true,
		MaxIdleConns:          100,
		IdleConnTimeout:       90 * time.Second,
		TLSHandshakeTimeout:   10 * time.Second,
		ExpectContinueTimeout: 1 * time.Second,
	}
	outReq := new(http.Request)
	*outReq = *req
	res, err := transport.RoundTrip(outReq)
	if err != nil {
		rw.WriteHeader(http.StatusBadGateway)
		return
	}

	for key, value := range res.Header {
		for _, v := range value {
			rw.Header().Add(key, v)
		}
	}

	rw.WriteHeader(res.StatusCode)
	io.Copy(rw, res.Body)
	res.Body.Close()
}

func main() {
	pxy := NewProxy()
	err := http.ListenAndServe(fmt.Sprintf(":%d", 8080), pxy)
	if err != nil {
		panic(err)
	}
}

以上只是一个简单的案例,可以在其中添加自定义逻辑,如流量统计,动态路由等

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值