Fix: statedump handle oom case, handle errors
[lttng-modules.git] / lttng-statedump-impl.c
1 /*
2 * lttng-statedump.c
3 *
4 * Linux Trace Toolkit Next Generation Kernel State Dump
5 *
6 * Copyright 2005 Jean-Hugues Deschenes <jean-hugues.deschenes@polymtl.ca>
7 * Copyright 2006-2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
8 *
9 * This library is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU Lesser General Public
11 * License as published by the Free Software Foundation; only
12 * version 2.1 of the License.
13 *
14 * This library is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 * Lesser General Public License for more details.
18 *
19 * You should have received a copy of the GNU Lesser General Public
20 * License along with this library; if not, write to the Free Software
21 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 *
23 * Changes:
24 * Eric Clement: Add listing of network IP interface
25 * 2006, 2007 Mathieu Desnoyers Fix kernel threads
26 * Various updates
27 */
28
29 #include <linux/init.h>
30 #include <linux/module.h>
31 #include <linux/netlink.h>
32 #include <linux/inet.h>
33 #include <linux/ip.h>
34 #include <linux/kthread.h>
35 #include <linux/proc_fs.h>
36 #include <linux/file.h>
37 #include <linux/interrupt.h>
38 #include <linux/irqnr.h>
39 #include <linux/cpu.h>
40 #include <linux/netdevice.h>
41 #include <linux/inetdevice.h>
42 #include <linux/sched.h>
43 #include <linux/mm.h>
44 #include <linux/fdtable.h>
45 #include <linux/swap.h>
46 #include <linux/wait.h>
47 #include <linux/mutex.h>
48 #include <linux/device.h>
49
50 #include "lttng-events.h"
51 #include "lttng-tracer.h"
52 #include "wrapper/irqdesc.h"
53 #include "wrapper/spinlock.h"
54 #include "wrapper/fdtable.h"
55 #include "wrapper/nsproxy.h"
56 #include "wrapper/irq.h"
57 #include "wrapper/tracepoint.h"
58 #include "wrapper/genhd.h"
59
60 #ifdef CONFIG_LTTNG_HAS_LIST_IRQ
61 #include <linux/irq.h>
62 #endif
63
64 /* Define the tracepoints, but do not build the probes */
65 #define CREATE_TRACE_POINTS
66 #define TRACE_INCLUDE_PATH ../instrumentation/events/lttng-module
67 #define TRACE_INCLUDE_FILE lttng-statedump
68 #include "instrumentation/events/lttng-module/lttng-statedump.h"
69
70 DEFINE_TRACE(lttng_statedump_block_device);
71 DEFINE_TRACE(lttng_statedump_end);
72 DEFINE_TRACE(lttng_statedump_interrupt);
73 DEFINE_TRACE(lttng_statedump_file_descriptor);
74 DEFINE_TRACE(lttng_statedump_start);
75 DEFINE_TRACE(lttng_statedump_process_state);
76 DEFINE_TRACE(lttng_statedump_network_interface);
77
78 struct lttng_fd_ctx {
79 char *page;
80 struct lttng_session *session;
81 struct task_struct *p;
82 struct fdtable *fdt;
83 };
84
85 /*
86 * Protected by the trace lock.
87 */
88 static struct delayed_work cpu_work[NR_CPUS];
89 static DECLARE_WAIT_QUEUE_HEAD(statedump_wq);
90 static atomic_t kernel_threads_to_run;
91
92 enum lttng_thread_type {
93 LTTNG_USER_THREAD = 0,
94 LTTNG_KERNEL_THREAD = 1,
95 };
96
97 enum lttng_execution_mode {
98 LTTNG_USER_MODE = 0,
99 LTTNG_SYSCALL = 1,
100 LTTNG_TRAP = 2,
101 LTTNG_IRQ = 3,
102 LTTNG_SOFTIRQ = 4,
103 LTTNG_MODE_UNKNOWN = 5,
104 };
105
106 enum lttng_execution_submode {
107 LTTNG_NONE = 0,
108 LTTNG_UNKNOWN = 1,
109 };
110
111 enum lttng_process_status {
112 LTTNG_UNNAMED = 0,
113 LTTNG_WAIT_FORK = 1,
114 LTTNG_WAIT_CPU = 2,
115 LTTNG_EXIT = 3,
116 LTTNG_ZOMBIE = 4,
117 LTTNG_WAIT = 5,
118 LTTNG_RUN = 6,
119 LTTNG_DEAD = 7,
120 };
121
122 static
123 int lttng_enumerate_block_devices(struct lttng_session *session)
124 {
125 struct class *ptr_block_class;
126 struct device_type *ptr_disk_type;
127 struct class_dev_iter iter;
128 struct device *dev;
129
130 ptr_block_class = wrapper_get_block_class();
131 if (!ptr_block_class)
132 return -ENOSYS;
133 ptr_disk_type = wrapper_get_disk_type();
134 if (!ptr_disk_type) {
135 return -ENOSYS;
136 }
137 class_dev_iter_init(&iter, ptr_block_class, NULL, ptr_disk_type);
138 while ((dev = class_dev_iter_next(&iter))) {
139 struct disk_part_iter piter;
140 struct gendisk *disk = dev_to_disk(dev);
141 struct hd_struct *part;
142
143 /*
144 * Don't show empty devices or things that have been
145 * suppressed
146 */
147 if (get_capacity(disk) == 0 ||
148 (disk->flags & GENHD_FL_SUPPRESS_PARTITION_INFO))
149 continue;
150
151 disk_part_iter_init(&piter, disk, DISK_PITER_INCL_PART0);
152 while ((part = disk_part_iter_next(&piter))) {
153 char name_buf[BDEVNAME_SIZE];
154 char *p;
155
156 p = wrapper_disk_name(disk, part->partno, name_buf);
157 if (!p) {
158 disk_part_iter_exit(&piter);
159 class_dev_iter_exit(&iter);
160 return -ENOSYS;
161 }
162 trace_lttng_statedump_block_device(session,
163 part_devt(part), name_buf);
164 }
165 disk_part_iter_exit(&piter);
166 }
167 class_dev_iter_exit(&iter);
168 return 0;
169 }
170
171 #ifdef CONFIG_INET
172
173 static
174 void lttng_enumerate_device(struct lttng_session *session,
175 struct net_device *dev)
176 {
177 struct in_device *in_dev;
178 struct in_ifaddr *ifa;
179
180 if (dev->flags & IFF_UP) {
181 in_dev = in_dev_get(dev);
182 if (in_dev) {
183 for (ifa = in_dev->ifa_list; ifa != NULL;
184 ifa = ifa->ifa_next) {
185 trace_lttng_statedump_network_interface(
186 session, dev, ifa);
187 }
188 in_dev_put(in_dev);
189 }
190 } else {
191 trace_lttng_statedump_network_interface(
192 session, dev, NULL);
193 }
194 }
195
196 static
197 int lttng_enumerate_network_ip_interface(struct lttng_session *session)
198 {
199 struct net_device *dev;
200
201 read_lock(&dev_base_lock);
202 for_each_netdev(&init_net, dev)
203 lttng_enumerate_device(session, dev);
204 read_unlock(&dev_base_lock);
205
206 return 0;
207 }
208 #else /* CONFIG_INET */
209 static inline
210 int lttng_enumerate_network_ip_interface(struct lttng_session *session)
211 {
212 return 0;
213 }
214 #endif /* CONFIG_INET */
215
216 static
217 int lttng_dump_one_fd(const void *p, struct file *file, unsigned int fd)
218 {
219 const struct lttng_fd_ctx *ctx = p;
220 const char *s = d_path(&file->f_path, ctx->page, PAGE_SIZE);
221 unsigned int flags = file->f_flags;
222
223 /*
224 * We don't expose kernel internal flags, only userspace-visible
225 * flags.
226 */
227 flags &= ~FMODE_NONOTIFY;
228 if (test_bit(fd, ctx->fdt->close_on_exec))
229 flags |= O_CLOEXEC;
230 if (IS_ERR(s)) {
231 struct dentry *dentry = file->f_path.dentry;
232
233 /* Make sure we give at least some info */
234 spin_lock(&dentry->d_lock);
235 trace_lttng_statedump_file_descriptor(ctx->session, ctx->p, fd,
236 dentry->d_name.name, flags, file->f_mode);
237 spin_unlock(&dentry->d_lock);
238 goto end;
239 }
240 trace_lttng_statedump_file_descriptor(ctx->session, ctx->p, fd, s,
241 flags, file->f_mode);
242 end:
243 return 0;
244 }
245
246 static
247 void lttng_enumerate_task_fd(struct lttng_session *session,
248 struct task_struct *p, char *tmp)
249 {
250 struct lttng_fd_ctx ctx = { .page = tmp, .session = session, .p = p };
251
252 task_lock(p);
253 ctx.fdt = files_fdtable(p->files);
254 lttng_iterate_fd(p->files, 0, lttng_dump_one_fd, &ctx);
255 task_unlock(p);
256 }
257
258 static
259 int lttng_enumerate_file_descriptors(struct lttng_session *session)
260 {
261 struct task_struct *p;
262 char *tmp;
263
264 tmp = (char *) __get_free_page(GFP_KERNEL);
265 if (!tmp)
266 return -ENOMEM;
267
268 /* Enumerate active file descriptors */
269 rcu_read_lock();
270 for_each_process(p)
271 lttng_enumerate_task_fd(session, p, tmp);
272 rcu_read_unlock();
273 free_page((unsigned long) tmp);
274 return 0;
275 }
276
277 #if 0
278 /*
279 * FIXME: we cannot take a mmap_sem while in a RCU read-side critical section
280 * (scheduling in atomic). Normally, the tasklist lock protects this kind of
281 * iteration, but it is not exported to modules.
282 */
283 static
284 void lttng_enumerate_task_vm_maps(struct lttng_session *session,
285 struct task_struct *p)
286 {
287 struct mm_struct *mm;
288 struct vm_area_struct *map;
289 unsigned long ino;
290
291 /* get_task_mm does a task_lock... */
292 mm = get_task_mm(p);
293 if (!mm)
294 return;
295
296 map = mm->mmap;
297 if (map) {
298 down_read(&mm->mmap_sem);
299 while (map) {
300 if (map->vm_file)
301 ino = map->vm_file->f_dentry->d_inode->i_ino;
302 else
303 ino = 0;
304 trace_lttng_statedump_vm_map(session, p, map, ino);
305 map = map->vm_next;
306 }
307 up_read(&mm->mmap_sem);
308 }
309 mmput(mm);
310 }
311
312 static
313 int lttng_enumerate_vm_maps(struct lttng_session *session)
314 {
315 struct task_struct *p;
316
317 rcu_read_lock();
318 for_each_process(p)
319 lttng_enumerate_task_vm_maps(session, p);
320 rcu_read_unlock();
321 return 0;
322 }
323 #endif
324
325 #ifdef CONFIG_LTTNG_HAS_LIST_IRQ
326
327 #if (LINUX_VERSION_CODE < KERNEL_VERSION(2,6,39))
328 #define irq_desc_get_chip(desc) get_irq_desc_chip(desc)
329 #endif
330
331 static
332 int lttng_list_interrupts(struct lttng_session *session)
333 {
334 unsigned int irq;
335 unsigned long flags = 0;
336 struct irq_desc *desc;
337
338 #define irq_to_desc wrapper_irq_to_desc
339 /* needs irq_desc */
340 for_each_irq_desc(irq, desc) {
341 struct irqaction *action;
342 const char *irq_chip_name =
343 irq_desc_get_chip(desc)->name ? : "unnamed_irq_chip";
344
345 local_irq_save(flags);
346 wrapper_desc_spin_lock(&desc->lock);
347 for (action = desc->action; action; action = action->next) {
348 trace_lttng_statedump_interrupt(session,
349 irq, irq_chip_name, action);
350 }
351 wrapper_desc_spin_unlock(&desc->lock);
352 local_irq_restore(flags);
353 }
354 return 0;
355 #undef irq_to_desc
356 }
357 #else
358 static inline
359 int lttng_list_interrupts(struct lttng_session *session)
360 {
361 return 0;
362 }
363 #endif
364
365 static
366 void lttng_statedump_process_ns(struct lttng_session *session,
367 struct task_struct *p,
368 enum lttng_thread_type type,
369 enum lttng_execution_mode mode,
370 enum lttng_execution_submode submode,
371 enum lttng_process_status status)
372 {
373 struct nsproxy *proxy;
374 struct pid_namespace *pid_ns;
375
376 rcu_read_lock();
377 proxy = task_nsproxy(p);
378 if (proxy) {
379 pid_ns = lttng_get_proxy_pid_ns(proxy);
380 do {
381 trace_lttng_statedump_process_state(session,
382 p, type, mode, submode, status, pid_ns);
383 pid_ns = pid_ns->parent;
384 } while (pid_ns);
385 } else {
386 trace_lttng_statedump_process_state(session,
387 p, type, mode, submode, status, NULL);
388 }
389 rcu_read_unlock();
390 }
391
392 static
393 int lttng_enumerate_process_states(struct lttng_session *session)
394 {
395 struct task_struct *g, *p;
396
397 rcu_read_lock();
398 for_each_process(g) {
399 p = g;
400 do {
401 enum lttng_execution_mode mode =
402 LTTNG_MODE_UNKNOWN;
403 enum lttng_execution_submode submode =
404 LTTNG_UNKNOWN;
405 enum lttng_process_status status;
406 enum lttng_thread_type type;
407
408 task_lock(p);
409 if (p->exit_state == EXIT_ZOMBIE)
410 status = LTTNG_ZOMBIE;
411 else if (p->exit_state == EXIT_DEAD)
412 status = LTTNG_DEAD;
413 else if (p->state == TASK_RUNNING) {
414 /* Is this a forked child that has not run yet? */
415 if (list_empty(&p->rt.run_list))
416 status = LTTNG_WAIT_FORK;
417 else
418 /*
419 * All tasks are considered as wait_cpu;
420 * the viewer will sort out if the task
421 * was really running at this time.
422 */
423 status = LTTNG_WAIT_CPU;
424 } else if (p->state &
425 (TASK_INTERRUPTIBLE | TASK_UNINTERRUPTIBLE)) {
426 /* Task is waiting for something to complete */
427 status = LTTNG_WAIT;
428 } else
429 status = LTTNG_UNNAMED;
430 submode = LTTNG_NONE;
431
432 /*
433 * Verification of t->mm is to filter out kernel
434 * threads; Viewer will further filter out if a
435 * user-space thread was in syscall mode or not.
436 */
437 if (p->mm)
438 type = LTTNG_USER_THREAD;
439 else
440 type = LTTNG_KERNEL_THREAD;
441 lttng_statedump_process_ns(session,
442 p, type, mode, submode, status);
443 task_unlock(p);
444 } while_each_thread(g, p);
445 }
446 rcu_read_unlock();
447
448 return 0;
449 }
450
451 static
452 void lttng_statedump_work_func(struct work_struct *work)
453 {
454 if (atomic_dec_and_test(&kernel_threads_to_run))
455 /* If we are the last thread, wake up do_lttng_statedump */
456 wake_up(&statedump_wq);
457 }
458
459 static
460 int do_lttng_statedump(struct lttng_session *session)
461 {
462 int cpu, ret;
463
464 trace_lttng_statedump_start(session);
465 ret = lttng_enumerate_process_states(session);
466 if (ret)
467 return ret;
468 ret = lttng_enumerate_file_descriptors(session);
469 if (ret)
470 return ret;
471 /*
472 * FIXME
473 * ret = lttng_enumerate_vm_maps(session);
474 * if (ret)
475 * return ret;
476 */
477 ret = lttng_list_interrupts(session);
478 if (ret)
479 return ret;
480 ret = lttng_enumerate_network_ip_interface(session);
481 if (ret)
482 return ret;
483 ret = lttng_enumerate_block_devices(session);
484 switch (ret) {
485 case -ENOSYS:
486 printk(KERN_WARNING "LTTng: block device enumeration is not supported by kernel\n");
487 break;
488 default:
489 return ret;
490 }
491
492 /* TODO lttng_dump_idt_table(session); */
493 /* TODO lttng_dump_softirq_vec(session); */
494 /* TODO lttng_list_modules(session); */
495 /* TODO lttng_dump_swap_files(session); */
496
497 /*
498 * Fire off a work queue on each CPU. Their sole purpose in life
499 * is to guarantee that each CPU has been in a state where is was in
500 * syscall mode (i.e. not in a trap, an IRQ or a soft IRQ).
501 */
502 get_online_cpus();
503 atomic_set(&kernel_threads_to_run, num_online_cpus());
504 for_each_online_cpu(cpu) {
505 INIT_DELAYED_WORK(&cpu_work[cpu], lttng_statedump_work_func);
506 schedule_delayed_work_on(cpu, &cpu_work[cpu], 0);
507 }
508 /* Wait for all threads to run */
509 __wait_event(statedump_wq, (atomic_read(&kernel_threads_to_run) == 0));
510 put_online_cpus();
511 /* Our work is done */
512 trace_lttng_statedump_end(session);
513 return 0;
514 }
515
516 /*
517 * Called with session mutex held.
518 */
519 int lttng_statedump_start(struct lttng_session *session)
520 {
521 return do_lttng_statedump(session);
522 }
523 EXPORT_SYMBOL_GPL(lttng_statedump_start);
524
525 static
526 int __init lttng_statedump_init(void)
527 {
528 /*
529 * Allow module to load even if the fixup cannot be done. This
530 * will allow seemless transition when the underlying issue fix
531 * is merged into the Linux kernel, and when tracepoint.c
532 * "tracepoint_module_notify" is turned into a static function.
533 */
534 (void) wrapper_lttng_fixup_sig(THIS_MODULE);
535 return 0;
536 }
537
538 module_init(lttng_statedump_init);
539
540 static
541 void __exit lttng_statedump_exit(void)
542 {
543 }
544
545 module_exit(lttng_statedump_exit);
546
547 MODULE_LICENSE("GPL and additional rights");
548 MODULE_AUTHOR("Jean-Hugues Deschenes");
549 MODULE_DESCRIPTION("Linux Trace Toolkit Next Generation Statedump");
550 MODULE_VERSION(__stringify(LTTNG_MODULES_MAJOR_VERSION) "."
551 __stringify(LTTNG_MODULES_MINOR_VERSION) "."
552 __stringify(LTTNG_MODULES_PATCHLEVEL_VERSION)
553 LTTNG_MODULES_EXTRAVERSION);
This page took 0.040908 seconds and 5 git commands to generate.