<?php
// 允许跨域请求
header('Access-Control-Allow-Origin: *');
header('Access-Control-Allow-Headers: Origin, X-Requested-With, Content-Type, Accept');
header('Access-Control-Allow-Methods: GET, POST, OPTIONS');

// 如果是OPTIONS请求，直接返回
if ($_SERVER['REQUEST_METHOD'] == 'OPTIONS') {
    exit(0);
}

// Ping测试
if (isset($_GET['action']) && $_GET['action'] == 'ping') {
    // 返回当前时间戳作为ping响应
    echo time();
    exit;
}

// 下载测速
if (isset($_GET['action']) && $_GET['action'] == 'download') {
    // 获取请求的文件大小，默认5MB
    $size = isset($_GET['size']) ? intval($_GET['size']) : 5 * 1024 * 1024;
    
    // 设置响应头
    header('Content-Type: application/octet-stream');
    header('Content-Length: ' . $size);
    header('Content-Disposition: attachment; filename="test.bin"');
    
    // 禁用输出缓冲
    if (ob_get_level()) {
        ob_end_clean();
    }
    
    // 生成随机数据并输出
    $chunkSize = 8192; // 8KB每次
    $remaining = $size;
    
    while ($remaining > 0) {
        $chunk = min($chunkSize, $remaining);
        echo random_bytes($chunk);
        $remaining -= $chunk;
        
        // 立即刷新输出缓冲区
        if (ob_get_level()) {
            ob_flush();
        }
        flush();
    }
    exit;
}

// 上传测速
if (isset($_GET['action']) && $_GET['action'] == 'upload' && $_SERVER['REQUEST_METHOD'] == 'POST') {
    // 直接读取并丢弃上传的数据，不保存到服务器
    $input = fopen('php://input', 'r');
    $total = 0;
    while ($data = fread($input, 8192)) {
        $total += strlen($data);
    }
    fclose($input);
    
    // 返回成功响应
    echo 'OK';
    exit;
}

// 如果没有匹配的操作，返回404
http_response_code(404);
echo 'Not Found';
?>