Description:
box uses user time
Commit status:
[Not Reviewed]
References:
Comments:
0 Commit comments 0 Inline Comments
Unresolved TODOs:
There are no unresolved TODOs
Add another comment

r130:db5312fad511 - - 1 file changed: 26 inserted, 4 deleted

@@ -298,100 +298,122
298 298
299 299 int execute(char *exname, char *inname, char *outname, double t, int max_mem)
300 300 {
301 301 STARTUPINFO si;
302 302 PROCESS_INFORMATION pi;
303 303 int ifsuccess = EXE_RESULT_OK;
304 304
305 305 ZeroMemory(&si, sizeof(si));
306 306 si.cb = sizeof(si);
307 307 ZeroMemory(&pi, sizeof(pi));
308 308
309 309 setstartupinfo(&si, inname, outname);
310 310
311 311 if(!CreateProcess( NULL, // No module name (use command line).
312 312 TEXT(exname), // Command line.
313 313 NULL, // Process handle not inheritable.
314 314 NULL, // Thread handle not inheritable.
315 315 TRUE, // Set handle inheritance to FALSE.
316 316 0, // No creation flags.
317 317 NULL, // Use parent's environment block.
318 318 NULL, // Use parent's starting directory.
319 319 &si, // Pointer to STARTUPINFO structure.
320 320 &pi)) // Pointer to PROCESS_INFORMATION structure.
321 321 {
322 322 //printf( "CreateProcess failed (%d).\n", GetLastError() );
323 323 fprintf(stderr, "Process creation error.\n");
324 324 report_stat(0,0);
325 325 return EXE_RESULT_ERROR;
326 326 }
327 327 //fprintf(stderr,"Process ID: %ld\n",pi.dwProcessId);
328 328 //fprintf(stderr,"time limit = %d\n",t);
329 329
330 330 // checking memory usage
331 331 // wait 0.1 sec before checking mem usage
332 332
333 333 SetProcessWorkingSetSize(pi.hProcess,
334 334 1,
335 335 max_mem);
336 336 int actual_memory_usage = 0;
337 337
338 338 Sleep(INITIAL_WAIT_FOR_MEM_CHECK);
339 339 if(!check_memory_usage(pi.dwProcessId,max_mem,&actual_memory_usage)) {
340 340 // using too much memory
341 341 fprintf(stderr,"Memory limit exceeded.\n");
342 342 //PrintMemoryInfo(pi.dwProcessId);
343 343 ifsuccess = EXE_RESULT_MEMORY;
344 344 }
345 345
346 - if((ifsuccess == EXE_RESULT_MEMORY) ||
347 - (WaitForSingleObject(pi.hProcess,
348 - (int)(t*1000) + 1
349 - - INITIAL_WAIT_FOR_MEM_CHECK)==WAIT_TIMEOUT)) {
346 + //printf("PID: %d\n", pi.dwProcessId);
347 +
348 + if(ifsuccess != EXE_RESULT_MEMORY) {
349 + int based_time = (int)(t*1000) + 1 - INITIAL_WAIT_FOR_MEM_CHECK;
350 + bool major_timed_out = (WaitForSingleObject(pi.hProcess,
351 + based_time)==WAIT_TIMEOUT);
352 + if(major_timed_out) {
353 + // wait some more for user time.
354 + double time_used = get_process_time_usage(pi.hProcess);
355 + while(time_used <= t) {
356 + int iter_time = 100;
357 + if(t - time_used < 200)
358 + iter_time = 20;
359 + bool iter_timed_out = (WaitForSingleObject(pi.hProcess,
360 + iter_time)==WAIT_TIMEOUT);
361 + if(!iter_timed_out)
362 + break;
363 +
364 + time_used = get_process_time_usage(pi.hProcess);
365 + //printf("%lf\n",time_used);
366 + }
367 + ifsuccess = EXE_RESULT_TIMEOUT;
368 + }
369 + }
370 +
371 + if((ifsuccess == EXE_RESULT_MEMORY) || (ifsuccess == EXE_RESULT_TIMEOUT)) {
350 372 // Kill process, because (1) it used too much memory, or (2) time limit
351 373 HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, FALSE, pi.dwProcessId);
352 374
353 375 if(ifsuccess != EXE_RESULT_MEMORY)
354 376 fprintf(stderr,"Time limit exceeded.\n");
355 377 if(hProcess != NULL) {
356 378 fprintf(stderr,"killing pid: %ld\n",pi.dwProcessId);
357 379 TerminateProcess(hProcess, 0);
358 380 wait_dialog();
359 381 } else {
360 382 DWORD dwNtvdmId = get_ntvdm_pid();
361 383 fprintf(stderr,"killing (ntvdm) pid: %ld\n",dwNtvdmId);
362 384 if(dwNtvdmId!=0) {
363 385 hProcess = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwNtvdmId);
364 386 TerminateProcess(hProcess, 0);
365 387 } else {
366 388 fprintf(stderr,"killing process error\n");
367 389 }
368 390
369 391 if(get_ntvdm_pid()!=0) {
370 392 fprintf(stderr,"killing error, ntvdm.exe still remains;");
371 393 fprintf(stderr,"please MANUALLY kill it.");
372 394 fflush(stderr);
373 395 do {
374 396 Sleep(1000);
375 397 } while(get_ntvdm_pid()!=0);
376 398 fprintf(stderr,"... done\n");
377 399 wait_dialog();
378 400 }
379 401 }
380 402 if(ifsuccess != EXE_RESULT_MEMORY)
381 403 ifsuccess = EXE_RESULT_TIMEOUT;
382 404 }
383 405
384 406 // check memory after terminated
385 407 if((ifsuccess==EXE_RESULT_OK) &&
386 408 (!check_memory_usage(pi.dwProcessId,max_mem, &actual_memory_usage))) {
387 409 // using too much memory
388 410 ifsuccess = EXE_RESULT_MEMORY;
389 411 }
390 412
391 413 // check return code
392 414 if(ifsuccess==EXE_RESULT_OK) {
393 415 DWORD exitcode;
394 416 GetExitCodeProcess(pi.hProcess, &exitcode);
395 417 if(exitcode!=0) {
396 418 fprintf(stderr,"Exit status %d.\n", (int)exitcode);
397 419 ifsuccess = EXE_RESULT_ERROR;
You need to be logged in to leave comments. Login now