int MemoryUsage(pid_t cpid){
char a[10000], *p, *q;
int data, stack;
int rss; /* ZP */
int n, v, fd;
p = a;
sprintf (p, "/proc/%d/status", cpid);
fd = open (p, O_RDONLY);
if (fd < 0)
ToPipe("IE ERROR Failed to open /proc/pid/status") ;
do{
n = read (fd, p, 10000);
} while ((n < 0) && (errno == EINTR));
if (n < 0)
ToPipe("IE ERROR Failed to read /proc/pid/status") ;
do{
v = close (fd);
} while ((v < 0) && (errno == EINTR));
if (v < 0)
ToPipe("IE ERROR Failed to close file descriptor for /proc/pid/status") ;
rss = data = stack = 0;
q = strstr (p, "VmData:");
char tmp[10000];
sprintf(tmp, "%s\n", p);
//ToPipe(tmp);
if (q != NULL)
{
sscanf (q, "%*s %d", &data);
q = strstr (q, "VmStk:");
if (q != NULL)
sscanf (q, "%*s %d\n", &stack);
q = strstr (q, "VmRSS:");
if (q != NULL)
sscanf (q, "%*s %d\n", &rss);
else
rss = data + stack;
}
printf("The memory usage is about to be printed \n");
printf("Memory Usage: data - %d, stack - %d, rss - %d\n", data, stack, rss);
return data + stack;
/* return (data + stack); */
}
/* high resolution (microsecond) sleep */
void MicroSecSleep (int ms){
struct timeval tv;
int w;
do{
tv.tv_sec = ms / 1000;
tv.tv_usec = (ms % 1000) * 1000;
w = select (0, NULL, NULL, NULL, &tv);
/* The value of the timeout is undefined after the select returns */
} while ((w < 0) && (errno == EINTR));
if (w < 0)
ToPipe("IE ERROR Failed while calling select in microsleep");
}
int main(int args, char *argv[]){
if(args<4){
ToPipe("IE ERROR Not enough arguments.");
return 1;
}
int TimeLimit, MemoryLimit, TestCaseFileId, FileId;
char InputFile[25], lang[10];
struct timeval start,finish;
struct sigaction SignalAction;
pid_t w;
int status;
long long t_sec, t_usec;
FileId = atoi(argv[1]), strcpy(InputFile, argv[2]);
TestCaseFileId = atoi(argv[3]), TimeLimit = atoi(argv[4]), MemoryLimit = atoi(argv[5]);
strcpy(lang, argv[6]);
SignalAction.sa_flags = SA_NOCLDSTOP | SA_NOCLDWAIT;
SignalAction.sa_handler = SignalHandler;
if(sigaction(SIGALRM, &SignalAction, NULL)==-1){
ToPipe("IE ERROR Could not register handler for SIGALRM");
return 1;
}
if(sigaction(SIGXCPU, &SignalAction, NULL)==-1){
ToPipe("IE ERROR Could not register handler fo SIGXCPU");
return 1;
}
gettimeofday(&start,NULL);
cpid = fork();
if (cpid == -1) { perror("fork"); exit(EXIT_FAILURE); }
if (cpid == 0) { /* Code executed by child */
pid_t ChildProcessId = getpid();
passwd* UserDetails = getpwnam("nobody");
char dir[25];
sprintf(dir, "%s%d/", FILEPATH, FileId);
//ToPipe(dir);
if( chdir(dir) == -1){
printf("%d", errno);
ToPipe("IE ERROR Cannot change directory to that of file");
return 1;
}
//ToPipe(get_current_dir_name());
/*if(chroot(get_current_dir_name())!=0){
ToPipe("IE ERROR Cannot jail the program");
return 1;
}*/
if(( setpgid( ChildProcessId, ChildProcessId) ) == -1 ){
ToPipe("IE ERROR process group could not be created\n");
return 1;
}
if(UserDetails!=NULL){
gid_t GroupId = UserDetails->pw_gid;
if( GroupId > 0 ){
setgid( GroupId );
}
else{
ToPipe("IE ERROR Could not set groupid as groupid is invalid\n");
return 1;
}
uid_t UserId = UserDetails->pw_uid;
if( UserId > 0 ){
setuid(UserId);
}
else{
ToPipe("IE ERROR Failed to set userid as it is invalid\n");
return 1;
}
}
else{
ToPipe("IE ERROR No user id associated with user nobody\n");
return 1;
}
char TestCaseFile[25], OutputFile[25];
sprintf(TestCaseFile, "%d.txt", TestCaseFileId);
sprintf(OutputFile, "%do.txt", TestCaseFileId);
if(alarm(TimeLimit)!=0){
ToPipe("IE ERROR Could not set alarm.");
}
//printf("Execution aginst timelimit: %d", TimeLimit);
if(strcmp(lang,"java")==0){
SetResourceLimitValuesJava(TimeLimit);
FILE *fpipe;
char command[100];
sprintf(command, "java -Djava.security.manager -Djava.security.policy==./policy.txt -Xms1024M Main < %s > %s", TestCaseFile, OutputFile);
//printf("%s\n", command);
char line[256];
if ( !(fpipe = (FILE*)popen(command,"r")) ){
perror("Problems with pipe");
//Logs::WriteLine("Problems with pipe");
}
else{
if ( fgets( line, sizeof line, fpipe)){
printf("%s\n", line);
}
}
pclose(fpipe);
exit(0);
}
if(freopen(TestCaseFile, "r", stdin)==NULL){
ToPipe("IE ERROR Could not open test case file\n");
}
if(freopen(OutputFile, "w", stdout)==NULL){
ToPipe("IE ERROR Could not open output file\n");
}
if(freopen("/dev/null", "w", stderr)==NULL){
ToPipe("IE ERROR Could not redirect error stream to null\n");
}
/*
if(strcmp(lang,"java")==0){
SetResourceLimitValuesJava(TimeLimit);
if(execl("/usr/bin/java", "/usr/bin/java", "-Xms20M -Xmx500M", "-classpath", InputFile, (char *) NULL) == -1){
fclose(stdout);
ToPipe("IE ERROR File not present or some other error.");
}
}
*/
//printf("execution about to begin");
SetResourceLimitValues(TimeLimit);
if(strcmp(lang, "python")==0){
strcat(InputFile, ".pyc");
if(execl("/usr/bin/python", "/usr/bin/python", InputFile, (char *) NULL) == -1){
fclose(stdout);
ToPipe("IE ERROR File not present or some other error.");
}
}
else if(strcmp(lang, "perl")==0){
strcat(InputFile, ".pl");
if(execl("/usr/bin/perl", "/usr/bin/perl", InputFile, (char *) NULL) == -1){
fclose(stdout);
ToPipe("IE ERROR File not present or some other error.");
}
}
else if(strcmp(lang, "php")==0){
strcat(InputFile, ".php");
printf("%s", InputFile);
if(execl("/usr/bin/php", "/usr/bin/php", InputFile, (char *) NULL) == -1){
fclose(stdout);
ToPipe("IE ERROR File not present or some other error.");
}
}
else if(execl(InputFile, InputFile,(char *) NULL) == -1){
fclose(stdout);
ToPipe("IE ERROR File not present or some other error.");
}
}
else { /* Code executed by parent */
struct rusage resourceUsage;
int MemoryUsed = 64;
do{
MicroSecSleep(MicroSecSleepInterval);
MemoryUsed = max(MemoryUsed, MemoryUsage(cpid));
if(MemoryUsed > MemoryLimit && strcmp(lang, "java")!=0){
kill(cpid, SIGKILL);
}
w = wait4 (cpid, &status, WUNTRACED | WCONTINUED, &resourceUsage);
//w = wait4 (cpid, &status, NULL, &resourceUsage);
} while(w==0);
gettimeofday(&finish,NULL);
t_sec = finish.tv_sec-start.tv_sec;
t_sec=finish.tv_sec-start.tv_sec;
t_usec=finish.tv_usec-start.tv_usec;
if(t_usec<0){
t_sec--;
t_usec+=1000000;
}
float TimeUsed = (float)( t_sec * 1000000 + (int) t_usec ) / 1000000 ;
bool sigkill = false, sigalrm = false;
char executionstatus[4], detailstatus[100];
detailstatus[0] = '\0';
if(MemoryUsed > MemoryLimit && strcmp(lang, "java")!=0){
strcpy(executionstatus, "MLE");
}
else if( WIFEXITED(status) == true ){
if( WEXITSTATUS(status) !=0 ){
strcpy(executionstatus, "RE");
strcpy(detailstatus, "NZEC");
}
else{
strcpy(executionstatus, "AC");
}
}
else if( WIFSIGNALED(status) == true ){
if (WTERMSIG (status) == SIGKILL ){
strcpy(executionstatus, "TLE");
sigkill = true;
}
else if (WTERMSIG(status) == SIGALRM){
strcpy(executionstatus, "TLE");
sigalrm = true;
}
else{
strcpy(executionstatus, "RE");
if (WTERMSIG (status) == SIGXFSZ)
strcpy(detailstatus, "SIGXFSZ");
else if (WTERMSIG (status) == SIGSEGV)
strcpy(detailstatus, "SIGSEGV");
else if (WTERMSIG (status) == SIGFPE)
strcpy(detailstatus, "SIGFPE");
else if (WTERMSIG (status) == SIGABRT)
strcpy(detailstatus, "SIGABRT");
else if (WTERMSIG (status) == SIGHUP){
strcpy(executionstatus, "IE");
strcpy(detailstatus, "SIGHUP");
}
else if (WTERMSIG (status) == SIGPIPE){
strcpy(executionstatus, "IE");
strcpy(detailstatus, "SIGPIPE");
}
else{
strcpy(detailstatus, "OTHER");
printf("%d\n", WTERMSIG(status));
}
}
}
char ExecutionStat[20];
sprintf(ExecutionStat, "status: %s", executionstatus);
ToPipe(ExecutionStat);
sprintf(ExecutionStat, "detailstatus: %s", detailstatus);
ToPipe(ExecutionStat);
sprintf(ExecutionStat, "timeused: %0.4f", TimeUsed);
ToPipe(ExecutionStat);
sprintf(ExecutionStat, "memoryused: %d", MemoryUsed);
ToPipe(ExecutionStat);
//printf("%d %d\n", (int)tv.tv_sec, (int)tv.tv_usec);
if(sigkill) printf("SIGKILL");
else if(sigalrm) printf("SIGALRM");
}
}
int main(int args, char *argv[]){
if(args<4){
ToPipe("IE ERROR Not enough arguments.");
return 1;
}
int TimeLimit, MemoryLimit, InputFileId, FileId;
struct timeval start,finish;
struct sigaction signalAction;
pid_t w;
int status;
long long t_sec, t_usec;
FileId = atoi(argv[1]), InputFileId = atoi(argv[2]), TimeLimit = atoi(argv[3]), MemoryLimit = atoi(argv[4]);
signalAction.sa_flags = SA_NOCLDSTOP | SA_NOCLDWAIT;
signalAction.sa_handler = signalHandler;
if(sigaction(SIGALRM, &signalAction, NULL)==-1){
ToPipe("IE ERROR Could not register handler for sigalrm");
return 1;
}
if(sigaction(SIGXCPU, &signalAction, NULL)==-1){
ToPipe("IE ERROR Could not register handler fo rSIGXCPU");
return 1;
}
gettimeofday(&start,NULL);
cpid = fork();
if (cpid == -1) { perror("fork"); exit(EXIT_FAILURE); }
if (cpid == 0) { /* Code executed by child */
pid_t ChildProcessId = getpid();
passwd* UserDetails = getpwnam("nobody");
char dir[10];
sprintf(dir, "%s%d/", FILEPATH, FileId);
//ToPipe(dir);
if( chdir(dir) == -1){
printf("%d", errno);
ToPipe("IE ERROR Cannot change directory to that of file");
return 1;
}
//ToPipe(get_current_dir_name());
/*if(chroot(get_current_dir_name())!=0){
ToPipe("IE ERROR Cannot jail the program");
return 1;
}*/
if(( setpgid( ChildProcessId, ChildProcessId) ) == -1 ){
ToPipe("IE ERROR process group could not be created\n");
return 1;
}
if(UserDetails!=NULL){
gid_t GroupId = UserDetails->pw_gid;
if( GroupId > 0 ){
setgid( GroupId );
}
else{
ToPipe("IE ERROR Could not set groupid as groupid is invalid\n");
return 1;
}
uid_t UserId = UserDetails->pw_uid;
if( UserId > 0 ){
setuid(UserId);
}
else{
ToPipe("IE ERROR Failed to set userid as it is invalid\n");
return 1;
}
}
else{
ToPipe("IE ERROR No user id associated with user nobody\n");
return 1;
}
char InputFile[10], TestCaseFile[10], OutputFile[10];
sprintf(InputFile, "main");
sprintf(TestCaseFile, "%d.txt", InputFileId);
sprintf(OutputFile, "%do.txt", InputFileId);
if(freopen(TestCaseFile, "r", stdin)==NULL){
ToPipe("IE ERROR Could not open test case file\n");
}
if(freopen(OutputFile, "w", stdout)==NULL){
ToPipe("IE ERROR Could not open output file\n");
}
if(freopen("/dev/null", "w", stderr)==NULL){
ToPipe("IE ERROR Could not redirect error stream to null\n");
}
setResourceLimitWrapper(TimeLimit, MemoryLimit);
alarm(TimeLimit);
if(execl(InputFile,InputFile,(char *) NULL) == -1){
fclose(stdout);
ToPipe("IE ERROR File not present or some other error.");
}
}
else { /* Code executed by parent */
struct rusage resourceUsage;
w = wait4 (cpid, &status, WUNTRACED | WCONTINUED, &resourceUsage);
gettimeofday(&finish,NULL);
t_sec = finish.tv_sec-start.tv_sec;
t_sec=finish.tv_sec-start.tv_sec;
t_usec=finish.tv_usec-start.tv_usec;
if(t_usec<0){
t_sec--;
t_usec+=1000000;
}
timeval tv = resourceUsage.ru_utime;
float SysTime = (float)( tv.tv_sec * 1000000 + (int) tv.tv_usec ) / 1000000 ;
float TimeUsed = (float)( t_sec * 1000000 + (int) t_usec ) / 1000000 ;
if( WIFEXITED(status) == true ){
if( WEXITSTATUS(status) !=0 ){
ToPipe("RE NZEC");
}
else{
ToPipe("AC");
}
}
bool sigkill = false, sigalrm = false;
if( WIFSIGNALED(status) == true ){
if (WTERMSIG (status) == SIGKILL ){
ToPipe("TLE");
sigkill = true;
}
else if (WTERMSIG(status) == SIGALRM){
ToPipe("TLE");
sigalrm = true;
}
else if (WTERMSIG (status) == SIGXFSZ)
ToPipe("RE SIGXFSZ");
else if (WTERMSIG (status) == SIGSEGV)
ToPipe("RE SIGSEGV");
else if (WTERMSIG (status) == SIGFPE)
ToPipe("RE SIGFPE");
else if (WTERMSIG (status) == SIGABRT)
ToPipe("RE SIGABRT");
else if (WTERMSIG (status) == SIGHUP)
ToPipe("IE SIGHUP");
else if (WTERMSIG (status) == SIGPIPE)
ToPipe("IE SIGPIPE");
else{
ToPipe("RE OTHER");
printf("%d\n", WTERMSIG(status));
}
}
char tmp[10];
sprintf(tmp, "%0.4f", TimeUsed);
ToPipe(tmp);
printf("%d %d\n", (int)tv.tv_sec, (int)tv.tv_usec);
if(sigkill) printf("SIGKILL");
else if(sigalrm) printf("SIGALRM");
}
}
