src/modules-load/modules-load.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2010 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU Lesser General Public License as published by
  10   the Free Software Foundation; either version 2.1 of the License, or
  11   (at your option) any later version.
  12
  13   systemd is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   Lesser General Public License for more details.
  17
  18   You should have received a copy of the GNU Lesser General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <unistd.h>
  23 #include <fcntl.h>
  24 #include <errno.h>
  25 #include <string.h>
  26 #include <sys/stat.h>
  27 #include <limits.h>
  28 #include <dirent.h>
  29 #include <libkmod.h>
  30
  31 #include "log.h"
  32 #include "util.h"
  33 #include "strv.h"
  34
  35 #pragma GCC diagnostic push
  36 #pragma GCC diagnostic ignored "-Wformat-nonliteral"
  37 static void systemd_kmod_log(void *data, int priority, const char *file, int line,
  38                              const char *fn, const char *format, va_list args)
  39 {
  40         log_meta(priority, file, line, fn, format, args);
  41 }
  42 #pragma GCC diagnostic pop
  43
  44 int main(int argc, char *argv[]) {
  45         int r = EXIT_FAILURE;
  46         char **files, **fn;
  47         struct kmod_ctx *ctx;
  48         const int probe_flags = KMOD_PROBE_APPLY_BLACKLIST|KMOD_PROBE_IGNORE_LOADED;
  49
  50         if (argc > 1) {
  51                 log_error("This program takes no argument.");
  52                 return EXIT_FAILURE;
  53         }
  54
  55         log_set_target(LOG_TARGET_AUTO);
  56         log_parse_environment();
  57         log_open();
  58
  59         umask(0022);
  60
  61         ctx = kmod_new(NULL, NULL);
  62         if (!ctx) {
  63                 log_error("Failed to allocate memory for kmod.");
  64                 goto finish;
  65         }
  66
  67         kmod_load_resources(ctx);
  68
  69         kmod_set_log_fn(ctx, systemd_kmod_log, NULL);
  70
  71         if (conf_files_list(&files, ".conf",
  72                             "/etc/modules-load.d",
  73                             "/run/modules-load.d",
  74                             "/usr/local/lib/modules-load.d",
  75                             "/usr/lib/modules-load.d",
  76 #ifdef HAVE_SPLIT_USR
  77                             "/lib/modules-load.d",
  78 #endif
  79                             NULL) < 0) {
  80                 log_error("Failed to enumerate modules-load.d files: %s", strerror(-r));
  81                 goto finish;
  82         }
  83
  84         r = EXIT_SUCCESS;
  85
  86         STRV_FOREACH(fn, files) {
  87                 FILE *f;
  88
  89                 f = fopen(*fn, "re");
  90                 if (!f) {
  91                         if (errno == ENOENT)
  92                                 continue;
  93
  94                         log_error("Failed to open %s: %m", *fn);
  95                         r = EXIT_FAILURE;
  96                         continue;
  97                 }
  98
  99                 log_debug("apply: %s\n", *fn);
 100                 for (;;) {
 101                         char line[LINE_MAX], *l;
 102                         struct kmod_list *itr, *modlist = NULL;
 103                         int err;
 104
 105                         if (!fgets(line, sizeof(line), f))
 106                                 break;
 107
 108                         l = strstrip(line);
 109                         if (*l == '#' || *l == 0)
 110                                 continue;
 111
 112                         err = kmod_module_new_from_lookup(ctx, l, &modlist);
 113                         if (err < 0) {
 114                                 log_error("Failed to lookup alias '%s'", l);
 115                                 r = EXIT_FAILURE;
 116                                 continue;
 117                         }
 118
 119                         kmod_list_foreach(itr, modlist) {
 120                                 struct kmod_module *mod;
 121
 122                                 mod = kmod_module_get_module(itr);
 123                                 err = kmod_module_probe_insert_module(mod, probe_flags,
 124                                                                       NULL, NULL, NULL, NULL);
 125
 126                                 if (err == 0)
 127                                         log_info("Inserted module '%s'", kmod_module_get_name(mod));
 128                                 else if (err == KMOD_PROBE_APPLY_BLACKLIST)
 129                                         log_info("Module '%s' is blacklisted", kmod_module_get_name(mod));
 130                                 else {
 131                                         log_error("Failed to insert '%s': %s", kmod_module_get_name(mod),
 132                                                         strerror(-err));
 133                                         r = EXIT_FAILURE;
 134                                 }
 135
 136                                 kmod_module_unref(mod);
 137                         }
 138
 139                         kmod_module_unref_list(modlist);
 140                 }
 141
 142                 if (ferror(f)) {
 143                         log_error("Failed to read from file: %m");
 144                         r = EXIT_FAILURE;
 145                 }
 146
 147                 fclose(f);
 148         }
 149
 150 finish:
 151         strv_free(files);
 152         kmod_unref(ctx);
 153
 154         return r;
 155 }