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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
# Copyright (c) Meta Platforms, Inc. and affiliates.
# SPDX-License-Identifier: LGPL-2.1-or-later
import contextlib
import os
from drgn import NULL, cast
from drgn.helpers.linux.fs import fget
from drgn.helpers.linux.kernfs import (
kernfs_name,
kernfs_parent,
kernfs_path,
kernfs_root,
kernfs_walk,
)
from drgn.helpers.linux.pid import find_task
from tests.linux_kernel import LinuxKernelTestCase
class TestKernfs(LinuxKernelTestCase):
@classmethod
def kernfs_node_from_fd(cls, fd):
file = fget(find_task(cls.prog, os.getpid()), fd)
return cast("struct kernfs_node *", file.f_inode.i_private)
def test_kernfs_parent(self):
with contextlib.ExitStack() as exit_stack:
fd = os.open("/sys/kernel/vmcoreinfo", os.O_RDONLY)
exit_stack.callback(os.close, fd)
dfd = os.open("/sys/kernel", os.O_RDONLY)
exit_stack.callback(os.close, dfd)
self.assertEqual(
kernfs_parent(self.kernfs_node_from_fd(fd)),
self.kernfs_node_from_fd(dfd),
)
def test_kernfs_root(self):
for path in ("/sys", "/sys/kernel", "/sys/kernel/vmcoreinfo"):
with self.subTest(path=path):
fd = os.open(path, os.O_RDONLY)
try:
self.assertEqual(
kernfs_root(self.kernfs_node_from_fd(fd)),
self.prog["sysfs_root"],
)
finally:
os.close(fd)
def test_kernfs_name(self):
with open("/sys/kernel/vmcoreinfo", "r") as f:
kn = self.kernfs_node_from_fd(f.fileno())
self.assertEqual(kernfs_name(kn), b"vmcoreinfo")
def test_kernfs_path(self):
with open("/sys/kernel/vmcoreinfo", "r") as f:
kn = self.kernfs_node_from_fd(f.fileno())
self.assertEqual(kernfs_path(kn), b"/kernel/vmcoreinfo")
def test_kernfs_walk(self):
fds = []
try:
for path in ("/sys", "/sys/kernel", "/sys/kernel/vmcoreinfo"):
fds.append(os.open(path, os.O_RDONLY))
kns = [self.kernfs_node_from_fd(fd) for fd in fds]
self.assertEqual(kernfs_walk(kns[0], ""), kns[0])
self.assertEqual(kernfs_walk(kns[0], "/"), kns[0])
self.assertEqual(kernfs_walk(kns[0], "kernel"), kns[1])
self.assertEqual(kernfs_walk(kns[0], "kernel/vmcoreinfo"), kns[2])
self.assertEqual(
kernfs_walk(kns[0], "kernel/foobar"),
NULL(self.prog, "struct kernfs_node *"),
)
self.assertEqual(
kernfs_walk(kns[0], "kernel/foo/bar"),
NULL(self.prog, "struct kernfs_node *"),
)
finally:
for fd in fds:
os.close(fd)
|
