arch/x86/kernel/process_64.c

   1 /*
   2  *  Copyright (C) 1995  Linus Torvalds
   3  *
   4  *  Pentium III FXSR, SSE support
   5  *      Gareth Hughes <gareth@valinux.com>, May 2000
   6  *
   7  *  X86-64 port
   8  *      Andi Kleen.
   9  *
  10  *      CPU hotplug support - ashok.raj@intel.com
  11  */
  12
  13 /*
  14  * This file handles the architecture-dependent parts of process handling..
  15  */
  16
  17 #include <linux/stackprotector.h>
  18 #include <linux/cpu.h>
  19 #include <linux/errno.h>
  20 #include <linux/sched.h>
  21 #include <linux/fs.h>
  22 #include <linux/kernel.h>
  23 #include <linux/mm.h>
  24 #include <linux/elfcore.h>
  25 #include <linux/smp.h>
  26 #include <linux/slab.h>
  27 #include <linux/user.h>
  28 #include <linux/interrupt.h>
  29 #include <linux/delay.h>
  30 #include <linux/module.h>
  31 #include <linux/ptrace.h>
  32 #include <linux/notifier.h>
  33 #include <linux/kprobes.h>
  34 #include <linux/kdebug.h>
  35 #include <linux/tick.h>
  36 #include <linux/prctl.h>
  37 #include <linux/uaccess.h>
  38 #include <linux/io.h>
  39 #include <linux/ftrace.h>
  40 #include <linux/cpuidle.h>
  41
  42 #include <asm/pgtable.h>
  43 #include <asm/system.h>
  44 #include <asm/processor.h>
  45 #include <asm/i387.h>
  46 #include <asm/mmu_context.h>
  47 #include <asm/prctl.h>
  48 #include <asm/desc.h>
  49 #include <asm/proto.h>
  50 #include <asm/ia32.h>
  51 #include <asm/idle.h>
  52 #include <asm/syscalls.h>
  53 #include <asm/debugreg.h>
  54 #include <asm/nmi.h>
  55
  56 asmlinkage extern void ret_from_fork(void);
  57
  58 DEFINE_PER_CPU(unsigned long, old_rsp);
  59 static DEFINE_PER_CPU(unsigned char, is_idle);
  60
  61 static ATOMIC_NOTIFIER_HEAD(idle_notifier);
  62
  63 void idle_notifier_register(struct notifier_block *n)
  64 {
  65         atomic_notifier_chain_register(&idle_notifier, n);
  66 }
  67 EXPORT_SYMBOL_GPL(idle_notifier_register);
  68
  69 void idle_notifier_unregister(struct notifier_block *n)
  70 {
  71         atomic_notifier_chain_unregister(&idle_notifier, n);
  72 }
  73 EXPORT_SYMBOL_GPL(idle_notifier_unregister);
  74
  75 void enter_idle(void)
  76 {
  77         percpu_write(is_idle, 1);
  78         atomic_notifier_call_chain(&idle_notifier, IDLE_START, NULL);
  79 }
  80
  81 static void __exit_idle(void)
  82 {
  83         if (x86_test_and_clear_bit_percpu(0, is_idle) == 0)
  84                 return;
  85         atomic_notifier_call_chain(&idle_notifier, IDLE_END, NULL);
  86 }
  87
  88 /* Called from interrupts to signify idle end */
  89 void exit_idle(void)
  90 {
  91         /* idle loop has pid 0 */
  92         if (current->pid)
  93                 return;
  94         __exit_idle();
  95 }
  96
  97 #ifndef CONFIG_SMP
  98 static inline void play_dead(void)
  99 {
 100         BUG();
 101 }
 102 #endif
 103
 104 /*
 105  * The idle thread. There's no useful work to be
 106  * done, so just try to conserve power and have a
 107  * low exit latency (ie sit in a loop waiting for
 108  * somebody to say that they'd like to reschedule)
 109  */
 110 void cpu_idle(void)
 111 {
 112         current_thread_info()->status |= TS_POLLING;
 113
 114         /*
 115          * If we're the non-boot CPU, nothing set the stack canary up
 116          * for us.  CPU0 already has it initialized but no harm in
 117          * doing it again.  This is a good place for updating it, as
 118          * we wont ever return from this function (so the invalid
 119          * canaries already on the stack wont ever trigger).
 120          */
 121         boot_init_stack_canary();
 122
 123         /* endless idle loop with no priority at all */
 124         while (1) {
 125                 tick_nohz_idle_enter();
 126                 while (!need_resched()) {
 127
 128                         rmb();
 129
 130                         if (cpu_is_offline(smp_processor_id()))
 131                                 play_dead();
 132                         /*
 133                          * Idle routines should keep interrupts disabled
 134                          * from here on, until they go to idle.
 135                          * Otherwise, idle callbacks can misfire.
 136                          */
 137                         local_touch_nmi();
 138                         local_irq_disable();
 139                         enter_idle();
 140                         /* Don't trace irqs off for idle */
 141                         stop_critical_timings();
 142
 143                         /* enter_idle() needs rcu for notifiers */
 144                         rcu_idle_enter();
 145
 146                         if (cpuidle_idle_call())
 147                                 pm_idle();
 148
 149                         rcu_idle_exit();
 150                         start_critical_timings();
 151
 152                         /* In many cases the interrupt that ended idle
 153                            has already called exit_idle. But some idle
 154                            loops can be woken up without interrupt. */
 155                         __exit_idle();
 156                 }
 157
 158                 tick_nohz_idle_exit();
 159                 preempt_enable_no_resched();
 160                 schedule();
 161                 preempt_disable();
 162         }
 163 }
 164
 165 /* Prints also some state that isn't saved in the pt_regs */
 166 void __show_regs(struct pt_regs *regs, int all)
 167 {
 168         unsigned long cr0 = 0L, cr2 = 0L, cr3 = 0L, cr4 = 0L, fs, gs, shadowgs;
 169         unsigned long d0, d1, d2, d3, d6, d7;
 170         unsigned int fsindex, gsindex;
 171         unsigned int ds, cs, es;
 172
 173         show_regs_common();
 174         printk(KERN_DEFAULT "RIP: %04lx:[<%016lx>] ", regs->cs & 0xffff, regs->ip);
 175         printk_address(regs->ip, 1);
 176         printk(KERN_DEFAULT "RSP: %04lx:%016lx  EFLAGS: %08lx\n", regs->ss,
 177                         regs->sp, regs->flags);
 178         printk(KERN_DEFAULT "RAX: %016lx RBX: %016lx RCX: %016lx\n",
 179                regs->ax, regs->bx, regs->cx);
 180         printk(KERN_DEFAULT "RDX: %016lx RSI: %016lx RDI: %016lx\n",
 181                regs->dx, regs->si, regs->di);
 182         printk(KERN_DEFAULT "RBP: %016lx R08: %016lx R09: %016lx\n",
 183                regs->bp, regs->r8, regs->r9);
 184         printk(KERN_DEFAULT "R10: %016lx R11: %016lx R12: %016lx\n",
 185                regs->r10, regs->r11, regs->r12);
 186         printk(KERN_DEFAULT "R13: %016lx R14: %016lx R15: %016lx\n",
 187                regs->r13, regs->r14, regs->r15);
 188
 189         asm("movl %%ds,%0" : "=r" (ds));
 190         asm("movl %%cs,%0" : "=r" (cs));
 191         asm("movl %%es,%0" : "=r" (es));
 192         asm("movl %%fs,%0" : "=r" (fsindex));
 193         asm("movl %%gs,%0" : "=r" (gsindex));
 194
 195         rdmsrl(MSR_FS_BASE, fs);
 196         rdmsrl(MSR_GS_BASE, gs);
 197         rdmsrl(MSR_KERNEL_GS_BASE, shadowgs);
 198
 199         if (!all)
 200                 return;
 201
 202         cr0 = read_cr0();
 203         cr2 = read_cr2();
 204         cr3 = read_cr3();
 205         cr4 = read_cr4();
 206
 207         printk(KERN_DEFAULT "FS:  %016lx(%04x) GS:%016lx(%04x) knlGS:%016lx\n",
 208                fs, fsindex, gs, gsindex, shadowgs);
 209         printk(KERN_DEFAULT "CS:  %04x DS: %04x ES: %04x CR0: %016lx\n", cs, ds,
 210                         es, cr0);
 211         printk(KERN_DEFAULT "CR2: %016lx CR3: %016lx CR4: %016lx\n", cr2, cr3,
 212                         cr4);
 213
 214         get_debugreg(d0, 0);
 215         get_debugreg(d1, 1);
 216         get_debugreg(d2, 2);
 217         printk(KERN_DEFAULT "DR0: %016lx DR1: %016lx DR2: %016lx\n", d0, d1, d2);
 218         get_debugreg(d3, 3);
 219         get_debugreg(d6, 6);
 220         get_debugreg(d7, 7);
 221         printk(KERN_DEFAULT "DR3: %016lx DR6: %016lx DR7: %016lx\n", d3, d6, d7);
 222 }
 223
 224 void release_thread(struct task_struct *dead_task)
 225 {
 226         if (dead_task->mm) {
 227                 if (dead_task->mm->context.size) {
 228                         printk("WARNING: dead process %8s still has LDT? <%p/%d>\n",
 229                                         dead_task->comm,
 230                                         dead_task->mm->context.ldt,
 231                                         dead_task->mm->context.size);
 232                         BUG();
 233                 }
 234         }
 235 }
 236
 237 static inline void set_32bit_tls(struct task_struct *t, int tls, u32 addr)
 238 {
 239         struct user_desc ud = {
 240                 .base_addr = addr,
 241                 .limit = 0xfffff,
 242                 .seg_32bit = 1,
 243                 .limit_in_pages = 1,
 244                 .useable = 1,
 245         };
 246         struct desc_struct *desc = t->thread.tls_array;
 247         desc += tls;
 248         fill_ldt(desc, &ud);
 249 }
 250
 251 static inline u32 read_32bit_tls(struct task_struct *t, int tls)
 252 {
 253         return get_desc_base(&t->thread.tls_array[tls]);
 254 }
 255
 256 /*
 257  * This gets called before we allocate a new thread and copy
 258  * the current task into it.
 259  */
 260 void prepare_to_copy(struct task_struct *tsk)
 261 {
 262         unlazy_fpu(tsk);
 263 }
 264
 265 int copy_thread(unsigned long clone_flags, unsigned long sp,
 266                 unsigned long unused,
 267         struct task_struct *p, struct pt_regs *regs)
 268 {
 269         int err;
 270         struct pt_regs *childregs;
 271         struct task_struct *me = current;
 272
 273         childregs = ((struct pt_regs *)
 274                         (THREAD_SIZE + task_stack_page(p))) - 1;
 275         *childregs = *regs;
 276
 277         childregs->ax = 0;
 278         if (user_mode(regs))
 279                 childregs->sp = sp;
 280         else
 281                 childregs->sp = (unsigned long)childregs;
 282
 283         p->thread.sp = (unsigned long) childregs;
 284         p->thread.sp0 = (unsigned long) (childregs+1);
 285         p->thread.usersp = me->thread.usersp;
 286
 287         set_tsk_thread_flag(p, TIF_FORK);
 288
 289         p->fpu_counter = 0;
 290         p->thread.io_bitmap_ptr = NULL;
 291
 292         savesegment(gs, p->thread.gsindex);
 293         p->thread.gs = p->thread.gsindex ? 0 : me->thread.gs;
 294         savesegment(fs, p->thread.fsindex);
 295         p->thread.fs = p->thread.fsindex ? 0 : me->thread.fs;
 296         savesegment(es, p->thread.es);
 297         savesegment(ds, p->thread.ds);
 298
 299         err = -ENOMEM;
 300         memset(p->thread.ptrace_bps, 0, sizeof(p->thread.ptrace_bps));
 301
 302         if (unlikely(test_tsk_thread_flag(me, TIF_IO_BITMAP))) {
 303                 p->thread.io_bitmap_ptr = kmemdup(me->thread.io_bitmap_ptr,
 304                                                   IO_BITMAP_BYTES, GFP_KERNEL);
 305                 if (!p->thread.io_bitmap_ptr) {
 306                         p->thread.io_bitmap_max = 0;
 307                         return -ENOMEM;
 308                 }
 309                 set_tsk_thread_flag(p, TIF_IO_BITMAP);
 310         }
 311
 312         /*
 313          * Set a new TLS for the child thread?
 314          */
 315         if (clone_flags & CLONE_SETTLS) {
 316 #ifdef CONFIG_IA32_EMULATION
 317                 if (test_thread_flag(TIF_IA32))
 318                         err = do_set_thread_area(p, -1,
 319                                 (struct user_desc __user *)childregs->si, 0);
 320                 else
 321 #endif
 322                         err = do_arch_prctl(p, ARCH_SET_FS, childregs->r8);
 323                 if (err)
 324                         goto out;
 325         }
 326         err = 0;
 327 out:
 328         if (err && p->thread.io_bitmap_ptr) {
 329                 kfree(p->thread.io_bitmap_ptr);
 330                 p->thread.io_bitmap_max = 0;
 331         }
 332
 333         return err;
 334 }
 335
 336 static void
 337 start_thread_common(struct pt_regs *regs, unsigned long new_ip,
 338                     unsigned long new_sp,
 339                     unsigned int _cs, unsigned int _ss, unsigned int _ds)
 340 {
 341         loadsegment(fs, 0);
 342         loadsegment(es, _ds);
 343         loadsegment(ds, _ds);
 344         load_gs_index(0);
 345         regs->ip                = new_ip;
 346         regs->sp                = new_sp;
 347         percpu_write(old_rsp, new_sp);
 348         regs->cs                = _cs;
 349         regs->ss                = _ss;
 350         regs->flags             = X86_EFLAGS_IF;
 351         /*
 352          * Free the old FP and other extended state
 353          */
 354         free_thread_xstate(current);
 355 }
 356
 357 void
 358 start_thread(struct pt_regs *regs, unsigned long new_ip, unsigned long new_sp)
 359 {
 360         start_thread_common(regs, new_ip, new_sp,
 361                             __USER_CS, __USER_DS, 0);
 362 }
 363
 364 #ifdef CONFIG_IA32_EMULATION
 365 void start_thread_ia32(struct pt_regs *regs, u32 new_ip, u32 new_sp)
 366 {
 367         start_thread_common(regs, new_ip, new_sp,
 368                             __USER32_CS, __USER32_DS, __USER32_DS);
 369 }
 370 #endif
 371
 372 /*
 373  *      switch_to(x,y) should switch tasks from x to y.
 374  *
 375  * This could still be optimized:
 376  * - fold all the options into a flag word and test it with a single test.
 377  * - could test fs/gs bitsliced
 378  *
 379  * Kprobes not supported here. Set the probe on schedule instead.
 380  * Function graph tracer not supported too.
 381  */
 382 __notrace_funcgraph struct task_struct *
 383 __switch_to(struct task_struct *prev_p, struct task_struct *next_p)
 384 {
 385         struct thread_struct *prev = &prev_p->thread;
 386         struct thread_struct *next = &next_p->thread;
 387         int cpu = smp_processor_id();
 388         struct tss_struct *tss = &per_cpu(init_tss, cpu);
 389         unsigned fsindex, gsindex;
 390         fpu_switch_t fpu;
 391
 392         fpu = switch_fpu_prepare(prev_p, next_p);
 393
 394         /*
 395          * Reload esp0, LDT and the page table pointer:
 396          */
 397         load_sp0(tss, next);
 398
 399         /*
 400          * Switch DS and ES.
 401          * This won't pick up thread selector changes, but I guess that is ok.
 402          */
 403         savesegment(es, prev->es);
 404         if (unlikely(next->es | prev->es))
 405                 loadsegment(es, next->es);
 406
 407         savesegment(ds, prev->ds);
 408         if (unlikely(next->ds | prev->ds))
 409                 loadsegment(ds, next->ds);
 410
 411
 412         /* We must save %fs and %gs before load_TLS() because
 413          * %fs and %gs may be cleared by load_TLS().
 414          *
 415          * (e.g. xen_load_tls())
 416          */
 417         savesegment(fs, fsindex);
 418         savesegment(gs, gsindex);
 419
 420         load_TLS(next, cpu);
 421
 422         /*
 423          * Leave lazy mode, flushing any hypercalls made here.
 424          * This must be done before restoring TLS segments so
 425          * the GDT and LDT are properly updated, and must be
 426          * done before math_state_restore, so the TS bit is up
 427          * to date.
 428          */
 429         arch_end_context_switch(next_p);
 430
 431         /*
 432          * Switch FS and GS.
 433          *
 434          * Segment register != 0 always requires a reload.  Also
 435          * reload when it has changed.  When prev process used 64bit
 436          * base always reload to avoid an information leak.
 437          */
 438         if (unlikely(fsindex | next->fsindex | prev->fs)) {
 439                 loadsegment(fs, next->fsindex);
 440                 /*
 441                  * Check if the user used a selector != 0; if yes
 442                  *  clear 64bit base, since overloaded base is always
 443                  *  mapped to the Null selector
 444                  */
 445                 if (fsindex)
 446                         prev->fs = 0;
 447         }
 448         /* when next process has a 64bit base use it */
 449         if (next->fs)
 450                 wrmsrl(MSR_FS_BASE, next->fs);
 451         prev->fsindex = fsindex;
 452
 453         if (unlikely(gsindex | next->gsindex | prev->gs)) {
 454                 load_gs_index(next->gsindex);
 455                 if (gsindex)
 456                         prev->gs = 0;
 457         }
 458         if (next->gs)
 459                 wrmsrl(MSR_KERNEL_GS_BASE, next->gs);
 460         prev->gsindex = gsindex;
 461
 462         switch_fpu_finish(next_p, fpu);
 463
 464         /*
 465          * Switch the PDA and FPU contexts.
 466          */
 467         prev->usersp = percpu_read(old_rsp);
 468         percpu_write(old_rsp, next->usersp);
 469         percpu_write(current_task, next_p);
 470
 471         percpu_write(kernel_stack,
 472                   (unsigned long)task_stack_page(next_p) +
 473                   THREAD_SIZE - KERNEL_STACK_OFFSET);
 474
 475         /*
 476          * Now maybe reload the debug registers and handle I/O bitmaps
 477          */
 478         if (unlikely(task_thread_info(next_p)->flags & _TIF_WORK_CTXSW_NEXT ||
 479                      task_thread_info(prev_p)->flags & _TIF_WORK_CTXSW_PREV))
 480                 __switch_to_xtra(prev_p, next_p, tss);
 481
 482         return prev_p;
 483 }
 484
 485 void set_personality_64bit(void)
 486 {
 487         /* inherit personality from parent */
 488
 489         /* Make sure to be in 64bit mode */
 490         clear_thread_flag(TIF_IA32);
 491
 492         /* Ensure the corresponding mm is not marked. */
 493         if (current->mm)
 494                 current->mm->context.ia32_compat = 0;
 495
 496         /* TBD: overwrites user setup. Should have two bits.
 497            But 64bit processes have always behaved this way,
 498            so it's not too bad. The main problem is just that
 499            32bit childs are affected again. */
 500         current->personality &= ~READ_IMPLIES_EXEC;
 501 }
 502
 503 void set_personality_ia32(void)
 504 {
 505         /* inherit personality from parent */
 506
 507         /* Make sure to be in 32bit mode */
 508         set_thread_flag(TIF_IA32);
 509         current->personality |= force_personality32;
 510
 511         /* Mark the associated mm as containing 32-bit tasks. */
 512         if (current->mm)
 513                 current->mm->context.ia32_compat = 1;
 514
 515         /* Prepare the first "return" to user space */
 516         current_thread_info()->status |= TS_COMPAT;
 517 }
 518
 519 unsigned long get_wchan(struct task_struct *p)
 520 {
 521         unsigned long stack;
 522         u64 fp, ip;
 523         int count = 0;
 524
 525         if (!p || p == current || p->state == TASK_RUNNING)
 526                 return 0;
 527         stack = (unsigned long)task_stack_page(p);
 528         if (p->thread.sp < stack || p->thread.sp >= stack+THREAD_SIZE)
 529                 return 0;
 530         fp = *(u64 *)(p->thread.sp);
 531         do {
 532                 if (fp < (unsigned long)stack ||
 533                     fp >= (unsigned long)stack+THREAD_SIZE)
 534                         return 0;
 535                 ip = *(u64 *)(fp+8);
 536                 if (!in_sched_functions(ip))
 537                         return ip;
 538                 fp = *(u64 *)fp;
 539         } while (count++ < 16);
 540         return 0;
 541 }
 542
 543 long do_arch_prctl(struct task_struct *task, int code, unsigned long addr)
 544 {
 545         int ret = 0;
 546         int doit = task == current;
 547         int cpu;
 548
 549         switch (code) {
 550         case ARCH_SET_GS:
 551                 if (addr >= TASK_SIZE_OF(task))
 552                         return -EPERM;
 553                 cpu = get_cpu();
 554                 /* handle small bases via the GDT because that's faster to
 555                    switch. */
 556                 if (addr <= 0xffffffff) {
 557                         set_32bit_tls(task, GS_TLS, addr);
 558                         if (doit) {
 559                                 load_TLS(&task->thread, cpu);
 560                                 load_gs_index(GS_TLS_SEL);
 561                         }
 562                         task->thread.gsindex = GS_TLS_SEL;
 563                         task->thread.gs = 0;
 564                 } else {
 565                         task->thread.gsindex = 0;
 566                         task->thread.gs = addr;
 567                         if (doit) {
 568                                 load_gs_index(0);
 569                                 ret = checking_wrmsrl(MSR_KERNEL_GS_BASE, addr);
 570                         }
 571                 }
 572                 put_cpu();
 573                 break;
 574         case ARCH_SET_FS:
 575                 /* Not strictly needed for fs, but do it for symmetry
 576                    with gs */
 577                 if (addr >= TASK_SIZE_OF(task))
 578                         return -EPERM;
 579                 cpu = get_cpu();
 580                 /* handle small bases via the GDT because that's faster to
 581                    switch. */
 582                 if (addr <= 0xffffffff) {
 583                         set_32bit_tls(task, FS_TLS, addr);
 584                         if (doit) {
 585                                 load_TLS(&task->thread, cpu);
 586                                 loadsegment(fs, FS_TLS_SEL);
 587                         }
 588                         task->thread.fsindex = FS_TLS_SEL;
 589                         task->thread.fs = 0;
 590                 } else {
 591                         task->thread.fsindex = 0;
 592                         task->thread.fs = addr;
 593                         if (doit) {
 594                                 /* set the selector to 0 to not confuse
 595                                    __switch_to */
 596                                 loadsegment(fs, 0);
 597                                 ret = checking_wrmsrl(MSR_FS_BASE, addr);
 598                         }
 599                 }
 600                 put_cpu();
 601                 break;
 602         case ARCH_GET_FS: {
 603                 unsigned long base;
 604                 if (task->thread.fsindex == FS_TLS_SEL)
 605                         base = read_32bit_tls(task, FS_TLS);
 606                 else if (doit)
 607                         rdmsrl(MSR_FS_BASE, base);
 608                 else
 609                         base = task->thread.fs;
 610                 ret = put_user(base, (unsigned long __user *)addr);
 611                 break;
 612         }
 613         case ARCH_GET_GS: {
 614                 unsigned long base;
 615                 unsigned gsindex;
 616                 if (task->thread.gsindex == GS_TLS_SEL)
 617                         base = read_32bit_tls(task, GS_TLS);
 618                 else if (doit) {
 619                         savesegment(gs, gsindex);
 620                         if (gsindex)
 621                                 rdmsrl(MSR_KERNEL_GS_BASE, base);
 622                         else
 623                                 base = task->thread.gs;
 624                 } else
 625                         base = task->thread.gs;
 626                 ret = put_user(base, (unsigned long __user *)addr);
 627                 break;
 628         }
 629
 630         default:
 631                 ret = -EINVAL;
 632                 break;
 633         }
 634
 635         return ret;
 636 }
 637
 638 long sys_arch_prctl(int code, unsigned long addr)
 639 {
 640         return do_arch_prctl(current, code, addr);
 641 }
 642
 643 unsigned long KSTK_ESP(struct task_struct *task)
 644 {
 645         return (test_tsk_thread_flag(task, TIF_IA32)) ?
 646                         (task_pt_regs(task)->sp) : ((task)->thread.usersp);
 647 }