Zpool: add NetBSD support

Signed-off-by: hoshinomori <hoshinomori@owarisekai.moe>

Author: uzuki314

CMakeLists.txt

@@ -92,7 +92,7 @@ cmake_dependent_option(ENABLE_DDCUTIL "Enable ddcutil" ON "LINUX" OFF)
 cmake_dependent_option(ENABLE_DIRECTX_HEADERS "Enable DirectX headers for WSL" ON "LINUX" OFF)
 cmake_dependent_option(ENABLE_ELF "Enable libelf" ON "LINUX OR ANDROID OR DragonFly OR Haiku OR GNU" OFF)
 cmake_dependent_option(ENABLE_THREADS "Enable multithreading" ON "Threads_FOUND" OFF)
-cmake_dependent_option(ENABLE_LIBZFS "Enable libzfs" ON "LINUX OR FreeBSD OR SunOS" OFF)
+cmake_dependent_option(ENABLE_LIBZFS "Enable libzfs" ON "LINUX OR FreeBSD OR NetBSD OR SunOS" OFF)
 cmake_dependent_option(ENABLE_PCIACCESS "Enable libpciaccess" ON "GNU" OFF)
 
 option(ENABLE_ZLIB "Enable zlib" ON)
@@ -1726,6 +1726,7 @@ elseif(NetBSD)
         PRIVATE "bluetooth"
         PRIVATE "m"
         PRIVATE "prop"
+        PRIVATE $<$<BOOL:${ENABLE_LIBZFS}>:zfs;nvpair>
     )
 elseif(SunOS)
     target_link_libraries(libfastfetch

src/detection/zpool/libzfs_simplified.h

@@ -2,8 +2,6 @@
 
 #include "fastfetch.h"
 
-// From https://github.com/openzfs/zfs/blob/master/include/libzfs.h
-
 /*
  * CDDL HEADER START
  *
@@ -25,6 +23,18 @@
  * CDDL HEADER END
  */
 
+// From https://github.com/NetBSD/src/blob/trunk/external/cddl/osnet/sys/sys/opentypes.h
+
+#ifndef __NetBSD__
+typedef long long longlong_t;
+#endif
+typedef longlong_t hrtime_t;
+typedef unsigned int uint_t;
+
+
+// From https://github.com/openzfs/zfs/blob/master/include/libzfs.h
+// From https://github.com/NetBSD/src/blob/trunk/external/cddl/osnet/dist/uts/common/sys/fs/zfs.h
+
 typedef enum {
     ZPOOL_PROP_INVAL = -1,
     ZPOOL_PROP_NAME,
@@ -46,13 +56,16 @@ typedef enum {
     ZPOOL_PROP_FREE,
     ZPOOL_PROP_ALLOCATED,
     ZPOOL_PROP_READONLY,
+#ifndef __NetBSD__
     ZPOOL_PROP_ASHIFT,
+#endif
     ZPOOL_PROP_COMMENT,
     ZPOOL_PROP_EXPANDSZ,
     ZPOOL_PROP_FREEING,
     ZPOOL_PROP_FRAGMENTATION,
     ZPOOL_PROP_LEAKED,
     ZPOOL_PROP_MAXBLOCKSIZE,
+#ifndef __NetBSD__
     ZPOOL_PROP_TNAME,
     ZPOOL_PROP_MAXDNODESIZE,
     ZPOOL_PROP_MULTIHOST,
@@ -63,6 +76,7 @@ typedef enum {
     ZPOOL_PROP_BCLONEUSED,
     ZPOOL_PROP_BCLONESAVED,
     ZPOOL_PROP_BCLONERATIO,
+#endif
     ZPOOL_NUM_PROPS
 } zpool_prop_t;
 
@@ -75,15 +89,158 @@ typedef enum {
     ZPROP_SRC_RECEIVED = 0x20
 } zprop_source_t;
 
-typedef bool boolean_t;
+#ifdef __NetBSD__
+
+#define	ZPOOL_CONFIG_VDEV_TREE "vdev_tree"
+#define ZPOOL_CONFIG_VDEV_STATS "vdev_stats"
+
+typedef enum vdev_state {
+    VDEV_STATE_UNKNOWN = 0,
+    VDEV_STATE_CLOSED,
+    VDEV_STATE_OFFLINE,
+    VDEV_STATE_REMOVED,
+    VDEV_STATE_CANT_OPEN,
+    VDEV_STATE_FAULTED,
+    VDEV_STATE_DEGRADED,
+    VDEV_STATE_HEALTHY
+} vdev_state_t;
+
+typedef enum vdev_aux {
+    VDEV_AUX_NONE,
+    VDEV_AUX_OPEN_FAILED,
+    VDEV_AUX_CORRUPT_DATA,
+    VDEV_AUX_NO_REPLICAS,
+    VDEV_AUX_BAD_GUID_SUM,
+    VDEV_AUX_TOO_SMALL,
+    VDEV_AUX_BAD_LABEL,
+    VDEV_AUX_VERSION_NEWER,
+    VDEV_AUX_VERSION_OLDER,
+    VDEV_AUX_UNSUP_FEAT,
+    VDEV_AUX_SPARED,
+    VDEV_AUX_ERR_EXCEEDED,
+    VDEV_AUX_IO_FAILURE,
+    VDEV_AUX_BAD_LOG,
+    VDEV_AUX_EXTERNAL,
+    VDEV_AUX_SPLIT_POOL,
+    VDEV_AUX_ASHIFT_TOO_BIG
+} vdev_aux_t;
+
+typedef enum pool_state {
+    POOL_STATE_ACTIVE = 0,
+    POOL_STATE_EXPORTED,
+    POOL_STATE_DESTROYED,
+    POOL_STATE_SPARE,
+    POOL_STATE_L2CACHE,
+    POOL_STATE_UNINITIALIZED,
+    POOL_STATE_UNAVAIL,
+    POOL_STATE_POTENTIALLY_ACTIVE
+} pool_state_t;
+
+typedef enum zio_type {
+    ZIO_TYPE_NULL = 0,
+    ZIO_TYPE_READ,
+    ZIO_TYPE_WRITE,
+    ZIO_TYPE_FREE,
+    ZIO_TYPE_CLAIM,
+    ZIO_TYPE_IOCTL,
+    ZIO_TYPES
+} zio_type_t;
+
+typedef struct vdev_stat {
+    hrtime_t vs_timestamp;
+    uint64_t vs_state;
+    uint64_t vs_aux;
+    uint64_t vs_alloc;
+    uint64_t vs_space;
+    uint64_t vs_dspace;
+    uint64_t vs_rsize;
+    uint64_t vs_esize;
+    uint64_t vs_ops[ZIO_TYPES];
+    uint64_t vs_bytes[ZIO_TYPES];
+    uint64_t vs_read_errors;
+    uint64_t vs_write_errors;
+    uint64_t vs_checksum_errors;
+    uint64_t vs_self_healed;
+    uint64_t vs_scan_removing;
+    uint64_t vs_scan_processed;
+    uint64_t vs_configured_ashift;
+    uint64_t vs_logical_ashift;
+    uint64_t vs_physical_ashift;
+    uint64_t vs_fragmentation;
+} vdev_stat_t;
+
+// From https://github.com/NetBSD/src/blob/trunk/external/cddl/osnet/uts/common/sys/nvpair.h
+
+typedef struct nvlist {
+    int32_t nvl_version;
+    uint32_t nvl_nvflag;
+    uint64_t nvl_priv;
+    uint32_t nvl_flag;
+    int32_t nvl_pad;
+} nvlist_t;
+
+// From https://github.com/NetBSD/src/blob/trunk/external/cddl/osnet/dist/lib/libzfs/common/libzfs.h
+
+typedef enum {
+    ZPOOL_STATUS_CORRUPT_CACHE,
+    ZPOOL_STATUS_MISSING_DEV_R,
+    ZPOOL_STATUS_MISSING_DEV_NR,
+    ZPOOL_STATUS_CORRUPT_LABEL_R,
+    ZPOOL_STATUS_CORRUPT_LABEL_NR,
+    ZPOOL_STATUS_BAD_GUID_SUM,
+    ZPOOL_STATUS_CORRUPT_POOL,
+    ZPOOL_STATUS_CORRUPT_DATA,
+    ZPOOL_STATUS_FAILING_DEV,
+    ZPOOL_STATUS_VERSION_NEWER,
+    ZPOOL_STATUS_HOSTID_MISMATCH,
+    ZPOOL_STATUS_IO_FAILURE_WAIT,
+    ZPOOL_STATUS_IO_FAILURE_CONTINUE,
+    ZPOOL_STATUS_BAD_LOG,
+    ZPOOL_STATUS_UNSUP_FEAT_READ,
+    ZPOOL_STATUS_UNSUP_FEAT_WRITE,
+    ZPOOL_STATUS_FAULTED_DEV_R,
+    ZPOOL_STATUS_FAULTED_DEV_NR,
+    ZPOOL_STATUS_VERSION_OLDER,
+    ZPOOL_STATUS_FEAT_DISABLED,
+    ZPOOL_STATUS_RESILVERING,
+    ZPOOL_STATUS_OFFLINE_DEV,
+    ZPOOL_STATUS_REMOVED_DEV,
+    ZPOOL_STATUS_NON_NATIVE_ASHIFT,
+    ZPOOL_STATUS_OK
+} zpool_status_t;
+#endif /*__NetBSD__*/
 
-typedef struct libzfs_handle libzfs_handle_t;
 typedef struct zpool_handle zpool_handle_t;
+typedef struct libzfs_handle libzfs_handle_t;
 typedef int (*zpool_iter_f)(zpool_handle_t *, void *);
-extern int zpool_iter(libzfs_handle_t *, zpool_iter_f, void *);
 
+extern int zpool_iter(libzfs_handle_t *, zpool_iter_f, void *);
 extern libzfs_handle_t *libzfs_init(void);
 extern void libzfs_fini(libzfs_handle_t *);
 extern uint64_t zpool_get_prop_int(zpool_handle_t *, zpool_prop_t, zprop_source_t *);
 extern const char *zpool_get_name(zpool_handle_t *);
 extern const char *zpool_get_state_str(zpool_handle_t *);
+
+#ifdef __NetBSD__
+extern int zpool_get_state(zpool_handle_t *);
+extern const char *zpool_state_to_name(vdev_state_t, vdev_aux_t);
+extern zpool_status_t zpool_get_status(zpool_handle_t *, char **);
+extern nvlist_t *zpool_get_config(zpool_handle_t *, nvlist_t **);
+
+// From https://github.com/NetBSD/src/blob/trunk/external/cddl/osnet/dist/uts/common/sys/nvpair.h
+
+extern nvlist_t *opensolaris_fnvlist_lookup_nvlist(nvlist_t *, const char *);
+static inline nvlist_t *fnvlist_lookup_nvlist(nvlist_t *nvl, const char *name) {
+    return opensolaris_fnvlist_lookup_nvlist(nvl, name);
+}
+
+extern int opensolaris_nvlist_lookup_uint64_array(nvlist_t *, const char *, uint64_t **, uint_t *);
+static inline int nvlist_lookup_uint64_array(nvlist_t *nvl, const char *name, uint64_t **a, uint_t *n) {
+    return opensolaris_nvlist_lookup_uint64_array(nvl, name, a, n);
+}
+
+extern int opensolaris_nvlist_lookup_uint32_array(nvlist_t *, const char *, uint32_t **, uint_t *);
+static inline int nvlist_lookup_uint32_array(nvlist_t *nvl, const char *name, uint32_t **a, uint_t *n) {
+    return opensolaris_nvlist_lookup_uint32_array(nvl, name, a, n);
+}
+#endif /*__NetBSD__*/

src/detection/zpool/zpool_linux.c

@@ -4,21 +4,30 @@
 #include "util/kmod.h"
 
 #ifdef __sun
-#define FF_DISABLE_DLOPEN
 #include <libzfs.h>
+#else
+#include "libzfs_simplified.h"
+#endif
 
+#if defined(__sun) || defined(__NetBSD__)
+#define FF_DISABLE_DLOPEN
 const char* zpool_get_state_str(zpool_handle_t* zpool)
 {
     if (zpool_get_state(zpool) == POOL_STATE_UNAVAIL)
         return "FAULTED";
     else
     {
         const char *str;
+        #ifdef __sun
         zpool_errata_t errata;
         zpool_status_t status = zpool_get_status(zpool, (char**) &str, &errata);
-        if (status == ZPOOL_STATUS_IO_FAILURE_WAIT ||
-            status == ZPOOL_STATUS_IO_FAILURE_CONTINUE ||
-            status == ZPOOL_STATUS_IO_FAILURE_MMP)
+        if (status == ZPOOL_STATUS_IO_FAILURE_MMP ||
+        #else
+        zpool_status_t status = zpool_get_status(zpool, (char**) &str);
+        if (
+        #endif
+            status == ZPOOL_STATUS_IO_FAILURE_WAIT ||
+            status == ZPOOL_STATUS_IO_FAILURE_CONTINUE)
             return "SUSPENDED";
         else
         {
@@ -36,9 +45,7 @@ const char* zpool_get_state_str(zpool_handle_t* zpool)
         }
     }
 }
-#else
-#include "libzfs_simplified.h"
-#endif
+#endif /* __sun || __NetBSD__ */
 
 #include "common/library.h"