/*
   * (C) Copyright David Gibson <dwg@au1.ibm.com>, IBM Corporation.  2005.
   *
   *
   * This program is free software; you can redistribute it and/or
   * modify it under the terms of the GNU General Public License as
   * published by the Free Software Foundation; either version 2 of the
   * License, or (at your option) any later version.
   *
   *  This program is distributed in the hope that it will be useful,
   *  but WITHOUT ANY WARRANTY; without even the implied warranty of
   *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   *  General Public License for more details.
   *
   *  You should have received a copy of the GNU General Public License
   *  along with this program; if not, write to the Free Software
   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
   *                                                                   USA
   */
  
  #include "dtc.h"
  #include "srcpos.h"
  
  #include "version_gen.h"
  
  /*
   * Command line options
   */
  int quiet;		/* Level of quietness */
  int reservenum;		/* Number of memory reservation slots */
  int minsize;		/* Minimum blob size */
  int padsize;		/* Additional padding to blob */

  int phandle_format = PHANDLE_BOTH;	/* Use linux,phandle or phandle properties */

  static void fill_fullpaths(struct node *tree, const char *prefix)

  {
  	struct node *child;
  	const char *unit;
  
  	tree->fullpath = join_path(prefix, tree->name);
  
  	unit = strchr(tree->name, '@');
  	if (unit)
  		tree->basenamelen = unit - tree->name;
  	else
  		tree->basenamelen = strlen(tree->name);
  
  	for_each_child(tree, child)
  		fill_fullpaths(child, tree->fullpath);
  }
  
  static void  __attribute__ ((noreturn)) usage(void)
  {
  	fprintf(stderr, "Usage:
  ");
  	fprintf(stderr, "\tdtc [options] <input file>
  ");
  	fprintf(stderr, "
  Options:
  ");
  	fprintf(stderr, "\t-h
  ");
  	fprintf(stderr, "\t\tThis help text
  ");
  	fprintf(stderr, "\t-q
  ");
  	fprintf(stderr, "\t\tQuiet: -q suppress warnings, -qq errors, -qqq all
  ");
  	fprintf(stderr, "\t-I <input format>
  ");
  	fprintf(stderr, "\t\tInput formats are:
  ");
  	fprintf(stderr, "\t\t\tdts - device tree source text
  ");
  	fprintf(stderr, "\t\t\tdtb - device tree blob
  ");
  	fprintf(stderr, "\t\t\tfs - /proc/device-tree style directory
  ");
  	fprintf(stderr, "\t-o <output file>
  ");
  	fprintf(stderr, "\t-O <output format>
  ");
  	fprintf(stderr, "\t\tOutput formats are:
  ");
  	fprintf(stderr, "\t\t\tdts - device tree source text
  ");
  	fprintf(stderr, "\t\t\tdtb - device tree blob
  ");
  	fprintf(stderr, "\t\t\tasm - assembler source
  ");
  	fprintf(stderr, "\t-V <output version>
  ");
  	fprintf(stderr, "\t\tBlob version to produce, defaults to %d (relevant for dtb
  \t\tand asm output only)
  ", DEFAULT_FDT_VERSION);

  	fprintf(stderr, "\t-d <output dependency file>
  ");

  	fprintf(stderr, "\t-R <number>
  ");
  	fprintf(stderr, "\t\tMake space for <number> reserve map entries (relevant for 
  \t\tdtb and asm output only)
  ");
  	fprintf(stderr, "\t-S <bytes>
  ");
  	fprintf(stderr, "\t\tMake the blob at least <bytes> long (extra space)
  ");
  	fprintf(stderr, "\t-p <bytes>
  ");
  	fprintf(stderr, "\t\tAdd padding to the blob of <bytes> long (extra space)
  ");
  	fprintf(stderr, "\t-b <number>
  ");
  	fprintf(stderr, "\t\tSet the physical boot cpu
  ");
  	fprintf(stderr, "\t-f
  ");
  	fprintf(stderr, "\t\tForce - try to produce output even if the input tree has errors
  ");

  	fprintf(stderr, "\t-s
  ");
  	fprintf(stderr, "\t\tSort nodes and properties before outputting (only useful for
  \t\tcomparing trees)
  ");

  	fprintf(stderr, "\t-v
  ");
  	fprintf(stderr, "\t\tPrint DTC version and exit
  ");

  	fprintf(stderr, "\t-H <phandle format>
  ");
  	fprintf(stderr, "\t\tphandle formats are:
  ");
  	fprintf(stderr, "\t\t\tlegacy - \"linux,phandle\" properties only
  ");
  	fprintf(stderr, "\t\t\tepapr - \"phandle\" properties only
  ");
  	fprintf(stderr, "\t\t\tboth - Both \"linux,phandle\" and \"phandle\" properties
  ");

  	exit(3);

  }
  
  int main(int argc, char *argv[])
  {
  	struct boot_info *bi;
  	const char *inform = "dts";
  	const char *outform = "dts";
  	const char *outname = "-";

  	const char *depname = NULL;

  	int force = 0, check = 0, sort = 0;

  	const char *arg;
  	int opt;

  	FILE *outf = NULL;
  	int outversion = DEFAULT_FDT_VERSION;

  	long long cmdline_boot_cpuid = -1;

  
  	quiet      = 0;
  	reservenum = 0;
  	minsize    = 0;
  	padsize    = 0;

  	while ((opt = getopt(argc, argv, "hI:O:o:V:d:R:S:p:fcqb:vH:s"))
  			!= EOF) {

  		switch (opt) {
  		case 'I':
  			inform = optarg;
  			break;
  		case 'O':
  			outform = optarg;
  			break;
  		case 'o':
  			outname = optarg;
  			break;
  		case 'V':
  			outversion = strtol(optarg, NULL, 0);
  			break;

  		case 'd':
  			depname = optarg;
  			break;

  		case 'R':
  			reservenum = strtol(optarg, NULL, 0);
  			break;
  		case 'S':
  			minsize = strtol(optarg, NULL, 0);
  			break;
  		case 'p':
  			padsize = strtol(optarg, NULL, 0);
  			break;
  		case 'f':
  			force = 1;
  			break;
  		case 'c':
  			check = 1;
  			break;
  		case 'q':
  			quiet++;
  			break;
  		case 'b':

  			cmdline_boot_cpuid = strtoll(optarg, NULL, 0);

  			break;
  		case 'v':

  			printf("Version: %s
  ", DTC_VERSION);
  			exit(0);

  		case 'H':
  			if (streq(optarg, "legacy"))
  				phandle_format = PHANDLE_LEGACY;
  			else if (streq(optarg, "epapr"))
  				phandle_format = PHANDLE_EPAPR;
  			else if (streq(optarg, "both"))
  				phandle_format = PHANDLE_BOTH;
  			else
  				die("Invalid argument \"%s\" to -H option
  ",
  				    optarg);
  			break;
  
  		case 's':
  			sort = 1;
  			break;

  		case 'h':
  		default:
  			usage();
  		}
  	}
  
  	if (argc > (optind+1))
  		usage();
  	else if (argc < (optind+1))
  		arg = "-";
  	else
  		arg = argv[optind];
  
  	/* minsize and padsize are mutually exclusive */

  	if (minsize && padsize)

  		die("Can't set both -p and -S
  ");

  	if (minsize)
  		fprintf(stderr, "DTC: Use of \"-S\" is deprecated; it will be removed soon, use \"-p\" instead
  ");

  	fprintf(stderr, "DTC: %s->%s  on file \"%s\"
  ",
  		inform, outform, arg);

  	if (depname) {
  		depfile = fopen(depname, "w");
  		if (!depfile)
  			die("Couldn't open dependency file %s: %s
  ", depname,
  			    strerror(errno));
  		fprintf(depfile, "%s:", outname);
  	}

  	if (streq(inform, "dts"))

  		bi = dt_from_source(arg);

  	else if (streq(inform, "fs"))

  		bi = dt_from_fs(arg);

  	else if(streq(inform, "dtb"))
  		bi = dt_from_blob(arg);
  	else

  		die("Unknown input format \"%s\"
  ", inform);

  	if (depfile) {
  		fputc('
  ', depfile);
  		fclose(depfile);
  	}

  	if (cmdline_boot_cpuid != -1)
  		bi->boot_cpuid_phys = cmdline_boot_cpuid;

  	fill_fullpaths(bi->dt, "");
  	process_checks(force, bi);

  	if (sort)
  		sort_tree(bi);

  
  	if (streq(outname, "-")) {
  		outf = stdout;
  	} else {
  		outf = fopen(outname, "w");
  		if (! outf)
  			die("Couldn't open output file %s: %s
  ",
  			    outname, strerror(errno));
  	}
  
  	if (streq(outform, "dts")) {
  		dt_to_source(outf, bi);
  	} else if (streq(outform, "dtb")) {

  		dt_to_blob(outf, bi, outversion);

  	} else if (streq(outform, "asm")) {

  		dt_to_asm(outf, bi, outversion);

  	} else if (streq(outform, "null")) {
  		/* do nothing */
  	} else {
  		die("Unknown output format \"%s\"
  ", outform);
  	}
  
  	exit(0);
  }