获取子进程C ++的执行结果

时间:2012-06-29 16:10:06

标签: c++ linux fork

我一直在努力实施在线评判,但我被困了.. 我使用fork来创建子进程,然后从子进程中使用system命令启动我应该测试的程序。

这是我用来测试评分者的程序。它处于无限循环中,但我只是评论该行以使其有效。

#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);
    }

1 个答案:

答案 0 :(得分:2)

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

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

N.B个人我不会将system()用于那样的文件重定向,我会设置管道然后在子节点中使用exec(),因此平地机实际上在孩子中运行。或者我只使用我的pstreams库。