golang实现正向代理http_proxy和https_proxy

Go 复制代码
package main

import (
	"bytes"
	"fmt"
	"io"
	"log"
	"net"
	"net/url"
	"strings"
)

func main() {
	// tcp 连接,监听 8080 端口
	l, err := net.Listen("tcp", ":8080")
	if err != nil {
		log.Panic(err)
	}

	// 死循环,每当遇到连接时,调用 handle
	for {
		client, err := l.Accept()
		if err != nil {
			log.Panic(err)
		}

		go handle(client)
	}
}

func handle(client net.Conn) {
	if client == nil {
		return
	}
	defer client.Close()

	log.Printf("remote addr: %v\n", client.RemoteAddr())

	// 用来存放客户端数据的缓冲区
	var b [1024]byte
	//从客户端获取数据
	n, err := client.Read(b[:])
	if err != nil {
		log.Println(err)
		return
	}

	var method, URL, address string
	// 从客户端数据读入 method,url
	fmt.Sscanf(string(b[:bytes.IndexByte(b[:], '\n')]), "%s%s", &method, &URL)
	hostPortURL, err := url.Parse(URL)
	if err != nil {
		log.Println(err)
		return
	}

	// 如果方法是 CONNECT,则为 https 协议
	if method == "CONNECT" {
		address = hostPortURL.Scheme + ":" + hostPortURL.Opaque
	} else { //否则为 http 协议
		address = hostPortURL.Host
		// 如果 host 不带端口,则默认为 80
		if strings.Index(hostPortURL.Host, ":") == -1 { //host 不带端口, 默认 80
			address = hostPortURL.Host + ":80"
		}
	}

	//获得了请求的 host 和 port,向服务端发起 tcp 连接
	server, err := net.Dial("tcp", address)
	if err != nil {
		log.Println(err)
		return
	}
	//如果使用 https 协议,需先向客户端表示连接建立完毕
	if method == "CONNECT" {
		fmt.Fprint(client, "HTTP/1.1 200 Connection established\r\n\r\n")
	} else { //如果使用 http 协议,需将从客户端得到的 http 请求转发给服务端
		server.Write(b[:n])
	}

	//将客户端的请求转发至服务端,将服务端的响应转发给客户端。io.Copy 为阻塞函数,文件描述符不关闭就不停止
	go io.Copy(server, client)
	io.Copy(client, server)
}
相关推荐
李少兄1 小时前
Unirest:优雅的Java HTTP客户端库
java·开发语言·http
lmryBC495 小时前
golang接口-interface
java·前端·golang
浮尘笔记6 小时前
go-zero使用elasticsearch踩坑记:时间存储和展示问题
大数据·elasticsearch·golang·go
冷琅辞9 小时前
Go语言的嵌入式网络
开发语言·后端·golang
徐小黑ACG12 小时前
GO语言 使用protobuf
开发语言·后端·golang·protobuf
Aa美少女战士16 小时前
单域名 vs 通配符:如何选择最适合你的 SSL 证书?
网络协议·https·ssl
咕噜签名16 小时前
如何申请p12证书
网络协议·https·ssl
2a3b4c16 小时前
SSL/TLS
网络协议·https·ssl
沫夕残雪17 小时前
HTTP,请求响应报头,以及抓包工具的讨论
网络·vscode·网络协议·http
硪就是硪21 小时前
内网环境将nginx的http改完https访问
nginx·http·https