golang 编写 http AES 加密代理-Go服务器开发

下面是一个简单的 golang 编写的 http AES 加密代理示例代码：

package main

import (
    "crypto/aes"
    "crypto/cipher"
    "encoding/base64"
    "fmt"
    "io"
    "log"
    "net/http"
    "net/url"
)

const key = "1234567890123456" // 16字节的AES加密密钥

func encrypt(text string) string {
    block, err := aes.NewCipher([]byte(key))
    if err != nil {
        panic(err)
    }
    plaintext := []byte(text)
    ciphertext := make([]byte, aes.BlockSize+len(plaintext))
    iv := ciphertext[:aes.BlockSize]
    if _, err := io.ReadFull(rand.Reader, iv); err != nil {
        panic(err)
    }
    stream := cipher.NewCTR(block, iv)
    stream.XORKeyStream(ciphertext[aes.BlockSize:], plaintext)
    return base64.URLEncoding.EncodeToString(ciphertext)
}

func decrypt(text string) (string, error) {
    ciphertext, err := base64.URLEncoding.DecodeString(text)
    if err != nil {
        return "", err
    }
    block, err := aes.NewCipher([]byte(key))
    if err != nil {
        return "", err
    }
    if len(ciphertext) < aes.BlockSize {
        return "", fmt.Errorf("ciphertext too short")
    }
    iv := ciphertext[:aes.BlockSize]
    ciphertext = ciphertext[aes.BlockSize:]
    stream := cipher.NewCTR(block, iv)
    stream.XORKeyStream(ciphertext, ciphertext)
    return string(ciphertext), nil
}

func handleProxy(w http.ResponseWriter, r *http.Request) {
    targetUrl := r.URL.Query().Get("url")
    u, err := url.Parse(targetUrl)
    if err != nil {
        http.Error(w, err.Error(), http.StatusBadRequest)
        return
    }
    req, err := http.NewRequest(r.Method, u.String(), r.Body)
    if err != nil {
        http.Error(w, err.Error(), http.StatusBadRequest)
        return
    }
    resp, err := http.DefaultClient.Do(req)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }
    defer resp.Body.Close()
    w.Header().Set("Content-Type", "application/json")
    io.Copy(w, resp.Body)
}

func handleEncryptProxy(w http.ResponseWriter, r *http.Request) {
    targetUrl := r.URL.Query().Get("url")
    u, err := url.Parse(targetUrl)
    if err != nil {
        http.Error(w, err.Error(), http.StatusBadRequest)
        return
    }
    req, err := http.NewRequest(r.Method, u.String(), r.Body)
    if err != nil {
        http.Error(w, err.Error(), http.StatusBadRequest)
        return
    }
    resp, err := http.DefaultClient.Do(req)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }
    defer resp.Body.Close()
    w.Header().Set("Content-Type", "application/json")
    var encryptedBody string
    io.Copy(&encryptedBody, resp.Body)
    fmt.Fprintf(w, `{"data": "%s"}`, encrypt(encryptedBody))
}

func main() {
    http.HandleFunc("/proxy/", handleProxy)
    http.HandleFunc("/encrypt-proxy/", handleEncryptProxy)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

这段代码实现了两个路由：

/proxy/ ：直接代理访问目标地址，并返回原始响应内容。
/encrypt-proxy/ ：代理访问目标地址，加密响应内容后返回。

其中，AES 加密使用的是 CTR 模式。在 encrypt 函数中，先生成一个随机的 16 字节的初始向量（IV），再用 AES 加密算法对原始数据进行加密，并将 IV 和密文拼接成一个字节数组，并使用 base64 编码转换成字符串返回。在 decrypt 函数中，先将密文字符串解码为字节数组，然后从中提取出 IV 和密文部分，再用 AES 解密算法对密文进行解密，最后返回解密出来的明文字符串。

需要注意的是，由于原始的响应内容可能包含二进制数据，在将响应内容加密后，需要将加密后的字节数组转换成字符串类型返回给客户端。在 handleEncryptProxy 中，我使用了 fmt.Fprintf 来格式化输出 JSON 响应结果，但实际情况下，这种方式可能不太适合复杂的 JSON 结构，需要根据实际情况选择更适合的方式。