/* radare - LGPL - Copyright 2011-2017 pancake<nopcode.org> */

#include <r_io.h>
#include <r_fs.h>
#include "grubfs.h"
#include <stdio.h>
#include <string.h>

static RIOBind *bio = NULL;
static ut64 delta = 0;

static void *empty(int sz) {
	void *p = malloc (sz);
	if (p) {
		memset (p, '\0', sz);
	}
	return p;
}

static grub_err_t read_foo(struct grub_disk *disk, grub_disk_addr_t sector, grub_size_t size, char *buf) {
	if (!disk) {
		eprintf ("oops. no disk\n");
		return 1;
	}
	const int blocksize = 512; // TODO unhardcode 512
	RIOBind *iob = disk->data;
	if (bio) {
		iob = bio;
	}
	// printf ("io %p\n", file->root->iob.io);
	const int len = size * blocksize;
	return iob->read_at (iob->io, delta + (blocksize * sector), (ut8 *)buf, len) != len;
}

GrubFS *grubfs_new(struct grub_fs *myfs, void *data) {
	struct grub_file *file;
	GrubFS *gfs = empty (sizeof (GrubFS));
	// hacky mallocs :D
	gfs->file = file = empty (sizeof (struct grub_file));
	file->device = empty (sizeof (struct grub_device) + 1024);
	file->device->disk = empty (sizeof (struct grub_disk));
	file->device->disk->dev = (grub_disk_dev_t)file->device; // hack!
	file->device->disk->dev->read = read_foo; // grub_disk_dev
	file->device->disk->data = data;
	// file->device->disk->read_hook = read_foo; //read_hook;
	file->fs = myfs;
	return gfs;
}

grub_disk_t grubfs_disk(void *data) {
	struct grub_disk *disk = empty (sizeof (struct grub_disk));
	disk->dev = empty (sizeof (struct grub_disk_dev));
	disk->dev->read = read_foo; // grub_disk_dev
	disk->data = data;
	return disk;
}

void grubfs_free(GrubFS *gf) {
	if (gf) {
		if (gf->file && gf->file->device) {
			free (gf->file->device->disk);
		}
		// free (gf->file->device);
		free (gf->file);
		free (gf);
	}
}

void grubfs_bind_io(RIOBind *iob, ut64 _delta) {
	bio = iob;
	delta = _delta;
}