深蓝色的空间

Online Judge是一个根据程序的输入输出测试程序正确性的Web程序，一般用于ACM/ICPC的训练。这是我做的测试部分的源代码，主要是限定程序的运行时间和占用的空间，以及程序的输入输出重定向。限时做得很粗糙，仅仅使用了WaitForSingleObject，不能得到真正的CPU时间。

用法：test <progname> <input> <output> <time> <memory>
<progname> 要测试的程序名
<input> 输入数据
<output> 程序输出
<time> 时间限制（单位：ms）
<memory> 存储空间限制（单位：KB）
返回值：正确结束返回0；超时——1，超出存储限制——2。

#include <windows.h>
#include <psapi.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
int main(int argc, char **argv){
	char *app_name=argv[1];
	char *input_file=argv[2];
	char *output_file=argv[3];
	char *time_str=argv[4];
	char *mem_str=argv[5];
	int time_limit=atoi(time_str);
	DWORD mem_limit=atoi(mem_str);
	mem_limit<<=10;
	
	SECURITY_ATTRIBUTES sa;
	sa.nLength=sizeof(SECURITY_ATTRIBUTES);
	sa.lpSecurityDescriptor=NULL;
	sa.bInheritHandle=TRUE;
	
	HANDLE hInput=CreateFile(input_file,GENERIC_READ,0,&sa,OPEN_ALWAYS,FILE_ATTRIBUTE_ARCHIVE,NULL);
	HANDLE hOutput=CreateFile(output_file,GENERIC_WRITE,0,&sa,CREATE_ALWAYS,FILE_ATTRIBUTE_ARCHIVE,NULL);
	
	STARTUPINFO si={sizeof(si)};
	GetStartupInfo(&si);
	
	si.dwFlags=STARTF_USESTDHANDLES;
	si.hStdInput=hInput;
	si.hStdOutput=hOutput;
	
	PROCESS_INFORMATION pi;
	ZeroMemory(&pi,sizeof(pi));
	char *szCommandLine;
	szCommandLine=(char *)calloc(sizeof(char),(strlen(app_name)+1));
	strcpy(szCommandLine,app_name);
	
	CreateProcess(NULL,szCommandLine,NULL,NULL,TRUE,CREATE_NO_WINDOW,NULL,NULL,&si,&pi);
	DWORD status=WaitForSingleObject(pi.hProcess,time_limit);
	PROCESS_MEMORY_COUNTERS memInfo;
	GetProcessMemoryInfo(pi.hProcess,&memInfo,sizeof(memInfo));
	int ret;
	if (status!=WAIT_OBJECT_0){
		ret=1;
		TerminateProcess(pi.hProcess,0);
	}else{
		if (memInfo.PeakWorkingSetSize>mem_limit){
			ret=2;
		}else{
			ret=0;
		}
	}
	CloseHandle(pi.hThread);
	CloseHandle(pi.hProcess);
	CloseHandle(hInput);
	CloseHandle(hOutput);
	return ret;
}