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 #include "conf-files.h"
  35
  36 #pragma GCC diagnostic push
  37 #pragma GCC diagnostic ignored "-Wformat-nonliteral"
  38 static void systemd_kmod_log(void *data, int priority, const char *file, int line,
  39                              const char *fn, const char *format, va_list args)
  40 {
  41         log_meta(priority, file, line, fn, format, args);
  42 }
  43 #pragma GCC diagnostic pop
  44
  45 int main(int argc, char *argv[]) {
  46         int r = EXIT_FAILURE;
  47         char **files, **fn;
  48         struct kmod_ctx *ctx;
  49         const int probe_flags = KMOD_PROBE_APPLY_BLACKLIST|KMOD_PROBE_IGNORE_LOADED;
  50
  51         if (argc > 1) {
  52                 log_error("This program takes no argument.");
  53                 return EXIT_FAILURE;
  54         }
  55
  56         log_set_target(LOG_TARGET_AUTO);
  57         log_parse_environment();
  58         log_open();
  59
  60         umask(0022);
  61
  62         ctx = kmod_new(NULL, NULL);
  63         if (!ctx) {
  64                 log_error("Failed to allocate memory for kmod.");
  65                 goto finish;
  66         }
  67
  68         kmod_load_resources(ctx);
  69
  70         kmod_set_log_fn(ctx, systemd_kmod_log, NULL);
  71
  72         if (conf_files_list(&files, ".conf",
  73                             "/etc/modules-load.d",
  74                             "/run/modules-load.d",
  75                             "/usr/local/lib/modules-load.d",
  76                             "/usr/lib/modules-load.d",
  77 #ifdef HAVE_SPLIT_USR
  78                             "/lib/modules-load.d",
  79 #endif
  80                             NULL) < 0) {
  81                 log_error("Failed to enumerate modules-load.d files: %s", strerror(-r));
  82                 goto finish;
  83         }
  84
  85         r = EXIT_SUCCESS;
  86
  87         STRV_FOREACH(fn, files) {
  88                 FILE *f;
  89
  90                 f = fopen(*fn, "re");
  91                 if (!f) {
  92                         if (errno == ENOENT)
  93                                 continue;
  94
  95                         log_error("Failed to open %s: %m", *fn);
  96                         r = EXIT_FAILURE;
  97                         continue;
  98                 }
  99
 100                 log_debug("apply: %s\n", *fn);
 101                 for (;;) {
 102                         char line[LINE_MAX], *l;
 103                         struct kmod_list *itr, *modlist = NULL;
 104                         int err;
 105
 106                         if (!fgets(line, sizeof(line), f))
 107                                 break;
 108
 109                         l = strstrip(line);
 110                         if (*l == '#' || *l == 0)
 111                                 continue;
 112
 113                         err = kmod_module_new_from_lookup(ctx, l, &modlist);
 114                         if (err < 0) {
 115                                 log_error("Failed to lookup alias '%s'", l);
 116                                 r = EXIT_FAILURE;
 117                                 continue;
 118                         }
 119
 120                         kmod_list_foreach(itr, modlist) {
 121                                 struct kmod_module *mod;
 122
 123                                 mod = kmod_module_get_module(itr);
 124                                 err = kmod_module_probe_insert_module(mod, probe_flags,
 125                                                                       NULL, NULL, NULL, NULL);
 126
 127                                 if (err == 0)
 128                                         log_info("Inserted module '%s'", kmod_module_get_name(mod));
 129                                 else if (err == KMOD_PROBE_APPLY_BLACKLIST)
 130                                         log_info("Module '%s' is blacklisted", kmod_module_get_name(mod));
 131                                 else {
 132                                         log_error("Failed to insert '%s': %s", kmod_module_get_name(mod),
 133                                                         strerror(-err));
 134                                         r = EXIT_FAILURE;
 135                                 }
 136
 137                                 kmod_module_unref(mod);
 138                         }
 139
 140                         kmod_module_unref_list(modlist);
 141                 }
 142
 143                 if (ferror(f)) {
 144                         log_error("Failed to read from file: %m");
 145                         r = EXIT_FAILURE;
 146                 }
 147
 148                 fclose(f);
 149         }
 150
 151 finish:
 152         strv_free(files);
 153         kmod_unref(ctx);
 154
 155         return r;
 156 }