C语言中用户文件路径输入的动态内存安全处理咨询

现有代码的安全性说明 #

你当前的代码用%1024s确实比无限制的%s更安全，因为它限制了读取的字符数，避免缓冲区溢出。但存在两个问题：一是固定1024的缓冲区可能无法容纳超长路径，二是scanf("%s")会在空格、制表符或换行处停止读取，导致包含空格的文件路径被截断。

1. 重构代码实现动态内存分配 #

推荐使用POSIX标准的getline函数（GCC、Clang及多数Unix-like系统支持，Windows下可通过MinGW或MSVC最新版本支持），它会自动分配足够的内存存储整行输入，无需手动管理初始缓冲区大小。

重构后的代码示例：

#include <stdio.h>
#include <stdlib.h>

int main() {
    char *filePath = NULL;
    size_t bufSize = 0;
    ssize_t readLen;

    printf("Please provide file to encrypt (File Path): ");
    // getline自动分配/扩容内存，读取整行输入（含空格）
    readLen = getline(&filePath, &bufSize, stdin);

    if (readLen == -1) {
        perror("Failed to read input");
        free(filePath);
        return 1;
    }

    // 去掉换行符（getline会读取输入末尾的\n）
    if (readLen > 0 && filePath[readLen - 1] == '\n') {
        filePath[readLen - 1] = '\0';
    }

    printf("You entered: %s\n", filePath);

    // 释放getline分配的内存
    free(filePath);
    return 0;
}

如果需要兼容不支持getline的环境，可手动实现动态读取逻辑：

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

int main() {
    char *filePath = malloc(128); // 初始分配128字节缓冲区
    if (!filePath) {
        perror("Failed to allocate memory");
        return 1;
    }
    size_t bufSize = 128;
    size_t currentLen = 0;

    printf("Please provide file to encrypt (File Path): ");

    // 逐字符读取，直到换行或EOF
    int c;
    while ((c = getchar()) != '\n' && c != EOF) {
        if (currentLen + 1 >= bufSize) {
            // 缓冲区扩容（通常翻倍）
            bufSize *= 2;
            char *temp = realloc(filePath, bufSize);
            if (!temp) {
                perror("Failed to reallocate memory");
                free(filePath);
                return 1;
            }
            filePath = temp;
        }
        filePath[currentLen++] = (char)c;
    }
    filePath[currentLen] = '\0'; // 添加字符串结束符

    printf("You entered: %s\n", filePath);

    free(filePath);
    return 0;
}

2. 输入长度未知时的动态内存分配处理逻辑 #

核心原则是初始分配小缓冲区，按需扩容，具体步骤：

• 初始化一个合理大小的缓冲区（如128或256字节），避免初始分配过大内存造成浪费。
• 读取输入时，检查缓冲区剩余空间（需留一个位置给\0）。
• 若缓冲区已满且输入未结束，用realloc扩容（通常将大小翻倍，平衡扩容次数与内存浪费）。
• 读取完成后添加字符串结束符，使用完毕后必须释放内存，避免泄漏。
• 异常处理：内存分配失败时，及时释放已分配内存，防止泄漏。

3. 接受文件路径输入的最安全方式 #

结合上述内容，最安全的方式包含以下几点：

• 读取整行输入：不要用scanf("%s")，它会截断含空格的路径，改用getline或逐字符读取，确保捕获完整路径。
• 动态分配内存：避免固定缓冲区的限制，同时不会浪费内存。
• 清理无效字符：去掉输入末尾的换行符，文件路径中不会包含该字符。
• 验证输入有效性：读取路径后，可通过access或stat函数检查路径是否存在、是否有读写权限，提前发现无效路径。
• 严格管理内存：所有动态分配的内存必须在使用完成后释放，避免内存泄漏。

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