go实现http代理

func main() {
	var port string = "8009"
	listener, err := net.Listen("tcp", port) //监听端口号
	if err != nil {
		log.Println(err)
		os.Exit(1)
	}
	for {
		conn, err := listener.Accept() //接收请求
		if err != nil {
			log.Println(err)
			continue
		}
		go handleConnect(conn) //处理请求
	}
}

func handleConnect(conn net.Conn) {
	defer conn.Close() //关闭连接

	request, err := http.ReadRequest(bufio.NewReader(conn)) //读取请求数据
	if err != nil {
		log.Println(err)
		return
	}
	log.Println("URL:", request.URL.String())
	log.Println("Host:", request.Host)
	client, err := net.Dial("tcp", request.Host) //建立连接
	if err != nil {
		log.Println(err)
		return
	}
	defer client.Close()
	err = request.Write(client) //替换Host头并发送请求
	if err != nil {
		log.Println(err)
		return
	}
	response, err := http.ReadResponse(bufio.NewReader(client), request) //读取响应
	if err != nil {
		log.Println(response)
		return
	}
	defer response.Body.Close()
	for k, v := range response.Header {
		for _, vv := range v {
			conn.Write([]byte(k + ":" + vv + " "))
		}
	}
	conn.Write([]byte(" "))
	io.Copy(conn, response.Body)
}

以上代码实现了一个简单的HTTP代理服务,监听指定的端口,并通过net.Listen获取客户端的连接请求。然后,使用http.ReadRequest读取请求,并通过net.Dial建立与目标服务器的连接。接着,使用http.Write向目标服务器发送请求,并通过http.ReadResponse读取响应。最后,将响应头和响应体写回客户端。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值