kernel/trace/trace_selftest.c

   1 /* Include in trace.c */
   2
   3 #include <linux/stringify.h>
   4 #include <linux/kthread.h>
   5 #include <linux/delay.h>
   6 #include <linux/slab.h>
   7
   8 static inline int trace_valid_entry(struct trace_entry *entry)
   9 {
  10         switch (entry->type) {
  11         case TRACE_FN:
  12         case TRACE_CTX:
  13         case TRACE_WAKE:
  14         case TRACE_STACK:
  15         case TRACE_PRINT:
  16         case TRACE_BRANCH:
  17         case TRACE_GRAPH_ENT:
  18         case TRACE_GRAPH_RET:
  19                 return 1;
  20         }
  21         return 0;
  22 }
  23
  24 static int trace_test_buffer_cpu(struct trace_buffer *buf, int cpu)
  25 {
  26         struct ring_buffer_event *event;
  27         struct trace_entry *entry;
  28         unsigned int loops = 0;
  29
  30         while ((event = ring_buffer_consume(buf->buffer, cpu, NULL, NULL))) {
  31                 entry = ring_buffer_event_data(event);
  32
  33                 /*
  34                  * The ring buffer is a size of trace_buf_size, if
  35                  * we loop more than the size, there's something wrong
  36                  * with the ring buffer.
  37                  */
  38                 if (loops++ > trace_buf_size) {
  39                         printk(KERN_CONT ".. bad ring buffer ");
  40                         goto failed;
  41                 }
  42                 if (!trace_valid_entry(entry)) {
  43                         printk(KERN_CONT ".. invalid entry %d ",
  44                                 entry->type);
  45                         goto failed;
  46                 }
  47         }
  48         return 0;
  49
  50  failed:
  51         /* disable tracing */
  52         tracing_disabled = 1;
  53         printk(KERN_CONT ".. corrupted trace buffer .. ");
  54         return -1;
  55 }
  56
  57 /*
  58  * Test the trace buffer to see if all the elements
  59  * are still sane.
  60  */
  61 static int trace_test_buffer(struct trace_buffer *buf, unsigned long *count)
  62 {
  63         unsigned long flags, cnt = 0;
  64         int cpu, ret = 0;
  65
  66         /* Don't allow flipping of max traces now */
  67         local_irq_save(flags);
  68         arch_spin_lock(&buf->tr->max_lock);
  69
  70         cnt = ring_buffer_entries(buf->buffer);
  71
  72         /*
  73          * The trace_test_buffer_cpu runs a while loop to consume all data.
  74          * If the calling tracer is broken, and is constantly filling
  75          * the buffer, this will run forever, and hard lock the box.
  76          * We disable the ring buffer while we do this test to prevent
  77          * a hard lock up.
  78          */
  79         tracing_off();
  80         for_each_possible_cpu(cpu) {
  81                 ret = trace_test_buffer_cpu(buf, cpu);
  82                 if (ret)
  83                         break;
  84         }
  85         tracing_on();
  86         arch_spin_unlock(&buf->tr->max_lock);
  87         local_irq_restore(flags);
  88
  89         if (count)
  90                 *count = cnt;
  91
  92         return ret;
  93 }
  94
  95 static inline void warn_failed_init_tracer(struct tracer *trace, int init_ret)
  96 {
  97         printk(KERN_WARNING "Failed to init %s tracer, init returned %d\n",
  98                 trace->name, init_ret);
  99 }
 100 #ifdef CONFIG_FUNCTION_TRACER
 101
 102 #ifdef CONFIG_DYNAMIC_FTRACE
 103
 104 static int trace_selftest_test_probe1_cnt;
 105 static void trace_selftest_test_probe1_func(unsigned long ip,
 106                                             unsigned long pip,
 107                                             struct ftrace_ops *op,
 108                                             struct pt_regs *pt_regs)
 109 {
 110         trace_selftest_test_probe1_cnt++;
 111 }
 112
 113 static int trace_selftest_test_probe2_cnt;
 114 static void trace_selftest_test_probe2_func(unsigned long ip,
 115                                             unsigned long pip,
 116                                             struct ftrace_ops *op,
 117                                             struct pt_regs *pt_regs)
 118 {
 119         trace_selftest_test_probe2_cnt++;
 120 }
 121
 122 static int trace_selftest_test_probe3_cnt;
 123 static void trace_selftest_test_probe3_func(unsigned long ip,
 124                                             unsigned long pip,
 125                                             struct ftrace_ops *op,
 126                                             struct pt_regs *pt_regs)
 127 {
 128         trace_selftest_test_probe3_cnt++;
 129 }
 130
 131 static int trace_selftest_test_global_cnt;
 132 static void trace_selftest_test_global_func(unsigned long ip,
 133                                             unsigned long pip,
 134                                             struct ftrace_ops *op,
 135                                             struct pt_regs *pt_regs)
 136 {
 137         trace_selftest_test_global_cnt++;
 138 }
 139
 140 static int trace_selftest_test_dyn_cnt;
 141 static void trace_selftest_test_dyn_func(unsigned long ip,
 142                                          unsigned long pip,
 143                                          struct ftrace_ops *op,
 144                                          struct pt_regs *pt_regs)
 145 {
 146         trace_selftest_test_dyn_cnt++;
 147 }
 148
 149 static struct ftrace_ops test_probe1 = {
 150         .func                   = trace_selftest_test_probe1_func,
 151         .flags                  = FTRACE_OPS_FL_RECURSION_SAFE,
 152 };
 153
 154 static struct ftrace_ops test_probe2 = {
 155         .func                   = trace_selftest_test_probe2_func,
 156         .flags                  = FTRACE_OPS_FL_RECURSION_SAFE,
 157 };
 158
 159 static struct ftrace_ops test_probe3 = {
 160         .func                   = trace_selftest_test_probe3_func,
 161         .flags                  = FTRACE_OPS_FL_RECURSION_SAFE,
 162 };
 163
 164 static void print_counts(void)
 165 {
 166         printk("(%d %d %d %d %d) ",
 167                trace_selftest_test_probe1_cnt,
 168                trace_selftest_test_probe2_cnt,
 169                trace_selftest_test_probe3_cnt,
 170                trace_selftest_test_global_cnt,
 171                trace_selftest_test_dyn_cnt);
 172 }
 173
 174 static void reset_counts(void)
 175 {
 176         trace_selftest_test_probe1_cnt = 0;
 177         trace_selftest_test_probe2_cnt = 0;
 178         trace_selftest_test_probe3_cnt = 0;
 179         trace_selftest_test_global_cnt = 0;
 180         trace_selftest_test_dyn_cnt = 0;
 181 }
 182
 183 static int trace_selftest_ops(struct trace_array *tr, int cnt)
 184 {
 185         int save_ftrace_enabled = ftrace_enabled;
 186         struct ftrace_ops *dyn_ops;
 187         char *func1_name;
 188         char *func2_name;
 189         int len1;
 190         int len2;
 191         int ret = -1;
 192
 193         printk(KERN_CONT "PASSED\n");
 194         pr_info("Testing dynamic ftrace ops #%d: ", cnt);
 195
 196         ftrace_enabled = 1;
 197         reset_counts();
 198
 199         /* Handle PPC64 '.' name */
 200         func1_name = "*" __stringify(DYN_FTRACE_TEST_NAME);
 201         func2_name = "*" __stringify(DYN_FTRACE_TEST_NAME2);
 202         len1 = strlen(func1_name);
 203         len2 = strlen(func2_name);
 204
 205         /*
 206          * Probe 1 will trace function 1.
 207          * Probe 2 will trace function 2.
 208          * Probe 3 will trace functions 1 and 2.
 209          */
 210         ftrace_set_filter(&test_probe1, func1_name, len1, 1);
 211         ftrace_set_filter(&test_probe2, func2_name, len2, 1);
 212         ftrace_set_filter(&test_probe3, func1_name, len1, 1);
 213         ftrace_set_filter(&test_probe3, func2_name, len2, 0);
 214
 215         register_ftrace_function(&test_probe1);
 216         register_ftrace_function(&test_probe2);
 217         register_ftrace_function(&test_probe3);
 218         /* First time we are running with main function */
 219         if (cnt > 1) {
 220                 ftrace_init_array_ops(tr, trace_selftest_test_global_func);
 221                 register_ftrace_function(tr->ops);
 222         }
 223
 224         DYN_FTRACE_TEST_NAME();
 225
 226         print_counts();
 227
 228         if (trace_selftest_test_probe1_cnt != 1)
 229                 goto out;
 230         if (trace_selftest_test_probe2_cnt != 0)
 231                 goto out;
 232         if (trace_selftest_test_probe3_cnt != 1)
 233                 goto out;
 234         if (cnt > 1) {
 235                 if (trace_selftest_test_global_cnt == 0)
 236                         goto out;
 237         }
 238
 239         DYN_FTRACE_TEST_NAME2();
 240
 241         print_counts();
 242
 243         if (trace_selftest_test_probe1_cnt != 1)
 244                 goto out;
 245         if (trace_selftest_test_probe2_cnt != 1)
 246                 goto out;
 247         if (trace_selftest_test_probe3_cnt != 2)
 248                 goto out;
 249
 250         /* Add a dynamic probe */
 251         dyn_ops = kzalloc(sizeof(*dyn_ops), GFP_KERNEL);
 252         if (!dyn_ops) {
 253                 printk("MEMORY ERROR ");
 254                 goto out;
 255         }
 256
 257         dyn_ops->func = trace_selftest_test_dyn_func;
 258
 259         register_ftrace_function(dyn_ops);
 260
 261         trace_selftest_test_global_cnt = 0;
 262
 263         DYN_FTRACE_TEST_NAME();
 264
 265         print_counts();
 266
 267         if (trace_selftest_test_probe1_cnt != 2)
 268                 goto out_free;
 269         if (trace_selftest_test_probe2_cnt != 1)
 270                 goto out_free;
 271         if (trace_selftest_test_probe3_cnt != 3)
 272                 goto out_free;
 273         if (cnt > 1) {
 274                 if (trace_selftest_test_global_cnt == 0)
 275                         goto out;
 276         }
 277         if (trace_selftest_test_dyn_cnt == 0)
 278                 goto out_free;
 279
 280         DYN_FTRACE_TEST_NAME2();
 281
 282         print_counts();
 283
 284         if (trace_selftest_test_probe1_cnt != 2)
 285                 goto out_free;
 286         if (trace_selftest_test_probe2_cnt != 2)
 287                 goto out_free;
 288         if (trace_selftest_test_probe3_cnt != 4)
 289                 goto out_free;
 290
 291         ret = 0;
 292  out_free:
 293         unregister_ftrace_function(dyn_ops);
 294         kfree(dyn_ops);
 295
 296  out:
 297         /* Purposely unregister in the same order */
 298         unregister_ftrace_function(&test_probe1);
 299         unregister_ftrace_function(&test_probe2);
 300         unregister_ftrace_function(&test_probe3);
 301         if (cnt > 1)
 302                 unregister_ftrace_function(tr->ops);
 303         ftrace_reset_array_ops(tr);
 304
 305         /* Make sure everything is off */
 306         reset_counts();
 307         DYN_FTRACE_TEST_NAME();
 308         DYN_FTRACE_TEST_NAME();
 309
 310         if (trace_selftest_test_probe1_cnt ||
 311             trace_selftest_test_probe2_cnt ||
 312             trace_selftest_test_probe3_cnt ||
 313             trace_selftest_test_global_cnt ||
 314             trace_selftest_test_dyn_cnt)
 315                 ret = -1;
 316
 317         ftrace_enabled = save_ftrace_enabled;
 318
 319         return ret;
 320 }
 321
 322 /* Test dynamic code modification and ftrace filters */
 323 static int trace_selftest_startup_dynamic_tracing(struct tracer *trace,
 324                                                   struct trace_array *tr,
 325                                                   int (*func)(void))
 326 {
 327         int save_ftrace_enabled = ftrace_enabled;
 328         unsigned long count;
 329         char *func_name;
 330         int ret;
 331
 332         /* The ftrace test PASSED */
 333         printk(KERN_CONT "PASSED\n");
 334         pr_info("Testing dynamic ftrace: ");
 335
 336         /* enable tracing, and record the filter function */
 337         ftrace_enabled = 1;
 338
 339         /* passed in by parameter to fool gcc from optimizing */
 340         func();
 341
 342         /*
 343          * Some archs *cough*PowerPC*cough* add characters to the
 344          * start of the function names. We simply put a '*' to
 345          * accommodate them.
 346          */
 347         func_name = "*" __stringify(DYN_FTRACE_TEST_NAME);
 348
 349         /* filter only on our function */
 350         ftrace_set_global_filter(func_name, strlen(func_name), 1);
 351
 352         /* enable tracing */
 353         ret = tracer_init(trace, tr);
 354         if (ret) {
 355                 warn_failed_init_tracer(trace, ret);
 356                 goto out;
 357         }
 358
 359         /* Sleep for a 1/10 of a second */
 360         msleep(100);
 361
 362         /* we should have nothing in the buffer */
 363         ret = trace_test_buffer(&tr->trace_buffer, &count);
 364         if (ret)
 365                 goto out;
 366
 367         if (count) {
 368                 ret = -1;
 369                 printk(KERN_CONT ".. filter did not filter .. ");
 370                 goto out;
 371         }
 372
 373         /* call our function again */
 374         func();
 375
 376         /* sleep again */
 377         msleep(100);
 378
 379         /* stop the tracing. */
 380         tracing_stop();
 381         ftrace_enabled = 0;
 382
 383         /* check the trace buffer */
 384         ret = trace_test_buffer(&tr->trace_buffer, &count);
 385         tracing_start();
 386
 387         /* we should only have one item */
 388         if (!ret && count != 1) {
 389                 trace->reset(tr);
 390                 printk(KERN_CONT ".. filter failed count=%ld ..", count);
 391                 ret = -1;
 392                 goto out;
 393         }
 394
 395         /* Test the ops with global tracing running */
 396         ret = trace_selftest_ops(tr, 1);
 397         trace->reset(tr);
 398
 399  out:
 400         ftrace_enabled = save_ftrace_enabled;
 401
 402         /* Enable tracing on all functions again */
 403         ftrace_set_global_filter(NULL, 0, 1);
 404
 405         /* Test the ops with global tracing off */
 406         if (!ret)
 407                 ret = trace_selftest_ops(tr, 2);
 408
 409         return ret;
 410 }
 411
 412 static int trace_selftest_recursion_cnt;
 413 static void trace_selftest_test_recursion_func(unsigned long ip,
 414                                                unsigned long pip,
 415                                                struct ftrace_ops *op,
 416                                                struct pt_regs *pt_regs)
 417 {
 418         /*
 419          * This function is registered without the recursion safe flag.
 420          * The ftrace infrastructure should provide the recursion
 421          * protection. If not, this will crash the kernel!
 422          */
 423         if (trace_selftest_recursion_cnt++ > 10)
 424                 return;
 425         DYN_FTRACE_TEST_NAME();
 426 }
 427
 428 static void trace_selftest_test_recursion_safe_func(unsigned long ip,
 429                                                     unsigned long pip,
 430                                                     struct ftrace_ops *op,
 431                                                     struct pt_regs *pt_regs)
 432 {
 433         /*
 434          * We said we would provide our own recursion. By calling
 435          * this function again, we should recurse back into this function
 436          * and count again. But this only happens if the arch supports
 437          * all of ftrace features and nothing else is using the function
 438          * tracing utility.
 439          */
 440         if (trace_selftest_recursion_cnt++)
 441                 return;
 442         DYN_FTRACE_TEST_NAME();
 443 }
 444
 445 static struct ftrace_ops test_rec_probe = {
 446         .func                   = trace_selftest_test_recursion_func,
 447 };
 448
 449 static struct ftrace_ops test_recsafe_probe = {
 450         .func                   = trace_selftest_test_recursion_safe_func,
 451         .flags                  = FTRACE_OPS_FL_RECURSION_SAFE,
 452 };
 453
 454 static int
 455 trace_selftest_function_recursion(void)
 456 {
 457         int save_ftrace_enabled = ftrace_enabled;
 458         char *func_name;
 459         int len;
 460         int ret;
 461
 462         /* The previous test PASSED */
 463         pr_cont("PASSED\n");
 464         pr_info("Testing ftrace recursion: ");
 465
 466
 467         /* enable tracing, and record the filter function */
 468         ftrace_enabled = 1;
 469
 470         /* Handle PPC64 '.' name */
 471         func_name = "*" __stringify(DYN_FTRACE_TEST_NAME);
 472         len = strlen(func_name);
 473
 474         ret = ftrace_set_filter(&test_rec_probe, func_name, len, 1);
 475         if (ret) {
 476                 pr_cont("*Could not set filter* ");
 477                 goto out;
 478         }
 479
 480         ret = register_ftrace_function(&test_rec_probe);
 481         if (ret) {
 482                 pr_cont("*could not register callback* ");
 483                 goto out;
 484         }
 485
 486         DYN_FTRACE_TEST_NAME();
 487
 488         unregister_ftrace_function(&test_rec_probe);
 489
 490         ret = -1;
 491         if (trace_selftest_recursion_cnt != 1) {
 492                 pr_cont("*callback not called once (%d)* ",
 493                         trace_selftest_recursion_cnt);
 494                 goto out;
 495         }
 496
 497         trace_selftest_recursion_cnt = 1;
 498
 499         pr_cont("PASSED\n");
 500         pr_info("Testing ftrace recursion safe: ");
 501
 502         ret = ftrace_set_filter(&test_recsafe_probe, func_name, len, 1);
 503         if (ret) {
 504                 pr_cont("*Could not set filter* ");
 505                 goto out;
 506         }
 507
 508         ret = register_ftrace_function(&test_recsafe_probe);
 509         if (ret) {
 510                 pr_cont("*could not register callback* ");
 511                 goto out;
 512         }
 513
 514         DYN_FTRACE_TEST_NAME();
 515
 516         unregister_ftrace_function(&test_recsafe_probe);
 517
 518         ret = -1;
 519         if (trace_selftest_recursion_cnt != 2) {
 520                 pr_cont("*callback not called expected 2 times (%d)* ",
 521                         trace_selftest_recursion_cnt);
 522                 goto out;
 523         }
 524
 525         ret = 0;
 526 out:
 527         ftrace_enabled = save_ftrace_enabled;
 528
 529         return ret;
 530 }
 531 #else
 532 # define trace_selftest_startup_dynamic_tracing(trace, tr, func) ({ 0; })
 533 # define trace_selftest_function_recursion() ({ 0; })
 534 #endif /* CONFIG_DYNAMIC_FTRACE */
 535
 536 static enum {
 537         TRACE_SELFTEST_REGS_START,
 538         TRACE_SELFTEST_REGS_FOUND,
 539         TRACE_SELFTEST_REGS_NOT_FOUND,
 540 } trace_selftest_regs_stat;
 541
 542 static void trace_selftest_test_regs_func(unsigned long ip,
 543                                           unsigned long pip,
 544                                           struct ftrace_ops *op,
 545                                           struct pt_regs *pt_regs)
 546 {
 547         if (pt_regs)
 548                 trace_selftest_regs_stat = TRACE_SELFTEST_REGS_FOUND;
 549         else
 550                 trace_selftest_regs_stat = TRACE_SELFTEST_REGS_NOT_FOUND;
 551 }
 552
 553 static struct ftrace_ops test_regs_probe = {
 554         .func           = trace_selftest_test_regs_func,
 555         .flags          = FTRACE_OPS_FL_RECURSION_SAFE | FTRACE_OPS_FL_SAVE_REGS,
 556 };
 557
 558 static int
 559 trace_selftest_function_regs(void)
 560 {
 561         int save_ftrace_enabled = ftrace_enabled;
 562         char *func_name;
 563         int len;
 564         int ret;
 565         int supported = 0;
 566
 567 #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
 568         supported = 1;
 569 #endif
 570
 571         /* The previous test PASSED */
 572         pr_cont("PASSED\n");
 573         pr_info("Testing ftrace regs%s: ",
 574                 !supported ? "(no arch support)" : "");
 575
 576         /* enable tracing, and record the filter function */
 577         ftrace_enabled = 1;
 578
 579         /* Handle PPC64 '.' name */
 580         func_name = "*" __stringify(DYN_FTRACE_TEST_NAME);
 581         len = strlen(func_name);
 582
 583         ret = ftrace_set_filter(&test_regs_probe, func_name, len, 1);
 584         /*
 585          * If DYNAMIC_FTRACE is not set, then we just trace all functions.
 586          * This test really doesn't care.
 587          */
 588         if (ret && ret != -ENODEV) {
 589                 pr_cont("*Could not set filter* ");
 590                 goto out;
 591         }
 592
 593         ret = register_ftrace_function(&test_regs_probe);
 594         /*
 595          * Now if the arch does not support passing regs, then this should
 596          * have failed.
 597          */
 598         if (!supported) {
 599                 if (!ret) {
 600                         pr_cont("*registered save-regs without arch support* ");
 601                         goto out;
 602                 }
 603                 test_regs_probe.flags |= FTRACE_OPS_FL_SAVE_REGS_IF_SUPPORTED;
 604                 ret = register_ftrace_function(&test_regs_probe);
 605         }
 606         if (ret) {
 607                 pr_cont("*could not register callback* ");
 608                 goto out;
 609         }
 610
 611
 612         DYN_FTRACE_TEST_NAME();
 613
 614         unregister_ftrace_function(&test_regs_probe);
 615
 616         ret = -1;
 617
 618         switch (trace_selftest_regs_stat) {
 619         case TRACE_SELFTEST_REGS_START:
 620                 pr_cont("*callback never called* ");
 621                 goto out;
 622
 623         case TRACE_SELFTEST_REGS_FOUND:
 624                 if (supported)
 625                         break;
 626                 pr_cont("*callback received regs without arch support* ");
 627                 goto out;
 628
 629         case TRACE_SELFTEST_REGS_NOT_FOUND:
 630                 if (!supported)
 631                         break;
 632                 pr_cont("*callback received NULL regs* ");
 633                 goto out;
 634         }
 635
 636         ret = 0;
 637 out:
 638         ftrace_enabled = save_ftrace_enabled;
 639
 640         return ret;
 641 }
 642
 643 /*
 644  * Simple verification test of ftrace function tracer.
 645  * Enable ftrace, sleep 1/10 second, and then read the trace
 646  * buffer to see if all is in order.
 647  */
 648 __init int
 649 trace_selftest_startup_function(struct tracer *trace, struct trace_array *tr)
 650 {
 651         int save_ftrace_enabled = ftrace_enabled;
 652         unsigned long count;
 653         int ret;
 654
 655 #ifdef CONFIG_DYNAMIC_FTRACE
 656         if (ftrace_filter_param) {
 657                 printk(KERN_CONT " ... kernel command line filter set: force PASS ... ");
 658                 return 0;
 659         }
 660 #endif
 661
 662         /* make sure msleep has been recorded */
 663         msleep(1);
 664
 665         /* start the tracing */
 666         ftrace_enabled = 1;
 667
 668         ret = tracer_init(trace, tr);
 669         if (ret) {
 670                 warn_failed_init_tracer(trace, ret);
 671                 goto out;
 672         }
 673
 674         /* Sleep for a 1/10 of a second */
 675         msleep(100);
 676         /* stop the tracing. */
 677         tracing_stop();
 678         ftrace_enabled = 0;
 679
 680         /* check the trace buffer */
 681         ret = trace_test_buffer(&tr->trace_buffer, &count);
 682         trace->reset(tr);
 683         tracing_start();
 684
 685         if (!ret && !count) {
 686                 printk(KERN_CONT ".. no entries found ..");
 687                 ret = -1;
 688                 goto out;
 689         }
 690
 691         ret = trace_selftest_startup_dynamic_tracing(trace, tr,
 692                                                      DYN_FTRACE_TEST_NAME);
 693         if (ret)
 694                 goto out;
 695
 696         ret = trace_selftest_function_recursion();
 697         if (ret)
 698                 goto out;
 699
 700         ret = trace_selftest_function_regs();
 701  out:
 702         ftrace_enabled = save_ftrace_enabled;
 703
 704         /* kill ftrace totally if we failed */
 705         if (ret)
 706                 ftrace_kill();
 707
 708         return ret;
 709 }
 710 #endif /* CONFIG_FUNCTION_TRACER */
 711
 712
 713 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
 714
 715 /* Maximum number of functions to trace before diagnosing a hang */
 716 #define GRAPH_MAX_FUNC_TEST     100000000
 717
 718 static unsigned int graph_hang_thresh;
 719
 720 /* Wrap the real function entry probe to avoid possible hanging */
 721 static int trace_graph_entry_watchdog(struct ftrace_graph_ent *trace)
 722 {
 723         /* This is harmlessly racy, we want to approximately detect a hang */
 724         if (unlikely(++graph_hang_thresh > GRAPH_MAX_FUNC_TEST)) {
 725                 ftrace_graph_stop();
 726                 printk(KERN_WARNING "BUG: Function graph tracer hang!\n");
 727                 if (ftrace_dump_on_oops) {
 728                         ftrace_dump(DUMP_ALL);
 729                         /* ftrace_dump() disables tracing */
 730                         tracing_on();
 731                 }
 732                 return 0;
 733         }
 734
 735         return trace_graph_entry(trace);
 736 }
 737
 738 /*
 739  * Pretty much the same than for the function tracer from which the selftest
 740  * has been borrowed.
 741  */
 742 __init int
 743 trace_selftest_startup_function_graph(struct tracer *trace,
 744                                         struct trace_array *tr)
 745 {
 746         int ret;
 747         unsigned long count;
 748
 749 #ifdef CONFIG_DYNAMIC_FTRACE
 750         if (ftrace_filter_param) {
 751                 printk(KERN_CONT " ... kernel command line filter set: force PASS ... ");
 752                 return 0;
 753         }
 754 #endif
 755
 756         /*
 757          * Simulate the init() callback but we attach a watchdog callback
 758          * to detect and recover from possible hangs
 759          */
 760         tracing_reset_online_cpus(&tr->trace_buffer);
 761         set_graph_array(tr);
 762         ret = register_ftrace_graph(&trace_graph_return,
 763                                     &trace_graph_entry_watchdog);
 764         if (ret) {
 765                 warn_failed_init_tracer(trace, ret);
 766                 goto out;
 767         }
 768         tracing_start_cmdline_record();
 769
 770         /* Sleep for a 1/10 of a second */
 771         msleep(100);
 772
 773         /* Have we just recovered from a hang? */
 774         if (graph_hang_thresh > GRAPH_MAX_FUNC_TEST) {
 775                 tracing_selftest_disabled = true;
 776                 ret = -1;
 777                 goto out;
 778         }
 779
 780         tracing_stop();
 781
 782         /* check the trace buffer */
 783         ret = trace_test_buffer(&tr->trace_buffer, &count);
 784
 785         trace->reset(tr);
 786         tracing_start();
 787
 788         if (!ret && !count) {
 789                 printk(KERN_CONT ".. no entries found ..");
 790                 ret = -1;
 791                 goto out;
 792         }
 793
 794         /* Don't test dynamic tracing, the function tracer already did */
 795
 796 out:
 797         /* Stop it if we failed */
 798         if (ret)
 799                 ftrace_graph_stop();
 800
 801         return ret;
 802 }
 803 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
 804
 805
 806 #ifdef CONFIG_IRQSOFF_TRACER
 807 int
 808 trace_selftest_startup_irqsoff(struct tracer *trace, struct trace_array *tr)
 809 {
 810         unsigned long save_max = tr->max_latency;
 811         unsigned long count;
 812         int ret;
 813
 814         /* start the tracing */
 815         ret = tracer_init(trace, tr);
 816         if (ret) {
 817                 warn_failed_init_tracer(trace, ret);
 818                 return ret;
 819         }
 820
 821         /* reset the max latency */
 822         tr->max_latency = 0;
 823         /* disable interrupts for a bit */
 824         local_irq_disable();
 825         udelay(100);
 826         local_irq_enable();
 827
 828         /*
 829          * Stop the tracer to avoid a warning subsequent
 830          * to buffer flipping failure because tracing_stop()
 831          * disables the tr and max buffers, making flipping impossible
 832          * in case of parallels max irqs off latencies.
 833          */
 834         trace->stop(tr);
 835         /* stop the tracing. */
 836         tracing_stop();
 837         /* check both trace buffers */
 838         ret = trace_test_buffer(&tr->trace_buffer, NULL);
 839         if (!ret)
 840                 ret = trace_test_buffer(&tr->max_buffer, &count);
 841         trace->reset(tr);
 842         tracing_start();
 843
 844         if (!ret && !count) {
 845                 printk(KERN_CONT ".. no entries found ..");
 846                 ret = -1;
 847         }
 848
 849         tr->max_latency = save_max;
 850
 851         return ret;
 852 }
 853 #endif /* CONFIG_IRQSOFF_TRACER */
 854
 855 #ifdef CONFIG_PREEMPT_TRACER
 856 int
 857 trace_selftest_startup_preemptoff(struct tracer *trace, struct trace_array *tr)
 858 {
 859         unsigned long save_max = tr->max_latency;
 860         unsigned long count;
 861         int ret;
 862
 863         /*
 864          * Now that the big kernel lock is no longer preemptable,
 865          * and this is called with the BKL held, it will always
 866          * fail. If preemption is already disabled, simply
 867          * pass the test. When the BKL is removed, or becomes
 868          * preemptible again, we will once again test this,
 869          * so keep it in.
 870          */
 871         if (preempt_count()) {
 872                 printk(KERN_CONT "can not test ... force ");
 873                 return 0;
 874         }
 875
 876         /* start the tracing */
 877         ret = tracer_init(trace, tr);
 878         if (ret) {
 879                 warn_failed_init_tracer(trace, ret);
 880                 return ret;
 881         }
 882
 883         /* reset the max latency */
 884         tr->max_latency = 0;
 885         /* disable preemption for a bit */
 886         preempt_disable();
 887         udelay(100);
 888         preempt_enable();
 889
 890         /*
 891          * Stop the tracer to avoid a warning subsequent
 892          * to buffer flipping failure because tracing_stop()
 893          * disables the tr and max buffers, making flipping impossible
 894          * in case of parallels max preempt off latencies.
 895          */
 896         trace->stop(tr);
 897         /* stop the tracing. */
 898         tracing_stop();
 899         /* check both trace buffers */
 900         ret = trace_test_buffer(&tr->trace_buffer, NULL);
 901         if (!ret)
 902                 ret = trace_test_buffer(&tr->max_buffer, &count);
 903         trace->reset(tr);
 904         tracing_start();
 905
 906         if (!ret && !count) {
 907                 printk(KERN_CONT ".. no entries found ..");
 908                 ret = -1;
 909         }
 910
 911         tr->max_latency = save_max;
 912
 913         return ret;
 914 }
 915 #endif /* CONFIG_PREEMPT_TRACER */
 916
 917 #if defined(CONFIG_IRQSOFF_TRACER) && defined(CONFIG_PREEMPT_TRACER)
 918 int
 919 trace_selftest_startup_preemptirqsoff(struct tracer *trace, struct trace_array *tr)
 920 {
 921         unsigned long save_max = tr->max_latency;
 922         unsigned long count;
 923         int ret;
 924
 925         /*
 926          * Now that the big kernel lock is no longer preemptable,
 927          * and this is called with the BKL held, it will always
 928          * fail. If preemption is already disabled, simply
 929          * pass the test. When the BKL is removed, or becomes
 930          * preemptible again, we will once again test this,
 931          * so keep it in.
 932          */
 933         if (preempt_count()) {
 934                 printk(KERN_CONT "can not test ... force ");
 935                 return 0;
 936         }
 937
 938         /* start the tracing */
 939         ret = tracer_init(trace, tr);
 940         if (ret) {
 941                 warn_failed_init_tracer(trace, ret);
 942                 goto out_no_start;
 943         }
 944
 945         /* reset the max latency */
 946         tr->max_latency = 0;
 947
 948         /* disable preemption and interrupts for a bit */
 949         preempt_disable();
 950         local_irq_disable();
 951         udelay(100);
 952         preempt_enable();
 953         /* reverse the order of preempt vs irqs */
 954         local_irq_enable();
 955
 956         /*
 957          * Stop the tracer to avoid a warning subsequent
 958          * to buffer flipping failure because tracing_stop()
 959          * disables the tr and max buffers, making flipping impossible
 960          * in case of parallels max irqs/preempt off latencies.
 961          */
 962         trace->stop(tr);
 963         /* stop the tracing. */
 964         tracing_stop();
 965         /* check both trace buffers */
 966         ret = trace_test_buffer(&tr->trace_buffer, NULL);
 967         if (ret)
 968                 goto out;
 969
 970         ret = trace_test_buffer(&tr->max_buffer, &count);
 971         if (ret)
 972                 goto out;
 973
 974         if (!ret && !count) {
 975                 printk(KERN_CONT ".. no entries found ..");
 976                 ret = -1;
 977                 goto out;
 978         }
 979
 980         /* do the test by disabling interrupts first this time */
 981         tr->max_latency = 0;
 982         tracing_start();
 983         trace->start(tr);
 984
 985         preempt_disable();
 986         local_irq_disable();
 987         udelay(100);
 988         preempt_enable();
 989         /* reverse the order of preempt vs irqs */
 990         local_irq_enable();
 991
 992         trace->stop(tr);
 993         /* stop the tracing. */
 994         tracing_stop();
 995         /* check both trace buffers */
 996         ret = trace_test_buffer(&tr->trace_buffer, NULL);
 997         if (ret)
 998                 goto out;
 999
1000         ret = trace_test_buffer(&tr->max_buffer, &count);
1001
1002         if (!ret && !count) {
1003                 printk(KERN_CONT ".. no entries found ..");
1004                 ret = -1;
1005                 goto out;
1006         }
1007
1008 out:
1009         tracing_start();
1010 out_no_start:
1011         trace->reset(tr);
1012         tr->max_latency = save_max;
1013
1014         return ret;
1015 }
1016 #endif /* CONFIG_IRQSOFF_TRACER && CONFIG_PREEMPT_TRACER */
1017
1018 #ifdef CONFIG_NOP_TRACER
1019 int
1020 trace_selftest_startup_nop(struct tracer *trace, struct trace_array *tr)
1021 {
1022         /* What could possibly go wrong? */
1023         return 0;
1024 }
1025 #endif
1026
1027 #ifdef CONFIG_SCHED_TRACER
1028 static int trace_wakeup_test_thread(void *data)
1029 {
1030         /* Make this a -deadline thread */
1031         static const struct sched_attr attr = {
1032                 .sched_policy = SCHED_DEADLINE,
1033                 .sched_runtime = 100000ULL,
1034                 .sched_deadline = 10000000ULL,
1035                 .sched_period = 10000000ULL
1036         };
1037         struct completion *x = data;
1038
1039         sched_setattr(current, &attr);
1040
1041         /* Make it know we have a new prio */
1042         complete(x);
1043
1044         /* now go to sleep and let the test wake us up */
1045         set_current_state(TASK_INTERRUPTIBLE);
1046         schedule();
1047
1048         complete(x);
1049
1050         /* we are awake, now wait to disappear */
1051         while (!kthread_should_stop()) {
1052                 /*
1053                  * This will likely be the system top priority
1054                  * task, do short sleeps to let others run.
1055                  */
1056                 msleep(100);
1057         }
1058
1059         return 0;
1060 }
1061
1062 int
1063 trace_selftest_startup_wakeup(struct tracer *trace, struct trace_array *tr)
1064 {
1065         unsigned long save_max = tr->max_latency;
1066         struct task_struct *p;
1067         struct completion is_ready;
1068         unsigned long count;
1069         int ret;
1070
1071         init_completion(&is_ready);
1072
1073         /* create a -deadline thread */
1074         p = kthread_run(trace_wakeup_test_thread, &is_ready, "ftrace-test");
1075         if (IS_ERR(p)) {
1076                 printk(KERN_CONT "Failed to create ftrace wakeup test thread ");
1077                 return -1;
1078         }
1079
1080         /* make sure the thread is running at -deadline policy */
1081         wait_for_completion(&is_ready);
1082
1083         /* start the tracing */
1084         ret = tracer_init(trace, tr);
1085         if (ret) {
1086                 warn_failed_init_tracer(trace, ret);
1087                 return ret;
1088         }
1089
1090         /* reset the max latency */
1091         tr->max_latency = 0;
1092
1093         while (p->on_rq) {
1094                 /*
1095                  * Sleep to make sure the -deadline thread is asleep too.
1096                  * On virtual machines we can't rely on timings,
1097                  * but we want to make sure this test still works.
1098                  */
1099                 msleep(100);
1100         }
1101
1102         init_completion(&is_ready);
1103
1104         wake_up_process(p);
1105
1106         /* Wait for the task to wake up */
1107         wait_for_completion(&is_ready);
1108
1109         /* stop the tracing. */
1110         tracing_stop();
1111         /* check both trace buffers */
1112         ret = trace_test_buffer(&tr->trace_buffer, NULL);
1113         printk("ret = %d\n", ret);
1114         if (!ret)
1115                 ret = trace_test_buffer(&tr->max_buffer, &count);
1116
1117
1118         trace->reset(tr);
1119         tracing_start();
1120
1121         tr->max_latency = save_max;
1122
1123         /* kill the thread */
1124         kthread_stop(p);
1125
1126         if (!ret && !count) {
1127                 printk(KERN_CONT ".. no entries found ..");
1128                 ret = -1;
1129         }
1130
1131         return ret;
1132 }
1133 #endif /* CONFIG_SCHED_TRACER */
1134
1135 #ifdef CONFIG_CONTEXT_SWITCH_TRACER
1136 int
1137 trace_selftest_startup_sched_switch(struct tracer *trace, struct trace_array *tr)
1138 {
1139         unsigned long count;
1140         int ret;
1141
1142         /* start the tracing */
1143         ret = tracer_init(trace, tr);
1144         if (ret) {
1145                 warn_failed_init_tracer(trace, ret);
1146                 return ret;
1147         }
1148
1149         /* Sleep for a 1/10 of a second */
1150         msleep(100);
1151         /* stop the tracing. */
1152         tracing_stop();
1153         /* check the trace buffer */
1154         ret = trace_test_buffer(&tr->trace_buffer, &count);
1155         trace->reset(tr);
1156         tracing_start();
1157
1158         if (!ret && !count) {
1159                 printk(KERN_CONT ".. no entries found ..");
1160                 ret = -1;
1161         }
1162
1163         return ret;
1164 }
1165 #endif /* CONFIG_CONTEXT_SWITCH_TRACER */
1166
1167 #ifdef CONFIG_BRANCH_TRACER
1168 int
1169 trace_selftest_startup_branch(struct tracer *trace, struct trace_array *tr)
1170 {
1171         unsigned long count;
1172         int ret;
1173
1174         /* start the tracing */
1175         ret = tracer_init(trace, tr);
1176         if (ret) {
1177                 warn_failed_init_tracer(trace, ret);
1178                 return ret;
1179         }
1180
1181         /* Sleep for a 1/10 of a second */
1182         msleep(100);
1183         /* stop the tracing. */
1184         tracing_stop();
1185         /* check the trace buffer */
1186         ret = trace_test_buffer(&tr->trace_buffer, &count);
1187         trace->reset(tr);
1188         tracing_start();
1189
1190         if (!ret && !count) {
1191                 printk(KERN_CONT ".. no entries found ..");
1192                 ret = -1;
1193         }
1194
1195         return ret;
1196 }
1197 #endif /* CONFIG_BRANCH_TRACER */
1198