1 files changed, 49 insertions, 0 deletions
diff --git a/pcilib/unit.c b/pcilib/unit.c
new file mode 100644
index 0000000..2dd7113
--- /dev/null
+++ b/pcilib/unit.c
@@ -0,0 +1,49 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <strings.h>
+
+#include "pci.h"
+#include "pcilib.h"
+#include "unit.h"
+#include "error.h"
+
+
+pcilib_unit_t pcilib_find_unit_by_name(pcilib_t *ctx, const char *unit) {
+    pcilib_unit_t i;
+
+    for(i = 0; ctx->units[i].name; i++) {
+        if (!strcasecmp(ctx->units[i].name, unit))
+	    return i;
+    }
+    return PCILIB_UNIT_INVALID;
+}
+
+
+int pcilib_add_units(pcilib_t *ctx, size_t n, const pcilib_unit_description_t *desc) {
+    if (!n) {
+        for (n = 0; desc[n].name; n++);
+    }
+
+    if ((ctx->num_units + n + 1) > ctx->alloc_units) {
+	size_t size;
+	pcilib_unit_description_t *units;
+
+        for (size = ctx->alloc_units; size < 2 * (n + ctx->num_units + 1); size <<= 1);
+
+        units = (pcilib_unit_description_t*)realloc(ctx->units, size * sizeof(pcilib_unit_description_t));
+        if (!units) return PCILIB_ERROR_MEMORY;
+
+        ctx->units = units;
+        ctx->alloc_units = size;
+
+        ctx->model_info.units = units;
+    }
+
+    memcpy(ctx->units + ctx->num_units, desc, n * sizeof(pcilib_unit_description_t));
+    memset(ctx->units + ctx->num_units + n, 0, sizeof(pcilib_unit_description_t));
+
+    ctx->num_units += n;
+
+    return 0;
+}