gpt-auto-generator.c revision e48fdd84432bbf9c2ecc339183258c7c33116032
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering/***
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering This file is part of systemd.
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering Copyright 2013 Lennart Poettering
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering systemd is free software; you can redistribute it and/or modify it
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering under the terms of the GNU Lesser General Public License as published by
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering the Free Software Foundation; either version 2.1 of the License, or
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering (at your option) any later version.
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering systemd is distributed in the hope that it will be useful, but
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering WITHOUT ANY WARRANTY; without even the implied warranty of
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering Lesser General Public License for more details.
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering You should have received a copy of the GNU Lesser General Public License
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering along with systemd; If not, see <http://www.gnu.org/licenses/>.
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering***/
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek#include <unistd.h>
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek#include <stdlib.h>
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek#include <fcntl.h>
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek#include <sys/ioctl.h>
143bfdaf0b890fa7acadf02d1eafacaef1b696bdHolger Hans Peter Freyther#include <sys/statfs.h>
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek#include <blkid/blkid.h>
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering#ifdef HAVE_LINUX_BTRFS_H
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek#include <linux/btrfs.h>
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering#endif
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering#include "path-util.h"
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek#include "util.h"
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering#include "mkdir.h"
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering#include "missing.h"
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering#include "sd-id128.h"
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering#include "libudev.h"
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering#include "udev-util.h"
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering#include "special.h"
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering#include "unit-name.h"
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering#include "virt.h"
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering#include "generator.h"
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering#include "gpt.h"
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering/* TODO:
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering *
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering * - Properly handle cryptsetup partitions
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering * - Define new partition type for encrypted swap
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering * - Make /home automount rather than mount
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering *
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering */
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poetteringstatic const char *arg_dest = "/tmp";
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart PoetteringDEFINE_TRIVIAL_CLEANUP_FUNC(blkid_probe, blkid_free_probe);
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering#define _cleanup_blkid_freep_probe_ _cleanup_(blkid_free_probep)
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poetteringstatic int verify_gpt_partition(const char *node, sd_id128_t *type, unsigned *nr, char **fstype) {
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering _cleanup_blkid_freep_probe_ blkid_probe b = NULL;
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering const char *v;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering int r;
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering errno = 0;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering b = blkid_new_probe_from_filename(node);
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering if (!b)
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering return errno != 0 ? -errno : -ENOMEM;
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering blkid_probe_enable_superblocks(b, 1);
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering blkid_probe_set_superblocks_flags(b, BLKID_SUBLKS_TYPE);
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering blkid_probe_enable_partitions(b, 1);
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering blkid_probe_set_partitions_flags(b, BLKID_PARTS_ENTRY_DETAILS);
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering errno = 0;
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering r = blkid_do_safeprobe(b);
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering if (r == -2 || r == 1) /* no result or uncertain */
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering return -EBADSLT;
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering else if (r != 0)
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering return errno ? -errno : -EIO;
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering errno = 0;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering r = blkid_probe_lookup_value(b, "PART_ENTRY_SCHEME", &v, NULL);
d7bd3de0654669e65b9642c248c5fa6d1d9a9f61Lennart Poettering if (r != 0)
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering /* return 0 if we're not on GPT */
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering return errno ? -errno : 0;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering if (strcmp(v, "gpt") != 0)
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering return 0;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering if (type) {
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek errno = 0;
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek r = blkid_probe_lookup_value(b, "PART_ENTRY_TYPE", &v, NULL);
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering if (r != 0)
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering return errno ? -errno : -EIO;
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering r = sd_id128_from_string(v, type);
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering if (r < 0)
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering return r;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering }
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering if (nr) {
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering errno = 0;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering r = blkid_probe_lookup_value(b, "PART_ENTRY_NUMBER", &v, NULL);
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering if (r != 0)
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering return errno ? -errno : -EIO;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering r = safe_atou(v, nr);
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering if (r < 0)
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering return r;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering }
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering if (fstype) {
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering errno = 0;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering r = blkid_probe_lookup_value(b, "TYPE", &v, NULL);
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering if (r != 0)
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering *fstype = NULL;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering else {
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering char *fst;
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering fst = strdup(v);
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering if (!fst)
374ec6abf31ada6ca554cc8ea99b282373fac010Lennart Poettering return -ENOMEM;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering *fstype = fst;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering }
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering }
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering return 1;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering}
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
7027ff61a34a12487712b382a061c654acc3a679Lennart Poetteringstatic int add_swap(const char *path, const char *fstype) {
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering _cleanup_free_ char *name = NULL, *unit = NULL, *lnk = NULL;
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering _cleanup_fclose_ FILE *f = NULL;
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering log_debug("Adding swap: %s %s", path, fstype);
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering name = unit_name_from_path(path, ".swap");
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering if (!name)
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering return log_oom();
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering unit = strjoin(arg_dest, "/", name, NULL);
1021b21bc6f8dd522b46116e8598b17f9f93f1b7Lennart Poettering if (!unit)
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering return log_oom();
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering f = fopen(unit, "wxe");
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering if (!f) {
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering log_error("Failed to create unit file %s: %m", unit);
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering return -errno;
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering }
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering fprintf(f,
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering "# Automatically generated by systemd-gpt-auto-generator\n\n"
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering "[Swap]\n"
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering "What=%s\n",
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering path);
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering fflush(f);
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering if (ferror(f)) {
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering log_error("Failed to write unit file %s: %m", unit);
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering return -errno;
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering }
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering lnk = strjoin(arg_dest, "/" SPECIAL_SWAP_TARGET ".wants/", name, NULL);
1021b21bc6f8dd522b46116e8598b17f9f93f1b7Lennart Poettering if (!lnk)
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering return log_oom();
1021b21bc6f8dd522b46116e8598b17f9f93f1b7Lennart Poettering
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering mkdir_parents_label(lnk, 0755);
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering if (symlink(unit, lnk) < 0) {
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering log_error("Failed to create symlink %s: %m", lnk);
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering return -errno;
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering }
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering return 0;
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering}
1021b21bc6f8dd522b46116e8598b17f9f93f1b7Lennart Poettering
1021b21bc6f8dd522b46116e8598b17f9f93f1b7Lennart Poetteringstatic int add_mount(const char *what, const char *where, const char *fstype, const char *description) {
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering _cleanup_free_ char *unit = NULL, *lnk = NULL, *p = NULL;
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering _cleanup_fclose_ FILE *f = NULL;
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering int r;
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering if (dir_is_empty(where) <= 0) {
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering log_debug("%s already populated, ignoring.", where);
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering return 0;
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering }
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering log_debug("Adding %s: %s %s", where, what, fstype);
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering unit = unit_name_from_path(where, ".mount");
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering if (!unit)
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering return log_oom();
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering
a0ab566574303be1ca12cdb334f284cfd407caa5Lennart Poettering p = strjoin(arg_dest, "/", unit, NULL);
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering if (!p)
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering return log_oom();
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering f = fopen(p, "wxe");
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering if (!f) {
a0ab566574303be1ca12cdb334f284cfd407caa5Lennart Poettering log_error("Failed to create unit file %s: %m", unit);
a0ab566574303be1ca12cdb334f284cfd407caa5Lennart Poettering return -errno;
a0ab566574303be1ca12cdb334f284cfd407caa5Lennart Poettering }
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering
ae018d9bc900d6355dea4af05119b49c67945184Lennart Poettering fprintf(f,
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering "# Automatically generated by systemd-gpt-auto-generator\n\n"
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering "[Unit]\n"
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering "Description=%s\n",
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering description);
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering r = generator_write_fsck_deps(f, arg_dest, what, where, fstype);
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering if (r < 0)
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering return r;
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering fprintf(f,
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering "\n"
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering "[Mount]\n"
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering "What=%s\n"
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering "Where=%s\n"
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering "Type=%s\n",
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering what, where, fstype);
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering fflush(f);
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering if (ferror(f)) {
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering log_error("Failed to write unit file %s: %m", unit);
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering return -errno;
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering }
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering lnk = strjoin(arg_dest, "/" SPECIAL_LOCAL_FS_TARGET ".requires/", unit, NULL);
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering if (!lnk)
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering return log_oom();
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering mkdir_parents_label(lnk, 0755);
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering if (symlink(unit, lnk) < 0) {
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering log_error("Failed to create symlink %s: %m", lnk);
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering return -errno;
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering }
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering return 0;
6c03089c32c251d823173bda4d809a9e643219f0Lennart Poettering}
7027ff61a34a12487712b382a061c654acc3a679Lennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poetteringstatic int enumerate_partitions(struct udev *udev, dev_t dev) {
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering _cleanup_udev_enumerate_unref_ struct udev_enumerate *e = NULL;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering _cleanup_udev_device_unref_ struct udev_device *d = NULL;
143bfdaf0b890fa7acadf02d1eafacaef1b696bdHolger Hans Peter Freyther _cleanup_free_ char *home = NULL, *home_fstype = NULL, *srv = NULL, *srv_fstype = NULL;
aff38e74bd776471f15ba54b305a24b0251eb865Lennart Poettering unsigned home_nr = (unsigned) -1, srv_nr = (unsigned )-1;
143bfdaf0b890fa7acadf02d1eafacaef1b696bdHolger Hans Peter Freyther struct udev_list_entry *first, *item;
78edb35ab4f4227485cb9ec816b43c37e0d5e62aLennart Poettering struct udev_device *parent = NULL;
a016b9228f338cb9b380ce7e00826ef462767d98Lennart Poettering int r;
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek e = udev_enumerate_new(udev);
96cde13ace6406582688028f3df5668a172ba628Zbigniew Jędrzejewski-Szmek if (!e)
return log_oom();
d = udev_device_new_from_devnum(udev, 'b', dev);
if (!d)
return log_oom();
parent = udev_device_get_parent(d);
if (!parent)
return log_oom();
r = udev_enumerate_add_match_parent(e, parent);
if (r < 0)
return log_oom();
r = udev_enumerate_add_match_subsystem(e, "block");
if (r < 0)
return log_oom();
r = udev_enumerate_scan_devices(e);
if (r < 0) {
log_error("Failed to enumerate partitions on /dev/block/%u:%u: %s", major(dev), minor(dev), strerror(-r));
return r;
}
first = udev_enumerate_get_list_entry(e);
udev_list_entry_foreach(item, first) {
_cleanup_udev_device_unref_ struct udev_device *q;
_cleanup_free_ char *fstype = NULL;
const char *node = NULL;
sd_id128_t type_id;
unsigned nr;
q = udev_device_new_from_syspath(udev, udev_list_entry_get_name(item));
if (!q)
return log_oom();
if (udev_device_get_devnum(q) == udev_device_get_devnum(d))
continue;
if (udev_device_get_devnum(q) == udev_device_get_devnum(parent))
continue;
node = udev_device_get_devnode(q);
if (!node)
return log_oom();
r = verify_gpt_partition(node, &type_id, &nr, &fstype);
if (r < 0) {
/* skip child devices which are not detected properly */
if (r == -EBADSLT)
continue;
log_error("Failed to verify GPT partition %s: %s", node, strerror(-r));
return r;
}
if (r == 0)
continue;
if (sd_id128_equal(type_id, GPT_SWAP))
add_swap(node, fstype);
else if (sd_id128_equal(type_id, GPT_HOME)) {
/* We only care for the first /home partition */
if (home && nr >= home_nr)
continue;
home_nr = nr;
free(home);
home = strdup(node);
if (!home)
return log_oom();
free(home_fstype);
home_fstype = fstype;
fstype = NULL;
} else if (sd_id128_equal(type_id, GPT_SRV)) {
/* We only care for the first /srv partition */
if (srv && nr >= srv_nr)
continue;
srv_nr = nr;
free(srv);
srv = strdup(node);
if (!srv)
return log_oom();
free(srv_fstype);
srv_fstype = fstype;
fstype = NULL;
}
}
if (home && home_fstype)
add_mount(home, "/home", home_fstype, "Home Partition");
if (srv && srv_fstype)
add_mount(srv, "/srv", srv_fstype, "Server Data Partition");
return r;
}
static int get_btrfs_block_device(const char *path, dev_t *dev) {
struct btrfs_ioctl_fs_info_args fsi = {};
_cleanup_close_ int fd = -1;
uint64_t id;
assert(path);
assert(dev);
fd = open(path, O_DIRECTORY|O_CLOEXEC);
if (fd < 0)
return -errno;
if (ioctl(fd, BTRFS_IOC_FS_INFO, &fsi) < 0)
return -errno;
/* We won't do this for btrfs RAID */
if (fsi.num_devices != 1)
return 0;
for (id = 1; id <= fsi.max_id; id++) {
struct btrfs_ioctl_dev_info_args di = {
.devid = id,
};
struct stat st;
if (ioctl(fd, BTRFS_IOC_DEV_INFO, &di) < 0) {
if (errno == ENODEV)
continue;
return -errno;
}
if (stat((char*) di.path, &st) < 0)
return -errno;
if (!S_ISBLK(st.st_mode))
return -ENODEV;
if (major(st.st_rdev) == 0)
return -ENODEV;
*dev = st.st_rdev;
return 1;
}
return -ENODEV;
}
static int get_block_device(const char *path, dev_t *dev) {
struct stat st;
struct statfs sfs;
assert(path);
assert(dev);
if (lstat(path, &st))
return -errno;
if (major(st.st_dev) != 0) {
*dev = st.st_dev;
return 1;
}
if (statfs(path, &sfs) < 0)
return -errno;
if (F_TYPE_EQUAL(sfs.f_type, BTRFS_SUPER_MAGIC))
return get_btrfs_block_device(path, dev);
return 0;
}
static int devno_to_devnode(struct udev *udev, dev_t devno, char **ret) {
_cleanup_udev_device_unref_ struct udev_device *d;
const char *t;
char *n;
d = udev_device_new_from_devnum(udev, 'b', devno);
if (!d)
return log_oom();
t = udev_device_get_devnode(d);
if (!t)
return -ENODEV;
n = strdup(t);
if (!n)
return -ENOMEM;
*ret = n;
return 0;
}
int main(int argc, char *argv[]) {
_cleanup_udev_unref_ struct udev *udev = NULL;
_cleanup_free_ char *node = NULL;
dev_t devno;
int r = 0;
if (argc > 1 && argc != 4) {
log_error("This program takes three or no arguments.");
return EXIT_FAILURE;
}
if (argc > 1)
arg_dest = argv[3];
log_set_target(LOG_TARGET_SAFE);
log_parse_environment();
log_open();
umask(0022);
if (in_initrd()) {
log_debug("In initrd, exiting.");
return EXIT_SUCCESS;
}
if (detect_container(NULL) > 0) {
log_debug("In a container, exiting.");
return EXIT_SUCCESS;
}
r = get_block_device("/", &devno);
if (r < 0) {
log_error("Failed to determine block device of root file system: %s", strerror(-r));
return EXIT_FAILURE;
} else if (r == 0) {
log_debug("Root file system not on a (single) block device.");
return EXIT_SUCCESS;
}
udev = udev_new();
if (!udev) {
log_oom();
return EXIT_FAILURE;
}
r = devno_to_devnode(udev, devno, &node);
if (r < 0) {
log_error("Failed to determine block device node from major/minor: %s", strerror(-r));
return EXIT_FAILURE;
}
log_debug("Root device %s.", node);
r = verify_gpt_partition(node, NULL, NULL, NULL);
if (r < 0) {
log_error("Failed to verify GPT partition %s: %s", node, strerror(-r));
return EXIT_FAILURE;
}
if (r == 0) {
log_debug("Not a GPT disk, exiting.");
return EXIT_SUCCESS;
}
r = enumerate_partitions(udev, devno);
if (r < 0)
return EXIT_FAILURE;
return EXIT_SUCCESS;
}