/*	$NetBSD: Locore.c,v 1.7 2000/08/20 07:04:59 tsubai Exp $	*/

/*
 * Copyright (C) 1995, 1996 Wolfgang Solfrank.
 * Copyright (C) 1995, 1996 TooLs GmbH.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *	This product includes software developed by TooLs GmbH.
 * 4. The name of TooLs GmbH may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY TOOLS GMBH ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL TOOLS GMBH BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
/*
 * Copyright (C) 2000 Benno Rice.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY Benno Rice ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL TOOLS GMBH BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * $FreeBSD: head/sys/dev/ofw/openfirm.c 86557 2001-11-18 20:38:44Z tmm $
 */

#include <sys/param.h>
#include <sys/kernel.h>
#include <sys/malloc.h>
#include <sys/systm.h>

#include <machine/stdarg.h>

#include <dev/ofw/openfirm.h>

MALLOC_DEFINE(M_OFWPROP, "openfirm", "OpenFirmware properties");

static ihandle_t stdin;
static ihandle_t stdout;

char	*OF_buf;

void	ofbcopy(const void *, void *, size_t);

/* Initialiaser */

void
OF_init(int (*openfirm)(void *))
{
	phandle_t	chosen;

	set_openfirm_callback(openfirm);
	if ((chosen = OF_finddevice("/chosen")) == -1)
		OF_exit();
	OF_getprop(chosen, "stdin", &stdin, sizeof(stdin));
	OF_getprop(chosen, "stdout", &stdout, sizeof(stdout));
}

void
OF_printf(const char *fmt, ...)
{
	va_list	va;
	char	buf[1024];

	va_start(va, fmt);
	vsprintf(buf, fmt, va);
	OF_write(stdout, buf, strlen(buf));
	va_end(va);
}

/*
 * Generic functions
 */

/* Test to see if a service exists. */
int
OF_test(char *name)
{
	static struct {
		cell_t	name;
		cell_t	nargs;
		cell_t	nreturns;
		cell_t	service;
		cell_t	missing;
	} args = {
		(cell_t)"test",
		1,
		1,
	};

	args.service = (cell_t)name;
	if (openfirmware(&args) == -1)
		return -1;
	return args.missing;
}

/* Return firmware millisecond count. */
int
OF_milliseconds()
{
	static struct {
		cell_t	name;
		cell_t	nargs;
		cell_t	nreturns;
		cell_t	ms;
	} args = {
		(cell_t)"milliseconds",
		0,
		1,
	};
	
	openfirmware(&args);
	return args.ms;
}

/*
 * Device tree functions
 */

/* Return the next sibling of this node or 0. */
phandle_t
OF_peer(phandle_t node)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		node;
		cell_t		next;
	} args = {
		(cell_t)"peer",
		1,
		1,
	};

	args.node = (cell_t)node;
	if (openfirmware(&args) == -1)
		return -1;
	return args.next;
}

/* Return the first child of this node or 0. */
phandle_t
OF_child(phandle_t node)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		node;
		cell_t		child;
	} args = {
		(cell_t)"child",
		1,
		1,
	};

	args.node = (cell_t)node;
	if (openfirmware(&args) == -1)
		return -1;
	return args.child;
}

/* Return the parent of this node or 0. */
phandle_t
OF_parent(phandle_t node)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		node;
		cell_t		parent;
	} args = {
		(cell_t)"parent",
		1,
		1,
	};

	args.node = (cell_t)node;
	if (openfirmware(&args) == -1)
		return -1;
	return args.parent;
}

/* Return the package handle that corresponds to an instance handle. */
phandle_t
OF_instance_to_package(ihandle_t instance)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		instance;
		cell_t		package;
	} args = {
		(cell_t)"instance-to-package",
		1,
		1,
	};
	
	args.instance = (cell_t)instance;
	if (openfirmware(&args) == -1)
		return -1;
	return args.package;
}

/* Get the length of a property of a package. */
int
OF_getproplen(phandle_t package, char *propname)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		package;
		cell_t 		propname;
		cell_t		proplen;
	} args = {
		(cell_t)"getproplen",
		2,
		1,
	};

	args.package = (cell_t)package;
	args.propname = (cell_t)propname;
	if (openfirmware(&args) == -1)
		return -1;
	return args.proplen;
}

/* Get the value of a property of a package. */
int
OF_getprop(phandle_t package, char *propname, void *buf, int buflen)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		package;
		cell_t		propname;
		cell_t		buf;
		cell_t		buflen;
		cell_t		size;
	} args = {
		(cell_t)"getprop",
		4,
		1,
	};
	
	args.package = (cell_t)package;
	args.propname = (cell_t)propname;
	args.buf = (cell_t)buf;
	args.buflen = (cell_t)buflen;
	if (openfirmware(&args) == -1)
		return -1;
	return args.size;
}

/*
 * Store the value of a property of a package into newly allocated memory (using
 * the M_OFWPROP malloc pool and M_WAITOK). elsz is the size of a single element,
 * the number of elements is return in number.
 */
int
OF_getprop_alloc(phandle_t package, char *propname, int elsz, void **buf)
{
	int len;

	*buf = NULL;
	if ((len = OF_getproplen(package, propname)) == -1 ||
	    len % elsz != 0)
		return (-1);

	*buf = malloc(len, M_OFWPROP, M_WAITOK);
	if (OF_getprop(package, propname, *buf, len) == -1) {
		free(*buf, M_OFWPROP);
		*buf = NULL;
		return (-1);
	}
	return (len / elsz);
}

/* Get the next property of a package. */
int
OF_nextprop(phandle_t package, char *previous, char *buf)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		package;
		cell_t 		previous;
		cell_t		buf;
		cell_t		flag;
	} args = {
		(cell_t)"nextprop",
		3,
		1,
	};

	args.package = (cell_t)package;
	args.previous = (cell_t)previous;
	args.buf = (cell_t)buf;
	if (openfirmware(&args) == -1)
		return -1;
	return args.flag;
}

/* Set the value of a property of a package. */
/* XXX Has a bug on FirePower */
int
OF_setprop(phandle_t package, char *propname, void *buf, int len)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		package;
		cell_t		propname;
		cell_t		buf;;
		cell_t		len;
		cell_t		size;
	} args = {
		(cell_t)"setprop",
		4,
		1,
	};
	
	args.package = (cell_t)package;
	args.propname = (cell_t)propname;
	args.buf = (cell_t)buf;
	args.len = (cell_t)len;
	if (openfirmware(&args) == -1)
		return -1;
	return args.size;
}

/* Convert a device specifier to a fully qualified pathname. */
int
OF_canon(const char *device, char *buf, int len)
{
	static struct {
		cell_t	name;
		cell_t	nargs;
		cell_t	nreturns;
		cell_t	device;
		cell_t	buf;
		cell_t	len;
		cell_t	size;
	} args = {
		(cell_t)"canon",
		3,
		1,
	};
	
	args.device = (cell_t)device;
	args.buf = (cell_t)buf;
	args.len = (cell_t)len;
	if (openfirmware(&args) == -1)
		return -1;
	return args.size;
}

/* Return a package handle for the specified device. */
phandle_t
OF_finddevice(const char *device)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		device;
		cell_t		package;
	} args = {
		(cell_t)"finddevice",
		1,
		1,
	};	
	
	args.device = (cell_t)device;
	if (openfirmware(&args) == -1)
		return -1;
	return args.package;
}

/* Return the fully qualified pathname corresponding to an instance. */
int
OF_instance_to_path(ihandle_t instance, char *buf, int len)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		instance;
		cell_t		buf;
		cell_t		len;
		cell_t		size;
	} args = {
		(cell_t)"instance-to-path",
		3,
		1,
	};

	args.instance = (cell_t)instance;
	args.buf = (cell_t)buf;
	args.len = (cell_t)len;
	if (openfirmware(&args) == -1)
		return -1;
	return(args.size);
}

/* Return the fully qualified pathname corresponding to a package. */
int
OF_package_to_path(phandle_t package, char *buf, int len)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		package;
		cell_t		buf;
		cell_t		len;
		cell_t		size;
	} args = {
		(cell_t)"package-to-path",
		3,
		1,
	};

	args.package = (cell_t)package;
	args.buf = (cell_t)buf;
	args.len = (cell_t)len;
	if (openfirmware(&args) == -1)
		return -1;
	return(args.size);
}

/*  Call the method in the scope of a given instance. */
int
OF_call_method(char *method, ihandle_t instance, int nargs, int nreturns, ...)
{
	va_list ap;
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		method;
		cell_t		instance;
		cell_t		args_n_results[12];
	} args = {
		(cell_t)"call-method",
		2,
		1,
	};
	cell_t *ip;
	int n;

	if (nargs > 6)
		return -1;
	args.nargs = (cell_t)nargs + 2;
	args.nreturns = (cell_t)nreturns + 1;
	args.method = (cell_t)method;
	args.instance = (cell_t)instance;
	va_start(ap, nreturns);
	for (ip = args.args_n_results + (n = (cell_t)nargs); --n >= 0;)
		*--ip = va_arg(ap, cell_t);

	if (openfirmware(&args) == -1)
		return -1;
	if (args.args_n_results[nargs])
		return args.args_n_results[nargs];
	for (ip = args.args_n_results + nargs + (n = args.nreturns); --n > 0;)
		*va_arg(ap, cell_t *) = *--ip;
	va_end(ap);
	return 0;
}

/*
 * Device I/O functions.
 */

/* Open an instance for a device. */
ihandle_t
OF_open(char *device)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		device;
		cell_t		instance;
	} args = {
		(cell_t)"open",
		1,
		1,
	};
	
	args.device = (cell_t)device;
	if (openfirmware(&args) == -1 || args.instance == 0) {
		return -1;
	}
	return args.instance;
}

/* Close an instance. */
void
OF_close(ihandle_t instance)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		instance;
	} args = {
		(cell_t)"close",
		1,
		0,
	};
	
	args.instance = (cell_t)instance;
	openfirmware(&args);
}

/* Read from an instance. */
int
OF_read(ihandle_t instance, void *addr, int len)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		instance;
		cell_t		addr;
		cell_t		len;
		cell_t		actual;
	} args = {
		(cell_t)"read",
		3,
		1,
	};

	args.instance = (cell_t)instance;
	args.addr = (cell_t)addr;
	args.len = (cell_t)len;
	if (openfirmware(&args) == -1)
		return -1;

	return args.actual;
}

/* Write to an instance. */
int
OF_write(ihandle_t instance, void *addr, int len)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		instance;
		cell_t		addr;
		cell_t		len;
		cell_t		actual;
	} args = {
		(cell_t)"write",
		3,
		1,
	};

	args.instance = (cell_t)instance;
	args.addr = (cell_t)addr;
	args.len = (cell_t)len;
	if (openfirmware(&args) == -1)
		return -1;
	return args.actual;
}

/* Seek to a position. */
int
OF_seek(ihandle_t instance, u_int64_t pos)
{
	static struct {
		cell_t		name;
		cell_t		nargs;
		cell_t		nreturns;
		cell_t		instance;
		cell_t		poshi;
		cell_t		poslo;
		cell_t		status;
	} args = {
		(cell_t)"seek",
		3,
		1,
	};
	
	args.instance = (cell_t)instance;
	args.poshi = (cell_t)(pos >> 32);
	args.poslo = (cell_t)pos;
	if (openfirmware(&args) == -1)
		return -1;
	return args.status;
}

/*
 * Memory functions.
 */

/* Claim an area of memory. */
void *
OF_claim(void *virt, u_int size, u_int align)
{
	static struct {
		cell_t	 name;
		cell_t	 nargs;
		cell_t	 nreturns;
		cell_t	 virt;
		cell_t	 size;
		cell_t	 align;
		cell_t	 baseaddr;
	} args = {
		(cell_t)"claim",
		3,
		1,
	};

	args.virt = (cell_t)virt;
	args.size = (cell_t)size;
	args.align = (cell_t)align;
	if (openfirmware(&args) == -1)
		return (void *)-1;
	return (void *)args.baseaddr;
}

/* Release an area of memory. */
void
OF_release(void *virt, u_int size)
{
	static struct {
		cell_t	name;
		cell_t	nargs;
		cell_t	nreturns;
		cell_t	virt;
		cell_t	size;
	} args = {
		(cell_t)"release",
		2,
		0,
	};
	
	args.virt = (cell_t)virt;
	args.size = (cell_t)size;
	openfirmware(&args);
}

/*
 * Control transfer functions.
 */

/* Reset the system and call "boot <bootspec>". */
void
OF_boot(char *bootspec)
{
	static struct {
		cell_t	name;
		cell_t	nargs;
		cell_t	nreturns;
		cell_t	bootspec;
	} args = {
		(cell_t)"boot",
		1,
		0,
	};

	args.bootspec = (cell_t)bootspec;
	openfirmware(&args);
	for (;;);			/* just in case */
}

/* Suspend and drop back to the OpenFirmware interface. */
void
OF_enter()
{
	static struct {
		cell_t name;
		cell_t nargs;
		cell_t nreturns;
	} args = {
		(cell_t)"enter",
		0,
		0
	};

	openfirmware(&args);
	return;				/* We may come back. */
}

/* Shut down and drop back to the OpenFirmware interface. */
void
OF_exit()
{
	static struct {
		cell_t name;
		cell_t nargs;
		cell_t nreturns;
	} args = {
		(cell_t)"exit",
		0,
		0
	};

	openfirmware(&args);
	for (;;);			/* just in case */
}

/* Free <size> bytes starting at <virt>, then call <entry> with <arg>. */
#ifdef	__notyet__
void
OF_chain(void *virt, u_int size, void (*entry)(), void *arg, u_int len)
{
	static struct {
		cell_t	name;
		cell_t	nargs;
		cell_t	nreturns;
		cell_t	virt;
		cell_t	size;
		call_t	entry;
		cell_t	arg;
		cell_t	len;
	} args = {
		(cell_t)"chain",
		5,
		0,
	};

	args.virt = (cell_t)virt;
	args.size = (cell_t)size;
	args.entry = (cell_t)entry;
	args.arg = (cell_t)arg;
	args.len = (cell_t)len;
	openfirmware(&args);
}
#else
void
OF_chain(void *virt, u_int size,
    void (*entry)(void *, u_int, void *, void *, u_int), void *arg, u_int len)
{
	/*
	 * This is a REALLY dirty hack till the firmware gets this going
	 */
#if 0
	OF_release(virt, size);
#endif
	entry(0, 0, openfirmware, arg, len);
}
#endif

void
ofbcopy(const void *src, void *dst, size_t len)
{
	const char *sp = src;
	char *dp = dst;

	if (src == dst)
		return;

	while (len-- > 0)
		*dp++ = *sp++;
}