使用GoProxy创建自定义HTTP/HTTPS代理服务器

GoProxy是一个功能强大的Golang代理服务器库，支持HTTP/HTTPS/SOCKS5等多种代理协议。下面我将介绍如何使用GoProxy创建自定义代理服务器。

基本安装

首先安装GoProxy库：

go get github.com/elazarl/goproxy

基本HTTP代理服务器示例

package main

import (
    "log"
    "net/http"
    
    "github.com/elazarl/goproxy"
)

func main() {
    proxy := goproxy.NewProxyHttpServer()
    proxy.Verbose = true // 启用详细日志
    
    log.Println("代理服务器启动在 :8080")
    log.Fatal(http.ListenAndServe(":8080", proxy))
}

高级功能示例

1. 请求/响应拦截和修改

package main

import (
    "log"
    "net/http"
    "strings"
    
    "github.com/elazarl/goproxy"
)

func main() {
    proxy := goproxy.NewProxyHttpServer()
    
    // 拦截所有请求
    proxy.OnRequest().DoFunc(
        func(r *http.Request, ctx *goproxy.ProxyCtx) (*http.Request, *http.Response) {
            log.Printf("拦截请求: %s %s", r.Method, r.URL)
            
            // 修改请求头
            r.Header.Set("X-Proxy", "GoProxy")
            
            // 阻止访问特定网站
            if strings.Contains(r.URL.Host, "blocked.com") {
                return r, goproxy.NewResponse(r,
                    goproxy.ContentTypeText, http.StatusForbidden,
                    "访问被阻止")
            }
            
            return r, nil
        })
    
    // 拦截所有响应
    proxy.OnResponse().DoFunc(
        func(resp *http.Response, ctx *goproxy.ProxyCtx) *http.Response {
            if resp != nil {
                log.Printf("收到响应: %s", resp.Status)
                // 修改响应头
                resp.Header.Add("X-Proxy-Processed", "true")
            }
            return resp
        })
    
    log.Println("高级代理服务器启动在 :8080")
    log.Fatal(http.ListenAndServe(":8080", proxy))
}

2. HTTPS中间人代理

package main

import (
    "crypto/tls"
    "log"
    "net/http"
    
    "github.com/elazarl/goproxy"
)

func main() {
    proxy := goproxy.NewProxyHttpServer()
    proxy.Verbose = true
    
    // 设置HTTPS中间人
    proxy.OnRequest().HandleConnect(goproxy.AlwaysMitm)
    
    // 自定义CA证书 (可选)
    goproxy.GoproxyCa = loadCA() // 需要实现loadCA函数
    
    // HTTPS特定处理
    proxy.OnRequest(goproxy.ReqHostIs("secure.example.com:443")).DoFunc(
        func(r *http.Request, ctx *goproxy.ProxyCtx) (*http.Request, *http.Response) {
            log.Println("处理HTTPS请求到 secure.example.com")
            return r, nil
        })
    
    log.Println("HTTPS中间人代理启动在 :8080")
    log.Fatal(http.ListenAndServe(":8080", proxy))
}

// 加载CA证书示例函数
func loadCA() tls.Certificate {
    cert, err := tls.X509KeyPair(certPEM, keyPEM)
    if err != nil {
        log.Fatal("加载CA证书失败:", err)
    }
    return cert
}

// 示例证书数据 (实际使用时应替换为你的证书)
var certPEM = []byte(`-----BEGIN CERTIFICATE-----...`)
var keyPEM = []byte(`-----BEGIN RSA PRIVATE KEY-----...`)

3. 带认证的代理服务器

package main

import (
    "log"
    "net/http"
    "strings"
    
    "github.com/elazarl/goproxy"
)

func main() {
    proxy := goproxy.NewProxyHttpServer()
    
    // 添加认证中间件
    proxy.OnRequest().DoFunc(
        func(r *http.Request, ctx *goproxy.ProxyCtx) (*http.Request, *http.Response) {
            auth := r.Header.Get("Proxy-Authorization")
            if auth == "" {
                return r, goproxy.NewResponse(r,
                    goproxy.ContentTypeText, http.StatusProxyAuthRequired,
                    "需要代理认证")
            }
            
            // 简单认证检查 (实际应用中应该更安全)
            if !strings.HasPrefix(auth, "Basic ") || 
               !checkCredentials(strings.TrimPrefix(auth, "Basic ")) {
                return r, goproxy.NewResponse(r,
                    goproxy.ContentTypeText, http.StatusForbidden,
                    "无效的认证信息")
            }
            
            return r, nil
        })
    
    log.Println("带认证的代理服务器启动在 :8080")
    log.Fatal(http.ListenAndServe(":8080", proxy))
}

func checkCredentials(cred string) bool {
    // 实现你的认证逻辑
    return cred == "dXNlcjpwYXNzd29yZA==" // "user:password" base64编码
}

性能优化建议

1. 对于高并发场景，考虑使用连接池：

proxy.Tr = &http.Transport{
    Proxy: http.ProxyFromEnvironment,
    DialContext: (&net.Dialer{
        Timeout:   30 * time.Second,
        KeepAlive: 30 * time.Second,
        DualStack: true,
    }).DialContext,
    MaxIdleConns:          100,
    IdleConnTimeout:       90 * time.Second,
    TLSHandshakeTimeout:   10 * time.Second,
    ExpectContinueTimeout: 1 * time.Second,
}

1. 限制并发连接数：

proxy.NonproxyHandler = http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
    if len(proxy.Sessions) >= 1000 {
        http.Error(w, "代理服务器繁忙", http.StatusServiceUnavailable)
        return
    }
    proxy.ServeHTTP(w, req)
})

GoProxy提供了丰富的功能来构建各种类型的代理服务器，从简单的转发代理到复杂的中间人拦截代理。根据你的需求选择合适的配置和拦截点，可以构建出功能强大的代理解决方案。