Enter DiskMan

1 files changed, 81 insertions, 0 deletions

diff --git a/diskman/diskman.cc b/diskman/diskman.cc
new file mode 100644
index 0000000..32a2c40
--- /dev/null
+++ b/diskman/diskman.cc
@@ -0,0 +1,81 @@
+/*
+ * diskman.cc - Common routines for the DiskMan library
+ * diskman, the Disk Manipulation library for
+ * Project Horizon
+ *
+ * Copyright (c) 2020 Adélie Linux and contributors.  All rights reserved.
+ * This code is licensed under the AGPL 3.0 license, as noted in the
+ * LICENSE-code file in the root directory of this repository.
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+#include "diskman.hh"
+
+#include <iostream>
+#include <libudev.h>
+
+namespace Horizon {
+namespace DiskMan {
+
+struct DiskMan::impl {
+    struct udev *udev;
+
+    impl() {
+        udev = udev_new();
+    }
+    ~impl() {
+        udev_unref(udev);
+    }
+    impl(const impl &other) {
+        this->udev = udev_ref(other.udev);
+    }
+};
+
+DiskMan::DiskMan() : pImpl {std::make_unique<impl>()} {
+}
+
+DiskMan::~DiskMan() = default;
+
+std::vector<Disk> DiskMan::find_disks(bool include_part, bool include_vg,
+                                      bool include_lvm) {
+    struct udev_enumerate *disk_enum = udev_enumerate_new(pImpl->udev);
+    struct udev_list_entry *first, *item;
+    struct udev_device *device = nullptr;
+    std::vector<Disk> disks;
+
+    if(disk_enum == nullptr) {
+        std::cerr << "Couldn't connect to udev" << std::endl;
+        return {};
+    }
+    udev_enumerate_add_match_subsystem(disk_enum, "block");
+    udev_enumerate_add_match_property(disk_enum, "DEVTYPE", "disk");
+    udev_enumerate_scan_devices(disk_enum);
+    first = udev_enumerate_get_list_entry(disk_enum);
+    if(first == NULL) {
+        std::cerr << "No block devices found" << std::endl;
+        return {};
+    }
+
+    udev_list_entry_foreach(item, first) {
+        const char *path = udev_list_entry_get_name(item);
+        if(device != NULL) udev_device_unref(device);
+        device = udev_device_new_from_syspath(pImpl->udev, path);
+        std::string name(udev_device_get_sysname(device));
+        if(name.compare(0, 4, "loop") == 0
+                || name.compare(0, 3, "ram") == 0) {
+            /* Don't include loop or ram devices */
+            continue;
+        }
+        if(!include_lvm && name.compare(0, 3, "dm-") == 0) {
+            /* Skip LVM volumes if requested. */
+            continue;
+        }
+        disks.push_back(Disk(device, include_part));
+    }
+
+    return disks;
+}
+
+}
+}