1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include "config2.h"
#include "common.h"
#include "sctcore.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;
}
|
