_do_fork（）如何返回两个不同的PID（一个用于父进程，一个用于子进程）

Question

我正在查看_do_fork()函数（），试图理解fork()如何返回父进程的子PID和子进程的0。

我认为nr包含子进程的PID（将返回给调用者进程），但是我无法看到它如何能够将0返回给子进程。 / p>

答案How does fork() know when to return 0?表示返回值是在为新进程创建的堆栈上传递的，但是（除了没有真正理解它）我在代码中找不到 EM>

那么，哪里为子流程设置了返回值0

_do_fork()函数的代码复制如下：

long _do_fork(unsigned long clone_flags,
          unsigned long stack_start,
          unsigned long stack_size,
          int __user *parent_tidptr,
          int __user *child_tidptr,
          unsigned long tls)
{
    struct task_struct *p;
    int trace = 0;
    long nr;

    /*
     * Determine whether and which event to report to ptracer.  When
     * called from kernel_thread or CLONE_UNTRACED is explicitly
     * requested, no event is reported; otherwise, report if the event
     * for the type of forking is enabled.
     */
    if (!(clone_flags & CLONE_UNTRACED)) {
        if (clone_flags & CLONE_VFORK)
            trace = PTRACE_EVENT_VFORK;
        else if ((clone_flags & CSIGNAL) != SIGCHLD)
            trace = PTRACE_EVENT_CLONE;
        else
            trace = PTRACE_EVENT_FORK;

        if (likely(!ptrace_event_enabled(current, trace)))
            trace = 0;
    }

    p = copy_process(clone_flags, stack_start, stack_size,
             child_tidptr, NULL, trace, tls, NUMA_NO_NODE);
    add_latent_entropy();
    /*
     * Do this prior waking up the new thread - the thread pointer
     * might get invalid after that point, if the thread exits quickly.
     */
    if (!IS_ERR(p)) {
        struct completion vfork;
        struct pid *pid;

        trace_sched_process_fork(current, p);

        pid = get_task_pid(p, PIDTYPE_PID);
        nr = pid_vnr(pid);

        if (clone_flags & CLONE_PARENT_SETTID)
            put_user(nr, parent_tidptr);

        if (clone_flags & CLONE_VFORK) {
            p->vfork_done = &vfork;
            init_completion(&vfork);
            get_task_struct(p);
        }

        wake_up_new_task(p);

        /* forking complete and child started to run, tell ptracer */
        if (unlikely(trace))
            ptrace_event_pid(trace, pid);

        if (clone_flags & CLONE_VFORK) {
            if (!wait_for_vfork_done(p, &vfork))
                ptrace_event_pid(PTRACE_EVENT_VFORK_DONE, pid);
        }

        put_pid(pid);
    } else {
        nr = PTR_ERR(p);
    }
    return nr;
}

Answer 1

您已使用return nr正确识别了新流程ID如何返回到父流程。但是，由于此代码在父线程上执行，因此您永远不会在任何地方看到return 0。此代码不适用于创建的新进程。

现在让我们检查一下_do_fork函数。

...
}
p = copy_process(clone_flags, stack_start, stack_size,
         child_tidptr, NULL, trace, tls, NUMA_NO_NODE);
add_latent_entropy();
...

这就是所有魔法发生的地方。当您致电copy_process 时，它会在内部调用copy_thread，这是一个特定目标代码。该函数负责处理与线程相关的数据结构。

现在说我们将目标作为X86_64，其调用约定是在%rax寄存器中返回返回值。然后，此函数将0复制到%rax，并将return_from_fork地址的值复制到%rip（指令指针）。

在其他平台上，ABI可能要求返回值进入堆栈。在这种情况下，0被放置在堆栈上。 copy_thread是特定于目标的，但copy_process不是。

This是X86_64的copy_thread实现。您可以在第160行周围看到正在设置的sp寄存器。在第182行，您可以看到%ax（％rax的子注册）被设置为0。

我希望这可以解决一些困惑。