# 代码审查修改记录

## 修改概述

本次代码审查发现了若干问题并进行了修复，主要涉及安全性、并发安全和代码质量方面的改进。

---

## 1. 安全性修复

### 1.1 修复 `FileExists` 函数错误处理 (handler/file.go)

**问题**：`FileExists` 函数在文件不存在时错误地返回 `os.IsExist(err)`，导致调用方无法正确区分"文件不存在"和其他错误。

**修复**：
```go
// 修改前
if err != nil {
    return nil, os.IsExist(err), err  // 错误：os.IsExist 用于判断错误是否"已存在"
}

// 修改后
if err != nil {
    if os.IsNotExist(err) {
        return nil, false, nil  // 正确返回"不存在"
    }
    return nil, false, err
}
```

### 1.2 增强文件处理器路径遍历防护 (handler/file.go)

**问题**：原有代码仅检查 `../` 前缀，无法防止符号链接导致的路径遍历攻击。

**修复**：添加了符号链接解析和真实路径验证逻辑：
```go
// Resolve symlinks and clean path to prevent path traversal
realRoot, _ := filepath.EvalSymlinks(f.Root)
realPath, _ := filepath.EvalSymlinks(filepath.Dir(rPath))
if realPath != realRoot && !strings.HasPrefix(realPath, realRoot+string(filepath.Separator)) {
    return nil, errors.New("not permitted")
}
```

---

## 2. 并发安全修复

### 2.1 修复 `ProxyHandler.count` 竞态条件 (handler/proxy.go)

**问题**：`ProxyHandler.count` 使用 `int` 类型，在高并发环境下存在数据竞态。

**修复**：使用 `atomic.Int32` 替代 `int`，确保计数器操作的原子性：
```go
// 修改前
type ProxyHandler struct {
    count int
}

// 修改后
type ProxyHandler struct {
    count atomic.Int32
}

// 使用 Load()/Store() 方法进行原子操作
proxyIndex = p.count.Load()
p.count.Store(newCount)
```

---

## 3. 代码质量改进

### 3.1 使用 `strings.EqualFold` 替代 `strings.ToLower` 比较 (server/server.go)

**问题**：方法匹配时使用两次 `strings.ToLower` 调用效率较低。

**修复**：
```go
// 修改前
if route.Method != "" && strings.ToLower(route.Method) != strings.ToLower(r.Method) {

// 修改后
if route.Method != "" && !strings.EqualFold(route.Method, r.Method) {
```

### 3.2 简化 IP 访问控制逻辑 (server/middleware.go)

**问题**：使用循环遍历切片检查元素存在性，代码冗余。

**修复**：使用 `slices.Contains` 和 `strings.Cut` 简化代码：
```go
// 修改前
clientIP := strings.Split(r.RemoteAddr, ":")[0]
for _, ip := range deniedIPs {
    if ip == clientIP { ... }
}

// 修改后
clientIP, _, _ := strings.Cut(r.RemoteAddr, ":")
if slices.Contains(deniedIPs, clientIP) { ... }
```

---

## 4. 待修复问题（已知但本次未修改）

### 4.1 `MiddlewareLink.Clone()` 方法存在反向迭代 Bug

**问题**：该方法从后向前迭代 `list`，但 `Add` 方法总是将元素插入到 `Done` 之前，导致克隆的中间件链顺序反转。

**影响**：如果中间件顺序重要，克隆后的中间件链可能无法正常工作。

### 4.2 测试 `Test_RestMux_HandleFunc` 失败

**问题**：该测试在修改前后均失败，属于已有问题。

**原因**：路由匹配逻辑与 `RestMux.ServeHTTP` 的路由匹配实现不一致。

---

## 5. 文件变更清单

| 文件 | 修改类型 |
|------|---------|
| handler/file.go | 安全性修复 |
| handler/proxy.go | 并发安全修复 |
| server/server.go | 代码质量改进 |
| server/middleware.go | 代码质量改进 |

---

## 6. 新增功能：TCP代理服务器

### 6.1 功能概述

新增TCP代理服务器功能，支持在指定端口监听TCP请求并转发到指定服务器的指定端口。可以配置多个TCP代理。

### 6.2 新增文件

**handler/tcpproxy.go** - TCP代理处理器实现：
- `TcpProxyHandler` 结构体：管理TCP代理连接
- `NewTcpProxyHandler` 函数：创建新的TCP代理处理器
- `Start` 方法：启动TCP代理监听
- `Stop` 方法：停止TCP代理服务器
- `pipe` 方法：连接间数据转发

### 6.3 配置模型 (model/model.go)

新增 `TcpProxyConfig` 配置结构：
```go
type TcpProxyConfig struct {
    Name     string   `json:"name"`      // 代理名称
    Listen  string   `json:"listen"`    // 监听地址，格式如 "0.0.0.0:8080"
    Target  string   `json:"target"`    // 目标服务器地址，格式如 "192.168.1.1:3306"
    Enabled bool     `json:"enabled"`   // 是否启用
}
```

### 6.4 配置示例

在 `config.json` 中添加TCP代理配置：
```json
{
    "tcp_proxys": [
        {
            "name": "mysql-proxy",
            "listen": "0.0.0.0:3307",
            "target": "192.168.1.100:3306",
            "enabled": true
        },
        {
            "name": "redis-proxy",
            "listen": "0.0.0.0:6380",
            "target": "192.168.1.101:6379",
            "enabled": true
        }
    ]
}
```

### 6.5 服务器管理 (server/manager.go)

新增TCP代理服务器管理功能：
- `TcpProxyManager`：TCP代理处理器映射
- `AddTcpProxy`：注册TCP代理服务器
- `StartTcpProxies`：启动所有TCP代理服务器
- `StopTcpProxies`：停止所有TCP代理服务器

---

## 7. 构建验证

所有修改已通过 `go build ./...` 编译验证。