/*
 * Broiler Emulate Machine
 *
 * (C) 2022.07.04 BuddyZhang1 <buddy.zhang@aliyun.com>
 * (C) 2022.07.04 BiscuitOS   <buddy.zhang@aliyun.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */
#include "broiler/broiler.h"
#include <getopt.h>

/* global data */
struct broiler *broiler;

void usage(const char *program_name) 
{
	printf("%s 1.0.0(2022-06-01)\n", program_name);
	printf("This is a program BEMU\n");
	printf("Usage:%s --kernel <kernel> --rootfs <rootfs> "
		"--memory <memory:MiB> "
		"--cpu <cpu> --cmdline <cmdline>\n", program_name);
}

int main(int argc, char *argv[])
{
	const char *short_opts = "hk:r:d:m:c:";
	const struct option long_opts[] = {
		{ "help", no_argument, NULL, 'h'},
		{ "kernel", required_argument, NULL, 'k'},
		{ "rootfs", required_argument, NULL, 'r'},
		{ "cmdline", required_argument, NULL, 'd'},
		{ "memory", required_argument, NULL, 'm'},
		{ "cpu", required_argument, NULL, 'c'},
		{ 0, 0, 0, 0 }
	};
	int hflag = 0;
	int c;
	opterr = 0;

	broiler = calloc(sizeof(*broiler), 1);
	if (!broiler) {
		printf("NO free memory create broiler.\n");
		abort();
	}

	while((c = getopt_long(argc, argv, short_opts,
						long_opts, NULL)) != -1) {
		switch(c) {
		case 'h': /* Helper */
			usage(argv[0]);
			return 0;
		case 'k': /* Kernel Image */
			broiler->kernel_name = optarg;
			break;
		case 'r': /* Rootfs */
			broiler->rootfs_name = optarg;
			broiler->disk_name[broiler->nr_disks] = optarg;
			broiler->nr_disks++;
			break;
		case 'd': /* Kernel CMDLINE */
			broiler->cmdline = optarg;
			break;
		case 'm': /* Memory */
			broiler->ram_size = strtoll(optarg, NULL, 0);
			broiler->ram_size <<= 20; /* MiB */
			break;
		case 'c': /* CPU nr */
			broiler->nr_cpu = strtol(optarg, NULL, 0);
			break;
		case '?':
			printf("Error:unknow option '-%c'\n", optopt);
			break;
		default:
			abort();
		}
	}

	if (broiler_base_init(broiler) < 0)
		goto err;
	return 0;

err:
	free(broiler);
	return -EINVAL;
}