1 files changed, 80 insertions, 0 deletions
diff --git a/src/checktle.c b/src/checktle.c
new file mode 100644
index 0000000..2226551
--- /dev/null
+++ b/src/checktle.c
@@ -0,0 +1,80 @@
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
+
+#include "config2.h"
+#include "common.h"
+#include "core.h"
+
+#include <string.h>
+#include <unistd.h>
+#include <signal.h>
+
+#include <pthread.h>
+#include <semaphore.h>
+
+volatile sig_atomic_t break_flag;
+
+static void break_handler(int signo){
+	break_flag = 1;
+}
+
+void* sctjudge_checktle(void* arg){
+	pid_t pidcopy;
+	long long sleeptime = (long long)(*(int*)arg) * 1000000;
+	struct sigaction break_catch;
+	struct timespec timelimit, timeinit, timecur, timetmp;
+	const struct timespec nanslparg = {0, SCT_CHECKTLE_INTERVAL};
+
+	pthread_mutex_lock(&tkill_mx);
+	tkill_yes = 1;
+	pthread_mutex_unlock(&tkill_mx);
+
+	pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL);
+	pthread_setcanceltype(PTHREAD_CANCEL_ASYNCHRONOUS, NULL);
+
+	sem_wait(&addthr);
+	clock_gettime(CLOCK_REALTIME, &timeinit);
+
+#ifndef HAVE_CONF_CAP
+	enable_setuid();
+#endif
+
+	break_flag = 0;
+	memset(&break_catch, 0, sizeof(break_catch));
+	break_catch.sa_handler = &break_handler;
+	sigaction(SIGINT, &break_catch, NULL);
+	sigaction(SIGTERM, &break_catch, NULL);
+
+	pthread_mutex_lock(&pidmutex);
+	pidcopy = pidchild;
+	pthread_mutex_unlock(&pidmutex);
+
+	timelimit.tv_sec = timeinit.tv_sec + sleeptime / 1000000000;
+	timelimit.tv_nsec = timeinit.tv_nsec + sleeptime % 1000000000;
+
+	for(clock_gettime(CLOCK_REALTIME, &timecur);
+			comparetimespec(&timecur, &timelimit) < 0 && !break_flag;
+			clock_gettime(CLOCK_REALTIME, &timecur)){
+		difftimespec(&timecur, &timelimit, &timetmp);
+		if(comparetimespec(&timetmp, &nanslparg) > 0){
+			nanosleep(&nanslparg, NULL);
+		}else{
+			nanosleep(&timetmp, NULL);
+		}
+	}
+
+	if(!break_flag){
+		pthread_mutex_lock(&judge_tle_mx);
+		judge_tle = 1;
+		pthread_mutex_unlock(&judge_tle_mx);
+	}
+
+	kill(pidcopy, SIGKILL);
+
+	pthread_mutex_lock(&tkill_mx);
+	tkill_yes = 0;
+	pthread_mutex_unlock(&tkill_mx);
+	return NULL;
+}
+