1#include "zpool.h"
  2
  3#if FF_HAVE_LIBZFS
  4
  5    #include "common/kmod.h"
  6
  7    #ifdef __sun
  8        #include <libzfs.h>
  9        #ifndef __illumos__
 10            // On Solaris 11, zpool_get_prop has only 5 arguments. #2173
 11            #define ffzpool_get_prop(zhp, prop, buf, len, srctype, literal) \
 12                ffzpool_get_prop(zhp, prop, buf, len, srctype)
 13        #endif
 14    #else
 15        #include "libzfs_simplified.h"
 16    #endif
 17
 18    #include "common/library.h"
 19
 20typedef struct FFZfsData {
 21    FF_LIBRARY_SYMBOL(libzfs_fini)
 22    FF_LIBRARY_SYMBOL(zpool_get_prop_int)
 23    FF_LIBRARY_SYMBOL(zpool_get_prop)
 24    FF_LIBRARY_SYMBOL(zpool_close)
 25
 26    // The fields in this struct store property IDs returned by `zpool_name_to_prop`,
 27    // not the property values themselves.
 28    struct {
 29        int name;
 30        int health;
 31        int guid;
 32        int size;
 33        int free;
 34        int allocated;
 35        int fragmentation;
 36        int readonly;
 37    } props;
 38
 39    libzfs_handle_t* handle;
 40    FFlist* result;
 41} FFZfsData;
 42
 43static inline void cleanLibzfs(FFZfsData* data) {
 44    if (data->fflibzfs_fini && data->handle) {
 45        data->fflibzfs_fini(data->handle);
 46        data->handle = nullptr;
 47    }
 48}
 49
 50static int enumZpoolCallback(zpool_handle_t* zpool, void* param) {
 51    FFZfsData* data = (FFZfsData*) param;
 52    zprop_source_t source;
 53    FFZpoolResult* item = FF_LIST_ADD(FFZpoolResult, *data->result);
 54    char buf[1024];
 55    if (data->ffzpool_get_prop(zpool, data->props.name, buf, ARRAY_SIZE(buf), &source, false) == 0) {
 56        ffStrbufInitS(&item->name, buf);
 57    } else {
 58        ffStrbufInitStatic(&item->name, "unknown");
 59    }
 60    if (data->ffzpool_get_prop(zpool, data->props.health, buf, ARRAY_SIZE(buf), &source, false) == 0) {
 61        ffStrbufInitS(&item->state, buf);
 62    } else {
 63        ffStrbufInitStatic(&item->state, "unknown");
 64    }
 65    item->guid = data->ffzpool_get_prop_int(zpool, data->props.guid, &source);
 66    item->total = data->ffzpool_get_prop_int(zpool, data->props.size, &source);
 67    item->used = item->total - data->ffzpool_get_prop_int(zpool, data->props.free, &source);
 68    item->allocated = data->ffzpool_get_prop_int(zpool, data->props.allocated, &source);
 69    uint64_t fragmentation = data->ffzpool_get_prop_int(zpool, data->props.fragmentation, &source);
 70    item->fragmentation = fragmentation == UINT64_MAX ? -DBL_MAX : (double) fragmentation;
 71    item->readOnly = (bool) data->ffzpool_get_prop_int(zpool, data->props.readonly, &source);
 72    data->ffzpool_close(zpool);
 73    return 0;
 74}
 75
 76const char* ffDetectZpool(FFlist* result /* list of FFZpoolResult */) {
 77    FF_LIBRARY_LOAD_MESSAGE(libzfs, "libzfs" FF_LIBRARY_EXTENSION, 6);
 78    FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libzfs, libzfs_init);
 79
 80    libzfs_handle_t* handle = fflibzfs_init();
 81    if (!handle) {
 82        if (!ffKmodLoaded("zfs")) {
 83            return "`zfs` kernel module is not loaded";
 84        }
 85        return "libzfs_init() failed";
 86    }
 87
 88    [[gnu::cleanup(cleanLibzfs)]] FFZfsData data = {
 89        .handle = handle,
 90        .result = result,
 91    };
 92
 93    FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libzfs, zpool_name_to_prop);
 94
 95    #define FF_QUERY_ZPOOL_PROP_FROM_NAME(prop_name)                 \
 96        do {                                                         \
 97            data.props.prop_name = ffzpool_name_to_prop(#prop_name); \
 98            if (data.props.prop_name < 0)                            \
 99                return "Failed to query prop: " #prop_name;          \
100        } while (false)
101    FF_QUERY_ZPOOL_PROP_FROM_NAME(name);
102    FF_QUERY_ZPOOL_PROP_FROM_NAME(health);
103    FF_QUERY_ZPOOL_PROP_FROM_NAME(guid);
104    FF_QUERY_ZPOOL_PROP_FROM_NAME(size);
105    FF_QUERY_ZPOOL_PROP_FROM_NAME(free);
106    FF_QUERY_ZPOOL_PROP_FROM_NAME(allocated);
107    FF_QUERY_ZPOOL_PROP_FROM_NAME(fragmentation);
108    FF_QUERY_ZPOOL_PROP_FROM_NAME(readonly);
109    #undef FF_QUERY_ZPOOL_PROP_FROM_NAME
110
111    FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libzfs, zpool_iter);
112    FF_LIBRARY_LOAD_SYMBOL_VAR_MESSAGE(libzfs, data, libzfs_fini);
113    FF_LIBRARY_LOAD_SYMBOL_VAR_MESSAGE(libzfs, data, zpool_get_prop_int);
114    FF_LIBRARY_LOAD_SYMBOL_VAR_MESSAGE(libzfs, data, zpool_get_prop);
115    FF_LIBRARY_LOAD_SYMBOL_VAR_MESSAGE(libzfs, data, zpool_close);
116
117    if (ffzpool_iter(handle, enumZpoolCallback, &data) < 0) {
118        return "zpool_iter() failed";
119    }
120
121    return nullptr;
122}
123
124#else
125
126const char* ffDetectZpool([[maybe_unused]] FFlist* result) {
127    return "fastfetch was compiled without libzfs support";
128}
129
130#endif