/* Copyright 2012 The ChromiumOS Authors
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */

#define __STDC_FORMAT_MACROS

#include <getopt.h>
#include <inttypes.h>
#include <string.h>

#include "cgpt.h"
#include "vboot_host.h"

extern const char *progname;

static void Usage(void)
{
	printf("\nUsage: %s show [OPTIONS] DRIVE\n\n"
	       "Display the GPT table.\n\n"
	       "Units are blocks by default.\n\n"
	       "Options:\n"
	       "  -D NUM       Size (in bytes) of the disk where partitions reside;\n"
	       "                 default 0, meaning partitions and GPT structs are\n"
	       "                 both on DRIVE\n"
	       "  -n           Numeric output only\n"
	       "  -v           Verbose output\n"
	       "  -q           Quick output\n"
	       "  -i NUM       Show specified partition only\n"
	       "  -d           Debug output (including invalid headers)\n"
	       "\n"
	       "When using -i, specific fields may be displayed using one of:\n"
	       "  -b  first block (a.k.a. start of partition)\n"
	       "  -s  partition size (in blocks)\n"
	       "  -t  type guid\n"
	       "  -u  unique guid\n"
	       "  -l  label\n"
	       "  -S  Successful flag\n"
	       "  -T  Tries flag\n"
	       "  -P  Priority flag\n"
	       "  -R  Required flag\n"
	       "  -B  Legacy Boot flag\n"
	       "  -A  raw 16-bit attribute value (bits 48-63)\n"
	       "\n",
	       progname);
}

int cmd_show(int argc, char *argv[])
{
	CgptShowParams params;
	memset(&params, 0, sizeof(params));

	int c;
	int errorcnt = 0;
	char *e = 0;

	opterr = 0; // quiet, you
	while ((c = getopt(argc, argv, ":hnvqi:bstulSTPRBAdD:")) != -1) {
		switch (c) {
		case 'D':
			params.drive_size = strtoull(optarg, &e, 0);
			errorcnt += check_int_parse(c, e);
			break;
		case 'n':
			params.numeric = 1;
			break;
		case 'v':
			params.verbose = 1;
			break;
		case 'q':
			params.quick = 1;
			break;
		case 'i':
			params.partition = (uint32_t)strtoul(optarg, &e, 0);
			errorcnt += check_int_parse(c, e);
			if (params.partition <= 0) {
				Error("-i requires a number between 1 and 128 (inclusive)\n");
				errorcnt++;
			}
			break;
		case 'b':
		case 's':
		case 't':
		case 'u':
		case 'l':
		case 'S':
		case 'T':
		case 'P':
		case 'R':
		case 'B':
		case 'A':
			if (params.single_item) {
				Error("-%c already specified; rejecting additional -%c\n",
				      params.single_item, c);
				Error("Only a single item may be displayed at a time\n");
				errorcnt++;
			}
			params.single_item = c;
			break;

		case 'd':
			params.debug = 1;
			break;

		case 'h':
			Usage();
			return CGPT_OK;
		case '?':
			Error("unrecognized option: -%c\n", optopt);
			errorcnt++;
			break;
		case ':':
			Error("missing argument to -%c\n", optopt);
			errorcnt++;
			break;
		default:
			errorcnt++;
			break;
		}
	}
	if (!params.partition && params.single_item) {
		Error("-i required when displaying a single item\n");
		errorcnt++;
	}
	if (errorcnt) {
		Usage();
		return CGPT_FAILED;
	}

	if (optind >= argc) {
		Error("missing drive argument\n");
		Usage();
		return CGPT_FAILED;
	}

	params.drive_name = argv[optind];

	return CgptShow(&params);
}