对拍程序

借鉴sandy老哥的对拍程序

https://www.cnblogs.com/sandychn/p/10355239.html

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cerrno>
#include <ctime>

// 设置区
namespace Settings {
    const int MAX_LEN = 20; // 当发现错误时,显示错误行数据的最大长度
    const int CASE_CNT = 100; // 对拍测试次数
    const int BUFFER_SIZE = 1 << 10; // 每行的缓冲区大小
    const char *const userFile = "1.cpp"; // 需要测试的源代码文件名(相对路径)
    const char *const stdFile = "2.cpp"; // 标准答案的源代码文件名(相对路径)
    const char *const dataFile = "3.cpp"; // 随机生成数据的源代码文件名(相对路径)
}

using namespace Settings;
using namespace std;
char cmd[BUFFER_SIZE], info[BUFFER_SIZE * 3];
char buf1[BUFFER_SIZE], buf2[BUFFER_SIZE];

template <typename ...T>
int run(const char *str, T ...args) {
    sprintf(cmd, str, args...);
    return system(cmd);
}

bool fileCompare(const char *stdOutput, const char *userOutput) {
    FILE *fp_std = fopen(stdOutput, "r"), *fp_user = fopen(userOutput, "r");
    if (fp_std == nullptr || fp_user == nullptr) {
        sprintf(info, "Open file failed: %s
", strerror(errno));
        return false;
    }
    bool flag = false;
    for (int i = 1; !feof(fp_std) && !feof(fp_user); ++i) {
        char *p1 = fgets(buf1, BUFFER_SIZE, fp_std);
        char *p2 = fgets(buf2, BUFFER_SIZE, fp_user);
        if (p1 == nullptr && p2 == nullptr) {
            flag = true;
            break;
        } else if (!p1 || !p2 || strcmp(p1, p2)) {
            sprintf(info,
                    "Difference found in line %d:
  std: %.*s user: %.*s",
                    i, MAX_LEN, p1 ? p1 : "(EOF Detected)
",
                    MAX_LEN, p2 ? p2 : "(EOF Detected)
");
            break;
        }
    }
    fclose(fp_std), fclose(fp_user);
    return flag;
}

int main() {
    printf("Compiling... ");
    run("g++ %s -o user.exe", userFile);
    run("g++ %s -o std.exe", stdFile);
    run("g++ %s -o data.exe", dataFile);
    printf("Finished.
");
    clock_t st, ed;
    for (int i = 1; i <= CASE_CNT; ++i) {
        printf("Case %03d: ", i);
        run("data.exe >data.txt");
        run("std.exe <data.txt >std.txt");
        st = clock();
        int ret = run("user.exe <data.txt >user.txt");
        if (ret) {
            printf("Runtime Error - Exit code is %d
", ret);
            break;
        }
        ed = clock();
        if (!fileCompare("std.txt", "user.txt")) {
            printf("Wrong Answer.
------------------------------
");
            puts(info);
            printf("------------------------------
Note: At most %d characters will be shown for both lines.
", MAX_LEN);
            printf("Press anykey to exit.
");
            getchar();
            break;
        } else {
            printf("Accepted. Time: %dms.
", int(ed - st));
        }
    }
    return 0;
}