/*
 * flash_{lock,unlock}
 *
 * utilities for locking/unlocking sectors of flash devices
 */

#ifndef PROGRAM_NAME
#define PROGRAM_NAME "flash_unlock"
#define FLASH_MSG    "unlock"
#define FLASH_UNLOCK 1
#else
#define FLASH_MSG    "lock"
#define FLASH_UNLOCK 0
#endif

#include <getopt.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <fcntl.h>
#include <time.h>
#include <sys/ioctl.h>
#include <sys/mount.h>
#include <string.h>

#include "common.h"
#include <mtd/mtd-user.h>

static void usage(int status)
{
	fprintf(status ? stderr : stdout,
		"Usage: %s [options] [--] <mtd device> [offset [block count]]\n"
		"\n"
		"Options:\n"
		" -h         --help              Display this help and exit\n"
		"            --version           Display version information and exit\n"
		"\n"
		"If offset is not specified, it defaults to 0.\n"
		"If block count is not specified, it defaults to all blocks.\n",
		PROGRAM_NAME);
	exit(status);
}

static const char short_opts[] = "h";
static const struct option long_opts[] = {
	{ "help",	no_argument,	0, 'h' },
	{ "version",	no_argument,	0, 'v' },
	{ NULL,		0,		0, 0 },
};

/* Program arguments */
static const char *dev, *offs_s, *count_s;

static void process_args(int argc, char *argv[])
{
	int arg_idx;

	for (;;) {
		int c;

		c = getopt_long(argc, argv, short_opts, long_opts, NULL);
		if (c == EOF)
			break;

		switch (c) {
		case 'h':
			usage(0);
			break;
		case 'v':
			common_print_version();
			exit(0);
		default:
			usage(1);
			break;
		}
	}

	arg_idx = optind;

	/* Sanity checks */
	if (argc - arg_idx < 1) {
		errmsg("too few arguments");
		usage(1);
	} else if (argc - arg_idx > 3) {
		errmsg("too many arguments");
		usage(1);
	}

	/* First non-option argument */
	dev = argv[arg_idx++];

	/* Second non-option argument */
	if (arg_idx < argc)
		offs_s = argv[arg_idx++];
	else
		offs_s = NULL;

	/* Third non-option argument */
	if (arg_idx < argc)
		count_s = argv[arg_idx++];
	else
		count_s = NULL;

}

int main(int argc, char *argv[])
{
	int fd, request;
	struct mtd_info_user mtdInfo;
	struct erase_info_user mtdLockInfo;
	int count;

	process_args(argc, argv);

	/* Get the device info to compare to command line sizes */
	fd = open(dev, O_RDWR);
	if (fd < 0)
		sys_errmsg_die("could not open: %s", dev);

	if (ioctl(fd, MEMGETINFO, &mtdInfo))
		sys_errmsg_die("could not get mtd info: %s", dev);

	/* Make sure user options are valid */
	if (offs_s)
		mtdLockInfo.start = strtol(offs_s, NULL, 0);
	else
		mtdLockInfo.start = 0;
	if (mtdLockInfo.start > mtdInfo.size)
		errmsg_die("%#x is beyond device size %#x",
			mtdLockInfo.start, mtdInfo.size);

	if (count_s) {
		count = strtol(count_s, NULL, 0);
		if (count == -1)
			mtdLockInfo.length = mtdInfo.size;
		else
			mtdLockInfo.length = mtdInfo.erasesize * count;
	} else
		mtdLockInfo.length = mtdInfo.size;
	if (mtdLockInfo.start + mtdLockInfo.length > mtdInfo.size)
		errmsg_die("range is more than device supports: %#x + %#x > %#x",
			mtdLockInfo.start, mtdLockInfo.length, mtdInfo.size);

	/* Finally do the operation */
	request = FLASH_UNLOCK ? MEMUNLOCK : MEMLOCK;
	if (ioctl(fd, request, &mtdLockInfo))
		sys_errmsg_die("could not %s device: %s\n",
			FLASH_MSG, dev);

	return 0;
}