2

我一直在尝试实现在线判断，但我被卡住了。我使用 fork 创建一个子进程，然后从子进程中我使用系统命令启动我应该测试的程序。

这是我用来测试分级机的程序。它处于无限循环中，但我只是评论该行以使其工作。

#include <stdio.h>
int main () {
    int a, b;
    scanf ("%d %d", &a, &b);
    while (1);
    printf ("%d\n", a + b);

    return 0;
}

到目前为止我编码的分级机是

struct timeval start, end;
    struct rlimit rl, default_memory, default_time;
    getrlimit (RLIMIT_AS, &default_memory);
    getrlimit (RLIMIT_CPU, &default_time);

    pid_t pid = fork ();

    gettimeofday (&start, NULL);

    if (pid < 0) printf ("Error\n");
    else if (pid == 0) { // child process
        if (memory_limit > 0) {
            rl.rlim_cur = rl.rlim_max = memory_limit;
            setrlimit (RLIMIT_AS, &rl);
        }

        if (time_limit > 0) {
            rl.rlim_cur = rl.rlim_max = time_limit / 1000 + 1;
            setrlimit (RLIMIT_CPU, &rl);
        }


        ptrace (PTRACE_TRACEME, pid, NULL, NULL);
        system ("./sum < in.txt > out.txt");

        setrlimit (RLIMIT_CPU, &default_time);
        setrlimit (RLIMIT_AS, &default_memory);
    }
    else { // parent process
        int *status;


            // Is this ok?
            // Because here http://linux.die.net/man/2/ptrace it says that
            // PTRACE_GETSIGINFO is used to get information about the signal that
            // caused the stop. And I'm interested If the program was killed because of
            // rlimit (exceeding time or memory limit) or if it was something else.
        ptrace (PTRACE_GETSIGINFO, pid, NULL, NULL);
        if (wait (status) == -1) return 1; // waiting for the child process to finish

        if (WIFSIGNALED (status)) {
            int exit_signal = WTERMSIG (status);
            printf ("Terminated with signal: %d.\n", exit_signal);
            switch (exit_signal) {
                case 0:
                    printf ("OK\n");
                    break;
                case SIGSEGV: // RLIMIT_AS -- SIGSEGV is sent, when the memory limit is exceeded
                    printf ("Memory Limit Exceeded\n");
                    break;
                case SIGXCPU: // RLIMIT_CPU -- when soft limit is reached SIGXCPU is sent
                case SIGKILL: // when hard limit is reached SIGKILL is sent
                    printf ("Time Limit Exceeded\n");
                    break;
                default: // I didn't handle all the signals, so I made everything else a run time error
                    printf ("Run Time Error\n");
                    break;
            }
        }

        ptrace (PTRACE_KILL, pid, NULL, NULL); // Kill the process If it is not killed

        gettimeofday (&end, NULL); // This is working fine
        double time = (end.tv_sec - start.tv_sec) + 1e-6 * (end.tv_usec - start.tv_usec);
        printf ("%lf\n", time);
    }
4

1 回答 1

2

您不需要使用 ptrace，您可以使用 向孩子发送信号kill(2)并通过 获取退出状态wait(2)，如果它因为使用 WTERMSIG 的信号而退出。

问题可能是您使用的是 system()，因此评分器实际上是在第三个进程中运行，而不是在子进程中。

注意我个人不会使用 system() 和这样的文件重定向，我会设置管道然后在孩子中使用 exec() ，所以评分器实际上在孩子中运行。或者我只是使用我的pstreams库。

于 2012-06-29T16:27:46.423 回答