/*
 * 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 <c2/audit.h>
#include <c2/audit_kernel.h>
#include <c2/audit_record.h>
#include <sys/kmem.h>
#include <sys/param.h>
#include <sys/systm.h>
#include <sys/taskq.h>
#include <sys/t_lock.h>
#include <sys/thread.h>
#include <sys/types.h>
#include <sys/zone.h>

zone_key_t au_zone_key;

/*ARGSUSED*/
static void *
au_zone_init(zoneid_t zone)
{
        au_kcontext_t   *kctx = kmem_zalloc(sizeof (au_kcontext_t), KM_SLEEP);
        static au_kcontext_t    *global_kctx = NULL;

        /*
         * INGLOBALZONE(curproc) is invalid at this point, so check for
         * zone 0
         */

        if (zone == 0) {
                global_kctx = kctx;
                global_zone->zone_audit_kctxt = kctx;
        } else {
                kctx->auk_policy = global_kctx->auk_policy;
                curproc->p_zone->zone_audit_kctxt = kctx;
        }
        kctx->auk_valid = AUK_VALID;
        kctx->auk_zid = zone;

        kctx->auk_info.ai_termid.at_type = AU_IPv4;
        kctx->auk_info.ai_auid = AU_NOAUDITID;
        kctx->auk_auditstate = AUC_INIT_AUDIT;

        /* setup defaults for audit queue flow control */
        kctx->auk_queue.hiwater = AQ_HIWATER;
        kctx->auk_queue.lowater = AQ_LOWATER;
        kctx->auk_queue.bufsz   = AQ_BUFSZ;
        kctx->auk_queue.buflen  = AQ_BUFSZ;
        kctx->auk_queue.delay   = AQ_DELAY;

        /* statistics per zone */
        kctx->auk_statistics.as_version  = TOKEN_VERSION;
        kctx->auk_statistics.as_numevent = MAX_KEVENTS;

        /* door IO buffer: */
        kctx->auk_dbuffer =
            kmem_alloc(AU_DBUF_HEADER + kctx->auk_queue.bufsz, KM_SLEEP);

        /* locks and cv's */

        mutex_init(&(kctx->auk_eagain_mutex), NULL, MUTEX_DEFAULT, NULL);
        cv_init(&(kctx->auk_eagain_cv), NULL, CV_DRIVER, NULL);

        mutex_init(&(kctx->auk_svc_lock), NULL, MUTEX_DEFAULT, NULL);

        mutex_init(&(kctx->auk_queue.lock), NULL, MUTEX_DEFAULT, NULL);
        cv_init(&(kctx->auk_queue.write_cv), NULL, CV_DRIVER, NULL);
        cv_init(&(kctx->auk_queue.read_cv), NULL, CV_DRIVER, NULL);

        return (kctx);
}

/*ARGSUSED*/
static void
au_zone_shutdown(zoneid_t zone, void *arg)
{
        au_kcontext_t   *kctx = arg;

        if (audit_active == C2AUDIT_LOADED && (kctx->auk_zid == GLOBAL_ZONEID ||
            (audit_policy | AUDIT_PERZONE)) && (kctx->auk_current_vp != NULL))
                (void) au_doormsg(kctx, AU_DBUF_SHUTDOWN, NULL);

        kctx->auk_valid = AUK_INVALID;

        /* shutdown the output thread if it is still running */
        kctx->auk_auditstate = AUC_NOAUDIT;

        if (kctx->auk_output_active) {
                mutex_enter(&(kctx->auk_queue.lock));
                cv_broadcast(&(kctx->auk_queue.read_cv));
                mutex_exit(&(kctx->auk_queue.lock));

                taskq_destroy(kctx->auk_taskq);
        }
}

/*ARGSUSED*/
static void
au_zone_destroy(zoneid_t zone, void *arg)
{
        au_kcontext_t   *kctx = arg;

        ASSERT(kctx->auk_auditstate == AUC_NOAUDIT);

        mutex_destroy(&(kctx->auk_eagain_mutex));
        cv_destroy(&(kctx->auk_eagain_cv));

        mutex_destroy(&(kctx->auk_svc_lock));

        mutex_enter(&(kctx->auk_queue.lock));
        if (kctx->auk_queue.head != NULL) {
                au_free_rec(kctx->auk_queue.head);
        }
        mutex_exit(&(kctx->auk_queue.lock));

        mutex_destroy(&(kctx->auk_queue.lock));

        cv_destroy(&(kctx->auk_queue.write_cv));
        cv_destroy(&(kctx->auk_queue.read_cv));

        kmem_free(kctx->auk_dbuffer, AU_DBUF_HEADER + kctx->auk_queue.buflen);

        kmem_free(kctx, sizeof (au_kcontext_t));
}

void
au_zone_setup()
{
        zone_key_create(&au_zone_key, au_zone_init, au_zone_shutdown,
            au_zone_destroy);

}

int
au_zone_getstate(const au_kcontext_t *context)
{
        au_kcontext_t *tcontext;

        if (context != NULL)
                return (context->auk_auditstate);
        tcontext = GET_KCTX_PZ;
        return (tcontext->auk_auditstate);
}