Rust实现简易HTTP服务器时出现栈溢出问题求助

一、栈溢出的可能原因 #

结合你描述的场景（请求favicon.ico时触发，index.html正常），常见原因有以下几种：

• 栈上分配了过大的缓冲区：如果处理TcpStream的函数里在栈上声明了超大数组（比如let mut buf = [0; 1024*1024];），单次请求可能没问题，但favicon.ico是浏览器自动发起的请求，可能和主请求并发处理，多个大栈缓冲区叠加直接耗尽栈空间。Rust默认栈大小通常只有几MB，建议把缓冲区移到堆上，用Vec<u8>代替栈数组。
• 递归调用无终止条件：检查处理函数是否存在递归调用逻辑，比如解析请求出错后又再次调用自身处理同一个流，而favicon.ico的请求格式或读取过程触发了这个错误递归，导致栈帧不断累积直至溢出。
• 请求解析逻辑的无限循环：读取流数据时没正确处理EOF（文件结束）或错误状态，导致循环内反复调用函数或持续压入栈帧。比如尝试读取固定长度数据但一直读不到，进入死循环并占用栈空间。
• 错误处理不当导致栈帧堆积：处理favicon.ico请求时，某个错误分支没有正确关闭流或终止处理，反而重复执行处理逻辑，导致栈上局部变量反复创建，最终耗尽栈空间。

二、便捷设置Content-Type的方法 #

因为你要求不使用高阶crate，最直接的方式是手动维护后缀名与Content-Type的映射：

1. 提取请求文件的后缀名：通过分割文件名字符串获取最后一个.后的部分。
2. 用匹配或哈希表返回对应的MIME类型。

示例1：用match表达式实现（无额外依赖） #

fn get_content_type(filename: &str) -> &str {
    match filename.split('.').last() {
        Some("html") => "text/html; charset=utf-8",
        Some("ico") => "image/x-icon",
        Some("css") => "text/css",
        Some("js") => "application/javascript",
        Some("png") => "image/png",
        Some("jpg") | Some("jpeg") => "image/jpeg",
        _ => "application/octet-stream",
    }
}

示例2：用哈希表实现（适合扩展更多类型） #

如果需要支持更多文件类型，用哈希表更易维护：

use std::collections::HashMap;

fn get_content_type(filename: &str) -> &str {
    // 懒加载哈希表，避免每次调用重复初始化
    static MIME_MAP: once_cell::sync::Lazy<HashMap<&str, &str>> = once_cell::sync::Lazy::new(|| {
        let mut map = HashMap::new();
        map.insert("html", "text/html; charset=utf-8");
        map.insert("ico", "image/x-icon");
        map.insert("css", "text/css");
        map.insert("js", "application/javascript");
        map.insert("png", "image/png");
        map.insert("jpg", "image/jpeg");
        map.insert("svg", "image/svg+xml");
        // 可继续添加更多类型
        map
    });

    filename.split('.')
        .last()
        .and_then(|ext| MIME_MAP.get(ext))
        .copied()
        .unwrap_or("application/octet-stream")
}

注：once_cell是轻量级工具库，若不想引入，也可以在程序启动时初始化全局哈希表。

内容的提问来源于stack exchange，提问作者DreadFog