程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> 內核進程的復制

內核進程的復制

編輯:C++入門知識

在通過fork系統調用創建進程時,最終會進入內核的do_fork函數,這個函數的大部分工作都是進程的復制,就是把大部分工作都委托給函數copy_process函數來完成。本博文主要討論進程的復制工作。

下面分成幾個段,所在代碼包含了整個copy_process函數

一,標志檢查

[cpp] 
static struct task_struct *copy_process(unsigned long clone_flags, 
                    unsigned long stack_start, 
                    struct pt_regs *regs, 
                    unsigned long stack_size, 
                    int __user *child_tidptr, 
                    struct pid *pid) 

    int retval; 
    struct task_struct *p; 
    int cgroup_callbacks_done = 0; 
 
    if ((clone_flags & (CLONE_NEWNS|CLONE_FS)) == (CLONE_NEWNS|CLONE_FS)) 
        return ERR_PTR(-EINVAL); 
 
    /*
     * Thread groups must share signals as well, and detached threads
     * can only be started up within the thread group.
     */ 
    if ((clone_flags & CLONE_THREAD) && !(clone_flags & CLONE_SIGHAND)) 
        return ERR_PTR(-EINVAL); 
 
    /*
     * Shared signal handlers imply shared VM. By way of the above,
     * thread groups also imply shared VM. Blocking this case allows
     * for various simplifications in other code.
     */ 
    if ((clone_flags & CLONE_SIGHAND) && !(clone_flags & CLONE_VM)) 
        return ERR_PTR(-EINVAL); 
 
    retval = security_task_create(clone_flags); 
    if (retval) 
        goto fork_out; 
 這是函數的開始部分,先進行傳入的參數檢查,主要是:
如果創建進程的時候,要求創建一個新的命名空間(CLONE_NEWNS),並且同時要求與父進程共享所有的文件系統信息(CLONE_FS),這是不允許的。此時是要求共享其文件系統。
在用CLONE_THREAD標志時,必須使用CLONE_SIGHAND標志,後者表示共享相同的信號處理表。
在使用CLONE_SIGHAND標志時,必須使用CLONE_VM標志,後者表示子進程和你進程共享虛擬地址空間,也只有這個時候,才能提供共享的信號處理程序。
二,建立副本
[cpp] 
    retval = -ENOMEM; 
    p = dup_task_struct(current); 
<span>  </span>if (!p) 
<span>      </span>goto fork_out; 
 dup_task_struct用來建立父進程的副本,函數如下:
[cpp]
static struct task_struct *dup_task_struct(struct task_struct *orig) 

    struct task_struct *tsk; 
    struct thread_info *ti; 
    int err; 
 
    prepare_to_copy(orig); 
 
    tsk = alloc_task_struct(); 
    if (!tsk) 
        return NULL; 
 
    ti = alloc_thread_info(tsk); 
    if (!ti) { 
        free_task_struct(tsk); 
        return NULL; 
    } 
 
    *tsk = *orig;//將父進程的內容填充新的進程 
    tsk->stack = ti; 
 
    err = prop_local_init_single(&tsk->dirties); 
    if (err) { 
        free_thread_info(ti); 
        free_task_struct(tsk); 
        return NULL; 
    } 
 
    setup_thread_stack(tsk, orig); 
 
#ifdef CONFIG_CC_STACKPROTECTOR 
    tsk->stack_canary = get_random_int(); 
#endif 
 
    /* One for us, one for whoever does the "release_task()" (usually parent) */ 
    atomic_set(&tsk->usage,2);//使用計數器置為2,表示當前進程描述符處於活動狀態。 
    atomic_set(&tsk->fs_excl, 0); 
#ifdef CONFIG_BLK_DEV_IO_TRACE 
    tsk->btrace_seq = 0; 
#endif 
    tsk->splice_pipe = NULL; 
    return tsk; 

調用alloc_task_struct為新進程分配進程結構,返回tsk指針。
為新的進程分配一個核心態棧,也就是tsk->stack。棧和thread_info一同保存在一個聯合結構中。thread_info用於保存進程所需的特定於處理器的底層信息,定義如下:
[cpp]
struct thread_info { 
    struct task_struct  *task;      /* 不前的主進程 */ 
    unsigned long       flags; 
    struct exec_domain  *exec_domain;   /* 執行區間 */ 
    int         preempt_count;  /* 內核搶占所需的一個計數器*/ 
    __u32 cpu; /* 進程正在其上執行的CPU數目 */ 
    struct restart_block    restart_block;//用於實現信號機制 
}; 

而進程的棧和thread_info的聯合體定義如下:
[cpp] 
union thread_union { 
    struct thread_info thread_info; 
    unsigned long stack[THREAD_SIZE/sizeof(long)]; 
}; 
在分配了棧後,調用setup_thread_stack確定棧內的布局。這個函數完成的操作是:把父進程的thread_info(進程描述結構)值復制給tsk的進程描述結構。然後將tsk的進程描述符中的task域改為tsk。
[cpp]
#define task_thread_info(task)  ((struct thread_info *)(task)->stack) 
 
static inline void setup_thread_stack(struct task_struct *p, struct task_struct *org) 

    *task_thread_info(p) = *task_thread_info(org); 
    task_thread_info(p)->task = p; 

在執行完setup_thread_stack之後,父子進程除了stack的指針之外是完全一樣的。
三,檢查進程數創建限制
[cpp]
    rt_mutex_init_task(p);//互斥鎖初始化 
 
#ifdef CONFIG_TRACE_IRQFLAGS 
    DEBUG_LOCKS_WARN_ON(!p->hardirqs_enabled); 
    DEBUG_LOCKS_WARN_ON(!p->softirqs_enabled); 
#endif 
    retval = -EAGAIN; 
    if (atomic_read(&p->user->processes) >= 
            p->signal->rlim[RLIMIT_NPROC].rlim_cur) { 
        if (!capable(CAP_SYS_ADMIN) && !capable(CAP_SYS_RESOURCE) && 
            p->user != current->nsproxy->user_ns->root_user) 
            goto bad_fork_free; 
    } 
<span>  </span>atomic_inc(&p->user->__count); 
<span>  </span>atomic_inc(&p->user->processes); 
<span>  </span>get_group_info(p->group_info);增加組的使用計數 
 進程結構task_struct中有一個域,名為:user_struct,這個域保存了當前用戶的使用資源計數。
[cpp] 
struct task_struct{ 
...... 
     struct user_struct *user; 
...... 
}; 
 在user_struct結構中包含processes表示當前的用戶能夠創建最多的進程數。如果超過限制,就放棄創建進程。root用戶除外。如果沒有超過限制,就將user_struct結構的引用計數加1,並將已經創建的進程數加1。
四,線程檢查
檢查線程創建是否超過最大限制,這個比較簡單
[cpp] 
/*
 * If multiple threads are within copy_process(), then this check
 * triggers too late. This doesn't hurt, the check is only there
 * to stop root fork bombs.
 */ 
if (nr_threads >= max_threads) 
    goto bad_fork_cleanup_count; 
五,寫入數據
在這裡,進程描述符task_struct已經建立好了,其和父進程是一樣的,除了棧地址不一樣之外,現在開始修改一些值。
[cpp]
    if (!try_module_get(task_thread_info(p)->exec_domain->module)) 
        goto bad_fork_cleanup_count; 
 
    if (p->binfmt && !try_module_get(p->binfmt->module)) 
        goto bad_fork_cleanup_put_domain; 
 
    p->did_exec = 0;當前還有加載任何執行的系統調用,所以為0以標識 
    delayacct_tsk_init(p);  /* Must remain after dup_task_struct() */ 
    copy_flags(clone_flags, p); 
    INIT_LIST_HEAD(&p->children); 
    INIT_LIST_HEAD(&p->sibling); 
    p->vfork_done = NULL; 
    spin_lock_init(&p->alloc_lock); 
 
    clear_tsk_thread_flag(p, TIF_SIGPENDING); 
    init_sigpending(&p->pending); 
 
    p->utime = cputime_zero; 
    p->stime = cputime_zero; 
    p->gtime = cputime_zero; 
    p->utimescaled = cputime_zero; 
    p->stimescaled = cputime_zero; 
    p->prev_utime = cputime_zero; 
    p->prev_stime = cputime_zero; 
 
#ifdef CONFIG_TASK_XACCT 
    p->rchar = 0;        /* I/O counter: bytes read */ 
    p->wchar = 0;        /* I/O counter: bytes written */ 
    p->syscr = 0;        /* I/O counter: read syscalls */ 
    p->syscw = 0;        /* I/O counter: write syscalls */ 
#endif 
    task_io_accounting_init(p); 
    acct_clear_integrals(p); 
 
    p->it_virt_expires = cputime_zero; 
    p->it_prof_expires = cputime_zero; 
    p->it_sched_expires = 0; 
    INIT_LIST_HEAD(&p->cpu_timers[0]); 
    INIT_LIST_HEAD(&p->cpu_timers[1]); 
    INIT_LIST_HEAD(&p->cpu_timers[2]); 
 
    p->lock_depth = -1;      /* -1 = no lock */ 
    do_posix_clock_monotonic_gettime(&p->start_time); 
    p->real_start_time = p->start_time; 
    monotonic_to_bootbased(&p->real_start_time); 
#ifdef CONFIG_SECURITY 
    p->security = NULL; 
#endif 
    p->io_context = NULL; 
    p->audit_context = NULL; 
    cgroup_fork(p); 
#ifdef CONFIG_NUMA 
    p->mempolicy = mpol_copy(p->mempolicy); 
    if (IS_ERR(p->mempolicy)) { 
        retval = PTR_ERR(p->mempolicy); 
        p->mempolicy = NULL; 
        goto bad_fork_cleanup_cgroup; 
    } 
    mpol_fix_fork_child_flag(p); 
#endif 
#ifdef CONFIG_TRACE_IRQFLAGS 
    p->irq_events = 0; 
#ifdef __ARCH_WANT_INTERRUPTS_ON_CTXSW 
    p->hardirqs_enabled = 1; 
#else 
    p->hardirqs_enabled = 0; 
#endif 
    p->hardirq_enable_ip = 0; 
    p->hardirq_enable_event = 0; 
    p->hardirq_disable_ip = _THIS_IP_; 
    p->hardirq_disable_event = 0; 
    p->softirqs_enabled = 1; 
    p->softirq_enable_ip = _THIS_IP_; 
    p->softirq_enable_event = 0; 
    p->softirq_disable_ip = 0; 
    p->softirq_disable_event = 0; 
    p->hardirq_context = 0; 
    p->softirq_context = 0; 
#endif 
#ifdef CONFIG_LOCKDEP 
    p->lockdep_depth = 0; /* no locks held yet */ 
    p->curr_chain_key = 0; 
    p->lockdep_recursion = 0; 
#endif 
#ifdef CONFIG_DEBUG_MUTEXES 
    p->blocked_on = NULL; /* not blocked yet */ 
#endif 
這裡是對一些值進行初始化,和各種策略相關的值,如調度等等,有些值是很重要的。
六,加入調度
[cpp]
/* Perform scheduler related setup. Assign this task to a CPU. */ 
sched_fork(p, clone_flags); 
這可以使用是進程可以參加調度,但此時進程的狀態改為正在TASK_RUNNING,這以防止內核的其他部分將其改為可運行狀態,因為我們對進程的設置還沒有完成,這樣調度進程會有問題。

七,開始復制
[cpp] 
if ((retval = security_task_alloc(p))) 
    goto bad_fork_cleanup_policy; 
if ((retval = audit_alloc(p))) 
    goto bad_fork_cleanup_security; 
/* copy all the process information */ 
if ((retval = copy_semundo(clone_flags, p)))System V信號量 
    goto bad_fork_cleanup_audit; 
if ((retval = copy_files(clone_flags, p)))文件描述符 
    goto bad_fork_cleanup_semundo; 
if ((retval = copy_fs(clone_flags, p)))文件系統上下文  
    goto bad_fork_cleanup_files; 
if ((retval = copy_sighand(clone_flags, p)))進程信息處理程序 
    goto bad_fork_cleanup_fs; 
if ((retval = copy_signal(clone_flags, p))) 
    goto bad_fork_cleanup_sighand; 
if ((retval = copy_mm(clone_flags, p)))地址空間 
    goto bad_fork_cleanup_signal; 
if ((retval = copy_keys(clone_flags, p))) 
    goto bad_fork_cleanup_mm; 
if ((retval = copy_namespaces(clone_flags, p))) 
    goto bad_fork_cleanup_keys; 
retval = copy_thread(0, clone_flags, stack_start, stack_size, p, regs); 
if (retval) 
    goto bad_fork_cleanup_namespaces; 
這裡有各種Copy。

八,分配PID等操作
[cpp]
    if (pid != &init_struct_pid) { 
        retval = -ENOMEM; 
        pid = alloc_pid(task_active_pid_ns(p)); 
        if (!pid) 
            goto bad_fork_cleanup_namespaces; 
 
        if (clone_flags & CLONE_NEWPID) { 
            retval = pid_ns_prepare_proc(task_active_pid_ns(p)); 
            if (retval < 0) 
                goto bad_fork_free_pid; 
        } 
    } 
 
    p->pid = pid_nr(pid); 
    p->tgid = p->pid; 
    if (clone_flags & CLONE_THREAD) 
        p->tgid = current->tgid; 
 
    p->set_child_tid = (clone_flags & CLONE_CHILD_SETTID) ? child_tidptr : NULL; 
    /*
     * Clear TID on mm_release()?
     */ 
    p->clear_child_tid = (clone_flags & CLONE_CHILD_CLEARTID) ? child_tidptr: NULL; 
#ifdef CONFIG_FUTEX 
    p->robust_list = NULL; 
#ifdef CONFIG_COMPAT 
    p->compat_robust_list = NULL; 
#endif 
    INIT_LIST_HEAD(&p->pi_state_list); 
    p->pi_state_cache = NULL; 
#endif 
    /*
     * sigaltstack should be cleared when sharing the same VM
     */ 
    if ((clone_flags & (CLONE_VM|CLONE_VFORK)) == CLONE_VM) 
        p->sas_ss_sp = p->sas_ss_size = 0; 
 
    /*
     * Syscall tracing should be turned off in the child regardless
     * of CLONE_PTRACE.
     */ 
    clear_tsk_thread_flag(p, TIF_SYSCALL_TRACE); 
#ifdef TIF_SYSCALL_EMU 
    clear_tsk_thread_flag(p, TIF_SYSCALL_EMU); 
#endif 
 
    /* Our parent execution domain becomes current domain
       These must match for thread signalling to apply */ 
    p->parent_exec_id = p->self_exec_id; 
 
    /* ok, now we should be set up.. */ 
    p->exit_signal = (clone_flags & CLONE_THREAD) ? -1 : (clone_flags & CSIGNAL); 
    p->pdeath_signal = 0; 
    p->exit_state = 0; 
 
    /*
     * Ok, make it visible to the rest of the system.
     * We dont wake it up yet.
     */ 
    p->group_leader = p; 
    INIT_LIST_HEAD(&p->thread_group); 
    INIT_LIST_HEAD(&p->ptrace_children); 
    INIT_LIST_HEAD(&p->ptrace_list); 
 
    /* Now that the task is set up, run cgroup callbacks if
     * necessary. We need to run them before the task is visible
     * on the tasklist. */ 
    cgroup_fork_callbacks(p); 
    cgroup_callbacks_done = 1; 
 
    /* Need tasklist lock for parent etc handling! */ 
    write_lock_irq(&tasklist_lock); 
 
    /* for sys_ioprio_set(IOPRIO_WHO_PGRP) */ 
    p->ioprio = current->ioprio; 
 
    /*
     * The task hasn't been attached yet, so its cpus_allowed mask will
     * not be changed, nor will its assigned CPU.
     *
     * The cpus_allowed mask of the parent may have changed after it was
     * copied first time - so re-copy it here, then check the child's CPU
     * to ensure it is on a valid CPU (and if not, just force it back to
     * parent's CPU). This avoids alot of nasty races.
     */ 
    p->cpus_allowed = current->cpus_allowed; 
    if (unlikely(!cpu_isset(task_cpu(p), p->cpus_allowed) || 
            !cpu_online(task_cpu(p)))) 
        set_task_cpu(p, smp_processor_id()); 
九,初始化父子關系
[cpp] 
/* CLONE_PARENT re-uses the old parent */ 
if (clone_flags & (CLONE_PARENT|CLONE_THREAD)) 
    p->real_parent = current->real_parent; 
else 
    p->real_parent = current; 
p->parent = p->real_parent; 
 
spin_lock(&current->sighand->siglock); 
 
/*
 * Process group and session signals need to be delivered to just the
 * parent before the fork or both the parent and the child after the
 * fork. Restart if a signal comes in before we add the new process to
 * it's process group.
 * A fatal signal pending means that current will exit, so the new
 * thread can't slip out of an OOM kill (or normal SIGKILL).
     */ 
recalc_sigpending(); 
if (signal_pending(current)) { 
    spin_unlock(¤t->sighand->siglock); 
    write_unlock_irq(&tasklist_lock); 
    retval = -ERESTARTNOINTR; 
    goto bad_fork_free_pid; 

如果創建的是線程或者是PARENT選項,則其父進程為current的父進程。

十,線程創建部分
如果本次創建的是線程,一些另於進程創建的操作。
[cpp]
if (clone_flags & CLONE_THREAD) { 
    p->group_leader = current->group_leader;//線程組長就是當前線程的組長 
    list_add_tail_rcu(&p->thread_group, &p->group_leader->thread_group); 
 
    if (!cputime_eq(current->signal->it_virt_expires, 
            cputime_zero) || 
        !cputime_eq(current->signal->it_prof_expires, 
            cputime_zero) || 
        current->signal->rlim[RLIMIT_CPU].rlim_cur != RLIM_INFINITY || 
        !list_empty(&current->signal->cpu_timers[0]) || 
        !list_empty(&current->signal->cpu_timers[1]) || 
        !list_empty(&current->signal->cpu_timers[2])) { 
        /*
         * Have child wake up on its first tick to check
         * for process CPU timers.
         */ 
        p->it_prof_expires = jiffies_to_cputime(1); 
    } 

十一,新進程插入進程鏈表
[cpp] 
if (likely(p->pid)) { 
    add_parent(p); 
    if (unlikely(p->ptrace & PT_PTRACED)) 
        __ptrace_link(p, current->parent); 
 
    if (thread_group_leader(p)) { 
        if (clone_flags & CLONE_NEWPID) 
            p->nsproxy->pid_ns->child_reaper = p;如果創建新的命名空間,則將命名空間的child_reaper設為當前創建的進程,這個進程就是這個創建的命名空間的init進程。 
 
        p->signal->tty = current->signal->tty; 
        set_task_pgrp(p, task_pgrp_nr(current)); 
        set_task_session(p, task_session_nr(current)); 
        attach_pid(p, PIDTYPE_PGID, task_pgrp(current)); 
        attach_pid(p, PIDTYPE_SID, task_session(current)); 
        list_add_tail_rcu(&p->tasks, &init_task.tasks); 
        __get_cpu_var(process_counts)++; 
    } 
    attach_pid(p, PIDTYPE_PID, pid); 
    nr_threads++; 

add_parent宏將進程的children鏈表與父進程連接,實現如下
[cpp] 
#define add_parent(p)       list_add_tail(&(p)->sibling,&(p)->parent->children) 
 
十二,最後
[cpp] 
    total_forks++; 
    spin_unlock(&current->sighand->siglock); 
    write_unlock_irq(&tasklist_lock); 
    proc_fork_connector(p); 
    cgroup_post_fork(p); 
    return p; 
 
bad_fork_free_pid: 
    if (pid != &init_struct_pid) 
        free_pid(pid); 
bad_fork_cleanup_namespaces: 
    exit_task_namespaces(p); 
bad_fork_cleanup_keys: 
    exit_keys(p); 
bad_fork_cleanup_mm: 
    if (p->mm) 
        mmput(p->mm); 
bad_fork_cleanup_signal: 
    cleanup_signal(p); 
bad_fork_cleanup_sighand: 
    __cleanup_sighand(p->sighand); 
bad_fork_cleanup_fs: 
    exit_fs(p); /* blocking */ 
bad_fork_cleanup_files: 
    exit_files(p); /* blocking */ 
bad_fork_cleanup_semundo: 
    exit_sem(p); 
bad_fork_cleanup_audit: 
    audit_free(p); 
bad_fork_cleanup_security: 
    security_task_free(p); 
bad_fork_cleanup_policy: 
#ifdef CONFIG_NUMA 
    mpol_free(p->mempolicy); 
bad_fork_cleanup_cgroup: 
#endif 
    cgroup_exit(p, cgroup_callbacks_done); 
    delayacct_tsk_free(p); 
    if (p->binfmt) 
        module_put(p->binfmt->module); 
bad_fork_cleanup_put_domain: 
    module_put(task_thread_info(p)->exec_domain->module); 
bad_fork_cleanup_count: 
    put_group_info(p->group_info); 
    atomic_dec(&p->user->processes); 
    free_uid(p->user); 
bad_fork_free: 
    free_task(p); 
fork_out: 
    return ERR_PTR(retval); 

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved