1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
#!/usr/bin/env python3
# Copyright (c) PLUMgrid, Inc.
# Licensed under the Apache License, Version 2.0 (the "License")
from ctypes import c_uint, c_ulong, Structure
from bcc import BPF
from time import sleep
import sys
from unittest import main, TestCase
from utils import mayFail
arg1 = sys.argv.pop(1).encode()
arg2 = b""
if len(sys.argv) > 1:
arg2 = sys.argv.pop(1)
class TestBlkRequest(TestCase):
def setUp(self):
b = BPF(arg1, arg2, debug=0)
self.latency = b.get_table(b"latency", c_uint, c_ulong)
if BPF.get_kprobe_functions(b"blk_start_request"):
b.attach_kprobe(event=b"blk_start_request",
fn_name=b"probe_blk_start_request")
b.attach_kprobe(event=b"blk_mq_start_request",
fn_name=b"probe_blk_start_request")
b.attach_kprobe(event=b"blk_update_request",
fn_name=b"probe_blk_update_request")
def test_blk1(self):
import subprocess
import os
# use /opt instead of /tmp so that it hits a real disk
for i in range(0, 2):
subprocess.call(["dd", "if=/dev/zero", "of=/opt/trace3.txt",
"count=1024", "bs=4096"])
subprocess.call(["sync"])
os.unlink("/opt/trace3.txt")
for key, leaf in self.latency.items():
print("latency %u:" % key.value, "count %u" % leaf.value)
sys.stdout.flush()
self.assertEqual(len(list(self.latency.keys())), len(self.latency))
if __name__ == "__main__":
main()
|
