自制fgets()与标准库fgets()对比：自制函数是否更高效？

嘿，这个问题问到点子上了——很多开发者都会对标准库函数的小瑕疵耿耿于怀，自己动手造轮子也是很正常的思路。咱们来聊聊你的自制函数到底能不能更高效，以及那些容易被忽略的潜在坑点。

一、效率：大概率不如标准库的fgets #

先给你泼点冷水：自制函数几乎不可能比标准库的fgets更高效。原因很简单：

• 标准库函数是经过编译器/平台厂商高度优化的，很多底层实现用了汇编指令、批量缓冲读取等技巧，甚至会利用CPU的缓存特性来加速IO操作。
• 你的自制函数依赖strcspn来处理换行符，这意味着你可能先读取一批数据到缓冲区，再遍历找换行——相当于多了一次字符串扫描的开销。而fgets在读取过程中就会检查换行符，一旦遇到就停止读取，逻辑上更直接。
• 标准IO的缓冲机制是经过长期调优的，比如默认的全缓冲、行缓冲策略能大幅减少系统调用次数；如果你的自制函数没有做好缓冲管理（比如每次只读一个字符），那系统调用的开销会直接把效率拉低。

如果你的程序瓶颈真的在读取文件行上，与其自己造轮子，不如先检查一下IO缓冲的设置（比如用setvbuf调整缓冲大小），或者考虑用更高效的批量读取方式。

二、那些容易踩的潜在坑点 #

你解决了fgets的两个小问题，但可能引入了新的隐患，下面是几个常见的坑：

1. 参数类型的“修复”可能藏着溢出风险 #

fgets的第二个参数用int确实有点反直觉，但这是历史遗留的兼容设计（早期C标准还没有size_t）。如果你把参数改成size_t，要注意：

• 如果调用者不小心传入了负数（比如把一个错误的int值转成size_t），会被解释成超大的无符号数，直接导致缓冲区溢出。而fgets遇到n<=0的情况是未定义行为，但至少不会直接用超大值去访问内存。
• 如果你的缓冲区大小超过INT_MAX，调用fgets的时候需要做截断处理（因为fgets的int参数无法容纳这么大的值），这部分逻辑如果没处理好，也会出问题。

2. 换行符处理的边界情况 #

用strcspn找'\n'来终止数组，要考虑这些场景：

• 超长行无换行：如果一行的长度超过缓冲区大小，且没有换行符，strcspn会返回缓冲区的长度，这时候你需要确保缓冲区最后有'\0'终止符，否则会导致字符串溢出。另外，后续的读取会不会把剩下的行内容当成新行处理？
• 无换行的最后一行：如果文件最后一行没有换行符，fgets会正常返回整行内容；你的自制函数能不能正确识别这种情况，而不是误以为没读完？
• Windows风格换行（\r\n）：strcspn只会匹配'\n'，那'\r'会留在缓冲区里，后续处理字符串的时候可能会出现奇怪的问题（比如打印出来有多余的回车）。

3. 错误与EOF处理的疏漏 #

fgets的返回值规则很清晰：返回NULL表示EOF或读取错误，同时设置errno；如果读到部分数据后遇到EOF，会返回这部分数据并在末尾加'\0'。你的自制函数有没有做到：

• 正确区分EOF和读取错误？比如调用feof和ferror来判断，而不是简单返回NULL。
• 读取到部分数据后遇到EOF时，是否正确终止字符串？
• 有没有正确设置errno，方便调用者排查问题？

4. 线程安全与重入性 #

标准库的fgets在多线程环境下是线程安全的（只要每个线程用自己的FILE指针），但如果你的自制函数用到了全局变量、静态缓冲区，或者没有正确处理FILE的锁（如果用的是标准IO函数），那在多线程程序里会出现数据竞争的问题。

5. 兼容性与可维护性 #

fgets是所有C环境都支持的标准函数，你的自制函数只能在你的程序里用。如果以后要移植到其他平台（比如嵌入式系统、不同的编译器），可能需要调整代码；而其他开发者接手你的代码时，也需要花时间理解你的自制函数逻辑，不如标准库函数直观。

三、更稳妥的替代方案 #

如果你只是想解决fgets的两个小缺点，其实不用完全重写，封装fgets就够了：

#include <stdio.h>
#include <string.h>
#include <limits.h>

char *my_fgets(char *buf, size_t buf_size, FILE *stream) {
    if (buf == NULL || buf_size == 0 || stream == NULL) return NULL;
    // 把size_t转成int，确保不超过INT_MAX
    int n = (buf_size > INT_MAX) ? INT_MAX : (int)buf_size;
    char *ret = fgets(buf, n, stream);
    if (ret != NULL) {
        // 去掉尾随的换行符（包括\n和\r\n）
        buf[strcspn(buf, "\n\r")] = '\0';
    }
    return ret;
}

这个封装既利用了fgets的高效和稳定性，又解决了换行符问题，同时把参数改成了更合理的size_t，还处理了大缓冲区的情况。

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