#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <linux/reboot.h>
#include <sys/reboot.h>

int main(int argc, char *argv[])
{
	int mode = LINUX_REBOOT_CMD_HALT;

	if (argc != 2) {
		printf("Please provide a reboot mode\n");
		return -1;
	}
	switch (atoi(argv[1])) {
	case	1:
		mode = LINUX_REBOOT_CMD_HALT;
		break;
	case	2:
		mode = LINUX_REBOOT_CMD_POWER_OFF;
		break;
	case	3:
		mode = LINUX_REBOOT_CMD_RESTART;
		break;
	case	4:
		mode = LINUX_REBOOT_CMD_RESTART2;
		break;
	default:
		break;
	}
	//reboot(LINUX_REBOOT_MAGIC1, LINUX_REBOOT_MAGIC2, mode, NULL);
	reboot(mode);
	return 0;
}