/*
 * 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 2009 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*
 * This file contains all the functions that get/set fields
 * in a GRUB menu entry.
 */
#include <stdio.h>
#include <errno.h>
#include <stdlib.h>
#include <strings.h>
#include <unistd.h>
#include <sys/types.h>
#include <assert.h>
#include <ctype.h>

#include "libgrub_cmd.def"
#include "libgrub_impl.h"

typedef int (*barg_parsef_t)(const grub_line_t *, grub_barg_t *);
static const  barg_parsef_t barg_parse[] = {
#define menu_cmd(cmd, num, flag, parsef)        parsef,
#include "libgrub_cmd.def"
};

/*
 * Remove extra '/', stops at first isspace character.
 * Return new string length.
 */
size_t
clean_path(char *path)
{
        int     i, c;
        size_t  k, n;

        n = strlen(path) + 1;

        for (i = 0; (c = path[i]) != 0 && !isspace(c); i++) {
                if (c == '/' && (k = strspn(path + i, "/") - 1) != 0) {
                        /* bcopy should deal with overlapping buffers */
                        n -= k;
                        bcopy(path + i + k, path + i, n - i);
                }
        }
        return (n - 1);
}

/*
 * Construct boot command line from the ge_barg field
 */
static size_t
barg_cmdline(const grub_barg_t *barg, char *cmd, size_t size)
{
        size_t n;
        const grub_fsdesc_t *fsd;

        if (!IS_BARG_VALID(barg) ||
            (fsd = grub_get_rootfsd(&barg->gb_root)) == NULL)
                return ((size_t)-1);

        /* if disk/top dataset is mounted, use mount point */
        if (fsd->gfs_mountp[0] != 0) {
                if ((n = snprintf(cmd, size, "%s%s", fsd->gfs_mountp,
                    barg->gb_kernel)) >= size)
                        return (n);
                return (clean_path(cmd));
        } else
                return (snprintf(cmd, size, "%s %s", fsd->gfs_dev,
                    barg->gb_kernel));
}


/*
 * Construct ge_barg field based on the other fields of the entry.
 * Return 0 on success, errno on failure.
 */
int
grub_entry_construct_barg(grub_entry_t *ent)
{
        int ret = 0;
        grub_barg_t *barg;
        grub_line_t *lp, *lend;
        grub_menu_t *mp;

        assert(ent);

        barg = &ent->ge_barg;
        mp = ent->ge_menu;

        assert(barg);
        assert(mp);

        (void) memset(barg, 0, sizeof (*barg));
        barg->gb_entry = ent;
        (void) bcopy(&mp->gm_root, &barg->gb_root, sizeof (barg->gb_root));

        lend = ent->ge_end->gl_next;
        for (lp = ent->ge_start; lp != lend; lp = lp->gl_next) {
                if (lp->gl_cmdtp >= GRBM_CMD_NUM)
                        ret = EG_INVALIDCMD;
                else
                        ret = barg_parse[lp->gl_cmdtp](lp, barg);

                if (ret != 0)
                        break;
        }

        barg->gb_errline = lp;
        if (ret == 0) {
                /* at least kernel and module should be defined */
                if (barg->gb_kernel[0] != 0 && barg->gb_module[0] != 0)
                        barg->gb_flags |= GRBM_VALID_FLAG;
        }

        return (ret);
}

const char *
grub_entry_get_fstyp(const grub_entry_t *ent)
{
        if (IS_ENTRY_BARG_VALID(ent))
                return (ent->ge_barg.gb_root.gr_fstyp);
        else
                return (NULL);
}

const char *
grub_entry_get_kernel(const grub_entry_t *ent)
{
        if (IS_ENTRY_BARG_VALID(ent))
                return (ent->ge_barg.gb_kernel);
        else
                return (NULL);
}

const char *
grub_entry_get_module(const grub_entry_t *ent)
{
        if (IS_ENTRY_BARG_VALID(ent))
                return (ent->ge_barg.gb_module);
        else
                return (NULL);
}

const char *
grub_entry_get_error_desc(const grub_entry_t *ent)
{
        assert(ent != NULL);
        return ("Not implemented");
}

const grub_fsdesc_t *
grub_entry_get_rootfs(const grub_entry_t *ent)
{
        if (IS_ENTRY_BARG_VALID(ent))
                return (grub_get_rootfsd(&ent->ge_barg.gb_root));
        else
                return (NULL);
}

size_t
grub_entry_get_cmdline(grub_entry_t *ent, char *cmdline, size_t size)
{
        if (IS_ENTRY_VALID(ent) && (grub_entry_construct_barg(ent) == 0))
                return (barg_cmdline(&ent->ge_barg, cmdline, size));
        else
                return ((size_t)-1);

}