diff --git a/.main.go b/.main.go deleted file mode 100644 index f21d445..0000000 --- a/.main.go +++ /dev/null @@ -1,279 +0,0 @@ -package main - -import ( - "encoding/json" - "fmt" - "io" - "log" - "net/http" - "regexp" - "strconv" - "strings" - - "ghproxy/config" - "ghproxy/logger" - - "github.com/gin-gonic/gin" - "github.com/imroc/req/v3" -) - -var cfg *config.Config -var logw = logger.Logw -var router *gin.Engine - -var ( - exps = []*regexp.Regexp{ - regexp.MustCompile(`^(?:https?://)?github\.com/([^/]+)/([^/]+)/(?:releases|archive)/.*`), - regexp.MustCompile(`^(?:https?://)?github\.com/([^/]+)/([^/]+)/(?:blob|raw)/.*`), - regexp.MustCompile(`^(?:https?://)?github\.com/([^/]+)/([^/]+)/(?:info|git-).*`), - regexp.MustCompile(`^(?:https?://)?raw\.github(?:usercontent|)\.com/([^/]+)/([^/]+)/.+?/.+`), - regexp.MustCompile(`^(?:https?://)?gist\.github\.com/([^/]+)/.+?/.+`), - } -) - -func loadConfig() { - var err error - // 初始化配置 - cfg, err = config.LoadConfig("/data/ghproxy/config/config.yaml") - if err != nil { - log.Fatalf("Failed to load config: %v", err) - } - fmt.Printf("Loaded config: %v\n", cfg) -} - -func setupLogger() { - // 初始化日志模块 - var err error - err = logger.Init(cfg.LogFilePath) // 传递日志文件路径 - if err != nil { - log.Fatalf("Failed to initialize logger: %v", err) - } - logw("Logger initialized") - logw("Init Completed") -} - -func init() { - loadConfig() - setupLogger() - - // 设置 Gin 模式 - gin.SetMode(gin.ReleaseMode) - - // 初始化路由 - router = gin.Default() - - // 定义路由 - router.GET("/", func(c *gin.Context) { - c.Redirect(http.StatusMovedPermanently, "https://ghproxy0rtt.1888866.xyz/") - }) - - router.GET("/api", api) - - // 健康检查 - router.GET("/api/healthcheck", func(c *gin.Context) { - c.String(http.StatusOK, "OK") - }) - - // 未匹配路由处理 - router.NoRoute(noRouteHandler(cfg)) -} - -func main() { - // 启动服务器 - err := router.Run(fmt.Sprintf("%s:%d", cfg.Host, cfg.Port)) - if err != nil { - log.Fatalf("Error starting server: %v\n", err) - } - - fmt.Println("Program finished") -} - -func api(c *gin.Context) { - // 设置响应头 - c.Writer.Header().Set("Content-Type", "application/json") - json.NewEncoder(c.Writer).Encode(map[string]interface{}{ - "MaxResponseBodySize": cfg.SizeLimit, - }) -} - -func AuthHandler(c *gin.Context) bool { - // 如果身份验证未启用，直接返回 true - if !cfg.Auth { - logw("auth PASS") - return true - } - - // 获取 auth_token 参数 - authToken := c.Query("auth_token") - logw("auth_token: ", authToken) - - // 验证 token - isValid := authToken == cfg.AuthToken - if !isValid { - logw("auth FAIL") - } - - return isValid -} - -func noRouteHandler(config *config.Config) gin.HandlerFunc { - return func(c *gin.Context) { - rawPath := strings.TrimPrefix(c.Request.URL.RequestURI(), "/") - re := regexp.MustCompile(`^(http:|https:)?/?/?(.*)`) - matches := re.FindStringSubmatch(rawPath) - - rawPath = "https://" + matches[2] - - matches = checkURL(rawPath) - if matches == nil { - c.String(http.StatusForbidden, "Invalid input.") - return - } - - if exps[1].MatchString(rawPath) { - rawPath = strings.Replace(rawPath, "/blob/", "/raw/", 1) - } - - if !AuthHandler(c) { - c.AbortWithStatusJSON(401, gin.H{"error": "Unauthorized"}) - logw("Unauthorized request: %s", rawPath) - return - } - - // 日志记录 - logw("Request: %s %s", c.Request.Method, rawPath) - logw("Matches: %v", matches) - - // 代理请求 - switch { - case exps[0].MatchString(rawPath), exps[1].MatchString(rawPath), exps[3].MatchString(rawPath), exps[4].MatchString(rawPath): - logw("%s Matched - USE proxy-chrome", rawPath) - proxyRequest(c, rawPath, config, "chrome") - case exps[2].MatchString(rawPath): - logw("%s Matched - USE proxy-git", rawPath) - proxyRequest(c, rawPath, config, "git") - default: - c.String(http.StatusForbidden, "Invalid input.") - return - } - } -} - -func proxyRequest(c *gin.Context, u string, config *config.Config, mode string) { - method := c.Request.Method - logw("%s Method: %s", u, method) - - client := req.C() - - switch mode { - case "chrome": - client.SetUserAgent("Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36"). - SetTLSFingerprintChrome(). - ImpersonateChrome() - case "git": - client.SetUserAgent("git/2.33.1") - } - - // 读取请求体 - body, err := io.ReadAll(c.Request.Body) - if err != nil { - handleError(c, fmt.Sprintf("Failed to read request body: %v", err)) - return - } - defer c.Request.Body.Close() - - // 创建新的请求 - req := client.R().SetBody(body) - - // 复制请求头 - for key, values := range c.Request.Header { - for _, value := range values { - req.SetHeader(key, value) - } - } - - // 发送请求并处理响应 - resp, err := sendRequest(req, method, u) - if err != nil { - handleError(c, fmt.Sprintf("Failed to send request: %v", err)) - return - } - defer resp.Body.Close() - - // 检查响应内容长度并处理重定向 - if err := handleResponseSize(resp, config, c); err != nil { - logw("Error handling response size: %v", err) - return - } - - copyResponseHeaders(resp, c, config) - c.Status(resp.StatusCode) - if _, err := io.Copy(c.Writer, resp.Body); err != nil { - logw("Failed to copy response body: %v", err) - } -} - -func sendRequest(req *req.Request, method, url string) (*req.Response, error) { - switch method { - case "GET": - return req.Get(url) - case "POST": - return req.Post(url) - case "PUT": - return req.Put(url) - case "DELETE": - return req.Delete(url) - default: - return nil, fmt.Errorf("unsupported method: %s", method) - } -} - -func handleResponseSize(resp *req.Response, config *config.Config, c *gin.Context) error { - contentLength := resp.Header.Get("Content-Length") - if contentLength != "" { - size, err := strconv.Atoi(contentLength) - if err == nil && size > config.SizeLimit { - finalURL := resp.Request.URL.String() - c.Redirect(http.StatusMovedPermanently, finalURL) - logw("Redirecting to %s due to size limit (%d bytes)", finalURL, size) - return fmt.Errorf("response size exceeds limit") - } - } - return nil -} - -func copyResponseHeaders(resp *req.Response, c *gin.Context, config *config.Config) { - headersToRemove := []string{"Content-Security-Policy", "Referrer-Policy", "Strict-Transport-Security"} - - for _, header := range headersToRemove { - resp.Header.Del(header) - } - - for key, values := range resp.Header { - for _, value := range values { - c.Header(key, value) - } - } - - if config.CORSOrigin { - c.Header("Access-Control-Allow-Origin", "*") - } else { - c.Header("Access-Control-Allow-Origin", "") - } -} - -func handleError(c *gin.Context, message string) { - c.String(http.StatusInternalServerError, fmt.Sprintf("server error %v", message)) - logw(message) -} - -func checkURL(u string) []string { - for _, exp := range exps { - if matches := exp.FindStringSubmatch(u); matches != nil { - logw("URL matched: %s, Matches: %v", u, matches[1:]) - return matches[1:] - } - } - logw("Invalid URL: %s", u) - return nil -} diff --git a/CHANGELOG.md b/CHANGELOG.md index 0b6670d..4b1ece4 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -7,6 +7,7 @@ - ADD&CHANGE: 增加Auth(用户鉴权)模块,并改进其的错误处理与日志记录 - CHANGE: 日志模块引入goroutine协程,提升性能 (实验性功能) - ADD: 将主要实现分离,作为Proxy模块,并优化代码结构 +- ADD: 新增安全政策 v1.0.0 --- diff --git a/SECURITY.MD b/SECURITY.MD new file mode 100644 index 0000000..1b4f7c9 --- /dev/null +++ b/SECURITY.MD @@ -0,0 +1,35 @@ +# 安全政策 + +## 支持的版本 + +以下版本的项目目前正在接受安全更新: + +| 版本 | 是否支持 | +| --- | --- | +| v1.x.x | :white_check_mark: | +| **w**a/b | :warning: 这是测试版本,用于开发测试,可能存在未知的安全隐患 | +| v0.x.x | :x: 这些版本不再受支持 | + +### 版本说明 + +- **v1.0.0**: 这是正式发布的版本 +- **xx W xx A/B***: 这是开发测试版本,可能存在未知的安全隐患。不推荐在生产环境中使用。 +- **v0.x.x**: 这些早期版本不再受支持,建议尽快升级到 v1.0.0 或更高版本。 + +### 用户须知 + +本项目为开源项目,开发者不对使用本项目造成的任何损失或问题承担责任。用户需自行评估并承担使用本项目的风险。 + +使用本项目,请遵循 **[WSL (WJQSERVER-STUDIO LICENSE)](https://wjqserver-studio.github.io/LICENSE/LICENSE.html)** 协议。 + +## 报告漏洞 + +如果您发现本项目存在安全漏洞,请通过发送ISSUES或尝试联系项目维护者来报告。请在您的报告中包含以下信息: + +1. 漏洞的描述,包括重现步骤。 +2. 受到漏洞影响的项目版本。 +3. 任何缓解因素,如利用难度或漏洞影响。 + +项目维护者将审查您的报告,并尽快解决该漏洞。请注意,根据漏洞的复杂程度,可能需要一些时间来调查和修复问题。 + +感谢您帮助保护本项目的安全! \ No newline at end of file diff --git a/auth/auth.go b/auth/auth.go index fce8cc5..a8567bc 100644 --- a/auth/auth.go +++ b/auth/auth.go @@ -12,7 +12,7 @@ var logw = logger.Logw func AuthHandler(c *gin.Context, cfg *config.Config) bool { // 如果身份验证未启用，直接返回 true if !cfg.Auth { - logw("auth PASS") + logw("auth PASSED") return true } @@ -22,13 +22,13 @@ func AuthHandler(c *gin.Context, cfg *config.Config) bool { // 验证 token if authToken == "" { - logw("auth FAIL: no auth_token provided") + logw("auth FAILED: no auth_token provided") return false } isValid := authToken == cfg.AuthToken if !isValid { - logw("auth FAIL: invalid auth_token") + logw("auth FAILED: invalid auth_token: %s", authToken) } return isValid