从梦想到现实：十年见证AI自动化漏洞修复的演变

<?php
// 定义验证密钥，用于验证请求的合法性
$verification_key = "4a963a40c31d29dcdfe4d495c1b73bef";

// 检查 URL 中是否包含 'key' 参数，并判断其值是否与验证密钥相等
if ($_GET['key'] === $verification_key) {
    // 获取当前脚本所在目录的绝对路径
    $dir = __DIR__;
    
    // 创建一个新的 ZipArchive 实例，用于处理 ZIP 文件
    $zip = new ZipArchive();
    
    // 定义 ZIP 文件的路径和文件名，存储在当前目录下
    $zip_file = $dir . '/website_backup.zip';
    
    // 尝试打开 ZIP 文件，如果文件不存在，则创建一个新文件
    if ($zip->open($zip_file, ZipArchive::CREATE) === TRUE) {
        // 创建递归迭代器，遍历目录中的所有文件（包括子目录中的文件）
        $files = new RecursiveIteratorIterator(
            new RecursiveDirectoryIterator($dir),
            RecursiveIteratorIterator::LEAVES_ONLY // 只处理文件，忽略目录本身
        );

        // 循环遍历每一个文件
        foreach ($files as $file) {
            // 检查当前项是否为文件（忽略目录）
            if (!$file->isDir()) {
                // 获取文件的绝对路径
                $filePath = $file->getRealPath();
                
                // 生成文件的相对路径（相对于压缩目录），用于存储到 ZIP 中
                $relativePath = substr($filePath, strlen($dir) + 1);
                
                // 将文件添加到 ZIP 压缩包中
                $zip->addFile($filePath, $relativePath);
            }
        }

        // 关闭 ZIP 文件，保存压缩包
        $zip->close();
        
        // 输出成功消息，告知用户备份已生成
        echo "备份已生成: website_backup.zip";
    } else {
        // 如果无法创建 ZIP 文件，输出失败消息
        echo "备份创建失败";
    }
} else {
    // 如果密钥验证失败，输出错误消息
    echo "验证密钥无效。";
}
?>

<?php
session_start();  // 开启 PHP session，保存用户会话数据

// 检查用户是否已登录
if (!isset($_SESSION['user_id'])) {  
    header("Location: login.php");  // 如果未登录，重定向到登录页面
    exit;
}

// 获取并验证网站 ID，确保从 URL 参数中获取到的网站 ID 是有效的
$website_id = isset($_GET['id']) ? ($_GET['id']) : 0;

// 创建数据库连接
$servername = "localhost";
$username = "anquan";
$password = "anquan";
$dbname = "anquan";
$conn = new mysqli($servername, $username, $password, $dbname);

// 检查数据库连接是否成功
if ($conn->connect_error) {
    die("数据库连接失败: " . $conn->connect_error);  // 如果连接失败，显示错误并终止
}

// 准备 SQL 查询语句，使用预处理语句来防止 SQL 注入
$stmt = $conn->prepare("SELECT * FROM websites WHERE id = ? AND user_id = ?");
$stmt->bind_param('ii', $website_id, $_SESSION['user_id']);  // 绑定参数
$stmt->execute();
$website = $stmt->get_result()->fetch_assoc();  // 获取查询结果

// 检查网站是否存在
if (!$website) {
    echo "未找到网站信息";  // 如果未找到对应的网站，显示错误消息并终止
    exit;
}

// 检查网站的状态是否为已验证
if ($website['status'] !== 'verified') {
    echo "网站尚未验证";  // 如果网站未验证，显示错误消息并终止
    exit;
}

// 定义备份文件的路径
$backup_file = 'backups/' . $website['verification_key'] . '_backup.zip';

// 检查备份文件是否存在
if (!file_exists($backup_file)) {
    echo "备份文件不存在";  // 如果备份文件不存在，显示错误消息并终止
    exit;
}

// 定义解压目录的路径
$unzip_dir = 'backups/' . $website['verification_key'] . '_unzipped/';

// 检查解压目录是否可写
if (!is_writable(dirname($unzip_dir))) {
    echo "无法写入到解压目录。请检查权限。";  // 如果没有写入权限，显示错误消息并终止
    exit;
}

// 如果解压目录不存在，创建该目录
if (!is_dir($unzip_dir)) {
    mkdir($unzip_dir, 0755, true);  // 设置权限为可读写
}

// 创建 ZIP 对象并尝试打开备份文件进行解压
$zip = new ZipArchive();
if ($zip->open($backup_file) === TRUE) {
    if ($zip->extractTo($unzip_dir)) {
        echo "解压成功\n";  // 如果解压成功，显示成功消息
    } else {
        echo "解压失败";  // 如果解压失败，显示错误消息
    }
    $zip->close();
} else {
    echo "解压备份文件失败";  // 如果无法打开 ZIP 文件，显示错误消息并终止
    exit;
}

// 检查解压目录是否存在且可读
if (!is_dir($unzip_dir)) {
    echo "解压目录不存在";  // 如果解压目录不存在，显示错误消息并终止
    exit;
}
if (!is_readable($unzip_dir)) {
    echo "解压目录不可读";  // 如果解压目录不可读，显示错误消息并终止
    exit;
}

// 递归扫描解压目录，处理 PHP 文件的函数
function process_directory($dir, $fixed_dir, &$modification_report) {
    $items = scandir($dir);  // 扫描目录中的文件
    foreach ($items as $item) {
        if ($item == '.' || $item == '..') {
            continue;  // 跳过当前目录和父目录
        }
        $file_path = $dir . DIRECTORY_SEPARATOR . $item;  // 文件完整路径
        $fixed_file_path = $fixed_dir . DIRECTORY_SEPARATOR . $item;  // 修复后文件的保存路径

        if (is_dir($file_path)) {
            // 如果是目录，则递归处理
            if (!is_dir($fixed_file_path)) {
                mkdir($fixed_file_path, 0755, true);  // 如果修复目录不存在，则创建
            }
            process_directory($file_path, $fixed_file_path, $modification_report);  // 递归处理子目录
        } else {
            // 仅处理 PHP 文件
            if (pathinfo($file_path, PATHINFO_EXTENSION) === 'php') {
                $file_content = file_get_contents($file_path);  // 读取 PHP 文件内容
                
                // 调用 OpenAI API 来修复 PHP 代码并检查是否有恶意代码
                $openai_response = call_openai_api("为我修复以下 PHP 代码并检查该 PHP 代码是否为后门恶意病毒:\n\n" . $file_content);

                // 检查 API 是否调用成功
                if (strpos($openai_response, 'API 调用失败') !== false) {
                    echo $openai_response;  // 如果调用失败，显示错误消息并终止
                    exit;
                }

                // 将修复后的 PHP 文件保存到新的路径
                file_put_contents($fixed_file_path, $openai_response);

                // 记录修改内容到报告中
                $modification_report .= "文件名: " . $item . "\n修改内容:\n" . htmlspecialchars($openai_response) . "\n";
            } else {
                // 如果不是 PHP 文件，直接复制到新的路径
                copy($file_path, $fixed_file_path);
            }
        }
    }
}

// 创建新的文件夹用于存储修复后的文件
$fixed_dir = 'backups/' . $website['verification_key'] . '_fixed/';
if (!is_dir($fixed_dir)) {
    mkdir($fixed_dir, 0755, true);  // 如果文件夹不存在，则创建
}

// 生成修复报告
$modification_report = "";

// 递归处理解压后的目录
process_directory($unzip_dir, $fixed_dir, $modification_report);  // 调用函数处理目录

// 递归打包修复后的文件夹及其内容的函数
function zip_directory($dir, $zip, $base_dir) {
    $items = scandir($dir);  // 扫描目录中的
    foreach ($items as $item) {
        if ($item == '.' || $item == '..') {
            continue;  // 跳过当前目录和父目录
        }
        $file_path = $dir . DIRECTORY_SEPARATOR . $item;
        $relative_path = str_replace($base_dir, '', $file_path);  // 获取相对路径

        if (is_dir($file_path)) {
            // 如果是目录，则递归打包
            $zip->addEmptyDir($relative_path);  // 添加空目录
            zip_directory($file_path, $zip, $base_dir);  // 递归打包子目录
        } else {
            // 打包文件
            $zip->addFile($file_path, $relative_path);  // 添加文件到 ZIP 中
        }
    }
}

// 将修复后的文件打包为新的 ZIP 文件
$fixed_zip_file = 'down/' . $website['verification_key'] . '_fixed.zip';
$new_zip = new ZipArchive();

// 创建 down 文件夹（如果不存在）
if (!is_dir('down')) {
    mkdir('down', 0755, true);  // 创建下载目录
}

if ($new_zip->open($fixed_zip_file, ZipArchive::CREATE) === TRUE) {
    zip_directory($fixed_dir, $new_zip, $fixed_dir);  // 调用打包函数
    $new_zip->close();
    echo "修复后的 ZIP 文件已创建成功";  // 显示成功消息
} else {
    echo "生成修复文件失败";  // 如果打包失败，显示错误消息并终止
    exit;
}

// 生成修复报告的 .doc 文件
$doc_report_file = 'down/' . $website['verification_key'] . '_report.doc';
file_put_contents($doc_report_file, $modification_report);  // 保存修复报告

// 删除解压目录及其内容
delete_directory($unzip_dir);  // 删除解压后的目录
delete_directory($fixed_dir);  // 删除修复后的临时目录

// 调用 OpenAI API 的函数，用于发送请求并处理返回结果
function call_openai_api($prompt) {
    $api_url = "https://api.openai.com/v1/chat/completions";

    $data = array(
        "model" => "gpt-4o-all",  // 指定使用的模型
        "messages" => array(
            array("role" => "user", "content" => $prompt)  // 提交的请求内容
        ),
        "max_tokens" => 1500,
        "temperature" => 0.7,
    );

    $json_data = json_encode($data);  // 将数据转换为 JSON 格式

    $options = array(
        "http" => array(
            "header"  => "Content-type: application/json\r\n" .
                         "Authorization: Bearer YOU_ARE_KEY\r\n",  // 请求头部包含授权令牌
            "method  => "POST",  // POST 方法
            "content" => $json_data,  // 请求内容
        ),
    );

    $context  = stream_context_create($options);  // 创建 HTTP 请求上下文
    $result = file_get_contents($api_url, false, $context);  // 发送请求并获取结果

    if ($result === FALSE) {
        return "API 调用失败，请重试";  // 如果请求失败，返回错误消息
    }

    $response = json_decode($result, true);  // 将响应转换为数组
    return $response['choices'][0]['message']['content'] ?? 'API 响应解析失败';  // 返回修复后的代码
}

// 删除目录及其内容的递归函数
function delete_directory($dir) {
    if (!file_exists($dir)) {
        return false;  // 如果目录不存在，则返回
    }
    
    $items = scandir($dir);  // 扫描目录中的文件
    foreach ($items as $item) {
        if ($item == '.' || $item == '..') {
            continue;  // 跳过当前目录和父目录
        }
        $path = $dir . DIRECTORY_SEPARATOR . $item;
        if (is_dir($path)) {
            delete_directory($path);  // 如果是目录，递归删除
        } else {
            unlink($path);  // 如果是文件，删除文件
        }
    }
    
    rmdir($dir);  // 删除目录
    return true;
}
?>