Fix: syscall event rule: emission sites not compared in is_equal
[lttng-tools.git] / src / bin / lttng-sessiond / modprobe.cpp
CommitLineData
096102bd 1/*
ab5be9fa
MJ
2 * Copyright (C) 2011 David Goulet <dgoulet@efficios.com>
3 * Copyright (C) 2014 Jan Glauber <jan.glauber@gmail.com>
096102bd 4 *
ab5be9fa 5 * SPDX-License-Identifier: GPL-2.0-only
096102bd 6 *
096102bd
DG
7 */
8
d11b2027
MJ
9/**
10 * @file modprobe.c
11 *
12 * @brief modprobe related functions.
13 *
14 */
15
6c1c0768 16#define _LGPL_SOURCE
28ab034a
JG
17#include "kern-modules.hpp"
18#include "lttng-sessiond.hpp"
19#include "modprobe.hpp"
096102bd 20
c9e313bc
SM
21#include <common/common.hpp>
22#include <common/utils.hpp>
096102bd 23
28ab034a
JG
24#include <stdio.h>
25#include <stdlib.h>
26#include <sys/wait.h>
096102bd 27
ab57d7d3
JG
28/* LTTng kernel tracer mandatory core modules list */
29struct kern_modules_param kern_modules_control_core[] = {
adfe4f5e
JG
30 {
31 .name = (char *) "lttng-ring-buffer-client-discard",
32 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED,
1c9a0b0e 33 .loaded = false,
adfe4f5e
JG
34 },
35 {
36 .name = (char *) "lttng-ring-buffer-client-overwrite",
37 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED,
1c9a0b0e 38 .loaded = false,
adfe4f5e
JG
39 },
40 {
41 .name = (char *) "lttng-ring-buffer-metadata-client",
42 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED,
1c9a0b0e 43 .loaded = false,
adfe4f5e
JG
44 },
45 {
46 .name = (char *) "lttng-ring-buffer-client-mmap-discard",
47 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED,
1c9a0b0e 48 .loaded = false,
adfe4f5e
JG
49 },
50 {
51 .name = (char *) "lttng-ring-buffer-client-mmap-overwrite",
52 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED,
1c9a0b0e 53 .loaded = false,
adfe4f5e
JG
54 },
55 {
56 .name = (char *) "lttng-ring-buffer-metadata-mmap-client",
57 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED,
1c9a0b0e 58 .loaded = false,
adfe4f5e 59 },
87a15032
JR
60 {
61 .name = (char *) "lttng-ring-buffer-event_notifier-client",
62 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 63 .loaded = false,
87a15032 64 },
90aa04a1
FD
65 {
66 .name = (char *) "lttng-counter-client-percpu-64-modular",
67 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 68 .loaded = false,
90aa04a1
FD
69 },
70 {
71 .name = (char *) "lttng-counter-client-percpu-32-modular",
72 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 73 .loaded = false,
90aa04a1 74 },
ab57d7d3
JG
75};
76
adfe4f5e 77/* LTTng kerneltracer probe modules list */
fbb9748b 78struct kern_modules_param kern_modules_probes_default[] = {
adfe4f5e
JG
79 {
80 .name = (char *) "lttng-probe-asoc",
81 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 82 .loaded = false,
adfe4f5e
JG
83 },
84 {
85 .name = (char *) "lttng-probe-block",
86 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 87 .loaded = false,
adfe4f5e
JG
88 },
89 {
90 .name = (char *) "lttng-probe-btrfs",
91 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 92 .loaded = false,
adfe4f5e
JG
93 },
94 {
95 .name = (char *) "lttng-probe-compaction",
96 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 97 .loaded = false,
adfe4f5e
JG
98 },
99 {
100 .name = (char *) "lttng-probe-ext3",
101 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 102 .loaded = false,
adfe4f5e
JG
103 },
104 {
105 .name = (char *) "lttng-probe-ext4",
106 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 107 .loaded = false,
adfe4f5e
JG
108 },
109 {
110 .name = (char *) "lttng-probe-gpio",
111 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 112 .loaded = false,
adfe4f5e
JG
113 },
114 {
115 .name = (char *) "lttng-probe-i2c",
116 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 117 .loaded = false,
adfe4f5e
JG
118 },
119 {
120 .name = (char *) "lttng-probe-irq",
121 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 122 .loaded = false,
adfe4f5e
JG
123 },
124 {
125 .name = (char *) "lttng-probe-jbd",
126 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 127 .loaded = false,
adfe4f5e
JG
128 },
129 {
130 .name = (char *) "lttng-probe-jbd2",
131 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 132 .loaded = false,
adfe4f5e
JG
133 },
134 {
135 .name = (char *) "lttng-probe-kmem",
136 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 137 .loaded = false,
adfe4f5e
JG
138 },
139 {
140 .name = (char *) "lttng-probe-kvm",
141 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 142 .loaded = false,
adfe4f5e
JG
143 },
144 {
145 .name = (char *) "lttng-probe-kvm-x86",
146 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 147 .loaded = false,
adfe4f5e
JG
148 },
149 {
150 .name = (char *) "lttng-probe-kvm-x86-mmu",
151 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 152 .loaded = false,
adfe4f5e
JG
153 },
154 {
155 .name = (char *) "lttng-probe-lock",
156 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 157 .loaded = false,
adfe4f5e
JG
158 },
159 {
160 .name = (char *) "lttng-probe-module",
161 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 162 .loaded = false,
adfe4f5e
JG
163 },
164 {
165 .name = (char *) "lttng-probe-napi",
166 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 167 .loaded = false,
adfe4f5e
JG
168 },
169 {
170 .name = (char *) "lttng-probe-net",
171 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 172 .loaded = false,
adfe4f5e
JG
173 },
174 {
175 .name = (char *) "lttng-probe-power",
176 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 177 .loaded = false,
adfe4f5e
JG
178 },
179 {
180 .name = (char *) "lttng-probe-preemptirq",
181 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 182 .loaded = false,
adfe4f5e
JG
183 },
184 {
185 .name = (char *) "lttng-probe-printk",
186 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 187 .loaded = false,
adfe4f5e
JG
188 },
189 {
190 .name = (char *) "lttng-probe-random",
191 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 192 .loaded = false,
adfe4f5e
JG
193 },
194 {
195 .name = (char *) "lttng-probe-rcu",
196 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 197 .loaded = false,
adfe4f5e
JG
198 },
199 {
200 .name = (char *) "lttng-probe-regmap",
201 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 202 .loaded = false,
adfe4f5e
JG
203 },
204 {
205 .name = (char *) "lttng-probe-regulator",
206 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 207 .loaded = false,
adfe4f5e
JG
208 },
209 {
210 .name = (char *) "lttng-probe-rpm",
211 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 212 .loaded = false,
adfe4f5e
JG
213 },
214 {
215 .name = (char *) "lttng-probe-sched",
216 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 217 .loaded = false,
adfe4f5e
JG
218 },
219 {
220 .name = (char *) "lttng-probe-scsi",
221 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 222 .loaded = false,
adfe4f5e
JG
223 },
224 {
225 .name = (char *) "lttng-probe-signal",
226 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 227 .loaded = false,
adfe4f5e
JG
228 },
229 {
230 .name = (char *) "lttng-probe-skb",
231 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 232 .loaded = false,
adfe4f5e
JG
233 },
234 {
235 .name = (char *) "lttng-probe-sock",
236 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 237 .loaded = false,
adfe4f5e
JG
238 },
239 {
240 .name = (char *) "lttng-probe-statedump",
241 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 242 .loaded = false,
adfe4f5e
JG
243 },
244 {
245 .name = (char *) "lttng-probe-sunrpc",
246 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 247 .loaded = false,
adfe4f5e
JG
248 },
249 {
250 .name = (char *) "lttng-probe-timer",
251 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 252 .loaded = false,
adfe4f5e
JG
253 },
254 {
255 .name = (char *) "lttng-probe-udp",
256 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 257 .loaded = false,
adfe4f5e
JG
258 },
259 {
260 .name = (char *) "lttng-probe-vmscan",
261 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 262 .loaded = false,
adfe4f5e
JG
263 },
264 {
265 .name = (char *) "lttng-probe-v4l2",
266 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 267 .loaded = false,
adfe4f5e
JG
268 },
269 {
270 .name = (char *) "lttng-probe-workqueue",
271 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 272 .loaded = false,
adfe4f5e
JG
273 },
274 {
275 .name = (char *) "lttng-probe-writeback",
276 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 277 .loaded = false,
adfe4f5e
JG
278 },
279 {
280 .name = (char *) "lttng-probe-x86-irq-vectors",
281 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 282 .loaded = false,
adfe4f5e
JG
283 },
284 {
285 .name = (char *) "lttng-probe-x86-exceptions",
286 .load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL,
1c9a0b0e 287 .loaded = false,
adfe4f5e 288 },
096102bd
DG
289};
290
fbb9748b
JG
291/* dynamic probe modules list */
292static struct kern_modules_param *probes;
293static int nr_probes;
c9d42407 294static int probes_capacity;
fbb9748b 295
ce7fc42f 296#ifdef HAVE_KMOD
234170ac 297#include <libkmod.h>
866c17ce 298
d11b2027
MJ
299/**
300 * @brief Logging function for libkmod integration.
301 */
28ab034a
JG
302static ATTR_FORMAT_PRINTF(6, 0) void log_kmod(void *data __attribute__((unused)),
303 int priority __attribute__((unused)),
304 const char *file __attribute__((unused)),
305 int line __attribute__((unused)),
306 const char *fn __attribute__((unused)),
307 const char *format,
308 va_list args)
234170ac
UTL
309{
310 char *str;
311
312 if (vasprintf(&str, format, args) < 0) {
313 return;
314 }
315
316 DBG("libkmod: %s", str);
317 free(str);
318}
866c17ce 319
d11b2027
MJ
320/**
321 * @brief Setup the libkmod context.
322 *
323 * Create the context, add a custom logging function and preload the
324 * ressources for faster operation.
325 *
326 * @returns \c 0 on success
327 * \c < 0 on error
328 */
866c17ce 329static int setup_kmod_ctx(struct kmod_ctx **ctx)
234170ac 330{
866c17ce 331 int ret = 0;
234170ac 332
cd9adb8b 333 *ctx = kmod_new(nullptr, nullptr);
234170ac
UTL
334 if (!ctx) {
335 PERROR("Unable to create kmod library context");
336 ret = -ENOMEM;
337 goto error;
338 }
339
d22ad5f8
SM
340 /*
341 * Parameter 2 of kmod_set_log_fn generates a
342 * -Wsuggest-attribute=formatkmod_set_log_fn warning that we can't fix,
343 * ignore it.
344 */
345 DIAGNOSTIC_PUSH
346 DIAGNOSTIC_IGNORE_SUGGEST_ATTRIBUTE_FORMAT
cd9adb8b 347 kmod_set_log_fn(*ctx, log_kmod, nullptr);
d22ad5f8 348 DIAGNOSTIC_POP
866c17ce
MJ
349 ret = kmod_load_resources(*ctx);
350 if (ret < 0) {
351 ERR("Failed to load kmod library resources");
352 goto error;
353 }
354
355error:
356 return ret;
357}
358
d11b2027
MJ
359/**
360 * @brief Loads the kernel modules in \p modules
361 *
362 * @param modules List of modules to load
363 * @param entries Number of modules in the list
d11b2027
MJ
364 *
365 * If the modules are required, we will return with error after the
366 * first failed module load, otherwise we continue loading.
367 *
368 * @returns \c 0 on success
369 * \c < 0 on error
370 */
28ab034a 371static int modprobe_lttng(struct kern_modules_param *modules, int entries)
866c17ce
MJ
372{
373 int ret = 0, i;
374 struct kmod_ctx *ctx;
375
376 ret = setup_kmod_ctx(&ctx);
377 if (ret < 0) {
378 goto error;
379 }
234170ac
UTL
380
381 for (i = 0; i < entries; i++) {
cd9adb8b 382 struct kmod_module *mod = nullptr;
234170ac
UTL
383
384 ret = kmod_module_new_from_name(ctx, modules[i].name, &mod);
385 if (ret < 0) {
386 PERROR("Failed to create kmod module for %s", modules[i].name);
387 goto error;
388 }
389
cd9adb8b 390 ret = kmod_module_probe_insert_module(mod, 0, nullptr, nullptr, nullptr, nullptr);
0b1e16b8
MJ
391 if (ret == -EEXIST) {
392 DBG("Module %s is already loaded", modules[i].name);
393 ret = 0;
394 } else if (ret < 0) {
adfe4f5e 395 if (modules[i].load_policy == KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED) {
7d46777b 396 PERROR("Unable to load required module `%s`", modules[i].name)
16c2e854
PP
397 goto error;
398 } else {
7d46777b
KS
399 PERROR("Unable to load optional module `%s`; continuing",
400 modules[i].name);
16c2e854
PP
401 ret = 0;
402 }
234170ac
UTL
403 } else {
404 DBG("Modprobe successfully %s", modules[i].name);
4ad664a0 405 modules[i].loaded = true;
234170ac
UTL
406 }
407
408 kmod_module_unref(mod);
409 }
410
411error:
412 if (ctx) {
413 kmod_unref(ctx);
414 }
415 return ret;
416}
417
d11b2027
MJ
418/**
419 * @brief Recursively unload modules.
420 *
421 * This function implements the same modules unloading behavior as
422 * 'modprobe -r' or rmmod, it will recursevily go trought the \p module
423 * dependencies and unload modules with a refcount of 0.
424 *
425 * @param mod The module to unload
426 *
427 * @returns \c 0 on success
428 * \c < 0 on error
429 */
28ab034a
JG
430static int rmmod_recurse(struct kmod_module *mod)
431{
866c17ce
MJ
432 int ret = 0;
433 struct kmod_list *deps, *itr;
434
435 if (kmod_module_get_initstate(mod) == KMOD_MODULE_BUILTIN) {
436 DBG("Module %s is builtin", kmod_module_get_name(mod));
437 return ret;
438 }
439
440 ret = kmod_module_remove_module(mod, 0);
441
442 deps = kmod_module_get_dependencies(mod);
cd9adb8b 443 if (deps != nullptr) {
28ab034a
JG
444 kmod_list_foreach(itr, deps)
445 {
866c17ce
MJ
446 struct kmod_module *dep = kmod_module_get_module(itr);
447 if (kmod_module_get_refcnt(dep) == 0) {
28ab034a 448 DBG("Recursive remove module %s", kmod_module_get_name(dep));
866c17ce
MJ
449 rmmod_recurse(dep);
450 }
451 kmod_module_unref(dep);
452 }
453 kmod_module_unref_list(deps);
454 }
455
456 return ret;
457}
458
d11b2027
MJ
459/**
460 * @brief Unloads the kernel modules in \p modules
461 *
462 * @param modules List of modules to unload
463 * @param entries Number of modules in the list
d11b2027
MJ
464 *
465 */
28ab034a 466static void modprobe_remove_lttng(const struct kern_modules_param *modules, int entries)
866c17ce
MJ
467{
468 int ret = 0, i;
469 struct kmod_ctx *ctx;
470
471 ret = setup_kmod_ctx(&ctx);
472 if (ret < 0) {
473 goto error;
474 }
475
476 for (i = entries - 1; i >= 0; i--) {
cd9adb8b 477 struct kmod_module *mod = nullptr;
866c17ce 478
4ad664a0
JG
479 if (!modules[i].loaded) {
480 continue;
481 }
482
866c17ce
MJ
483 ret = kmod_module_new_from_name(ctx, modules[i].name, &mod);
484 if (ret < 0) {
485 PERROR("Failed to create kmod module for %s", modules[i].name);
486 goto error;
487 }
488
489 ret = rmmod_recurse(mod);
490 if (ret == -EEXIST) {
491 DBG("Module %s is not in kernel.", modules[i].name);
28ab034a
JG
492 } else if (modules[i].load_policy == KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED &&
493 ret < 0) {
7d46777b 494 PERROR("Unable to remove module `%s`", modules[i].name);
866c17ce 495 } else {
28ab034a 496 DBG("Modprobe removal successful %s", modules[i].name);
866c17ce
MJ
497 }
498
499 kmod_module_unref(mod);
500 }
501
502error:
503 if (ctx) {
504 kmod_unref(ctx);
505 }
506}
507
234170ac
UTL
508#else /* HAVE_KMOD */
509
28ab034a 510static int modprobe_lttng(struct kern_modules_param *modules, int entries)
096102bd
DG
511{
512 int ret = 0, i;
513 char modprobe[256];
514
e23b81ed 515 for (i = 0; i < entries; i++) {
28ab034a
JG
516 ret = snprintf(modprobe,
517 sizeof(modprobe),
518 "/sbin/modprobe %s%s",
519 modules[i].load_policy ==
520 KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED ?
521 "" :
522 "-q ",
523 modules[i].name);
096102bd
DG
524 if (ret < 0) {
525 PERROR("snprintf modprobe");
526 goto error;
527 }
528 modprobe[sizeof(modprobe) - 1] = '\0';
529 ret = system(modprobe);
530 if (ret == -1) {
adfe4f5e 531 if (modules[i].load_policy == KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED) {
16c2e854 532 ERR("Unable to launch modprobe for required module %s",
28ab034a 533 modules[i].name);
16c2e854
PP
534 goto error;
535 } else {
536 DBG("Unable to launch modprobe for optional module %s; continuing",
28ab034a 537 modules[i].name);
16c2e854
PP
538 ret = 0;
539 }
540 } else if (WEXITSTATUS(ret) != 0) {
adfe4f5e 541 if (modules[i].load_policy == KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED) {
28ab034a 542 ERR("Unable to load required module %s", modules[i].name);
49cddecd
KS
543 /* Force negative return code */
544 if (ret > 0) {
545 ret = -ret;
546 }
16c2e854
PP
547 goto error;
548 } else {
549 DBG("Unable to load optional module %s; continuing",
28ab034a 550 modules[i].name);
16c2e854
PP
551 ret = 0;
552 }
096102bd 553 } else {
ab57d7d3 554 DBG("Modprobe successfully %s", modules[i].name);
355d2778 555 modules[i].loaded = true;
096102bd
DG
556 }
557 }
558
559error:
560 return ret;
561}
562
28ab034a 563static void modprobe_remove_lttng(const struct kern_modules_param *modules, int entries)
35e090b7
MJ
564{
565 int ret = 0, i;
566 char modprobe[256];
567
568 for (i = entries - 1; i >= 0; i--) {
4ad664a0
JG
569 if (!modules[i].loaded) {
570 continue;
571 }
28ab034a
JG
572 ret = snprintf(
573 modprobe, sizeof(modprobe), "/sbin/modprobe -r -q %s", modules[i].name);
35e090b7
MJ
574 if (ret < 0) {
575 PERROR("snprintf modprobe -r");
576 return;
577 }
578 modprobe[sizeof(modprobe) - 1] = '\0';
579 ret = system(modprobe);
580 if (ret == -1) {
1d25334c
FD
581 if (modules[i].load_policy == KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED) {
582 ERR("Unable to launch modprobe -r for required module %s",
28ab034a 583 modules[i].name);
1d25334c
FD
584 } else {
585 DBG("Unable to launch modprobe -r for optional module %s",
28ab034a 586 modules[i].name);
1d25334c
FD
587 }
588 } else if (WEXITSTATUS(ret) != 0) {
589 if (modules[i].load_policy == KERNEL_MODULE_PROPERTY_LOAD_POLICY_REQUIRED) {
28ab034a 590 ERR("Unable to remove required module %s", modules[i].name);
1d25334c 591 } else {
28ab034a 592 DBG("Unable to remove optional module %s", modules[i].name);
1d25334c 593 }
35e090b7 594 } else {
1d25334c 595 DBG("Modprobe removal successful %s", modules[i].name);
35e090b7
MJ
596 }
597 }
598}
599
866c17ce
MJ
600#endif /* HAVE_KMOD */
601
35e090b7
MJ
602/*
603 * Remove control kernel module(s) in reverse load order.
604 */
cd9adb8b 605void modprobe_remove_lttng_control()
35e090b7 606{
28ab034a 607 modprobe_remove_lttng(kern_modules_control_core, ARRAY_SIZE(kern_modules_control_core));
35e090b7
MJ
608}
609
cd9adb8b 610static void free_probes()
35e090b7
MJ
611{
612 int i;
613
614 if (!probes) {
615 return;
616 }
617 for (i = 0; i < nr_probes; ++i) {
618 free(probes[i].name);
619 }
620 free(probes);
cd9adb8b 621 probes = nullptr;
35e090b7
MJ
622 nr_probes = 0;
623}
624
625/*
626 * Remove data kernel modules in reverse load order.
627 */
cd9adb8b 628void modprobe_remove_lttng_data()
35e090b7
MJ
629{
630 if (!probes) {
631 return;
632 }
adfe4f5e
JG
633
634 modprobe_remove_lttng(probes, nr_probes);
35e090b7
MJ
635 free_probes();
636}
637
638/*
639 * Remove all kernel modules in reverse order.
640 */
cd9adb8b 641void modprobe_remove_lttng_all()
35e090b7
MJ
642{
643 modprobe_remove_lttng_data();
644 modprobe_remove_lttng_control();
645}
646
e23b81ed
JG
647/*
648 * Load control kernel module(s).
649 */
cd9adb8b 650int modprobe_lttng_control()
e23b81ed 651{
28ab034a 652 return modprobe_lttng(kern_modules_control_core, ARRAY_SIZE(kern_modules_control_core));
e23b81ed 653}
ab57d7d3 654
c9d42407
PP
655/**
656 * Grow global list of probes (double capacity or set it to 1 if
657 * currently 0 and copy existing data).
096102bd 658 */
cd9adb8b 659static int grow_probes()
096102bd 660{
c9d42407
PP
661 int i;
662 struct kern_modules_param *tmp_probes;
fbb9748b 663
c9d42407
PP
664 /* Initialize capacity to 1 if 0. */
665 if (probes_capacity == 0) {
64803277 666 probes = zmalloc<kern_modules_param>();
c9d42407
PP
667 if (!probes) {
668 PERROR("malloc probe list");
669 return -ENOMEM;
670 }
671
672 probes_capacity = 1;
673 return 0;
fbb9748b
JG
674 }
675
c9d42407
PP
676 /* Double size. */
677 probes_capacity *= 2;
678
64803277 679 tmp_probes = calloc<kern_modules_param>(probes_capacity);
c9d42407 680 if (!tmp_probes) {
fbb9748b
JG
681 PERROR("malloc probe list");
682 return -ENOMEM;
683 }
684
c9d42407 685 for (i = 0; i < nr_probes; ++i) {
adfe4f5e
JG
686 /* Ownership of 'name' field is transferred. */
687 tmp_probes[i] = probes[i];
c9d42407
PP
688 }
689
690 /* Replace probes with larger copy. */
691 free(probes);
692 probes = tmp_probes;
693
694 return 0;
695}
696
697/*
698 * Appends a comma-separated list of probes to the global list
699 * of probes.
700 */
701static int append_list_to_probes(const char *list)
702{
703 char *next;
d3c04b7c 704 int ret;
50e0fa5f 705 char *tmp_list, *cur_list, *saveptr;
c9d42407 706
a0377dfe 707 LTTNG_ASSERT(list);
c9d42407 708
44603c80 709 cur_list = tmp_list = strdup(list);
c9d42407
PP
710 if (!tmp_list) {
711 PERROR("strdup temp list");
712 return -ENOMEM;
713 }
714
715 for (;;) {
fbb9748b 716 size_t name_len;
c9d42407 717 struct kern_modules_param *cur_mod;
fbb9748b 718
50e0fa5f 719 next = strtok_r(cur_list, ",", &saveptr);
fbb9748b 720 if (!next) {
c9d42407 721 break;
fbb9748b 722 }
cd9adb8b 723 cur_list = nullptr;
fbb9748b
JG
724
725 /* filter leading spaces */
726 while (*next == ' ') {
727 next++;
728 }
729
c9d42407
PP
730 if (probes_capacity <= nr_probes) {
731 ret = grow_probes();
732 if (ret) {
398d5459 733 goto error;
c9d42407
PP
734 }
735 }
736
fbb9748b
JG
737 /* Length 13 is "lttng-probe-" + \0 */
738 name_len = strlen(next) + 13;
739
d3c04b7c 740 cur_mod = &probes[nr_probes];
64803277 741 cur_mod->name = calloc<char>(name_len);
c9d42407 742 if (!cur_mod->name) {
fbb9748b 743 PERROR("malloc probe list");
398d5459
MD
744 ret = -ENOMEM;
745 goto error;
fbb9748b
JG
746 }
747
c9d42407 748 ret = snprintf(cur_mod->name, name_len, "lttng-probe-%s", next);
fbb9748b
JG
749 if (ret < 0) {
750 PERROR("snprintf modprobe name");
398d5459
MD
751 ret = -ENOMEM;
752 goto error;
fbb9748b 753 }
c9d42407 754
adfe4f5e
JG
755 cur_mod->load_policy = KERNEL_MODULE_PROPERTY_LOAD_POLICY_OPTIONAL;
756
c9d42407 757 nr_probes++;
fbb9748b
JG
758 }
759
c9d42407 760 free(tmp_list);
c9d42407 761 return 0;
398d5459
MD
762
763error:
764 free(tmp_list);
765 free_probes();
766 return ret;
c9d42407
PP
767}
768
769/*
770 * Load data kernel module(s).
771 */
cd9adb8b 772int modprobe_lttng_data()
c9d42407
PP
773{
774 int ret, i;
775 char *list;
776
777 /*
778 * Base probes: either from command line option, environment
779 * variable or default list.
780 */
412d7227 781 list = the_config.kmod_probes_list.value;
c9d42407
PP
782 if (list) {
783 /* User-specified probes. */
784 ret = append_list_to_probes(list);
c9d42407
PP
785 if (ret) {
786 return ret;
787 }
788 } else {
789 /* Default probes. */
790 int def_len = ARRAY_SIZE(kern_modules_probes_default);
c9d42407 791
64803277 792 probes = calloc<kern_modules_param>(def_len);
c9d42407
PP
793 if (!probes) {
794 PERROR("malloc probe list");
795 return -ENOMEM;
796 }
797
798 nr_probes = probes_capacity = def_len;
799
800 for (i = 0; i < def_len; ++i) {
28ab034a 801 char *name = strdup(kern_modules_probes_default[i].name);
c9d42407
PP
802
803 if (!name) {
804 PERROR("strdup probe item");
398d5459
MD
805 ret = -ENOMEM;
806 goto error;
c9d42407
PP
807 }
808
809 probes[i].name = name;
adfe4f5e 810 probes[i].load_policy = kern_modules_probes_default[i].load_policy;
c9d42407
PP
811 }
812 }
813
814 /*
815 * Extra modules? Append them to current probes list.
816 */
412d7227 817 list = the_config.kmod_extra_probes_list.value;
c9d42407
PP
818 if (list) {
819 ret = append_list_to_probes(list);
820 if (ret) {
398d5459 821 goto error;
c9d42407
PP
822 }
823 }
824
825 /*
826 * Load probes modules now.
827 */
adfe4f5e 828 ret = modprobe_lttng(probes, nr_probes);
398d5459
MD
829 if (ret) {
830 goto error;
831 }
832 return ret;
833
834error:
835 free_probes();
836 return ret;
096102bd 837}
This page took 0.158076 seconds and 5 git commands to generate.