/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright 2010 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */


#include "benv.h"
#include <sys/sunddi.h>
#include <sys/ddi_impldefs.h>
#include <sys/openpromio.h>
#include <stdio.h>

static int getpropval(struct openpromio *opp, char *prop);

static char *promdev = "/dev/openprom";
static int prom_fd;
static char *mfail = "malloc";

/*
 * 128 is the size of the largest (currently) property name
 * 16384 - MAXPROPSIZE - sizeof (int) is the size of the largest
 * (currently) property value that is allowed.
 * the sizeof (u_int) is from struct openpromio
 */

#define MAXPROPSIZE     128
#define MAXVALSIZE      (16384 - MAXPROPSIZE - sizeof (u_int))
#define BUFSIZE         (MAXPROPSIZE + MAXVALSIZE + sizeof (u_int))
#define MINVALSIZE      (4 * sizeof (u_long))
#define MINBUFSIZE      (MINVALSIZE + sizeof (u_long))

typedef union {
        char buf[BUFSIZE];
        struct openpromio opp;
} Oppbuf;

typedef union {
        char buf[MINVALSIZE + sizeof (u_int)];
        struct openpromio opp;
} Oppbuf_small;

static Oppbuf   oppbuf;

static unsigned long
next(unsigned long id)
{
        Oppbuf_small    oppbuf;
        struct openpromio *opp = &(oppbuf.opp);
        unsigned long *ip = (unsigned long *)(opp->oprom_array);

        memset(oppbuf.buf, 0, MINBUFSIZE);
        opp->oprom_size = MINVALSIZE;
        *ip = id;
        if (ioctl(prom_fd, OPROMNEXT, opp) < 0)
                return (0);
        return (*(unsigned long *)opp->oprom_array);
}

static unsigned long
child(unsigned long id)
{
        Oppbuf_small    oppbuf;
        struct openpromio *opp = &(oppbuf.opp);
        unsigned long *ip = (unsigned long *)(opp->oprom_array);

        memset(oppbuf.buf, 0, MINBUFSIZE);
        opp->oprom_size = MINVALSIZE;
        *ip = id;
        if (ioctl(prom_fd, OPROMCHILD, opp) < 0)
                return (0);
        return (*(unsigned long *)opp->oprom_array);
}

/*
 * Find a node by name from the prom device tree.
 * Return the id or 0 if it is not found.
 */
static unsigned long
prom_findnode_byname(unsigned long id, char *name)
{
        struct openpromio *opp = &(oppbuf.opp);
        unsigned long nid;

        if (id == 0)
                return (0);
        if (!getpropval(opp, "name"))
                return (0);
        if (strcmp(opp->oprom_array, name) == 0)
                return (id);
        if (nid = prom_findnode_byname(child(id), name))
                return (nid);
        if (nid = prom_findnode_byname(next(id), name))
                return (nid);
        return (0);
}

/*
 * Make the current prom node be the rootnode and return its id.
 */
static unsigned long
prom_rootnode()
{
        return (next(0));
}

static int
getpropval(struct openpromio *opp, char *prop)
{
        opp->oprom_size = MAXVALSIZE;

        (void) strlcpy(opp->oprom_array, prop, MAXPROPSIZE);
        if (ioctl(prom_fd, OPROMGETPROP, opp) < 0)
                return (0);
        if (opp->oprom_size == 0)
                return (0);
        return (1);
}

static int
getnextprop(struct openpromio *opp, char *prop)
{
        opp->oprom_size = MAXVALSIZE;

        (void) strlcpy(opp->oprom_array, prop, MAXPROPSIZE);
        if (ioctl(prom_fd, OPROMNXTPROP, opp) < 0)
                return (0);
        if (opp->oprom_size == 0)
                return (0);
        return (1);
}

char *
getbootcmd(void)
{
        struct openpromio *opp = &(oppbuf.opp);
        opp->oprom_size = MAXVALSIZE;
        if (ioctl(prom_fd, OPROMGETBOOTPATH, opp) < 0)
                return (NULL);
        return (opp->oprom_array);
}

/*
 * Get a pointer to the requested property from the current node.
 * The property is stored in static storage and the returned pointer
 * points into the static storage.  The property length is placed in
 * the location pointed to by the third argument.
 */
static unsigned char *
prom_getprop(char *prop, int *lenp)
{
        struct openpromio *opp = &(oppbuf.opp);

        if (!getpropval(opp, prop))
                return (NULL);
        *lenp = opp->oprom_size;
        return ((unsigned char *)opp->oprom_array);
}

static unsigned char *
prom_nextprop(char *prop)
{
        struct openpromio *opp = &(oppbuf.opp);

        if (!getnextprop(opp, prop))
                return ((unsigned char *)0);
        return ((unsigned char *)opp->oprom_array);
}

ddi_prop_t *
get_proplist(char *name)
{
        ddi_prop_t *plist, *npp, *plast;
        char *curprop, *newprop;
        unsigned char *propval;
        unsigned long id;

        plist = NULL;
        plast = NULL;
        id = prom_findnode_byname(prom_rootnode(), name);
        if (id == 0)
                return (plist);
        curprop = "";
        while (newprop = (char *)prom_nextprop(curprop)) {
                curprop = strdup(newprop);
                npp = (ddi_prop_t *)malloc(sizeof (ddi_prop_t));
                if (npp == 0)
                        exit(_error(PERROR, mfail));
                propval = prom_getprop(curprop, &npp->prop_len);
                npp->prop_name = curprop;
                if (propval != NULL) {
                        npp->prop_val = (char *)malloc(npp->prop_len);
                        if (npp->prop_val == 0)
                                exit(_error(PERROR, mfail));
                        memcpy(npp->prop_val, propval, npp->prop_len);
                } else
                        npp->prop_val = NULL;
                npp->prop_next = NULL;
                if (plast == NULL) {
                        plist = npp;
                } else {
                        plast->prop_next = npp;
                }
                plast = npp;
        }
        return (plist);
}

caddr_t
get_propval(char *name, char *node)
{
        ddi_prop_t *prop, *plist;

        if ((plist = get_proplist(node)) == NULL)
                return (NULL);

        for (prop = plist; prop != NULL; prop = prop->prop_next)
                if (strcmp(prop->prop_name, name) == 0)
                        return (prop->prop_val);

        return (NULL);
}

void
get_kbenv(void)
{
        if ((prom_fd = open(promdev, O_RDONLY)) < 0) {
                exit(_error(PERROR, "prom open failed"));
        }
}

void
close_kbenv(void)
{
        (void) close(prom_fd);
}