arch/x86/include/asm/switch_to.h

   1 #ifndef _ASM_X86_SWITCH_TO_H
   2 #define _ASM_X86_SWITCH_TO_H
   3
   4 struct task_struct; /* one of the stranger aspects of C forward declarations */
   5 __visible struct task_struct *__switch_to(struct task_struct *prev,
   6                                            struct task_struct *next);
   7 struct tss_struct;
   8 void __switch_to_xtra(struct task_struct *prev_p, struct task_struct *next_p,
   9                       struct tss_struct *tss);
  10
  11 #ifdef CONFIG_X86_32
  12
  13 #ifdef CONFIG_CC_STACKPROTECTOR
  14 #define __switch_canary                                                 \
  15         "movl %P[task_canary](%[next]), %%ebx\n\t"                      \
  16         "movl %%ebx, "__percpu_arg([stack_canary])"\n\t"
  17 #define __switch_canary_oparam                                          \
  18         , [stack_canary] "=m" (stack_canary.canary)
  19 #define __switch_canary_iparam                                          \
  20         , [task_canary] "i" (offsetof(struct task_struct, stack_canary))
  21 #else   /* CC_STACKPROTECTOR */
  22 #define __switch_canary
  23 #define __switch_canary_oparam
  24 #define __switch_canary_iparam
  25 #endif  /* CC_STACKPROTECTOR */
  26
  27 /*
  28  * Saving eflags is important. It switches not only IOPL between tasks,
  29  * it also protects other tasks from NT leaking through sysenter etc.
  30  */
  31 #define switch_to(prev, next, last)                                     \
  32 do {                                                                    \
  33         /*                                                              \
  34          * Context-switching clobbers all registers, so we clobber      \
  35          * them explicitly, via unused output variables.                \
  36          * (EAX and EBP is not listed because EBP is saved/restored     \
  37          * explicitly for wchan access and EAX is the return value of   \
  38          * __switch_to())                                               \
  39          */                                                             \
  40         unsigned long ebx, ecx, edx, esi, edi;                          \
  41                                                                         \
  42         asm volatile("pushfl\n\t"               /* save    flags */     \
  43                      "pushl %%ebp\n\t"          /* save    EBP   */     \
  44                      "movl %%esp,%[prev_sp]\n\t"        /* save    ESP   */ \
  45                      "movl %[next_sp],%%esp\n\t"        /* restore ESP   */ \
  46                      "movl $1f,%[prev_ip]\n\t"  /* save    EIP   */     \
  47                      "pushl %[next_ip]\n\t"     /* restore EIP   */     \
  48                      __switch_canary                                    \
  49                      "jmp __switch_to\n"        /* regparm call  */     \
  50                      "1:\t"                                             \
  51                      "popl %%ebp\n\t"           /* restore EBP   */     \
  52                      "popfl\n"                  /* restore flags */     \
  53                                                                         \
  54                      /* output parameters */                            \
  55                      : [prev_sp] "=m" (prev->thread.sp),                \
  56                        [prev_ip] "=m" (prev->thread.ip),                \
  57                        "=a" (last),                                     \
  58                                                                         \
  59                        /* clobbered output registers: */                \
  60                        "=b" (ebx), "=c" (ecx), "=d" (edx),              \
  61                        "=S" (esi), "=D" (edi)                           \
  62                                                                         \
  63                        __switch_canary_oparam                           \
  64                                                                         \
  65                        /* input parameters: */                          \
  66                      : [next_sp]  "m" (next->thread.sp),                \
  67                        [next_ip]  "m" (next->thread.ip),                \
  68                                                                         \
  69                        /* regparm parameters for __switch_to(): */      \
  70                        [prev]     "a" (prev),                           \
  71                        [next]     "d" (next)                            \
  72                                                                         \
  73                        __switch_canary_iparam                           \
  74                                                                         \
  75                      : /* reloaded segment registers */                 \
  76                         "memory");                                      \
  77 } while (0)
  78
  79 #else /* CONFIG_X86_32 */
  80
  81 /* frame pointer must be last for get_wchan */
  82 #define SAVE_CONTEXT    "pushf ; pushq %%rbp ; movq %%rsi,%%rbp\n\t"
  83 #define RESTORE_CONTEXT "movq %%rbp,%%rsi ; popq %%rbp ; popf\t"
  84
  85 #define __EXTRA_CLOBBER  \
  86         , "rcx", "rbx", "rdx", "r8", "r9", "r10", "r11", \
  87           "r12", "r13", "r14", "r15"
  88
  89 #ifdef CONFIG_CC_STACKPROTECTOR
  90 #define __switch_canary                                                   \
  91         "movq %P[task_canary](%%rsi),%%r8\n\t"                            \
  92         "movq %%r8,"__percpu_arg([gs_canary])"\n\t"
  93 #define __switch_canary_oparam                                            \
  94         , [gs_canary] "=m" (irq_stack_union.stack_canary)
  95 #define __switch_canary_iparam                                            \
  96         , [task_canary] "i" (offsetof(struct task_struct, stack_canary))
  97 #else   /* CC_STACKPROTECTOR */
  98 #define __switch_canary
  99 #define __switch_canary_oparam
 100 #define __switch_canary_iparam
 101 #endif  /* CC_STACKPROTECTOR */
 102
 103 /* Save restore flags to clear handle leaking NT */
 104 #define switch_to(prev, next, last) \
 105         asm volatile(SAVE_CONTEXT                                         \
 106              "movq %%rsp,%P[threadrsp](%[prev])\n\t" /* save RSP */       \
 107              "movq %P[threadrsp](%[next]),%%rsp\n\t" /* restore RSP */    \
 108              "call __switch_to\n\t"                                       \
 109              "movq "__percpu_arg([current_task])",%%rsi\n\t"              \
 110              __switch_canary                                              \
 111              "movq %P[thread_info](%%rsi),%%r8\n\t"                       \
 112              "movq %%rax,%%rdi\n\t"                                       \
 113              "testl  %[_tif_fork],%P[ti_flags](%%r8)\n\t"                 \
 114              "jnz   ret_from_fork\n\t"                                    \
 115              RESTORE_CONTEXT                                              \
 116              : "=a" (last)                                                \
 117                __switch_canary_oparam                                     \
 118              : [next] "S" (next), [prev] "D" (prev),                      \
 119                [threadrsp] "i" (offsetof(struct task_struct, thread.sp)), \
 120                [ti_flags] "i" (offsetof(struct thread_info, flags)),      \
 121                [_tif_fork] "i" (_TIF_FORK),                               \
 122                [thread_info] "i" (offsetof(struct task_struct, stack)),   \
 123                [current_task] "m" (current_task)                          \
 124                __switch_canary_iparam                                     \
 125              : "memory", "cc" __EXTRA_CLOBBER)
 126
 127 #endif /* CONFIG_X86_32 */
 128
 129 #endif /* _ASM_X86_SWITCH_TO_H */