arch/x86/kernel/ptrace.c

   1 /* By Ross Biro 1/23/92 */
   2 /*
   3  * Pentium III FXSR, SSE support
   4  *      Gareth Hughes <gareth@valinux.com>, May 2000
   5  *
   6  * BTS tracing
   7  *      Markus Metzger <markus.t.metzger@intel.com>, Dec 2007
   8  */
   9
  10 #include <linux/kernel.h>
  11 #include <linux/sched.h>
  12 #include <linux/mm.h>
  13 #include <linux/smp.h>
  14 #include <linux/errno.h>
  15 #include <linux/ptrace.h>
  16 #include <linux/regset.h>
  17 #include <linux/tracehook.h>
  18 #include <linux/user.h>
  19 #include <linux/elf.h>
  20 #include <linux/security.h>
  21 #include <linux/audit.h>
  22 #include <linux/seccomp.h>
  23 #include <linux/signal.h>
  24 #include <linux/workqueue.h>
  25 #include <linux/perf_event.h>
  26 #include <linux/hw_breakpoint.h>
  27
  28 #include <asm/uaccess.h>
  29 #include <asm/pgtable.h>
  30 #include <asm/system.h>
  31 #include <asm/processor.h>
  32 #include <asm/i387.h>
  33 #include <asm/debugreg.h>
  34 #include <asm/ldt.h>
  35 #include <asm/desc.h>
  36 #include <asm/prctl.h>
  37 #include <asm/proto.h>
  38 #include <asm/ds.h>
  39 #include <asm/hw_breakpoint.h>
  40
  41 #include "tls.h"
  42
  43 #define CREATE_TRACE_POINTS
  44 #include <trace/events/syscalls.h>
  45
  46 enum x86_regset {
  47         REGSET_GENERAL,
  48         REGSET_FP,
  49         REGSET_XFP,
  50         REGSET_IOPERM64 = REGSET_XFP,
  51         REGSET_TLS,
  52         REGSET_IOPERM32,
  53 };
  54
  55 struct pt_regs_offset {
  56         const char *name;
  57         int offset;
  58 };
  59
  60 #define REG_OFFSET_NAME(r) {.name = #r, .offset = offsetof(struct pt_regs, r)}
  61 #define REG_OFFSET_END {.name = NULL, .offset = 0}
  62
  63 static const struct pt_regs_offset regoffset_table[] = {
  64 #ifdef CONFIG_X86_64
  65         REG_OFFSET_NAME(r15),
  66         REG_OFFSET_NAME(r14),
  67         REG_OFFSET_NAME(r13),
  68         REG_OFFSET_NAME(r12),
  69         REG_OFFSET_NAME(r11),
  70         REG_OFFSET_NAME(r10),
  71         REG_OFFSET_NAME(r9),
  72         REG_OFFSET_NAME(r8),
  73 #endif
  74         REG_OFFSET_NAME(bx),
  75         REG_OFFSET_NAME(cx),
  76         REG_OFFSET_NAME(dx),
  77         REG_OFFSET_NAME(si),
  78         REG_OFFSET_NAME(di),
  79         REG_OFFSET_NAME(bp),
  80         REG_OFFSET_NAME(ax),
  81 #ifdef CONFIG_X86_32
  82         REG_OFFSET_NAME(ds),
  83         REG_OFFSET_NAME(es),
  84         REG_OFFSET_NAME(fs),
  85         REG_OFFSET_NAME(gs),
  86 #endif
  87         REG_OFFSET_NAME(orig_ax),
  88         REG_OFFSET_NAME(ip),
  89         REG_OFFSET_NAME(cs),
  90         REG_OFFSET_NAME(flags),
  91         REG_OFFSET_NAME(sp),
  92         REG_OFFSET_NAME(ss),
  93         REG_OFFSET_END,
  94 };
  95
  96 /**
  97  * regs_query_register_offset() - query register offset from its name
  98  * @name:       the name of a register
  99  *
 100  * regs_query_register_offset() returns the offset of a register in struct
 101  * pt_regs from its name. If the name is invalid, this returns -EINVAL;
 102  */
 103 int regs_query_register_offset(const char *name)
 104 {
 105         const struct pt_regs_offset *roff;
 106         for (roff = regoffset_table; roff->name != NULL; roff++)
 107                 if (!strcmp(roff->name, name))
 108                         return roff->offset;
 109         return -EINVAL;
 110 }
 111
 112 /**
 113  * regs_query_register_name() - query register name from its offset
 114  * @offset:     the offset of a register in struct pt_regs.
 115  *
 116  * regs_query_register_name() returns the name of a register from its
 117  * offset in struct pt_regs. If the @offset is invalid, this returns NULL;
 118  */
 119 const char *regs_query_register_name(unsigned int offset)
 120 {
 121         const struct pt_regs_offset *roff;
 122         for (roff = regoffset_table; roff->name != NULL; roff++)
 123                 if (roff->offset == offset)
 124                         return roff->name;
 125         return NULL;
 126 }
 127
 128 static const int arg_offs_table[] = {
 129 #ifdef CONFIG_X86_32
 130         [0] = offsetof(struct pt_regs, ax),
 131         [1] = offsetof(struct pt_regs, dx),
 132         [2] = offsetof(struct pt_regs, cx)
 133 #else /* CONFIG_X86_64 */
 134         [0] = offsetof(struct pt_regs, di),
 135         [1] = offsetof(struct pt_regs, si),
 136         [2] = offsetof(struct pt_regs, dx),
 137         [3] = offsetof(struct pt_regs, cx),
 138         [4] = offsetof(struct pt_regs, r8),
 139         [5] = offsetof(struct pt_regs, r9)
 140 #endif
 141 };
 142
 143 /**
 144  * regs_get_argument_nth() - get Nth argument at function call
 145  * @regs:       pt_regs which contains registers at function entry.
 146  * @n:          argument number.
 147  *
 148  * regs_get_argument_nth() returns @n th argument of a function call.
 149  * Since usually the kernel stack will be changed right after function entry,
 150  * you must use this at function entry. If the @n th entry is NOT in the
 151  * kernel stack or pt_regs, this returns 0.
 152  */
 153 unsigned long regs_get_argument_nth(struct pt_regs *regs, unsigned int n)
 154 {
 155         if (n < ARRAY_SIZE(arg_offs_table))
 156                 return *(unsigned long *)((char *)regs + arg_offs_table[n]);
 157         else {
 158                 /*
 159                  * The typical case: arg n is on the stack.
 160                  * (Note: stack[0] = return address, so skip it)
 161                  */
 162                 n -= ARRAY_SIZE(arg_offs_table);
 163                 return regs_get_kernel_stack_nth(regs, 1 + n);
 164         }
 165 }
 166
 167 /*
 168  * does not yet catch signals sent when the child dies.
 169  * in exit.c or in signal.c.
 170  */
 171
 172 /*
 173  * Determines which flags the user has access to [1 = access, 0 = no access].
 174  */
 175 #define FLAG_MASK_32            ((unsigned long)                        \
 176                                  (X86_EFLAGS_CF | X86_EFLAGS_PF |       \
 177                                   X86_EFLAGS_AF | X86_EFLAGS_ZF |       \
 178                                   X86_EFLAGS_SF | X86_EFLAGS_TF |       \
 179                                   X86_EFLAGS_DF | X86_EFLAGS_OF |       \
 180                                   X86_EFLAGS_RF | X86_EFLAGS_AC))
 181
 182 /*
 183  * Determines whether a value may be installed in a segment register.
 184  */
 185 static inline bool invalid_selector(u16 value)
 186 {
 187         return unlikely(value != 0 && (value & SEGMENT_RPL_MASK) != USER_RPL);
 188 }
 189
 190 #ifdef CONFIG_X86_32
 191
 192 #define FLAG_MASK               FLAG_MASK_32
 193
 194 static unsigned long *pt_regs_access(struct pt_regs *regs, unsigned long regno)
 195 {
 196         BUILD_BUG_ON(offsetof(struct pt_regs, bx) != 0);
 197         return &regs->bx + (regno >> 2);
 198 }
 199
 200 static u16 get_segment_reg(struct task_struct *task, unsigned long offset)
 201 {
 202         /*
 203          * Returning the value truncates it to 16 bits.
 204          */
 205         unsigned int retval;
 206         if (offset != offsetof(struct user_regs_struct, gs))
 207                 retval = *pt_regs_access(task_pt_regs(task), offset);
 208         else {
 209                 if (task == current)
 210                         retval = get_user_gs(task_pt_regs(task));
 211                 else
 212                         retval = task_user_gs(task);
 213         }
 214         return retval;
 215 }
 216
 217 static int set_segment_reg(struct task_struct *task,
 218                            unsigned long offset, u16 value)
 219 {
 220         /*
 221          * The value argument was already truncated to 16 bits.
 222          */
 223         if (invalid_selector(value))
 224                 return -EIO;
 225
 226         /*
 227          * For %cs and %ss we cannot permit a null selector.
 228          * We can permit a bogus selector as long as it has USER_RPL.
 229          * Null selectors are fine for other segment registers, but
 230          * we will never get back to user mode with invalid %cs or %ss
 231          * and will take the trap in iret instead.  Much code relies
 232          * on user_mode() to distinguish a user trap frame (which can
 233          * safely use invalid selectors) from a kernel trap frame.
 234          */
 235         switch (offset) {
 236         case offsetof(struct user_regs_struct, cs):
 237         case offsetof(struct user_regs_struct, ss):
 238                 if (unlikely(value == 0))
 239                         return -EIO;
 240
 241         default:
 242                 *pt_regs_access(task_pt_regs(task), offset) = value;
 243                 break;
 244
 245         case offsetof(struct user_regs_struct, gs):
 246                 if (task == current)
 247                         set_user_gs(task_pt_regs(task), value);
 248                 else
 249                         task_user_gs(task) = value;
 250         }
 251
 252         return 0;
 253 }
 254
 255 #else  /* CONFIG_X86_64 */
 256
 257 #define FLAG_MASK               (FLAG_MASK_32 | X86_EFLAGS_NT)
 258
 259 static unsigned long *pt_regs_access(struct pt_regs *regs, unsigned long offset)
 260 {
 261         BUILD_BUG_ON(offsetof(struct pt_regs, r15) != 0);
 262         return &regs->r15 + (offset / sizeof(regs->r15));
 263 }
 264
 265 static u16 get_segment_reg(struct task_struct *task, unsigned long offset)
 266 {
 267         /*
 268          * Returning the value truncates it to 16 bits.
 269          */
 270         unsigned int seg;
 271
 272         switch (offset) {
 273         case offsetof(struct user_regs_struct, fs):
 274                 if (task == current) {
 275                         /* Older gas can't assemble movq %?s,%r?? */
 276                         asm("movl %%fs,%0" : "=r" (seg));
 277                         return seg;
 278                 }
 279                 return task->thread.fsindex;
 280         case offsetof(struct user_regs_struct, gs):
 281                 if (task == current) {
 282                         asm("movl %%gs,%0" : "=r" (seg));
 283                         return seg;
 284                 }
 285                 return task->thread.gsindex;
 286         case offsetof(struct user_regs_struct, ds):
 287                 if (task == current) {
 288                         asm("movl %%ds,%0" : "=r" (seg));
 289                         return seg;
 290                 }
 291                 return task->thread.ds;
 292         case offsetof(struct user_regs_struct, es):
 293                 if (task == current) {
 294                         asm("movl %%es,%0" : "=r" (seg));
 295                         return seg;
 296                 }
 297                 return task->thread.es;
 298
 299         case offsetof(struct user_regs_struct, cs):
 300         case offsetof(struct user_regs_struct, ss):
 301                 break;
 302         }
 303         return *pt_regs_access(task_pt_regs(task), offset);
 304 }
 305
 306 static int set_segment_reg(struct task_struct *task,
 307                            unsigned long offset, u16 value)
 308 {
 309         /*
 310          * The value argument was already truncated to 16 bits.
 311          */
 312         if (invalid_selector(value))
 313                 return -EIO;
 314
 315         switch (offset) {
 316         case offsetof(struct user_regs_struct,fs):
 317                 /*
 318                  * If this is setting fs as for normal 64-bit use but
 319                  * setting fs_base has implicitly changed it, leave it.
 320                  */
 321                 if ((value == FS_TLS_SEL && task->thread.fsindex == 0 &&
 322                      task->thread.fs != 0) ||
 323                     (value == 0 && task->thread.fsindex == FS_TLS_SEL &&
 324                      task->thread.fs == 0))
 325                         break;
 326                 task->thread.fsindex = value;
 327                 if (task == current)
 328                         loadsegment(fs, task->thread.fsindex);
 329                 break;
 330         case offsetof(struct user_regs_struct,gs):
 331                 /*
 332                  * If this is setting gs as for normal 64-bit use but
 333                  * setting gs_base has implicitly changed it, leave it.
 334                  */
 335                 if ((value == GS_TLS_SEL && task->thread.gsindex == 0 &&
 336                      task->thread.gs != 0) ||
 337                     (value == 0 && task->thread.gsindex == GS_TLS_SEL &&
 338                      task->thread.gs == 0))
 339                         break;
 340                 task->thread.gsindex = value;
 341                 if (task == current)
 342                         load_gs_index(task->thread.gsindex);
 343                 break;
 344         case offsetof(struct user_regs_struct,ds):
 345                 task->thread.ds = value;
 346                 if (task == current)
 347                         loadsegment(ds, task->thread.ds);
 348                 break;
 349         case offsetof(struct user_regs_struct,es):
 350                 task->thread.es = value;
 351                 if (task == current)
 352                         loadsegment(es, task->thread.es);
 353                 break;
 354
 355                 /*
 356                  * Can't actually change these in 64-bit mode.
 357                  */
 358         case offsetof(struct user_regs_struct,cs):
 359                 if (unlikely(value == 0))
 360                         return -EIO;
 361 #ifdef CONFIG_IA32_EMULATION
 362                 if (test_tsk_thread_flag(task, TIF_IA32))
 363                         task_pt_regs(task)->cs = value;
 364 #endif
 365                 break;
 366         case offsetof(struct user_regs_struct,ss):
 367                 if (unlikely(value == 0))
 368                         return -EIO;
 369 #ifdef CONFIG_IA32_EMULATION
 370                 if (test_tsk_thread_flag(task, TIF_IA32))
 371                         task_pt_regs(task)->ss = value;
 372 #endif
 373                 break;
 374         }
 375
 376         return 0;
 377 }
 378
 379 #endif  /* CONFIG_X86_32 */
 380
 381 static unsigned long get_flags(struct task_struct *task)
 382 {
 383         unsigned long retval = task_pt_regs(task)->flags;
 384
 385         /*
 386          * If the debugger set TF, hide it from the readout.
 387          */
 388         if (test_tsk_thread_flag(task, TIF_FORCED_TF))
 389                 retval &= ~X86_EFLAGS_TF;
 390
 391         return retval;
 392 }
 393
 394 static int set_flags(struct task_struct *task, unsigned long value)
 395 {
 396         struct pt_regs *regs = task_pt_regs(task);
 397
 398         /*
 399          * If the user value contains TF, mark that
 400          * it was not "us" (the debugger) that set it.
 401          * If not, make sure it stays set if we had.
 402          */
 403         if (value & X86_EFLAGS_TF)
 404                 clear_tsk_thread_flag(task, TIF_FORCED_TF);
 405         else if (test_tsk_thread_flag(task, TIF_FORCED_TF))
 406                 value |= X86_EFLAGS_TF;
 407
 408         regs->flags = (regs->flags & ~FLAG_MASK) | (value & FLAG_MASK);
 409
 410         return 0;
 411 }
 412
 413 static int putreg(struct task_struct *child,
 414                   unsigned long offset, unsigned long value)
 415 {
 416         switch (offset) {
 417         case offsetof(struct user_regs_struct, cs):
 418         case offsetof(struct user_regs_struct, ds):
 419         case offsetof(struct user_regs_struct, es):
 420         case offsetof(struct user_regs_struct, fs):
 421         case offsetof(struct user_regs_struct, gs):
 422         case offsetof(struct user_regs_struct, ss):
 423                 return set_segment_reg(child, offset, value);
 424
 425         case offsetof(struct user_regs_struct, flags):
 426                 return set_flags(child, value);
 427
 428 #ifdef CONFIG_X86_64
 429         case offsetof(struct user_regs_struct,fs_base):
 430                 if (value >= TASK_SIZE_OF(child))
 431                         return -EIO;
 432                 /*
 433                  * When changing the segment base, use do_arch_prctl
 434                  * to set either thread.fs or thread.fsindex and the
 435                  * corresponding GDT slot.
 436                  */
 437                 if (child->thread.fs != value)
 438                         return do_arch_prctl(child, ARCH_SET_FS, value);
 439                 return 0;
 440         case offsetof(struct user_regs_struct,gs_base):
 441                 /*
 442                  * Exactly the same here as the %fs handling above.
 443                  */
 444                 if (value >= TASK_SIZE_OF(child))
 445                         return -EIO;
 446                 if (child->thread.gs != value)
 447                         return do_arch_prctl(child, ARCH_SET_GS, value);
 448                 return 0;
 449 #endif
 450         }
 451
 452         *pt_regs_access(task_pt_regs(child), offset) = value;
 453         return 0;
 454 }
 455
 456 static unsigned long getreg(struct task_struct *task, unsigned long offset)
 457 {
 458         switch (offset) {
 459         case offsetof(struct user_regs_struct, cs):
 460         case offsetof(struct user_regs_struct, ds):
 461         case offsetof(struct user_regs_struct, es):
 462         case offsetof(struct user_regs_struct, fs):
 463         case offsetof(struct user_regs_struct, gs):
 464         case offsetof(struct user_regs_struct, ss):
 465                 return get_segment_reg(task, offset);
 466
 467         case offsetof(struct user_regs_struct, flags):
 468                 return get_flags(task);
 469
 470 #ifdef CONFIG_X86_64
 471         case offsetof(struct user_regs_struct, fs_base): {
 472                 /*
 473                  * do_arch_prctl may have used a GDT slot instead of
 474                  * the MSR.  To userland, it appears the same either
 475                  * way, except the %fs segment selector might not be 0.
 476                  */
 477                 unsigned int seg = task->thread.fsindex;
 478                 if (task->thread.fs != 0)
 479                         return task->thread.fs;
 480                 if (task == current)
 481                         asm("movl %%fs,%0" : "=r" (seg));
 482                 if (seg != FS_TLS_SEL)
 483                         return 0;
 484                 return get_desc_base(&task->thread.tls_array[FS_TLS]);
 485         }
 486         case offsetof(struct user_regs_struct, gs_base): {
 487                 /*
 488                  * Exactly the same here as the %fs handling above.
 489                  */
 490                 unsigned int seg = task->thread.gsindex;
 491                 if (task->thread.gs != 0)
 492                         return task->thread.gs;
 493                 if (task == current)
 494                         asm("movl %%gs,%0" : "=r" (seg));
 495                 if (seg != GS_TLS_SEL)
 496                         return 0;
 497                 return get_desc_base(&task->thread.tls_array[GS_TLS]);
 498         }
 499 #endif
 500         }
 501
 502         return *pt_regs_access(task_pt_regs(task), offset);
 503 }
 504
 505 static int genregs_get(struct task_struct *target,
 506                        const struct user_regset *regset,
 507                        unsigned int pos, unsigned int count,
 508                        void *kbuf, void __user *ubuf)
 509 {
 510         if (kbuf) {
 511                 unsigned long *k = kbuf;
 512                 while (count > 0) {
 513                         *k++ = getreg(target, pos);
 514                         count -= sizeof(*k);
 515                         pos += sizeof(*k);
 516                 }
 517         } else {
 518                 unsigned long __user *u = ubuf;
 519                 while (count > 0) {
 520                         if (__put_user(getreg(target, pos), u++))
 521                                 return -EFAULT;
 522                         count -= sizeof(*u);
 523                         pos += sizeof(*u);
 524                 }
 525         }
 526
 527         return 0;
 528 }
 529
 530 static int genregs_set(struct task_struct *target,
 531                        const struct user_regset *regset,
 532                        unsigned int pos, unsigned int count,
 533                        const void *kbuf, const void __user *ubuf)
 534 {
 535         int ret = 0;
 536         if (kbuf) {
 537                 const unsigned long *k = kbuf;
 538                 while (count > 0 && !ret) {
 539                         ret = putreg(target, pos, *k++);
 540                         count -= sizeof(*k);
 541                         pos += sizeof(*k);
 542                 }
 543         } else {
 544                 const unsigned long  __user *u = ubuf;
 545                 while (count > 0 && !ret) {
 546                         unsigned long word;
 547                         ret = __get_user(word, u++);
 548                         if (ret)
 549                                 break;
 550                         ret = putreg(target, pos, word);
 551                         count -= sizeof(*u);
 552                         pos += sizeof(*u);
 553                 }
 554         }
 555         return ret;
 556 }
 557
 558 static void ptrace_triggered(struct perf_event *bp, void *data)
 559 {
 560         int i;
 561         struct thread_struct *thread = &(current->thread);
 562
 563         /*
 564          * Store in the virtual DR6 register the fact that the breakpoint
 565          * was hit so the thread's debugger will see it.
 566          */
 567         for (i = 0; i < HBP_NUM; i++) {
 568                 if (thread->ptrace_bps[i] == bp)
 569                         break;
 570         }
 571
 572         thread->debugreg6 |= (DR_TRAP0 << i);
 573 }
 574
 575 /*
 576  * Walk through every ptrace breakpoints for this thread and
 577  * build the dr7 value on top of their attributes.
 578  *
 579  */
 580 static unsigned long ptrace_get_dr7(struct perf_event *bp[])
 581 {
 582         int i;
 583         int dr7 = 0;
 584         struct arch_hw_breakpoint *info;
 585
 586         for (i = 0; i < HBP_NUM; i++) {
 587                 if (bp[i] && !bp[i]->attr.disabled) {
 588                         info = counter_arch_bp(bp[i]);
 589                         dr7 |= encode_dr7(i, info->len, info->type);
 590                 }
 591         }
 592
 593         return dr7;
 594 }
 595
 596 static struct perf_event *
 597 ptrace_modify_breakpoint(struct perf_event *bp, int len, int type,
 598                          struct task_struct *tsk, int disabled)
 599 {
 600         int err;
 601         int gen_len, gen_type;
 602         DEFINE_BREAKPOINT_ATTR(attr);
 603
 604         /*
 605          * We shoud have at least an inactive breakpoint at this
 606          * slot. It means the user is writing dr7 without having
 607          * written the address register first
 608          */
 609         if (!bp)
 610                 return ERR_PTR(-EINVAL);
 611
 612         err = arch_bp_generic_fields(len, type, &gen_len, &gen_type);
 613         if (err)
 614                 return ERR_PTR(err);
 615
 616         attr = bp->attr;
 617         attr.bp_len = gen_len;
 618         attr.bp_type = gen_type;
 619         attr.disabled = disabled;
 620
 621         return modify_user_hw_breakpoint(bp, &attr);
 622 }
 623
 624 /*
 625  * Handle ptrace writes to debug register 7.
 626  */
 627 static int ptrace_write_dr7(struct task_struct *tsk, unsigned long data)
 628 {
 629         struct thread_struct *thread = &(tsk->thread);
 630         unsigned long old_dr7;
 631         int i, orig_ret = 0, rc = 0;
 632         int enabled, second_pass = 0;
 633         unsigned len, type;
 634         struct perf_event *bp;
 635
 636         data &= ~DR_CONTROL_RESERVED;
 637         old_dr7 = ptrace_get_dr7(thread->ptrace_bps);
 638 restore:
 639         /*
 640          * Loop through all the hardware breakpoints, making the
 641          * appropriate changes to each.
 642          */
 643         for (i = 0; i < HBP_NUM; i++) {
 644                 enabled = decode_dr7(data, i, &len, &type);
 645                 bp = thread->ptrace_bps[i];
 646
 647                 if (!enabled) {
 648                         if (bp) {
 649                                 /*
 650                                  * Don't unregister the breakpoints right-away,
 651                                  * unless all register_user_hw_breakpoint()
 652                                  * requests have succeeded. This prevents
 653                                  * any window of opportunity for debug
 654                                  * register grabbing by other users.
 655                                  */
 656                                 if (!second_pass)
 657                                         continue;
 658
 659                                 thread->ptrace_bps[i] = NULL;
 660                                 bp = ptrace_modify_breakpoint(bp, len, type,
 661                                                               tsk, 1);
 662                                 if (IS_ERR(bp)) {
 663                                         rc = PTR_ERR(bp);
 664                                         thread->ptrace_bps[i] = NULL;
 665                                         break;
 666                                 }
 667                                 thread->ptrace_bps[i] = bp;
 668                         }
 669                         continue;
 670                 }
 671
 672                 bp = ptrace_modify_breakpoint(bp, len, type, tsk, 0);
 673
 674                 /* Incorrect bp, or we have a bug in bp API */
 675                 if (IS_ERR(bp)) {
 676                         rc = PTR_ERR(bp);
 677                         thread->ptrace_bps[i] = NULL;
 678                         break;
 679                 }
 680                 thread->ptrace_bps[i] = bp;
 681         }
 682         /*
 683          * Make a second pass to free the remaining unused breakpoints
 684          * or to restore the original breakpoints if an error occurred.
 685          */
 686         if (!second_pass) {
 687                 second_pass = 1;
 688                 if (rc < 0) {
 689                         orig_ret = rc;
 690                         data = old_dr7;
 691                 }
 692                 goto restore;
 693         }
 694         return ((orig_ret < 0) ? orig_ret : rc);
 695 }
 696
 697 /*
 698  * Handle PTRACE_PEEKUSR calls for the debug register area.
 699  */
 700 static unsigned long ptrace_get_debugreg(struct task_struct *tsk, int n)
 701 {
 702         struct thread_struct *thread = &(tsk->thread);
 703         unsigned long val = 0;
 704
 705         if (n < HBP_NUM) {
 706                 struct perf_event *bp;
 707                 bp = thread->ptrace_bps[n];
 708                 if (!bp)
 709                         return 0;
 710                 val = bp->hw.info.address;
 711         } else if (n == 6) {
 712                 val = thread->debugreg6;
 713          } else if (n == 7) {
 714                 val = ptrace_get_dr7(thread->ptrace_bps);
 715         }
 716         return val;
 717 }
 718
 719 static int ptrace_set_breakpoint_addr(struct task_struct *tsk, int nr,
 720                                       unsigned long addr)
 721 {
 722         struct perf_event *bp;
 723         struct thread_struct *t = &tsk->thread;
 724         DEFINE_BREAKPOINT_ATTR(attr);
 725
 726         if (!t->ptrace_bps[nr]) {
 727                 /*
 728                  * Put stub len and type to register (reserve) an inactive but
 729                  * correct bp
 730                  */
 731                 attr.bp_addr = addr;
 732                 attr.bp_len = HW_BREAKPOINT_LEN_1;
 733                 attr.bp_type = HW_BREAKPOINT_W;
 734                 attr.disabled = 1;
 735
 736                 bp = register_user_hw_breakpoint(&attr, ptrace_triggered, tsk);
 737         } else {
 738                 bp = t->ptrace_bps[nr];
 739                 t->ptrace_bps[nr] = NULL;
 740
 741                 attr = bp->attr;
 742                 attr.bp_addr = addr;
 743                 bp = modify_user_hw_breakpoint(bp, &attr);
 744         }
 745         /*
 746          * CHECKME: the previous code returned -EIO if the addr wasn't a
 747          * valid task virtual addr. The new one will return -EINVAL in this
 748          * case.
 749          * -EINVAL may be what we want for in-kernel breakpoints users, but
 750          * -EIO looks better for ptrace, since we refuse a register writing
 751          * for the user. And anyway this is the previous behaviour.
 752          */
 753         if (IS_ERR(bp))
 754                 return PTR_ERR(bp);
 755
 756         t->ptrace_bps[nr] = bp;
 757
 758         return 0;
 759 }
 760
 761 /*
 762  * Handle PTRACE_POKEUSR calls for the debug register area.
 763  */
 764 int ptrace_set_debugreg(struct task_struct *tsk, int n, unsigned long val)
 765 {
 766         struct thread_struct *thread = &(tsk->thread);
 767         int rc = 0;
 768
 769         /* There are no DR4 or DR5 registers */
 770         if (n == 4 || n == 5)
 771                 return -EIO;
 772
 773         if (n == 6) {
 774                 thread->debugreg6 = val;
 775                 goto ret_path;
 776         }
 777         if (n < HBP_NUM) {
 778                 rc = ptrace_set_breakpoint_addr(tsk, n, val);
 779                 if (rc)
 780                         return rc;
 781         }
 782         /* All that's left is DR7 */
 783         if (n == 7)
 784                 rc = ptrace_write_dr7(tsk, val);
 785
 786 ret_path:
 787         return rc;
 788 }
 789
 790 /*
 791  * These access the current or another (stopped) task's io permission
 792  * bitmap for debugging or core dump.
 793  */
 794 static int ioperm_active(struct task_struct *target,
 795                          const struct user_regset *regset)
 796 {
 797         return target->thread.io_bitmap_max / regset->size;
 798 }
 799
 800 static int ioperm_get(struct task_struct *target,
 801                       const struct user_regset *regset,
 802                       unsigned int pos, unsigned int count,
 803                       void *kbuf, void __user *ubuf)
 804 {
 805         if (!target->thread.io_bitmap_ptr)
 806                 return -ENXIO;
 807
 808         return user_regset_copyout(&pos, &count, &kbuf, &ubuf,
 809                                    target->thread.io_bitmap_ptr,
 810                                    0, IO_BITMAP_BYTES);
 811 }
 812
 813 #ifdef CONFIG_X86_PTRACE_BTS
 814 /*
 815  * A branch trace store context.
 816  *
 817  * Contexts may only be installed by ptrace_bts_config() and only for
 818  * ptraced tasks.
 819  *
 820  * Contexts are destroyed when the tracee is detached from the tracer.
 821  * The actual destruction work requires interrupts enabled, so the
 822  * work is deferred and will be scheduled during __ptrace_unlink().
 823  *
 824  * Contexts hold an additional task_struct reference on the traced
 825  * task, as well as a reference on the tracer's mm.
 826  *
 827  * Ptrace already holds a task_struct for the duration of ptrace operations,
 828  * but since destruction is deferred, it may be executed after both
 829  * tracer and tracee exited.
 830  */
 831 struct bts_context {
 832         /* The branch trace handle. */
 833         struct bts_tracer       *tracer;
 834
 835         /* The buffer used to store the branch trace and its size. */
 836         void                    *buffer;
 837         unsigned int            size;
 838
 839         /* The mm that paid for the above buffer. */
 840         struct mm_struct        *mm;
 841
 842         /* The task this context belongs to. */
 843         struct task_struct      *task;
 844
 845         /* The signal to send on a bts buffer overflow. */
 846         unsigned int            bts_ovfl_signal;
 847
 848         /* The work struct to destroy a context. */
 849         struct work_struct      work;
 850 };
 851
 852 static int alloc_bts_buffer(struct bts_context *context, unsigned int size)
 853 {
 854         void *buffer = NULL;
 855         int err = -ENOMEM;
 856
 857         err = account_locked_memory(current->mm, current->signal->rlim, size);
 858         if (err < 0)
 859                 return err;
 860
 861         buffer = kzalloc(size, GFP_KERNEL);
 862         if (!buffer)
 863                 goto out_refund;
 864
 865         context->buffer = buffer;
 866         context->size = size;
 867         context->mm = get_task_mm(current);
 868
 869         return 0;
 870
 871  out_refund:
 872         refund_locked_memory(current->mm, size);
 873         return err;
 874 }
 875
 876 static inline void free_bts_buffer(struct bts_context *context)
 877 {
 878         if (!context->buffer)
 879                 return;
 880
 881         kfree(context->buffer);
 882         context->buffer = NULL;
 883
 884         refund_locked_memory(context->mm, context->size);
 885         context->size = 0;
 886
 887         mmput(context->mm);
 888         context->mm = NULL;
 889 }
 890
 891 static void free_bts_context_work(struct work_struct *w)
 892 {
 893         struct bts_context *context;
 894
 895         context = container_of(w, struct bts_context, work);
 896
 897         ds_release_bts(context->tracer);
 898         put_task_struct(context->task);
 899         free_bts_buffer(context);
 900         kfree(context);
 901 }
 902
 903 static inline void free_bts_context(struct bts_context *context)
 904 {
 905         INIT_WORK(&context->work, free_bts_context_work);
 906         schedule_work(&context->work);
 907 }
 908
 909 static inline struct bts_context *alloc_bts_context(struct task_struct *task)
 910 {
 911         struct bts_context *context = kzalloc(sizeof(*context), GFP_KERNEL);
 912         if (context) {
 913                 context->task = task;
 914                 task->bts = context;
 915
 916                 get_task_struct(task);
 917         }
 918
 919         return context;
 920 }
 921
 922 static int ptrace_bts_read_record(struct task_struct *child, size_t index,
 923                                   struct bts_struct __user *out)
 924 {
 925         struct bts_context *context;
 926         const struct bts_trace *trace;
 927         struct bts_struct bts;
 928         const unsigned char *at;
 929         int error;
 930
 931         context = child->bts;
 932         if (!context)
 933                 return -ESRCH;
 934
 935         trace = ds_read_bts(context->tracer);
 936         if (!trace)
 937                 return -ESRCH;
 938
 939         at = trace->ds.top - ((index + 1) * trace->ds.size);
 940         if ((void *)at < trace->ds.begin)
 941                 at += (trace->ds.n * trace->ds.size);
 942
 943         if (!trace->read)
 944                 return -EOPNOTSUPP;
 945
 946         error = trace->read(context->tracer, at, &bts);
 947         if (error < 0)
 948                 return error;
 949
 950         if (copy_to_user(out, &bts, sizeof(bts)))
 951                 return -EFAULT;
 952
 953         return sizeof(bts);
 954 }
 955
 956 static int ptrace_bts_drain(struct task_struct *child,
 957                             long size,
 958                             struct bts_struct __user *out)
 959 {
 960         struct bts_context *context;
 961         const struct bts_trace *trace;
 962         const unsigned char *at;
 963         int error, drained = 0;
 964
 965         context = child->bts;
 966         if (!context)
 967                 return -ESRCH;
 968
 969         trace = ds_read_bts(context->tracer);
 970         if (!trace)
 971                 return -ESRCH;
 972
 973         if (!trace->read)
 974                 return -EOPNOTSUPP;
 975
 976         if (size < (trace->ds.top - trace->ds.begin))
 977                 return -EIO;
 978
 979         for (at = trace->ds.begin; (void *)at < trace->ds.top;
 980              out++, drained++, at += trace->ds.size) {
 981                 struct bts_struct bts;
 982
 983                 error = trace->read(context->tracer, at, &bts);
 984                 if (error < 0)
 985                         return error;
 986
 987                 if (copy_to_user(out, &bts, sizeof(bts)))
 988                         return -EFAULT;
 989         }
 990
 991         memset(trace->ds.begin, 0, trace->ds.n * trace->ds.size);
 992
 993         error = ds_reset_bts(context->tracer);
 994         if (error < 0)
 995                 return error;
 996
 997         return drained;
 998 }
 999
1000 static int ptrace_bts_config(struct task_struct *child,
1001                              long cfg_size,
1002                              const struct ptrace_bts_config __user *ucfg)
1003 {
1004         struct bts_context *context;
1005         struct ptrace_bts_config cfg;
1006         unsigned int flags = 0;
1007
1008         if (cfg_size < sizeof(cfg))
1009                 return -EIO;
1010
1011         if (copy_from_user(&cfg, ucfg, sizeof(cfg)))
1012                 return -EFAULT;
1013
1014         context = child->bts;
1015         if (!context)
1016                 context = alloc_bts_context(child);
1017         if (!context)
1018                 return -ENOMEM;
1019
1020         if (cfg.flags & PTRACE_BTS_O_SIGNAL) {
1021                 if (!cfg.signal)
1022                         return -EINVAL;
1023
1024                 return -EOPNOTSUPP;
1025                 context->bts_ovfl_signal = cfg.signal;
1026         }
1027
1028         ds_release_bts(context->tracer);
1029         context->tracer = NULL;
1030
1031         if ((cfg.flags & PTRACE_BTS_O_ALLOC) && (cfg.size != context->size)) {
1032                 int err;
1033
1034                 free_bts_buffer(context);
1035                 if (!cfg.size)
1036                         return 0;
1037
1038                 err = alloc_bts_buffer(context, cfg.size);
1039                 if (err < 0)
1040                         return err;
1041         }
1042
1043         if (cfg.flags & PTRACE_BTS_O_TRACE)
1044                 flags |= BTS_USER;
1045
1046         if (cfg.flags & PTRACE_BTS_O_SCHED)
1047                 flags |= BTS_TIMESTAMPS;
1048
1049         context->tracer =
1050                 ds_request_bts_task(child, context->buffer, context->size,
1051                                     NULL, (size_t)-1, flags);
1052         if (unlikely(IS_ERR(context->tracer))) {
1053                 int error = PTR_ERR(context->tracer);
1054
1055                 free_bts_buffer(context);
1056                 context->tracer = NULL;
1057                 return error;
1058         }
1059
1060         return sizeof(cfg);
1061 }
1062
1063 static int ptrace_bts_status(struct task_struct *child,
1064                              long cfg_size,
1065                              struct ptrace_bts_config __user *ucfg)
1066 {
1067         struct bts_context *context;
1068         const struct bts_trace *trace;
1069         struct ptrace_bts_config cfg;
1070
1071         context = child->bts;
1072         if (!context)
1073                 return -ESRCH;
1074
1075         if (cfg_size < sizeof(cfg))
1076                 return -EIO;
1077
1078         trace = ds_read_bts(context->tracer);
1079         if (!trace)
1080                 return -ESRCH;
1081
1082         memset(&cfg, 0, sizeof(cfg));
1083         cfg.size        = trace->ds.end - trace->ds.begin;
1084         cfg.signal      = context->bts_ovfl_signal;
1085         cfg.bts_size    = sizeof(struct bts_struct);
1086
1087         if (cfg.signal)
1088                 cfg.flags |= PTRACE_BTS_O_SIGNAL;
1089
1090         if (trace->ds.flags & BTS_USER)
1091                 cfg.flags |= PTRACE_BTS_O_TRACE;
1092
1093         if (trace->ds.flags & BTS_TIMESTAMPS)
1094                 cfg.flags |= PTRACE_BTS_O_SCHED;
1095
1096         if (copy_to_user(ucfg, &cfg, sizeof(cfg)))
1097                 return -EFAULT;
1098
1099         return sizeof(cfg);
1100 }
1101
1102 static int ptrace_bts_clear(struct task_struct *child)
1103 {
1104         struct bts_context *context;
1105         const struct bts_trace *trace;
1106
1107         context = child->bts;
1108         if (!context)
1109                 return -ESRCH;
1110
1111         trace = ds_read_bts(context->tracer);
1112         if (!trace)
1113                 return -ESRCH;
1114
1115         memset(trace->ds.begin, 0, trace->ds.n * trace->ds.size);
1116
1117         return ds_reset_bts(context->tracer);
1118 }
1119
1120 static int ptrace_bts_size(struct task_struct *child)
1121 {
1122         struct bts_context *context;
1123         const struct bts_trace *trace;
1124
1125         context = child->bts;
1126         if (!context)
1127                 return -ESRCH;
1128
1129         trace = ds_read_bts(context->tracer);
1130         if (!trace)
1131                 return -ESRCH;
1132
1133         return (trace->ds.top - trace->ds.begin) / trace->ds.size;
1134 }
1135
1136 /*
1137  * Called from __ptrace_unlink() after the child has been moved back
1138  * to its original parent.
1139  */
1140 void ptrace_bts_untrace(struct task_struct *child)
1141 {
1142         if (unlikely(child->bts)) {
1143                 free_bts_context(child->bts);
1144                 child->bts = NULL;
1145         }
1146 }
1147 #endif /* CONFIG_X86_PTRACE_BTS */
1148
1149 /*
1150  * Called by kernel/ptrace.c when detaching..
1151  *
1152  * Make sure the single step bit is not set.
1153  */
1154 void ptrace_disable(struct task_struct *child)
1155 {
1156         user_disable_single_step(child);
1157 #ifdef TIF_SYSCALL_EMU
1158         clear_tsk_thread_flag(child, TIF_SYSCALL_EMU);
1159 #endif
1160 }
1161
1162 #if defined CONFIG_X86_32 || defined CONFIG_IA32_EMULATION
1163 static const struct user_regset_view user_x86_32_view; /* Initialized below. */
1164 #endif
1165
1166 long arch_ptrace(struct task_struct *child, long request, long addr, long data)
1167 {
1168         int ret;
1169         unsigned long __user *datap = (unsigned long __user *)data;
1170
1171         switch (request) {
1172         /* read the word at location addr in the USER area. */
1173         case PTRACE_PEEKUSR: {
1174                 unsigned long tmp;
1175
1176                 ret = -EIO;
1177                 if ((addr & (sizeof(data) - 1)) || addr < 0 ||
1178                     addr >= sizeof(struct user))
1179                         break;
1180
1181                 tmp = 0;  /* Default return condition */
1182                 if (addr < sizeof(struct user_regs_struct))
1183                         tmp = getreg(child, addr);
1184                 else if (addr >= offsetof(struct user, u_debugreg[0]) &&
1185                          addr <= offsetof(struct user, u_debugreg[7])) {
1186                         addr -= offsetof(struct user, u_debugreg[0]);
1187                         tmp = ptrace_get_debugreg(child, addr / sizeof(data));
1188                 }
1189                 ret = put_user(tmp, datap);
1190                 break;
1191         }
1192
1193         case PTRACE_POKEUSR: /* write the word at location addr in the USER area */
1194                 ret = -EIO;
1195                 if ((addr & (sizeof(data) - 1)) || addr < 0 ||
1196                     addr >= sizeof(struct user))
1197                         break;
1198
1199                 if (addr < sizeof(struct user_regs_struct))
1200                         ret = putreg(child, addr, data);
1201                 else if (addr >= offsetof(struct user, u_debugreg[0]) &&
1202                          addr <= offsetof(struct user, u_debugreg[7])) {
1203                         addr -= offsetof(struct user, u_debugreg[0]);
1204                         ret = ptrace_set_debugreg(child,
1205                                                   addr / sizeof(data), data);
1206                 }
1207                 break;
1208
1209         case PTRACE_GETREGS:    /* Get all gp regs from the child. */
1210                 return copy_regset_to_user(child,
1211                                            task_user_regset_view(current),
1212                                            REGSET_GENERAL,
1213                                            0, sizeof(struct user_regs_struct),
1214                                            datap);
1215
1216         case PTRACE_SETREGS:    /* Set all gp regs in the child. */
1217                 return copy_regset_from_user(child,
1218                                              task_user_regset_view(current),
1219                                              REGSET_GENERAL,
1220                                              0, sizeof(struct user_regs_struct),
1221                                              datap);
1222
1223         case PTRACE_GETFPREGS:  /* Get the child FPU state. */
1224                 return copy_regset_to_user(child,
1225                                            task_user_regset_view(current),
1226                                            REGSET_FP,
1227                                            0, sizeof(struct user_i387_struct),
1228                                            datap);
1229
1230         case PTRACE_SETFPREGS:  /* Set the child FPU state. */
1231                 return copy_regset_from_user(child,
1232                                              task_user_regset_view(current),
1233                                              REGSET_FP,
1234                                              0, sizeof(struct user_i387_struct),
1235                                              datap);
1236
1237 #ifdef CONFIG_X86_32
1238         case PTRACE_GETFPXREGS: /* Get the child extended FPU state. */
1239                 return copy_regset_to_user(child, &user_x86_32_view,
1240                                            REGSET_XFP,
1241                                            0, sizeof(struct user_fxsr_struct),
1242                                            datap) ? -EIO : 0;
1243
1244         case PTRACE_SETFPXREGS: /* Set the child extended FPU state. */
1245                 return copy_regset_from_user(child, &user_x86_32_view,
1246                                              REGSET_XFP,
1247                                              0, sizeof(struct user_fxsr_struct),
1248                                              datap) ? -EIO : 0;
1249 #endif
1250
1251 #if defined CONFIG_X86_32 || defined CONFIG_IA32_EMULATION
1252         case PTRACE_GET_THREAD_AREA:
1253                 if (addr < 0)
1254                         return -EIO;
1255                 ret = do_get_thread_area(child, addr,
1256                                          (struct user_desc __user *) data);
1257                 break;
1258
1259         case PTRACE_SET_THREAD_AREA:
1260                 if (addr < 0)
1261                         return -EIO;
1262                 ret = do_set_thread_area(child, addr,
1263                                          (struct user_desc __user *) data, 0);
1264                 break;
1265 #endif
1266
1267 #ifdef CONFIG_X86_64
1268                 /* normal 64bit interface to access TLS data.
1269                    Works just like arch_prctl, except that the arguments
1270                    are reversed. */
1271         case PTRACE_ARCH_PRCTL:
1272                 ret = do_arch_prctl(child, data, addr);
1273                 break;
1274 #endif
1275
1276         /*
1277          * These bits need more cooking - not enabled yet:
1278          */
1279 #ifdef CONFIG_X86_PTRACE_BTS
1280         case PTRACE_BTS_CONFIG:
1281                 ret = ptrace_bts_config
1282                         (child, data, (struct ptrace_bts_config __user *)addr);
1283                 break;
1284
1285         case PTRACE_BTS_STATUS:
1286                 ret = ptrace_bts_status
1287                         (child, data, (struct ptrace_bts_config __user *)addr);
1288                 break;
1289
1290         case PTRACE_BTS_SIZE:
1291                 ret = ptrace_bts_size(child);
1292                 break;
1293
1294         case PTRACE_BTS_GET:
1295                 ret = ptrace_bts_read_record
1296                         (child, data, (struct bts_struct __user *) addr);
1297                 break;
1298
1299         case PTRACE_BTS_CLEAR:
1300                 ret = ptrace_bts_clear(child);
1301                 break;
1302
1303         case PTRACE_BTS_DRAIN:
1304                 ret = ptrace_bts_drain
1305                         (child, data, (struct bts_struct __user *) addr);
1306                 break;
1307 #endif /* CONFIG_X86_PTRACE_BTS */
1308
1309         default:
1310                 ret = ptrace_request(child, request, addr, data);
1311                 break;
1312         }
1313
1314         return ret;
1315 }
1316
1317 #ifdef CONFIG_IA32_EMULATION
1318
1319 #include <linux/compat.h>
1320 #include <linux/syscalls.h>
1321 #include <asm/ia32.h>
1322 #include <asm/user32.h>
1323
1324 #define R32(l,q)                                                        \
1325         case offsetof(struct user32, regs.l):                           \
1326                 regs->q = value; break
1327
1328 #define SEG32(rs)                                                       \
1329         case offsetof(struct user32, regs.rs):                          \
1330                 return set_segment_reg(child,                           \
1331                                        offsetof(struct user_regs_struct, rs), \
1332                                        value);                          \
1333                 break
1334
1335 static int putreg32(struct task_struct *child, unsigned regno, u32 value)
1336 {
1337         struct pt_regs *regs = task_pt_regs(child);
1338
1339         switch (regno) {
1340
1341         SEG32(cs);
1342         SEG32(ds);
1343         SEG32(es);
1344         SEG32(fs);
1345         SEG32(gs);
1346         SEG32(ss);
1347
1348         R32(ebx, bx);
1349         R32(ecx, cx);
1350         R32(edx, dx);
1351         R32(edi, di);
1352         R32(esi, si);
1353         R32(ebp, bp);
1354         R32(eax, ax);
1355         R32(eip, ip);
1356         R32(esp, sp);
1357
1358         case offsetof(struct user32, regs.orig_eax):
1359                 /*
1360                  * A 32-bit debugger setting orig_eax means to restore
1361                  * the state of the task restarting a 32-bit syscall.
1362                  * Make sure we interpret the -ERESTART* codes correctly
1363                  * in case the task is not actually still sitting at the
1364                  * exit from a 32-bit syscall with TS_COMPAT still set.
1365                  */
1366                 regs->orig_ax = value;
1367                 if (syscall_get_nr(child, regs) >= 0)
1368                         task_thread_info(child)->status |= TS_COMPAT;
1369                 break;
1370
1371         case offsetof(struct user32, regs.eflags):
1372                 return set_flags(child, value);
1373
1374         case offsetof(struct user32, u_debugreg[0]) ...
1375                 offsetof(struct user32, u_debugreg[7]):
1376                 regno -= offsetof(struct user32, u_debugreg[0]);
1377                 return ptrace_set_debugreg(child, regno / 4, value);
1378
1379         default:
1380                 if (regno > sizeof(struct user32) || (regno & 3))
1381                         return -EIO;
1382
1383                 /*
1384                  * Other dummy fields in the virtual user structure
1385                  * are ignored
1386                  */
1387                 break;
1388         }
1389         return 0;
1390 }
1391
1392 #undef R32
1393 #undef SEG32
1394
1395 #define R32(l,q)                                                        \
1396         case offsetof(struct user32, regs.l):                           \
1397                 *val = regs->q; break
1398
1399 #define SEG32(rs)                                                       \
1400         case offsetof(struct user32, regs.rs):                          \
1401                 *val = get_segment_reg(child,                           \
1402                                        offsetof(struct user_regs_struct, rs)); \
1403                 break
1404
1405 static int getreg32(struct task_struct *child, unsigned regno, u32 *val)
1406 {
1407         struct pt_regs *regs = task_pt_regs(child);
1408
1409         switch (regno) {
1410
1411         SEG32(ds);
1412         SEG32(es);
1413         SEG32(fs);
1414         SEG32(gs);
1415
1416         R32(cs, cs);
1417         R32(ss, ss);
1418         R32(ebx, bx);
1419         R32(ecx, cx);
1420         R32(edx, dx);
1421         R32(edi, di);
1422         R32(esi, si);
1423         R32(ebp, bp);
1424         R32(eax, ax);
1425         R32(orig_eax, orig_ax);
1426         R32(eip, ip);
1427         R32(esp, sp);
1428
1429         case offsetof(struct user32, regs.eflags):
1430                 *val = get_flags(child);
1431                 break;
1432
1433         case offsetof(struct user32, u_debugreg[0]) ...
1434                 offsetof(struct user32, u_debugreg[7]):
1435                 regno -= offsetof(struct user32, u_debugreg[0]);
1436                 *val = ptrace_get_debugreg(child, regno / 4);
1437                 break;
1438
1439         default:
1440                 if (regno > sizeof(struct user32) || (regno & 3))
1441                         return -EIO;
1442
1443                 /*
1444                  * Other dummy fields in the virtual user structure
1445                  * are ignored
1446                  */
1447                 *val = 0;
1448                 break;
1449         }
1450         return 0;
1451 }
1452
1453 #undef R32
1454 #undef SEG32
1455
1456 static int genregs32_get(struct task_struct *target,
1457                          const struct user_regset *regset,
1458                          unsigned int pos, unsigned int count,
1459                          void *kbuf, void __user *ubuf)
1460 {
1461         if (kbuf) {
1462                 compat_ulong_t *k = kbuf;
1463                 while (count > 0) {
1464                         getreg32(target, pos, k++);
1465                         count -= sizeof(*k);
1466                         pos += sizeof(*k);
1467                 }
1468         } else {
1469                 compat_ulong_t __user *u = ubuf;
1470                 while (count > 0) {
1471                         compat_ulong_t word;
1472                         getreg32(target, pos, &word);
1473                         if (__put_user(word, u++))
1474                                 return -EFAULT;
1475                         count -= sizeof(*u);
1476                         pos += sizeof(*u);
1477                 }
1478         }
1479
1480         return 0;
1481 }
1482
1483 static int genregs32_set(struct task_struct *target,
1484                          const struct user_regset *regset,
1485                          unsigned int pos, unsigned int count,
1486                          const void *kbuf, const void __user *ubuf)
1487 {
1488         int ret = 0;
1489         if (kbuf) {
1490                 const compat_ulong_t *k = kbuf;
1491                 while (count > 0 && !ret) {
1492                         ret = putreg32(target, pos, *k++);
1493                         count -= sizeof(*k);
1494                         pos += sizeof(*k);
1495                 }
1496         } else {
1497                 const compat_ulong_t __user *u = ubuf;
1498                 while (count > 0 && !ret) {
1499                         compat_ulong_t word;
1500                         ret = __get_user(word, u++);
1501                         if (ret)
1502                                 break;
1503                         ret = putreg32(target, pos, word);
1504                         count -= sizeof(*u);
1505                         pos += sizeof(*u);
1506                 }
1507         }
1508         return ret;
1509 }
1510
1511 long compat_arch_ptrace(struct task_struct *child, compat_long_t request,
1512                         compat_ulong_t caddr, compat_ulong_t cdata)
1513 {
1514         unsigned long addr = caddr;
1515         unsigned long data = cdata;
1516         void __user *datap = compat_ptr(data);
1517         int ret;
1518         __u32 val;
1519
1520         switch (request) {
1521         case PTRACE_PEEKUSR:
1522                 ret = getreg32(child, addr, &val);
1523                 if (ret == 0)
1524                         ret = put_user(val, (__u32 __user *)datap);
1525                 break;
1526
1527         case PTRACE_POKEUSR:
1528                 ret = putreg32(child, addr, data);
1529                 break;
1530
1531         case PTRACE_GETREGS:    /* Get all gp regs from the child. */
1532                 return copy_regset_to_user(child, &user_x86_32_view,
1533                                            REGSET_GENERAL,
1534                                            0, sizeof(struct user_regs_struct32),
1535                                            datap);
1536
1537         case PTRACE_SETREGS:    /* Set all gp regs in the child. */
1538                 return copy_regset_from_user(child, &user_x86_32_view,
1539                                              REGSET_GENERAL, 0,
1540                                              sizeof(struct user_regs_struct32),
1541                                              datap);
1542
1543         case PTRACE_GETFPREGS:  /* Get the child FPU state. */
1544                 return copy_regset_to_user(child, &user_x86_32_view,
1545                                            REGSET_FP, 0,
1546                                            sizeof(struct user_i387_ia32_struct),
1547                                            datap);
1548
1549         case PTRACE_SETFPREGS:  /* Set the child FPU state. */
1550                 return copy_regset_from_user(
1551                         child, &user_x86_32_view, REGSET_FP,
1552                         0, sizeof(struct user_i387_ia32_struct), datap);
1553
1554         case PTRACE_GETFPXREGS: /* Get the child extended FPU state. */
1555                 return copy_regset_to_user(child, &user_x86_32_view,
1556                                            REGSET_XFP, 0,
1557                                            sizeof(struct user32_fxsr_struct),
1558                                            datap);
1559
1560         case PTRACE_SETFPXREGS: /* Set the child extended FPU state. */
1561                 return copy_regset_from_user(child, &user_x86_32_view,
1562                                              REGSET_XFP, 0,
1563                                              sizeof(struct user32_fxsr_struct),
1564                                              datap);
1565
1566         case PTRACE_GET_THREAD_AREA:
1567         case PTRACE_SET_THREAD_AREA:
1568 #ifdef CONFIG_X86_PTRACE_BTS
1569         case PTRACE_BTS_CONFIG:
1570         case PTRACE_BTS_STATUS:
1571         case PTRACE_BTS_SIZE:
1572         case PTRACE_BTS_GET:
1573         case PTRACE_BTS_CLEAR:
1574         case PTRACE_BTS_DRAIN:
1575 #endif /* CONFIG_X86_PTRACE_BTS */
1576                 return arch_ptrace(child, request, addr, data);
1577
1578         default:
1579                 return compat_ptrace_request(child, request, addr, data);
1580         }
1581
1582         return ret;
1583 }
1584
1585 #endif  /* CONFIG_IA32_EMULATION */
1586
1587 #ifdef CONFIG_X86_64
1588
1589 static const struct user_regset x86_64_regsets[] = {
1590         [REGSET_GENERAL] = {
1591                 .core_note_type = NT_PRSTATUS,
1592                 .n = sizeof(struct user_regs_struct) / sizeof(long),
1593                 .size = sizeof(long), .align = sizeof(long),
1594                 .get = genregs_get, .set = genregs_set
1595         },
1596         [REGSET_FP] = {
1597                 .core_note_type = NT_PRFPREG,
1598                 .n = sizeof(struct user_i387_struct) / sizeof(long),
1599                 .size = sizeof(long), .align = sizeof(long),
1600                 .active = xfpregs_active, .get = xfpregs_get, .set = xfpregs_set
1601         },
1602         [REGSET_IOPERM64] = {
1603                 .core_note_type = NT_386_IOPERM,
1604                 .n = IO_BITMAP_LONGS,
1605                 .size = sizeof(long), .align = sizeof(long),
1606                 .active = ioperm_active, .get = ioperm_get
1607         },
1608 };
1609
1610 static const struct user_regset_view user_x86_64_view = {
1611         .name = "x86_64", .e_machine = EM_X86_64,
1612         .regsets = x86_64_regsets, .n = ARRAY_SIZE(x86_64_regsets)
1613 };
1614
1615 #else  /* CONFIG_X86_32 */
1616
1617 #define user_regs_struct32      user_regs_struct
1618 #define genregs32_get           genregs_get
1619 #define genregs32_set           genregs_set
1620
1621 #define user_i387_ia32_struct   user_i387_struct
1622 #define user32_fxsr_struct      user_fxsr_struct
1623
1624 #endif  /* CONFIG_X86_64 */
1625
1626 #if defined CONFIG_X86_32 || defined CONFIG_IA32_EMULATION
1627 static const struct user_regset x86_32_regsets[] = {
1628         [REGSET_GENERAL] = {
1629                 .core_note_type = NT_PRSTATUS,
1630                 .n = sizeof(struct user_regs_struct32) / sizeof(u32),
1631                 .size = sizeof(u32), .align = sizeof(u32),
1632                 .get = genregs32_get, .set = genregs32_set
1633         },
1634         [REGSET_FP] = {
1635                 .core_note_type = NT_PRFPREG,
1636                 .n = sizeof(struct user_i387_ia32_struct) / sizeof(u32),
1637                 .size = sizeof(u32), .align = sizeof(u32),
1638                 .active = fpregs_active, .get = fpregs_get, .set = fpregs_set
1639         },
1640         [REGSET_XFP] = {
1641                 .core_note_type = NT_PRXFPREG,
1642                 .n = sizeof(struct user32_fxsr_struct) / sizeof(u32),
1643                 .size = sizeof(u32), .align = sizeof(u32),
1644                 .active = xfpregs_active, .get = xfpregs_get, .set = xfpregs_set
1645         },
1646         [REGSET_TLS] = {
1647                 .core_note_type = NT_386_TLS,
1648                 .n = GDT_ENTRY_TLS_ENTRIES, .bias = GDT_ENTRY_TLS_MIN,
1649                 .size = sizeof(struct user_desc),
1650                 .align = sizeof(struct user_desc),
1651                 .active = regset_tls_active,
1652                 .get = regset_tls_get, .set = regset_tls_set
1653         },
1654         [REGSET_IOPERM32] = {
1655                 .core_note_type = NT_386_IOPERM,
1656                 .n = IO_BITMAP_BYTES / sizeof(u32),
1657                 .size = sizeof(u32), .align = sizeof(u32),
1658                 .active = ioperm_active, .get = ioperm_get
1659         },
1660 };
1661
1662 static const struct user_regset_view user_x86_32_view = {
1663         .name = "i386", .e_machine = EM_386,
1664         .regsets = x86_32_regsets, .n = ARRAY_SIZE(x86_32_regsets)
1665 };
1666 #endif
1667
1668 const struct user_regset_view *task_user_regset_view(struct task_struct *task)
1669 {
1670 #ifdef CONFIG_IA32_EMULATION
1671         if (test_tsk_thread_flag(task, TIF_IA32))
1672 #endif
1673 #if defined CONFIG_X86_32 || defined CONFIG_IA32_EMULATION
1674                 return &user_x86_32_view;
1675 #endif
1676 #ifdef CONFIG_X86_64
1677         return &user_x86_64_view;
1678 #endif
1679 }
1680
1681 void send_sigtrap(struct task_struct *tsk, struct pt_regs *regs,
1682                                          int error_code, int si_code)
1683 {
1684         struct siginfo info;
1685
1686         tsk->thread.trap_no = 1;
1687         tsk->thread.error_code = error_code;
1688
1689         memset(&info, 0, sizeof(info));
1690         info.si_signo = SIGTRAP;
1691         info.si_code = si_code;
1692
1693         /* User-mode ip? */
1694         info.si_addr = user_mode_vm(regs) ? (void __user *) regs->ip : NULL;
1695
1696         /* Send us the fake SIGTRAP */
1697         force_sig_info(SIGTRAP, &info, tsk);
1698 }
1699
1700
1701 #ifdef CONFIG_X86_32
1702 # define IS_IA32        1
1703 #elif defined CONFIG_IA32_EMULATION
1704 # define IS_IA32        is_compat_task()
1705 #else
1706 # define IS_IA32        0
1707 #endif
1708
1709 /*
1710  * We must return the syscall number to actually look up in the table.
1711  * This can be -1L to skip running any syscall at all.
1712  */
1713 asmregparm long syscall_trace_enter(struct pt_regs *regs)
1714 {
1715         long ret = 0;
1716
1717         /*
1718          * If we stepped into a sysenter/syscall insn, it trapped in
1719          * kernel mode; do_debug() cleared TF and set TIF_SINGLESTEP.
1720          * If user-mode had set TF itself, then it's still clear from
1721          * do_debug() and we need to set it again to restore the user
1722          * state.  If we entered on the slow path, TF was already set.
1723          */
1724         if (test_thread_flag(TIF_SINGLESTEP))
1725                 regs->flags |= X86_EFLAGS_TF;
1726
1727         /* do the secure computing check first */
1728         secure_computing(regs->orig_ax);
1729
1730         if (unlikely(test_thread_flag(TIF_SYSCALL_EMU)))
1731                 ret = -1L;
1732
1733         if ((ret || test_thread_flag(TIF_SYSCALL_TRACE)) &&
1734             tracehook_report_syscall_entry(regs))
1735                 ret = -1L;
1736
1737         if (unlikely(test_thread_flag(TIF_SYSCALL_TRACEPOINT)))
1738                 trace_sys_enter(regs, regs->orig_ax);
1739
1740         if (unlikely(current->audit_context)) {
1741                 if (IS_IA32)
1742                         audit_syscall_entry(AUDIT_ARCH_I386,
1743                                             regs->orig_ax,
1744                                             regs->bx, regs->cx,
1745                                             regs->dx, regs->si);
1746 #ifdef CONFIG_X86_64
1747                 else
1748                         audit_syscall_entry(AUDIT_ARCH_X86_64,
1749                                             regs->orig_ax,
1750                                             regs->di, regs->si,
1751                                             regs->dx, regs->r10);
1752 #endif
1753         }
1754
1755         return ret ?: regs->orig_ax;
1756 }
1757
1758 asmregparm void syscall_trace_leave(struct pt_regs *regs)
1759 {
1760         if (unlikely(current->audit_context))
1761                 audit_syscall_exit(AUDITSC_RESULT(regs->ax), regs->ax);
1762
1763         if (unlikely(test_thread_flag(TIF_SYSCALL_TRACEPOINT)))
1764                 trace_sys_exit(regs, regs->ax);
1765
1766         if (test_thread_flag(TIF_SYSCALL_TRACE))
1767                 tracehook_report_syscall_exit(regs, 0);
1768
1769         /*
1770          * If TIF_SYSCALL_EMU is set, we only get here because of
1771          * TIF_SINGLESTEP (i.e. this is PTRACE_SYSEMU_SINGLESTEP).
1772          * We already reported this syscall instruction in
1773          * syscall_trace_enter(), so don't do any more now.
1774          */
1775         if (unlikely(test_thread_flag(TIF_SYSCALL_EMU)))
1776                 return;
1777
1778         /*
1779          * If we are single-stepping, synthesize a trap to follow the
1780          * system call instruction.
1781          */
1782         if (test_thread_flag(TIF_SINGLESTEP) &&
1783             tracehook_consider_fatal_signal(current, SIGTRAP))
1784                 send_sigtrap(current, regs, 0, TRAP_BRKPT);
1785 }