/* $NetBSD: keylock.c,v 1.3 2014/02/25 18:30:09 pooka Exp $ */

/*
 * Copyright (c) 2009 Marc Balmer <marc@msys.ch>
 * 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 THE AUTHOR ``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 THE AUTHOR 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.
 */

#include "opt_secmodel_keylock.h"

/* Support for multi-position electro-mechanical keylocks */

#include <sys/param.h>
#include <sys/kernel.h>
#include <sys/sysctl.h>

#include <dev/keylock.h>

#ifdef secmodel_keylock
#include <sys/kauth.h>
#include <secmodel/keylock/keylock.h>
#endif

static int (*keylock_pos_cb)(void *) = NULL;
static void *keylock_pos_cb_arg = NULL;
static int keylock_npos = 0;
static int keylock_order = 0;

int keylock_pos_sysctl(SYSCTLFN_PROTO);
int keylock_state_sysctl(SYSCTLFN_PROTO);
int keylock_order_sysctl(SYSCTLFN_PROTO);

SYSCTL_SETUP(sysctl_keylock_setup, "sysctl keylock setup")
{
        const struct sysctlnode *node = NULL;

        sysctl_createv(clog, 0, NULL, &node,
            CTLFLAG_PERMANENT,
            CTLTYPE_NODE, "keylock",
            SYSCTL_DESCR("Keylock state"),
            NULL, 0, NULL, 0,
            CTL_HW, CTL_CREATE, CTL_EOL);

        if (node == NULL)
                return;

        sysctl_createv(clog, 0, &node, NULL,
            CTLFLAG_PERMANENT | CTLFLAG_READONLY,
            CTLTYPE_INT, "pos",
            SYSCTL_DESCR("Current keylock position"),
            keylock_pos_sysctl, 0, NULL, 0,
            CTL_CREATE, CTL_EOL);
        sysctl_createv(clog, 0, &node, NULL,
            CTLFLAG_PERMANENT | CTLFLAG_READONLY,
            CTLTYPE_INT, "npos",
            SYSCTL_DESCR("Number of keylock positions"),
            NULL, 0, &keylock_npos, 0,
            CTL_CREATE, CTL_EOL);
        sysctl_createv(clog, 0, &node, NULL,
            CTLFLAG_PERMANENT | CTLFLAG_READONLY,
            CTLTYPE_INT, "state",
            SYSCTL_DESCR("Keylock state"),
            keylock_state_sysctl, 0, NULL, 0,
            CTL_CREATE, CTL_EOL);
       sysctl_createv(clog, 0, &node, NULL,
            CTLFLAG_PERMANENT | CTLFLAG_READWRITE, 
            CTLTYPE_INT, "order", 
            SYSCTL_DESCR("Keylock closedness order"),
            keylock_order_sysctl, 0, NULL, 0,
            CTL_CREATE, CTL_EOL);
}

int
keylock_register(void *cb_arg, int npos, int (*cb)(void *))
{
        if (keylock_pos_cb != NULL)
                return -1;

        keylock_pos_cb = cb;
        keylock_pos_cb_arg = cb_arg;
        keylock_npos = npos;
#ifdef secmodel_keylock
        secmodel_keylock_start();
#endif
        return 0;
}

void
keylock_unregister(void *cb_arg, int (*cb)(void *))
{
        if (keylock_pos_cb != cb || keylock_pos_cb_arg != cb_arg)
                return;

#ifdef secmodel_keylock
        secmodel_keylock_stop();
#endif
        keylock_pos_cb = NULL;
        keylock_pos_cb_arg = NULL;
        keylock_npos = 0;
}

int
keylock_position(void)
{
        if (keylock_pos_cb == NULL)
                return 0;

        return (*keylock_pos_cb)(keylock_pos_cb_arg);
}

int
keylock_num_positions(void)
{
        return keylock_npos;
}

int
keylock_state(void)
{
        int pos;

        if (keylock_npos == 0)
                return KEYLOCK_ABSENT;

        pos = keylock_position();
        if (pos == 0)
                return KEYLOCK_TAMPER;

        /*
         * XXX How should the intermediate positions be handled?
         * At the moment only the ultimate positions are properly handled,
         * we need to think about what we do with the intermediate positions.
         * For now we return KEYLOCK_SEMIOPEN for them.
         */
        if (pos == 1)
                return keylock_order == 0 ? KEYLOCK_CLOSE : KEYLOCK_OPEN;
        else if (pos == keylock_npos)
                return keylock_order == 0 ? KEYLOCK_OPEN : KEYLOCK_CLOSE;
        return KEYLOCK_SEMIOPEN;
}

int
keylock_pos_sysctl(SYSCTLFN_ARGS)
{
        struct sysctlnode node;
        int val;

        node = *rnode;
        node.sysctl_data = &val;

        val = keylock_position();
        return sysctl_lookup(SYSCTLFN_CALL(&node));
}

int
keylock_state_sysctl(SYSCTLFN_ARGS)
{
        struct sysctlnode node;
        int val;

        node = *rnode;
        node.sysctl_data = &val;

        val = keylock_state();
        return sysctl_lookup(SYSCTLFN_CALL(&node));
}

int
keylock_order_sysctl(SYSCTLFN_ARGS)
{
        struct sysctlnode node;
        int val, error;

        node = *rnode;
        node.sysctl_data = &val;

        val = keylock_order;
        error = sysctl_lookup(SYSCTLFN_CALL(&node));
        if (error || newp == NULL)
                return error;
        if (keylock_state() != KEYLOCK_OPEN)
                return -1;

        keylock_order = val;
        return 0;
}