forked from shashankkumar/CodeRunner
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Resolving merge conflicts with sandbox
- Loading branch information
Showing
7 changed files
with
295 additions
and
38 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,250 @@ | ||
| #include "includes.h" | ||
| #include "resources.h" | ||
| #include "config.h" | ||
|
|
||
| pid_t cpid; | ||
| const int MicroSecSleepInterval = 71; | ||
| void ToPipe(const char* str){ | ||
| printf("%s\n", str); | ||
| return; | ||
| } | ||
|
|
||
| void SignalHandler(int signum){ | ||
| if(cpid!=0){ | ||
| killpg(cpid,SIGKILL); | ||
| } | ||
| } | ||
|
|
||
| /* 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 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:"); | ||
| 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; | ||
| } | ||
|
|
||
| return min(data + stack, rss); | ||
| /* return (data + stack); */ | ||
| } | ||
|
|
||
| int main(int args, char *argv[]){ | ||
|
|
||
| if(args<4){ | ||
| ToPipe("IE ERROR Not enough arguments."); | ||
| return 1; | ||
| } | ||
|
|
||
| int TimeLimit, MemoryLimit, TestCaseFileId, FileId; | ||
| char InputFile[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]); | ||
|
|
||
| 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[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 TestCaseFile[10], OutputFile[10]; | ||
| sprintf(TestCaseFile, "%d.txt", TestCaseFileId); | ||
| sprintf(OutputFile, "%do.txt", TestCaseFileId); | ||
| 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"); | ||
| } | ||
| SetResourceLimitValues(TimeLimit); | ||
| 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; | ||
|
|
||
| int MemoryUsed = 0; | ||
|
|
||
| do{ | ||
| MicroSecSleep(MicroSecSleepInterval); | ||
| MemoryUsed = max(MemoryUsed, MemoryUsage(cpid)); | ||
| if(MemoryUsed > MemoryLimit){ | ||
| kill(cpid, SIGKILL); | ||
| } | ||
| w = wait4 (cpid, &status, WUNTRACED | WCONTINUED, &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; | ||
| } | ||
|
|
||
| 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 ; | ||
|
|
||
| bool sigkill = false, sigalrm = false; | ||
|
|
||
| if(MemoryUsed > MemoryLimit){ | ||
| ; | ||
| } | ||
| else if( WIFEXITED(status) == true ){ | ||
| if( WEXITSTATUS(status) !=0 ){ | ||
| ToPipe("RE NZEC"); | ||
| } | ||
| else{ | ||
| ToPipe("AC"); | ||
| } | ||
| } | ||
| else 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); | ||
| sprintf(tmp, "%d", MemoryUsed); | ||
| ToPipe(tmp); | ||
| //printf("%d %d\n", (int)tv.tv_sec, (int)tv.tv_usec); | ||
| if(sigkill) printf("SIGKILL"); | ||
| else if(sigalrm) printf("SIGALRM"); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.