基于golang的http(s)與socks5代理伺服器的代碼實作

背景

很多制作公司為了安全起見，大部分内部設計人員的電腦是沒法聯網的，隻有少數部分電腦可以連外網（比如制片人員的電腦）。但是在外包這種模式下，為了能夠讓設計人員友善的接包任務，并回傳結果，需要設計人員的電腦能夠通過代理服務（部署在制片的電腦上）上指定的網站，這個時候就需要我們實作代理請求轉發的功能。

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代理，你需要執行以下步驟：

1. 擷取用戶端請求: 使用Golang的net包接收用戶端請求。
2. 轉發請求: 使用Golang的http包将請求轉發到服務端。
3. 擷取服務端響應: 從服務端接收響應并将其傳回給用戶端。

以下是一個示例代碼，實作了一個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)
    }
}