Wangyt

导航

对于Linux内核执行过程的理解(基于fork、execve、schedule等函数)

382 + 原创作品转载请注明出处 + https://github.com/mengning/linuxkernel/

一、实验环境

  win10 -> VMware -> Ubuntu16.04 + GDB -> QEMU -> linux-3.9.4 + MenuOS

二、实验目的

  1、理解Linux内核中创建一个进程的过程;

  2、理解Linux内核加载可执行程序的过程;

  3、理解Linux内核中的进程调度时机以及进程切换过程。

三、实验过程及结果

  1、对于Linux内核中进程创建过程的理解

   对于每个进程来说,都有其对应的进程控制块(PCB),在内核文件中,对于PCB结构体【task_struct】的定义比较复杂,包含的参数众多,这里就不详细进行解读【具体的定义存放于:/linux-3.9.4/inlude/sched.h】,一般情况下进程控制块中的信息都会包含进程状态、进程堆栈栈顶指针、进程唯一标识符、进程相关的链表指针等信息。 

  接下来是基于fork函数来理解内核的进程创建过程。进程的创建,大体上是将父进程的信息复制给子进程,子进程再对一些参数进行针对性的修改。如果想要创建一个进程的话,大致可以通过四种方式进行创建,分别是fork、vfork、clone三个系统调用和keinel_thread内核函数,但是这四种方式最终都是通过调用do_fork函数实现进程的实际创建,只是对应的传入的参数有所不同。

  以下是do_fork函数的部分源码【/linux-3.9.4/kernel/fork.c】。

long do_fork(unsigned long clone_flags,
          unsigned long stack_start,
          unsigned long stack_size,
          int __user *parent_tidptr,
          int __user *child_tidptr)
{
    struct task_struct *p;              //创建进程描述符指针
    int trace = 0;
    long nr;                               //子进程pid

    ...
        //创建子进程的描述符和执行时所需的其他数据结构
    p = copy_process(clone_flags, stack_start, stack_size,
             child_tidptr, NULL, trace);    

    if (!IS_ERR(p)) {                              //copy_process执行成功
        struct completion vfork;           //定义完成量(一个执行单元等待另一个执行单元完成) 

        trace_sched_process_fork(current, p);

        nr = task_pid_vnr(p);           //获取pid
    
        ...
        //如果clone_flags包含CLONE_VFORK标识,将vfork完成量赋给进程描述符
        if (clone_flags & CLONE_VFORK) {
            p->vfork_done = &vfork;
            init_completion(&vfork);
            get_task_struct(p);
        }

        wake_up_new_task(p);         //将子进程添加到调度器的队列

        ...

        //如果clone_flags包含CLONE_VFORK标识,将父进程插入等待队列,直到子进程调用exec函数或退出
        if (clone_flags & CLONE_VFORK) {
            if (!wait_for_vfork_done(p, &vfork))
                ptrace_event(PTRACE_EVENT_VFORK_DONE, nr);
        }
    } else {
        nr = PTR_ERR(p);   //错误处理
    }
    return nr;                     //返回子进程pid(此处的pid为子进程的pid)
}   

  do_fork()主要完成了调用copy_process()复制父进程的相关信息,调用wake_up_new_task函数将子进程加入到调度器队列中。

  以下是copy_process()函数的部分源码【/linux-3.9.4/kernel/fork.c】。

static struct task_struct *copy_process(unsigned long clone_flags,
                    unsigned long stack_start,
                    unsigned long stack_size,
                    int __user *child_tidptr,
                    struct pid *pid,
                    int trace)
{
    int retval;
    struct task_struct *p;

    ...

    retval = security_task_create(clone_flags);    //安全性检查
    
        ...
    
    p = dup_task_struct(current);     //复制PCB,为子进程创建内核栈、进程描述符
    if (!p)
        goto fork_out;

    ftrace_graph_init_task(p);

    ...

    retval = -EAGAIN;
    if (nr_threads >= max_threads)    //用户进程数目是否超出限制
        goto bad_fork_cleanup_count;

    if (!try_module_get(task_thread_info(p)->exec_domain->module))
        goto bad_fork_cleanup_count;

    ...
    
    /* 复制进程信息 */
    retval = copy_semundo(clone_flags, p);
    ...
    retval = copy_files(clone_flags, p);
    ...
    retval = copy_fs(clone_flags, p);
    ...
    retval = copy_sighand(clone_flags, p);
    ...
    retval = copy_signal(clone_flags, p);
    ...
    retval = copy_mm(clone_flags, p);
    ...
    retval = copy_namespaces(clone_flags, p);
    ...
    retval = copy_io(clone_flags, p);
    ...
    retval = copy_thread(clone_flags, stack_start, stack_size, p);
    ...
  //如果传入的pid与init_struct_pid地址不同,为子进程分配新的pid
    if (pid != &init_struct_pid) {
        retval = -ENOMEM;
        pid = alloc_pid(p->nsproxy->pid_ns);
        if (!pid)
            goto bad_fork_cleanup_io;
    }
 
    p->pid = pid_nr(pid);     //根据pid结构体获取pid
    p->tgid = p->pid;
    if (clone_flags & CLONE_THREAD)   //子进程与父进程在同一线程组
        p->tgid = current->tgid;      //子进程继承父进程的tpid   

    ...

    if (likely(p->pid)) {
        ptrace_init_task(p, (clone_flags & CLONE_PTRACE) || trace);

        if (thread_group_leader(p)) {
            ...
       //将子进程加入其所在组的散列表中    
            attach_pid(p, PIDTYPE_PGID, task_pgrp(current));
            attach_pid(p, PIDTYPE_SID, task_session(current));
            list_add_tail(&p->sibling, &p->real_parent->children);
            list_add_tail_rcu(&p->tasks, &init_task.tasks);
            __this_cpu_inc(process_counts);
        }
        attach_pid(p, PIDTYPE_PID, pid);
        nr_threads++;      //系统中进程数目++
    }

    ...

    trace_task_newtask(p, clone_flags);

    return p;     //返回子进程描述符p

...
}
    

  copy_process()函数主要完成了调用dup_task_struct函数来复制父进程的进程描述符(tsak_struct)、信息检查、初始化、设置子进程的状态、利用写时复制技术进行进程资源的复制、调用copy_thread来初始化子进程内核堆栈、设置进程pid等。

  接下来是实际的验证环节。此次验证过程中,根文件系统需要重做,具体的流程,可参见https://github.com/mengning/linuxkernel/

图 1 GDB设置的断点

图 2 执行fork功能结果1

图 2 执行fork功能结果2(断点)

图3 执行fork功能结果3

  2、对于Linux内核中可执行程序加载过程的理解

  目标文件也叫ABI(Application Binary Interface,应用程序二进制接口)。最开始的目标文件格式是*.out,后来发展成了COFF格式,现如今,常用的目标文件格式主要有两种,一种是Windows平台下的PE,一种是Linux平台下的ELF。

  ELF(Executable and Linkable Format),可执行和可连接的格式。ELF格式的文件主要用于存储Linux程序。具体来说,ELF格式文件有三种不同类型的目标文件:可重定位文件、可执行文件、共享目标文件。ELF文件参与程序的链接和执行。

  一个源码,需要成为可执行文件之前,需要经历预处理、编译、汇编、链接等几个步骤【以hello.c为例】。

  预处理:gcc -E hello.c -o hello.i 。

  预处理时,编译器主要的工作有:删除所有注释“//”、“/**/”;删除所有的“#define”,展开所有的宏定义;处理所有的条件预编译指令;处理“#include”预编译指令,将被包含的文件插入该预编译指令的位置;添加行号和标识。处理完之后,hello.i属于文本文件。

  编译:gcc -S hello.i -o hello.s -m32 。【-S:只编译,不进行汇编;-m32:生成32位平台格式文件,与64位平台使用不同的寄存器名和指令集】

  编译过程中,gcc首先检查代码的正确性和规范性,以进一步确定代码实际要完成的工作。在检查无误之后,gcc把代码翻译成汇编语言。hello,s任然是文本文件。

  汇编:gcc -c hello.s -o hello.o.m32 -m32

  汇编结束后生成的.o文件为ELF格式的文件。目标文件至少含有三个3个节区(Section),分别是.text、.data和.bss。.bss段(bss segment)(BlockStarted by Symbol),主要是用来存放程序中未初始化的全局变量的一块内存区域,属于静态内存分配,当程序开始运行时,系统将用0来初始化这片内存区域;.data段(data segment,数据段),通常是指用来存放程序中已经初始化的全局变量的一块内存区域,属于静态分配;.text段(code segment/text segment),通常指用来存放程序执行代码的一块内存区域,内存区域大小在程序运行前就已经确定,通常属于只读。除了上述的3个主要的节区之外,还有一些其他常见的节:用于存放C中的字符串和#define定义的常量,包含着只读数据的.rodata段;包含版本控制信息的.comment段;包含动态链接信息的.dymanic段;包含动态链接符号表的.dynsym段;包含用于初始化进程的可执行代码.init段,也就是执行main函数之前需要执行的部分程序。

  链接:gcc hello.0.m32 -o hello.m32.static -m32 -static

  链接是将各种代码和数据部分收集起来并组合成一个单一文件的过程,这个文件可以为加载到内存中运行。在链接的过程当中,有静态链接和动态链接两种方式。静态链接,直接将需要的执行代码复制到最终的可执行文件中,优点是代码的装载速度快,执行速度也较快,对外部环境依赖度低,缺点是程序占用的内存比较大;动态链接,不直接复治需要的代码,而是通过一系列的符号和参数,在程序运行或加载时将这些信息传递给OS,OS负责将需要的动态库加载到内存中,然后程序在运行指定的代码时,去共享执行内存中已经加载的动态库去执行代码,最终达到运行时链接的目的,动态链接的有点事多个程序可以共享一个代码段,占用内存少,缺点是运行时加载,影响执行性能,对库的依赖度比较高,容易出现版本不兼容的问题。另外,对于动态链接,还可以细分为可执行程序装载时动态链接和运行时动态链接。

  对于可执行程序的执行环境,一般是shell程序来启动一个可执行程序,shell环境会执行相应的函数来执行可执行文件。

  在Linux中,对于一个可执行文件的执行,可以通过的方式主要有:execl、execlp、execle、execv、exexvp、execve等6个函数,其使用差异主要体现在命令行参数和环境变量参数的不同,这些函数最终都需要调用系统调用函数sys_execve()来实现执行可执行文件的目的。

  sys_execve()函数代码如下。

 1 SYSCALL_DEFINE3(execve,
 2         const char __user *, filename,
 3         const char __user *const __user *, argv,
 4         const char __user *const __user *, envp)
 5 {
 6     struct filename *path = getname(filename);
 7     int error = PTR_ERR(path);
 8     if (!IS_ERR(path)) {
 9         error = do_execve(path->name, argv, envp);
10         putname(path);
11     }
12     return error;
13 }

  do_execve()函数代码如下。

1 int do_execve(const char *filename,
2     const char __user *const __user *__argv,
3     const char __user *const __user *__envp)
4 {
5     struct user_arg_ptr argv = { .ptr.native = __argv };
6     struct user_arg_ptr envp = { .ptr.native = __envp };
7     return do_execve_common(filename, argv, envp);
8 }

  do_execve_common()函数关键代码如下。

 1 /*
 2  * sys_execve() executes a new program.
 3  */
 4 static int do_execve_common(const char *filename,
 5                 struct user_arg_ptr argv,
 6                 struct user_arg_ptr envp)
 7 {
 8     struct linux_binprm *bprm;
 9     struct file *file;
10     struct files_struct *displaced;
11     bool clear_in_exec;
12     int retval;
13     const struct cred *cred = current_cred();
14 
15     ...
16 
17     file = open_exec(filename);     //打开要加载的可执行文件,加载文件头部,判断文件类型
18 
19     ...
20 
21   //创建一个bprm结构体,把环境变量和命令行参数都复制到结构体中
22     bprm->file = file;
23     bprm->filename = filename;
24     bprm->interp = filename;
25 
26     ...
27 
28   //把传入的shell上下文复制到bprm中
29     retval = copy_strings(bprm->envc, envp, bprm);
30     if (retval < 0)
31         goto out;
32   //把传入的命令行参数复制到bprm中
33     retval = copy_strings(bprm->argc, argv, bprm);
34     if (retval < 0)
35         goto out;
36   //根据读入的文件头部,寻找可执行文件的处理函数
37     retval = search_binary_handler(bprm);
38 
39     ...
40 
41 } 
int search_binary_handler(struct linux_binprm *bprm)
{
    unsigned int depth = bprm->recursion_depth;
    int try,retval;
    struct linux_binfmt *fmt;
    pid_t old_pid, old_vpid;

        ...

    retval = security_bprm_check(bprm);//安全检查
    ...

//循环寻找可以解析当前可执行文件的代码
        list_for_each_entry(fmt, &formats, lh) {
            int (*fn)(struct linux_binprm *) = fmt->load_binary;
            ...
        }
        ...
    return retval;
}

   在找到可以解析当前可执行文件的代码之后,会调用start_kernel()开启一个新的进程进行可执行文件的执行。

  综合上述的理解,对于系统调用sys_execve的系统调用的主题流程为:sys_execve() -> do_execve() -> do_execve_common() -> search_binary_handler() ->start_thread()。

  以下是具体的验证过程。

图 4 断点设置

图 5 sys_execve() 跟踪结果1

图 6 sys_execve() 跟踪结果2

  3、Linux内核中进程调度的时机

  进程调度与中断关系密切。在OS中,中断主要有两类,一类是硬中断,也可以认为是外部中断,中断信号由外部产生,经CPU引脚传入内部,比较常用的是时钟中断、键盘、硬盘中断等;一类是软中断或异常,由OS内部产生,主要的类型包括系统调用、调试断点等特殊情况,导致CPU跳转至异常处理程序。具体来说,OS内部异常有3种,分别是故障(Fault)、退出(Abort)、陷阱(Trap)。故障(Fault):有问题,但是可以恢复至当前指令,常见的有缺页中断、除0错误等;退出(Abort):不可恢复的严重故障,导致程序无法运行,只能退出,比较常见的是连续发生故障;陷阱(Trap):程序主动产生的异常,在执行当前指令时发生,比较常见的是系统调用。

  Linux内核用过schedule函数实现进程的调度,换句话说,进程调度的实际,就是调用schedule函数的时机。调用schedule函数主要通过两种方式,一种是进程主动调用,比如进程需要等待某事件的发生,会主动调用schedule函数;一种是松散调用,内核代码可以随时调用schedule函数切换进程,比如中断处理程序或者某些内部线程。

  总体上来说,进程调度的时机主要有:用户进程通过特定的系统调用主动让出CPU;中断处理程序在内核返回用户态时进行调度;内核线程主动调用schedule函数让出CPU;中断处理程序主动调用schedule函数让出CPU,这类情况涵盖了上述中的第一种、第二种情况。

   schedule()函数源码如下【/linux-3.9.4/kernel/sched/core.c】。

1 asmlinkage void __sched schedule(void)
2 {
3     struct task_struct *tsk = current;
4 
5     sched_submit_work(tsk);
6     __schedule();
7 }

   sched_submit_work()函数源码如下【/linux-3.9.4/kernel/sched/core.c】。

 1 static inline void sched_submit_work(struct task_struct *tsk)
 2 {
 3     if (!tsk->state || tsk_is_pi_blocked(tsk))
 4         return;
 5     /*
 6      * If we are going to sleep and we have plugged IO queued,
 7      * make sure to submit it to avoid deadlocks.
 8      */
 9     if (blk_needs_flush_plug(tsk))
10         blk_schedule_flush_plug(tsk);
11 }

   __schedule()函数源码如下【/linux-3.9.4/kernel/sched/core.c】。

 1 static void __sched __schedule(void)
 2 {
 3     struct task_struct *prev, *next;
 4     unsigned long *switch_count;
 5     struct rq *rq;
 6     int cpu;
 7 
 8 need_resched:
 9     preempt_disable();
10     cpu = smp_processor_id();
11     rq = cpu_rq(cpu);
12     rcu_note_context_switch(cpu);
13     prev = rq->curr;
14 
15     schedule_debug(prev);
16 
17     if (sched_feat(HRTICK))
18         hrtick_clear(rq);
19 
20     raw_spin_lock_irq(&rq->lock);
21 
22     switch_count = &prev->nivcsw;
23     if (prev->state && !(preempt_count() & PREEMPT_ACTIVE)) {
24         if (unlikely(signal_pending_state(prev->state, prev))) {
25             prev->state = TASK_RUNNING;
26         } else {
27             deactivate_task(rq, prev, DEQUEUE_SLEEP);
28             prev->on_rq = 0;
29 
30             /*
31              * If a worker went to sleep, notify and ask workqueue
32              * whether it wants to wake up a task to maintain
33              * concurrency.
34              */
35             if (prev->flags & PF_WQ_WORKER) {
36                 struct task_struct *to_wakeup;
37 
38                 to_wakeup = wq_worker_sleeping(prev, cpu);
39                 if (to_wakeup)
40                     try_to_wake_up_local(to_wakeup);
41             }
42         }
43         switch_count = &prev->nvcsw;
44     }
45 
46     pre_schedule(rq, prev);
47 
48     if (unlikely(!rq->nr_running))
49         idle_balance(cpu, rq);
50 
51     put_prev_task(rq, prev);
52     next = pick_next_task(rq);
53     clear_tsk_need_resched(prev);
54     rq->skip_clock_update = 0;
55 
56     if (likely(prev != next)) {
57         rq->nr_switches++;
58         rq->curr = next;
59         ++*switch_count;
60 
61         context_switch(rq, prev, next); /* unlocks the rq */
62         /*
63          * The context switch have flipped the stack from under us
64          * and restored the local variables which were saved when
65          * this task called schedule() in the past. prev == current
66          * is still correct, but it can be moved to another cpu/rq.
67          */
68         cpu = smp_processor_id();
69         rq = cpu_rq(cpu);
70     } else
71         raw_spin_unlock_irq(&rq->lock);
72 
73     post_schedule(rq);
74 
75     sched_preempt_enable_no_resched();
76     if (need_resched())
77         goto need_resched;
78 }

  在__schedule()函数中,会调用context_switch()函数进行上下文切换,在context_switch()函数中会调用switch_to()函数进行实质上的进程切换。

  4、对于Linux内核中进程切换过程的理解

  以下是switch_to()函数的源码部分【/linux-3.9.4/arch/x86/include/asm/switch_to.h】

 1 #define switch_to(prev, next, last)                    \
 2 do {                                    \
 3     /*                                \
 4      * Context-switching clobbers all registers, so we clobber    \
 5      * them explicitly, via unused output variables.        \
 6      * (EAX and EBP is not listed because EBP is saved/restored    \
 7      * explicitly for wchan access and EAX is the return value of    \
 8      * __switch_to())                        \
 9      */                                \
10     unsigned long ebx, ecx, edx, esi, edi;                \
11                                     \
12     asm volatile("pushfl\n\t"        /* save    flags */    \
13              "pushl %%ebp\n\t"        /* save    EBP   */    \
14              "movl %%esp,%[prev_sp]\n\t"    /* save    ESP   */ \
15              "movl %[next_sp],%%esp\n\t"    /* restore ESP   */ \
16              "movl $1f,%[prev_ip]\n\t"    /* save    EIP   */    \
17              "pushl %[next_ip]\n\t"    /* restore EIP   */    \
18              __switch_canary                    \
19              "jmp __switch_to\n"    /* regparm call  */    \
20              "1:\t"                        \
21              "popl %%ebp\n\t"        /* restore EBP   */    \
22              "popfl\n"            /* restore flags */    \
23                                     \
24              /* output parameters */                \
25              : [prev_sp] "=m" (prev->thread.sp),        \
26                [prev_ip] "=m" (prev->thread.ip),        \
27                "=a" (last),                    \
28                                     \
29                /* clobbered output registers: */        \
30                "=b" (ebx), "=c" (ecx), "=d" (edx),        \
31                "=S" (esi), "=D" (edi)                \
32                                            \
33                __switch_canary_oparam                \
34                                     \
35                /* input parameters: */                \
36              : [next_sp]  "m" (next->thread.sp),        \
37                [next_ip]  "m" (next->thread.ip),        \
38                                            \
39                /* regparm parameters for __switch_to(): */    \
40                [prev]     "a" (prev),                \
41                [next]     "d" (next)                \
42                                     \
43                __switch_canary_iparam                \
44                                     \
45              : /* reloaded segment registers */            \
46             "memory");                    \
47 } while (0)

  通过对上述的源码我们知道,witch_to()函数才是本质上实现进程切换的函数。函数内保存了当前进程的ip、堆栈栈顶等信息,还将下一进程的相关信息取出并执行。 

四、总结与结论

  本次实验,总体上对于进程创建、可执行文件的装载和执行、进程调度时机、进程切换等方面有了一定程度的了解,但是,了解的还是不够深入,具体的细节还是存在着很多疑问,需要进一步的消化。

 

posted on 2019-03-26 18:54  Wangyt  阅读(1137)  评论(0编辑  收藏  举报