update
This commit is contained in:
parent
5215cdc0ad
commit
989c44891d
84
src/main.cpp
84
src/main.cpp
@ -59,10 +59,18 @@ struct Test{
|
||||
}
|
||||
};
|
||||
|
||||
#ifdef _WIN32
|
||||
struct ProgramRet{
|
||||
time_point<steady_clock> start;
|
||||
time_point<steady_clock> end;
|
||||
};
|
||||
void executeCommandWithRedirect(const std::string &command, const std::string &inputFile, const std::string &outputFile);
|
||||
#else
|
||||
struct ProgramRet{
|
||||
time_point<system_clock> start;
|
||||
time_point<system_clock> end;
|
||||
};
|
||||
#endif
|
||||
|
||||
template<typename T>
|
||||
T getValueOrPanic(T value);
|
||||
@ -143,23 +151,33 @@ int main(int argc, char* argv[]) {
|
||||
const string exePathStr = getProtectPath(exePath);
|
||||
for (auto i : tests) {
|
||||
tasks.push([i, &exePathStr, &testsPath]() -> ProgramRet {
|
||||
string commands;
|
||||
std::stringstream ss;
|
||||
ss<<i.id;
|
||||
string id_str;
|
||||
ss>>id_str;
|
||||
string id_str = std::to_string(i.id);
|
||||
path outPath = testsPath / (id_str + ".out");
|
||||
commands = commands+exePathStr+" < "+getProtectPath(i.in)+" > "+getProtectPath(outPath);
|
||||
LOG(commands)
|
||||
|
||||
// 构建命令
|
||||
string commands = exePathStr + " < " + getProtectPath(i.in) + " > " + getProtectPath(outPath);
|
||||
|
||||
// 输出构建的命令进行调试
|
||||
LOG(commands);
|
||||
|
||||
// 执行命令
|
||||
auto start = high_resolution_clock::now();
|
||||
system(commands.c_str());
|
||||
#ifdef _WIN32
|
||||
executeCommandWithRedirect(commands, getProtectPath(i.in), getProtectPath(outPath));
|
||||
#else
|
||||
system(commands.c_str())
|
||||
#endif
|
||||
auto end = high_resolution_clock::now();
|
||||
|
||||
// AS_EM(result, 0,"error in exec commands")
|
||||
|
||||
return ProgramRet{
|
||||
.start = start,
|
||||
.end = end
|
||||
};
|
||||
});
|
||||
}
|
||||
|
||||
while(tasks.size()>0){
|
||||
auto i = tasks.front();
|
||||
tasks.pop();
|
||||
@ -169,10 +187,60 @@ int main(int argc, char* argv[]) {
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef _WIN32
|
||||
|
||||
void executeCommandWithRedirect(const std::string &command, const std::string &inputFile, const std::string &outputFile) {
|
||||
// 创建输入输出文件的句柄
|
||||
HANDLE hInput = CreateFile(inputFile.c_str(), GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
|
||||
if (hInput == INVALID_HANDLE_VALUE) {
|
||||
throw std::runtime_error("Failed to open input file: " + inputFile);
|
||||
}
|
||||
|
||||
HANDLE hOutput = CreateFile(outputFile.c_str(), GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
||||
if (hOutput == INVALID_HANDLE_VALUE) {
|
||||
CloseHandle(hInput);
|
||||
throw std::runtime_error("Failed to open output file: " + outputFile);
|
||||
}
|
||||
|
||||
STARTUPINFO si;
|
||||
PROCESS_INFORMATION pi;
|
||||
ZeroMemory(&si, sizeof(si));
|
||||
si.cb = sizeof(si);
|
||||
si.dwFlags |= STARTF_USESTDHANDLES; // 使用标准句柄
|
||||
si.hStdInput = hInput;
|
||||
si.hStdOutput = hOutput;
|
||||
si.hStdError = GetStdHandle(STD_ERROR_HANDLE); // 重定向错误输出
|
||||
|
||||
ZeroMemory(&pi, sizeof(pi));
|
||||
|
||||
// 将命令转换为 LPSTR 类型
|
||||
LPSTR cmd = const_cast<LPSTR>(command.c_str());
|
||||
|
||||
// 创建进程
|
||||
if (!CreateProcess(NULL, cmd, NULL, NULL, TRUE, 0, NULL, NULL, &si, &pi)) {
|
||||
CloseHandle(hInput);
|
||||
CloseHandle(hOutput);
|
||||
throw std::runtime_error("CreateProcess failed with error: " + std::to_string(GetLastError()));
|
||||
}
|
||||
|
||||
// 等待子进程结束
|
||||
WaitForSingleObject(pi.hProcess, INFINITE);
|
||||
|
||||
// 关闭句柄
|
||||
CloseHandle(pi.hProcess);
|
||||
CloseHandle(pi.hThread);
|
||||
CloseHandle(hInput);
|
||||
CloseHandle(hOutput);
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
|
||||
string getProtectPath(const std::filesystem::path &p){
|
||||
return "\""+p.string()+"\"";
|
||||
}
|
||||
|
||||
|
||||
template<typename T>
|
||||
T getValueOrPanic(T value){
|
||||
AS_NE(value, Json::nullValue)
|
||||
|
Loading…
Reference in New Issue
Block a user