Compare commits
6 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
3e40146281 | ||
|
|
ac7e1e43b5 | ||
|
|
f134d22540 | ||
|
|
79153c0f7d | ||
|
|
4fd47812f7 | ||
|
|
17c49d534b |
12
CHANGELOG.md
12
CHANGELOG.md
@@ -1,5 +1,17 @@
|
|||||||
# 更新日志
|
# 更新日志
|
||||||
|
|
||||||
|
2.5.0 - 2025-03-17
|
||||||
|
---
|
||||||
|
- ADD: 加入脚本嵌套加速功能
|
||||||
|
- CHANGE: 改进Auth模块
|
||||||
|
|
||||||
|
25w19a - 2025-03-16
|
||||||
|
---
|
||||||
|
- PRE-RELEASE: 此版本是v2.5.0的预发布版本,请勿在生产环境中使用;
|
||||||
|
- ADD: 加入脚本嵌套加速功能
|
||||||
|
- CHANGE: 改进Auth模块
|
||||||
|
- CHANGE: 将handler模块化改进
|
||||||
|
|
||||||
2.4.2 - 2025-03-14
|
2.4.2 - 2025-03-14
|
||||||
---
|
---
|
||||||
- CHANGE: 在GitClone Cache模式下, 相关请求会使用独立httpc client
|
- CHANGE: 在GitClone Cache模式下, 相关请求会使用独立httpc client
|
||||||
|
|||||||
@@ -1 +1 @@
|
|||||||
25w18a
|
25w19a
|
||||||
14
README.md
14
README.md
@@ -20,6 +20,7 @@
|
|||||||
- 使用[Gin](https://github.com/gin-gonic/gin)作为Web框架
|
- 使用[Gin](https://github.com/gin-gonic/gin)作为Web框架
|
||||||
- 使用[Touka-HTTPC](https://github.com/satomitouka/touka-httpc)作为HTTP客户端
|
- 使用[Touka-HTTPC](https://github.com/satomitouka/touka-httpc)作为HTTP客户端
|
||||||
- 支持Git clone,raw,realeases等文件拉取
|
- 支持Git clone,raw,realeases等文件拉取
|
||||||
|
- 支持Git Clone缓存(配合组件)
|
||||||
- 支持Docker部署
|
- 支持Docker部署
|
||||||
- 支持速率限制
|
- 支持速率限制
|
||||||
- 支持用户鉴权
|
- 支持用户鉴权
|
||||||
@@ -31,8 +32,9 @@
|
|||||||
**本项目是[WJQSERVER-STUDIO/ghproxy-go](https://github.com/WJQSERVER-STUDIO/ghproxy-go)的重构版本,实现了原项目原定功能的同时,进一步优化了性能**
|
**本项目是[WJQSERVER-STUDIO/ghproxy-go](https://github.com/WJQSERVER-STUDIO/ghproxy-go)的重构版本,实现了原项目原定功能的同时,进一步优化了性能**
|
||||||
关于此项目的详细开发过程,请参看Commit记录与[CHANGELOG.md](https://github.com/WJQSERVER-STUDIO/ghproxy/blob/main/CHANGELOG.md)
|
关于此项目的详细开发过程,请参看Commit记录与[CHANGELOG.md](https://github.com/WJQSERVER-STUDIO/ghproxy/blob/main/CHANGELOG.md)
|
||||||
|
|
||||||
- V2.0.0 对`proxy`核心模块进行了重构,大幅优化内存占用
|
- v2.4.1 对路径匹配进行优化
|
||||||
- V1.0.0 迁移至本仓库,并再次重构内容实现
|
- v2.0.0 对`proxy`核心模块进行了重构,大幅优化内存占用
|
||||||
|
- v1.0.0 迁移至本仓库,并再次重构内容实现
|
||||||
- v0.2.0 重构项目实现
|
- v0.2.0 重构项目实现
|
||||||
|
|
||||||
### LICENSE
|
### LICENSE
|
||||||
@@ -48,9 +50,11 @@
|
|||||||
```
|
```
|
||||||
# 下载文件
|
# 下载文件
|
||||||
https://ghproxy.1888866.xyz/raw.githubusercontent.com/WJQSERVER-STUDIO/tools-stable/main/tools-stable-ghproxy.sh
|
https://ghproxy.1888866.xyz/raw.githubusercontent.com/WJQSERVER-STUDIO/tools-stable/main/tools-stable-ghproxy.sh
|
||||||
|
https://ghproxy.1888866.xyz/https://raw.githubusercontent.com/WJQSERVER-STUDIO/tools-stable/main/tools-stable-ghproxy.sh
|
||||||
|
|
||||||
# 克隆仓库
|
# 克隆仓库
|
||||||
git clone https://ghproxy.1888866.xyz/github.com/WJQSERVER-STUDIO/ghproxy.git
|
git clone https://ghproxy.1888866.xyz/github.com/WJQSERVER-STUDIO/ghproxy.git
|
||||||
|
git clone https://ghproxy.1888866.xyz/https://github.com/WJQSERVER-STUDIO/ghproxy.git
|
||||||
```
|
```
|
||||||
|
|
||||||
## 部署说明
|
## 部署说明
|
||||||
@@ -107,6 +111,9 @@ mode = "bypass" # bypass / cache 运行模式, cache模式依赖smart-git
|
|||||||
smartGitAddr = "http://127.0.0.1:8080" # smart-git组件地址
|
smartGitAddr = "http://127.0.0.1:8080" # smart-git组件地址
|
||||||
ForceH2C = false # 强制使用H2C连接
|
ForceH2C = false # 强制使用H2C连接
|
||||||
|
|
||||||
|
[shell]
|
||||||
|
editor = false # 脚本嵌套加速
|
||||||
|
|
||||||
[pages]
|
[pages]
|
||||||
mode = "internal" # "internal" or "external" 内部/外部 前端 默认内部
|
mode = "internal" # "internal" or "external" 内部/外部 前端 默认内部
|
||||||
theme = "bootstrap" # "bootstrap" or "nebula" 内置主题
|
theme = "bootstrap" # "bootstrap" or "nebula" 内置主题
|
||||||
@@ -121,6 +128,7 @@ level = "info" # 日志级别 dump, debug, info, warn, error, none
|
|||||||
authMethod = "parameters" # 鉴权方式,支持parameters,header
|
authMethod = "parameters" # 鉴权方式,支持parameters,header
|
||||||
authToken = "token" # 用户鉴权Token
|
authToken = "token" # 用户鉴权Token
|
||||||
enabled = false # 是否开启用户鉴权
|
enabled = false # 是否开启用户鉴权
|
||||||
|
ForceAllowApi = false # 在不开启Header鉴权的情况下允许api代理
|
||||||
|
|
||||||
[blacklist]
|
[blacklist]
|
||||||
blacklistFile = "/data/ghproxy/config/blacklist.json" # 黑名单文件路径
|
blacklistFile = "/data/ghproxy/config/blacklist.json" # 黑名单文件路径
|
||||||
@@ -189,6 +197,8 @@ url = "socks5://127.0.0.1:1080" # "http://127.0.0.1:7890" 支持Socks5/HTTP(S)
|
|||||||
|
|
||||||
爱发电: https://afdian.com/a/wjqserver
|
爱发电: https://afdian.com/a/wjqserver
|
||||||
|
|
||||||
|
USDT(TRC20): `TNfSYG6F2vkiibd6J6mhhHNWDgWgNdF5hN`
|
||||||
|
|
||||||
### 捐赠列表
|
### 捐赠列表
|
||||||
|
|
||||||
虚位以待...
|
虚位以待...
|
||||||
|
|||||||
@@ -7,23 +7,21 @@ import (
|
|||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
)
|
)
|
||||||
|
|
||||||
func AuthHeaderHandler(c *gin.Context, cfg *config.Config) (isValid bool, err string) {
|
func AuthHeaderHandler(c *gin.Context, cfg *config.Config) (isValid bool, err error) {
|
||||||
if !cfg.Auth.Enabled {
|
if !cfg.Auth.Enabled {
|
||||||
return true, ""
|
return true, nil
|
||||||
}
|
}
|
||||||
// 获取"GH-Auth"的值
|
// 获取"GH-Auth"的值
|
||||||
authToken := c.GetHeader("GH-Auth")
|
authToken := c.GetHeader("GH-Auth")
|
||||||
logDebug("%s %s %s %s %s AUTH_TOKEN: %s", c.Request.Method, c.Request.Host, c.Request.URL.Path, c.Request.Proto, c.Request.RemoteAddr, authToken)
|
logDebug("%s %s %s %s %s AUTH_TOKEN: %s", c.Request.Method, c.Request.Host, c.Request.URL.Path, c.Request.Proto, c.Request.RemoteAddr, authToken)
|
||||||
if authToken == "" {
|
if authToken == "" {
|
||||||
err := "Auth Header == nil"
|
return false, fmt.Errorf("Auth token not found")
|
||||||
return false, err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
isValid = authToken == cfg.Auth.AuthToken
|
isValid = authToken == cfg.Auth.AuthToken
|
||||||
if !isValid {
|
if !isValid {
|
||||||
err := fmt.Sprintf("Auth token incorrect: %s", authToken)
|
return false, fmt.Errorf("Auth token incorrect")
|
||||||
return false, err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return isValid, ""
|
return isValid, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,24 +7,22 @@ import (
|
|||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
)
|
)
|
||||||
|
|
||||||
func AuthParametersHandler(c *gin.Context, cfg *config.Config) (isValid bool, err string) {
|
func AuthParametersHandler(c *gin.Context, cfg *config.Config) (isValid bool, err error) {
|
||||||
if !cfg.Auth.Enabled {
|
if !cfg.Auth.Enabled {
|
||||||
return true, ""
|
return true, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
authToken := c.Query("auth_token")
|
authToken := c.Query("auth_token")
|
||||||
logDebug("%s %s %s %s %s AUTH_TOKEN: %s", c.ClientIP(), c.Request.Method, c.Request.URL.Path, c.Request.UserAgent(), c.Request.Proto, authToken)
|
logDebug("%s %s %s %s %s AUTH_TOKEN: %s", c.ClientIP(), c.Request.Method, c.Request.URL.Path, c.Request.UserAgent(), c.Request.Proto, authToken)
|
||||||
|
|
||||||
if authToken == "" {
|
if authToken == "" {
|
||||||
err := "Auth token == nil"
|
return false, fmt.Errorf("Auth token not found")
|
||||||
return false, err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
isValid = authToken == cfg.Auth.AuthToken
|
isValid = authToken == cfg.Auth.AuthToken
|
||||||
if !isValid {
|
if !isValid {
|
||||||
err := fmt.Sprintf("Auth token incorrect: %s", authToken)
|
return false, fmt.Errorf("Auth token invalid")
|
||||||
return false, err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return isValid, ""
|
return isValid, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
package auth
|
package auth
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"ghproxy/config"
|
"ghproxy/config"
|
||||||
|
|
||||||
"github.com/WJQSERVER-STUDIO/go-utils/logger"
|
"github.com/WJQSERVER-STUDIO/go-utils/logger"
|
||||||
@@ -34,7 +35,7 @@ func Init(cfg *config.Config) {
|
|||||||
logDebug("Auth Init")
|
logDebug("Auth Init")
|
||||||
}
|
}
|
||||||
|
|
||||||
func AuthHandler(c *gin.Context, cfg *config.Config) (isValid bool, err string) {
|
func AuthHandler(c *gin.Context, cfg *config.Config) (isValid bool, err error) {
|
||||||
if cfg.Auth.AuthMethod == "parameters" {
|
if cfg.Auth.AuthMethod == "parameters" {
|
||||||
isValid, err = AuthParametersHandler(c, cfg)
|
isValid, err = AuthParametersHandler(c, cfg)
|
||||||
return isValid, err
|
return isValid, err
|
||||||
@@ -43,9 +44,9 @@ func AuthHandler(c *gin.Context, cfg *config.Config) (isValid bool, err string)
|
|||||||
return isValid, err
|
return isValid, err
|
||||||
} else if cfg.Auth.AuthMethod == "" {
|
} else if cfg.Auth.AuthMethod == "" {
|
||||||
logError("Auth method not set")
|
logError("Auth method not set")
|
||||||
return true, ""
|
return true, nil
|
||||||
} else {
|
} else {
|
||||||
logError("Auth method not supported")
|
logError("Auth method not supported")
|
||||||
return false, "Auth method not supported"
|
return false, fmt.Errorf(fmt.Sprintf("Auth method %s not supported", cfg.Auth.AuthMethod))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ type Config struct {
|
|||||||
Server ServerConfig
|
Server ServerConfig
|
||||||
Httpc HttpcConfig
|
Httpc HttpcConfig
|
||||||
GitClone GitCloneConfig
|
GitClone GitCloneConfig
|
||||||
|
Shell ShellConfig
|
||||||
Pages PagesConfig
|
Pages PagesConfig
|
||||||
Log LogConfig
|
Log LogConfig
|
||||||
Auth AuthConfig
|
Auth AuthConfig
|
||||||
@@ -62,6 +63,14 @@ type GitCloneConfig struct {
|
|||||||
ForceH2C bool `toml:"ForceH2C"`
|
ForceH2C bool `toml:"ForceH2C"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
[shell]
|
||||||
|
editor = true
|
||||||
|
*/
|
||||||
|
type ShellConfig struct {
|
||||||
|
Editor bool `toml:"editor"`
|
||||||
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
[pages]
|
[pages]
|
||||||
mode = "internal" # "internal" or "external"
|
mode = "internal" # "internal" or "external"
|
||||||
@@ -82,11 +91,20 @@ type LogConfig struct {
|
|||||||
Level string `toml:"level"`
|
Level string `toml:"level"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
[auth]
|
||||||
|
authMethod = "parameters" # "header" or "parameters"
|
||||||
|
authToken = "token"
|
||||||
|
enabled = false
|
||||||
|
passThrough = false
|
||||||
|
ForceAllowApi = true
|
||||||
|
*/
|
||||||
type AuthConfig struct {
|
type AuthConfig struct {
|
||||||
Enabled bool `toml:"enabled"`
|
Enabled bool `toml:"enabled"`
|
||||||
AuthMethod string `toml:"authMethod"`
|
AuthMethod string `toml:"authMethod"`
|
||||||
AuthToken string `toml:"authToken"`
|
AuthToken string `toml:"authToken"`
|
||||||
PassThrough bool `toml:"passThrough"`
|
PassThrough bool `toml:"passThrough"`
|
||||||
|
ForceAllowApi bool `toml:"ForceAllowApi"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type BlacklistConfig struct {
|
type BlacklistConfig struct {
|
||||||
|
|||||||
@@ -17,6 +17,9 @@ mode = "bypass" # bypass / cache
|
|||||||
smartGitAddr = "http://127.0.0.1:8080"
|
smartGitAddr = "http://127.0.0.1:8080"
|
||||||
ForceH2C = false
|
ForceH2C = false
|
||||||
|
|
||||||
|
[shell]
|
||||||
|
editor = false
|
||||||
|
|
||||||
[pages]
|
[pages]
|
||||||
mode = "internal" # "internal" or "external"
|
mode = "internal" # "internal" or "external"
|
||||||
theme = "bootstrap" # "bootstrap" or "nebula"
|
theme = "bootstrap" # "bootstrap" or "nebula"
|
||||||
@@ -32,6 +35,7 @@ authMethod = "parameters" # "header" or "parameters"
|
|||||||
authToken = "token"
|
authToken = "token"
|
||||||
enabled = false
|
enabled = false
|
||||||
passThrough = false
|
passThrough = false
|
||||||
|
ForceAllowApi = false
|
||||||
|
|
||||||
[blacklist]
|
[blacklist]
|
||||||
blacklistFile = "/data/ghproxy/config/blacklist.json"
|
blacklistFile = "/data/ghproxy/config/blacklist.json"
|
||||||
|
|||||||
@@ -17,6 +17,9 @@ mode = "bypass" # bypass / cache
|
|||||||
smartGitAddr = "http://127.0.0.1:8080"
|
smartGitAddr = "http://127.0.0.1:8080"
|
||||||
ForceH2C = false
|
ForceH2C = false
|
||||||
|
|
||||||
|
[shell]
|
||||||
|
editor = false
|
||||||
|
|
||||||
[pages]
|
[pages]
|
||||||
mode = "internal" # "internal" or "external"
|
mode = "internal" # "internal" or "external"
|
||||||
theme = "bootstrap" # "bootstrap" or "nebula"
|
theme = "bootstrap" # "bootstrap" or "nebula"
|
||||||
@@ -32,6 +35,7 @@ authMethod = "parameters" # "header" or "parameters"
|
|||||||
authToken = "token"
|
authToken = "token"
|
||||||
enabled = false
|
enabled = false
|
||||||
passThrough = false
|
passThrough = false
|
||||||
|
ForceAllowApi = false
|
||||||
|
|
||||||
[blacklist]
|
[blacklist]
|
||||||
blacklistFile = "/usr/local/ghproxy/config/blacklist.json"
|
blacklistFile = "/usr/local/ghproxy/config/blacklist.json"
|
||||||
|
|||||||
@@ -12,7 +12,7 @@ import (
|
|||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
)
|
)
|
||||||
|
|
||||||
func ChunkedProxyRequest(c *gin.Context, u string, cfg *config.Config, mode string, runMode string) {
|
func ChunkedProxyRequest(c *gin.Context, u string, cfg *config.Config, matcher string) {
|
||||||
method := c.Request.Method
|
method := c.Request.Method
|
||||||
|
|
||||||
// 发送HEAD请求, 预获取Content-Length
|
// 发送HEAD请求, 预获取Content-Length
|
||||||
@@ -23,6 +23,7 @@ func ChunkedProxyRequest(c *gin.Context, u string, cfg *config.Config, mode stri
|
|||||||
}
|
}
|
||||||
setRequestHeaders(c, headReq)
|
setRequestHeaders(c, headReq)
|
||||||
removeWSHeader(headReq) // 删除Conection Upgrade头, 避免与HTTP/2冲突(检查是否存在Upgrade头)
|
removeWSHeader(headReq) // 删除Conection Upgrade头, 避免与HTTP/2冲突(检查是否存在Upgrade头)
|
||||||
|
reWriteEncodeHeader(headReq)
|
||||||
AuthPassThrough(c, cfg, headReq)
|
AuthPassThrough(c, cfg, headReq)
|
||||||
|
|
||||||
headResp, err := client.Do(headReq)
|
headResp, err := client.Do(headReq)
|
||||||
@@ -64,6 +65,7 @@ func ChunkedProxyRequest(c *gin.Context, u string, cfg *config.Config, mode stri
|
|||||||
}
|
}
|
||||||
setRequestHeaders(c, req)
|
setRequestHeaders(c, req)
|
||||||
removeWSHeader(req) // 删除Conection Upgrade头, 避免与HTTP/2冲突(检查是否存在Upgrade头)
|
removeWSHeader(req) // 删除Conection Upgrade头, 避免与HTTP/2冲突(检查是否存在Upgrade头)
|
||||||
|
reWriteEncodeHeader(req)
|
||||||
AuthPassThrough(c, cfg, req)
|
AuthPassThrough(c, cfg, req)
|
||||||
|
|
||||||
resp, err := client.Do(req)
|
resp, err := client.Do(req)
|
||||||
@@ -106,6 +108,9 @@ func ChunkedProxyRequest(c *gin.Context, u string, cfg *config.Config, mode stri
|
|||||||
resp.Header.Del(header)
|
resp.Header.Del(header)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
//c.Header("Accept-Encoding", "gzip")
|
||||||
|
//c.Header("Content-Encoding", "gzip")
|
||||||
|
|
||||||
/*
|
/*
|
||||||
if cfg.CORS.Enabled {
|
if cfg.CORS.Enabled {
|
||||||
c.Header("Access-Control-Allow-Origin", "*")
|
c.Header("Access-Control-Allow-Origin", "*")
|
||||||
@@ -127,12 +132,30 @@ func ChunkedProxyRequest(c *gin.Context, u string, cfg *config.Config, mode stri
|
|||||||
|
|
||||||
c.Status(resp.StatusCode)
|
c.Status(resp.StatusCode)
|
||||||
|
|
||||||
//_, err = io.CopyBuffer(c.Writer, resp.Body, nil)
|
if MatcherShell(u) && matchString(matcher, matchedMatchers) && cfg.Shell.Editor {
|
||||||
_, err = copyb.CopyBuffer(c.Writer, resp.Body, nil)
|
// 判断body是不是gzip
|
||||||
if err != nil {
|
var compress string
|
||||||
logError("%s %s %s %s %s Failed to copy response body: %v", c.ClientIP(), method, u, c.Request.Header.Get("User-Agent"), c.Request.Proto, err)
|
if resp.Header.Get("Content-Encoding") == "gzip" {
|
||||||
return
|
compress = "gzip"
|
||||||
|
}
|
||||||
|
|
||||||
|
logInfo("Is Shell: %s %s %s %s %s", c.ClientIP(), method, u, c.Request.Header.Get("User-Agent"), c.Request.Proto)
|
||||||
|
c.Header("Content-Length", "")
|
||||||
|
_, err = processLinks(resp.Body, c.Writer, compress, c.Request.Host, cfg)
|
||||||
|
if err != nil {
|
||||||
|
logError("%s %s %s %s %s Failed to copy response body: %v", c.ClientIP(), method, u, c.Request.Header.Get("User-Agent"), c.Request.Proto, err)
|
||||||
|
return
|
||||||
|
} else {
|
||||||
|
c.Writer.Flush() // 确保刷入
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
c.Writer.Flush() // 确保刷入
|
//_, err = io.CopyBuffer(c.Writer, resp.Body, nil)
|
||||||
|
_, err = copyb.CopyBuffer(c.Writer, resp.Body, nil)
|
||||||
|
if err != nil {
|
||||||
|
logError("%s %s %s %s %s Failed to copy response body: %v", c.ClientIP(), method, u, c.Request.Header.Get("User-Agent"), c.Request.Proto, err)
|
||||||
|
return
|
||||||
|
} else {
|
||||||
|
c.Writer.Flush() // 确保刷入
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
package proxy
|
package proxy
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"ghproxy/auth"
|
"ghproxy/auth"
|
||||||
"ghproxy/config"
|
"ghproxy/config"
|
||||||
@@ -65,89 +66,20 @@ func NoRouteHandler(cfg *config.Config, limiter *rate.RateLimiter, iplimiter *ra
|
|||||||
// 制作url
|
// 制作url
|
||||||
rawPath = "https://" + matches[2]
|
rawPath = "https://" + matches[2]
|
||||||
|
|
||||||
var (
|
user, repo, matcher, err := Matcher(rawPath, cfg)
|
||||||
user string
|
if err != nil {
|
||||||
repo string
|
if errors.Is(err, ErrInvalidURL) {
|
||||||
matcher string
|
|
||||||
)
|
|
||||||
|
|
||||||
// 匹配 "https://github.com"开头的链接
|
|
||||||
if strings.HasPrefix(rawPath, "https://github.com") {
|
|
||||||
remainingPath := strings.TrimPrefix(rawPath, "https://github.com")
|
|
||||||
if strings.HasPrefix(remainingPath, "/") {
|
|
||||||
remainingPath = strings.TrimPrefix(remainingPath, "/")
|
|
||||||
}
|
|
||||||
// 预期格式/user/repo/more...
|
|
||||||
// 取出user和repo和最后部分
|
|
||||||
parts := strings.Split(remainingPath, "/")
|
|
||||||
if len(parts) <= 2 {
|
|
||||||
logWarning("%s %s %s %s %s Invalid URL", c.ClientIP(), c.Request.Method, rawPath, c.Request.Header.Get("User-Agent"), c.Request.Proto)
|
|
||||||
c.String(http.StatusForbidden, "Invalid URL Format. Path: %s", rawPath)
|
c.String(http.StatusForbidden, "Invalid URL Format. Path: %s", rawPath)
|
||||||
|
logWarning(err.Error())
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
user = parts[0]
|
if errors.Is(err, ErrAuthHeaderUnavailable) {
|
||||||
repo = parts[1]
|
c.String(http.StatusForbidden, "AuthHeader Unavailable")
|
||||||
// 匹配 "https://github.com"开头的链接
|
logWarning(err.Error())
|
||||||
if len(parts) >= 3 {
|
|
||||||
switch parts[2] {
|
|
||||||
case "releases", "archive":
|
|
||||||
matcher = "releases"
|
|
||||||
case "blob", "raw":
|
|
||||||
matcher = "blob"
|
|
||||||
case "info", "git-upload-pack":
|
|
||||||
matcher = "clone"
|
|
||||||
default:
|
|
||||||
fmt.Println("Invalid URL: Unknown type")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// 匹配 "https://raw"开头的链接
|
|
||||||
if strings.HasPrefix(rawPath, "https://raw") {
|
|
||||||
remainingPath := strings.TrimPrefix(rawPath, "https://")
|
|
||||||
parts := strings.Split(remainingPath, "/")
|
|
||||||
if len(parts) <= 3 {
|
|
||||||
logWarning("%s %s %s %s %s Invalid URL", c.ClientIP(), c.Request.Method, rawPath, c.Request.Header.Get("User-Agent"), c.Request.Proto)
|
|
||||||
c.String(http.StatusForbidden, "Invalid URL Format. Path: %s", rawPath)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
user = parts[1]
|
|
||||||
repo = parts[2]
|
|
||||||
matcher = "raw"
|
|
||||||
}
|
|
||||||
// 匹配 "https://gist"开头的链接
|
|
||||||
if strings.HasPrefix(rawPath, "https://gist") {
|
|
||||||
remainingPath := strings.TrimPrefix(rawPath, "https://")
|
|
||||||
parts := strings.Split(remainingPath, "/")
|
|
||||||
if len(parts) <= 3 {
|
|
||||||
logWarning("%s %s %s %s %s Invalid URL", c.ClientIP(), c.Request.Method, rawPath, c.Request.Header.Get("User-Agent"), c.Request.Proto)
|
|
||||||
c.String(http.StatusForbidden, "Invalid URL Format. Path: %s", rawPath)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
user = parts[1]
|
|
||||||
matcher = "gist"
|
|
||||||
}
|
|
||||||
// 匹配 "https://api.github.com/"开头的链接
|
|
||||||
if strings.HasPrefix(rawPath, "https://api.github.com/") {
|
|
||||||
matcher = "api"
|
|
||||||
remainingPath := strings.TrimPrefix(rawPath, "https://api.github.com/")
|
|
||||||
|
|
||||||
parts := strings.Split(remainingPath, "/")
|
|
||||||
if parts[0] == "repos" {
|
|
||||||
user = parts[1]
|
|
||||||
repo = parts[2]
|
|
||||||
}
|
|
||||||
if parts[0] == "users" {
|
|
||||||
user = parts[1]
|
|
||||||
}
|
|
||||||
if cfg.Auth.AuthMethod != "header" || !cfg.Auth.Enabled {
|
|
||||||
c.JSON(http.StatusForbidden, gin.H{"error": "HeaderAuth is not enabled."})
|
|
||||||
logError("%s %s %s %s %s HeaderAuth-Error: HeaderAuth is not enabled.", c.ClientIP(), c.Request.Method, rawPath, c.Request.Header.Get("User-Agent"), c.Request.Proto)
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
username := user
|
username := user
|
||||||
//username, repo := MatchUserRepo(rawPath, cfg, c, matches) // 匹配用户名和仓库名
|
|
||||||
|
|
||||||
logInfo("%s %s %s %s %s Matched-Username: %s, Matched-Repo: %s", c.ClientIP(), c.Request.Method, rawPath, c.Request.Header.Get("User-Agent"), c.Request.Proto, username, repo)
|
logInfo("%s %s %s %s %s Matched-Username: %s, Matched-Repo: %s", c.ClientIP(), c.Request.Method, rawPath, c.Request.Header.Get("User-Agent"), c.Request.Proto, username, repo)
|
||||||
// dump log 记录详细信息 c.ClientIP(), c.Request.Method, rawPath, c.Request.Header.Get("User-Agent"), c.Request.Proto, full Header
|
// dump log 记录详细信息 c.ClientIP(), c.Request.Method, rawPath, c.Request.Header.Get("User-Agent"), c.Request.Proto, full Header
|
||||||
@@ -195,7 +127,8 @@ func NoRouteHandler(cfg *config.Config, limiter *rate.RateLimiter, iplimiter *ra
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 鉴权
|
// 鉴权
|
||||||
authcheck, err := auth.AuthHandler(c, cfg)
|
var authcheck bool
|
||||||
|
authcheck, err = auth.AuthHandler(c, cfg)
|
||||||
if !authcheck {
|
if !authcheck {
|
||||||
c.AbortWithStatusJSON(401, gin.H{"error": "Unauthorized"})
|
c.AbortWithStatusJSON(401, gin.H{"error": "Unauthorized"})
|
||||||
logWarning("%s %s %s %s %s Auth-Error: %v", c.ClientIP(), c.Request.Method, rawPath, c.Request.Header.Get("User-Agent"), c.Request.Proto, err)
|
logWarning("%s %s %s %s %s Auth-Error: %v", c.ClientIP(), c.Request.Method, rawPath, c.Request.Header.Get("User-Agent"), c.Request.Proto, err)
|
||||||
@@ -207,8 +140,7 @@ func NoRouteHandler(cfg *config.Config, limiter *rate.RateLimiter, iplimiter *ra
|
|||||||
|
|
||||||
switch matcher {
|
switch matcher {
|
||||||
case "releases", "blob", "raw", "gist", "api":
|
case "releases", "blob", "raw", "gist", "api":
|
||||||
//ProxyRequest(c, rawPath, cfg, "chrome", runMode)
|
ChunkedProxyRequest(c, rawPath, cfg, matcher)
|
||||||
ChunkedProxyRequest(c, rawPath, cfg, "chrome", runMode) // dev test chunk
|
|
||||||
case "clone":
|
case "clone":
|
||||||
//ProxyRequest(c, rawPath, cfg, "git", runMode)
|
//ProxyRequest(c, rawPath, cfg, "git", runMode)
|
||||||
GitReq(c, rawPath, cfg, "git", runMode)
|
GitReq(c, rawPath, cfg, "git", runMode)
|
||||||
|
|||||||
@@ -35,20 +35,6 @@ func InitReq(cfg *config.Config) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func initHTTPClient(cfg *config.Config) {
|
func initHTTPClient(cfg *config.Config) {
|
||||||
/*
|
|
||||||
ctr = &http.Transport{
|
|
||||||
MaxIdleConns: 100,
|
|
||||||
MaxConnsPerHost: 60,
|
|
||||||
IdleConnTimeout: 20 * time.Second,
|
|
||||||
TLSHandshakeTimeout: 10 * time.Second,
|
|
||||||
ExpectContinueTimeout: 1 * time.Second,
|
|
||||||
ResponseHeaderTimeout: 10 * time.Second,
|
|
||||||
DialContext: (&net.Dialer{
|
|
||||||
Timeout: 30 * time.Second,
|
|
||||||
KeepAlive: 30 * time.Second,
|
|
||||||
}).DialContext,
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
var proTolcols = new(http.Protocols)
|
var proTolcols = new(http.Protocols)
|
||||||
proTolcols.SetHTTP1(true)
|
proTolcols.SetHTTP1(true)
|
||||||
proTolcols.SetHTTP2(true)
|
proTolcols.SetHTTP2(true)
|
||||||
|
|||||||
@@ -1,34 +1,286 @@
|
|||||||
package proxy
|
package proxy
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bufio"
|
||||||
|
"compress/gzip"
|
||||||
"fmt"
|
"fmt"
|
||||||
"ghproxy/config"
|
"ghproxy/config"
|
||||||
"net/http"
|
"io"
|
||||||
"regexp"
|
"regexp"
|
||||||
|
"strings"
|
||||||
"github.com/gin-gonic/gin"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// 预定义regex
|
// 定义错误类型, error承载描述, 便于处理
|
||||||
var (
|
type MatcherErrors struct {
|
||||||
pathRegex = regexp.MustCompile(`^([^/]+)/([^/]+)/([^/]+)/.*`) // 匹配路径
|
Code int
|
||||||
gistRegex = regexp.MustCompile(`^(?:https?://)?gist\.github(?:usercontent|)\.com/([^/]+)/([^/]+)/.*`) // 匹配gist路径
|
Msg string
|
||||||
)
|
Err error
|
||||||
|
}
|
||||||
// 提取用户名和仓库名
|
|
||||||
func MatchUserRepo(rawPath string, cfg *config.Config, c *gin.Context, matches []string) (string, string) {
|
var (
|
||||||
if gistMatches := gistRegex.FindStringSubmatch(rawPath); len(gistMatches) == 3 {
|
ErrInvalidURL = &MatcherErrors{
|
||||||
logDump("%s %s %s %s %s Matched-Username: %s", c.ClientIP(), c.Request.Method, rawPath, c.Request.Header.Get("User-Agent"), c.Request.Proto, gistMatches[1])
|
Code: 403,
|
||||||
return gistMatches[1], ""
|
Msg: "Invalid URL Format",
|
||||||
}
|
}
|
||||||
// 定义路径
|
ErrAuthHeaderUnavailable = &MatcherErrors{
|
||||||
if pathMatches := pathRegex.FindStringSubmatch(matches[2]); len(pathMatches) >= 4 {
|
Code: 403,
|
||||||
return pathMatches[2], pathMatches[3]
|
Msg: "AuthHeader Unavailable",
|
||||||
}
|
}
|
||||||
|
)
|
||||||
// 返回错误信息
|
|
||||||
errMsg := fmt.Sprintf("%s %s %s %s %s Invalid URL", c.ClientIP(), c.Request.Method, rawPath, c.Request.Header.Get("User-Agent"), c.Request.Proto)
|
func (e *MatcherErrors) Error() string {
|
||||||
logWarning(errMsg)
|
if e.Err != nil {
|
||||||
c.String(http.StatusForbidden, "Invalid path; expected username/repo, Path: %s", rawPath)
|
return fmt.Sprintf("Code: %d, Msg: %s, Err: %s", e.Code, e.Msg, e.Err.Error())
|
||||||
return "", ""
|
}
|
||||||
|
return fmt.Sprintf("Code: %d, Msg: %s", e.Code, e.Msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *MatcherErrors) Unwrap() error {
|
||||||
|
return e.Err
|
||||||
|
}
|
||||||
|
|
||||||
|
func Matcher(rawPath string, cfg *config.Config) (string, string, string, error) {
|
||||||
|
var (
|
||||||
|
user string
|
||||||
|
repo string
|
||||||
|
matcher string
|
||||||
|
)
|
||||||
|
// 匹配 "https://github.com"开头的链接
|
||||||
|
if strings.HasPrefix(rawPath, "https://github.com") {
|
||||||
|
remainingPath := strings.TrimPrefix(rawPath, "https://github.com")
|
||||||
|
if strings.HasPrefix(remainingPath, "/") {
|
||||||
|
remainingPath = strings.TrimPrefix(remainingPath, "/")
|
||||||
|
}
|
||||||
|
// 预期格式/user/repo/more...
|
||||||
|
// 取出user和repo和最后部分
|
||||||
|
parts := strings.Split(remainingPath, "/")
|
||||||
|
if len(parts) <= 2 {
|
||||||
|
return "", "", "", ErrInvalidURL
|
||||||
|
}
|
||||||
|
user = parts[0]
|
||||||
|
repo = parts[1]
|
||||||
|
// 匹配 "https://github.com"开头的链接
|
||||||
|
if len(parts) >= 3 {
|
||||||
|
switch parts[2] {
|
||||||
|
case "releases", "archive":
|
||||||
|
matcher = "releases"
|
||||||
|
case "blob", "raw":
|
||||||
|
matcher = "blob"
|
||||||
|
case "info", "git-upload-pack":
|
||||||
|
matcher = "clone"
|
||||||
|
default:
|
||||||
|
return "", "", "", ErrInvalidURL
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return user, repo, matcher, nil
|
||||||
|
}
|
||||||
|
// 匹配 "https://raw"开头的链接
|
||||||
|
if strings.HasPrefix(rawPath, "https://raw") {
|
||||||
|
remainingPath := strings.TrimPrefix(rawPath, "https://")
|
||||||
|
parts := strings.Split(remainingPath, "/")
|
||||||
|
if len(parts) <= 3 {
|
||||||
|
return "", "", "", ErrInvalidURL
|
||||||
|
}
|
||||||
|
user = parts[1]
|
||||||
|
repo = parts[2]
|
||||||
|
matcher = "raw"
|
||||||
|
|
||||||
|
return user, repo, matcher, nil
|
||||||
|
}
|
||||||
|
// 匹配 "https://gist"开头的链接
|
||||||
|
if strings.HasPrefix(rawPath, "https://gist") {
|
||||||
|
remainingPath := strings.TrimPrefix(rawPath, "https://")
|
||||||
|
parts := strings.Split(remainingPath, "/")
|
||||||
|
if len(parts) <= 3 {
|
||||||
|
return "", "", "", ErrInvalidURL
|
||||||
|
}
|
||||||
|
user = parts[1]
|
||||||
|
repo = ""
|
||||||
|
matcher = "gist"
|
||||||
|
return user, repo, matcher, nil
|
||||||
|
}
|
||||||
|
// 匹配 "https://api.github.com/"开头的链接
|
||||||
|
if strings.HasPrefix(rawPath, "https://api.github.com/") {
|
||||||
|
matcher = "api"
|
||||||
|
remainingPath := strings.TrimPrefix(rawPath, "https://api.github.com/")
|
||||||
|
|
||||||
|
parts := strings.Split(remainingPath, "/")
|
||||||
|
if parts[0] == "repos" {
|
||||||
|
user = parts[1]
|
||||||
|
repo = parts[2]
|
||||||
|
}
|
||||||
|
if parts[0] == "users" {
|
||||||
|
user = parts[1]
|
||||||
|
}
|
||||||
|
if !cfg.Auth.ForceAllowApi {
|
||||||
|
if cfg.Auth.AuthMethod != "header" || !cfg.Auth.Enabled {
|
||||||
|
return "", "", "", ErrAuthHeaderUnavailable
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return user, repo, matcher, nil
|
||||||
|
}
|
||||||
|
return "", "", "", ErrInvalidURL
|
||||||
|
}
|
||||||
|
|
||||||
|
func EditorMatcher(rawPath string, cfg *config.Config) (bool, string, error) {
|
||||||
|
var (
|
||||||
|
matcher string
|
||||||
|
)
|
||||||
|
// 匹配 "https://github.com"开头的链接
|
||||||
|
if strings.HasPrefix(rawPath, "https://github.com") {
|
||||||
|
remainingPath := strings.TrimPrefix(rawPath, "https://github.com")
|
||||||
|
if strings.HasPrefix(remainingPath, "/") {
|
||||||
|
remainingPath = strings.TrimPrefix(remainingPath, "/")
|
||||||
|
}
|
||||||
|
return true, "", nil
|
||||||
|
}
|
||||||
|
// 匹配 "https://raw.githubusercontent.com"开头的链接
|
||||||
|
if strings.HasPrefix(rawPath, "https://raw.githubusercontent.com") {
|
||||||
|
return true, matcher, nil
|
||||||
|
}
|
||||||
|
// 匹配 "https://raw.github.com"开头的链接
|
||||||
|
if strings.HasPrefix(rawPath, "https://raw.github.com") {
|
||||||
|
return true, matcher, nil
|
||||||
|
}
|
||||||
|
// 匹配 "https://gist.githubusercontent.com"开头的链接
|
||||||
|
if strings.HasPrefix(rawPath, "https://gist.githubusercontent.com") {
|
||||||
|
return true, matcher, nil
|
||||||
|
}
|
||||||
|
// 匹配 "https://gist.github.com"开头的链接
|
||||||
|
if strings.HasPrefix(rawPath, "https://gist.github.com") {
|
||||||
|
return true, matcher, nil
|
||||||
|
}
|
||||||
|
// 匹配 "https://api.github.com/"开头的链接
|
||||||
|
if strings.HasPrefix(rawPath, "https://api.github.com") {
|
||||||
|
matcher = "api"
|
||||||
|
return true, matcher, nil
|
||||||
|
}
|
||||||
|
return false, "", ErrInvalidURL
|
||||||
|
}
|
||||||
|
|
||||||
|
// 匹配文件扩展名是sh的rawPath
|
||||||
|
func MatcherShell(rawPath string) bool {
|
||||||
|
if strings.HasSuffix(rawPath, ".sh") {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// LinkProcessor 是一个函数类型,用于处理提取到的链接。
|
||||||
|
type LinkProcessor func(string) string
|
||||||
|
|
||||||
|
// 自定义 URL 修改函数
|
||||||
|
func modifyURL(url string, host string, cfg *config.Config) string {
|
||||||
|
// 去除url内的https://或http://
|
||||||
|
matched, _, err := EditorMatcher(url, cfg)
|
||||||
|
if err != nil {
|
||||||
|
logDump("Invalid URL: %s", url)
|
||||||
|
return url
|
||||||
|
}
|
||||||
|
if matched {
|
||||||
|
|
||||||
|
u := strings.TrimPrefix(url, "https://")
|
||||||
|
u = strings.TrimPrefix(url, "http://")
|
||||||
|
logDump("Modified URL: %s", "https://"+host+"/"+u)
|
||||||
|
return "https://" + host + "/" + u
|
||||||
|
}
|
||||||
|
return url
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
matchedMatchers = []string{
|
||||||
|
"blob",
|
||||||
|
"raw",
|
||||||
|
"gist",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
// matchString 检查目标字符串是否在给定的字符串集合中
|
||||||
|
func matchString(target string, stringsToMatch []string) bool {
|
||||||
|
matchMap := make(map[string]struct{}, len(stringsToMatch))
|
||||||
|
for _, str := range stringsToMatch {
|
||||||
|
matchMap[str] = struct{}{}
|
||||||
|
}
|
||||||
|
_, exists := matchMap[target]
|
||||||
|
return exists
|
||||||
|
}
|
||||||
|
|
||||||
|
// processLinks 处理链接并将结果写入输出流
|
||||||
|
func processLinks(input io.Reader, output io.Writer, compress string, host string, cfg *config.Config) (written int64, err error) {
|
||||||
|
var reader *bufio.Reader
|
||||||
|
|
||||||
|
if compress == "gzip" {
|
||||||
|
// 解压gzip
|
||||||
|
gzipReader, err := gzip.NewReader(input)
|
||||||
|
if err != nil {
|
||||||
|
return 0, fmt.Errorf("gzip解压错误: %v", err)
|
||||||
|
}
|
||||||
|
defer gzipReader.Close()
|
||||||
|
reader = bufio.NewReader(gzipReader)
|
||||||
|
} else {
|
||||||
|
reader = bufio.NewReader(input)
|
||||||
|
}
|
||||||
|
|
||||||
|
var writer *bufio.Writer
|
||||||
|
var gzipWriter *gzip.Writer
|
||||||
|
|
||||||
|
// 根据是否gzip确定 writer 的创建
|
||||||
|
if compress == "gzip" {
|
||||||
|
gzipWriter = gzip.NewWriter(output)
|
||||||
|
writer = bufio.NewWriterSize(gzipWriter, 4096) //设置缓冲区大小
|
||||||
|
} else {
|
||||||
|
writer = bufio.NewWriterSize(output, 4096)
|
||||||
|
}
|
||||||
|
|
||||||
|
//确保writer关闭
|
||||||
|
defer func() {
|
||||||
|
var closeErr error // 局部变量,用于保存defer中可能发生的错误
|
||||||
|
|
||||||
|
if gzipWriter != nil {
|
||||||
|
if closeErr = gzipWriter.Close(); closeErr != nil {
|
||||||
|
logError("gzipWriter close failed %v", closeErr)
|
||||||
|
// 如果已经存在错误,则保留。否则,记录此错误。
|
||||||
|
if err == nil {
|
||||||
|
err = closeErr
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if flushErr := writer.Flush(); flushErr != nil {
|
||||||
|
logError("writer flush failed %v", flushErr)
|
||||||
|
// 如果已经存在错误,则保留。否则,记录此错误。
|
||||||
|
if err == nil {
|
||||||
|
err = flushErr
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
// 使用正则表达式匹配 http 和 https 链接
|
||||||
|
urlPattern := regexp.MustCompile(`https?://[^\s'"]+`)
|
||||||
|
for {
|
||||||
|
line, err := reader.ReadString('\n')
|
||||||
|
if err != nil {
|
||||||
|
if err == io.EOF {
|
||||||
|
break // 文件结束
|
||||||
|
}
|
||||||
|
return written, fmt.Errorf("读取行错误: %v", err) // 传递错误
|
||||||
|
}
|
||||||
|
|
||||||
|
// 替换所有匹配的 URL
|
||||||
|
modifiedLine := urlPattern.ReplaceAllStringFunc(line, func(originalURL string) string {
|
||||||
|
return modifyURL(originalURL, host, cfg)
|
||||||
|
})
|
||||||
|
|
||||||
|
n, werr := writer.WriteString(modifiedLine)
|
||||||
|
written += int64(n) // 更新写入的字节数
|
||||||
|
if werr != nil {
|
||||||
|
return written, fmt.Errorf("写入文件错误: %v", werr) // 传递错误
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 在返回之前,再刷新一次
|
||||||
|
if fErr := writer.Flush(); fErr != nil {
|
||||||
|
return written, fErr
|
||||||
|
}
|
||||||
|
|
||||||
|
return written, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ package proxy
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
)
|
)
|
||||||
@@ -19,3 +20,31 @@ func removeWSHeader(req *http.Request) {
|
|||||||
req.Header.Del("Upgrade")
|
req.Header.Del("Upgrade")
|
||||||
req.Header.Del("Connection")
|
req.Header.Del("Connection")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func reWriteEncodeHeader(req *http.Request) {
|
||||||
|
|
||||||
|
if isGzipAccepted(req.Header) {
|
||||||
|
req.Header.Set("Content-Encoding", "gzip")
|
||||||
|
req.Header.Set("Accept-Encoding", "gzip")
|
||||||
|
} else {
|
||||||
|
req.Header.Del("Content-Encoding")
|
||||||
|
req.Header.Del("Accept-Encoding")
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// isGzipAccepted 检查 Accept-Encoding 头部中是否包含 gzip
|
||||||
|
func isGzipAccepted(header http.Header) bool {
|
||||||
|
// 获取 Accept-Encoding 的值
|
||||||
|
encodings := header["Accept-Encoding"]
|
||||||
|
for _, encoding := range encodings {
|
||||||
|
// 将 encoding 字符串拆分为多个编码
|
||||||
|
for _, enc := range strings.Split(encoding, ",") {
|
||||||
|
// 去除空格并检查是否为 gzip
|
||||||
|
if strings.TrimSpace(enc) == "gzip" {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user