Question

我在C中读到了关于内存使用量计算的article，并且有问题。

我编写了一个简单的测试程序，它可能工作超过一秒，并且使用的内存超过1 KB。

#include <stdio.h>
#include <stdlib.h>

int main(void)
{
int a;
int f[1000000];
sleep(1);
 scanf("%d",&a);
 printf("%d %d\n",a/10,a%10);

return 0;
}

然后我将其编译为某些main.exe并执行文章

中的程序操作

pid = fork();
if (pid == 0) 
{
    struct rlimit rlim;
    rlim.rlim_cur = rlim.rlim_max = TIME_LIMIT;
    setrlimit(RLIMIT_CPU, &rlim);
    execv("./main.exe",NULL);
}
else 
{
        struct rusage resource_usage;
        // set arbitrary lower limit value of memory used
        int memory_used = 128;
        pid_t pid2;

        do {
            memory_used = max(memory_used, get_memory_usage(pid));
            if ((memory_used > memory_limit)
                kill(pid, SIGKILL);

           // wait for the child process to change state
            pid2 = wait4(pid, &status, WUNTRACED | WCONTINUED, &resource_usage);
        } while (pid2 == 0);
}

来自文章

的函数get_memory_usage()

int get_memory_usage(pid_t pid) {
    int fd, data, stack;
    char buf[4096], status_child[NAME_MAX];
    char *vm;

    sprintf(status_child, "/proc/%d/status", pid);
    if ((fd = open(status_child, O_RDONLY)) < 0)
        return -1;

    read(fd, buf, 4095);
    buf[4095] = '\0';
    close(fd);

    data = stack = 0;

    vm = strstr(buf, "VmData:");
    if (vm) {
        sscanf(vm, "%*s %d", &data);
    }
    vm = strstr(buf, "VmStk:");
    if (vm) {
        sscanf(vm, "%*s %d", &stack);
    }

    return data + stack;    
}

但字符串中的问题是pid2 = wait4(pid, &status, WUNTRACED | WCONTINUED, &resource_usage);。 while只进行一次迭代并等待进程结束。但是我需要计算内存，rusage resource_usage不要给我这个信息。在子进程运行时，如何使while工作，并在停止的地方停止。 ZOMBIE时，儿童过程的状态存在更多问题。它没有记忆。我也需要抓住它。因为如果我的main.exe我使用这个测试程序：

#include <stdio.h>
#include <stdlib.h>

int main(void)
{
   int a;
   int f[1000000];
   sleep(1);
   scanf("%d",&a);
   printf("%d %d\n",a/10,a%10);
   return 0;
}

当我在无限get_memory_usage中从while做一些输出时，会显示所有/proc/[pid]/status输出。而且我看到，那个孩子的过程是

Name:   check.exe
State:  R (running)

之后它会

Name:   main.exe
State:  Z (zombie)

这意味着proc无法捕获main.exe正在运行的信息。

Answer 1

根据waitpid()手册页，您对wait4()的调用会阻止，直到进程停止，或者在停止后再次恢复。这与阻止等待输入不同，这意味着它已被信号（SIGSTOP）停止。

您需要WNOHANG来阻止wait4()阻止并导致其立即返回，可能是这样的：

do 
{
   // All the stuff you want to do
   pid2 = wait4(pid, &status, WNOHANG, &resource_usage);
} while (pid2 == 0);

NB我根本没有测试过上面的内容，甚至没有对它进行编译。

C中子进程的内存使用情况

1 个答案: