149 lines
3.1 KiB
C++
149 lines
3.1 KiB
C++
#include <stdio.h>
|
||
#include <unistd.h>
|
||
#include <stdlib.h>
|
||
#include <signal.h>
|
||
#include <string.h>
|
||
#include <sys/wait.h>
|
||
#include <time.h>
|
||
//--------------------------------------------------------------------------------------------
|
||
#include "Process_r.h"
|
||
|
||
Process_r task; //задача
|
||
char* coup_de_grace=NULL;
|
||
|
||
void hdl_abort_by_user(int sig)
|
||
{
|
||
if ((coup_de_grace!=NULL)&&strlen(coup_de_grace)!=0)
|
||
system(coup_de_grace);
|
||
FILE * killed = fopen("INTERRUPT","w");
|
||
fprintf(killed,"task was interrupted\n");
|
||
fclose(killed);
|
||
killpg(0, SIGKILL);
|
||
exit(0);
|
||
}
|
||
|
||
void set_handlers(){
|
||
|
||
struct sigaction act;
|
||
|
||
memset(&act, 0, sizeof(act)); //обнуление полей структуры
|
||
act.sa_handler = hdl_abort_by_user; //установка обработчика в поле стурктуры
|
||
|
||
sigset_t set; //маска сигналов, которые буудт блокироваться во время выполнения обработчика
|
||
sigemptyset(&set); //предваритальеное обнуление всех сигналов
|
||
sigaddset(&set, SIGINT);
|
||
act.sa_mask = set;
|
||
|
||
sigaction(SIGINT, &act, 0); //установка обработчика
|
||
|
||
}
|
||
|
||
|
||
//argv[0] - имя программы
|
||
//argv[0] - время
|
||
//argv[1] путь к драйверу dvm
|
||
//argv[2] f/c(компиляция) или run(запуск)
|
||
//argv[3] имя программы
|
||
//argv[4] .. argv [argc-1] либо флаги компиляции, либо измерения решетки по одному
|
||
|
||
|
||
int main(int argc, char ** argv){
|
||
|
||
int i;
|
||
time_t time1;
|
||
time_t time2;
|
||
|
||
// for (i=0; i<argc; ++i){
|
||
// printf(("argv["+String_::ToString(i)+"] = '%s'\n").string,argv[i]);
|
||
|
||
// }
|
||
|
||
// printf("\n");
|
||
char ** args = new char* [argc] ;
|
||
|
||
int maxtime = atoi(argv[0]);
|
||
coup_de_grace =argv[1];
|
||
/*
|
||
printf("\nmaxtime = %d\n",maxtime);
|
||
printf("\ncoup_de_grace = %'s'\n",coup_de_grace);
|
||
printf("dvm_drv = %'s'\n",argv[2]);
|
||
*/
|
||
|
||
|
||
|
||
for ( i=2; i<argc; ++i)
|
||
args [i-2] = argv[i];
|
||
|
||
args[argc-2] = NULL;
|
||
args[argc-1] = NULL;
|
||
/*
|
||
for (i=0; i<argc; ++i){
|
||
printf("args %d = '%s'\n",i,args[i]);
|
||
}
|
||
*/
|
||
|
||
#if 1
|
||
int pid = getpid();
|
||
|
||
task.Start();
|
||
|
||
//потомок - задача
|
||
if (task.pid ==0 )
|
||
execvp (argv[2],args);
|
||
|
||
|
||
time1 = time((time_t *)0);
|
||
|
||
|
||
int s;
|
||
|
||
double total_time = 0.;
|
||
|
||
|
||
|
||
int timer_pid = fork();
|
||
|
||
if (timer_pid ==0){
|
||
|
||
sleep(maxtime);
|
||
|
||
FILE * killed = fopen("TIMEOUT","w");
|
||
fprintf(killed,"task was killed by timer\n");
|
||
fclose(killed);
|
||
|
||
kill(pid,SIGINT);
|
||
|
||
exit(0);
|
||
|
||
}
|
||
|
||
set_handlers();
|
||
|
||
waitpid(task.pid,&s,WUNTRACED);
|
||
kill(timer_pid,SIGKILL);
|
||
|
||
|
||
time2 = time((time_t *)0);
|
||
|
||
total_time = difftime(time2, time1);
|
||
|
||
|
||
FILE * tt = fopen("total_time","w");
|
||
fprintf(tt,"%.5lf",total_time);
|
||
fclose(tt);
|
||
|
||
FILE * done = fopen("DONE","w");
|
||
fprintf(done,"task finished correctly");
|
||
fclose(done);
|
||
|
||
//Добивание. не факт что и сам тест будет корректно написан. от греха.
|
||
if ((coup_de_grace!=NULL)&&strlen(coup_de_grace)!=0)
|
||
system(coup_de_grace);
|
||
killpg(0, SIGKILL);
|
||
#endif
|
||
|
||
return 0;
|
||
}
|
||
|
||
|