/*
 * Copyright (c) 1997 - 2006 Kungliga Tekniska Högskolan
 * (Royal Institute of Technology, Stockholm, Sweden).
 * 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.
 *
 * 3. Neither the name of the Institute nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE 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 INSTITUTE 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.
 */

#include "kadm5_locl.h"

RCSID("$Id$");

static kadm5_ret_t
add_tl_data(kadm5_principal_ent_t ent, int16_t type,
            const void *data, size_t size)
{
    krb5_tl_data *tl;

    tl = calloc(1, sizeof(*tl));
    if (tl == NULL)
        return _kadm5_error_code(ENOMEM);

    tl->tl_data_type = type;
    tl->tl_data_length = size;
    tl->tl_data_contents = malloc(size);
    if (tl->tl_data_contents == NULL && size != 0) {
        free(tl);
        return _kadm5_error_code(ENOMEM);
    }
    memcpy(tl->tl_data_contents, data, size);

    tl->tl_data_next = ent->tl_data;
    ent->tl_data = tl;
    ent->n_tl_data++;

    return 0;
}

KRB5_LIB_FUNCTION krb5_ssize_t KRB5_LIB_CALL
_krb5_put_int(void *buffer, unsigned long value, size_t size); /* XXX */

kadm5_ret_t
kadm5_s_get_principal(void *server_handle,
                      krb5_principal princ,
                      kadm5_principal_ent_t out,
                      uint32_t mask)
{
    kadm5_server_context *context = server_handle;
    kadm5_ret_t ret;
    hdb_entry_ex ent;

    memset(&ent, 0, sizeof(ent));
    ret = context->db->hdb_open(context->context, context->db, O_RDONLY, 0);
    if(ret)
        return ret;
    ret = context->db->hdb_fetch_kvno(context->context, context->db, princ,
                                      HDB_F_DECRYPT|HDB_F_GET_ANY|HDB_F_ADMIN_DATA, 0, &ent);
    context->db->hdb_close(context->context, context->db);
    if(ret)
        return _kadm5_error_code(ret);

    memset(out, 0, sizeof(*out));
    if(mask & KADM5_PRINCIPAL)
        ret  = krb5_copy_principal(context->context, ent.entry.principal,
                                   &out->principal);
    if(ret)
        goto out;
    if(mask & KADM5_PRINC_EXPIRE_TIME && ent.entry.valid_end)
        out->princ_expire_time = *ent.entry.valid_end;
    if(mask & KADM5_PW_EXPIRATION && ent.entry.pw_end)
        out->pw_expiration = *ent.entry.pw_end;
    if(mask & KADM5_LAST_PWD_CHANGE)
        hdb_entry_get_pw_change_time(&ent.entry, &out->last_pwd_change);
    if(mask & KADM5_ATTRIBUTES){
        out->attributes |= ent.entry.flags.postdate ? 0 : KRB5_KDB_DISALLOW_POSTDATED;
        out->attributes |= ent.entry.flags.forwardable ? 0 : KRB5_KDB_DISALLOW_FORWARDABLE;
        out->attributes |= ent.entry.flags.initial ? KRB5_KDB_DISALLOW_TGT_BASED : 0;
        out->attributes |= ent.entry.flags.renewable ? 0 : KRB5_KDB_DISALLOW_RENEWABLE;
        out->attributes |= ent.entry.flags.proxiable ? 0 : KRB5_KDB_DISALLOW_PROXIABLE;
        out->attributes |= ent.entry.flags.invalid ? KRB5_KDB_DISALLOW_ALL_TIX : 0;
        out->attributes |= ent.entry.flags.require_preauth ? KRB5_KDB_REQUIRES_PRE_AUTH : 0;
        out->attributes |= ent.entry.flags.server ? 0 : KRB5_KDB_DISALLOW_SVR;
        out->attributes |= ent.entry.flags.change_pw ? KRB5_KDB_PWCHANGE_SERVICE : 0;
        out->attributes |= ent.entry.flags.ok_as_delegate ? KRB5_KDB_OK_AS_DELEGATE : 0;
        out->attributes |= ent.entry.flags.trusted_for_delegation ? KRB5_KDB_TRUSTED_FOR_DELEGATION : 0;
        out->attributes |= ent.entry.flags.allow_kerberos4 ? KRB5_KDB_ALLOW_KERBEROS4 : 0;
        out->attributes |= ent.entry.flags.allow_digest ? KRB5_KDB_ALLOW_DIGEST : 0;
    }
    if(mask & KADM5_MAX_LIFE) {
        if(ent.entry.max_life)
            out->max_life = *ent.entry.max_life;
        else
            out->max_life = INT_MAX;
    }
    if(mask & KADM5_MOD_TIME) {
        if(ent.entry.modified_by)
            out->mod_date = ent.entry.modified_by->time;
        else
            out->mod_date = ent.entry.created_by.time;
    }
    if(mask & KADM5_MOD_NAME) {
        if(ent.entry.modified_by) {
            if (ent.entry.modified_by->principal != NULL)
                ret = krb5_copy_principal(context->context,
                                          ent.entry.modified_by->principal,
                                          &out->mod_name);
        } else if(ent.entry.created_by.principal != NULL)
            ret = krb5_copy_principal(context->context,
                                      ent.entry.created_by.principal,
                                      &out->mod_name);
        else
            out->mod_name = NULL;
    }
    if(ret)
        goto out;

    if(mask & KADM5_KVNO)
        out->kvno = ent.entry.kvno;
    if(mask & KADM5_MKVNO) {
        size_t n;
        out->mkvno = 0; /* XXX */
        for(n = 0; n < ent.entry.keys.len; n++)
            if(ent.entry.keys.val[n].mkvno) {
                out->mkvno = *ent.entry.keys.val[n].mkvno; /* XXX this isn't right */
                break;
            }
    }
#if 0 /* XXX implement */
    if(mask & KADM5_AUX_ATTRIBUTES)
        ;
    if(mask & KADM5_LAST_SUCCESS)
        ;
    if(mask & KADM5_LAST_FAILED)
        ;
    if(mask & KADM5_FAIL_AUTH_COUNT)
        ;
#endif
    if(mask & KADM5_POLICY)
        out->policy = NULL;
    if(mask & KADM5_MAX_RLIFE) {
        if(ent.entry.max_renew)
            out->max_renewable_life = *ent.entry.max_renew;
        else
            out->max_renewable_life = INT_MAX;
    }
    if(mask & KADM5_KEY_DATA){
        size_t i;
        Key *key;
        krb5_key_data *kd;
        krb5_salt salt;
        krb5_data *sp;
        krb5_get_pw_salt(context->context, ent.entry.principal, &salt);
        out->key_data = malloc(ent.entry.keys.len * sizeof(*out->key_data));
        if (out->key_data == NULL && ent.entry.keys.len != 0) {
            ret = ENOMEM;
            goto out;
        }
        for(i = 0; i < ent.entry.keys.len; i++){
            key = &ent.entry.keys.val[i];
            kd = &out->key_data[i];
            kd->key_data_ver = 2;
            kd->key_data_kvno = ent.entry.kvno;
            kd->key_data_type[0] = key->key.keytype;
            if(key->salt)
                kd->key_data_type[1] = key->salt->type;
            else
                kd->key_data_type[1] = KRB5_PADATA_PW_SALT;
            /* setup key */
            kd->key_data_length[0] = key->key.keyvalue.length;
            kd->key_data_contents[0] = malloc(kd->key_data_length[0]);
            if(kd->key_data_contents[0] == NULL && kd->key_data_length[0] != 0){
                ret = ENOMEM;
                break;
            }
            memcpy(kd->key_data_contents[0], key->key.keyvalue.data,
                   kd->key_data_length[0]);
            /* setup salt */
            if(key->salt)
                sp = &key->salt->salt;
            else
                sp = &salt.saltvalue;
            kd->key_data_length[1] = sp->length;
            kd->key_data_contents[1] = malloc(kd->key_data_length[1]);
            if(kd->key_data_length[1] != 0
               && kd->key_data_contents[1] == NULL) {
                memset(kd->key_data_contents[0], 0, kd->key_data_length[0]);
                ret = ENOMEM;
                break;
            }
            memcpy(kd->key_data_contents[1], sp->data, kd->key_data_length[1]);
            out->n_key_data = i + 1;
        }
        krb5_free_salt(context->context, salt);
    }
    if(ret){
        kadm5_free_principal_ent(context, out);
        goto out;
    }
    if(mask & KADM5_TL_DATA) {
        time_t last_pw_expire;
        const HDB_Ext_PKINIT_acl *acl;
        const HDB_Ext_Aliases *aliases;

        ret = hdb_entry_get_pw_change_time(&ent.entry, &last_pw_expire);
        if (ret == 0 && last_pw_expire) {
            unsigned char buf[4];
            _krb5_put_int(buf, last_pw_expire, sizeof(buf));
            ret = add_tl_data(out, KRB5_TL_LAST_PWD_CHANGE, buf, sizeof(buf));
            if (ret)
                goto out;
        }
        if(ret){
            kadm5_free_principal_ent(context, out);
            goto out;
        }
        /*
         * If the client was allowed to get key data, let it have the
         * password too.
         */
        if(mask & KADM5_KEY_DATA) {
            heim_utf8_string pw;

            ret = hdb_entry_get_password(context->context,
                                         context->db, &ent.entry, &pw);
            if (ret == 0) {
                (void) add_tl_data(out, KRB5_TL_PASSWORD, pw, strlen(pw) + 1);
                free(pw);
            }
            krb5_clear_error_message(context->context);
        }

        ret = hdb_entry_get_pkinit_acl(&ent.entry, &acl);
        if (ret == 0 && acl) {
            krb5_data buf;
            size_t len;

            ASN1_MALLOC_ENCODE(HDB_Ext_PKINIT_acl, buf.data, buf.length,
                                acl, &len, ret);
            if (ret) {
                kadm5_free_principal_ent(context, out);
                goto out;
            }
            if (len != buf.length)
                krb5_abortx(context->context,
                            "internal ASN.1 encoder error");
            ret = add_tl_data(out, KRB5_TL_PKINIT_ACL, buf.data, buf.length);
            free(buf.data);
            if (ret) {
                kadm5_free_principal_ent(context, out);
                goto out;
            }
        }
        if(ret){
            kadm5_free_principal_ent(context, out);
            goto out;
        }

        ret = hdb_entry_get_aliases(&ent.entry, &aliases);
        if (ret == 0 && aliases) {
            krb5_data buf;
            size_t len;

            ASN1_MALLOC_ENCODE(HDB_Ext_Aliases, buf.data, buf.length,
                               aliases, &len, ret);
            if (ret) {
                kadm5_free_principal_ent(context, out);
                goto out;
            }
            if (len != buf.length)
                krb5_abortx(context->context,
                            "internal ASN.1 encoder error");
            ret = add_tl_data(out, KRB5_TL_ALIASES, buf.data, buf.length);
            free(buf.data);
            if (ret) {
                kadm5_free_principal_ent(context, out);
                goto out;
            }
        }
        if(ret){
            kadm5_free_principal_ent(context, out);
            goto out;
        }

    }
out:
    hdb_free_entry(context->context, &ent);

    return _kadm5_error_code(ret);
}