diff options
Diffstat (limited to 'src/SctMain.c')
| -rw-r--r-- | src/SctMain.c | 645 |
1 files changed, 645 insertions, 0 deletions
diff --git a/src/SctMain.c b/src/SctMain.c new file mode 100644 index 0000000..05adc52 --- /dev/null +++ b/src/SctMain.c @@ -0,0 +1,645 @@ +#ifdef HAVE_CONFIG_H +# include "SctConfig.h" +#endif + +#include "SctConst.h" +#include "SctVersion.h" +#include "SctCommon.h" +#include "ProcCommon.h" +#include "JudgeCommon.h" +#include "CliCommon.h" + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <locale.h> +#include <time.h> +#include <errno.h> +#include <sys/time.h> +#include <sys/resource.h> +#include <sys/types.h> + +#include <pthread.h> +#include <semaphore.h> + +#ifdef HAVE_CONF_UGIDNAME +#include <grp.h> +#include <pwd.h> +#endif + +#include <l4darr.h> +#include <l4arg.h> + +#define SCTJUDGE_CLIARG_CHECK_MISSING \ + if(++i >= argc){ \ + fprintf(stderr, "%s: 選項「%s」之後需要一個參數\n", \ + argv[0], argv[i-1]); \ + exit(SCTJUDGE_EXIT_SYNTAX); \ + } + +#define SCTJUDGE_CLIARG_CHECK_QARG_MISSING \ + if(!l4qarg_hasvalue(qarglist[j])){ \ + fprintf(stderr, "%s: 在「%s」之後缺少一個值\n", \ + argv[0], qarglist[j].arg_name); \ + exit(SCTJUDGE_EXIT_SYNTAX); \ + } + +#define SCTJUDGE_CLIARG_CHECK_QARG_EXTRA \ + if(l4qarg_hasvalue(qarglist[j])){ \ + fprintf(stderr, "%s: 在「%s」之後並不需要參數\n", \ + argv[0], qarglist[j].arg_name); \ + exit(SCTJUDGE_EXIT_SYNTAX); \ + } \ + +typedef struct misc_option_flag_and_number { + char* id; + unsigned* num; + unsigned flag; +} MISCOPT; + +int main(int argc, char* argv[]){ + /* 照理說是 zh_TW.utf8，但考慮作業系統差異還是從環境變數抓 + * （反正出錯是使用者的事） */ + setlocale(LC_ALL, ""); + + int i, j, k; + int verbose = 0; /* 0 是安靜模式（機器模式）、1 是 verbose、2 是 debug */ + int ui = SCTJUDGE_UI_CLI; + _Bool parseok = 0; + char* tmpstr; + JUDGEINFO sjopt; /* 無設定檔模式下，傳送這個給 CliMain() */ + L4QARG* qarglist; + +#ifdef HAVE_CONF_UGIDNAME + struct passwd* pwinfo; + struct group* grinfo; +#endif + + save_uids(); + +#ifndef HAVE_CONF_CAP + /* 即使有 setuid root，還是不能讓每個使用者拿到 root 權限 + * 所以說只有在 fork 的時候才要把這個權限開起來，其他就關掉吧 */ + disable_setuid(); +#endif + + /* 定義 -m 時的選項清單 + * 當然有些特殊的像是 errfile 和 uid/gid 以及 arg 這裡沒辦法 */ + const MISCOPT miscopt[] = { + {"time", &(sjopt.procinfo.i_limit_time), 0}, + {"timelimit", &(sjopt.procinfo.i_limit_time), 0}, + {"time-limit", &(sjopt.procinfo.i_limit_time), 0}, + {"tle", &(sjopt.procinfo.i_limit_time), 0}, + + {"stderr", NULL, PROCINFO_FLAG_REDIR_STDERR}, + {"redir-stderr", NULL, PROCINFO_FLAG_REDIR_STDERR}, + + {"secure", NULL, PROCINFO_FLAG_SECURE_CHECK}, + {"secure-check", NULL, PROCINFO_FLAG_SECURE_CHECK}, + + {"mem", &(sjopt.procinfo.i_limit_mem), PROCINFO_FLAG_LIMIT_MEM}, + {"memory", &(sjopt.procinfo.i_limit_mem), PROCINFO_FLAG_LIMIT_MEM}, + {"ram", &(sjopt.procinfo.i_limit_mem), PROCINFO_FLAG_LIMIT_MEM}, + {"as", &(sjopt.procinfo.i_limit_mem), PROCINFO_FLAG_LIMIT_MEM}, + {"addr-space", &(sjopt.procinfo.i_limit_mem), PROCINFO_FLAG_LIMIT_MEM}, + {"address-space", &(sjopt.procinfo.i_limit_mem), PROCINFO_FLAG_LIMIT_MEM}, + {"vm", &(sjopt.procinfo.i_limit_mem), PROCINFO_FLAG_LIMIT_MEM}, + {"virtual-memory", &(sjopt.procinfo.i_limit_mem), PROCINFO_FLAG_LIMIT_MEM}, + + {"core", &(sjopt.procinfo.i_limit_core), PROCINFO_FLAG_LIMIT_CORE}, + {"coredump", &(sjopt.procinfo.i_limit_core), PROCINFO_FLAG_LIMIT_CORE}, + {"corefile", &(sjopt.procinfo.i_limit_core), PROCINFO_FLAG_LIMIT_CORE}, + + {"cputime", &(sjopt.procinfo.i_limit_cpu), PROCINFO_FLAG_LIMIT_CPU}, + {"cputime-limit", &(sjopt.procinfo.i_limit_cpu), PROCINFO_FLAG_LIMIT_CPU}, + + {"outlimit", &(sjopt.procinfo.i_limit_out), PROCINFO_FLAG_LIMIT_OUT}, + {"out-limit", &(sjopt.procinfo.i_limit_out), PROCINFO_FLAG_LIMIT_OUT}, + {"output-limit", &(sjopt.procinfo.i_limit_out), PROCINFO_FLAG_LIMIT_OUT}, + + {"openfile", &(sjopt.procinfo.i_limit_file), PROCINFO_FLAG_LIMIT_FILE}, + {"fdlimit", &(sjopt.procinfo.i_limit_file), PROCINFO_FLAG_LIMIT_FILE}, + {"fdmax", &(sjopt.procinfo.i_limit_file), PROCINFO_FLAG_LIMIT_FILE}, + {"filelimit", &(sjopt.procinfo.i_limit_file), PROCINFO_FLAG_LIMIT_FILE}, + {"file-limit", &(sjopt.procinfo.i_limit_file), PROCINFO_FLAG_LIMIT_FILE}, + + {"process", &(sjopt.procinfo.i_limit_proc), PROCINFO_FLAG_LIMIT_PROC}, + {"proc", &(sjopt.procinfo.i_limit_proc), PROCINFO_FLAG_LIMIT_PROC}, + {"proc-limit", &(sjopt.procinfo.i_limit_proc), PROCINFO_FLAG_LIMIT_PROC}, + + {"stack", &(sjopt.procinfo.i_limit_stack), PROCINFO_FLAG_LIMIT_STACK}, + {"stack-size", &(sjopt.procinfo.i_limit_stack), PROCINFO_FLAG_LIMIT_STACK}, + {"stack-limit", &(sjopt.procinfo.i_limit_stack), PROCINFO_FLAG_LIMIT_STACK} + + }; + + + + /* 將無設定檔模式的結構設定為預設值 */ + sctjudge_setdefault(&sjopt); + sjopt.procinfo.i_argv = l4da_create(sizeof(char*), 2); + tmpstr = NULL; + l4da_pushback(sjopt.procinfo.i_argv, &tmpstr); + + /* 解析命令列的時間到了 + * 為了方便與安全起見，所有 sjopt 裡面的字串都會用 strdup() 複製一次 */ + for(i=1; i<argc; i++){ + if(argv[i][0] == '-'){ + if(!strcmp(&argv[i][1], "help") || !strcmp(&argv[i][1], "-help")){ + printf(SCTJUDGE_TITLEBAR"\n\n" + "用法：%s [選項]\n" + "選項：\n\n" + " -v/-verbose\n" + " 顯示更多訊息（第二個 -v 可以顯示所有重要指令執行過" + "程）\n\n" + " -V/version\n" + " 顯示版本資訊並查看可使用的額外功能列表\n\n" + " -n/-dryrun\n" + " 只列出設定值而不要執行（包含 -v）\n\n" + " -f/-force\n" + " 忽略警告訊息並強制執行\n\n" + " -t/-time <時間>\n" + " 受測程式時間限制為<時間>毫秒\n\n" + " -e/-exec <檔案>\n" + " 指定受測程式可執行檔名稱（務必靜態連結！）\n\n" + " -i/-input <檔案>\n" + " 指定要導向至受測程式標準輸入的檔案，若未指定則為 " + WITH_NULL "\n\n" + " -o/-output <檔案>\n" + " 指定要導向至受測程式標準輸出的檔案，若未指定則為 " + WITH_NULL "\n\n" + " -r/-root <目錄>\n" + " 受測程式將以<目錄>為根目錄執行，若無此選項則關" + "閉 chroot 功能\n\n" + " -m/-misc 選項1[=值],選項2[=值],選項3[=值]...\n" + " 設定額外參數：\n\n" + " mem=<大小>\n" + " 設定受測程式記憶體上限為<大小>位元組\n\n" + " outlimit=<大小>\n" + " 受測程式最多只能輸出<大小>位元組，若未指定則不" + "限制\n" + " （無限制時請確定有足夠的磁碟空間！）\n\n" + " stderr\n" + " 將受測程式的標準錯誤也導向至輸出檔。若未指定，" + "則只有標準輸\n" + " 出會寫入輸出檔案，標準錯誤則被導向至 " + WITH_NULL "\n\n" + " nocopy\n" + " 如果啟用了 chroot 功能，預設情況下本程式會自動" + "將檔案複製\n" + " 到新的根目錄中，並在結束時自動刪除檔案。\n" + " 使用此選項則取消自動複製的功能，但請注意：此時" + " file 指定的\n" + " 檔案名稱是相對於新的根目錄而不是現在的根目錄！" + "\n\n" + " uid=<UID>\n" + " 受測程式將以 UID 為 <UID> 的使用者身份執行\n\n" + " gid=<GID>\n" + " 受測程式將以 GID 為 <GID> 的群組身份執行\n" + " 此選項會同時設定 real/effective/supplementary " + "GID(s)\n" + " 原有的 supplementry GIDs 會被清空，只剩下這裡指" + "定的數值\n\n" + , argv[0]); + return SCTJUDGE_EXIT_SUCCESS; + + }else if(!strcmp(&argv[i][1], "V") || + !strcmp(&argv[i][1], "version") || + !strcmp(&argv[i][1], "-version")){ + fputs(SCTJUDGE_ABOUT_STRING, stdout); + exit(SCTJUDGE_EXIT_SUCCESS); + + }else if(!strcmp(&argv[i][1], "v") || + !strcmp(&argv[i][1], "verbose")){ + verbose++; + sjopt.procinfo.i_verbose = verbose; + + }else if(!strcmp(&argv[i][1], "d") || + !strcmp(&argv[i][1], "debug")){ + verbose += 2; + sjopt.procinfo.i_verbose = verbose; + + }else if(!strcmp(&argv[i][1], "q") || + !strcmp(&argv[i][1], "quiet") || + !strcmp(&argv[i][1], "silent")){ + verbose = 0; + sjopt.procinfo.i_verbose = verbose; + + }else if(!strcmp(&argv[i][1], "n") || + !strcmp(&argv[i][1], "dryrun")){ + verbose = (verbose < 1) ? 1 : verbose; + sjopt.procinfo.i_flag |= PROCINFO_FLAG_DRYRUN; + + }else if(!strcmp(&argv[i][1], "f") || + !strcmp(&argv[i][1], "force")){ + sjopt.procinfo.i_flag |= PROCINFO_FLAG_FORCE; + + }else if(!strcmp(&argv[i][1], "t") || + !strcmp(&argv[i][1], "time")){ + SCTJUDGE_CLIARG_CHECK_MISSING; + if(str_to_u(argv[i], &sjopt.procinfo.i_limit_time) < 0){ + fprintf(stderr, "%s: 「%s」不是正確的時間設定值\n", + argv[0], argv[i]); + exit(SCTJUDGE_EXIT_SYNTAX); + } + + }else if(!strcmp(&argv[i][1], "e") || + !strcmp(&argv[i][1], "exec")){ + SCTJUDGE_CLIARG_CHECK_MISSING; + FREE_IF_NOT_NULL(sjopt.procinfo.i_exec); + sjopt.procinfo.i_exec = strdup(argv[i]); + + }else if(!strcmp(&argv[i][1], "i") || + !strcmp(&argv[i][1], "in") || + !strcmp(&argv[i][1], "input")){ + SCTJUDGE_CLIARG_CHECK_MISSING; + FREE_IF_NOT_NULL(sjopt.procinfo.i_infile); + sjopt.procinfo.i_infile = strdup(argv[i]); + + }else if(!strcmp(&argv[i][1], "o") || + !strcmp(&argv[i][1], "out") || + !strcmp(&argv[i][1], "output")){ + SCTJUDGE_CLIARG_CHECK_MISSING; + FREE_IF_NOT_NULL(sjopt.procinfo.i_outfile); + sjopt.procinfo.i_outfile = strdup(argv[i]); + + }else if(!strcmp(&argv[i][1], "r") || + !strcmp(&argv[i][1], "root") || + !strcmp(&argv[i][1], "chroot")){ + SCTJUDGE_CLIARG_CHECK_MISSING; + FREE_IF_NOT_NULL(sjopt.procinfo.i_chroot); + sjopt.procinfo.i_chroot = strdup(argv[i]); + + }else if(!strcmp(&argv[i][1], "u") || + !strcmp(&argv[i][1], "ui") || + !strcmp(&argv[i][1], "interface")){ + SCTJUDGE_CLIARG_CHECK_MISSING; + if(!strcmp(argv[i], "c") || + !strcmp(argv[i], "cli") || + !strcmp(argv[i], "cmd") || + !strcmp(argv[i], "cmdln") || + !strcmp(argv[i], "command") || + !strcmp(argv[i], "commandline")){ + ui = SCTJUDGE_UI_CLI; + }else if(!strcmp(argv[i], "t") || + !strcmp(argv[i], "tui") || + !strcmp(argv[i], "text") || + !strcmp(argv[i], "term") || + !strcmp(argv[i], "terminal") || + !strcmp(argv[i], "curses") || + !strcmp(argv[i], "ncurses")){ + ui = SCTJUDGE_UI_TUI; + }else if(!strcmp(argv[i], "g") || + !strcmp(argv[i], "gui") || + !strcmp(argv[i], "graph") || + !strcmp(argv[i], "graphic") || + !strcmp(argv[i], "graphics") || + !strcmp(argv[i], "graphical") || + !strcmp(argv[i], "x") || + !strcmp(argv[i], "xorg") || + !strcmp(argv[i], "xwin") || + !strcmp(argv[i], "xwindow")){ + ui = SCTJUDGE_UI_GUI; + }else{ + fprintf(stderr, "%s: 不明的使用者界面「%s」\n", + argv[0], argv[i]); + exit(SCTJUDGE_EXIT_SYNTAX); + } + + }else if(!strcmp(&argv[i][1], "m") || + !strcmp(&argv[i][1], "misc") || + !strcmp(&argv[i][1], "opt") || + !strcmp(&argv[i][1], "option")){ + SCTJUDGE_CLIARG_CHECK_MISSING; + qarglist = l4qarg_parse(argv[i]); + for(j=0; !l4qarg_end(qarglist[j]); j++){ + parseok = 0; + for(k=0; k < sizeof(miscopt) / sizeof(MISCOPT); k++){ + if(!strcmp(miscopt[k].id, qarglist[j].arg_name)){ + if(miscopt[k].num == NULL){ + SCTJUDGE_CLIARG_CHECK_QARG_EXTRA; + sjopt.procinfo.i_flag |= miscopt[k].flag; + parseok = 1; + break; + }else{ + SCTJUDGE_CLIARG_CHECK_QARG_MISSING; + sjopt.procinfo.i_flag |= miscopt[k].flag; + if(str_to_u( + qarglist[j].arg_value, miscopt[k].num) < 0) + { + fprintf(stderr, "%s: 「%s」並不是整數\n", + argv[0], qarglist[j].arg_value); + exit(SCTJUDGE_EXIT_SYNTAX); + } + parseok = 1; + break; + } + } + + /* 這裡是有加上 no 的選項 */ + if(strlen(qarglist[j].arg_name) >= 2 && + !strncmp(qarglist[j].arg_name, "no", 2) && + !strcmp((qarglist[j].arg_name)+2, miscopt[k].id)){ + SCTJUDGE_CLIARG_CHECK_QARG_EXTRA; + sjopt.procinfo.i_flag &= ~(miscopt[k].flag); + parseok = 1; + break; + } + + } + + if(parseok){ + continue; + } + + /* copy/nocopy 設在 JUDGEINFO 但不在 PROCINFO */ + if(!strcmp(qarglist[j].arg_name, "copy")){ + SCTJUDGE_CLIARG_CHECK_QARG_EXTRA; + sjopt.flag |= JUDGEINFO_FLAG_COPY; + }else if(!strcmp(qarglist[j].arg_name, "nocopy")){ + SCTJUDGE_CLIARG_CHECK_QARG_EXTRA; + sjopt.flag &= ~(JUDGEINFO_FLAG_COPY); + } + + /* 修改 uid / gid 的功能 */ + else if(!strcmp(qarglist[j].arg_name, "uid") || + !strcmp(qarglist[j].arg_name, "user")){ + SCTJUDGE_CLIARG_CHECK_QARG_MISSING; + sjopt.procinfo.i_flag |= PROCINFO_FLAG_SETUID; + if(str_to_u( + qarglist[j].arg_value, &sjopt.procinfo.i_uid) < 0){ +#ifdef HAVE_CONF_UGIDNAME + pwinfo = getpwnam(qarglist[j].arg_value); + if(pwinfo == NULL){ + fprintf(stderr, "%s: 「%s」不是正確的 UID 或" + "使用者名稱\n", + argv[0], qarglist[j].arg_value); + exit(SCTJUDGE_EXIT_SYNTAX); + } + sjopt.procinfo.i_uid = pwinfo->pw_uid; +#else + fprintf(stderr, "%s: 「%s」並不是整數\n", + argv[0], qarglist[j].arg_value); + exit(SCTJUDGE_EXIT_SYNTAX); +#endif + } + }else if(!strcmp(qarglist[j].arg_name, "gid") || + !strcmp(qarglist[j].arg_name, "group")){ + SCTJUDGE_CLIARG_CHECK_QARG_MISSING; + sjopt.procinfo.i_flag |= PROCINFO_FLAG_SETGID; + if(str_to_u( + qarglist[j].arg_value, &sjopt.procinfo.i_gid) < 0){ +#ifdef HAVE_CONF_UGIDNAME + grinfo = getgrnam(qarglist[j].arg_value); + if(grinfo == NULL){ + fprintf(stderr, "%s: 「%s」不是正確的 GID 或" + "群組名稱\n", + argv[0], qarglist[j].arg_value); + exit(SCTJUDGE_EXIT_SYNTAX); + } + sjopt.procinfo.i_gid = grinfo->gr_gid; +#else + fprintf(stderr, "%s: 「%s」並不是整數\n", + argv[0], qarglist[j].arg_value); + exit(SCTJUDGE_EXIT_SYNTAX); +#endif + } + } + + /* stderr 也可以指定個別的檔案 */ + else if(!strcmp(qarglist[j].arg_name, "err") || + !strcmp(qarglist[j].arg_name, "errfile")){ + SCTJUDGE_CLIARG_CHECK_QARG_MISSING; + FREE_IF_NOT_NULL(sjopt.procinfo.i_errfile); + sjopt.procinfo.i_errfile = + strdup(qarglist[j].arg_value); + } + + /* 可以指定程式執行時的參數 */ + else if(!strcmp(qarglist[j].arg_name, "arg") || + !strcmp(qarglist[j].arg_name, "argv") || + !strcmp(qarglist[j].arg_name, "addarg") || + !strcmp(qarglist[j].arg_name, "appendarg")){ + SCTJUDGE_CLIARG_CHECK_QARG_MISSING; + tmpstr = strdup(qarglist[j].arg_value); + l4da_pushback(sjopt.procinfo.i_argv, &tmpstr); + }else if(!strcmp(qarglist[j].arg_name, "arg0") || + !strcmp(qarglist[j].arg_name, "argv0")){ + SCTJUDGE_CLIARG_CHECK_QARG_MISSING; + tmpstr = strdup(qarglist[j].arg_value); + l4da_v(sjopt.procinfo.i_argv, char*, 0) = tmpstr; + } + else{ + fprintf(stderr, "%s: 「%s」是不明的選項\n", argv[0], + qarglist[j].arg_name); + exit(SCTJUDGE_EXIT_SYNTAX); + } + } + l4qarg_free(qarglist); + }else{ + fprintf(stderr, "%s: 不明的選項「%s」\n" + "請嘗試執行 %s -help 來取得說明\n" + , argv[0], argv[i], argv[0]); + return SCTJUDGE_EXIT_SYNTAX; + } + }else{ + fprintf(stderr, "%s: 抱歉，設定檔功能尚未實作\n", argv[0]); + return SCTJUDGE_EXIT_NA; + } + } + + if(l4da_v(sjopt.procinfo.i_argv, char*, 0) == NULL){ + /* 未明確指定 argv[0] 則直接將執行檔名稱複製過去 */ + if(sjopt.procinfo.i_exec != NULL){ + l4da_v(sjopt.procinfo.i_argv, char*, 0) = + strdup(sjopt.procinfo.i_exec); + } + } + + /* argv 尾端必須是 NULL */ + tmpstr = NULL; + l4da_pushback(sjopt.procinfo.i_argv, &tmpstr); + + /* 至此可以進入主程式了 */ + if(verbose >= 1){ + puts(SCTJUDGE_TITLEBAR"\n"); + } + + switch(ui){ + case SCTJUDGE_UI_CLI: + if(verbose >= 2){ + printf("DEBUG: %s: Call sctcli_main() [Entering CLI mode]\n", + __func__); + } + sctcli_main(NULL, &sjopt); + break; + + case SCTJUDGE_UI_TUI: + fprintf(stderr, "%s: 抱歉，curses 界面尚未實作\n", argv[0]); + exit(SCTJUDGE_EXIT_NA); + break; + + case SCTJUDGE_UI_GUI: + fprintf(stderr, "%s: 抱歉，圖形化使用者界面尚未實作\n", argv[0]); + exit(SCTJUDGE_EXIT_NA); + break; + + default: + fprintf(stderr, "%s: internal error: invalid ui %d\n", + argv[0], ui); + exit(SCTJUDGE_EXIT_NA); + } + + sctjudge_freestring(&sjopt); + + /********************************************************************** + * 以下都要清掉！！！ + * ********************************************************************/ + +#if 0 + /* 檢查與修正設定值 */ + if(verbose){ + puts("正在檢查設定值是否正確......"); + } + if(mcopt.executable == NULL){ + fputs("受測程式可執行檔名稱為必要參數，不可為空白\n", stderr); + exit(SCTEXIT_TOOFEW); + } + if(mcopt.inputfile == NULL){ + mcopt.inputfile = WITH_NULL; + } + if(mcopt.outputfile == NULL){ + fputs("輸出檔案名稱必要參數，不可為空白\n", stderr); + exit(SCTEXIT_TOOFEW); + } + if(mcopt.exectime <= 0){ + fputs("執行時間限制為必要參數，不可為空白！\n", stderr); + exit(SCTEXIT_TOOFEW); + } + if((mcopt.flags & SCTMC_SETUID)){ + if(mcopt.uid == 0){ + if(!force){ + fputs("將 UID 設為 0 並不安全！（加上 -f 來強制執行）\n", + stderr); + exit(SCTEXIT_BADID); + }else{ + if(procrealuid != 0){ + fputs("只有 root 可以將 UID 設定 0\n", stderr); + exit(SCTEXIT_BADID); + } + } + } + }else{ + if(procrealuid == 0 && !force){ + fputs("不允許以 root 身份執行本程式（加上 -f 來強制執行）\n", + stderr); + exit(SCTEXIT_BADID); + } + } + + /* 現在開始建立 thread 了！*/ + pthread_t tmain; /* 其他 thread 的 ID 放在全域變數 */ + struct makechildrval* mtreturn; /* 接收 sctjudge_makechild 的回傳值 */ + + /* 傳給另外兩個 thread 的參數 */ + int exectimelimit = mcopt.exectime; + /* XXX 抱歉 hard coding，我以後會改進 */ + long displayinterval = SCT_DISPTIME_INTERVAL; + + pthread_attr_t detstate, joistate; + /* 除了主要的 sctjudge_makechild 以外， + * 我想其他的 thread 沒有 join 的必要 */ + pthread_attr_init(&detstate); + pthread_attr_setdetachstate(&detstate, PTHREAD_CREATE_DETACHED); + pthread_attr_init(&joistate); + pthread_attr_setdetachstate(&joistate, PTHREAD_CREATE_JOINABLE); + pthread_mutex_init(&pidmutex, NULL); + pthread_mutex_init(&tkill_mx, NULL); + pthread_mutex_init(&tdisplay_mx, NULL); + pthread_mutex_init(&judge_tle_mx, NULL); + sem_init(&mcthr, 0, 0); + sem_init(&tlethr, 0, 0); + sem_init(&dispthr, 0, 0); + pthread_create(&tmain, &joistate, &sctjudge_makechild, (void*)&mcopt); + if(!dryrun){ + pthread_create(&tkill, &detstate, &sctjudge_checktle, + (void*)&exectimelimit); + } + if(!dryrun && verbose){ + pthread_create(&tdisplay, &detstate, &sctjudge_displaytime, + (void*)&displayinterval); + } + pthread_attr_destroy(&detstate); + pthread_attr_destroy(&joistate); + sem_post(&mcthr); + + pthread_join(tmain, (void**)&mtreturn); + /* XXX 底下這訊息很有可能讓使用者覺得很奇怪 */ + pthread_mutex_destroy(&pidmutex); + pthread_mutex_destroy(&tkill_mx); + pthread_mutex_destroy(&tdisplay_mx); + + sem_destroy(&mcthr); + sem_destroy(&tlethr); + sem_destroy(&dispthr); + + if(dryrun){ + return SCTEXIT_SUCCESS; + } + + /* 這裡就要顯示結果了 */ + if(mtreturn == NULL || mtreturn->mc_exitcode != 0){ + fprintf(stderr, "%s: 因錯誤發生而結束程式\n", argv[0]); + free((void*)mtreturn); + sem_destroy(&mcthr); + exit(SCTEXIT_THREAD); + }else{ + if(verbose){ + putchar('\r'); + puts("======================================================="); + fputs("評測結果：", stdout); + if(mtreturn->judge_result < 0 || + mtreturn->judge_result >= SCTRES_UD){ + mtreturn->judge_result = SCTRES_UD; + } + printf("%s (%s)\n", + sctres_text[0][mtreturn->judge_result], + sctres_text[1][mtreturn->judge_result]); + printf("執行時間：%ld.%03ld 秒\n", + mtreturn->judge_time.tv_sec, + mtreturn->judge_time.tv_nsec / 1000000); + printf("程式離開狀態：%d\n", mtreturn->judge_exitcode); + if(mtreturn->judge_result == SCTRES_RE){ + printf("程式因訊號 %d 而終止\n", mtreturn->judge_signal); + } + puts("-------------------------------------------------------"); + printf("User CPU time: %ld.%03ld\n" + "System CPU time: %ld.%03ld\n", + mtreturn->judge_rusage.ru_utime.tv_sec, + mtreturn->judge_rusage.ru_utime.tv_usec / 1000, + mtreturn->judge_rusage.ru_stime.tv_sec, + mtreturn->judge_rusage.ru_stime.tv_usec / 1000); + puts("======================================================="); + }else{ + /* 輸出格式： + * 第一行是結果代碼、時間、離開狀態 + * 第二行是收到的訊號 (只在 RE 時會有) */ + if(mtreturn->judge_result < 0 || + mtreturn->judge_result >= SCTRES_UD){ + mtreturn->judge_result = SCTRES_UD; + } + fputs(sctres_text[0][mtreturn->judge_result], stdout); + putchar(' '); + printf("%ld%03ld %d\n", mtreturn->judge_time.tv_sec, + mtreturn->judge_time.tv_nsec / 1000000, + mtreturn->judge_exitcode); + if(mtreturn->judge_result == SCTRES_RE){ + printf("%d\n", mtreturn->judge_signal); + } + } + free((void*)mtreturn); + } + +#endif + return SCTJUDGE_EXIT_SUCCESS; +} |