chiark / gitweb /
cryptsetup-generator: add a missing OOM check
[elogind.git] / src / cryptsetup / cryptsetup-generator.c
index ba59b49b014380ccb107fe885e69418981c636bd..5c4a2807a74059af059971f1874cd88fecb5cd74 100644 (file)
@@ -6,16 +6,16 @@
   Copyright 2010 Lennart Poettering
 
   systemd is free software; you can redistribute it and/or modify it
-  under the terms of the GNU General Public License as published by
-  the Free Software Foundation; either version 2 of the License, or
+  under the terms of the GNU Lesser General Public License as published by
+  the Free Software Foundation; either version 2.1 of the License, or
   (at your option) any later version.
 
   systemd is distributed in the hope that it will be useful, but
   WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
-  General Public License for more details.
+  Lesser General Public License for more details.
 
-  You should have received a copy of the GNU General Public License
+  You should have received a copy of the GNU Lesser General Public License
   along with systemd; If not, see <http://www.gnu.org/licenses/>.
 ***/
 
 #include "log.h"
 #include "util.h"
 #include "unit-name.h"
+#include "mkdir.h"
+#include "virt.h"
+#include "strv.h"
+#include "fileio.h"
 
-const char *arg_dest = "/tmp";
+static const char *arg_dest = "/tmp";
+static bool arg_enabled = true;
+static bool arg_read_crypttab = true;
 
 static bool has_option(const char *haystack, const char *needle) {
         const char *f = haystack;
@@ -64,9 +70,8 @@ static int create_disk(
                 const char *password,
                 const char *options) {
 
-        char *p = NULL, *n = NULL, *d = NULL, *u = NULL, *from = NULL, *to = NULL, *e = NULL;
-        int r;
-        FILE *f = NULL;
+        char _cleanup_free_ *p = NULL, *n = NULL, *d = NULL, *u = NULL, *from = NULL, *to = NULL, *e = NULL;
+        FILE _cleanup_fclose_ *f = NULL;
         bool noauto, nofail;
 
         assert(name);
@@ -75,42 +80,37 @@ static int create_disk(
         noauto = has_option(options, "noauto");
         nofail = has_option(options, "nofail");
 
-        if (!(n = unit_name_build_escape("cryptsetup", name, ".service"))) {
-                r = -ENOMEM;
-                log_error("Failed to allocate unit name.");
-                goto fail;
-        }
+        n = unit_name_from_path_instance("systemd-cryptsetup", name, ".service");
+        if (!n)
+                return log_oom();
 
-        if (asprintf(&p, "%s/%s", arg_dest, n) < 0) {
-                r = -ENOMEM;
-                log_error("Failed to allocate unit file name.");
-                goto fail;
-        }
+        p = strjoin(arg_dest, "/", n, NULL);
+        if (!p)
+                return log_oom();
 
-        if (!(u = fstab_node_to_udev_node(device))) {
-                r = -ENOMEM;
-                log_error("Failed to allocate device node.");
-                goto fail;
-        }
+        u = fstab_node_to_udev_node(device);
+        if (!u)
+                return log_oom();
 
-        if (!(d = unit_name_from_path(u, ".device"))) {
-                r = -ENOMEM;
-                log_error("Failed to allocate device name.");
-                goto fail;
-        }
+        d = unit_name_from_path(u, ".device");
+        if (!d)
+                return log_oom();
 
-        if (!(f = fopen(p, "wxe"))) {
-                r = -errno;
-                log_error("Failed to create unit file: %m");
-                goto fail;
+        f = fopen(p, "wxe");
+        if (!f) {
+                log_error("Failed to create unit file %s: %m", p);
+                return -errno;
         }
 
         fprintf(f,
+                "# Automatically generated by systemd-cryptsetup-generator\n\n"
                 "[Unit]\n"
                 "Description=Cryptography Setup for %%I\n"
+                "Documentation=man:systemd-cryptsetup@.service(8) man:crypttab(5)\n"
+                "SourcePath=/etc/crypttab\n"
                 "Conflicts=umount.target\n"
                 "DefaultDependencies=no\n"
-                "BindTo=%s dev-mapper-%%i.device\n"
+                "BindsTo=%s dev-mapper-%%i.device\n"
                 "After=systemd-readahead-collect.service systemd-readahead-replay.service %s\n"
                 "Before=umount.target\n",
                 d, d);
@@ -122,11 +122,9 @@ static int create_disk(
         if (password && (streq(password, "/dev/urandom") ||
                          streq(password, "/dev/random") ||
                          streq(password, "/dev/hw_random")))
-                fprintf(f,
-                        "After=systemd-random-seed-load.service\n");
+                fputs("After=systemd-random-seed-load.service\n", f);
         else
-                fprintf(f,
-                        "Before=local-fs.target\n");
+                fputs("Before=local-fs.target\n", f);
 
         fprintf(f,
                 "\n[Service]\n"
@@ -151,148 +149,271 @@ static int create_disk(
         fflush(f);
 
         if (ferror(f)) {
-                r = -errno;
-                log_error("Failed to write file: %m");
-                goto fail;
+                log_error("Failed to write file %s: %m", p);
+                return -errno;
         }
 
-        if (asprintf(&from, "../%s", n) < 0) {
-                r = -ENOMEM;
-                goto fail;
-        }
+        if (asprintf(&from, "../%s", n) < 0)
+                return log_oom();
 
         if (!noauto) {
 
-                if (asprintf(&to, "%s/%s.wants/%s", arg_dest, d, n) < 0) {
-                        r = -ENOMEM;
-                        goto fail;
-                }
-
-                mkdir_parents(to, 0755);
+                to = strjoin(arg_dest, "/", d, ".wants/", n, NULL);
+                if (!to)
+                        return log_oom();
 
+                mkdir_parents_label(to, 0755);
                 if (symlink(from, to) < 0) {
                         log_error("Failed to create symlink '%s' to '%s': %m", from, to);
-                        r = -errno;
-                        goto fail;
+                        return -errno;
                 }
 
                 free(to);
-                to = NULL;
-
                 if (!nofail)
-                        asprintf(&to, "%s/cryptsetup.target.requires/%s", arg_dest, n);
+                        to = strjoin(arg_dest, "/cryptsetup.target.requires/", n, NULL);
                 else
-                        asprintf(&to, "%s/cryptsetup.target.wants/%s", arg_dest, n);
-
-                if (!to) {
-                        r = -ENOMEM;
-                        goto fail;
-                }
-
-                mkdir_parents(to, 0755);
+                        to = strjoin(arg_dest, "/cryptsetup.target.wants/", n, NULL);
+                if (!to)
+                        return log_oom();
 
+                mkdir_parents_label(to, 0755);
                 if (symlink(from, to) < 0) {
                         log_error("Failed to create symlink '%s' to '%s': %m", from, to);
-                        r = -errno;
-                        goto fail;
+                        return -errno;
                 }
         }
 
-        free(to);
-        to = NULL;
-
         e = unit_name_escape(name);
-        if (asprintf(&to, "%s/dev-mapper-%s.device.requires/%s", arg_dest, e, n) < 0) {
-                r = -ENOMEM;
-                goto fail;
-        }
+        if (!e)
+                return log_oom();
 
-        mkdir_parents(to, 0755);
+        free(to);
+        to = strjoin(arg_dest, "/dev-mapper-", e, ".device.requires/", n, NULL);
+        if (!to)
+                return log_oom();
 
+        mkdir_parents_label(to, 0755);
         if (symlink(from, to) < 0) {
                 log_error("Failed to create symlink '%s' to '%s': %m", from, to);
-                r = -errno;
-                goto fail;
+                return -errno;
         }
 
-        r = 0;
+        return 0;
+}
 
-fail:
-        free(p);
-        free(n);
-        free(d);
-        free(e);
+static int parse_proc_cmdline(char ***arg_proc_cmdline_disks) {
+        char _cleanup_free_ *line = NULL;
+        char *w = NULL, *state = NULL;
+        int r;
+        size_t l;
 
-        free(from);
-        free(to);
+        if (detect_container(NULL) > 0)
+                return 0;
 
-        if (f)
-                fclose(f);
+        r = read_one_line_file("/proc/cmdline", &line);
+        if (r < 0) {
+                log_warning("Failed to read /proc/cmdline, ignoring: %s", strerror(-r));
+                return 0;
+        }
 
-        return r;
+        FOREACH_WORD_QUOTED(w, l, line, state) {
+                char _cleanup_free_ *word = NULL;
+
+                word = strndup(w, l);
+                if (!word)
+                        return log_oom();
+
+                if (startswith(word, "luks=")) {
+                        r = parse_boolean(word + 5);
+                        if (r < 0)
+                                log_warning("Failed to parse luks switch %s. Ignoring.", word + 5);
+                        else
+                                arg_enabled = r;
+
+                } else if (startswith(word, "rd.luks=")) {
+
+                        if (in_initrd()) {
+                                r = parse_boolean(word + 8);
+                                if (r < 0)
+                                        log_warning("Failed to parse luks switch %s. Ignoring.", word + 8);
+                                else
+                                        arg_enabled = r;
+                        }
+
+                } else if (startswith(word, "luks.crypttab=")) {
+                        r = parse_boolean(word + 14);
+                        if (r < 0)
+                                log_warning("Failed to parse luks crypttab switch %s. Ignoring.", word + 14);
+                        else
+                                arg_read_crypttab = r;
+
+                } else if (startswith(word, "rd.luks.crypttab=")) {
+
+                        if (in_initrd()) {
+                                r = parse_boolean(word + 17);
+                                if (r < 0)
+                                        log_warning("Failed to parse luks crypttab switch %s. Ignoring.", word + 17);
+                                else
+                                        arg_read_crypttab = r;
+                        }
+
+                } else if (startswith(word, "luks.uuid=")) {
+                        char **t;
+
+                        t = strv_append(*arg_proc_cmdline_disks, word + 10);
+                        if (!t)
+                                return log_oom();
+
+                        strv_free(*arg_proc_cmdline_disks);
+                        *arg_proc_cmdline_disks = t;
+
+                } else if (startswith(word, "rd.luks.uuid=")) {
+
+                        if (in_initrd()) {
+                                char **t;
+
+                                t = strv_append(*arg_proc_cmdline_disks, word + 13);
+                                if (!t)
+                                        return log_oom();
+
+                                strv_free(*arg_proc_cmdline_disks);
+                                *arg_proc_cmdline_disks = t;
+                        }
+
+                } else if (startswith(word, "luks.") ||
+                           (in_initrd() && startswith(word, "rd.luks."))) {
+
+                        log_warning("Unknown kernel switch %s. Ignoring.", word);
+                }
+        }
+
+        strv_uniq(*arg_proc_cmdline_disks);
+
+        return 0;
 }
 
 int main(int argc, char *argv[]) {
-        FILE *f;
-        int r = EXIT_SUCCESS;
+        FILE _cleanup_fclose_ *f = NULL;
         unsigned n = 0;
+        int r = EXIT_SUCCESS;
+        char **i;
+        char _cleanup_strv_free_ **arg_proc_cmdline_disks_done = NULL;
+        char _cleanup_strv_free_ **arg_proc_cmdline_disks = NULL;
 
-        if (argc > 2) {
-                log_error("This program takes one or no arguments.");
+        if (argc > 1 && argc != 4) {
+                log_error("This program takes three or no arguments.");
                 return EXIT_FAILURE;
         }
 
         if (argc > 1)
                 arg_dest = argv[1];
 
-        log_set_target(LOG_TARGET_AUTO);
+        log_set_target(LOG_TARGET_SAFE);
         log_parse_environment();
         log_open();
 
         umask(0022);
 
-        if (!(f = fopen("/etc/crypttab", "re"))) {
+        if (parse_proc_cmdline(&arg_proc_cmdline_disks) < 0)
+                return EXIT_FAILURE;
 
-                if (errno == ENOENT)
-                        r = EXIT_SUCCESS;
-                else {
-                        r = EXIT_FAILURE;
-                        log_error("Failed to open /etc/crypttab: %m");
+        if (!arg_enabled)
+                return EXIT_SUCCESS;
+
+        if (arg_read_crypttab) {
+                f = fopen("/etc/crypttab", "re");
+
+                if (!f) {
+                        if (errno == ENOENT)
+                                r = EXIT_SUCCESS;
+                        else {
+                                r = EXIT_FAILURE;
+                                log_error("Failed to open /etc/crypttab: %m");
+                        }
+                } else for (;;) {
+                        char line[LINE_MAX], *l;
+                        char _cleanup_free_ *name = NULL, *device = NULL, *password = NULL, *options = NULL;
+                        int k;
+
+                        if (!fgets(line, sizeof(line), f))
+                                break;
+
+                        n++;
+
+                        l = strstrip(line);
+                        if (*l == '#' || *l == 0)
+                                continue;
+
+                        k = sscanf(l, "%ms %ms %ms %ms", &name, &device, &password, &options);
+                        if (k < 2 || k > 4) {
+                                log_error("Failed to parse /etc/crypttab:%u, ignoring.", n);
+                                r = EXIT_FAILURE;
+                                continue;
+                        }
+
+                        if (arg_proc_cmdline_disks) {
+                                /*
+                                  If luks UUIDs are specified on the kernel command line, use them as a filter
+                                  for /etc/crypttab and only generate units for those.
+                                */
+                                STRV_FOREACH(i, arg_proc_cmdline_disks) {
+                                        char _cleanup_free_ *proc_device = NULL, *proc_name = NULL;
+                                        const char *p = *i;
+
+                                        if (startswith(p, "luks-"))
+                                                p += 5;
+
+                                        proc_name = strappend("luks-", p);
+                                        proc_device = strappend("UUID=", p);
+
+                                        if (!proc_name || !proc_device)
+                                                return log_oom();
+
+                                        if (streq(proc_device, device) || streq(proc_name, name)) {
+                                                char **t;
+
+                                                if (create_disk(name, device, password, options) < 0)
+                                                        r = EXIT_FAILURE;
+
+                                                t = strv_append(arg_proc_cmdline_disks_done, p);
+                                                if (!t)
+                                                        return log_oom();
+
+                                                strv_free(arg_proc_cmdline_disks_done);
+                                                arg_proc_cmdline_disks_done = t;
+                                        }
+                                }
+                        } else {
+                                if (create_disk(name, device, password, options) < 0)
+                                        r = EXIT_FAILURE;
+                        }
                 }
-
-                goto finish;
         }
 
-        for (;;) {
-                char line[LINE_MAX], *l;
-                char *name = NULL, *device = NULL, *password = NULL, *options = NULL;
-                int k;
+        STRV_FOREACH(i, arg_proc_cmdline_disks) {
+                /*
+                  Generate units for those UUIDs, which were specified
+                  on the kernel command line and not yet written.
+                */
 
-                if (!(fgets(line, sizeof(line), f)))
-                        break;
+                char _cleanup_free_ *name = NULL, *device = NULL;
+                const char *p = *i;
 
-                n++;
+                if (startswith(p, "luks-"))
+                        p += 5;
 
-                l = strstrip(line);
-                if (*l == '#' || *l == 0)
+                if (strv_contains(arg_proc_cmdline_disks_done, p))
                         continue;
 
-                if ((k = sscanf(l, "%ms %ms %ms %ms", &name, &device, &password, &options)) < 2 || k > 4) {
-                        log_error("Failed to parse /etc/crypttab:%u, ignoring.", n);
-                        r = EXIT_FAILURE;
-                        goto next;
-                }
+                name = strappend("luks-", p);
+                device = strappend("UUID=", p);
 
-                if (create_disk(name, device, password, options) < 0)
-                        r = EXIT_FAILURE;
+                if (!name || !device)
+                        return log_oom();
 
-        next:
-                free(name);
-                free(device);
-                free(password);
-                free(options);
+                if (create_disk(name, device, NULL, "timeout=0") < 0)
+                        r = EXIT_FAILURE;
         }
 
-finish:
         return r;
 }