1 files changed, 84 insertions, 0 deletions
diff --git a/initrd/initrd.c b/initrd/initrd.c
new file mode 100644
index 0000000..8fe0f01
--- /dev/null
+++ b/initrd/initrd.c
@@ -0,0 +1,84 @@
+#include "initrd.h"
+#include "tar.h"
+#include "string.h"
+#include "commandline.h"
+#include "memory.h"
+#include "print.h"
+#include "klib/rbtree.h"
+#include "klib/hash.h"
+#include "boot.h"
+
+static rbtree_t s_initrd_files;
+
+static size_t
+s_tar_oct_dec(const char *oct)
+{
+    size_t value = 0;
+    while(*oct != 0) {
+        value *= 8;
+        value += (*oct) - '0';
+        oct++;
+    }
+    return value;
+}
+
+static void
+s_initrd_parse(struct BootModule *module)
+{
+    struct tar_block *sector = (struct tar_block*)module->addr;
+    while(true) {
+        tar_header_t *header = (tar_header_t*)sector;
+        if(header->name[0] == 0) break;
+
+        initrd_file_t file_tmp;
+        initrd_file_t *file = rbtree_insert(
+                &s_initrd_files, 
+                string_hash(header->name), 
+                &file_tmp);
+
+        *file = (initrd_file_t){ 
+            .header = header,
+            .name = header->name,
+            .size = s_tar_oct_dec(header->size),
+            .data = &sector[1]
+        };
+        kdbgf("File %s size %i\n", file->name, file->size);
+ 
+        sector = &sector[(file->size / 512) + 1];
+        if(file->size % 512 > 0) sector = &sector[1];
+    }
+}
+
+initrd_file_t *
+ird_getfile(const char *path)
+{
+    initrd_file_t *file = rbtree_find(&s_initrd_files, string_hash(path));
+    return file;
+}
+
+void
+initrd_setup(void)
+{
+    const char *initrd_path = cmdline_get("initrd");
+    if(initrd_path == 0) {
+        kerrf("No initrd loaded!\n");
+        return;
+    }
+    kdbgf("Initrd located in module path %s\n", initrd_path);
+    struct BootModule *initrd_module = NULL;
+    for(size_t i = 0; i < boot_module_count; i++) {
+        struct BootModule *module = &boot_modules[i];
+        if(strcmp(module->path, initrd_path) == 0) {
+            initrd_module = module;
+            break;
+        }
+    }
+
+    if(initrd_module == NULL) {
+        klogf("Initrd not found in modules!\n");
+        return;
+    }
+
+    rbtree_new(&s_initrd_files, initrd_file_t);
+    s_initrd_parse(initrd_module);
+}