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
读取响应。最后,将响应头和响应体写回客户端。