/* -*- mode: c; c-basic-offset: 4; indent-tabs-mode: nil -*- */

/* plugins/kadm5_auth/test/main.c - test modules for kadm5_auth interface */

/*

 * Copyright (C) 2017 by the Massachusetts Institute of Technology.

 * All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 *

 * * Redistributions of source code must retain the above copyright

 *   notice, this list of conditions and the following disclaimer.

 *

 * * 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 COPYRIGHT HOLDERS AND CONTRIBUTORS

 * "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

 * COPYRIGHT HOLDER OR CONTRIBUTORS 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.

 */

/*

 * This file implements two testing kadm5_auth modules, the welcomer and the

 * bouncer.  The welcomer implements permissive behavior, while the bouncer

 * implements restrictive behavior.

 *

 * Module data objects and restrictions are adequately tested by the acl

 * module, so we do not test them here.  Focus instead on the ability to

 * examine principal and policy objects and to perform DB operations.

 */

#include "k5-int.h"

#include <kadm5/admin.h>

#include <krb5/kadm5_auth_plugin.h>

krb5_error_code

kadm5_auth_welcomer_initvt(krb5_context context, int maj_ver, int min_ver,

                           krb5_plugin_vtable vtable);

krb5_error_code

kadm5_auth_bouncer_initvt(krb5_context context, int maj_ver, int min_ver,

                          krb5_plugin_vtable vtable);

/* The welcomer authorizes all getprinc operations, since kadmin uses them as a

 * precursor to modprinc. */

static krb5_error_code

welcomer_getprinc(krb5_context context, kadm5_auth_moddata data,

                  krb5_const_principal client, krb5_const_principal target)

{

    return 0;

}

/* The welcomer authorizes addprinc operations which set a policy "VIP". */

static krb5_error_code

welcomer_addprinc(krb5_context context, kadm5_auth_moddata data,

                  krb5_const_principal client, krb5_const_principal target,

                  const struct _kadm5_principal_ent_t *ent, long mask,

                  struct kadm5_auth_restrictions **rs_out)

{

    if ((mask & KADM5_POLICY) && strcmp(ent->policy, "VIP") == 0)

        return 0;

    return KRB5_PLUGIN_NO_HANDLE;

}

/* The bouncer denies addprinc operations which include a maximum lifetime. */

static krb5_error_code

bouncer_addprinc(krb5_context context, kadm5_auth_moddata data,

                 krb5_const_principal client, krb5_const_principal target,

                 const struct _kadm5_principal_ent_t *ent, long mask,

                 struct kadm5_auth_restrictions **rs_out)

{

    return (mask & KADM5_MAX_LIFE) ? EPERM : KRB5_PLUGIN_NO_HANDLE;

}

/* The welcomer authorizes modprinc operations which only set maxrenewlife. */

static krb5_error_code

welcomer_modprinc(krb5_context context, kadm5_auth_moddata data,

                  krb5_const_principal client, krb5_const_principal target,

                  const struct _kadm5_principal_ent_t *ent, long mask,

                  struct kadm5_auth_restrictions **rs_out)

{

    return (mask == KADM5_MAX_RLIFE) ? 0 : KRB5_PLUGIN_NO_HANDLE;

}

/* The bouncer denies modprinc operations if the target principal has an even

 * number of components. */

static krb5_error_code

bouncer_modprinc(krb5_context context, kadm5_auth_moddata data,

                 krb5_const_principal client, krb5_const_principal target,

                 const struct _kadm5_principal_ent_t *ent, long mask,

                 struct kadm5_auth_restrictions **rs_out)

{

    return (target->length % 2 == 0) ? EPERM : KRB5_PLUGIN_NO_HANDLE;

}

/* The welcomer authorizes setstr operations for the attribute "note". */

static krb5_error_code

welcomer_setstr(krb5_context context, kadm5_auth_moddata data,

                krb5_const_principal client, krb5_const_principal target,

                const char *key, const char *value)

{

    return (strcmp(key, "note") == 0) ? 0 : KRB5_PLUGIN_NO_HANDLE;

}

/* The bouncer denies setstr operations if the value is more than 10 bytes. */

static krb5_error_code

bouncer_setstr(krb5_context context, kadm5_auth_moddata data,

               krb5_const_principal client, krb5_const_principal target,

               const char *key, const char *value)

{

    return (strlen(value) > 10) ? EPERM : KRB5_PLUGIN_NO_HANDLE;

}

/* The welcomer authorizes delprinc operations if the target principal starts

 * with "d". */

static krb5_error_code

welcomer_delprinc(krb5_context context, kadm5_auth_moddata data,

                  krb5_const_principal client, krb5_const_principal target)

{

    if (target->length > 0 && target->data[0].length > 0 &&

        *target->data[0].data == 'd')

        return 0;

    return KRB5_PLUGIN_NO_HANDLE;

}

/* The bouncer denies delprinc operations if the target principal has the

 * "nodelete" string attribute. */

static krb5_error_code

bouncer_delprinc(krb5_context context, kadm5_auth_moddata data,

                 krb5_const_principal client, krb5_const_principal target)

{

    krb5_error_code ret;

    krb5_db_entry *ent;

    char *val = NULL;

    if (krb5_db_get_principal(context, target, 0, &ent) != 0)

        return EPERM;

    ret = krb5_dbe_get_string(context, ent, "nodelete", &val;;

    krb5_db_free_principal(context, ent);

    ret = (ret != 0 || val != NULL) ? EPERM : KRB5_PLUGIN_NO_HANDLE;

    krb5_dbe_free_string(context, val);

    return ret;

}

/* The welcomer authorizes rename operations if the first components of the

 * principals have the same length. */

static krb5_error_code

welcomer_renprinc(krb5_context context, kadm5_auth_moddata data,

                  krb5_const_principal client, krb5_const_principal src,

                  krb5_const_principal dest)

{

    if (src->length > 0 && dest->length > 0 &&

        src->data[0].length == dest->data[0].length)

        return 0;

    return KRB5_PLUGIN_NO_HANDLE;

}

/* The bouncer denies rename operations if the source principal starts with

 * "a". */

static krb5_error_code

bouncer_renprinc(krb5_context context, kadm5_auth_moddata data,

                 krb5_const_principal client, krb5_const_principal src,

                 krb5_const_principal dest)

{

    if (src->length > 0 && src->data[0].length > 0 &&

        *src->data[0].data == 'a')

        return EPERM;

    return KRB5_PLUGIN_NO_HANDLE;

}

/* The welcomer authorizes addpol operations which set a minlength of 3. */

static krb5_error_code

welcomer_addpol(krb5_context context, kadm5_auth_moddata data,

                krb5_const_principal client, const char *policy,

                const struct _kadm5_policy_ent_t *ent, long mask)

{

    if ((mask & KADM5_PW_MIN_LENGTH) && ent->pw_min_length == 3)

        return 0;

    return KRB5_PLUGIN_NO_HANDLE;

}

/* The bouncer denies addpol operations if the name is 3 bytes or less. */

static krb5_error_code

bouncer_addpol(krb5_context context, kadm5_auth_moddata data,

               krb5_const_principal client, const char *policy,

               const struct _kadm5_policy_ent_t *ent, long mask)

{

    return (strlen(policy) <= 3) ? EPERM : KRB5_PLUGIN_NO_HANDLE;

}

/* The welcomer authorizes modpol operations which only change min_life. */

static krb5_error_code

welcomer_modpol(krb5_context context, kadm5_auth_moddata data,

                krb5_const_principal client, const char *policy,

                const struct _kadm5_policy_ent_t *ent, long mask)

{

    return (mask == KADM5_PW_MIN_LIFE) ? 0 : KRB5_PLUGIN_NO_HANDLE;

}

/* The bouncer denies modpol operations which set pw_min_life above 10. */

static krb5_error_code

bouncer_modpol(krb5_context context, kadm5_auth_moddata data,

               krb5_const_principal client, const char *policy,

               const struct _kadm5_policy_ent_t *ent, long mask)

{

    if ((mask & KADM5_PW_MIN_LIFE) && ent->pw_min_life > 10)

        return EPERM;

    return KRB5_PLUGIN_NO_HANDLE;

}

/* The welcomer authorizes getpol operations if the policy and client principal

 * policy have the same length. */

static krb5_error_code

welcomer_getpol(krb5_context context, kadm5_auth_moddata data,

                krb5_const_principal client, const char *policy,

                const char *client_policy)

{

    if (client_policy != NULL && strlen(policy) == strlen(client_policy))

        return 0;

    return KRB5_PLUGIN_NO_HANDLE;

}

/* The bouncer denies getpol operations if the policy name begins with 'x'. */

static krb5_error_code

bouncer_getpol(krb5_context context, kadm5_auth_moddata data,

               krb5_const_principal client, const char *policy,

               const char *client_policy)

{

    return (*policy == 'x') ? EPERM : KRB5_PLUGIN_NO_HANDLE;

}

/* The welcomer counts end calls by incrementing the "ends" string attribute on

 * the "opcount" principal, if it exists. */

static void

welcomer_end(krb5_context context, kadm5_auth_moddata data)

{

    krb5_principal princ = NULL;

    krb5_db_entry *ent = NULL;

    char *val = NULL, buf[10];

    if (krb5_parse_name(context, "opcount", &princ) != 0)

        goto cleanup;

    if (krb5_db_get_principal(context, princ, 0, &ent) != 0)

        goto cleanup;

    if (krb5_dbe_get_string(context, ent, "ends", &val) != 0 || val == NULL)

        goto cleanup;

    snprintf(buf, sizeof(buf), "%d", atoi(val) + 1);

    if (krb5_dbe_set_string(context, ent, "ends", buf) != 0)

        goto cleanup;

    ent->mask = KADM5_TL_DATA;

    krb5_db_put_principal(context, ent);

cleanup:

    krb5_dbe_free_string(context, val);

    krb5_db_free_principal(context, ent);

    krb5_free_principal(context, princ);

}

krb5_error_code

kadm5_auth_welcomer_initvt(krb5_context context, int maj_ver, int min_ver,

                           krb5_plugin_vtable vtable)

{

    kadm5_auth_vtable vt = (kadm5_auth_vtable)vtable;

    vt->name = "welcomer";

    vt->addprinc = welcomer_addprinc;

    vt->modprinc = welcomer_modprinc;

    vt->setstr = welcomer_setstr;

    vt->delprinc = welcomer_delprinc;

    vt->renprinc = welcomer_renprinc;

    vt->getprinc = welcomer_getprinc;

    vt->addpol = welcomer_addpol;

    vt->modpol = welcomer_modpol;

    vt->getpol = welcomer_getpol;

    vt->end = welcomer_end;

    return 0;

}

krb5_error_code

kadm5_auth_bouncer_initvt(krb5_context context, int maj_ver, int min_ver,

                          krb5_plugin_vtable vtable)

{

    kadm5_auth_vtable vt = (kadm5_auth_vtable)vtable;

    vt->name = "bouncer";

    vt->addprinc = bouncer_addprinc;

    vt->modprinc = bouncer_modprinc;

    vt->setstr = bouncer_setstr;

    vt->delprinc = bouncer_delprinc;

    vt->renprinc = bouncer_renprinc;

    vt->addpol = bouncer_addpol;

    vt->modpol = bouncer_modpol;

    vt->getpol = bouncer_getpol;

    return 0;

}