/*

 * (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"

#ifdef _WIN32

#include <fcntl.h>

#include <io.h>

#endif

#define FTF_FULLPATH	0x1

#define FTF_VARALIGN	0x2

#define FTF_NAMEPROPS	0x4

#define FTF_BOOTCPUID	0x8

#define FTF_STRTABSIZE	0x10

#define FTF_STRUCTSIZE	0x20

#define FTF_NOPS	0x40

static struct version_info {

	int version;

	int last_comp_version;

	int hdr_size;

	int flags;

} version_table[] = {

	{1, 1, FDT_V1_SIZE,

	 FTF_FULLPATH|FTF_VARALIGN|FTF_NAMEPROPS},

	{2, 1, FDT_V2_SIZE,

	 FTF_FULLPATH|FTF_VARALIGN|FTF_NAMEPROPS|FTF_BOOTCPUID},

	{3, 1, FDT_V3_SIZE,

	 FTF_FULLPATH|FTF_VARALIGN|FTF_NAMEPROPS|FTF_BOOTCPUID|FTF_STRTABSIZE},

	{16, 16, FDT_V3_SIZE,

	 FTF_BOOTCPUID|FTF_STRTABSIZE|FTF_NOPS},

	{17, 16, FDT_V17_SIZE,

	 FTF_BOOTCPUID|FTF_STRTABSIZE|FTF_STRUCTSIZE|FTF_NOPS},

};

struct emitter {

	void (*cell)(void *, cell_t);

	void (*string)(void *, char *, int);

	void (*align)(void *, int);

	void (*data)(void *, struct data);

	void (*beginnode)(void *, const char *);

	void (*endnode)(void *, const char *);

	void (*property)(void *, const char *);

};

static void bin_emit_cell(void *e, cell_t val)

{

	struct data *dtbuf = e;

	*dtbuf = data_append_cell(*dtbuf, val);

}

static void bin_emit_string(void *e, char *str, int len)

{

	struct data *dtbuf = e;

	if (len == 0)

		len = strlen(str);

	*dtbuf = data_append_data(*dtbuf, str, len);

	*dtbuf = data_append_byte(*dtbuf, '\0');

}

static void bin_emit_align(void *e, int a)

{

	struct data *dtbuf = e;

	*dtbuf = data_append_align(*dtbuf, a);

}

static void bin_emit_data(void *e, struct data d)

{

	struct data *dtbuf = e;

	*dtbuf = data_append_data(*dtbuf, d.val, d.len);

}

static void bin_emit_beginnode(void *e, const char *label)

{

	bin_emit_cell(e, FDT_BEGIN_NODE);

}

static void bin_emit_endnode(void *e, const char *label)

{

	bin_emit_cell(e, FDT_END_NODE);

}

static void bin_emit_property(void *e, const char *label)

{

	bin_emit_cell(e, FDT_PROP);

}

static struct emitter bin_emitter = {

	.cell = bin_emit_cell,

	.string = bin_emit_string,

	.align = bin_emit_align,

	.data = bin_emit_data,

	.beginnode = bin_emit_beginnode,

	.endnode = bin_emit_endnode,

	.property = bin_emit_property,

};

static void emit_label(FILE *f, const char *prefix, const char *label)

{

	fprintf(f, "\t.globl\t%s_%s\n", prefix, label);

	fprintf(f, "%s_%s:\n", prefix, label);

	fprintf(f, "_%s_%s:\n", prefix, label);

}

static void emit_offset_label(FILE *f, const char *label, int offset)

{

	fprintf(f, "\t.globl\t%s\n", label);

	fprintf(f, "%s\t= . + %d\n", label, offset);

}

static void asm_emit_cell(void *e, cell_t val)

{

	FILE *f = e;

	fprintf(f, "\t.long\t0x%x\n", val);

}

static void asm_emit_string(void *e, char *str, int len)

{

	FILE *f = e;

	char c = 0;

	if (len != 0) {

		/* XXX: ewww */

		c = str[len];

		str[len] = '\0';

	}

	fprintf(f, "\t.string\t\"%s\"\n", str);

	if (len != 0) {

		str[len] = c;

	}

}

static void asm_emit_align(void *e, int a)

{

	FILE *f = e;

	fprintf(f, "\t.balign\t%d\n", a);

}

static void asm_emit_data(void *e, struct data d)

{

	FILE *f = e;

	int off = 0;

	struct marker *m = d.markers;

	for_each_marker_of_type(m, LABEL)

		emit_offset_label(f, m->ref, m->offset);

	while ((d.len - off) >= sizeof(uint32_t)) {

		fprintf(f, "\t.long\t0x%x\n",

			fdt32_to_cpu(*((uint32_t *)(d.val+off))));

		off += sizeof(uint32_t);

	}

	while ((d.len - off) >= 1) {

		fprintf(f, "\t.byte\t0x%hhx\n", d.val[off]);

		off += 1;

	}

	assert(off == d.len);

}

static void asm_emit_beginnode(void *e, const char *label)

{

	FILE *f = e;

	if (label) {

		fprintf(f, "\t.globl\t%s\n", label);

		fprintf(f, "%s:\n", label);

	}

	fprintf(f, "\t.long\tFDT_BEGIN_NODE\n");

}

static void asm_emit_endnode(void *e, const char *label)

{

	FILE *f = e;

	fprintf(f, "\t.long\tFDT_END_NODE\n");

	if (label) {

		fprintf(f, "\t.globl\t%s_end\n", label);

		fprintf(f, "%s_end:\n", label);

	}

}

static void asm_emit_property(void *e, const char *label)

{

	FILE *f = e;

	if (label) {

		fprintf(f, "\t.globl\t%s\n", label);

		fprintf(f, "%s:\n", label);

	}

	fprintf(f, "\t.long\tFDT_PROP\n");

}

static struct emitter asm_emitter = {

	.cell = asm_emit_cell,

	.string = asm_emit_string,

	.align = asm_emit_align,

	.data = asm_emit_data,

	.beginnode = asm_emit_beginnode,

	.endnode = asm_emit_endnode,

	.property = asm_emit_property,

};

static int stringtable_insert(struct data *d, const char *str)

{

	int i;

	/* FIXME: do this more efficiently? */

	for (i = 0; i < d->len; i++) {

		if (streq(str, d->val + i))

			return i;

	}

	*d = data_append_data(*d, str, strlen(str)+1);

	return i;

}

static void flatten_tree(struct node *tree, struct emitter *emit,

			 void *etarget, struct data *strbuf,

			 struct version_info *vi)

{

	struct property *prop;

	struct node *child;

	int seen_name_prop = 0;

	emit->beginnode(etarget, tree->label);

	if (vi->flags & FTF_FULLPATH)

		emit->string(etarget, tree->fullpath, 0);

	else

		emit->string(etarget, tree->name, 0);

	emit->align(etarget, sizeof(cell_t));

	for_each_property(tree, prop) {

		int nameoff;

		if (streq(prop->name, "name"))

			seen_name_prop = 1;

		nameoff = stringtable_insert(strbuf, prop->name);

		emit->property(etarget, prop->label);

		emit->cell(etarget, prop->val.len);

		emit->cell(etarget, nameoff);

		if ((vi->flags & FTF_VARALIGN) && (prop->val.len >= 8))

			emit->align(etarget, 8);

		emit->data(etarget, prop->val);

		emit->align(etarget, sizeof(cell_t));

	}

	if ((vi->flags & FTF_NAMEPROPS) && !seen_name_prop) {

		emit->property(etarget, NULL);

		emit->cell(etarget, tree->basenamelen+1);

		emit->cell(etarget, stringtable_insert(strbuf, "name"));

		if ((vi->flags & FTF_VARALIGN) && ((tree->basenamelen+1) >= 8))

			emit->align(etarget, 8);

		emit->string(etarget, tree->name, tree->basenamelen);

		emit->align(etarget, sizeof(cell_t));

	}

	for_each_child(tree, child) {

		flatten_tree(child, emit, etarget, strbuf, vi);

	}

	emit->endnode(etarget, tree->label);

}

static struct data flatten_reserve_list(struct reserve_info *reservelist,

				 struct version_info *vi)

{

	struct reserve_info *re;

	struct data d = empty_data;

	static struct fdt_reserve_entry null_re = {0,0};

	int    j;

	for (re = reservelist; re; re = re->next) {

		d = data_append_re(d, &re->re);

	}

	/*

	 * Add additional reserved slots if the user asked for them.

	 */

	for (j = 0; j < reservenum; j++) {

		d = data_append_re(d, &null_re);

	}

	return d;

}

static void make_fdt_header(struct fdt_header *fdt,

			    struct version_info *vi,

			    int reservesize, int dtsize, int strsize,

			    int boot_cpuid_phys)

{

	int reserve_off;

	reservesize += sizeof(struct fdt_reserve_entry);

	memset(fdt, 0xff, sizeof(*fdt));

	fdt->magic = cpu_to_fdt32(FDT_MAGIC);

	fdt->version = cpu_to_fdt32(vi->version);

	fdt->last_comp_version = cpu_to_fdt32(vi->last_comp_version);

	/* Reserve map should be doubleword aligned */

	reserve_off = ALIGN(vi->hdr_size, 8);

	fdt->off_mem_rsvmap = cpu_to_fdt32(reserve_off);

	fdt->off_dt_struct = cpu_to_fdt32(reserve_off + reservesize);

	fdt->off_dt_strings = cpu_to_fdt32(reserve_off + reservesize

					  + dtsize);

	fdt->totalsize = cpu_to_fdt32(reserve_off + reservesize + dtsize + strsize);

	if (vi->flags & FTF_BOOTCPUID)

		fdt->boot_cpuid_phys = cpu_to_fdt32(boot_cpuid_phys);

	if (vi->flags & FTF_STRTABSIZE)

		fdt->size_dt_strings = cpu_to_fdt32(strsize);

	if (vi->flags & FTF_STRUCTSIZE)

		fdt->size_dt_struct = cpu_to_fdt32(dtsize);

}

void dt_to_blob(FILE *f, struct boot_info *bi, int version)

{

	struct version_info *vi = NULL;

	int i;

	struct data blob       = empty_data;

	struct data reservebuf = empty_data;

	struct data dtbuf      = empty_data;

	struct data strbuf     = empty_data;

	struct fdt_header fdt;

	int padlen = 0;

	for (i = 0; i < ARRAY_SIZE(version_table); i++) {

		if (version_table[i].version == version)

			vi = &version_table[i];

	}

	if (!vi)

		die("Unknown device tree blob version %d\n", version);

	flatten_tree(bi->dt, &bin_emitter, &dtbuf, &strbuf, vi);

	bin_emit_cell(&dtbuf, FDT_END);

	reservebuf = flatten_reserve_list(bi->reservelist, vi);

	/* Make header */

	make_fdt_header(&fdt, vi, reservebuf.len, dtbuf.len, strbuf.len,

			bi->boot_cpuid_phys);

	/*

	 * If the user asked for more space than is used, adjust the totalsize.

	 */

	if (minsize > 0) {

		padlen = minsize - fdt32_to_cpu(fdt.totalsize);

		if ((padlen < 0) && (quiet < 1))

			fprintf(stderr,

				"Warning: blob size %d >= minimum size %d\n",

				fdt32_to_cpu(fdt.totalsize), minsize);

	}

	if (padsize > 0)

		padlen = padsize;

	if (padlen > 0) {

		int tsize = fdt32_to_cpu(fdt.totalsize);

		tsize += padlen;

		fdt.totalsize = cpu_to_fdt32(tsize);

	}

	/*

	 * Assemble the blob: start with the header, add with alignment

	 * the reserve buffer, add the reserve map terminating zeroes,

	 * the device tree itself, and finally the strings.

	 */

	blob = data_append_data(blob, &fdt, vi->hdr_size);

	blob = data_append_align(blob, 8);

	blob = data_merge(blob, reservebuf);

	blob = data_append_zeroes(blob, sizeof(struct fdt_reserve_entry));

	blob = data_merge(blob, dtbuf);

	blob = data_merge(blob, strbuf);

	/*

	 * If the user asked for more space than is used, pad out the blob.

	 */

	if (padlen > 0)

		blob = data_append_zeroes(blob, padlen);

	fwrite(blob.val, blob.len, 1, f);

	if (ferror(f))

		die("Error writing device tree blob: %s\n", strerror(errno));

	/*

	 * data_merge() frees the right-hand element so only the blob

	 * remains to be freed.

	 */

	data_free(blob);

}

static void dump_stringtable_asm(FILE *f, struct data strbuf)

{

	const char *p;

	int len;

	p = strbuf.val;

	while (p < (strbuf.val + strbuf.len)) {

		len = strlen(p);

		fprintf(f, "\t.string \"%s\"\n", p);

		p += len+1;

	}

}

void dt_to_asm(FILE *f, struct boot_info *bi, int version)

{

	struct version_info *vi = NULL;

	int i;

	struct data strbuf = empty_data;

	struct reserve_info *re;

	const char *symprefix = "dt";

	for (i = 0; i < ARRAY_SIZE(version_table); i++) {

		if (version_table[i].version == version)

			vi = &version_table[i];

	}

	if (!vi)

		die("Unknown device tree blob version %d\n", version);

	fprintf(f, "/* autogenerated by dtc, do not edit */\n\n");

	fprintf(f, "#define FDT_MAGIC 0x%x\n", FDT_MAGIC);

	fprintf(f, "#define FDT_BEGIN_NODE 0x%x\n", FDT_BEGIN_NODE);

	fprintf(f, "#define FDT_END_NODE 0x%x\n", FDT_END_NODE);

	fprintf(f, "#define FDT_PROP 0x%x\n", FDT_PROP);

	fprintf(f, "#define FDT_END 0x%x\n", FDT_END);

	fprintf(f, "\n");

	emit_label(f, symprefix, "blob_start");

	emit_label(f, symprefix, "header");

	fprintf(f, "\t.long\tFDT_MAGIC\t\t\t\t/* magic */\n");

	fprintf(f, "\t.long\t_%s_blob_abs_end - _%s_blob_start\t/* totalsize */\n",

		symprefix, symprefix);

	fprintf(f, "\t.long\t_%s_struct_start - _%s_blob_start\t/* off_dt_struct */\n",

		symprefix, symprefix);

	fprintf(f, "\t.long\t_%s_strings_start - _%s_blob_start\t/* off_dt_strings */\n",

		symprefix, symprefix);

	fprintf(f, "\t.long\t_%s_reserve_map - _%s_blob_start\t/* off_dt_strings */\n",

		symprefix, symprefix);

	fprintf(f, "\t.long\t%d\t\t\t\t\t/* version */\n", vi->version);

	fprintf(f, "\t.long\t%d\t\t\t\t\t/* last_comp_version */\n",

		vi->last_comp_version);

	if (vi->flags & FTF_BOOTCPUID)

		fprintf(f, "\t.long\t%i\t\t\t\t\t/* boot_cpuid_phys */\n",

			bi->boot_cpuid_phys);

	if (vi->flags & FTF_STRTABSIZE)

		fprintf(f, "\t.long\t_%s_strings_end - _%s_strings_start\t/* size_dt_strings */\n",

			symprefix, symprefix);

	if (vi->flags & FTF_STRUCTSIZE)

		fprintf(f, "\t.long\t_%s_struct_end - _%s_struct_start\t/* size_dt_struct */\n",

			symprefix, symprefix);

	/*

	 * Reserve map entries.

	 * Align the reserve map to a doubleword boundary.

	 * Each entry is an (address, size) pair of u64 values.

	 * Always supply a zero-sized temination entry.

	 */

	asm_emit_align(f, 8);

	emit_label(f, symprefix, "reserve_map");

	fprintf(f, "/* Memory reserve map from source file */\n");

	/*

	 * Use .long on high and low halfs of u64s to avoid .quad

	 * as it appears .quad isn't available in some assemblers.

	 */

	for (re = bi->reservelist; re; re = re->next) {

		if (re->label) {

			fprintf(f, "\t.globl\t%s\n", re->label);

			fprintf(f, "%s:\n", re->label);

		}

		fprintf(f, "\t.long\t0x%08x, 0x%08x\n",

			(unsigned int)(re->re.address >> 32),

			(unsigned int)(re->re.address & 0xffffffff));

		fprintf(f, "\t.long\t0x%08x, 0x%08x\n",

			(unsigned int)(re->re.size >> 32),

			(unsigned int)(re->re.size & 0xffffffff));

	}

	for (i = 0; i < reservenum; i++) {

		fprintf(f, "\t.long\t0, 0\n\t.long\t0, 0\n");

	}

	fprintf(f, "\t.long\t0, 0\n\t.long\t0, 0\n");