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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
|
/*
* Linux内核诊断工具--字符设备实现，用于用户态与内核态之间的交互
*
* Copyright (C) 2020 Alibaba Ltd.
*
* 作者: Baoyou Xie <[email protected]>
*
* License terms: GNU General Public License (GPL) version 3
*
*/
#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/kernel.h>
#include <linux/types.h>
#include <linux/notifier.h>
#include <linux/sched.h>
#include <linux/kdebug.h>
#include <linux/nmi.h>
#include <linux/vmalloc.h>
#include <linux/uaccess.h>
#include <linux/ptrace.h>
#include <linux/fs.h>
#include <linux/mm.h>
#include <linux/cdev.h>
#include <linux/slab.h>
#include <linux/hrtimer.h>
#include <linux/ktime.h>
#include <linux/cpumask.h>
#include <linux/mm_types.h>
#include <linux/pid_namespace.h>
#include <net/net_namespace.h>
#include <linux/inetdevice.h>
#include <asm/irq_regs.h>
#include <asm/ptrace.h>
#include <linux/stacktrace.h>
#include <asm/stacktrace.h>
#include <asm/syscall.h>
#include <linux/compiler.h>
#include <linux/version.h>
#include <linux/mm.h>
#include <linux/highmem.h>
#include <linux/kallsyms.h>
#include <linux/hardirq.h>
#include "uapi/ali_diagnose.h"
static int diag_dev_major = -1;
static struct class *diag_dev_class = NULL;
static struct device *diag_dev = NULL;
struct diag_dev {
struct cdev cdev;
};
static long diag_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
{
int ret = -EINVAL;
int type, nr;
type = _IOC_TYPE(cmd);
nr = _IOC_NR(cmd);
switch (type) {
case DIAG_IOCTL_TYPE_TEST:
if (nr == 1) {
struct diag_ioctl_test val;
ret = copy_from_user(&val, (void *)arg, sizeof(struct diag_ioctl_test));
if (!ret) {
val.out = val.in + 1;
ret = copy_to_user((void *)arg, &val, sizeof(struct diag_ioctl_test));
}
}
break;
case DIAG_IOCTL_TYPE_VERSION:
if (nr == 1) {
ret = DIAG_VERSION;
}
break;
case DIAG_IOCTL_TYPE_PUPIL:
ret = diag_ioctl_pupil_task(nr, arg);
break;
case DIAG_IOCTL_TYPE_SYS_DELAY:
ret = diag_ioctl_sys_delay(nr, arg);
break;
case DIAG_IOCTL_TYPE_SYS_COST:
ret = diag_ioctl_sys_cost(nr, arg);
break;
case DIAG_IOCTL_TYPE_SCHED_DELAY:
ret = diag_ioctl_sched_delay(nr, arg);
break;
case DIAG_IOCTL_TYPE_THROTTLE_DELAY:
ret = diag_ioctl_throttle_delay(nr, arg);
break;
case DIAG_IOCTL_TYPE_IRQ_DELAY:
ret = diag_ioctl_irq_delay(nr, arg);
break;
case DIAG_IOCTL_TYPE_IRQ_STATS:
ret = diag_ioctl_irq_stats(nr, arg);
break;
case DIAG_IOCTL_TYPE_IRQ_TRACE:
ret = diag_ioctl_irq_trace(nr, arg);
break;
case DIAG_IOCTL_TYPE_LOAD_MONITOR:
ret = diag_ioctl_load_monitor(nr, arg);
break;
case DIAG_IOCTL_TYPE_RUN_TRACE:
ret = diag_ioctl_run_trace(nr, arg);
break;
case DIAG_IOCTL_TYPE_PERF:
ret = diag_ioctl_perf(nr, arg);
break;
case DIAG_IOCTL_TYPE_KPROBE:
ret = diag_ioctl_kprobe(nr, arg);
break;
case DIAG_IOCTL_TYPE_UPROBE:
ret = diag_ioctl_uprobe(nr, arg);
break;
case DIAG_IOCTL_TYPE_UTILIZATION:
ret = diag_ioctl_utilization(nr, arg);
break;
case DIAG_IOCTL_TYPE_EXIT_MONITOR:
ret = diag_ioctl_exit_monitor(nr, arg);
break;
case DIAG_IOCTL_TYPE_MUTEX_MONITOR:
ret = diag_ioctl_mutex_monitor(nr, arg);
break;
case DIAG_IOCTL_TYPE_EXEC_MONITOR:
ret = diag_ioctl_exec_monitor(nr, arg);
break;
case DIAG_IOCTL_TYPE_ALLOC_TOP:
ret = diag_ioctl_alloc_top(nr, arg);
break;
case DIAG_IOCTL_TYPE_HIGH_ORDER:
ret = diag_ioctl_high_order(nr, arg);
break;
case DIAG_IOCTL_TYPE_DROP_PACKET:
ret = diag_ioctl_drop_packet(nr, arg);
break;
case DIAG_IOCTL_TYPE_TCP_RETRANS:
ret = diag_ioctl_tcp_retrans(nr, arg);
break;
case DIAG_IOCTL_TYPE_PING_DELAY:
ret = diag_ioctl_ping_delay(nr, arg);
break;
case DIAG_IOCTL_TYPE_PING_DELAY6:
ret = diag_ioctl_ping_delay6(nr, arg);
break;
case DIAG_IOCTL_TYPE_RW_TOP:
ret = diag_ioctl_rw_top(nr, arg);
break;
case DIAG_IOCTL_TYPE_FS_SHM:
ret = diag_ioctl_fs_shm(nr, arg);
break;
case DIAG_IOCTL_TYPE_FS_ORPHAN:
ret = diag_ioctl_fs_orphan(nr, arg);
break;
case DIAG_IOCTL_TYPE_FS_CACHE:
ret = diag_ioctl_fs_cache(nr, arg);
break;
case DIAG_IOCTL_TYPE_REBOOT:
ret = diag_ioctl_reboot(nr, arg);
break;
case DIAG_IOCTL_TYPE_NET_BANDWIDTH:
ret = diag_ioctl_net_bandwidth(nr, arg);
break;
case DIAG_IOCTL_TYPE_SIG_INFO:
ret = diag_ioctl_sig_info(nr, arg);
break;
case DIAG_IOCTL_TYPE_TASK_MONITOR:
ret = diag_ioctl_task_monitor(nr, arg);
break;
case DIAG_IOCTL_TYPE_RW_SEM:
ret = diag_ioctl_rw_sem(nr, arg);
break;
case DIAG_IOCTL_TYPE_RSS_MONITOR:
ret = diag_ioctl_rss_monitor(nr, arg);
break;
case DIAG_IOCTL_TYPE_MM_LEAK:
ret = diag_ioctl_mm_leak(nr, arg);
break;
case DIAG_IOCTL_TYPE_MEMCG_STATS:
ret = diag_ioctl_memcg_stats(nr, arg);
break;
case DIAG_IOCTL_TYPE_TCP_CONNECT:
ret = diag_ioctl_tcp_connect(nr, arg);
break;
default:
break;
}
return ret;
}
static int diag_open(struct inode *inode, struct file *file)
{
__module_get(THIS_MODULE);
return 0;
}
static int diag_release(struct inode *inode, struct file *file)
{
module_put(THIS_MODULE);
return 0;
}
static const struct file_operations diag_fops = {
.open = diag_open,
.release = diag_release,
.unlocked_ioctl = diag_ioctl,
};
#if LINUX_VERSION_CODE < KERNEL_VERSION(3,10,0)
static char *diag_devnode(struct device *dev, mode_t *mode)
#else
static char *diag_devnode(struct device *dev, umode_t *mode)
#endif
{
if (mode)
*mode = S_IRUGO | S_IRWXUGO | S_IALLUGO;
return kstrdup("diagnose-tools", GFP_KERNEL);;
}
int diag_dev_init(void)
{
int ret = 0;
diag_dev_major = register_chrdev(0, DIAG_DEV_NAME, &diag_fops);;
if (diag_dev_major < 0) {
printk("diagnose-tools: failed to register device\n");
return diag_dev_major;
}
diag_dev_class = class_create(THIS_MODULE, DIAG_DEV_NAME);
if (IS_ERR(diag_dev_class)) {
ret = PTR_ERR(diag_dev_class);
printk(KERN_ERR "diagnose-tools: class_create err=%d", ret);
unregister_chrdev(diag_dev_major, DIAG_DEV_NAME);
return ret;
}
diag_dev_class->devnode = diag_devnode;
diag_dev = device_create(diag_dev_class, NULL, MKDEV(diag_dev_major, 0), NULL, DIAG_DEV_NAME);
if (IS_ERR(diag_dev)) {
ret = PTR_ERR(diag_dev);
printk(KERN_ERR "diagnose-tools: device_create err=%d", ret);
unregister_chrdev(diag_dev_major, DIAG_DEV_NAME);
class_destroy(diag_dev_class);
return ret;
}
return 0;
}
void diag_dev_cleanup(void)
{
if (diag_dev_major >= 0) {
unregister_chrdev(diag_dev_major, DIAG_DEV_NAME);
}
if (diag_dev != NULL) {
device_destroy(diag_dev_class, MKDEV(diag_dev_major, 0));
}
if (diag_dev_class != NULL) {
class_destroy(diag_dev_class);
}
diag_dev_major = -1;
diag_dev = NULL;
diag_dev_class = NULL;
}
|
