Description:
box uses user time
Commit status:
[Not Reviewed]
References:
Diff options:
Comments:
0 Commit comments
0 Inline Comments
Unresolved TODOs:
There are no unresolved TODOs
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