/*      $NetBSD: pam_afslog.c,v 1.2 2006/01/20 16:51:15 christos Exp $  */

/*-
 * Copyright 2005 Tyler C. Sarna <tsarna@netbsd.org>
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Tyler C. Sarna
 *
 * 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. Neither the name of The NetBSD Foundation 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 NETBSD FOUNDATION, INC. 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 FOUNDATION 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 <sys/cdefs.h>

__RCSID("$NetBSD: pam_afslog.c,v 1.2 2006/01/20 16:51:15 christos Exp $");

#include <krb5/krb5.h>
#include <krb5/kafs.h>

#define PAM_SM_AUTH
#define PAM_SM_CRED
#include <security/pam_appl.h>
#include <security/pam_modules.h>
#include <security/pam_mod_misc.h>

PAM_EXTERN int
pam_sm_authenticate(pam_handle_t *pamh, int flags __unused,
    int argc __unused, const char *argv[] __unused)
{
        return PAM_IGNORE;
}

PAM_EXTERN int
pam_sm_setcred(pam_handle_t *pamh, int flags,
    int argc __unused, const char *argv[] __unused)
{
        krb5_context ctx;
        krb5_ccache ccache;
        krb5_principal principal;
        krb5_error_code kret;
        const void *service = NULL;
        const char *ccname = NULL;
        int do_afslog = 0, ret = PAM_SUCCESS;
                                                
        pam_get_item(pamh, PAM_SERVICE, &service);
        if (service == NULL)
                service = "pam_afslog";

        kret = krb5_init_context(&ctx);
        if (kret != 0) {
                PAM_LOG("Error: krb5_init_context() failed");
                ret = PAM_SERVICE_ERR;
        } else {
                ccname = pam_getenv(pamh, "KRB5CCNAME");
                if (ccname)
                        kret = krb5_cc_resolve(ctx, ccname, &ccache);
                else
                        kret = krb5_cc_default(ctx, &ccache);
                if (kret != 0) {
                        PAM_LOG("Error: failed to open ccache");
                        ret = PAM_SERVICE_ERR;
                } else {
                        kret = krb5_cc_get_principal(ctx, ccache, &principal);
                        if (kret != 0) {
                                PAM_LOG("Error: krb5_cc_get_principal() failed");
                                ret = PAM_SERVICE_ERR;
                        } else {
                                krb5_appdefault_boolean(ctx,
                                        (const char *)service,
                                        krb5_principal_get_realm(
                                                ctx, principal),
                                        "afslog", FALSE, &do_afslog);

                                /* silently bail if not enabled */
        
                                if (do_afslog && k_hasafs()) {
                                        switch (flags & ~PAM_SILENT) {
                                        case 0:
                                        case PAM_ESTABLISH_CRED:
                                                k_setpag();
                                
                                                /* FALLTHROUGH */
        
                                        case PAM_REINITIALIZE_CRED:
                                        case PAM_REFRESH_CRED:
                                                krb5_afslog(ctx, ccache,
                                                        NULL, NULL);
                                                break;

                                        case PAM_DELETE_CRED:
                                                k_unlog();
                                                break;
                                        }
                                }

                                krb5_free_principal(ctx, principal);
                        }

                        krb5_cc_close(ctx, ccache);
                }

                krb5_free_context(ctx);
        }
        
        return ret;
}

PAM_MODULE_ENTRY("pam_afslog");