C语言中安全的C++ freopen等价实现咨询及代码评估

你提供的C++竞赛模板代码如下：

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>

using namespace std;

int main() {
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
    int a;
    cin >> a;
    cout << a * 4 << " " << a * a;
}

你自行实现的C语言版本存在安全隐患，同时有冗余操作可优化，具体分析如下：

安全性问题 #

• 未检查文件重定向结果：freopen打开文件失败时会返回NULL，你的代码没处理这种情况。一旦文件路径错误或权限不足，后续scanf、printf会操作失效流，导致程序崩溃或输出异常。
• 未校验输入有效性：scanf("%d", &a)遇到非整数输入会读取失败，此时a值未定义，后续计算结果错误，且错误输入会留在缓冲区引发后续操作异常。
• 多余的fclose(stdout)：程序退出时会自动关闭所有标准流，手动关闭不仅没必要，还可能在freopen失败时（此时stdout仍是控制台流）导致意外资源释放问题。

优化点 #

• 无需存储freopen的返回指针，我们只需要验证重定向是否成功，指针本身对后续操作无意义。
• 去掉手动关闭标准流的操作，简化代码逻辑。

改进后的安全最优实现 #

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

int main() {
    // 检查输入文件重定向
    if (freopen("input.txt", "r", stdin) == NULL) {
        perror("Open input file failed");
        exit(EXIT_FAILURE);
    }
    // 检查输出文件重定向
    if (freopen("output.txt", "w", stdout) == NULL) {
        perror("Open output file failed");
        exit(EXIT_FAILURE);
    }

    int a;
    // 校验整数输入是否成功
    if (scanf("%d", &a) != 1) {
        fprintf(stderr, "Invalid input: need an integer\n");
        exit(EXIT_FAILURE);
    }

    printf("%d %d", a * 4, a * a);
    return 0;
}

这个版本保留了原代码核心逻辑，通过错误检查避免潜在崩溃和异常，去掉冗余操作后更符合竞赛编程的简洁高效要求。

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