Alexey Ivanov | 777e802 | 2019-01-03 13:46:38 -0800 | [diff] [blame] | 1 | #!/usr/bin/env python |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 2 | # |
| 3 | # cachestat Count cache kernel function calls. |
| 4 | # For Linux, uses BCC, eBPF. See .c file. |
| 5 | # |
| 6 | # USAGE: cachestat |
| 7 | # Taken from funccount by Brendan Gregg |
| 8 | # This is a rewrite of cachestat from perf to bcc |
| 9 | # https://github.com/brendangregg/perf-tools/blob/master/fs/cachestat |
| 10 | # |
| 11 | # Copyright (c) 2016 Allan McAleavy. |
| 12 | # Copyright (c) 2015 Brendan Gregg. |
| 13 | # Licensed under the Apache License, Version 2.0 (the "License") |
| 14 | # |
| 15 | # 09-Sep-2015 Brendan Gregg Created this. |
| 16 | # 06-Nov-2015 Allan McAleavy |
| 17 | # 13-Jan-2016 Allan McAleavy run pep8 against program |
| 18 | |
| 19 | from __future__ import print_function |
| 20 | from bcc import BPF |
| 21 | from time import sleep, strftime |
Marko Myllynen | 27e7aea | 2018-09-26 20:09:07 +0300 | [diff] [blame] | 22 | import argparse |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 23 | import signal |
| 24 | import re |
| 25 | from sys import argv |
| 26 | |
| 27 | # signal handler |
| 28 | def signal_ignore(signal, frame): |
| 29 | print() |
| 30 | |
| 31 | # Function to gather data from /proc/meminfo |
| 32 | # return dictionary for quicker lookup of both values |
| 33 | def get_meminfo(): |
| 34 | result = dict() |
| 35 | |
| 36 | for line in open('/proc/meminfo'): |
| 37 | k = line.split(':', 3) |
| 38 | v = k[1].split() |
| 39 | result[k[0]] = int(v[0]) |
| 40 | return result |
| 41 | |
| 42 | # set global variables |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 43 | mpa = 0 |
| 44 | mbd = 0 |
| 45 | apcl = 0 |
| 46 | apd = 0 |
Joel Fernandes | 2b34870 | 2018-02-27 20:38:42 -0800 | [diff] [blame] | 47 | total = 0 |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 48 | misses = 0 |
Joel Fernandes | 2b34870 | 2018-02-27 20:38:42 -0800 | [diff] [blame] | 49 | hits = 0 |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 50 | debug = 0 |
| 51 | |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 52 | # arguments |
Marko Myllynen | 27e7aea | 2018-09-26 20:09:07 +0300 | [diff] [blame] | 53 | parser = argparse.ArgumentParser( |
| 54 | description="Count cache kernel function calls", |
| 55 | formatter_class=argparse.RawDescriptionHelpFormatter) |
| 56 | parser.add_argument("-T", "--timestamp", action="store_true", |
| 57 | help="include timestamp on output") |
| 58 | parser.add_argument("interval", nargs="?", default=5, |
| 59 | help="output interval, in seconds") |
| 60 | parser.add_argument("count", nargs="?", default=-1, |
| 61 | help="number of outputs") |
| 62 | parser.add_argument("--ebpf", action="store_true", |
| 63 | help=argparse.SUPPRESS) |
| 64 | args = parser.parse_args() |
| 65 | count = int(args.count) |
| 66 | tstamp = args.timestamp |
| 67 | interval = int(args.interval) |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 68 | |
Marko Myllynen | 27e7aea | 2018-09-26 20:09:07 +0300 | [diff] [blame] | 69 | # define BPF program |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 70 | bpf_text = """ |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 71 | #include <uapi/linux/ptrace.h> |
| 72 | struct key_t { |
| 73 | u64 ip; |
| 74 | }; |
| 75 | |
| 76 | BPF_HASH(counts, struct key_t); |
| 77 | |
| 78 | int do_count(struct pt_regs *ctx) { |
| 79 | struct key_t key = {}; |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 80 | u64 ip; |
| 81 | |
Naveen N. Rao | 4afa96a | 2016-05-03 14:54:21 +0530 | [diff] [blame] | 82 | key.ip = PT_REGS_IP(ctx); |
Javier Honduvilla Coto | 64bf965 | 2018-08-01 06:50:19 +0200 | [diff] [blame] | 83 | counts.increment(key); // update counter |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 84 | return 0; |
| 85 | } |
| 86 | |
| 87 | """ |
Marko Myllynen | 27e7aea | 2018-09-26 20:09:07 +0300 | [diff] [blame] | 88 | |
| 89 | if debug or args.ebpf: |
| 90 | print(bpf_text) |
| 91 | if args.ebpf: |
| 92 | exit() |
| 93 | |
| 94 | # load BPF program |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 95 | b = BPF(text=bpf_text) |
| 96 | b.attach_kprobe(event="add_to_page_cache_lru", fn_name="do_count") |
| 97 | b.attach_kprobe(event="mark_page_accessed", fn_name="do_count") |
| 98 | b.attach_kprobe(event="account_page_dirtied", fn_name="do_count") |
| 99 | b.attach_kprobe(event="mark_buffer_dirty", fn_name="do_count") |
| 100 | |
| 101 | # header |
| 102 | if tstamp: |
| 103 | print("%-8s " % "TIME", end="") |
Joel Fernandes | 2b34870 | 2018-02-27 20:38:42 -0800 | [diff] [blame] | 104 | print("%8s %8s %8s %8s %12s %10s" % |
| 105 | ("TOTAL", "MISSES", "HITS", "DIRTIES", "BUFFERS_MB", "CACHED_MB")) |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 106 | |
| 107 | loop = 0 |
| 108 | exiting = 0 |
| 109 | while 1: |
| 110 | if count > 0: |
| 111 | loop += 1 |
| 112 | if loop > count: |
| 113 | exit() |
| 114 | |
| 115 | try: |
| 116 | sleep(interval) |
| 117 | except KeyboardInterrupt: |
| 118 | exiting = 1 |
| 119 | # as cleanup can take many seconds, trap Ctrl-C: |
| 120 | signal.signal(signal.SIGINT, signal_ignore) |
| 121 | |
Marko Myllynen | 27e7aea | 2018-09-26 20:09:07 +0300 | [diff] [blame] | 122 | counts = b["counts"] |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 123 | for k, v in sorted(counts.items(), key=lambda counts: counts[1].value): |
| 124 | |
Brenden Blanco | 42d6098 | 2017-04-24 14:31:28 -0700 | [diff] [blame] | 125 | if re.match(b'mark_page_accessed', b.ksym(k.ip)) is not None: |
chantra | a2d669c | 2016-07-29 14:10:15 -0700 | [diff] [blame] | 126 | mpa = max(0, v.value) |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 127 | |
Brenden Blanco | 42d6098 | 2017-04-24 14:31:28 -0700 | [diff] [blame] | 128 | if re.match(b'mark_buffer_dirty', b.ksym(k.ip)) is not None: |
chantra | a2d669c | 2016-07-29 14:10:15 -0700 | [diff] [blame] | 129 | mbd = max(0, v.value) |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 130 | |
Brenden Blanco | 42d6098 | 2017-04-24 14:31:28 -0700 | [diff] [blame] | 131 | if re.match(b'add_to_page_cache_lru', b.ksym(k.ip)) is not None: |
chantra | a2d669c | 2016-07-29 14:10:15 -0700 | [diff] [blame] | 132 | apcl = max(0, v.value) |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 133 | |
Brenden Blanco | 42d6098 | 2017-04-24 14:31:28 -0700 | [diff] [blame] | 134 | if re.match(b'account_page_dirtied', b.ksym(k.ip)) is not None: |
chantra | a2d669c | 2016-07-29 14:10:15 -0700 | [diff] [blame] | 135 | apd = max(0, v.value) |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 136 | |
Joel Fernandes | 2b34870 | 2018-02-27 20:38:42 -0800 | [diff] [blame] | 137 | # total = total cache accesses without counting dirties |
| 138 | # misses = total of add to lru because of read misses |
| 139 | total = (mpa - mbd) |
| 140 | misses = (apcl - apd) |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 141 | |
Joel Fernandes | 2b34870 | 2018-02-27 20:38:42 -0800 | [diff] [blame] | 142 | if total < 0: |
| 143 | total = 0 |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 144 | |
Joel Fernandes | 2b34870 | 2018-02-27 20:38:42 -0800 | [diff] [blame] | 145 | if misses < 0: |
| 146 | misses = 0 |
| 147 | |
| 148 | hits = total - misses |
| 149 | |
| 150 | # If hits are < 0, then its possible misses are overestimated |
| 151 | # due to possibly page cache read ahead adding more pages than |
| 152 | # needed. In this case just assume misses as total and reset hits. |
| 153 | if hits < 0: |
| 154 | misses = total |
| 155 | hits = 0 |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 156 | |
| 157 | if debug: |
Joel Fernandes | 2b34870 | 2018-02-27 20:38:42 -0800 | [diff] [blame] | 158 | print("%d %d %d %d %d %d %d\n" % |
| 159 | (mpa, mbd, apcl, apd, total, misses, hits)) |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 160 | |
| 161 | counts.clear() |
| 162 | |
| 163 | # Get memory info |
| 164 | mem = get_meminfo() |
| 165 | cached = int(mem["Cached"]) / 1024 |
| 166 | buff = int(mem["Buffers"]) / 1024 |
| 167 | |
Marko Myllynen | 27e7aea | 2018-09-26 20:09:07 +0300 | [diff] [blame] | 168 | if tstamp: |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 169 | print("%-8s " % strftime("%H:%M:%S"), end="") |
Joel Fernandes | 2b34870 | 2018-02-27 20:38:42 -0800 | [diff] [blame] | 170 | print("%8d %8d %8d %8d %12.0f %10.0f" % |
| 171 | (total, misses, hits, mbd, buff, cached)) |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 172 | |
| 173 | mpa = 0 |
| 174 | mbd = 0 |
| 175 | apcl = 0 |
| 176 | apd = 0 |
Joel Fernandes | 2b34870 | 2018-02-27 20:38:42 -0800 | [diff] [blame] | 177 | total = 0 |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 178 | misses = 0 |
Joel Fernandes | 2b34870 | 2018-02-27 20:38:42 -0800 | [diff] [blame] | 179 | hits = 0 |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 180 | cached = 0 |
| 181 | buff = 0 |
unixtest | 57abe5b | 2016-01-31 10:47:03 +0000 | [diff] [blame] | 182 | |
| 183 | if exiting: |
| 184 | print("Detaching...") |
| 185 | exit() |