#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include "config2.h"
#include "version.h"
#include "common.h"
#include "sctcore.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <locale.h>
#include <time.h>
#include <grp.h>
#include <pwd.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <sys/types.h>
#include <pthread.h>
#include <semaphore.h>
#include <l4darr.h>
#include <l4arg.h>
#define SCTMAIN_ERRNOARG "%s: 選項「%s」之後需要一個參數\n"
#define SCTMAIN_ERRNOQARG "%s: 在「%s」之後缺少一個值\n"
#define SCTMAIN_ERREXTRAQARG "%s: 在「%s」之後並不需要參數\n"
#define SCTMAIN_CHECKARGEXIST \
do{ \
i++; \
if(i >= argc){ \
fprintf(stderr, SCTMAIN_ERRNOARG, argv[0], argv[i-1]); \
exit(SCTEXIT_SYNTAX); \
} \
}while(0)
#define SCTMAIN_CHECKQARGHAVEVALUE \
do{ \
if(!l4qarg_hasvalue(qarglist[j])){ \
fprintf(stderr, SCTMAIN_ERRNOQARG, \
argv[0], qarglist[j].arg_name); \
exit(SCTEXIT_SYNTAX); \
} \
}while(0)
#define SCTMAIN_CHECKQARGNOVALUE \
do{ \
if(l4qarg_hasvalue(qarglist[j])){ \
fprintf(stderr, SCTMAIN_ERREXTRAQARG, \
argv[0], qarglist[j].arg_name); \
exit(SCTEXIT_SYNTAX); \
} \
}while(0)
static const char* sctres_text[2][SCTRES_MAX] = {
{"OK", "RE", "TLE", "OLE", "SLE", "AB", "UD"},
{
"正常結束",
"執行過程中發生錯誤",
"超過時間限制",
"輸出超過限制",
"暫停次數太多",
"使用者中斷操作",
"未定義的結果"
}
};
int main(int argc, char* argv[]){
/* 照理說是 zh_TW.utf8，但考慮作業系統差異還是從環境變數抓
* （反正出錯是使用者的事） */
setlocale(LC_ALL, "");
int i, j;
struct makechildopt mcopt; /* 送給 sctjudge_makechild 當參數用 */
char verbose=0; /* 0 是安靜模式（機器模式）、1 是 verbose、2 是 debug */
char dryrun=0, force=0;
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
/* 預設值當然都是 0 */
memset(&mcopt, 0 ,sizeof(mcopt));
/* 據說有些 NULL 不是 0，所以還是設定一下比較保險 */
mcopt.executable = NULL;
mcopt.chrootdir = NULL;
mcopt.inputfile = NULL;
mcopt.outputfile = NULL;
/* 解析命令列的時間到了 */
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"
" 指定要導向至受測程式標準輸入的檔案，若未指定則為 "
NULL_DEVICE"\n\n"
" -o/-output <檔案>\n"
" 指定要導向至受測程式標準輸出的檔案，若未指定則為 "
NULL_DEVICE"\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"
" 出會寫入輸出檔案，標準錯誤則被導向至 "
NULL_DEVICE"\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 SCTEXIT_SUCCESS;
}else if(!strcmp(&argv[i][1], "V") ||
!strcmp(&argv[i][1], "version") ||
!strcmp(&argv[i][1], "-version")){
puts(SCTJUDGE_TITLEBAR);
putchar('\n');
printf("目前可用(+)和不可用(-)的功能列表：\n"
#ifdef HAVE_CONF_PROCMON
" + "
#else
" - "
#endif
"使用 Linux 的 proc 檔案系統來增強程序監視器的功能 (--"
#ifdef HAVE_CONF_PROCMON
"enable"
#else
"disable"
#endif
"-procmon)\n"
#ifdef HAVE_CONF_CAP
" + "
#else
" - "
#endif
"支援 Linux capabilities (--"
#ifdef HAVE_CONF_CAP
"enable"
#else
"disable"
#endif
"-cap)\n"
#ifdef HAVE_CONF_UGIDNAME
" + "
#else
" - "
#endif
"自動轉換使用者或群組名稱成 UID 或 GID "
"(自動偵測)\n"
);
putchar('\n');
printf("編譯時指定的參數列表：\n"
#ifdef NULL_DEVICE
" * 用於捨棄資料的裝置檔案："
NULL_DEVICE
" (--with-null)\n"
#endif
#ifdef PROC_PATH
" * Linux 的 proc 檔案系統路徑："
PROC_PATH
" (--with-proc)\n"
#endif
);
exit(SCTEXIT_SUCCESS);
}else if(!strcmp(&argv[i][1], "v") ||
!strcmp(&argv[i][1], "verbose")){
verbose++;
mcopt.flags |= SCTMC_VERBOSE;
if(verbose >= 2){
mcopt.flags |= SCTMC_DEBUG;
}
}else if(!strcmp(&argv[i][1], "n") ||
!strcmp(&argv[i][1], "dryrun")){
dryrun = 1, verbose = 1;
mcopt.flags |= (SCTMC_DRYRUN | SCTMC_VERBOSE);
}else if(!strcmp(&argv[i][1], "f") ||
!strcmp(&argv[i][1], "force")){
force = 1;
mcopt.flags |= (SCTMC_FORCE);
}else if(!strcmp(&argv[i][1], "t") ||
!strcmp(&argv[i][1], "time")){
SCTMAIN_CHECKARGEXIST;
if(sscanf(argv[i], "%u", &mcopt.exectime) <= 0 ||
mcopt.exectime <= 0){
fprintf(stderr, "%s: 「%s」不是正確的時間設定值\n",
argv[0], argv[i]);
exit(SCTEXIT_SYNTAX);
}
}else if(!strcmp(&argv[i][1], "e") ||
!strcmp(&argv[i][1], "exec")){
SCTMAIN_CHECKARGEXIST;
mcopt.executable = argv[i];
}else if(!strcmp(&argv[i][1], "i") ||
!strcmp(&argv[i][1], "in") ||
!strcmp(&argv[i][1], "input")){
SCTMAIN_CHECKARGEXIST;
mcopt.inputfile = argv[i];
}else if(!strcmp(&argv[i][1], "o") ||
!strcmp(&argv[i][1], "out") ||
!strcmp(&argv[i][1], "output")){
SCTMAIN_CHECKARGEXIST;
mcopt.outputfile = argv[i];
}else if(!strcmp(&argv[i][1], "r") ||
!strcmp(&argv[i][1], "root") ||
!strcmp(&argv[i][1], "chroot")){
SCTMAIN_CHECKARGEXIST;
mcopt.chrootdir = argv[i];
}else if(!strcmp(&argv[i][1], "m") ||
!strcmp(&argv[i][1], "misc") ||
!strcmp(&argv[i][1], "opt") ||
!strcmp(&argv[i][1], "option")){
SCTMAIN_CHECKARGEXIST;
qarglist = l4qarg_parse(argv[i]);
if(qarglist == NULL){
fprintf(stderr, "%s: 記憶體配置發生錯誤\n", argv[0]);
exit(SCTEXIT_MALLOC);
}
for(j=0; !l4qarg_end(qarglist[j]); j++){
if(!strcmp(qarglist[j].arg_name, "mem") ||
!strcmp(qarglist[j].arg_name, "memory")){
SCTMAIN_CHECKQARGHAVEVALUE;
if(sscanf(qarglist[j].arg_value, "%u",
&mcopt.memlimit) <= 0 ||
mcopt.memlimit <= 0){
fprintf(stderr, "%s: 「%s」不是正確的記憶體限制"
"設定值\n", argv[0],
qarglist[j].arg_value);
exit(SCTEXIT_SYNTAX);
}
}else if(!strcmp(qarglist[j].arg_name, "outlimit") ||
!strcmp(qarglist[j].arg_name, "outlim")){
SCTMAIN_CHECKQARGHAVEVALUE;
if(sscanf(qarglist[j].arg_value, "%u",
&mcopt.outlimit) <= 0 ||
mcopt.outlimit <= 0){
fprintf(stderr, "%s: 「%s」不是正確的輸出限制"
"設定\n", argv[0], qarglist[j].arg_value);
exit(SCTEXIT_SYNTAX);
}
}else if(!strcmp(qarglist[j].arg_name, "stderr")){
SCTMAIN_CHECKQARGNOVALUE;
mcopt.flags |= SCTMC_REDIR_STDERR;
}else if(!strcmp(qarglist[j].arg_name, "nocopy")){
SCTMAIN_CHECKQARGNOVALUE;
mcopt.flags |= SCTMC_NOCOPY;
}else if(!strcmp(qarglist[j].arg_name, "uid") ||
!strcmp(qarglist[j].arg_name, "user")){
SCTMAIN_CHECKQARGHAVEVALUE;
mcopt.flags |= SCTMC_SETUID;
if(sscanf(qarglist[j].arg_value, "%u", &mcopt.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(SCTEXIT_SYNTAX);
}
mcopt.uid = pwinfo->pw_uid;
#else
fprintf(stderr, "%s: 「%s」並不是整數\n",
argv[0], qarglist[j].arg_value);
exit(SCTEXIT_SYNTAX);
#endif
}
}else if(!strcmp(qarglist[j].arg_name, "gid") ||
!strcmp(qarglist[j].arg_name, "group")){
SCTMAIN_CHECKQARGHAVEVALUE;
mcopt.flags |= SCTMC_SETGID;
if(sscanf(qarglist[j].arg_value, "%u", &mcopt.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(SCTEXIT_SYNTAX);
}
mcopt.gid = grinfo->gr_gid;
#else
fprintf(stderr, "%s: 「%s」並不是整數\n",
argv[0], qarglist[j].arg_value);
exit(SCTEXIT_SYNTAX);
#endif
}
}else{
fprintf(stderr, "%s: 「%s」是不明的選項\n", argv[0],
qarglist[j].arg_name);
exit(SCTEXIT_SYNTAX);
}
}
l4qarg_free(qarglist);
}else{
fprintf(stderr, "%s: 不明的選項「%s」\n", argv[0], argv[i]);
return SCTEXIT_SYNTAX;
}
}else{
fprintf(stderr, "%s: 參數 %d「%s」是多餘的\n"
"請嘗試執行 %s -help 來取得說明\n"
, argv[0], i, argv[i], argv[0]);
return SCTEXIT_SYNTAX;
}
}
/* 至此可以進入主程式了 */
if(verbose){
puts(SCTJUDGE_TITLEBAR"\n");
}
/* 檢查與修正設定值 */
if(verbose){
puts("正在檢查設定值是否正確......");
}
if(mcopt.executable == NULL){
fputs("受測程式可執行檔名稱為必要參數，不可為空白\n", stderr);
exit(SCTEXIT_TOOFEW);
}
if(mcopt.inputfile == NULL){
mcopt.inputfile = NULL_DEVICE;
}
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);
}
return SCTEXIT_SUCCESS;
}