个人开发者的AI聊天“作死”行为大赏

1个月前发布

2783

1. “裸奔AJAX”式API

// functions.php 里的灾难代码
add_action('wp_ajax_nopriv_chat', 'handle_chat'); // nopriv！谁都能调用！
add_action('wp_ajax_chat', 'handle_chat');

function handle_chat() {
    // 没有任何验证！
    $prompt = $_POST['prompt']; // 直接使用$_POST，SQL注入在招手
    $response = call_gpt($prompt);
    
    wp_send_json_success($response); // 直接输出，XSS准备好了
}

锐评：这是把自家客厅改成了公共网吧，连监控摄像头都没装。攻击者可以用这个接口：

刷爆你的API额度
注入恶意代码
发起DDoS攻击
还能顺便帮你把网站搞崩

2. “密码硬编码”绝技

// 在插件根目录的config.php
define('OPENAI_API_KEY', 'sk-live-abc123...');
define('DB_PASSWORD', 'root123');

// 然后在JavaScript里直接使用
echo "<script>
    const apiKey = '" . OPENAI_API_KEY . "';
    // 恭喜，API Key现在前端可见了！
</script>";

锐评：这相当于把银行卡密码写在便利贴上，然后贴在网吧的显示器上。GitHub一搜一大把这种“送钱插件”。

3. “全开放式”REST API

// register_rest_route 的灾难用法
register_rest_route('ai-chat/v1', '/generate', [
    'methods' => 'POST',
    'callback' => 'generate_text',
    'permission_callback' => '__return_true' // 经典！谁都有权限！
]);

function generate_text($request) {
    $messages = $request->get_param('messages');
    // 没有任何额度检查
    return call_openai($messages); // 无限免费使用！
}

锐评：这已经不是开门揖盗了，这是开了个24小时自助银行，还贴了告示”钞票自取，不用登记”。

4. “佛系”Nonce验证

// 以为加了nonce就安全了
wp_localize_script('chat-script', 'ajax_object', [
    'ajax_url' => admin_url('admin-ajax.php'),
    'nonce' => wp_create_nonce('chat_nonce') // 但这个nonce对所有用户都一样！
]);

// 验证时
function handle_chat() {
    if (!wp_verify_nonce($_POST['nonce'], 'chat_nonce')) {
        die('Security check failed'); // 心理安慰罢了
    }
    // 继续处理...
}

锐评：用万能钥匙当门锁，还觉得自己家很安全。WordPress nonce不是为API安全设计的！

5. “日志泄露”全家桶

// 调试信息直接输出
function log_chat($user_id, $prompt, $response) {
    $log = "User $user_id asked: $prompt\nGPT replied: $response\n";
    file_put_contents(
        ABSPATH . 'chat_logs.txt',  // 放在网站根目录！
        $log,
        FILE_APPEND
    );
    
    // 更骚的操作
    error_log("信用卡支付成功: " . $_POST['card_number']); // 写到PHP错误日志
}

锐评：这不是日志系统，这是给黑客写的使用说明书。访问 yoursite.com/chat_logs.txt 就能看到所有对话记录。

WordPress AI插件安全防护正确姿势

1. 多层防护架构

// 1. 自定义API端点，不用admin-ajax
add_action('rest_api_init', function() {
    register_rest_route('secure-ai/v1', '/chat', [
        'methods' => 'POST',
        'callback' => 'secure_chat_handler',
        'permission_callback' => 'verify_api_request' // 关键！
    ]);
});

// 2. 严格的权限验证
function verify_api_request($request) {
    // 检查API密钥
    $api_key = $request->get_header('X-API-Key');
    if (!$api_key || !validate_api_key($api_key)) {
        return new WP_Error('forbidden', 'Invalid API key', ['status' => 403]);
    }
    
    // 检查用户状态
    $user_id = get_user_id_by_api_key($api_key);
    if (!user_can_access_ai($user_id)) {
        return new WP_Error('forbidden', 'Insufficient credits', ['status' => 402]);
    }
    
    // 频率限制
    if (is_rate_limited($user_id)) {
        return new WP_Error('too_many_requests', 'Rate limit exceeded', ['status' => 429]);
    }
    
    return true;
}

2. 完整的限流系统

class AI_Rate_Limiter {
    private $table_name;
    
    public function __construct() {
        global $wpdb;
        $this->table_name = $wpdb->prefix . 'ai_rate_limit';
    }
    
    public function check_limit($user_id, $cost = 1) {
        $current_hour = date('Y-m-d H:00:00');
        
        // 使用WordPress数据库
        $usage = $wpdb->get_var($wpdb->prepare(
            "SELECT SUM(cost) FROM {$this->table_name} 
             WHERE user_id = %d AND hour = %s",
            $user_id, $current_hour
        ));
        
        $user_limit = get_user_meta($user_id, 'ai_hourly_limit', true) ?: 1000;
        
        return ($usage + $cost) <= $user_limit;
    }
    
    public function record_usage($user_id, $tokens) {
        global $wpdb;
        
        $cost = ceil($tokens / 1000); // 每1000token计1次
        
        $wpdb->insert($this->table_name, [
            'user_id' => $user_id,
            'hour' => date('Y-m-d H:00:00'),
            'cost' => $cost,
            'created_at' => current_time('mysql')
        ]);
    }
}

3. 安全的密钥管理

// 使用WordPress选项API加密存储
class API_Key_Manager {
    const ENCRYPTION_KEY = 'your-encryption-key-here'; // 应该从wp-config.php定义
    
    public static function store_api_key($user_id, $api_key) {
        // 加密存储
        $encrypted = self::encrypt($api_key);
        update_user_meta($user_id, '_encrypted_openai_key', $encrypted);
        
        // 生成访问令牌
        $access_token = wp_generate_password(32, false);
        update_user_meta($user_id, '_ai_access_token', wp_hash_password($access_token));
        
        return $access_token; // 只返回一次
    }
    
    private static function encrypt($data) {
        // 使用openssl加密
        $iv = openssl_random_pseudo_bytes(openssl_cipher_iv_length('aes-256-cbc'));
        $encrypted = openssl_encrypt($data, 'aes-256-cbc', self::ENCRYPTION_KEY, 0, $iv);
        return base64_encode($iv . $encrypted);
    }
}

4. 输入验证和清理

function validate_chat_request($data) {
    // 1. 检查必需字段
    $required = ['messages', 'model'];
    foreach ($required as $field) {
        if (empty($data[$field])) {
            return new WP_Error('invalid_input', "Missing field: $field");
        }
    }
    
    // 2. 验证消息结构
    if (!is_array($data['messages'])) {
        return new WP_Error('invalid_input', 'Messages must be an array');
    }
    
    // 3. 限制消息数量
    if (count($data['messages']) > 20) {
        return new WP_Error('invalid_input', 'Too many messages (max 20)');
    }
    
    // 4. 清理内容
    foreach ($data['messages'] as &$message) {
        $message['content'] = wp_strip_all_tags($message['content']);
        $message['content'] = substr($message['content'], 0, 2000); // 长度限制
    }
    
    // 5. 模型白名单
    $allowed_models = ['gpt-3.5-turbo', 'gpt-4', 'gpt-4-turbo'];
    if (!in_array($data['model'], $allowed_models)) {
        return new WP_Error('invalid_input', 'Model not allowed');
    }
    
    return $data;
}

5. 完整的插件安全架构

/*
 * 安全AI聊天插件架构
 * 
 * 目录结构：
 * secure-ai-chat/
 * ├── includes/
 * │   ├── class-api-manager.php      # API密钥管理
 * │   ├── class-rate-limiter.php     # 限流系统
 * │   ├── class-audit-log.php        # 审计日志
 * │   └── class-billing.php          # 计费系统
 * ├── admin/
 * │   └── settings-page.php          # 后台设置
 * ├── public/
 * │   └── api-endpoints.php          # REST API端点
 * └── secure-ai-chat.php             # 主文件
 */
 
// 主文件示例
class Secure_AI_Chat {
    private $rate_limiter;
    private $api_manager;
    private $audit_log;
    
    public function __construct() {
        // 初始化所有组件
        $this->rate_limiter = new AI_Rate_Limiter();
        $this->api_manager = new API_Key_Manager();
        $this->audit_log = new Audit_Log();
        
        // 注册API端点
        add_action('rest_api_init', [$this, 'register_api_routes']);
        
        // 添加后台菜单
        add_action('admin_menu', [$this, 'add_admin_menu']);
        
        // 激活/停用钩子
        register_activation_hook(__FILE__, [$this, 'create_tables']);
        register_deactivation_hook(__FILE__, [$this, 'cleanup']);
    }
    
    public function register_api_routes() {
        // 所有API端点都需要验证
        register_rest_route('secure-ai/v1', '/chat', [
            'methods' => 'POST',
            'callback' => [$this, 'handle_chat_request'],
            'permission_callback' => [$this, 'validate_api_key']
        ]);
        
        // 用户额度查询
        register_rest_route('secure-ai/v1', '/usage', [
            'methods' => 'GET',
            'callback' => [$this, 'get_usage'],
            'permission_callback' => [$this, 'validate_api_key']
        ]);
    }
    
    public function validate_api_key($request) {
        // 多层验证逻辑
        // ...
    }
}

给WordPress AI插件开发者的血泪建议

必须做的：

永远不用 wp_ajax_nopriv：除非你想被刷破产
使用REST API + 权限回调：这是WordPress最安全的API方式
数据库操作必须用 $wpdb->prepare：防止SQL注入
输出必须用 esc_* 函数：防止XSS
敏感数据加密存储：不要用 update_option 直接存API密钥

强烈推荐的WordPress安全插件：

// 在插件中集成这些安全措施
add_action('init', function() {
    // 1. 隐藏WordPress版本
    remove_action('wp_head', 'wp_generator');
    
    // 2. 限制REST API访问
    if (!is_user_logged_in()) {
        add_filter('rest_authentication_errors', function($result) {
            // 只允许特定端点公开访问
            $route = $GLOBALS['wp']->query_vars['rest_route'] ?? '';
            $public_routes = ['/wp/v2/posts', '/secure-ai/v1/public'];
            
            if (!in_array($route, $public_routes)) {
                return new WP_Error('rest_forbidden', 'Access denied', ['status' => 403]);
            }
            return $result;
        });
    }
});

监控和告警：

// 简单但有效的监控
add_action('secure_ai_request_complete', function($user_id, $tokens_used, $success) {
    // 记录到WordPress日志
    $log_message = sprintf(
        'AI Request - User: %d, Tokens: %d, Success: %s',
        $user_id,
        $tokens_used,
        $success ? 'Yes' : 'No'
    );
    
    // 高风险操作告警
    if ($tokens_used > 10000) {
        wp_mail(
            get_option('admin_email'),
            '高额Token使用告警',
            "用户 {$user_id} 单次使用了 {$tokens_used} tokens"
        );
    }
    
    // 异常频率检测
    $hourly_requests = get_transient("user_{$user_id}_hourly_requests") ?: 0;
    $hourly_requests++;
    set_transient("user_{$user_id}_hourly_requests", $hourly_requests, HOUR_IN_SECONDS);
    
    if ($hourly_requests > 100) {
        // 自动限制该用户
        update_user_meta($user_id, '_ai_rate_limited', time() + 3600);
    }
}, 10, 3);

总结：WordPress AI插件安全黄金法则

假设所有输入都是恶意的：WordPress的$_POST、$_GET都可能被污染
权限检查要早于逻辑处理：先验证，再处理
使用WordPress自带的安全函数：wp_verify_nonce, $wpdb->prepare, esc_* 系列
不要信任前端：所有验证必须在服务端进行
日志要安全存储：不要放在web可访问目录
API密钥要加密：不要用明文存储在数据库
一定要限流：没有限流的AI API等于送钱

最后警告：如果你在WordPress插件目录里看到以下代码，请立即关闭网站并报警（程序员界的报警）：

// 自杀式代码三件套
eval($_POST['code']);  // 1. 代码执行
$wpdb->query("SELECT * FROM users WHERE id = {$_GET['id']}"); // 2. SQL注入
file_put_contents($_GET['file'], $_POST['content']); // 3. 任意文件写入