linux/tools/perf/builtin-kmem.c
<<
>>
Prefs
   1#include "builtin.h"
   2#include "perf.h"
   3
   4#include "util/evlist.h"
   5#include "util/evsel.h"
   6#include "util/util.h"
   7#include "util/config.h"
   8#include "util/symbol.h"
   9#include "util/thread.h"
  10#include "util/header.h"
  11#include "util/session.h"
  12#include "util/tool.h"
  13#include "util/callchain.h"
  14
  15#include <subcmd/parse-options.h>
  16#include "util/trace-event.h"
  17#include "util/data.h"
  18#include "util/cpumap.h"
  19
  20#include "util/debug.h"
  21
  22#include <linux/rbtree.h>
  23#include <linux/string.h>
  24#include <locale.h>
  25#include <regex.h>
  26
  27static int      kmem_slab;
  28static int      kmem_page;
  29
  30static long     kmem_page_size;
  31static enum {
  32        KMEM_SLAB,
  33        KMEM_PAGE,
  34} kmem_default = KMEM_SLAB;  /* for backward compatibility */
  35
  36struct alloc_stat;
  37typedef int (*sort_fn_t)(void *, void *);
  38
  39static int                      alloc_flag;
  40static int                      caller_flag;
  41
  42static int                      alloc_lines = -1;
  43static int                      caller_lines = -1;
  44
  45static bool                     raw_ip;
  46
  47struct alloc_stat {
  48        u64     call_site;
  49        u64     ptr;
  50        u64     bytes_req;
  51        u64     bytes_alloc;
  52        u32     hit;
  53        u32     pingpong;
  54
  55        short   alloc_cpu;
  56
  57        struct rb_node node;
  58};
  59
  60static struct rb_root root_alloc_stat;
  61static struct rb_root root_alloc_sorted;
  62static struct rb_root root_caller_stat;
  63static struct rb_root root_caller_sorted;
  64
  65static unsigned long total_requested, total_allocated;
  66static unsigned long nr_allocs, nr_cross_allocs;
  67
  68static int insert_alloc_stat(unsigned long call_site, unsigned long ptr,
  69                             int bytes_req, int bytes_alloc, int cpu)
  70{
  71        struct rb_node **node = &root_alloc_stat.rb_node;
  72        struct rb_node *parent = NULL;
  73        struct alloc_stat *data = NULL;
  74
  75        while (*node) {
  76                parent = *node;
  77                data = rb_entry(*node, struct alloc_stat, node);
  78
  79                if (ptr > data->ptr)
  80                        node = &(*node)->rb_right;
  81                else if (ptr < data->ptr)
  82                        node = &(*node)->rb_left;
  83                else
  84                        break;
  85        }
  86
  87        if (data && data->ptr == ptr) {
  88                data->hit++;
  89                data->bytes_req += bytes_req;
  90                data->bytes_alloc += bytes_alloc;
  91        } else {
  92                data = malloc(sizeof(*data));
  93                if (!data) {
  94                        pr_err("%s: malloc failed\n", __func__);
  95                        return -1;
  96                }
  97                data->ptr = ptr;
  98                data->pingpong = 0;
  99                data->hit = 1;
 100                data->bytes_req = bytes_req;
 101                data->bytes_alloc = bytes_alloc;
 102
 103                rb_link_node(&data->node, parent, node);
 104                rb_insert_color(&data->node, &root_alloc_stat);
 105        }
 106        data->call_site = call_site;
 107        data->alloc_cpu = cpu;
 108        return 0;
 109}
 110
 111static int insert_caller_stat(unsigned long call_site,
 112                              int bytes_req, int bytes_alloc)
 113{
 114        struct rb_node **node = &root_caller_stat.rb_node;
 115        struct rb_node *parent = NULL;
 116        struct alloc_stat *data = NULL;
 117
 118        while (*node) {
 119                parent = *node;
 120                data = rb_entry(*node, struct alloc_stat, node);
 121
 122                if (call_site > data->call_site)
 123                        node = &(*node)->rb_right;
 124                else if (call_site < data->call_site)
 125                        node = &(*node)->rb_left;
 126                else
 127                        break;
 128        }
 129
 130        if (data && data->call_site == call_site) {
 131                data->hit++;
 132                data->bytes_req += bytes_req;
 133                data->bytes_alloc += bytes_alloc;
 134        } else {
 135                data = malloc(sizeof(*data));
 136                if (!data) {
 137                        pr_err("%s: malloc failed\n", __func__);
 138                        return -1;
 139                }
 140                data->call_site = call_site;
 141                data->pingpong = 0;
 142                data->hit = 1;
 143                data->bytes_req = bytes_req;
 144                data->bytes_alloc = bytes_alloc;
 145
 146                rb_link_node(&data->node, parent, node);
 147                rb_insert_color(&data->node, &root_caller_stat);
 148        }
 149
 150        return 0;
 151}
 152
 153static int perf_evsel__process_alloc_event(struct perf_evsel *evsel,
 154                                           struct perf_sample *sample)
 155{
 156        unsigned long ptr = perf_evsel__intval(evsel, sample, "ptr"),
 157                      call_site = perf_evsel__intval(evsel, sample, "call_site");
 158        int bytes_req = perf_evsel__intval(evsel, sample, "bytes_req"),
 159            bytes_alloc = perf_evsel__intval(evsel, sample, "bytes_alloc");
 160
 161        if (insert_alloc_stat(call_site, ptr, bytes_req, bytes_alloc, sample->cpu) ||
 162            insert_caller_stat(call_site, bytes_req, bytes_alloc))
 163                return -1;
 164
 165        total_requested += bytes_req;
 166        total_allocated += bytes_alloc;
 167
 168        nr_allocs++;
 169        return 0;
 170}
 171
 172static int perf_evsel__process_alloc_node_event(struct perf_evsel *evsel,
 173                                                struct perf_sample *sample)
 174{
 175        int ret = perf_evsel__process_alloc_event(evsel, sample);
 176
 177        if (!ret) {
 178                int node1 = cpu__get_node(sample->cpu),
 179                    node2 = perf_evsel__intval(evsel, sample, "node");
 180
 181                if (node1 != node2)
 182                        nr_cross_allocs++;
 183        }
 184
 185        return ret;
 186}
 187
 188static int ptr_cmp(void *, void *);
 189static int slab_callsite_cmp(void *, void *);
 190
 191static struct alloc_stat *search_alloc_stat(unsigned long ptr,
 192                                            unsigned long call_site,
 193                                            struct rb_root *root,
 194                                            sort_fn_t sort_fn)
 195{
 196        struct rb_node *node = root->rb_node;
 197        struct alloc_stat key = { .ptr = ptr, .call_site = call_site };
 198
 199        while (node) {
 200                struct alloc_stat *data;
 201                int cmp;
 202
 203                data = rb_entry(node, struct alloc_stat, node);
 204
 205                cmp = sort_fn(&key, data);
 206                if (cmp < 0)
 207                        node = node->rb_left;
 208                else if (cmp > 0)
 209                        node = node->rb_right;
 210                else
 211                        return data;
 212        }
 213        return NULL;
 214}
 215
 216static int perf_evsel__process_free_event(struct perf_evsel *evsel,
 217                                          struct perf_sample *sample)
 218{
 219        unsigned long ptr = perf_evsel__intval(evsel, sample, "ptr");
 220        struct alloc_stat *s_alloc, *s_caller;
 221
 222        s_alloc = search_alloc_stat(ptr, 0, &root_alloc_stat, ptr_cmp);
 223        if (!s_alloc)
 224                return 0;
 225
 226        if ((short)sample->cpu != s_alloc->alloc_cpu) {
 227                s_alloc->pingpong++;
 228
 229                s_caller = search_alloc_stat(0, s_alloc->call_site,
 230                                             &root_caller_stat,
 231                                             slab_callsite_cmp);
 232                if (!s_caller)
 233                        return -1;
 234                s_caller->pingpong++;
 235        }
 236        s_alloc->alloc_cpu = -1;
 237
 238        return 0;
 239}
 240
 241static u64 total_page_alloc_bytes;
 242static u64 total_page_free_bytes;
 243static u64 total_page_nomatch_bytes;
 244static u64 total_page_fail_bytes;
 245static unsigned long nr_page_allocs;
 246static unsigned long nr_page_frees;
 247static unsigned long nr_page_fails;
 248static unsigned long nr_page_nomatch;
 249
 250static bool use_pfn;
 251static bool live_page;
 252static struct perf_session *kmem_session;
 253
 254#define MAX_MIGRATE_TYPES  6
 255#define MAX_PAGE_ORDER     11
 256
 257static int order_stats[MAX_PAGE_ORDER][MAX_MIGRATE_TYPES];
 258
 259struct page_stat {
 260        struct rb_node  node;
 261        u64             page;
 262        u64             callsite;
 263        int             order;
 264        unsigned        gfp_flags;
 265        unsigned        migrate_type;
 266        u64             alloc_bytes;
 267        u64             free_bytes;
 268        int             nr_alloc;
 269        int             nr_free;
 270};
 271
 272static struct rb_root page_live_tree;
 273static struct rb_root page_alloc_tree;
 274static struct rb_root page_alloc_sorted;
 275static struct rb_root page_caller_tree;
 276static struct rb_root page_caller_sorted;
 277
 278struct alloc_func {
 279        u64 start;
 280        u64 end;
 281        char *name;
 282};
 283
 284static int nr_alloc_funcs;
 285static struct alloc_func *alloc_func_list;
 286
 287static int funcmp(const void *a, const void *b)
 288{
 289        const struct alloc_func *fa = a;
 290        const struct alloc_func *fb = b;
 291
 292        if (fa->start > fb->start)
 293                return 1;
 294        else
 295                return -1;
 296}
 297
 298static int callcmp(const void *a, const void *b)
 299{
 300        const struct alloc_func *fa = a;
 301        const struct alloc_func *fb = b;
 302
 303        if (fb->start <= fa->start && fa->end < fb->end)
 304                return 0;
 305
 306        if (fa->start > fb->start)
 307                return 1;
 308        else
 309                return -1;
 310}
 311
 312static int build_alloc_func_list(void)
 313{
 314        int ret;
 315        struct map *kernel_map;
 316        struct symbol *sym;
 317        struct rb_node *node;
 318        struct alloc_func *func;
 319        struct machine *machine = &kmem_session->machines.host;
 320        regex_t alloc_func_regex;
 321        const char pattern[] = "^_?_?(alloc|get_free|get_zeroed)_pages?";
 322
 323        ret = regcomp(&alloc_func_regex, pattern, REG_EXTENDED);
 324        if (ret) {
 325                char err[BUFSIZ];
 326
 327                regerror(ret, &alloc_func_regex, err, sizeof(err));
 328                pr_err("Invalid regex: %s\n%s", pattern, err);
 329                return -EINVAL;
 330        }
 331
 332        kernel_map = machine__kernel_map(machine);
 333        if (map__load(kernel_map, NULL) < 0) {
 334                pr_err("cannot load kernel map\n");
 335                return -ENOENT;
 336        }
 337
 338        map__for_each_symbol(kernel_map, sym, node) {
 339                if (regexec(&alloc_func_regex, sym->name, 0, NULL, 0))
 340                        continue;
 341
 342                func = realloc(alloc_func_list,
 343                               (nr_alloc_funcs + 1) * sizeof(*func));
 344                if (func == NULL)
 345                        return -ENOMEM;
 346
 347                pr_debug("alloc func: %s\n", sym->name);
 348                func[nr_alloc_funcs].start = sym->start;
 349                func[nr_alloc_funcs].end   = sym->end;
 350                func[nr_alloc_funcs].name  = sym->name;
 351
 352                alloc_func_list = func;
 353                nr_alloc_funcs++;
 354        }
 355
 356        qsort(alloc_func_list, nr_alloc_funcs, sizeof(*func), funcmp);
 357
 358        regfree(&alloc_func_regex);
 359        return 0;
 360}
 361
 362/*
 363 * Find first non-memory allocation function from callchain.
 364 * The allocation functions are in the 'alloc_func_list'.
 365 */
 366static u64 find_callsite(struct perf_evsel *evsel, struct perf_sample *sample)
 367{
 368        struct addr_location al;
 369        struct machine *machine = &kmem_session->machines.host;
 370        struct callchain_cursor_node *node;
 371
 372        if (alloc_func_list == NULL) {
 373                if (build_alloc_func_list() < 0)
 374                        goto out;
 375        }
 376
 377        al.thread = machine__findnew_thread(machine, sample->pid, sample->tid);
 378        sample__resolve_callchain(sample, &callchain_cursor, NULL, evsel, &al, 16);
 379
 380        callchain_cursor_commit(&callchain_cursor);
 381        while (true) {
 382                struct alloc_func key, *caller;
 383                u64 addr;
 384
 385                node = callchain_cursor_current(&callchain_cursor);
 386                if (node == NULL)
 387                        break;
 388
 389                key.start = key.end = node->ip;
 390                caller = bsearch(&key, alloc_func_list, nr_alloc_funcs,
 391                                 sizeof(key), callcmp);
 392                if (!caller) {
 393                        /* found */
 394                        if (node->map)
 395                                addr = map__unmap_ip(node->map, node->ip);
 396                        else
 397                                addr = node->ip;
 398
 399                        return addr;
 400                } else
 401                        pr_debug3("skipping alloc function: %s\n", caller->name);
 402
 403                callchain_cursor_advance(&callchain_cursor);
 404        }
 405
 406out:
 407        pr_debug2("unknown callsite: %"PRIx64 "\n", sample->ip);
 408        return sample->ip;
 409}
 410
 411struct sort_dimension {
 412        const char              name[20];
 413        sort_fn_t               cmp;
 414        struct list_head        list;
 415};
 416
 417static LIST_HEAD(page_alloc_sort_input);
 418static LIST_HEAD(page_caller_sort_input);
 419
 420static struct page_stat *
 421__page_stat__findnew_page(struct page_stat *pstat, bool create)
 422{
 423        struct rb_node **node = &page_live_tree.rb_node;
 424        struct rb_node *parent = NULL;
 425        struct page_stat *data;
 426
 427        while (*node) {
 428                s64 cmp;
 429
 430                parent = *node;
 431                data = rb_entry(*node, struct page_stat, node);
 432
 433                cmp = data->page - pstat->page;
 434                if (cmp < 0)
 435                        node = &parent->rb_left;
 436                else if (cmp > 0)
 437                        node = &parent->rb_right;
 438                else
 439                        return data;
 440        }
 441
 442        if (!create)
 443                return NULL;
 444
 445        data = zalloc(sizeof(*data));
 446        if (data != NULL) {
 447                data->page = pstat->page;
 448                data->order = pstat->order;
 449                data->gfp_flags = pstat->gfp_flags;
 450                data->migrate_type = pstat->migrate_type;
 451
 452                rb_link_node(&data->node, parent, node);
 453                rb_insert_color(&data->node, &page_live_tree);
 454        }
 455
 456        return data;
 457}
 458
 459static struct page_stat *page_stat__find_page(struct page_stat *pstat)
 460{
 461        return __page_stat__findnew_page(pstat, false);
 462}
 463
 464static struct page_stat *page_stat__findnew_page(struct page_stat *pstat)
 465{
 466        return __page_stat__findnew_page(pstat, true);
 467}
 468
 469static struct page_stat *
 470__page_stat__findnew_alloc(struct page_stat *pstat, bool create)
 471{
 472        struct rb_node **node = &page_alloc_tree.rb_node;
 473        struct rb_node *parent = NULL;
 474        struct page_stat *data;
 475        struct sort_dimension *sort;
 476
 477        while (*node) {
 478                int cmp = 0;
 479
 480                parent = *node;
 481                data = rb_entry(*node, struct page_stat, node);
 482
 483                list_for_each_entry(sort, &page_alloc_sort_input, list) {
 484                        cmp = sort->cmp(pstat, data);
 485                        if (cmp)
 486                                break;
 487                }
 488
 489                if (cmp < 0)
 490                        node = &parent->rb_left;
 491                else if (cmp > 0)
 492                        node = &parent->rb_right;
 493                else
 494                        return data;
 495        }
 496
 497        if (!create)
 498                return NULL;
 499
 500        data = zalloc(sizeof(*data));
 501        if (data != NULL) {
 502                data->page = pstat->page;
 503                data->order = pstat->order;
 504                data->gfp_flags = pstat->gfp_flags;
 505                data->migrate_type = pstat->migrate_type;
 506
 507                rb_link_node(&data->node, parent, node);
 508                rb_insert_color(&data->node, &page_alloc_tree);
 509        }
 510
 511        return data;
 512}
 513
 514static struct page_stat *page_stat__find_alloc(struct page_stat *pstat)
 515{
 516        return __page_stat__findnew_alloc(pstat, false);
 517}
 518
 519static struct page_stat *page_stat__findnew_alloc(struct page_stat *pstat)
 520{
 521        return __page_stat__findnew_alloc(pstat, true);
 522}
 523
 524static struct page_stat *
 525__page_stat__findnew_caller(struct page_stat *pstat, bool create)
 526{
 527        struct rb_node **node = &page_caller_tree.rb_node;
 528        struct rb_node *parent = NULL;
 529        struct page_stat *data;
 530        struct sort_dimension *sort;
 531
 532        while (*node) {
 533                int cmp = 0;
 534
 535                parent = *node;
 536                data = rb_entry(*node, struct page_stat, node);
 537
 538                list_for_each_entry(sort, &page_caller_sort_input, list) {
 539                        cmp = sort->cmp(pstat, data);
 540                        if (cmp)
 541                                break;
 542                }
 543
 544                if (cmp < 0)
 545                        node = &parent->rb_left;
 546                else if (cmp > 0)
 547                        node = &parent->rb_right;
 548                else
 549                        return data;
 550        }
 551
 552        if (!create)
 553                return NULL;
 554
 555        data = zalloc(sizeof(*data));
 556        if (data != NULL) {
 557                data->callsite = pstat->callsite;
 558                data->order = pstat->order;
 559                data->gfp_flags = pstat->gfp_flags;
 560                data->migrate_type = pstat->migrate_type;
 561
 562                rb_link_node(&data->node, parent, node);
 563                rb_insert_color(&data->node, &page_caller_tree);
 564        }
 565
 566        return data;
 567}
 568
 569static struct page_stat *page_stat__find_caller(struct page_stat *pstat)
 570{
 571        return __page_stat__findnew_caller(pstat, false);
 572}
 573
 574static struct page_stat *page_stat__findnew_caller(struct page_stat *pstat)
 575{
 576        return __page_stat__findnew_caller(pstat, true);
 577}
 578
 579static bool valid_page(u64 pfn_or_page)
 580{
 581        if (use_pfn && pfn_or_page == -1UL)
 582                return false;
 583        if (!use_pfn && pfn_or_page == 0)
 584                return false;
 585        return true;
 586}
 587
 588struct gfp_flag {
 589        unsigned int flags;
 590        char *compact_str;
 591        char *human_readable;
 592};
 593
 594static struct gfp_flag *gfps;
 595static int nr_gfps;
 596
 597static int gfpcmp(const void *a, const void *b)
 598{
 599        const struct gfp_flag *fa = a;
 600        const struct gfp_flag *fb = b;
 601
 602        return fa->flags - fb->flags;
 603}
 604
 605/* see include/trace/events/mmflags.h */
 606static const struct {
 607        const char *original;
 608        const char *compact;
 609} gfp_compact_table[] = {
 610        { "GFP_TRANSHUGE",              "THP" },
 611        { "GFP_TRANSHUGE_LIGHT",        "THL" },
 612        { "GFP_HIGHUSER_MOVABLE",       "HUM" },
 613        { "GFP_HIGHUSER",               "HU" },
 614        { "GFP_USER",                   "U" },
 615        { "GFP_TEMPORARY",              "TMP" },
 616        { "GFP_KERNEL_ACCOUNT",         "KAC" },
 617        { "GFP_KERNEL",                 "K" },
 618        { "GFP_NOFS",                   "NF" },
 619        { "GFP_ATOMIC",                 "A" },
 620        { "GFP_NOIO",                   "NI" },
 621        { "GFP_NOWAIT",                 "NW" },
 622        { "GFP_DMA",                    "D" },
 623        { "__GFP_HIGHMEM",              "HM" },
 624        { "GFP_DMA32",                  "D32" },
 625        { "__GFP_HIGH",                 "H" },
 626        { "__GFP_ATOMIC",               "_A" },
 627        { "__GFP_IO",                   "I" },
 628        { "__GFP_FS",                   "F" },
 629        { "__GFP_COLD",                 "CO" },
 630        { "__GFP_NOWARN",               "NWR" },
 631        { "__GFP_REPEAT",               "R" },
 632        { "__GFP_NOFAIL",               "NF" },
 633        { "__GFP_NORETRY",              "NR" },
 634        { "__GFP_COMP",                 "C" },
 635        { "__GFP_ZERO",                 "Z" },
 636        { "__GFP_NOMEMALLOC",           "NMA" },
 637        { "__GFP_MEMALLOC",             "MA" },
 638        { "__GFP_HARDWALL",             "HW" },
 639        { "__GFP_THISNODE",             "TN" },
 640        { "__GFP_RECLAIMABLE",          "RC" },
 641        { "__GFP_MOVABLE",              "M" },
 642        { "__GFP_ACCOUNT",              "AC" },
 643        { "__GFP_NOTRACK",              "NT" },
 644        { "__GFP_WRITE",                "WR" },
 645        { "__GFP_RECLAIM",              "R" },
 646        { "__GFP_DIRECT_RECLAIM",       "DR" },
 647        { "__GFP_KSWAPD_RECLAIM",       "KR" },
 648        { "__GFP_OTHER_NODE",           "ON" },
 649};
 650
 651static size_t max_gfp_len;
 652
 653static char *compact_gfp_flags(char *gfp_flags)
 654{
 655        char *orig_flags = strdup(gfp_flags);
 656        char *new_flags = NULL;
 657        char *str, *pos = NULL;
 658        size_t len = 0;
 659
 660        if (orig_flags == NULL)
 661                return NULL;
 662
 663        str = strtok_r(orig_flags, "|", &pos);
 664        while (str) {
 665                size_t i;
 666                char *new;
 667                const char *cpt;
 668
 669                for (i = 0; i < ARRAY_SIZE(gfp_compact_table); i++) {
 670                        if (strcmp(gfp_compact_table[i].original, str))
 671                                continue;
 672
 673                        cpt = gfp_compact_table[i].compact;
 674                        new = realloc(new_flags, len + strlen(cpt) + 2);
 675                        if (new == NULL) {
 676                                free(new_flags);
 677                                return NULL;
 678                        }
 679
 680                        new_flags = new;
 681
 682                        if (!len) {
 683                                strcpy(new_flags, cpt);
 684                        } else {
 685                                strcat(new_flags, "|");
 686                                strcat(new_flags, cpt);
 687                                len++;
 688                        }
 689
 690                        len += strlen(cpt);
 691                }
 692
 693                str = strtok_r(NULL, "|", &pos);
 694        }
 695
 696        if (max_gfp_len < len)
 697                max_gfp_len = len;
 698
 699        free(orig_flags);
 700        return new_flags;
 701}
 702
 703static char *compact_gfp_string(unsigned long gfp_flags)
 704{
 705        struct gfp_flag key = {
 706                .flags = gfp_flags,
 707        };
 708        struct gfp_flag *gfp;
 709
 710        gfp = bsearch(&key, gfps, nr_gfps, sizeof(*gfps), gfpcmp);
 711        if (gfp)
 712                return gfp->compact_str;
 713
 714        return NULL;
 715}
 716
 717static int parse_gfp_flags(struct perf_evsel *evsel, struct perf_sample *sample,
 718                           unsigned int gfp_flags)
 719{
 720        struct pevent_record record = {
 721                .cpu = sample->cpu,
 722                .data = sample->raw_data,
 723                .size = sample->raw_size,
 724        };
 725        struct trace_seq seq;
 726        char *str, *pos = NULL;
 727
 728        if (nr_gfps) {
 729                struct gfp_flag key = {
 730                        .flags = gfp_flags,
 731                };
 732
 733                if (bsearch(&key, gfps, nr_gfps, sizeof(*gfps), gfpcmp))
 734                        return 0;
 735        }
 736
 737        trace_seq_init(&seq);
 738        pevent_event_info(&seq, evsel->tp_format, &record);
 739
 740        str = strtok_r(seq.buffer, " ", &pos);
 741        while (str) {
 742                if (!strncmp(str, "gfp_flags=", 10)) {
 743                        struct gfp_flag *new;
 744
 745                        new = realloc(gfps, (nr_gfps + 1) * sizeof(*gfps));
 746                        if (new == NULL)
 747                                return -ENOMEM;
 748
 749                        gfps = new;
 750                        new += nr_gfps++;
 751
 752                        new->flags = gfp_flags;
 753                        new->human_readable = strdup(str + 10);
 754                        new->compact_str = compact_gfp_flags(str + 10);
 755                        if (!new->human_readable || !new->compact_str)
 756                                return -ENOMEM;
 757
 758                        qsort(gfps, nr_gfps, sizeof(*gfps), gfpcmp);
 759                }
 760
 761                str = strtok_r(NULL, " ", &pos);
 762        }
 763
 764        trace_seq_destroy(&seq);
 765        return 0;
 766}
 767
 768static int perf_evsel__process_page_alloc_event(struct perf_evsel *evsel,
 769                                                struct perf_sample *sample)
 770{
 771        u64 page;
 772        unsigned int order = perf_evsel__intval(evsel, sample, "order");
 773        unsigned int gfp_flags = perf_evsel__intval(evsel, sample, "gfp_flags");
 774        unsigned int migrate_type = perf_evsel__intval(evsel, sample,
 775                                                       "migratetype");
 776        u64 bytes = kmem_page_size << order;
 777        u64 callsite;
 778        struct page_stat *pstat;
 779        struct page_stat this = {
 780                .order = order,
 781                .gfp_flags = gfp_flags,
 782                .migrate_type = migrate_type,
 783        };
 784
 785        if (use_pfn)
 786                page = perf_evsel__intval(evsel, sample, "pfn");
 787        else
 788                page = perf_evsel__intval(evsel, sample, "page");
 789
 790        nr_page_allocs++;
 791        total_page_alloc_bytes += bytes;
 792
 793        if (!valid_page(page)) {
 794                nr_page_fails++;
 795                total_page_fail_bytes += bytes;
 796
 797                return 0;
 798        }
 799
 800        if (parse_gfp_flags(evsel, sample, gfp_flags) < 0)
 801                return -1;
 802
 803        callsite = find_callsite(evsel, sample);
 804
 805        /*
 806         * This is to find the current page (with correct gfp flags and
 807         * migrate type) at free event.
 808         */
 809        this.page = page;
 810        pstat = page_stat__findnew_page(&this);
 811        if (pstat == NULL)
 812                return -ENOMEM;
 813
 814        pstat->nr_alloc++;
 815        pstat->alloc_bytes += bytes;
 816        pstat->callsite = callsite;
 817
 818        if (!live_page) {
 819                pstat = page_stat__findnew_alloc(&this);
 820                if (pstat == NULL)
 821                        return -ENOMEM;
 822
 823                pstat->nr_alloc++;
 824                pstat->alloc_bytes += bytes;
 825                pstat->callsite = callsite;
 826        }
 827
 828        this.callsite = callsite;
 829        pstat = page_stat__findnew_caller(&this);
 830        if (pstat == NULL)
 831                return -ENOMEM;
 832
 833        pstat->nr_alloc++;
 834        pstat->alloc_bytes += bytes;
 835
 836        order_stats[order][migrate_type]++;
 837
 838        return 0;
 839}
 840
 841static int perf_evsel__process_page_free_event(struct perf_evsel *evsel,
 842                                                struct perf_sample *sample)
 843{
 844        u64 page;
 845        unsigned int order = perf_evsel__intval(evsel, sample, "order");
 846        u64 bytes = kmem_page_size << order;
 847        struct page_stat *pstat;
 848        struct page_stat this = {
 849                .order = order,
 850        };
 851
 852        if (use_pfn)
 853                page = perf_evsel__intval(evsel, sample, "pfn");
 854        else
 855                page = perf_evsel__intval(evsel, sample, "page");
 856
 857        nr_page_frees++;
 858        total_page_free_bytes += bytes;
 859
 860        this.page = page;
 861        pstat = page_stat__find_page(&this);
 862        if (pstat == NULL) {
 863                pr_debug2("missing free at page %"PRIx64" (order: %d)\n",
 864                          page, order);
 865
 866                nr_page_nomatch++;
 867                total_page_nomatch_bytes += bytes;
 868
 869                return 0;
 870        }
 871
 872        this.gfp_flags = pstat->gfp_flags;
 873        this.migrate_type = pstat->migrate_type;
 874        this.callsite = pstat->callsite;
 875
 876        rb_erase(&pstat->node, &page_live_tree);
 877        free(pstat);
 878
 879        if (live_page) {
 880                order_stats[this.order][this.migrate_type]--;
 881        } else {
 882                pstat = page_stat__find_alloc(&this);
 883                if (pstat == NULL)
 884                        return -ENOMEM;
 885
 886                pstat->nr_free++;
 887                pstat->free_bytes += bytes;
 888        }
 889
 890        pstat = page_stat__find_caller(&this);
 891        if (pstat == NULL)
 892                return -ENOENT;
 893
 894        pstat->nr_free++;
 895        pstat->free_bytes += bytes;
 896
 897        if (live_page) {
 898                pstat->nr_alloc--;
 899                pstat->alloc_bytes -= bytes;
 900
 901                if (pstat->nr_alloc == 0) {
 902                        rb_erase(&pstat->node, &page_caller_tree);
 903                        free(pstat);
 904                }
 905        }
 906
 907        return 0;
 908}
 909
 910typedef int (*tracepoint_handler)(struct perf_evsel *evsel,
 911                                  struct perf_sample *sample);
 912
 913static int process_sample_event(struct perf_tool *tool __maybe_unused,
 914                                union perf_event *event,
 915                                struct perf_sample *sample,
 916                                struct perf_evsel *evsel,
 917                                struct machine *machine)
 918{
 919        int err = 0;
 920        struct thread *thread = machine__findnew_thread(machine, sample->pid,
 921                                                        sample->tid);
 922
 923        if (thread == NULL) {
 924                pr_debug("problem processing %d event, skipping it.\n",
 925                         event->header.type);
 926                return -1;
 927        }
 928
 929        dump_printf(" ... thread: %s:%d\n", thread__comm_str(thread), thread->tid);
 930
 931        if (evsel->handler != NULL) {
 932                tracepoint_handler f = evsel->handler;
 933                err = f(evsel, sample);
 934        }
 935
 936        thread__put(thread);
 937
 938        return err;
 939}
 940
 941static struct perf_tool perf_kmem = {
 942        .sample          = process_sample_event,
 943        .comm            = perf_event__process_comm,
 944        .mmap            = perf_event__process_mmap,
 945        .mmap2           = perf_event__process_mmap2,
 946        .ordered_events  = true,
 947};
 948
 949static double fragmentation(unsigned long n_req, unsigned long n_alloc)
 950{
 951        if (n_alloc == 0)
 952                return 0.0;
 953        else
 954                return 100.0 - (100.0 * n_req / n_alloc);
 955}
 956
 957static void __print_slab_result(struct rb_root *root,
 958                                struct perf_session *session,
 959                                int n_lines, int is_caller)
 960{
 961        struct rb_node *next;
 962        struct machine *machine = &session->machines.host;
 963
 964        printf("%.105s\n", graph_dotted_line);
 965        printf(" %-34s |",  is_caller ? "Callsite": "Alloc Ptr");
 966        printf(" Total_alloc/Per | Total_req/Per   | Hit      | Ping-pong | Frag\n");
 967        printf("%.105s\n", graph_dotted_line);
 968
 969        next = rb_first(root);
 970
 971        while (next && n_lines--) {
 972                struct alloc_stat *data = rb_entry(next, struct alloc_stat,
 973                                                   node);
 974                struct symbol *sym = NULL;
 975                struct map *map;
 976                char buf[BUFSIZ];
 977                u64 addr;
 978
 979                if (is_caller) {
 980                        addr = data->call_site;
 981                        if (!raw_ip)
 982                                sym = machine__find_kernel_function(machine, addr, &map, NULL);
 983                } else
 984                        addr = data->ptr;
 985
 986                if (sym != NULL)
 987                        snprintf(buf, sizeof(buf), "%s+%" PRIx64 "", sym->name,
 988                                 addr - map->unmap_ip(map, sym->start));
 989                else
 990                        snprintf(buf, sizeof(buf), "%#" PRIx64 "", addr);
 991                printf(" %-34s |", buf);
 992
 993                printf(" %9llu/%-5lu | %9llu/%-5lu | %8lu | %9lu | %6.3f%%\n",
 994                       (unsigned long long)data->bytes_alloc,
 995                       (unsigned long)data->bytes_alloc / data->hit,
 996                       (unsigned long long)data->bytes_req,
 997                       (unsigned long)data->bytes_req / data->hit,
 998                       (unsigned long)data->hit,
 999                       (unsigned long)data->pingpong,
1000                       fragmentation(data->bytes_req, data->bytes_alloc));
1001
1002                next = rb_next(next);
1003        }
1004
1005        if (n_lines == -1)
1006                printf(" ...                                | ...             | ...             | ...      | ...       | ...   \n");
1007
1008        printf("%.105s\n", graph_dotted_line);
1009}
1010
1011static const char * const migrate_type_str[] = {
1012        "UNMOVABL",
1013        "RECLAIM",
1014        "MOVABLE",
1015        "RESERVED",
1016        "CMA/ISLT",
1017        "UNKNOWN",
1018};
1019
1020static void __print_page_alloc_result(struct perf_session *session, int n_lines)
1021{
1022        struct rb_node *next = rb_first(&page_alloc_sorted);
1023        struct machine *machine = &session->machines.host;
1024        const char *format;
1025        int gfp_len = max(strlen("GFP flags"), max_gfp_len);
1026
1027        printf("\n%.105s\n", graph_dotted_line);
1028        printf(" %-16s | %5s alloc (KB) | Hits      | Order | Mig.type | %-*s | Callsite\n",
1029               use_pfn ? "PFN" : "Page", live_page ? "Live" : "Total",
1030               gfp_len, "GFP flags");
1031        printf("%.105s\n", graph_dotted_line);
1032
1033        if (use_pfn)
1034                format = " %16llu | %'16llu | %'9d | %5d | %8s | %-*s | %s\n";
1035        else
1036                format = " %016llx | %'16llu | %'9d | %5d | %8s | %-*s | %s\n";
1037
1038        while (next && n_lines--) {
1039                struct page_stat *data;
1040                struct symbol *sym;
1041                struct map *map;
1042                char buf[32];
1043                char *caller = buf;
1044
1045                data = rb_entry(next, struct page_stat, node);
1046                sym = machine__find_kernel_function(machine, data->callsite,
1047                                                    &map, NULL);
1048                if (sym && sym->name)
1049                        caller = sym->name;
1050                else
1051                        scnprintf(buf, sizeof(buf), "%"PRIx64, data->callsite);
1052
1053                printf(format, (unsigned long long)data->page,
1054                       (unsigned long long)data->alloc_bytes / 1024,
1055                       data->nr_alloc, data->order,
1056                       migrate_type_str[data->migrate_type],
1057                       gfp_len, compact_gfp_string(data->gfp_flags), caller);
1058
1059                next = rb_next(next);
1060        }
1061
1062        if (n_lines == -1) {
1063                printf(" ...              | ...              | ...       | ...   | ...      | %-*s | ...\n",
1064                       gfp_len, "...");
1065        }
1066
1067        printf("%.105s\n", graph_dotted_line);
1068}
1069
1070static void __print_page_caller_result(struct perf_session *session, int n_lines)
1071{
1072        struct rb_node *next = rb_first(&page_caller_sorted);
1073        struct machine *machine = &session->machines.host;
1074        int gfp_len = max(strlen("GFP flags"), max_gfp_len);
1075
1076        printf("\n%.105s\n", graph_dotted_line);
1077        printf(" %5s alloc (KB) | Hits      | Order | Mig.type | %-*s | Callsite\n",
1078               live_page ? "Live" : "Total", gfp_len, "GFP flags");
1079        printf("%.105s\n", graph_dotted_line);
1080
1081        while (next && n_lines--) {
1082                struct page_stat *data;
1083                struct symbol *sym;
1084                struct map *map;
1085                char buf[32];
1086                char *caller = buf;
1087
1088                data = rb_entry(next, struct page_stat, node);
1089                sym = machine__find_kernel_function(machine, data->callsite,
1090                                                    &map, NULL);
1091                if (sym && sym->name)
1092                        caller = sym->name;
1093                else
1094                        scnprintf(buf, sizeof(buf), "%"PRIx64, data->callsite);
1095
1096                printf(" %'16llu | %'9d | %5d | %8s | %-*s | %s\n",
1097                       (unsigned long long)data->alloc_bytes / 1024,
1098                       data->nr_alloc, data->order,
1099                       migrate_type_str[data->migrate_type],
1100                       gfp_len, compact_gfp_string(data->gfp_flags), caller);
1101
1102                next = rb_next(next);
1103        }
1104
1105        if (n_lines == -1) {
1106                printf(" ...              | ...       | ...   | ...      | %-*s | ...\n",
1107                       gfp_len, "...");
1108        }
1109
1110        printf("%.105s\n", graph_dotted_line);
1111}
1112
1113static void print_gfp_flags(void)
1114{
1115        int i;
1116
1117        printf("#\n");
1118        printf("# GFP flags\n");
1119        printf("# ---------\n");
1120        for (i = 0; i < nr_gfps; i++) {
1121                printf("# %08x: %*s: %s\n", gfps[i].flags,
1122                       (int) max_gfp_len, gfps[i].compact_str,
1123                       gfps[i].human_readable);
1124        }
1125}
1126
1127static void print_slab_summary(void)
1128{
1129        printf("\nSUMMARY (SLAB allocator)");
1130        printf("\n========================\n");
1131        printf("Total bytes requested: %'lu\n", total_requested);
1132        printf("Total bytes allocated: %'lu\n", total_allocated);
1133        printf("Total bytes wasted on internal fragmentation: %'lu\n",
1134               total_allocated - total_requested);
1135        printf("Internal fragmentation: %f%%\n",
1136               fragmentation(total_requested, total_allocated));
1137        printf("Cross CPU allocations: %'lu/%'lu\n", nr_cross_allocs, nr_allocs);
1138}
1139
1140static void print_page_summary(void)
1141{
1142        int o, m;
1143        u64 nr_alloc_freed = nr_page_frees - nr_page_nomatch;
1144        u64 total_alloc_freed_bytes = total_page_free_bytes - total_page_nomatch_bytes;
1145
1146        printf("\nSUMMARY (page allocator)");
1147        printf("\n========================\n");
1148        printf("%-30s: %'16lu   [ %'16"PRIu64" KB ]\n", "Total allocation requests",
1149               nr_page_allocs, total_page_alloc_bytes / 1024);
1150        printf("%-30s: %'16lu   [ %'16"PRIu64" KB ]\n", "Total free requests",
1151               nr_page_frees, total_page_free_bytes / 1024);
1152        printf("\n");
1153
1154        printf("%-30s: %'16"PRIu64"   [ %'16"PRIu64" KB ]\n", "Total alloc+freed requests",
1155               nr_alloc_freed, (total_alloc_freed_bytes) / 1024);
1156        printf("%-30s: %'16"PRIu64"   [ %'16"PRIu64" KB ]\n", "Total alloc-only requests",
1157               nr_page_allocs - nr_alloc_freed,
1158               (total_page_alloc_bytes - total_alloc_freed_bytes) / 1024);
1159        printf("%-30s: %'16lu   [ %'16"PRIu64" KB ]\n", "Total free-only requests",
1160               nr_page_nomatch, total_page_nomatch_bytes / 1024);
1161        printf("\n");
1162
1163        printf("%-30s: %'16lu   [ %'16"PRIu64" KB ]\n", "Total allocation failures",
1164               nr_page_fails, total_page_fail_bytes / 1024);
1165        printf("\n");
1166
1167        printf("%5s  %12s  %12s  %12s  %12s  %12s\n", "Order",  "Unmovable",
1168               "Reclaimable", "Movable", "Reserved", "CMA/Isolated");
1169        printf("%.5s  %.12s  %.12s  %.12s  %.12s  %.12s\n", graph_dotted_line,
1170               graph_dotted_line, graph_dotted_line, graph_dotted_line,
1171               graph_dotted_line, graph_dotted_line);
1172
1173        for (o = 0; o < MAX_PAGE_ORDER; o++) {
1174                printf("%5d", o);
1175                for (m = 0; m < MAX_MIGRATE_TYPES - 1; m++) {
1176                        if (order_stats[o][m])
1177                                printf("  %'12d", order_stats[o][m]);
1178                        else
1179                                printf("  %12c", '.');
1180                }
1181                printf("\n");
1182        }
1183}
1184
1185static void print_slab_result(struct perf_session *session)
1186{
1187        if (caller_flag)
1188                __print_slab_result(&root_caller_sorted, session, caller_lines, 1);
1189        if (alloc_flag)
1190                __print_slab_result(&root_alloc_sorted, session, alloc_lines, 0);
1191        print_slab_summary();
1192}
1193
1194static void print_page_result(struct perf_session *session)
1195{
1196        if (caller_flag || alloc_flag)
1197                print_gfp_flags();
1198        if (caller_flag)
1199                __print_page_caller_result(session, caller_lines);
1200        if (alloc_flag)
1201                __print_page_alloc_result(session, alloc_lines);
1202        print_page_summary();
1203}
1204
1205static void print_result(struct perf_session *session)
1206{
1207        if (kmem_slab)
1208                print_slab_result(session);
1209        if (kmem_page)
1210                print_page_result(session);
1211}
1212
1213static LIST_HEAD(slab_caller_sort);
1214static LIST_HEAD(slab_alloc_sort);
1215static LIST_HEAD(page_caller_sort);
1216static LIST_HEAD(page_alloc_sort);
1217
1218static void sort_slab_insert(struct rb_root *root, struct alloc_stat *data,
1219                             struct list_head *sort_list)
1220{
1221        struct rb_node **new = &(root->rb_node);
1222        struct rb_node *parent = NULL;
1223        struct sort_dimension *sort;
1224
1225        while (*new) {
1226                struct alloc_stat *this;
1227                int cmp = 0;
1228
1229                this = rb_entry(*new, struct alloc_stat, node);
1230                parent = *new;
1231
1232                list_for_each_entry(sort, sort_list, list) {
1233                        cmp = sort->cmp(data, this);
1234                        if (cmp)
1235                                break;
1236                }
1237
1238                if (cmp > 0)
1239                        new = &((*new)->rb_left);
1240                else
1241                        new = &((*new)->rb_right);
1242        }
1243
1244        rb_link_node(&data->node, parent, new);
1245        rb_insert_color(&data->node, root);
1246}
1247
1248static void __sort_slab_result(struct rb_root *root, struct rb_root *root_sorted,
1249                               struct list_head *sort_list)
1250{
1251        struct rb_node *node;
1252        struct alloc_stat *data;
1253
1254        for (;;) {
1255                node = rb_first(root);
1256                if (!node)
1257                        break;
1258
1259                rb_erase(node, root);
1260                data = rb_entry(node, struct alloc_stat, node);
1261                sort_slab_insert(root_sorted, data, sort_list);
1262        }
1263}
1264
1265static void sort_page_insert(struct rb_root *root, struct page_stat *data,
1266                             struct list_head *sort_list)
1267{
1268        struct rb_node **new = &root->rb_node;
1269        struct rb_node *parent = NULL;
1270        struct sort_dimension *sort;
1271
1272        while (*new) {
1273                struct page_stat *this;
1274                int cmp = 0;
1275
1276                this = rb_entry(*new, struct page_stat, node);
1277                parent = *new;
1278
1279                list_for_each_entry(sort, sort_list, list) {
1280                        cmp = sort->cmp(data, this);
1281                        if (cmp)
1282                                break;
1283                }
1284
1285                if (cmp > 0)
1286                        new = &parent->rb_left;
1287                else
1288                        new = &parent->rb_right;
1289        }
1290
1291        rb_link_node(&data->node, parent, new);
1292        rb_insert_color(&data->node, root);
1293}
1294
1295static void __sort_page_result(struct rb_root *root, struct rb_root *root_sorted,
1296                               struct list_head *sort_list)
1297{
1298        struct rb_node *node;
1299        struct page_stat *data;
1300
1301        for (;;) {
1302                node = rb_first(root);
1303                if (!node)
1304                        break;
1305
1306                rb_erase(node, root);
1307                data = rb_entry(node, struct page_stat, node);
1308                sort_page_insert(root_sorted, data, sort_list);
1309        }
1310}
1311
1312static void sort_result(void)
1313{
1314        if (kmem_slab) {
1315                __sort_slab_result(&root_alloc_stat, &root_alloc_sorted,
1316                                   &slab_alloc_sort);
1317                __sort_slab_result(&root_caller_stat, &root_caller_sorted,
1318                                   &slab_caller_sort);
1319        }
1320        if (kmem_page) {
1321                if (live_page)
1322                        __sort_page_result(&page_live_tree, &page_alloc_sorted,
1323                                           &page_alloc_sort);
1324                else
1325                        __sort_page_result(&page_alloc_tree, &page_alloc_sorted,
1326                                           &page_alloc_sort);
1327
1328                __sort_page_result(&page_caller_tree, &page_caller_sorted,
1329                                   &page_caller_sort);
1330        }
1331}
1332
1333static int __cmd_kmem(struct perf_session *session)
1334{
1335        int err = -EINVAL;
1336        struct perf_evsel *evsel;
1337        const struct perf_evsel_str_handler kmem_tracepoints[] = {
1338                /* slab allocator */
1339                { "kmem:kmalloc",               perf_evsel__process_alloc_event, },
1340                { "kmem:kmem_cache_alloc",      perf_evsel__process_alloc_event, },
1341                { "kmem:kmalloc_node",          perf_evsel__process_alloc_node_event, },
1342                { "kmem:kmem_cache_alloc_node", perf_evsel__process_alloc_node_event, },
1343                { "kmem:kfree",                 perf_evsel__process_free_event, },
1344                { "kmem:kmem_cache_free",       perf_evsel__process_free_event, },
1345                /* page allocator */
1346                { "kmem:mm_page_alloc",         perf_evsel__process_page_alloc_event, },
1347                { "kmem:mm_page_free",          perf_evsel__process_page_free_event, },
1348        };
1349
1350        if (!perf_session__has_traces(session, "kmem record"))
1351                goto out;
1352
1353        if (perf_session__set_tracepoints_handlers(session, kmem_tracepoints)) {
1354                pr_err("Initializing perf session tracepoint handlers failed\n");
1355                goto out;
1356        }
1357
1358        evlist__for_each_entry(session->evlist, evsel) {
1359                if (!strcmp(perf_evsel__name(evsel), "kmem:mm_page_alloc") &&
1360                    perf_evsel__field(evsel, "pfn")) {
1361                        use_pfn = true;
1362                        break;
1363                }
1364        }
1365
1366        setup_pager();
1367        err = perf_session__process_events(session);
1368        if (err != 0) {
1369                pr_err("error during process events: %d\n", err);
1370                goto out;
1371        }
1372        sort_result();
1373        print_result(session);
1374out:
1375        return err;
1376}
1377
1378/* slab sort keys */
1379static int ptr_cmp(void *a, void *b)
1380{
1381        struct alloc_stat *l = a;
1382        struct alloc_stat *r = b;
1383
1384        if (l->ptr < r->ptr)
1385                return -1;
1386        else if (l->ptr > r->ptr)
1387                return 1;
1388        return 0;
1389}
1390
1391static struct sort_dimension ptr_sort_dimension = {
1392        .name   = "ptr",
1393        .cmp    = ptr_cmp,
1394};
1395
1396static int slab_callsite_cmp(void *a, void *b)
1397{
1398        struct alloc_stat *l = a;
1399        struct alloc_stat *r = b;
1400
1401        if (l->call_site < r->call_site)
1402                return -1;
1403        else if (l->call_site > r->call_site)
1404                return 1;
1405        return 0;
1406}
1407
1408static struct sort_dimension callsite_sort_dimension = {
1409        .name   = "callsite",
1410        .cmp    = slab_callsite_cmp,
1411};
1412
1413static int hit_cmp(void *a, void *b)
1414{
1415        struct alloc_stat *l = a;
1416        struct alloc_stat *r = b;
1417
1418        if (l->hit < r->hit)
1419                return -1;
1420        else if (l->hit > r->hit)
1421                return 1;
1422        return 0;
1423}
1424
1425static struct sort_dimension hit_sort_dimension = {
1426        .name   = "hit",
1427        .cmp    = hit_cmp,
1428};
1429
1430static int bytes_cmp(void *a, void *b)
1431{
1432        struct alloc_stat *l = a;
1433        struct alloc_stat *r = b;
1434
1435        if (l->bytes_alloc < r->bytes_alloc)
1436                return -1;
1437        else if (l->bytes_alloc > r->bytes_alloc)
1438                return 1;
1439        return 0;
1440}
1441
1442static struct sort_dimension bytes_sort_dimension = {
1443        .name   = "bytes",
1444        .cmp    = bytes_cmp,
1445};
1446
1447static int frag_cmp(void *a, void *b)
1448{
1449        double x, y;
1450        struct alloc_stat *l = a;
1451        struct alloc_stat *r = b;
1452
1453        x = fragmentation(l->bytes_req, l->bytes_alloc);
1454        y = fragmentation(r->bytes_req, r->bytes_alloc);
1455
1456        if (x < y)
1457                return -1;
1458        else if (x > y)
1459                return 1;
1460        return 0;
1461}
1462
1463static struct sort_dimension frag_sort_dimension = {
1464        .name   = "frag",
1465        .cmp    = frag_cmp,
1466};
1467
1468static int pingpong_cmp(void *a, void *b)
1469{
1470        struct alloc_stat *l = a;
1471        struct alloc_stat *r = b;
1472
1473        if (l->pingpong < r->pingpong)
1474                return -1;
1475        else if (l->pingpong > r->pingpong)
1476                return 1;
1477        return 0;
1478}
1479
1480static struct sort_dimension pingpong_sort_dimension = {
1481        .name   = "pingpong",
1482        .cmp    = pingpong_cmp,
1483};
1484
1485/* page sort keys */
1486static int page_cmp(void *a, void *b)
1487{
1488        struct page_stat *l = a;
1489        struct page_stat *r = b;
1490
1491        if (l->page < r->page)
1492                return -1;
1493        else if (l->page > r->page)
1494                return 1;
1495        return 0;
1496}
1497
1498static struct sort_dimension page_sort_dimension = {
1499        .name   = "page",
1500        .cmp    = page_cmp,
1501};
1502
1503static int page_callsite_cmp(void *a, void *b)
1504{
1505        struct page_stat *l = a;
1506        struct page_stat *r = b;
1507
1508        if (l->callsite < r->callsite)
1509                return -1;
1510        else if (l->callsite > r->callsite)
1511                return 1;
1512        return 0;
1513}
1514
1515static struct sort_dimension page_callsite_sort_dimension = {
1516        .name   = "callsite",
1517        .cmp    = page_callsite_cmp,
1518};
1519
1520static int page_hit_cmp(void *a, void *b)
1521{
1522        struct page_stat *l = a;
1523        struct page_stat *r = b;
1524
1525        if (l->nr_alloc < r->nr_alloc)
1526                return -1;
1527        else if (l->nr_alloc > r->nr_alloc)
1528                return 1;
1529        return 0;
1530}
1531
1532static struct sort_dimension page_hit_sort_dimension = {
1533        .name   = "hit",
1534        .cmp    = page_hit_cmp,
1535};
1536
1537static int page_bytes_cmp(void *a, void *b)
1538{
1539        struct page_stat *l = a;
1540        struct page_stat *r = b;
1541
1542        if (l->alloc_bytes < r->alloc_bytes)
1543                return -1;
1544        else if (l->alloc_bytes > r->alloc_bytes)
1545                return 1;
1546        return 0;
1547}
1548
1549static struct sort_dimension page_bytes_sort_dimension = {
1550        .name   = "bytes",
1551        .cmp    = page_bytes_cmp,
1552};
1553
1554static int page_order_cmp(void *a, void *b)
1555{
1556        struct page_stat *l = a;
1557        struct page_stat *r = b;
1558
1559        if (l->order < r->order)
1560                return -1;
1561        else if (l->order > r->order)
1562                return 1;
1563        return 0;
1564}
1565
1566static struct sort_dimension page_order_sort_dimension = {
1567        .name   = "order",
1568        .cmp    = page_order_cmp,
1569};
1570
1571static int migrate_type_cmp(void *a, void *b)
1572{
1573        struct page_stat *l = a;
1574        struct page_stat *r = b;
1575
1576        /* for internal use to find free'd page */
1577        if (l->migrate_type == -1U)
1578                return 0;
1579
1580        if (l->migrate_type < r->migrate_type)
1581                return -1;
1582        else if (l->migrate_type > r->migrate_type)
1583                return 1;
1584        return 0;
1585}
1586
1587static struct sort_dimension migrate_type_sort_dimension = {
1588        .name   = "migtype",
1589        .cmp    = migrate_type_cmp,
1590};
1591
1592static int gfp_flags_cmp(void *a, void *b)
1593{
1594        struct page_stat *l = a;
1595        struct page_stat *r = b;
1596
1597        /* for internal use to find free'd page */
1598        if (l->gfp_flags == -1U)
1599                return 0;
1600
1601        if (l->gfp_flags < r->gfp_flags)
1602                return -1;
1603        else if (l->gfp_flags > r->gfp_flags)
1604                return 1;
1605        return 0;
1606}
1607
1608static struct sort_dimension gfp_flags_sort_dimension = {
1609        .name   = "gfp",
1610        .cmp    = gfp_flags_cmp,
1611};
1612
1613static struct sort_dimension *slab_sorts[] = {
1614        &ptr_sort_dimension,
1615        &callsite_sort_dimension,
1616        &hit_sort_dimension,
1617        &bytes_sort_dimension,
1618        &frag_sort_dimension,
1619        &pingpong_sort_dimension,
1620};
1621
1622static struct sort_dimension *page_sorts[] = {
1623        &page_sort_dimension,
1624        &page_callsite_sort_dimension,
1625        &page_hit_sort_dimension,
1626        &page_bytes_sort_dimension,
1627        &page_order_sort_dimension,
1628        &migrate_type_sort_dimension,
1629        &gfp_flags_sort_dimension,
1630};
1631
1632static int slab_sort_dimension__add(const char *tok, struct list_head *list)
1633{
1634        struct sort_dimension *sort;
1635        int i;
1636
1637        for (i = 0; i < (int)ARRAY_SIZE(slab_sorts); i++) {
1638                if (!strcmp(slab_sorts[i]->name, tok)) {
1639                        sort = memdup(slab_sorts[i], sizeof(*slab_sorts[i]));
1640                        if (!sort) {
1641                                pr_err("%s: memdup failed\n", __func__);
1642                                return -1;
1643                        }
1644                        list_add_tail(&sort->list, list);
1645                        return 0;
1646                }
1647        }
1648
1649        return -1;
1650}
1651
1652static int page_sort_dimension__add(const char *tok, struct list_head *list)
1653{
1654        struct sort_dimension *sort;
1655        int i;
1656
1657        for (i = 0; i < (int)ARRAY_SIZE(page_sorts); i++) {
1658                if (!strcmp(page_sorts[i]->name, tok)) {
1659                        sort = memdup(page_sorts[i], sizeof(*page_sorts[i]));
1660                        if (!sort) {
1661                                pr_err("%s: memdup failed\n", __func__);
1662                                return -1;
1663                        }
1664                        list_add_tail(&sort->list, list);
1665                        return 0;
1666                }
1667        }
1668
1669        return -1;
1670}
1671
1672static int setup_slab_sorting(struct list_head *sort_list, const char *arg)
1673{
1674        char *tok;
1675        char *str = strdup(arg);
1676        char *pos = str;
1677
1678        if (!str) {
1679                pr_err("%s: strdup failed\n", __func__);
1680                return -1;
1681        }
1682
1683        while (true) {
1684                tok = strsep(&pos, ",");
1685                if (!tok)
1686                        break;
1687                if (slab_sort_dimension__add(tok, sort_list) < 0) {
1688                        error("Unknown slab --sort key: '%s'", tok);
1689                        free(str);
1690                        return -1;
1691                }
1692        }
1693
1694        free(str);
1695        return 0;
1696}
1697
1698static int setup_page_sorting(struct list_head *sort_list, const char *arg)
1699{
1700        char *tok;
1701        char *str = strdup(arg);
1702        char *pos = str;
1703
1704        if (!str) {
1705                pr_err("%s: strdup failed\n", __func__);
1706                return -1;
1707        }
1708
1709        while (true) {
1710                tok = strsep(&pos, ",");
1711                if (!tok)
1712                        break;
1713                if (page_sort_dimension__add(tok, sort_list) < 0) {
1714                        error("Unknown page --sort key: '%s'", tok);
1715                        free(str);
1716                        return -1;
1717                }
1718        }
1719
1720        free(str);
1721        return 0;
1722}
1723
1724static int parse_sort_opt(const struct option *opt __maybe_unused,
1725                          const char *arg, int unset __maybe_unused)
1726{
1727        if (!arg)
1728                return -1;
1729
1730        if (kmem_page > kmem_slab ||
1731            (kmem_page == 0 && kmem_slab == 0 && kmem_default == KMEM_PAGE)) {
1732                if (caller_flag > alloc_flag)
1733                        return setup_page_sorting(&page_caller_sort, arg);
1734                else
1735                        return setup_page_sorting(&page_alloc_sort, arg);
1736        } else {
1737                if (caller_flag > alloc_flag)
1738                        return setup_slab_sorting(&slab_caller_sort, arg);
1739                else
1740                        return setup_slab_sorting(&slab_alloc_sort, arg);
1741        }
1742
1743        return 0;
1744}
1745
1746static int parse_caller_opt(const struct option *opt __maybe_unused,
1747                            const char *arg __maybe_unused,
1748                            int unset __maybe_unused)
1749{
1750        caller_flag = (alloc_flag + 1);
1751        return 0;
1752}
1753
1754static int parse_alloc_opt(const struct option *opt __maybe_unused,
1755                           const char *arg __maybe_unused,
1756                           int unset __maybe_unused)
1757{
1758        alloc_flag = (caller_flag + 1);
1759        return 0;
1760}
1761
1762static int parse_slab_opt(const struct option *opt __maybe_unused,
1763                          const char *arg __maybe_unused,
1764                          int unset __maybe_unused)
1765{
1766        kmem_slab = (kmem_page + 1);
1767        return 0;
1768}
1769
1770static int parse_page_opt(const struct option *opt __maybe_unused,
1771                          const char *arg __maybe_unused,
1772                          int unset __maybe_unused)
1773{
1774        kmem_page = (kmem_slab + 1);
1775        return 0;
1776}
1777
1778static int parse_line_opt(const struct option *opt __maybe_unused,
1779                          const char *arg, int unset __maybe_unused)
1780{
1781        int lines;
1782
1783        if (!arg)
1784                return -1;
1785
1786        lines = strtoul(arg, NULL, 10);
1787
1788        if (caller_flag > alloc_flag)
1789                caller_lines = lines;
1790        else
1791                alloc_lines = lines;
1792
1793        return 0;
1794}
1795
1796static int __cmd_record(int argc, const char **argv)
1797{
1798        const char * const record_args[] = {
1799        "record", "-a", "-R", "-c", "1",
1800        };
1801        const char * const slab_events[] = {
1802        "-e", "kmem:kmalloc",
1803        "-e", "kmem:kmalloc_node",
1804        "-e", "kmem:kfree",
1805        "-e", "kmem:kmem_cache_alloc",
1806        "-e", "kmem:kmem_cache_alloc_node",
1807        "-e", "kmem:kmem_cache_free",
1808        };
1809        const char * const page_events[] = {
1810        "-e", "kmem:mm_page_alloc",
1811        "-e", "kmem:mm_page_free",
1812        };
1813        unsigned int rec_argc, i, j;
1814        const char **rec_argv;
1815
1816        rec_argc = ARRAY_SIZE(record_args) + argc - 1;
1817        if (kmem_slab)
1818                rec_argc += ARRAY_SIZE(slab_events);
1819        if (kmem_page)
1820                rec_argc += ARRAY_SIZE(page_events) + 1; /* for -g */
1821
1822        rec_argv = calloc(rec_argc + 1, sizeof(char *));
1823
1824        if (rec_argv == NULL)
1825                return -ENOMEM;
1826
1827        for (i = 0; i < ARRAY_SIZE(record_args); i++)
1828                rec_argv[i] = strdup(record_args[i]);
1829
1830        if (kmem_slab) {
1831                for (j = 0; j < ARRAY_SIZE(slab_events); j++, i++)
1832                        rec_argv[i] = strdup(slab_events[j]);
1833        }
1834        if (kmem_page) {
1835                rec_argv[i++] = strdup("-g");
1836
1837                for (j = 0; j < ARRAY_SIZE(page_events); j++, i++)
1838                        rec_argv[i] = strdup(page_events[j]);
1839        }
1840
1841        for (j = 1; j < (unsigned int)argc; j++, i++)
1842                rec_argv[i] = argv[j];
1843
1844        return cmd_record(i, rec_argv, NULL);
1845}
1846
1847static int kmem_config(const char *var, const char *value, void *cb __maybe_unused)
1848{
1849        if (!strcmp(var, "kmem.default")) {
1850                if (!strcmp(value, "slab"))
1851                        kmem_default = KMEM_SLAB;
1852                else if (!strcmp(value, "page"))
1853                        kmem_default = KMEM_PAGE;
1854                else
1855                        pr_err("invalid default value ('slab' or 'page' required): %s\n",
1856                               value);
1857                return 0;
1858        }
1859
1860        return 0;
1861}
1862
1863int cmd_kmem(int argc, const char **argv, const char *prefix __maybe_unused)
1864{
1865        const char * const default_slab_sort = "frag,hit,bytes";
1866        const char * const default_page_sort = "bytes,hit";
1867        struct perf_data_file file = {
1868                .mode = PERF_DATA_MODE_READ,
1869        };
1870        const struct option kmem_options[] = {
1871        OPT_STRING('i', "input", &input_name, "file", "input file name"),
1872        OPT_INCR('v', "verbose", &verbose,
1873                    "be more verbose (show symbol address, etc)"),
1874        OPT_CALLBACK_NOOPT(0, "caller", NULL, NULL,
1875                           "show per-callsite statistics", parse_caller_opt),
1876        OPT_CALLBACK_NOOPT(0, "alloc", NULL, NULL,
1877                           "show per-allocation statistics", parse_alloc_opt),
1878        OPT_CALLBACK('s', "sort", NULL, "key[,key2...]",
1879                     "sort by keys: ptr, callsite, bytes, hit, pingpong, frag, "
1880                     "page, order, migtype, gfp", parse_sort_opt),
1881        OPT_CALLBACK('l', "line", NULL, "num", "show n lines", parse_line_opt),
1882        OPT_BOOLEAN(0, "raw-ip", &raw_ip, "show raw ip instead of symbol"),
1883        OPT_BOOLEAN('f', "force", &file.force, "don't complain, do it"),
1884        OPT_CALLBACK_NOOPT(0, "slab", NULL, NULL, "Analyze slab allocator",
1885                           parse_slab_opt),
1886        OPT_CALLBACK_NOOPT(0, "page", NULL, NULL, "Analyze page allocator",
1887                           parse_page_opt),
1888        OPT_BOOLEAN(0, "live", &live_page, "Show live page stat"),
1889        OPT_END()
1890        };
1891        const char *const kmem_subcommands[] = { "record", "stat", NULL };
1892        const char *kmem_usage[] = {
1893                NULL,
1894                NULL
1895        };
1896        struct perf_session *session;
1897        int ret = -1;
1898        const char errmsg[] = "No %s allocation events found.  Have you run 'perf kmem record --%s'?\n";
1899
1900        perf_config(kmem_config, NULL);
1901        argc = parse_options_subcommand(argc, argv, kmem_options,
1902                                        kmem_subcommands, kmem_usage, 0);
1903
1904        if (!argc)
1905                usage_with_options(kmem_usage, kmem_options);
1906
1907        if (kmem_slab == 0 && kmem_page == 0) {
1908                if (kmem_default == KMEM_SLAB)
1909                        kmem_slab = 1;
1910                else
1911                        kmem_page = 1;
1912        }
1913
1914        if (!strncmp(argv[0], "rec", 3)) {
1915                symbol__init(NULL);
1916                return __cmd_record(argc, argv);
1917        }
1918
1919        file.path = input_name;
1920
1921        kmem_session = session = perf_session__new(&file, false, &perf_kmem);
1922        if (session == NULL)
1923                return -1;
1924
1925        if (kmem_slab) {
1926                if (!perf_evlist__find_tracepoint_by_name(session->evlist,
1927                                                          "kmem:kmalloc")) {
1928                        pr_err(errmsg, "slab", "slab");
1929                        goto out_delete;
1930                }
1931        }
1932
1933        if (kmem_page) {
1934                struct perf_evsel *evsel;
1935
1936                evsel = perf_evlist__find_tracepoint_by_name(session->evlist,
1937                                                             "kmem:mm_page_alloc");
1938                if (evsel == NULL) {
1939                        pr_err(errmsg, "page", "page");
1940                        goto out_delete;
1941                }
1942
1943                kmem_page_size = pevent_get_page_size(evsel->tp_format->pevent);
1944                symbol_conf.use_callchain = true;
1945        }
1946
1947        symbol__init(&session->header.env);
1948
1949        if (!strcmp(argv[0], "stat")) {
1950                setlocale(LC_ALL, "");
1951
1952                if (cpu__setup_cpunode_map())
1953                        goto out_delete;
1954
1955                if (list_empty(&slab_caller_sort))
1956                        setup_slab_sorting(&slab_caller_sort, default_slab_sort);
1957                if (list_empty(&slab_alloc_sort))
1958                        setup_slab_sorting(&slab_alloc_sort, default_slab_sort);
1959                if (list_empty(&page_caller_sort))
1960                        setup_page_sorting(&page_caller_sort, default_page_sort);
1961                if (list_empty(&page_alloc_sort))
1962                        setup_page_sorting(&page_alloc_sort, default_page_sort);
1963
1964                if (kmem_page) {
1965                        setup_page_sorting(&page_alloc_sort_input,
1966                                           "page,order,migtype,gfp");
1967                        setup_page_sorting(&page_caller_sort_input,
1968                                           "callsite,order,migtype,gfp");
1969                }
1970                ret = __cmd_kmem(session);
1971        } else
1972                usage_with_options(kmem_usage, kmem_options);
1973
1974out_delete:
1975        perf_session__delete(session);
1976
1977        return ret;
1978}
1979
1980