基于Golang的Http(s)與Socks5代理服務器的代碼實現
作者：研道鳩摩智
在Go語言中，可以使用標準庫的 Net/Http 包來實現HTTP代理。
背景
很多制作公司為了安全起見，大部分內部設計人員的電腦是沒法聯網的，只有少數部分電腦可以連外網（比如制片人員的電腦）。但是在外包這種模式下，為了能夠讓設計人員方便的接包任務，并回傳結果，需要設計人員的電腦能夠通過代理服務（部署在制片的電腦上）上指定的網站，這個時候就需要我們實現代理請求轉發的功能。
HTTP代理
在Go語言中，可以使用標準庫的 net/http 包來實現HTTP代理。
下面是一個簡單的例子，實現了一個HTTP代理服務器，它可以代理客戶端的HTTP請求并返回請求的響應：
package main
import (
"net"
"net/http"
"net/http/httputil"
)
func handleHTTP(w http.ResponseWriter, req *http.Request) {
resp, err := http.DefaultTransport.RoundTrip(req)
if err != nil {
http.Error(w, err.Error(), http.StatusServiceUnavailable)
return
}
defer resp.Body.Close()
copyHeader(w.Header(), resp.Header)
w.WriteHeader(resp.StatusCode)
io.Copy(w, resp.Body)
}
func copyHeader(dst, src http.Header) {
for k, vv := range src {
for _, v := range vv {
dst.Add(k, v)
}
}
}HTTPS代理
Golang可以輕松實現一個https代理，你需要執行以下步驟：
- 獲取客戶端請求: 使用Golang的net包接收客戶端請求。
- 轉發請求: 使用Golang的http包將請求轉發到服務端。
- 獲取服務端響應: 從服務端接收響應并將其返回給客戶端。
以下是一個示例代碼，實現了一個https代理：
package main
import (
"bufio"
"io"
"net"
"net/http"
"net/http/httputil"
)
func handleHttps(w http.ResponseWriter, r *http.Request) {
dest_conn, err := net.DialTimeout("tcp", r.Host, 10*time.Second)
if err != nil {
http.Error(w, err.Error(), http.StatusServiceUnavailable)
return
}
w.WriteHeader(http.StatusOK)
hijacker, ok := w.(http.Hijacker)
if !ok {
http.Error(w, "Hijacking not supported", http.StatusInternalServerError)
return
}
client_conn, _, err := hijacker.Hijack()
if err != nil {
http.Error(w, err.Error(), http.StatusServiceUnavailable)
}
go transfer(dest_conn, client_conn)
go transfer(client_conn, dest_conn)
}
func transfer(destination io.WriteCloser, source io.ReadCloser) {
defer destination.Close()
defer source.Close()
io.Copy(destination, source)
}
func handleHTTP(w http.ResponseWriter, req *http.Request) {
resp, err := http.DefaultTransport.RoundTrip(req)
if err != nil {
http.Error(w, err.Error(), http.StatusServiceUnavailable)
return
}
defer resp.Body.Close()
copyHeader(w.Header(), resp.Header)
w.WriteHeader(resp.StatusCode)
io.Copy(w, resp.Body)
}
func copyHeader(dst, src http.Header) {
for k, vv := range src {
for _, v := range vv {
dst.Add(k, v)
}
}
}SOCKS代理
Socks5協議是一種靈活的代理協議,主要用于實現代理服務器的功能，允許客戶端通過代理服務器與其他網絡資源進行通信。Socks5協議比Socks4協議更先進，具有更多的特性，如它支持用戶名/密碼驗證，還支持TCP和UDP協議.下面實現的是Socks5代理協議
package main
import (
"bufio"
"fmt"
"net"
"os"
)
func main() {
l, err := net.Listen("tcp", ":1080")
if err != nil {
fmt.Println(err)
os.Exit(1)
}
defer l.Close()
for {
client, err := l.Accept()
if err != nil {
fmt.Println(err)
continue
}
go handleClientRequest(client)
}
}
func handleClientRequest(client net.Conn) {
if client == nil {
return
}
defer client.Close()
var b [1024]byte
n, err := client.Read(b[:])
if err != nil {
fmt.Println(err)
return
}
var host, port string
switch b[3] {
case 0x01: // IPv4
host = net.IPv4(b[4], b[5], b[6], b[7]).String()
port = fmt.Sprintf("%d", b[8]<<8|b[9])
case 0x03: // Domain name
host = string(b[5 : n-2])
port = fmt.Sprintf("%d", b[n-2]<<8|b[n-1])
case 0x04: // IPv6
host = net.IP{b[4], b[5], b[6], b[7], b[8], b[9], b[10], b[11], b[12], b[13], b[14], b[15], b[16], b[17], b[18], b[19]}.String()
port = fmt.Sprintf("%d", b[20]<<8|b[21])
}
server, err := net.Dial("tcp", host+":"+port)
if err != nil {
fmt.Println(err)
return
}
defer server.Close()
client.Write([]byte{0x05, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x08, 0x43})
go func() {
_, err := io.Copy(server, client)
if err != nil {
fmt.Println("io.Copy error:", err)
}
}()
_, err = io.Copy(client, server)
if err != nil {
fmt.Println("io.Copy error:", err)
}
}
責任編輯：姜華
來源：
今日頭條
