#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;
}