tools/perf/builtin-lock.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 #include <errno.h>
   3 #include <inttypes.h>
   4 #include "builtin.h"
   5 #include "perf.h"
   6
   7 #include "util/evlist.h" // for struct evsel_str_handler
   8 #include "util/evsel.h"
   9 #include "util/symbol.h"
  10 #include "util/thread.h"
  11 #include "util/header.h"
  12 #include "util/target.h"
  13 #include "util/callchain.h"
  14 #include "util/lock-contention.h"
  15 #include "util/bpf_skel/lock_data.h"
  16
  17 #include <subcmd/pager.h>
  18 #include <subcmd/parse-options.h>
  19 #include "util/trace-event.h"
  20 #include "util/tracepoint.h"
  21
  22 #include "util/debug.h"
  23 #include "util/session.h"
  24 #include "util/tool.h"
  25 #include "util/data.h"
  26 #include "util/string2.h"
  27 #include "util/map.h"
  28 #include "util/util.h"
  29
  30 #include <sys/types.h>
  31 #include <sys/prctl.h>
  32 #include <semaphore.h>
  33 #include <math.h>
  34 #include <limits.h>
  35 #include <ctype.h>
  36
  37 #include <linux/list.h>
  38 #include <linux/hash.h>
  39 #include <linux/kernel.h>
  40 #include <linux/zalloc.h>
  41 #include <linux/err.h>
  42 #include <linux/stringify.h>
  43
  44 static struct perf_session *session;
  45 static struct target target;
  46
  47 /* based on kernel/lockdep.c */
  48 #define LOCKHASH_BITS           12
  49 #define LOCKHASH_SIZE           (1UL << LOCKHASH_BITS)
  50
  51 static struct hlist_head lockhash_table[LOCKHASH_SIZE];
  52
  53 #define __lockhashfn(key)       hash_long((unsigned long)key, LOCKHASH_BITS)
  54 #define lockhashentry(key)      (lockhash_table + __lockhashfn((key)))
  55
  56 static struct rb_root           thread_stats;
  57
  58 static bool combine_locks;
  59 static bool show_thread_stats;
  60 static bool show_lock_addrs;
  61 static bool show_lock_owner;
  62 static bool use_bpf;
  63 static unsigned long bpf_map_entries = 10240;
  64 static int max_stack_depth = CONTENTION_STACK_DEPTH;
  65 static int stack_skip = CONTENTION_STACK_SKIP;
  66 static int print_nr_entries = INT_MAX / 2;
  67 static LIST_HEAD(callstack_filters);
  68
  69 struct callstack_filter {
  70         struct list_head list;
  71         char name[];
  72 };
  73
  74 static struct lock_filter filters;
  75
  76 static enum lock_aggr_mode aggr_mode = LOCK_AGGR_ADDR;
  77
  78 static bool needs_callstack(void)
  79 {
  80         return verbose > 0 || !list_empty(&callstack_filters);
  81 }
  82
  83 static struct thread_stat *thread_stat_find(u32 tid)
  84 {
  85         struct rb_node *node;
  86         struct thread_stat *st;
  87
  88         node = thread_stats.rb_node;
  89         while (node) {
  90                 st = container_of(node, struct thread_stat, rb);
  91                 if (st->tid == tid)
  92                         return st;
  93                 else if (tid < st->tid)
  94                         node = node->rb_left;
  95                 else
  96                         node = node->rb_right;
  97         }
  98
  99         return NULL;
 100 }
 101
 102 static void thread_stat_insert(struct thread_stat *new)
 103 {
 104         struct rb_node **rb = &thread_stats.rb_node;
 105         struct rb_node *parent = NULL;
 106         struct thread_stat *p;
 107
 108         while (*rb) {
 109                 p = container_of(*rb, struct thread_stat, rb);
 110                 parent = *rb;
 111
 112                 if (new->tid < p->tid)
 113                         rb = &(*rb)->rb_left;
 114                 else if (new->tid > p->tid)
 115                         rb = &(*rb)->rb_right;
 116                 else
 117                         BUG_ON("inserting invalid thread_stat\n");
 118         }
 119
 120         rb_link_node(&new->rb, parent, rb);
 121         rb_insert_color(&new->rb, &thread_stats);
 122 }
 123
 124 static struct thread_stat *thread_stat_findnew_after_first(u32 tid)
 125 {
 126         struct thread_stat *st;
 127
 128         st = thread_stat_find(tid);
 129         if (st)
 130                 return st;
 131
 132         st = zalloc(sizeof(struct thread_stat));
 133         if (!st) {
 134                 pr_err("memory allocation failed\n");
 135                 return NULL;
 136         }
 137
 138         st->tid = tid;
 139         INIT_LIST_HEAD(&st->seq_list);
 140
 141         thread_stat_insert(st);
 142
 143         return st;
 144 }
 145
 146 static struct thread_stat *thread_stat_findnew_first(u32 tid);
 147 static struct thread_stat *(*thread_stat_findnew)(u32 tid) =
 148         thread_stat_findnew_first;
 149
 150 static struct thread_stat *thread_stat_findnew_first(u32 tid)
 151 {
 152         struct thread_stat *st;
 153
 154         st = zalloc(sizeof(struct thread_stat));
 155         if (!st) {
 156                 pr_err("memory allocation failed\n");
 157                 return NULL;
 158         }
 159         st->tid = tid;
 160         INIT_LIST_HEAD(&st->seq_list);
 161
 162         rb_link_node(&st->rb, NULL, &thread_stats.rb_node);
 163         rb_insert_color(&st->rb, &thread_stats);
 164
 165         thread_stat_findnew = thread_stat_findnew_after_first;
 166         return st;
 167 }
 168
 169 /* build simple key function one is bigger than two */
 170 #define SINGLE_KEY(member)                                              \
 171         static int lock_stat_key_ ## member(struct lock_stat *one,      \
 172                                          struct lock_stat *two)         \
 173         {                                                               \
 174                 return one->member > two->member;                       \
 175         }
 176
 177 SINGLE_KEY(nr_acquired)
 178 SINGLE_KEY(nr_contended)
 179 SINGLE_KEY(avg_wait_time)
 180 SINGLE_KEY(wait_time_total)
 181 SINGLE_KEY(wait_time_max)
 182
 183 static int lock_stat_key_wait_time_min(struct lock_stat *one,
 184                                         struct lock_stat *two)
 185 {
 186         u64 s1 = one->wait_time_min;
 187         u64 s2 = two->wait_time_min;
 188         if (s1 == ULLONG_MAX)
 189                 s1 = 0;
 190         if (s2 == ULLONG_MAX)
 191                 s2 = 0;
 192         return s1 > s2;
 193 }
 194
 195 struct lock_key {
 196         /*
 197          * name: the value for specify by user
 198          * this should be simpler than raw name of member
 199          * e.g. nr_acquired -> acquired, wait_time_total -> wait_total
 200          */
 201         const char              *name;
 202         /* header: the string printed on the header line */
 203         const char              *header;
 204         /* len: the printing width of the field */
 205         int                     len;
 206         /* key: a pointer to function to compare two lock stats for sorting */
 207         int                     (*key)(struct lock_stat*, struct lock_stat*);
 208         /* print: a pointer to function to print a given lock stats */
 209         void                    (*print)(struct lock_key*, struct lock_stat*);
 210         /* list: list entry to link this */
 211         struct list_head        list;
 212 };
 213
 214 static void lock_stat_key_print_time(unsigned long long nsec, int len)
 215 {
 216         static const struct {
 217                 float base;
 218                 const char *unit;
 219         } table[] = {
 220                 { 1e9 * 3600, "h " },
 221                 { 1e9 * 60, "m " },
 222                 { 1e9, "s " },
 223                 { 1e6, "ms" },
 224                 { 1e3, "us" },
 225                 { 0, NULL },
 226         };
 227
 228         for (int i = 0; table[i].unit; i++) {
 229                 if (nsec < table[i].base)
 230                         continue;
 231
 232                 pr_info("%*.2f %s", len - 3, nsec / table[i].base, table[i].unit);
 233                 return;
 234         }
 235
 236         pr_info("%*llu %s", len - 3, nsec, "ns");
 237 }
 238
 239 #define PRINT_KEY(member)                                               \
 240 static void lock_stat_key_print_ ## member(struct lock_key *key,        \
 241                                            struct lock_stat *ls)        \
 242 {                                                                       \
 243         pr_info("%*llu", key->len, (unsigned long long)ls->member);     \
 244 }
 245
 246 #define PRINT_TIME(member)                                              \
 247 static void lock_stat_key_print_ ## member(struct lock_key *key,        \
 248                                            struct lock_stat *ls)        \
 249 {                                                                       \
 250         lock_stat_key_print_time((unsigned long long)ls->member, key->len);     \
 251 }
 252
 253 PRINT_KEY(nr_acquired)
 254 PRINT_KEY(nr_contended)
 255 PRINT_TIME(avg_wait_time)
 256 PRINT_TIME(wait_time_total)
 257 PRINT_TIME(wait_time_max)
 258
 259 static void lock_stat_key_print_wait_time_min(struct lock_key *key,
 260                                               struct lock_stat *ls)
 261 {
 262         u64 wait_time = ls->wait_time_min;
 263
 264         if (wait_time == ULLONG_MAX)
 265                 wait_time = 0;
 266
 267         lock_stat_key_print_time(wait_time, key->len);
 268 }
 269
 270
 271 static const char               *sort_key = "acquired";
 272
 273 static int                      (*compare)(struct lock_stat *, struct lock_stat *);
 274
 275 static struct rb_root           sorted; /* place to store intermediate data */
 276 static struct rb_root           result; /* place to store sorted data */
 277
 278 static LIST_HEAD(lock_keys);
 279 static const char               *output_fields;
 280
 281 #define DEF_KEY_LOCK(name, header, fn_suffix, len)                      \
 282         { #name, header, len, lock_stat_key_ ## fn_suffix, lock_stat_key_print_ ## fn_suffix, {} }
 283 static struct lock_key report_keys[] = {
 284         DEF_KEY_LOCK(acquired, "acquired", nr_acquired, 10),
 285         DEF_KEY_LOCK(contended, "contended", nr_contended, 10),
 286         DEF_KEY_LOCK(avg_wait, "avg wait", avg_wait_time, 12),
 287         DEF_KEY_LOCK(wait_total, "total wait", wait_time_total, 12),
 288         DEF_KEY_LOCK(wait_max, "max wait", wait_time_max, 12),
 289         DEF_KEY_LOCK(wait_min, "min wait", wait_time_min, 12),
 290
 291         /* extra comparisons much complicated should be here */
 292         { }
 293 };
 294
 295 static struct lock_key contention_keys[] = {
 296         DEF_KEY_LOCK(contended, "contended", nr_contended, 10),
 297         DEF_KEY_LOCK(wait_total, "total wait", wait_time_total, 12),
 298         DEF_KEY_LOCK(wait_max, "max wait", wait_time_max, 12),
 299         DEF_KEY_LOCK(wait_min, "min wait", wait_time_min, 12),
 300         DEF_KEY_LOCK(avg_wait, "avg wait", avg_wait_time, 12),
 301
 302         /* extra comparisons much complicated should be here */
 303         { }
 304 };
 305
 306 static int select_key(bool contention)
 307 {
 308         int i;
 309         struct lock_key *keys = report_keys;
 310
 311         if (contention)
 312                 keys = contention_keys;
 313
 314         for (i = 0; keys[i].name; i++) {
 315                 if (!strcmp(keys[i].name, sort_key)) {
 316                         compare = keys[i].key;
 317
 318                         /* selected key should be in the output fields */
 319                         if (list_empty(&keys[i].list))
 320                                 list_add_tail(&keys[i].list, &lock_keys);
 321
 322                         return 0;
 323                 }
 324         }
 325
 326         pr_err("Unknown compare key: %s\n", sort_key);
 327         return -1;
 328 }
 329
 330 static int add_output_field(bool contention, char *name)
 331 {
 332         int i;
 333         struct lock_key *keys = report_keys;
 334
 335         if (contention)
 336                 keys = contention_keys;
 337
 338         for (i = 0; keys[i].name; i++) {
 339                 if (strcmp(keys[i].name, name))
 340                         continue;
 341
 342                 /* prevent double link */
 343                 if (list_empty(&keys[i].list))
 344                         list_add_tail(&keys[i].list, &lock_keys);
 345
 346                 return 0;
 347         }
 348
 349         pr_err("Unknown output field: %s\n", name);
 350         return -1;
 351 }
 352
 353 static int setup_output_field(bool contention, const char *str)
 354 {
 355         char *tok, *tmp, *orig;
 356         int i, ret = 0;
 357         struct lock_key *keys = report_keys;
 358
 359         if (contention)
 360                 keys = contention_keys;
 361
 362         /* no output field given: use all of them */
 363         if (str == NULL) {
 364                 for (i = 0; keys[i].name; i++)
 365                         list_add_tail(&keys[i].list, &lock_keys);
 366                 return 0;
 367         }
 368
 369         for (i = 0; keys[i].name; i++)
 370                 INIT_LIST_HEAD(&keys[i].list);
 371
 372         orig = tmp = strdup(str);
 373         if (orig == NULL)
 374                 return -ENOMEM;
 375
 376         while ((tok = strsep(&tmp, ",")) != NULL){
 377                 ret = add_output_field(contention, tok);
 378                 if (ret < 0)
 379                         break;
 380         }
 381         free(orig);
 382
 383         return ret;
 384 }
 385
 386 static void combine_lock_stats(struct lock_stat *st)
 387 {
 388         struct rb_node **rb = &sorted.rb_node;
 389         struct rb_node *parent = NULL;
 390         struct lock_stat *p;
 391         int ret;
 392
 393         while (*rb) {
 394                 p = container_of(*rb, struct lock_stat, rb);
 395                 parent = *rb;
 396
 397                 if (st->name && p->name)
 398                         ret = strcmp(st->name, p->name);
 399                 else
 400                         ret = !!st->name - !!p->name;
 401
 402                 if (ret == 0) {
 403                         p->nr_acquired += st->nr_acquired;
 404                         p->nr_contended += st->nr_contended;
 405                         p->wait_time_total += st->wait_time_total;
 406
 407                         if (p->nr_contended)
 408                                 p->avg_wait_time = p->wait_time_total / p->nr_contended;
 409
 410                         if (p->wait_time_min > st->wait_time_min)
 411                                 p->wait_time_min = st->wait_time_min;
 412                         if (p->wait_time_max < st->wait_time_max)
 413                                 p->wait_time_max = st->wait_time_max;
 414
 415                         p->broken |= st->broken;
 416                         st->combined = 1;
 417                         return;
 418                 }
 419
 420                 if (ret < 0)
 421                         rb = &(*rb)->rb_left;
 422                 else
 423                         rb = &(*rb)->rb_right;
 424         }
 425
 426         rb_link_node(&st->rb, parent, rb);
 427         rb_insert_color(&st->rb, &sorted);
 428 }
 429
 430 static void insert_to_result(struct lock_stat *st,
 431                              int (*bigger)(struct lock_stat *, struct lock_stat *))
 432 {
 433         struct rb_node **rb = &result.rb_node;
 434         struct rb_node *parent = NULL;
 435         struct lock_stat *p;
 436
 437         if (combine_locks && st->combined)
 438                 return;
 439
 440         while (*rb) {
 441                 p = container_of(*rb, struct lock_stat, rb);
 442                 parent = *rb;
 443
 444                 if (bigger(st, p))
 445                         rb = &(*rb)->rb_left;
 446                 else
 447                         rb = &(*rb)->rb_right;
 448         }
 449
 450         rb_link_node(&st->rb, parent, rb);
 451         rb_insert_color(&st->rb, &result);
 452 }
 453
 454 /* returns left most element of result, and erase it */
 455 static struct lock_stat *pop_from_result(void)
 456 {
 457         struct rb_node *node = result.rb_node;
 458
 459         if (!node)
 460                 return NULL;
 461
 462         while (node->rb_left)
 463                 node = node->rb_left;
 464
 465         rb_erase(node, &result);
 466         return container_of(node, struct lock_stat, rb);
 467 }
 468
 469 struct lock_stat *lock_stat_find(u64 addr)
 470 {
 471         struct hlist_head *entry = lockhashentry(addr);
 472         struct lock_stat *ret;
 473
 474         hlist_for_each_entry(ret, entry, hash_entry) {
 475                 if (ret->addr == addr)
 476                         return ret;
 477         }
 478         return NULL;
 479 }
 480
 481 struct lock_stat *lock_stat_findnew(u64 addr, const char *name, int flags)
 482 {
 483         struct hlist_head *entry = lockhashentry(addr);
 484         struct lock_stat *ret, *new;
 485
 486         hlist_for_each_entry(ret, entry, hash_entry) {
 487                 if (ret->addr == addr)
 488                         return ret;
 489         }
 490
 491         new = zalloc(sizeof(struct lock_stat));
 492         if (!new)
 493                 goto alloc_failed;
 494
 495         new->addr = addr;
 496         new->name = strdup(name);
 497         if (!new->name) {
 498                 free(new);
 499                 goto alloc_failed;
 500         }
 501
 502         new->flags = flags;
 503         new->wait_time_min = ULLONG_MAX;
 504
 505         hlist_add_head(&new->hash_entry, entry);
 506         return new;
 507
 508 alloc_failed:
 509         pr_err("memory allocation failed\n");
 510         return NULL;
 511 }
 512
 513 bool match_callstack_filter(struct machine *machine, u64 *callstack)
 514 {
 515         struct map *kmap;
 516         struct symbol *sym;
 517         u64 ip;
 518
 519         if (list_empty(&callstack_filters))
 520                 return true;
 521
 522         for (int i = 0; i < max_stack_depth; i++) {
 523                 struct callstack_filter *filter;
 524
 525                 if (!callstack || !callstack[i])
 526                         break;
 527
 528                 ip = callstack[i];
 529                 sym = machine__find_kernel_symbol(machine, ip, &kmap);
 530                 if (sym == NULL)
 531                         continue;
 532
 533                 list_for_each_entry(filter, &callstack_filters, list) {
 534                         if (strstr(sym->name, filter->name))
 535                                 return true;
 536                 }
 537         }
 538         return false;
 539 }
 540
 541 struct trace_lock_handler {
 542         /* it's used on CONFIG_LOCKDEP */
 543         int (*acquire_event)(struct evsel *evsel,
 544                              struct perf_sample *sample);
 545
 546         /* it's used on CONFIG_LOCKDEP && CONFIG_LOCK_STAT */
 547         int (*acquired_event)(struct evsel *evsel,
 548                               struct perf_sample *sample);
 549
 550         /* it's used on CONFIG_LOCKDEP && CONFIG_LOCK_STAT */
 551         int (*contended_event)(struct evsel *evsel,
 552                                struct perf_sample *sample);
 553
 554         /* it's used on CONFIG_LOCKDEP */
 555         int (*release_event)(struct evsel *evsel,
 556                              struct perf_sample *sample);
 557
 558         /* it's used when CONFIG_LOCKDEP is off */
 559         int (*contention_begin_event)(struct evsel *evsel,
 560                                       struct perf_sample *sample);
 561
 562         /* it's used when CONFIG_LOCKDEP is off */
 563         int (*contention_end_event)(struct evsel *evsel,
 564                                     struct perf_sample *sample);
 565 };
 566
 567 static struct lock_seq_stat *get_seq(struct thread_stat *ts, u64 addr)
 568 {
 569         struct lock_seq_stat *seq;
 570
 571         list_for_each_entry(seq, &ts->seq_list, list) {
 572                 if (seq->addr == addr)
 573                         return seq;
 574         }
 575
 576         seq = zalloc(sizeof(struct lock_seq_stat));
 577         if (!seq) {
 578                 pr_err("memory allocation failed\n");
 579                 return NULL;
 580         }
 581         seq->state = SEQ_STATE_UNINITIALIZED;
 582         seq->addr = addr;
 583
 584         list_add(&seq->list, &ts->seq_list);
 585         return seq;
 586 }
 587
 588 enum broken_state {
 589         BROKEN_ACQUIRE,
 590         BROKEN_ACQUIRED,
 591         BROKEN_CONTENDED,
 592         BROKEN_RELEASE,
 593         BROKEN_MAX,
 594 };
 595
 596 static int bad_hist[BROKEN_MAX];
 597
 598 enum acquire_flags {
 599         TRY_LOCK = 1,
 600         READ_LOCK = 2,
 601 };
 602
 603 static int get_key_by_aggr_mode_simple(u64 *key, u64 addr, u32 tid)
 604 {
 605         switch (aggr_mode) {
 606         case LOCK_AGGR_ADDR:
 607                 *key = addr;
 608                 break;
 609         case LOCK_AGGR_TASK:
 610                 *key = tid;
 611                 break;
 612         case LOCK_AGGR_CALLER:
 613         default:
 614                 pr_err("Invalid aggregation mode: %d\n", aggr_mode);
 615                 return -EINVAL;
 616         }
 617         return 0;
 618 }
 619
 620 static u64 callchain_id(struct evsel *evsel, struct perf_sample *sample);
 621
 622 static int get_key_by_aggr_mode(u64 *key, u64 addr, struct evsel *evsel,
 623                                  struct perf_sample *sample)
 624 {
 625         if (aggr_mode == LOCK_AGGR_CALLER) {
 626                 *key = callchain_id(evsel, sample);
 627                 return 0;
 628         }
 629         return get_key_by_aggr_mode_simple(key, addr, sample->tid);
 630 }
 631
 632 static int report_lock_acquire_event(struct evsel *evsel,
 633                                      struct perf_sample *sample)
 634 {
 635         struct lock_stat *ls;
 636         struct thread_stat *ts;
 637         struct lock_seq_stat *seq;
 638         const char *name = evsel__strval(evsel, sample, "name");
 639         u64 addr = evsel__intval(evsel, sample, "lockdep_addr");
 640         int flag = evsel__intval(evsel, sample, "flags");
 641         u64 key;
 642         int ret;
 643
 644         ret = get_key_by_aggr_mode_simple(&key, addr, sample->tid);
 645         if (ret < 0)
 646                 return ret;
 647
 648         ls = lock_stat_findnew(key, name, 0);
 649         if (!ls)
 650                 return -ENOMEM;
 651
 652         ts = thread_stat_findnew(sample->tid);
 653         if (!ts)
 654                 return -ENOMEM;
 655
 656         seq = get_seq(ts, addr);
 657         if (!seq)
 658                 return -ENOMEM;
 659
 660         switch (seq->state) {
 661         case SEQ_STATE_UNINITIALIZED:
 662         case SEQ_STATE_RELEASED:
 663                 if (!flag) {
 664                         seq->state = SEQ_STATE_ACQUIRING;
 665                 } else {
 666                         if (flag & TRY_LOCK)
 667                                 ls->nr_trylock++;
 668                         if (flag & READ_LOCK)
 669                                 ls->nr_readlock++;
 670                         seq->state = SEQ_STATE_READ_ACQUIRED;
 671                         seq->read_count = 1;
 672                         ls->nr_acquired++;
 673                 }
 674                 break;
 675         case SEQ_STATE_READ_ACQUIRED:
 676                 if (flag & READ_LOCK) {
 677                         seq->read_count++;
 678                         ls->nr_acquired++;
 679                         goto end;
 680                 } else {
 681                         goto broken;
 682                 }
 683                 break;
 684         case SEQ_STATE_ACQUIRED:
 685         case SEQ_STATE_ACQUIRING:
 686         case SEQ_STATE_CONTENDED:
 687 broken:
 688                 /* broken lock sequence */
 689                 if (!ls->broken) {
 690                         ls->broken = 1;
 691                         bad_hist[BROKEN_ACQUIRE]++;
 692                 }
 693                 list_del_init(&seq->list);
 694                 free(seq);
 695                 goto end;
 696         default:
 697                 BUG_ON("Unknown state of lock sequence found!\n");
 698                 break;
 699         }
 700
 701         ls->nr_acquire++;
 702         seq->prev_event_time = sample->time;
 703 end:
 704         return 0;
 705 }
 706
 707 static int report_lock_acquired_event(struct evsel *evsel,
 708                                       struct perf_sample *sample)
 709 {
 710         struct lock_stat *ls;
 711         struct thread_stat *ts;
 712         struct lock_seq_stat *seq;
 713         u64 contended_term;
 714         const char *name = evsel__strval(evsel, sample, "name");
 715         u64 addr = evsel__intval(evsel, sample, "lockdep_addr");
 716         u64 key;
 717         int ret;
 718
 719         ret = get_key_by_aggr_mode_simple(&key, addr, sample->tid);
 720         if (ret < 0)
 721                 return ret;
 722
 723         ls = lock_stat_findnew(key, name, 0);
 724         if (!ls)
 725                 return -ENOMEM;
 726
 727         ts = thread_stat_findnew(sample->tid);
 728         if (!ts)
 729                 return -ENOMEM;
 730
 731         seq = get_seq(ts, addr);
 732         if (!seq)
 733                 return -ENOMEM;
 734
 735         switch (seq->state) {
 736         case SEQ_STATE_UNINITIALIZED:
 737                 /* orphan event, do nothing */
 738                 return 0;
 739         case SEQ_STATE_ACQUIRING:
 740                 break;
 741         case SEQ_STATE_CONTENDED:
 742                 contended_term = sample->time - seq->prev_event_time;
 743                 ls->wait_time_total += contended_term;
 744                 if (contended_term < ls->wait_time_min)
 745                         ls->wait_time_min = contended_term;
 746                 if (ls->wait_time_max < contended_term)
 747                         ls->wait_time_max = contended_term;
 748                 break;
 749         case SEQ_STATE_RELEASED:
 750         case SEQ_STATE_ACQUIRED:
 751         case SEQ_STATE_READ_ACQUIRED:
 752                 /* broken lock sequence */
 753                 if (!ls->broken) {
 754                         ls->broken = 1;
 755                         bad_hist[BROKEN_ACQUIRED]++;
 756                 }
 757                 list_del_init(&seq->list);
 758                 free(seq);
 759                 goto end;
 760         default:
 761                 BUG_ON("Unknown state of lock sequence found!\n");
 762                 break;
 763         }
 764
 765         seq->state = SEQ_STATE_ACQUIRED;
 766         ls->nr_acquired++;
 767         ls->avg_wait_time = ls->nr_contended ? ls->wait_time_total/ls->nr_contended : 0;
 768         seq->prev_event_time = sample->time;
 769 end:
 770         return 0;
 771 }
 772
 773 static int report_lock_contended_event(struct evsel *evsel,
 774                                        struct perf_sample *sample)
 775 {
 776         struct lock_stat *ls;
 777         struct thread_stat *ts;
 778         struct lock_seq_stat *seq;
 779         const char *name = evsel__strval(evsel, sample, "name");
 780         u64 addr = evsel__intval(evsel, sample, "lockdep_addr");
 781         u64 key;
 782         int ret;
 783
 784         ret = get_key_by_aggr_mode_simple(&key, addr, sample->tid);
 785         if (ret < 0)
 786                 return ret;
 787
 788         ls = lock_stat_findnew(key, name, 0);
 789         if (!ls)
 790                 return -ENOMEM;
 791
 792         ts = thread_stat_findnew(sample->tid);
 793         if (!ts)
 794                 return -ENOMEM;
 795
 796         seq = get_seq(ts, addr);
 797         if (!seq)
 798                 return -ENOMEM;
 799
 800         switch (seq->state) {
 801         case SEQ_STATE_UNINITIALIZED:
 802                 /* orphan event, do nothing */
 803                 return 0;
 804         case SEQ_STATE_ACQUIRING:
 805                 break;
 806         case SEQ_STATE_RELEASED:
 807         case SEQ_STATE_ACQUIRED:
 808         case SEQ_STATE_READ_ACQUIRED:
 809         case SEQ_STATE_CONTENDED:
 810                 /* broken lock sequence */
 811                 if (!ls->broken) {
 812                         ls->broken = 1;
 813                         bad_hist[BROKEN_CONTENDED]++;
 814                 }
 815                 list_del_init(&seq->list);
 816                 free(seq);
 817                 goto end;
 818         default:
 819                 BUG_ON("Unknown state of lock sequence found!\n");
 820                 break;
 821         }
 822
 823         seq->state = SEQ_STATE_CONTENDED;
 824         ls->nr_contended++;
 825         ls->avg_wait_time = ls->wait_time_total/ls->nr_contended;
 826         seq->prev_event_time = sample->time;
 827 end:
 828         return 0;
 829 }
 830
 831 static int report_lock_release_event(struct evsel *evsel,
 832                                      struct perf_sample *sample)
 833 {
 834         struct lock_stat *ls;
 835         struct thread_stat *ts;
 836         struct lock_seq_stat *seq;
 837         const char *name = evsel__strval(evsel, sample, "name");
 838         u64 addr = evsel__intval(evsel, sample, "lockdep_addr");
 839         u64 key;
 840         int ret;
 841
 842         ret = get_key_by_aggr_mode_simple(&key, addr, sample->tid);
 843         if (ret < 0)
 844                 return ret;
 845
 846         ls = lock_stat_findnew(key, name, 0);
 847         if (!ls)
 848                 return -ENOMEM;
 849
 850         ts = thread_stat_findnew(sample->tid);
 851         if (!ts)
 852                 return -ENOMEM;
 853
 854         seq = get_seq(ts, addr);
 855         if (!seq)
 856                 return -ENOMEM;
 857
 858         switch (seq->state) {
 859         case SEQ_STATE_UNINITIALIZED:
 860                 goto end;
 861         case SEQ_STATE_ACQUIRED:
 862                 break;
 863         case SEQ_STATE_READ_ACQUIRED:
 864                 seq->read_count--;
 865                 BUG_ON(seq->read_count < 0);
 866                 if (seq->read_count) {
 867                         ls->nr_release++;
 868                         goto end;
 869                 }
 870                 break;
 871         case SEQ_STATE_ACQUIRING:
 872         case SEQ_STATE_CONTENDED:
 873         case SEQ_STATE_RELEASED:
 874                 /* broken lock sequence */
 875                 if (!ls->broken) {
 876                         ls->broken = 1;
 877                         bad_hist[BROKEN_RELEASE]++;
 878                 }
 879                 goto free_seq;
 880         default:
 881                 BUG_ON("Unknown state of lock sequence found!\n");
 882                 break;
 883         }
 884
 885         ls->nr_release++;
 886 free_seq:
 887         list_del_init(&seq->list);
 888         free(seq);
 889 end:
 890         return 0;
 891 }
 892
 893 static int get_symbol_name_offset(struct map *map, struct symbol *sym, u64 ip,
 894                                   char *buf, int size)
 895 {
 896         u64 offset;
 897
 898         if (map == NULL || sym == NULL) {
 899                 buf[0] = '\0';
 900                 return 0;
 901         }
 902
 903         offset = map->map_ip(map, ip) - sym->start;
 904
 905         if (offset)
 906                 return scnprintf(buf, size, "%s+%#lx", sym->name, offset);
 907         else
 908                 return strlcpy(buf, sym->name, size);
 909 }
 910 static int lock_contention_caller(struct evsel *evsel, struct perf_sample *sample,
 911                                   char *buf, int size)
 912 {
 913         struct thread *thread;
 914         struct callchain_cursor *cursor = &callchain_cursor;
 915         struct machine *machine = &session->machines.host;
 916         struct symbol *sym;
 917         int skip = 0;
 918         int ret;
 919
 920         /* lock names will be replaced to task name later */
 921         if (show_thread_stats)
 922                 return -1;
 923
 924         thread = machine__findnew_thread(machine, -1, sample->pid);
 925         if (thread == NULL)
 926                 return -1;
 927
 928         /* use caller function name from the callchain */
 929         ret = thread__resolve_callchain(thread, cursor, evsel, sample,
 930                                         NULL, NULL, max_stack_depth);
 931         if (ret != 0) {
 932                 thread__put(thread);
 933                 return -1;
 934         }
 935
 936         callchain_cursor_commit(cursor);
 937         thread__put(thread);
 938
 939         while (true) {
 940                 struct callchain_cursor_node *node;
 941
 942                 node = callchain_cursor_current(cursor);
 943                 if (node == NULL)
 944                         break;
 945
 946                 /* skip first few entries - for lock functions */
 947                 if (++skip <= stack_skip)
 948                         goto next;
 949
 950                 sym = node->ms.sym;
 951                 if (sym && !machine__is_lock_function(machine, node->ip)) {
 952                         get_symbol_name_offset(node->ms.map, sym, node->ip,
 953                                                buf, size);
 954                         return 0;
 955                 }
 956
 957 next:
 958                 callchain_cursor_advance(cursor);
 959         }
 960         return -1;
 961 }
 962
 963 static u64 callchain_id(struct evsel *evsel, struct perf_sample *sample)
 964 {
 965         struct callchain_cursor *cursor = &callchain_cursor;
 966         struct machine *machine = &session->machines.host;
 967         struct thread *thread;
 968         u64 hash = 0;
 969         int skip = 0;
 970         int ret;
 971
 972         thread = machine__findnew_thread(machine, -1, sample->pid);
 973         if (thread == NULL)
 974                 return -1;
 975
 976         /* use caller function name from the callchain */
 977         ret = thread__resolve_callchain(thread, cursor, evsel, sample,
 978                                         NULL, NULL, max_stack_depth);
 979         thread__put(thread);
 980
 981         if (ret != 0)
 982                 return -1;
 983
 984         callchain_cursor_commit(cursor);
 985
 986         while (true) {
 987                 struct callchain_cursor_node *node;
 988
 989                 node = callchain_cursor_current(cursor);
 990                 if (node == NULL)
 991                         break;
 992
 993                 /* skip first few entries - for lock functions */
 994                 if (++skip <= stack_skip)
 995                         goto next;
 996
 997                 if (node->ms.sym && machine__is_lock_function(machine, node->ip))
 998                         goto next;
 999
1000                 hash ^= hash_long((unsigned long)node->ip, 64);
1001
1002 next:
1003                 callchain_cursor_advance(cursor);
1004         }
1005         return hash;
1006 }
1007
1008 static u64 *get_callstack(struct perf_sample *sample, int max_stack)
1009 {
1010         u64 *callstack;
1011         u64 i;
1012         int c;
1013
1014         callstack = calloc(max_stack, sizeof(*callstack));
1015         if (callstack == NULL)
1016                 return NULL;
1017
1018         for (i = 0, c = 0; i < sample->callchain->nr && c < max_stack; i++) {
1019                 u64 ip = sample->callchain->ips[i];
1020
1021                 if (ip >= PERF_CONTEXT_MAX)
1022                         continue;
1023
1024                 callstack[c++] = ip;
1025         }
1026         return callstack;
1027 }
1028
1029 static int report_lock_contention_begin_event(struct evsel *evsel,
1030                                               struct perf_sample *sample)
1031 {
1032         struct lock_stat *ls;
1033         struct thread_stat *ts;
1034         struct lock_seq_stat *seq;
1035         u64 addr = evsel__intval(evsel, sample, "lock_addr");
1036         unsigned int flags = evsel__intval(evsel, sample, "flags");
1037         u64 key;
1038         int i, ret;
1039         static bool kmap_loaded;
1040         struct machine *machine = &session->machines.host;
1041         struct map *kmap;
1042         struct symbol *sym;
1043
1044         ret = get_key_by_aggr_mode(&key, addr, evsel, sample);
1045         if (ret < 0)
1046                 return ret;
1047
1048         if (!kmap_loaded) {
1049                 unsigned long *addrs;
1050
1051                 /* make sure it loads the kernel map to find lock symbols */
1052                 map__load(machine__kernel_map(machine));
1053                 kmap_loaded = true;
1054
1055                 /* convert (kernel) symbols to addresses */
1056                 for (i = 0; i < filters.nr_syms; i++) {
1057                         sym = machine__find_kernel_symbol_by_name(machine,
1058                                                                   filters.syms[i],
1059                                                                   &kmap);
1060                         if (sym == NULL) {
1061                                 pr_warning("ignore unknown symbol: %s\n",
1062                                            filters.syms[i]);
1063                                 continue;
1064                         }
1065
1066                         addrs = realloc(filters.addrs,
1067                                         (filters.nr_addrs + 1) * sizeof(*addrs));
1068                         if (addrs == NULL) {
1069                                 pr_warning("memory allocation failure\n");
1070                                 return -ENOMEM;
1071                         }
1072
1073                         addrs[filters.nr_addrs++] = kmap->unmap_ip(kmap, sym->start);
1074                         filters.addrs = addrs;
1075                 }
1076         }
1077
1078         ls = lock_stat_find(key);
1079         if (!ls) {
1080                 char buf[128];
1081                 const char *name = "";
1082
1083                 switch (aggr_mode) {
1084                 case LOCK_AGGR_ADDR:
1085                         sym = machine__find_kernel_symbol(machine, key, &kmap);
1086                         if (sym)
1087                                 name = sym->name;
1088                         break;
1089                 case LOCK_AGGR_CALLER:
1090                         name = buf;
1091                         if (lock_contention_caller(evsel, sample, buf, sizeof(buf)) < 0)
1092                                 name = "Unknown";
1093                         break;
1094                 case LOCK_AGGR_TASK:
1095                 default:
1096                         break;
1097                 }
1098
1099                 ls = lock_stat_findnew(key, name, flags);
1100                 if (!ls)
1101                         return -ENOMEM;
1102         }
1103
1104         if (filters.nr_types) {
1105                 bool found = false;
1106
1107                 for (i = 0; i < filters.nr_types; i++) {
1108                         if (flags == filters.types[i]) {
1109                                 found = true;
1110                                 break;
1111                         }
1112                 }
1113
1114                 if (!found)
1115                         return 0;
1116         }
1117
1118         if (filters.nr_addrs) {
1119                 bool found = false;
1120
1121                 for (i = 0; i < filters.nr_addrs; i++) {
1122                         if (addr == filters.addrs[i]) {
1123                                 found = true;
1124                                 break;
1125                         }
1126                 }
1127
1128                 if (!found)
1129                         return 0;
1130         }
1131
1132         if (needs_callstack()) {
1133                 u64 *callstack = get_callstack(sample, max_stack_depth);
1134                 if (callstack == NULL)
1135                         return -ENOMEM;
1136
1137                 if (!match_callstack_filter(machine, callstack)) {
1138                         free(callstack);
1139                         return 0;
1140                 }
1141
1142                 if (ls->callstack == NULL)
1143                         ls->callstack = callstack;
1144                 else
1145                         free(callstack);
1146         }
1147
1148         ts = thread_stat_findnew(sample->tid);
1149         if (!ts)
1150                 return -ENOMEM;
1151
1152         seq = get_seq(ts, addr);
1153         if (!seq)
1154                 return -ENOMEM;
1155
1156         switch (seq->state) {
1157         case SEQ_STATE_UNINITIALIZED:
1158         case SEQ_STATE_ACQUIRED:
1159                 break;
1160         case SEQ_STATE_CONTENDED:
1161                 /*
1162                  * It can have nested contention begin with mutex spinning,
1163                  * then we would use the original contention begin event and
1164                  * ignore the second one.
1165                  */
1166                 goto end;
1167         case SEQ_STATE_ACQUIRING:
1168         case SEQ_STATE_READ_ACQUIRED:
1169         case SEQ_STATE_RELEASED:
1170                 /* broken lock sequence */
1171                 if (!ls->broken) {
1172                         ls->broken = 1;
1173                         bad_hist[BROKEN_CONTENDED]++;
1174                 }
1175                 list_del_init(&seq->list);
1176                 free(seq);
1177                 goto end;
1178         default:
1179                 BUG_ON("Unknown state of lock sequence found!\n");
1180                 break;
1181         }
1182
1183         if (seq->state != SEQ_STATE_CONTENDED) {
1184                 seq->state = SEQ_STATE_CONTENDED;
1185                 seq->prev_event_time = sample->time;
1186                 ls->nr_contended++;
1187         }
1188 end:
1189         return 0;
1190 }
1191
1192 static int report_lock_contention_end_event(struct evsel *evsel,
1193                                             struct perf_sample *sample)
1194 {
1195         struct lock_stat *ls;
1196         struct thread_stat *ts;
1197         struct lock_seq_stat *seq;
1198         u64 contended_term;
1199         u64 addr = evsel__intval(evsel, sample, "lock_addr");
1200         u64 key;
1201         int ret;
1202
1203         ret = get_key_by_aggr_mode(&key, addr, evsel, sample);
1204         if (ret < 0)
1205                 return ret;
1206
1207         ls = lock_stat_find(key);
1208         if (!ls)
1209                 return 0;
1210
1211         ts = thread_stat_find(sample->tid);
1212         if (!ts)
1213                 return 0;
1214
1215         seq = get_seq(ts, addr);
1216         if (!seq)
1217                 return -ENOMEM;
1218
1219         switch (seq->state) {
1220         case SEQ_STATE_UNINITIALIZED:
1221                 goto end;
1222         case SEQ_STATE_CONTENDED:
1223                 contended_term = sample->time - seq->prev_event_time;
1224                 ls->wait_time_total += contended_term;
1225                 if (contended_term < ls->wait_time_min)
1226                         ls->wait_time_min = contended_term;
1227                 if (ls->wait_time_max < contended_term)
1228                         ls->wait_time_max = contended_term;
1229                 break;
1230         case SEQ_STATE_ACQUIRING:
1231         case SEQ_STATE_ACQUIRED:
1232         case SEQ_STATE_READ_ACQUIRED:
1233         case SEQ_STATE_RELEASED:
1234                 /* broken lock sequence */
1235                 if (!ls->broken) {
1236                         ls->broken = 1;
1237                         bad_hist[BROKEN_ACQUIRED]++;
1238                 }
1239                 list_del_init(&seq->list);
1240                 free(seq);
1241                 goto end;
1242         default:
1243                 BUG_ON("Unknown state of lock sequence found!\n");
1244                 break;
1245         }
1246
1247         seq->state = SEQ_STATE_ACQUIRED;
1248         ls->nr_acquired++;
1249         ls->avg_wait_time = ls->wait_time_total/ls->nr_acquired;
1250 end:
1251         return 0;
1252 }
1253
1254 /* lock oriented handlers */
1255 /* TODO: handlers for CPU oriented, thread oriented */
1256 static struct trace_lock_handler report_lock_ops  = {
1257         .acquire_event          = report_lock_acquire_event,
1258         .acquired_event         = report_lock_acquired_event,
1259         .contended_event        = report_lock_contended_event,
1260         .release_event          = report_lock_release_event,
1261         .contention_begin_event = report_lock_contention_begin_event,
1262         .contention_end_event   = report_lock_contention_end_event,
1263 };
1264
1265 static struct trace_lock_handler contention_lock_ops  = {
1266         .contention_begin_event = report_lock_contention_begin_event,
1267         .contention_end_event   = report_lock_contention_end_event,
1268 };
1269
1270
1271 static struct trace_lock_handler *trace_handler;
1272
1273 static int evsel__process_lock_acquire(struct evsel *evsel, struct perf_sample *sample)
1274 {
1275         if (trace_handler->acquire_event)
1276                 return trace_handler->acquire_event(evsel, sample);
1277         return 0;
1278 }
1279
1280 static int evsel__process_lock_acquired(struct evsel *evsel, struct perf_sample *sample)
1281 {
1282         if (trace_handler->acquired_event)
1283                 return trace_handler->acquired_event(evsel, sample);
1284         return 0;
1285 }
1286
1287 static int evsel__process_lock_contended(struct evsel *evsel, struct perf_sample *sample)
1288 {
1289         if (trace_handler->contended_event)
1290                 return trace_handler->contended_event(evsel, sample);
1291         return 0;
1292 }
1293
1294 static int evsel__process_lock_release(struct evsel *evsel, struct perf_sample *sample)
1295 {
1296         if (trace_handler->release_event)
1297                 return trace_handler->release_event(evsel, sample);
1298         return 0;
1299 }
1300
1301 static int evsel__process_contention_begin(struct evsel *evsel, struct perf_sample *sample)
1302 {
1303         if (trace_handler->contention_begin_event)
1304                 return trace_handler->contention_begin_event(evsel, sample);
1305         return 0;
1306 }
1307
1308 static int evsel__process_contention_end(struct evsel *evsel, struct perf_sample *sample)
1309 {
1310         if (trace_handler->contention_end_event)
1311                 return trace_handler->contention_end_event(evsel, sample);
1312         return 0;
1313 }
1314
1315 static void print_bad_events(int bad, int total)
1316 {
1317         /* Output for debug, this have to be removed */
1318         int i;
1319         int broken = 0;
1320         const char *name[4] =
1321                 { "acquire", "acquired", "contended", "release" };
1322
1323         for (i = 0; i < BROKEN_MAX; i++)
1324                 broken += bad_hist[i];
1325
1326         if (quiet || total == 0 || (broken == 0 && verbose <= 0))
1327                 return;
1328
1329         pr_info("\n=== output for debug ===\n\n");
1330         pr_info("bad: %d, total: %d\n", bad, total);
1331         pr_info("bad rate: %.2f %%\n", (double)bad / (double)total * 100);
1332         pr_info("histogram of events caused bad sequence\n");
1333         for (i = 0; i < BROKEN_MAX; i++)
1334                 pr_info(" %10s: %d\n", name[i], bad_hist[i]);
1335 }
1336
1337 /* TODO: various way to print, coloring, nano or milli sec */
1338 static void print_result(void)
1339 {
1340         struct lock_stat *st;
1341         struct lock_key *key;
1342         char cut_name[20];
1343         int bad, total, printed;
1344
1345         if (!quiet) {
1346                 pr_info("%20s ", "Name");
1347                 list_for_each_entry(key, &lock_keys, list)
1348                         pr_info("%*s ", key->len, key->header);
1349                 pr_info("\n\n");
1350         }
1351
1352         bad = total = printed = 0;
1353         while ((st = pop_from_result())) {
1354                 total++;
1355                 if (st->broken)
1356                         bad++;
1357                 if (!st->nr_acquired)
1358                         continue;
1359
1360                 bzero(cut_name, 20);
1361
1362                 if (strlen(st->name) < 20) {
1363                         /* output raw name */
1364                         const char *name = st->name;
1365
1366                         if (show_thread_stats) {
1367                                 struct thread *t;
1368
1369                                 /* st->addr contains tid of thread */
1370                                 t = perf_session__findnew(session, st->addr);
1371                                 name = thread__comm_str(t);
1372                         }
1373
1374                         pr_info("%20s ", name);
1375                 } else {
1376                         strncpy(cut_name, st->name, 16);
1377                         cut_name[16] = '.';
1378                         cut_name[17] = '.';
1379                         cut_name[18] = '.';
1380                         cut_name[19] = '\0';
1381                         /* cut off name for saving output style */
1382                         pr_info("%20s ", cut_name);
1383                 }
1384
1385                 list_for_each_entry(key, &lock_keys, list) {
1386                         key->print(key, st);
1387                         pr_info(" ");
1388                 }
1389                 pr_info("\n");
1390
1391                 if (++printed >= print_nr_entries)
1392                         break;
1393         }
1394
1395         print_bad_events(bad, total);
1396 }
1397
1398 static bool info_threads, info_map;
1399
1400 static void dump_threads(void)
1401 {
1402         struct thread_stat *st;
1403         struct rb_node *node;
1404         struct thread *t;
1405
1406         pr_info("%10s: comm\n", "Thread ID");
1407
1408         node = rb_first(&thread_stats);
1409         while (node) {
1410                 st = container_of(node, struct thread_stat, rb);
1411                 t = perf_session__findnew(session, st->tid);
1412                 pr_info("%10d: %s\n", st->tid, thread__comm_str(t));
1413                 node = rb_next(node);
1414                 thread__put(t);
1415         }
1416 }
1417
1418 static int compare_maps(struct lock_stat *a, struct lock_stat *b)
1419 {
1420         int ret;
1421
1422         if (a->name && b->name)
1423                 ret = strcmp(a->name, b->name);
1424         else
1425                 ret = !!a->name - !!b->name;
1426
1427         if (!ret)
1428                 return a->addr < b->addr;
1429         else
1430                 return ret < 0;
1431 }
1432
1433 static void dump_map(void)
1434 {
1435         unsigned int i;
1436         struct lock_stat *st;
1437
1438         pr_info("Address of instance: name of class\n");
1439         for (i = 0; i < LOCKHASH_SIZE; i++) {
1440                 hlist_for_each_entry(st, &lockhash_table[i], hash_entry) {
1441                         insert_to_result(st, compare_maps);
1442                 }
1443         }
1444
1445         while ((st = pop_from_result()))
1446                 pr_info(" %#llx: %s\n", (unsigned long long)st->addr, st->name);
1447 }
1448
1449 static int dump_info(void)
1450 {
1451         int rc = 0;
1452
1453         if (info_threads)
1454                 dump_threads();
1455         else if (info_map)
1456                 dump_map();
1457         else {
1458                 rc = -1;
1459                 pr_err("Unknown type of information\n");
1460         }
1461
1462         return rc;
1463 }
1464
1465 static const struct evsel_str_handler lock_tracepoints[] = {
1466         { "lock:lock_acquire",   evsel__process_lock_acquire,   }, /* CONFIG_LOCKDEP */
1467         { "lock:lock_acquired",  evsel__process_lock_acquired,  }, /* CONFIG_LOCKDEP, CONFIG_LOCK_STAT */
1468         { "lock:lock_contended", evsel__process_lock_contended, }, /* CONFIG_LOCKDEP, CONFIG_LOCK_STAT */
1469         { "lock:lock_release",   evsel__process_lock_release,   }, /* CONFIG_LOCKDEP */
1470 };
1471
1472 static const struct evsel_str_handler contention_tracepoints[] = {
1473         { "lock:contention_begin", evsel__process_contention_begin, },
1474         { "lock:contention_end",   evsel__process_contention_end,   },
1475 };
1476
1477 static int process_event_update(struct perf_tool *tool,
1478                                 union perf_event *event,
1479                                 struct evlist **pevlist)
1480 {
1481         int ret;
1482
1483         ret = perf_event__process_event_update(tool, event, pevlist);
1484         if (ret < 0)
1485                 return ret;
1486
1487         /* this can return -EEXIST since we call it for each evsel */
1488         perf_session__set_tracepoints_handlers(session, lock_tracepoints);
1489         perf_session__set_tracepoints_handlers(session, contention_tracepoints);
1490         return 0;
1491 }
1492
1493 typedef int (*tracepoint_handler)(struct evsel *evsel,
1494                                   struct perf_sample *sample);
1495
1496 static int process_sample_event(struct perf_tool *tool __maybe_unused,
1497                                 union perf_event *event,
1498                                 struct perf_sample *sample,
1499                                 struct evsel *evsel,
1500                                 struct machine *machine)
1501 {
1502         int err = 0;
1503         struct thread *thread = machine__findnew_thread(machine, sample->pid,
1504                                                         sample->tid);
1505
1506         if (thread == NULL) {
1507                 pr_debug("problem processing %d event, skipping it.\n",
1508                         event->header.type);
1509                 return -1;
1510         }
1511
1512         if (evsel->handler != NULL) {
1513                 tracepoint_handler f = evsel->handler;
1514                 err = f(evsel, sample);
1515         }
1516
1517         thread__put(thread);
1518
1519         return err;
1520 }
1521
1522 static void combine_result(void)
1523 {
1524         unsigned int i;
1525         struct lock_stat *st;
1526
1527         if (!combine_locks)
1528                 return;
1529
1530         for (i = 0; i < LOCKHASH_SIZE; i++) {
1531                 hlist_for_each_entry(st, &lockhash_table[i], hash_entry) {
1532                         combine_lock_stats(st);
1533                 }
1534         }
1535 }
1536
1537 static void sort_result(void)
1538 {
1539         unsigned int i;
1540         struct lock_stat *st;
1541
1542         for (i = 0; i < LOCKHASH_SIZE; i++) {
1543                 hlist_for_each_entry(st, &lockhash_table[i], hash_entry) {
1544                         insert_to_result(st, compare);
1545                 }
1546         }
1547 }
1548
1549 static const struct {
1550         unsigned int flags;
1551         const char *str;
1552         const char *name;
1553 } lock_type_table[] = {
1554         { 0,                            "semaphore",    "semaphore" },
1555         { LCB_F_SPIN,                   "spinlock",     "spinlock" },
1556         { LCB_F_SPIN | LCB_F_READ,      "rwlock:R",     "rwlock" },
1557         { LCB_F_SPIN | LCB_F_WRITE,     "rwlock:W",     "rwlock" },
1558         { LCB_F_READ,                   "rwsem:R",      "rwsem" },
1559         { LCB_F_WRITE,                  "rwsem:W",      "rwsem" },
1560         { LCB_F_RT,                     "rt-mutex",     "rt-mutex" },
1561         { LCB_F_RT | LCB_F_READ,        "rwlock-rt:R",  "rwlock-rt" },
1562         { LCB_F_RT | LCB_F_WRITE,       "rwlock-rt:W",  "rwlock-rt" },
1563         { LCB_F_PERCPU | LCB_F_READ,    "pcpu-sem:R",   "percpu-rwsem" },
1564         { LCB_F_PERCPU | LCB_F_WRITE,   "pcpu-sem:W",   "percpu-rwsem" },
1565         { LCB_F_MUTEX,                  "mutex",        "mutex" },
1566         { LCB_F_MUTEX | LCB_F_SPIN,     "mutex",        "mutex" },
1567         /* alias for get_type_flag() */
1568         { LCB_F_MUTEX | LCB_F_SPIN,     "mutex-spin",   "mutex" },
1569 };
1570
1571 static const char *get_type_str(unsigned int flags)
1572 {
1573         flags &= LCB_F_MAX_FLAGS - 1;
1574
1575         for (unsigned int i = 0; i < ARRAY_SIZE(lock_type_table); i++) {
1576                 if (lock_type_table[i].flags == flags)
1577                         return lock_type_table[i].str;
1578         }
1579         return "unknown";
1580 }
1581
1582 static const char *get_type_name(unsigned int flags)
1583 {
1584         flags &= LCB_F_MAX_FLAGS - 1;
1585
1586         for (unsigned int i = 0; i < ARRAY_SIZE(lock_type_table); i++) {
1587                 if (lock_type_table[i].flags == flags)
1588                         return lock_type_table[i].name;
1589         }
1590         return "unknown";
1591 }
1592
1593 static unsigned int get_type_flag(const char *str)
1594 {
1595         for (unsigned int i = 0; i < ARRAY_SIZE(lock_type_table); i++) {
1596                 if (!strcmp(lock_type_table[i].name, str))
1597                         return lock_type_table[i].flags;
1598         }
1599         for (unsigned int i = 0; i < ARRAY_SIZE(lock_type_table); i++) {
1600                 if (!strcmp(lock_type_table[i].str, str))
1601                         return lock_type_table[i].flags;
1602         }
1603         return UINT_MAX;
1604 }
1605
1606 static void lock_filter_finish(void)
1607 {
1608         zfree(&filters.types);
1609         filters.nr_types = 0;
1610
1611         zfree(&filters.addrs);
1612         filters.nr_addrs = 0;
1613
1614         for (int i = 0; i < filters.nr_syms; i++)
1615                 free(filters.syms[i]);
1616
1617         zfree(&filters.syms);
1618         filters.nr_syms = 0;
1619 }
1620
1621 static void sort_contention_result(void)
1622 {
1623         sort_result();
1624 }
1625
1626 static void print_bpf_events(int total, struct lock_contention_fails *fails)
1627 {
1628         /* Output for debug, this have to be removed */
1629         int broken = fails->task + fails->stack + fails->time;
1630
1631         if (quiet || total == 0 || (broken == 0 && verbose <= 0))
1632                 return;
1633
1634         total += broken;
1635         pr_info("\n=== output for debug ===\n\n");
1636         pr_info("bad: %d, total: %d\n", broken, total);
1637         pr_info("bad rate: %.2f %%\n", (double)broken / (double)total * 100);
1638
1639         pr_info("histogram of failure reasons\n");
1640         pr_info(" %10s: %d\n", "task", fails->task);
1641         pr_info(" %10s: %d\n", "stack", fails->stack);
1642         pr_info(" %10s: %d\n", "time", fails->time);
1643 }
1644 static void print_contention_result(struct lock_contention *con)
1645 {
1646         struct lock_stat *st;
1647         struct lock_key *key;
1648         int bad, total, printed;
1649
1650         if (!quiet) {
1651                 list_for_each_entry(key, &lock_keys, list)
1652                         pr_info("%*s ", key->len, key->header);
1653
1654                 switch (aggr_mode) {
1655                 case LOCK_AGGR_TASK:
1656                         pr_info("  %10s   %s\n\n", "pid",
1657                                 show_lock_owner ? "owner" : "comm");
1658                         break;
1659                 case LOCK_AGGR_CALLER:
1660                         pr_info("  %10s   %s\n\n", "type", "caller");
1661                         break;
1662                 case LOCK_AGGR_ADDR:
1663                         pr_info("  %16s   %s\n\n", "address", "symbol");
1664                         break;
1665                 default:
1666                         break;
1667                 }
1668         }
1669
1670         bad = total = printed = 0;
1671
1672         while ((st = pop_from_result())) {
1673                 struct thread *t;
1674                 int pid;
1675
1676                 total += use_bpf ? st->nr_contended : 1;
1677                 if (st->broken)
1678                         bad++;
1679
1680                 if (!st->wait_time_total)
1681                         continue;
1682
1683                 list_for_each_entry(key, &lock_keys, list) {
1684                         key->print(key, st);
1685                         pr_info(" ");
1686                 }
1687
1688                 switch (aggr_mode) {
1689                 case LOCK_AGGR_CALLER:
1690                         pr_info("  %10s   %s\n", get_type_str(st->flags), st->name);
1691                         break;
1692                 case LOCK_AGGR_TASK:
1693                         pid = st->addr;
1694                         t = perf_session__findnew(session, pid);
1695                         pr_info("  %10d   %s\n",
1696                                 pid, pid == -1 ? "Unknown" : thread__comm_str(t));
1697                         break;
1698                 case LOCK_AGGR_ADDR:
1699                         pr_info("  %016llx   %s (%s)\n", (unsigned long long)st->addr,
1700                                 st->name, get_type_name(st->flags));
1701                         break;
1702                 default:
1703                         break;
1704                 }
1705
1706                 if (aggr_mode == LOCK_AGGR_CALLER && verbose > 0) {
1707                         struct map *kmap;
1708                         struct symbol *sym;
1709                         char buf[128];
1710                         u64 ip;
1711
1712                         for (int i = 0; i < max_stack_depth; i++) {
1713                                 if (!st->callstack || !st->callstack[i])
1714                                         break;
1715
1716                                 ip = st->callstack[i];
1717                                 sym = machine__find_kernel_symbol(con->machine, ip, &kmap);
1718                                 get_symbol_name_offset(kmap, sym, ip, buf, sizeof(buf));
1719                                 pr_info("\t\t\t%#lx  %s\n", (unsigned long)ip, buf);
1720                         }
1721                 }
1722
1723                 if (++printed >= print_nr_entries)
1724                         break;
1725         }
1726
1727         if (use_bpf)
1728                 print_bpf_events(total, &con->fails);
1729         else
1730                 print_bad_events(bad, total);
1731 }
1732
1733 static bool force;
1734
1735 static int __cmd_report(bool display_info)
1736 {
1737         int err = -EINVAL;
1738         struct perf_tool eops = {
1739                 .attr            = perf_event__process_attr,
1740                 .event_update    = process_event_update,
1741                 .sample          = process_sample_event,
1742                 .comm            = perf_event__process_comm,
1743                 .mmap            = perf_event__process_mmap,
1744                 .namespaces      = perf_event__process_namespaces,
1745                 .tracing_data    = perf_event__process_tracing_data,
1746                 .ordered_events  = true,
1747         };
1748         struct perf_data data = {
1749                 .path  = input_name,
1750                 .mode  = PERF_DATA_MODE_READ,
1751                 .force = force,
1752         };
1753
1754         session = perf_session__new(&data, &eops);
1755         if (IS_ERR(session)) {
1756                 pr_err("Initializing perf session failed\n");
1757                 return PTR_ERR(session);
1758         }
1759
1760         /* for lock function check */
1761         symbol_conf.sort_by_name = true;
1762         symbol_conf.allow_aliases = true;
1763         symbol__init(&session->header.env);
1764
1765         if (!data.is_pipe) {
1766                 if (!perf_session__has_traces(session, "lock record"))
1767                         goto out_delete;
1768
1769                 if (perf_session__set_tracepoints_handlers(session, lock_tracepoints)) {
1770                         pr_err("Initializing perf session tracepoint handlers failed\n");
1771                         goto out_delete;
1772                 }
1773
1774                 if (perf_session__set_tracepoints_handlers(session, contention_tracepoints)) {
1775                         pr_err("Initializing perf session tracepoint handlers failed\n");
1776                         goto out_delete;
1777                 }
1778         }
1779
1780         if (setup_output_field(false, output_fields))
1781                 goto out_delete;
1782
1783         if (select_key(false))
1784                 goto out_delete;
1785
1786         if (show_thread_stats)
1787                 aggr_mode = LOCK_AGGR_TASK;
1788
1789         err = perf_session__process_events(session);
1790         if (err)
1791                 goto out_delete;
1792
1793         setup_pager();
1794         if (display_info) /* used for info subcommand */
1795                 err = dump_info();
1796         else {
1797                 combine_result();
1798                 sort_result();
1799                 print_result();
1800         }
1801
1802 out_delete:
1803         perf_session__delete(session);
1804         return err;
1805 }
1806
1807 static void sighandler(int sig __maybe_unused)
1808 {
1809 }
1810
1811 static int check_lock_contention_options(const struct option *options,
1812                                          const char * const *usage)
1813
1814 {
1815         if (show_thread_stats && show_lock_addrs) {
1816                 pr_err("Cannot use thread and addr mode together\n");
1817                 parse_options_usage(usage, options, "threads", 0);
1818                 parse_options_usage(NULL, options, "lock-addr", 0);
1819                 return -1;
1820         }
1821
1822         if (show_lock_owner && !use_bpf) {
1823                 pr_err("Lock owners are available only with BPF\n");
1824                 parse_options_usage(usage, options, "lock-owner", 0);
1825                 parse_options_usage(NULL, options, "use-bpf", 0);
1826                 return -1;
1827         }
1828
1829         if (show_lock_owner && show_lock_addrs) {
1830                 pr_err("Cannot use owner and addr mode together\n");
1831                 parse_options_usage(usage, options, "lock-owner", 0);
1832                 parse_options_usage(NULL, options, "lock-addr", 0);
1833                 return -1;
1834         }
1835
1836         if (show_lock_owner)
1837                 show_thread_stats = true;
1838
1839         return 0;
1840 }
1841
1842 static int __cmd_contention(int argc, const char **argv)
1843 {
1844         int err = -EINVAL;
1845         struct perf_tool eops = {
1846                 .attr            = perf_event__process_attr,
1847                 .event_update    = process_event_update,
1848                 .sample          = process_sample_event,
1849                 .comm            = perf_event__process_comm,
1850                 .mmap            = perf_event__process_mmap,
1851                 .tracing_data    = perf_event__process_tracing_data,
1852                 .ordered_events  = true,
1853         };
1854         struct perf_data data = {
1855                 .path  = input_name,
1856                 .mode  = PERF_DATA_MODE_READ,
1857                 .force = force,
1858         };
1859         struct lock_contention con = {
1860                 .target = &target,
1861                 .result = &lockhash_table[0],
1862                 .map_nr_entries = bpf_map_entries,
1863                 .max_stack = max_stack_depth,
1864                 .stack_skip = stack_skip,
1865                 .filters = &filters,
1866                 .save_callstack = needs_callstack(),
1867                 .owner = show_lock_owner,
1868         };
1869
1870         session = perf_session__new(use_bpf ? NULL : &data, &eops);
1871         if (IS_ERR(session)) {
1872                 pr_err("Initializing perf session failed\n");
1873                 return PTR_ERR(session);
1874         }
1875
1876         con.machine = &session->machines.host;
1877
1878         con.aggr_mode = aggr_mode = show_thread_stats ? LOCK_AGGR_TASK :
1879                 show_lock_addrs ? LOCK_AGGR_ADDR : LOCK_AGGR_CALLER;
1880
1881         if (con.aggr_mode == LOCK_AGGR_CALLER)
1882                 con.save_callstack = true;
1883
1884         /* for lock function check */
1885         symbol_conf.sort_by_name = true;
1886         symbol_conf.allow_aliases = true;
1887         symbol__init(&session->header.env);
1888
1889         if (use_bpf) {
1890                 err = target__validate(&target);
1891                 if (err) {
1892                         char errbuf[512];
1893
1894                         target__strerror(&target, err, errbuf, 512);
1895                         pr_err("%s\n", errbuf);
1896                         goto out_delete;
1897                 }
1898
1899                 signal(SIGINT, sighandler);
1900                 signal(SIGCHLD, sighandler);
1901                 signal(SIGTERM, sighandler);
1902
1903                 con.evlist = evlist__new();
1904                 if (con.evlist == NULL) {
1905                         err = -ENOMEM;
1906                         goto out_delete;
1907                 }
1908
1909                 err = evlist__create_maps(con.evlist, &target);
1910                 if (err < 0)
1911                         goto out_delete;
1912
1913                 if (argc) {
1914                         err = evlist__prepare_workload(con.evlist, &target,
1915                                                        argv, false, NULL);
1916                         if (err < 0)
1917                                 goto out_delete;
1918                 }
1919
1920                 if (lock_contention_prepare(&con) < 0) {
1921                         pr_err("lock contention BPF setup failed\n");
1922                         goto out_delete;
1923                 }
1924         } else if (!data.is_pipe) {
1925                 if (!perf_session__has_traces(session, "lock record"))
1926                         goto out_delete;
1927
1928                 if (!evlist__find_evsel_by_str(session->evlist,
1929                                                "lock:contention_begin")) {
1930                         pr_err("lock contention evsel not found\n");
1931                         goto out_delete;
1932                 }
1933
1934                 if (perf_session__set_tracepoints_handlers(session,
1935                                                 contention_tracepoints)) {
1936                         pr_err("Initializing perf session tracepoint handlers failed\n");
1937                         goto out_delete;
1938                 }
1939         }
1940
1941         if (setup_output_field(true, output_fields))
1942                 goto out_delete;
1943
1944         if (select_key(true))
1945                 goto out_delete;
1946
1947         if (use_bpf) {
1948                 lock_contention_start();
1949                 if (argc)
1950                         evlist__start_workload(con.evlist);
1951
1952                 /* wait for signal */
1953                 pause();
1954
1955                 lock_contention_stop();
1956                 lock_contention_read(&con);
1957         } else {
1958                 err = perf_session__process_events(session);
1959                 if (err)
1960                         goto out_delete;
1961         }
1962
1963         setup_pager();
1964
1965         sort_contention_result();
1966         print_contention_result(&con);
1967
1968 out_delete:
1969         lock_filter_finish();
1970         evlist__delete(con.evlist);
1971         lock_contention_finish();
1972         perf_session__delete(session);
1973         return err;
1974 }
1975
1976
1977 static int __cmd_record(int argc, const char **argv)
1978 {
1979         const char *record_args[] = {
1980                 "record", "-R", "-m", "1024", "-c", "1", "--synth", "task",
1981         };
1982         const char *callgraph_args[] = {
1983                 "--call-graph", "fp," __stringify(CONTENTION_STACK_DEPTH),
1984         };
1985         unsigned int rec_argc, i, j, ret;
1986         unsigned int nr_tracepoints;
1987         unsigned int nr_callgraph_args = 0;
1988         const char **rec_argv;
1989         bool has_lock_stat = true;
1990
1991         for (i = 0; i < ARRAY_SIZE(lock_tracepoints); i++) {
1992                 if (!is_valid_tracepoint(lock_tracepoints[i].name)) {
1993                         pr_debug("tracepoint %s is not enabled. "
1994                                  "Are CONFIG_LOCKDEP and CONFIG_LOCK_STAT enabled?\n",
1995                                  lock_tracepoints[i].name);
1996                         has_lock_stat = false;
1997                         break;
1998                 }
1999         }
2000
2001         if (has_lock_stat)
2002                 goto setup_args;
2003
2004         for (i = 0; i < ARRAY_SIZE(contention_tracepoints); i++) {
2005                 if (!is_valid_tracepoint(contention_tracepoints[i].name)) {
2006                         pr_err("tracepoint %s is not enabled.\n",
2007                                contention_tracepoints[i].name);
2008                         return 1;
2009                 }
2010         }
2011
2012         nr_callgraph_args = ARRAY_SIZE(callgraph_args);
2013
2014 setup_args:
2015         rec_argc = ARRAY_SIZE(record_args) + nr_callgraph_args + argc - 1;
2016
2017         if (has_lock_stat)
2018                 nr_tracepoints = ARRAY_SIZE(lock_tracepoints);
2019         else
2020                 nr_tracepoints = ARRAY_SIZE(contention_tracepoints);
2021
2022         /* factor of 2 is for -e in front of each tracepoint */
2023         rec_argc += 2 * nr_tracepoints;
2024
2025         rec_argv = calloc(rec_argc + 1, sizeof(char *));
2026         if (!rec_argv)
2027                 return -ENOMEM;
2028
2029         for (i = 0; i < ARRAY_SIZE(record_args); i++)
2030                 rec_argv[i] = strdup(record_args[i]);
2031
2032         for (j = 0; j < nr_tracepoints; j++) {
2033                 const char *ev_name;
2034
2035                 if (has_lock_stat)
2036                         ev_name = strdup(lock_tracepoints[j].name);
2037                 else
2038                         ev_name = strdup(contention_tracepoints[j].name);
2039
2040                 if (!ev_name)
2041                         return -ENOMEM;
2042
2043                 rec_argv[i++] = "-e";
2044                 rec_argv[i++] = ev_name;
2045         }
2046
2047         for (j = 0; j < nr_callgraph_args; j++, i++)
2048                 rec_argv[i] = callgraph_args[j];
2049
2050         for (j = 1; j < (unsigned int)argc; j++, i++)
2051                 rec_argv[i] = argv[j];
2052
2053         BUG_ON(i != rec_argc);
2054
2055         ret = cmd_record(i, rec_argv);
2056         free(rec_argv);
2057         return ret;
2058 }
2059
2060 static int parse_map_entry(const struct option *opt, const char *str,
2061                             int unset __maybe_unused)
2062 {
2063         unsigned long *len = (unsigned long *)opt->value;
2064         unsigned long val;
2065         char *endptr;
2066
2067         errno = 0;
2068         val = strtoul(str, &endptr, 0);
2069         if (*endptr != '\0' || errno != 0) {
2070                 pr_err("invalid BPF map length: %s\n", str);
2071                 return -1;
2072         }
2073
2074         *len = val;
2075         return 0;
2076 }
2077
2078 static int parse_max_stack(const struct option *opt, const char *str,
2079                            int unset __maybe_unused)
2080 {
2081         unsigned long *len = (unsigned long *)opt->value;
2082         long val;
2083         char *endptr;
2084
2085         errno = 0;
2086         val = strtol(str, &endptr, 0);
2087         if (*endptr != '\0' || errno != 0) {
2088                 pr_err("invalid max stack depth: %s\n", str);
2089                 return -1;
2090         }
2091
2092         if (val < 0 || val > sysctl__max_stack()) {
2093                 pr_err("invalid max stack depth: %ld\n", val);
2094                 return -1;
2095         }
2096
2097         *len = val;
2098         return 0;
2099 }
2100
2101 static bool add_lock_type(unsigned int flags)
2102 {
2103         unsigned int *tmp;
2104
2105         tmp = realloc(filters.types, (filters.nr_types + 1) * sizeof(*filters.types));
2106         if (tmp == NULL)
2107                 return false;
2108
2109         tmp[filters.nr_types++] = flags;
2110         filters.types = tmp;
2111         return true;
2112 }
2113
2114 static int parse_lock_type(const struct option *opt __maybe_unused, const char *str,
2115                            int unset __maybe_unused)
2116 {
2117         char *s, *tmp, *tok;
2118         int ret = 0;
2119
2120         s = strdup(str);
2121         if (s == NULL)
2122                 return -1;
2123
2124         for (tok = strtok_r(s, ", ", &tmp); tok; tok = strtok_r(NULL, ", ", &tmp)) {
2125                 unsigned int flags = get_type_flag(tok);
2126
2127                 if (flags == -1U) {
2128                         pr_err("Unknown lock flags: %s\n", tok);
2129                         ret = -1;
2130                         break;
2131                 }
2132
2133                 if (!add_lock_type(flags)) {
2134                         ret = -1;
2135                         break;
2136                 }
2137         }
2138
2139         free(s);
2140         return ret;
2141 }
2142
2143 static bool add_lock_addr(unsigned long addr)
2144 {
2145         unsigned long *tmp;
2146
2147         tmp = realloc(filters.addrs, (filters.nr_addrs + 1) * sizeof(*filters.addrs));
2148         if (tmp == NULL) {
2149                 pr_err("Memory allocation failure\n");
2150                 return false;
2151         }
2152
2153         tmp[filters.nr_addrs++] = addr;
2154         filters.addrs = tmp;
2155         return true;
2156 }
2157
2158 static bool add_lock_sym(char *name)
2159 {
2160         char **tmp;
2161         char *sym = strdup(name);
2162
2163         if (sym == NULL) {
2164                 pr_err("Memory allocation failure\n");
2165                 return false;
2166         }
2167
2168         tmp = realloc(filters.syms, (filters.nr_syms + 1) * sizeof(*filters.syms));
2169         if (tmp == NULL) {
2170                 pr_err("Memory allocation failure\n");
2171                 free(sym);
2172                 return false;
2173         }
2174
2175         tmp[filters.nr_syms++] = sym;
2176         filters.syms = tmp;
2177         return true;
2178 }
2179
2180 static int parse_lock_addr(const struct option *opt __maybe_unused, const char *str,
2181                            int unset __maybe_unused)
2182 {
2183         char *s, *tmp, *tok;
2184         int ret = 0;
2185         u64 addr;
2186
2187         s = strdup(str);
2188         if (s == NULL)
2189                 return -1;
2190
2191         for (tok = strtok_r(s, ", ", &tmp); tok; tok = strtok_r(NULL, ", ", &tmp)) {
2192                 char *end;
2193
2194                 addr = strtoul(tok, &end, 16);
2195                 if (*end == '\0') {
2196                         if (!add_lock_addr(addr)) {
2197                                 ret = -1;
2198                                 break;
2199                         }
2200                         continue;
2201                 }
2202
2203                 /*
2204                  * At this moment, we don't have kernel symbols.  Save the symbols
2205                  * in a separate list and resolve them to addresses later.
2206                  */
2207                 if (!add_lock_sym(tok)) {
2208                         ret = -1;
2209                         break;
2210                 }
2211         }
2212
2213         free(s);
2214         return ret;
2215 }
2216
2217 static int parse_call_stack(const struct option *opt __maybe_unused, const char *str,
2218                            int unset __maybe_unused)
2219 {
2220         char *s, *tmp, *tok;
2221         int ret = 0;
2222
2223         s = strdup(str);
2224         if (s == NULL)
2225                 return -1;
2226
2227         for (tok = strtok_r(s, ", ", &tmp); tok; tok = strtok_r(NULL, ", ", &tmp)) {
2228                 struct callstack_filter *entry;
2229
2230                 entry = malloc(sizeof(*entry) + strlen(tok) + 1);
2231                 if (entry == NULL) {
2232                         pr_err("Memory allocation failure\n");
2233                         return -1;
2234                 }
2235
2236                 strcpy(entry->name, tok);
2237                 list_add_tail(&entry->list, &callstack_filters);
2238         }
2239
2240         free(s);
2241         return ret;
2242 }
2243
2244 int cmd_lock(int argc, const char **argv)
2245 {
2246         const struct option lock_options[] = {
2247         OPT_STRING('i', "input", &input_name, "file", "input file name"),
2248         OPT_INCR('v', "verbose", &verbose, "be more verbose (show symbol address, etc)"),
2249         OPT_BOOLEAN('D', "dump-raw-trace", &dump_trace, "dump raw trace in ASCII"),
2250         OPT_BOOLEAN('f', "force", &force, "don't complain, do it"),
2251         OPT_STRING(0, "vmlinux", &symbol_conf.vmlinux_name,
2252                    "file", "vmlinux pathname"),
2253         OPT_STRING(0, "kallsyms", &symbol_conf.kallsyms_name,
2254                    "file", "kallsyms pathname"),
2255         OPT_BOOLEAN('q', "quiet", &quiet, "Do not show any warnings or messages"),
2256         OPT_END()
2257         };
2258
2259         const struct option info_options[] = {
2260         OPT_BOOLEAN('t', "threads", &info_threads,
2261                     "dump thread list in perf.data"),
2262         OPT_BOOLEAN('m', "map", &info_map,
2263                     "map of lock instances (address:name table)"),
2264         OPT_PARENT(lock_options)
2265         };
2266
2267         const struct option report_options[] = {
2268         OPT_STRING('k', "key", &sort_key, "acquired",
2269                     "key for sorting (acquired / contended / avg_wait / wait_total / wait_max / wait_min)"),
2270         OPT_STRING('F', "field", &output_fields, NULL,
2271                     "output fields (acquired / contended / avg_wait / wait_total / wait_max / wait_min)"),
2272         /* TODO: type */
2273         OPT_BOOLEAN('c', "combine-locks", &combine_locks,
2274                     "combine locks in the same class"),
2275         OPT_BOOLEAN('t', "threads", &show_thread_stats,
2276                     "show per-thread lock stats"),
2277         OPT_INTEGER('E', "entries", &print_nr_entries, "display this many functions"),
2278         OPT_PARENT(lock_options)
2279         };
2280
2281         struct option contention_options[] = {
2282         OPT_STRING('k', "key", &sort_key, "wait_total",
2283                     "key for sorting (contended / wait_total / wait_max / wait_min / avg_wait)"),
2284         OPT_STRING('F', "field", &output_fields, "contended,wait_total,wait_max,avg_wait",
2285                     "output fields (contended / wait_total / wait_max / wait_min / avg_wait)"),
2286         OPT_BOOLEAN('t', "threads", &show_thread_stats,
2287                     "show per-thread lock stats"),
2288         OPT_BOOLEAN('b', "use-bpf", &use_bpf, "use BPF program to collect lock contention stats"),
2289         OPT_BOOLEAN('a', "all-cpus", &target.system_wide,
2290                     "System-wide collection from all CPUs"),
2291         OPT_STRING('C', "cpu", &target.cpu_list, "cpu",
2292                     "List of cpus to monitor"),
2293         OPT_STRING('p', "pid", &target.pid, "pid",
2294                    "Trace on existing process id"),
2295         OPT_STRING(0, "tid", &target.tid, "tid",
2296                    "Trace on existing thread id (exclusive to --pid)"),
2297         OPT_CALLBACK('M', "map-nr-entries", &bpf_map_entries, "num",
2298                      "Max number of BPF map entries", parse_map_entry),
2299         OPT_CALLBACK(0, "max-stack", &max_stack_depth, "num",
2300                      "Set the maximum stack depth when collecting lopck contention, "
2301                      "Default: " __stringify(CONTENTION_STACK_DEPTH), parse_max_stack),
2302         OPT_INTEGER(0, "stack-skip", &stack_skip,
2303                     "Set the number of stack depth to skip when finding a lock caller, "
2304                     "Default: " __stringify(CONTENTION_STACK_SKIP)),
2305         OPT_INTEGER('E', "entries", &print_nr_entries, "display this many functions"),
2306         OPT_BOOLEAN('l', "lock-addr", &show_lock_addrs, "show lock stats by address"),
2307         OPT_CALLBACK('Y', "type-filter", NULL, "FLAGS",
2308                      "Filter specific type of locks", parse_lock_type),
2309         OPT_CALLBACK('L', "lock-filter", NULL, "ADDRS/NAMES",
2310                      "Filter specific address/symbol of locks", parse_lock_addr),
2311         OPT_CALLBACK('S', "callstack-filter", NULL, "NAMES",
2312                      "Filter specific function in the callstack", parse_call_stack),
2313         OPT_BOOLEAN('o', "lock-owner", &show_lock_owner, "show lock owners instead of waiters"),
2314         OPT_PARENT(lock_options)
2315         };
2316
2317         const char * const info_usage[] = {
2318                 "perf lock info [<options>]",
2319                 NULL
2320         };
2321         const char *const lock_subcommands[] = { "record", "report", "script",
2322                                                  "info", "contention", NULL };
2323         const char *lock_usage[] = {
2324                 NULL,
2325                 NULL
2326         };
2327         const char * const report_usage[] = {
2328                 "perf lock report [<options>]",
2329                 NULL
2330         };
2331         const char * const contention_usage[] = {
2332                 "perf lock contention [<options>]",
2333                 NULL
2334         };
2335         unsigned int i;
2336         int rc = 0;
2337
2338         for (i = 0; i < LOCKHASH_SIZE; i++)
2339                 INIT_HLIST_HEAD(lockhash_table + i);
2340
2341         argc = parse_options_subcommand(argc, argv, lock_options, lock_subcommands,
2342                                         lock_usage, PARSE_OPT_STOP_AT_NON_OPTION);
2343         if (!argc)
2344                 usage_with_options(lock_usage, lock_options);
2345
2346         if (strlen(argv[0]) > 2 && strstarts("record", argv[0])) {
2347                 return __cmd_record(argc, argv);
2348         } else if (strlen(argv[0]) > 2 && strstarts("report", argv[0])) {
2349                 trace_handler = &report_lock_ops;
2350                 if (argc) {
2351                         argc = parse_options(argc, argv,
2352                                              report_options, report_usage, 0);
2353                         if (argc)
2354                                 usage_with_options(report_usage, report_options);
2355                 }
2356                 rc = __cmd_report(false);
2357         } else if (!strcmp(argv[0], "script")) {
2358                 /* Aliased to 'perf script' */
2359                 return cmd_script(argc, argv);
2360         } else if (!strcmp(argv[0], "info")) {
2361                 if (argc) {
2362                         argc = parse_options(argc, argv,
2363                                              info_options, info_usage, 0);
2364                         if (argc)
2365                                 usage_with_options(info_usage, info_options);
2366                 }
2367                 /* recycling report_lock_ops */
2368                 trace_handler = &report_lock_ops;
2369                 rc = __cmd_report(true);
2370         } else if (strlen(argv[0]) > 2 && strstarts("contention", argv[0])) {
2371                 trace_handler = &contention_lock_ops;
2372                 sort_key = "wait_total";
2373                 output_fields = "contended,wait_total,wait_max,avg_wait";
2374
2375 #ifndef HAVE_BPF_SKEL
2376                 set_option_nobuild(contention_options, 'b', "use-bpf",
2377                                    "NO_BPF_SKEL=1", false);
2378 #endif
2379                 if (argc) {
2380                         argc = parse_options(argc, argv, contention_options,
2381                                              contention_usage, 0);
2382                 }
2383
2384                 if (check_lock_contention_options(contention_options,
2385                                                   contention_usage) < 0)
2386                         return -1;
2387
2388                 rc = __cmd_contention(argc, argv);
2389         } else {
2390                 usage_with_options(lock_usage, lock_options);
2391         }
2392
2393         return rc;
2394 }