/*
 * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 * Copyright (c) 2016 by Delphix. All rights reserved.
 */


/*
 * lib/krb5/os/kuserok.c
 *
 * Copyright 1990,1993 by the Massachusetts Institute of Technology.
 * All Rights Reserved.
 *
 * Export of this software from the United States of America may
 *   require a specific license from the United States Government.
 *   It is the responsibility of any person or organization contemplating
 *   export to obtain such a license before exporting.
 *
 * WITHIN THAT CONSTRAINT, permission to use, copy, modify, and
 * distribute this software and its documentation for any purpose and
 * without fee is hereby granted, provided that the above copyright
 * notice appear in all copies and that both that copyright notice and
 * this permission notice appear in supporting documentation, and that
 * the name of M.I.T. not be used in advertising or publicity pertaining
 * to distribution of the software without specific, written prior
 * permission.  Furthermore if you modify this software you must label
 * your software as modified software and not distribute it in such a
 * fashion that it might be confused with the original M.I.T. software.
 * M.I.T. makes no representations about the suitability of
 * this software for any purpose.  It is provided "as is" without express
 * or implied warranty.
 *
 *
 * krb5_kuserok()
 */

#include "k5-int.h"
#if !defined(_WIN32)		/* Not yet for Windows */
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <pwd.h>
#include <libintl.h>
#include <gssapi/gssapi.h>
#include <gssapi/gssapi_ext.h>
#include <gssapi_krb5.h>
#include <gssapiP_krb5.h>
#include <syslog.h>

#if defined(_AIX) && defined(_IBMR2)
#include <sys/access.h>
/* xlc has a bug with "const" */
#define getpwnam(user) getpwnam((char *)user)
#endif

#define MAX_USERNAME 65
#define	CACHE_FILENAME_LEN 35

#if defined(__APPLE__) && defined(__MACH__)
#include <hfs/hfs_mount.h>	/* XXX */
#define FILE_OWNER_OK(UID)  ((UID) == 0 || (UID) == UNKNOWNUID)
#else
#define FILE_OWNER_OK(UID)  ((UID) == 0)
#endif

/* Solaris Kerberos */
extern void
gsscred_set_options();

extern OM_uint32
gsscred_name_to_unix_cred_ext();

extern int
safechown(const char *src, uid_t uid, gid_t gid, int mode);

extern const char *error_message(long);


krb5_data tgtname = {
	0,
	KRB5_TGS_NAME_SIZE,
	KRB5_TGS_NAME
};

/* Solaris Kerberos */
static krb5_error_code
krb5_move_ccache(krb5_context kcontext, krb5_principal client,
		struct passwd *pwd)
{
	char *name = 0;
	static char ccache_name_buf[CACHE_FILENAME_LEN];
	krb5_ccache ccache = NULL;
	krb5_error_code retval;

	name = getenv(KRB5_ENV_CCNAME);
	if (name == 0)
		/*
		 * This means that there was no forwarding
		 * of creds
		 */
		return (0);
	else {
		/*
		 * creds have been forwarded and stored in
		 * KRB5_ENV_CCNAME and now we need to store it
		 * under uid
		 */

		krb5_creds mcreds, save_v5creds;

		memset(&mcreds, 0, sizeof (mcreds));
		memset(&save_v5creds, 0, sizeof (save_v5creds));

		mcreds.client =  client;
		retval = krb5_build_principal_ext(kcontext, &mcreds.server,
				krb5_princ_realm(kcontext,  client)->length,
				krb5_princ_realm(kcontext,  client)->data,
				tgtname.length, tgtname.data,
				krb5_princ_realm(kcontext,  client)->length,
				krb5_princ_realm(kcontext,  client)->data,
				0);
		if (retval) {
			syslog(LOG_ERR,
				gettext("KRB5: %s while creating"
					"V5 krbtgt principal "),
				error_message(retval));
			return (retval);
		}

		mcreds.ticket_flags = 0;
		retval = krb5_cc_default(kcontext, &ccache);
		if (retval) {
			syslog(LOG_ERR,
				gettext("KRB5: %s while getting "
					"default cache "),
				error_message(retval));
			return (retval);
		}

		retval = krb5_cc_retrieve_cred(kcontext, ccache,
						0,
						&mcreds, &save_v5creds);
		if (retval) {
			syslog(LOG_ERR,
				gettext("KRB5: %s while retrieving "
					"cerdentials "),
				error_message(retval));
			return (retval);
		}
		/*
		 * reset the env variable and recreate the
		 * cache using the default cache name
		 */
		retval = krb5_cc_destroy(kcontext, ccache);
		if (retval) {
			syslog(LOG_ERR,
				gettext("KRB5: %s while destroying cache "),
				error_message(retval));
			return (retval);
		}
		krb5_unsetenv(KRB5_ENV_CCNAME);
		snprintf(ccache_name_buf,
			CACHE_FILENAME_LEN,
			"FILE:/tmp/krb5cc_%d", pwd->pw_uid);
		krb5_setenv(KRB5_ENV_CCNAME, ccache_name_buf, 1);
		retval =  krb5_cc_resolve(kcontext, ccache_name_buf, &ccache);
		if (retval) {
			syslog(LOG_ERR,
				gettext("KRB5: %s while resolving cache "),
				error_message(retval));
			return (retval);
		}
		retval = krb5_cc_initialize(kcontext, ccache, client);
		if (retval) {
			syslog(LOG_ERR,
				gettext("KRB5: %s while initializing cache "),
				error_message(retval));
			return (retval);
		}
		retval =  krb5_cc_store_cred(kcontext, ccache, &save_v5creds);
		if (retval) {
			syslog(LOG_ERR,
				gettext("KRB5: %s while storing creds "),
				error_message(retval));
			return (retval);
		}
		snprintf(ccache_name_buf,
			CACHE_FILENAME_LEN,
			"/tmp/krb5cc_%d", pwd->pw_uid);
		if (safechown(ccache_name_buf, pwd->pw_uid,
			pwd->pw_gid, -1) == -1) {
			syslog(LOG_ERR,
				gettext("KRB5: Can not change "
					"ownership of cache file, "
					"possible security breach\n"));
		}
	}

	return (0);
}


/*
 * Solaris Kerberos:
 * krb5_gsscred: Given a kerberos principal try to find the corresponding
 * local uid via the gss cred table. Return TRUE if the uid was found in the
 * cred table, otherwise return FALSE.
 */
static krb5_boolean
krb5_gsscred(krb5_principal principal, uid_t *uid)
{
	OM_uint32 minor, major;
	gss_name_t name;
	gss_buffer_desc name_buf;

	name_buf.value = &principal;
	name_buf.length = sizeof (principal);

	/*
	 * Convert the kerb principal in to a gss name
	 */
	major = gss_import_name(&minor, &name_buf,
				(gss_OID)gss_nt_krb5_principal, &name);

	if (major != GSS_S_COMPLETE)
		return (FALSE);

	gsscred_set_options();

	/*
	 * Get the uid mapping from the gsscred table.
	 * (but set flag to not call back into this mech as we do krb5
	 * auth_to_local name mapping from this module).
	 */
	major = gsscred_name_to_unix_cred_ext(name, (gss_OID)gss_mech_krb5,
					  uid, 0, 0, 0, 0);

	(void) gss_release_name(&minor, &name);

	if (major != GSS_S_COMPLETE)
		return (FALSE);

	return (TRUE);
}

/*
 * Given a Kerberos principal "principal", and a local username "luser",
 * determine whether user is authorized to login according to the
 * authorization file ("~luser/.k5login" by default).  Returns TRUE
 * if authorized, FALSE if not authorized.
 *
 * If there is no account for "luser" on the local machine, returns
 * FALSE.  If there is no authorization file, and the given Kerberos
 * name "server" translates to the same name as "luser" (using
 * krb5_aname_to_lname()), returns TRUE.  Otherwise, if the authorization file
 * can't be accessed, returns FALSE.  Otherwise, the file is read for
 * a matching principal name, instance, and realm.  If one is found,
 * returns TRUE, if none is found, returns FALSE.
 *
 * The file entries are in the format produced by krb5_unparse_name(),
 * one entry per line.
 *
 */

krb5_boolean KRB5_CALLCONV
krb5_kuserok(krb5_context context, krb5_principal principal, const char *luser)
{
    struct stat sbuf;
    struct passwd *pwd;
    char pbuf[MAXPATHLEN];
    krb5_boolean isok = FALSE;
    FILE *fp;
    char kuser[MAX_USERNAME];
    char *princname;
    char linebuf[BUFSIZ];
    char *newline;
    /* Solaris Kerberos */
    uid_t uid;
    int gobble;

    /* no account => no access */
    char pwbuf[BUFSIZ];
    struct passwd pwx;
    if (k5_getpwnam_r(luser, &pwx, pwbuf, sizeof(pwbuf), &pwd) != 0)
	return(FALSE);
    (void) strncpy(pbuf, pwd->pw_dir, sizeof(pbuf) - 1);
    pbuf[sizeof(pbuf) - 1] = '\0';
    (void) strncat(pbuf, "/.k5login", sizeof(pbuf) - 1 - strlen(pbuf));

    if (access(pbuf, F_OK)) {	 /* not accessible */
	/*
	 * if they're trying to log in as themself, and there is no .k5login file,
	 * let them.  First, have krb5 check it's rules.  If no success,
	 * search the gsscred table (the sequence here should be consistent
	 * with the uid mappings done for gssd).
	 */
	if (!(krb5_aname_to_localname(context, principal,
				      sizeof(kuser), kuser))
	    && (strcmp(kuser, luser) == 0)) {
		/* Solaris Kerberos */
		if (krb5_move_ccache(context, principal, pwd))
			return (FALSE);
	    	return(TRUE);
	}

	if (krb5_gsscred(principal, &uid)) {
#ifdef DEBUG
	    char *princname;

	    (void)krb5_unparse_name(context, principal, &princname);
	    syslog(LOG_DEBUG, "gsscred mapped %s to %d expecting %d (%s)\n",
		   princname, uid, pwd->pw_uid, luser);
	    free(princname);
#endif
	    if (uid == pwd->pw_uid) {
		if (krb5_move_ccache(context, principal, pwd))
			return (FALSE);
		return (TRUE);
	    }
	}

    }
    if (krb5_unparse_name(context, principal, &princname))
	return(FALSE);			/* no hope of matching */

    /* open ~/.k5login */
    /* Solaris Kerberos */
    if ((fp = fopen(pbuf, "rF")) == NULL) {
	free(princname);
	return(FALSE);
    }
    /*
     * For security reasons, the .k5login file must be owned either by
     * the user himself, or by root.  Otherwise, don't grant access.
     */
    if (fstat(fileno(fp), &sbuf)) {
	fclose(fp);
	free(princname);
	return(FALSE);
    }
    if (sbuf.st_uid != pwd->pw_uid && !FILE_OWNER_OK(sbuf.st_uid)) {
	fclose(fp);
	free(princname);
	return(FALSE);
    }

    /* check each line */
    while (!isok && (fgets(linebuf, BUFSIZ, fp) != NULL)) {
	/* null-terminate the input string */
	linebuf[BUFSIZ-1] = '\0';
	newline = NULL;
	/* nuke the newline if it exists */
	if ((newline = strchr(linebuf, '\n')))
	    *newline = '\0';
	if (!strcmp(linebuf, princname)) {
	    isok = TRUE;
	    /* Solaris Kerberos */
	    if (krb5_move_ccache(context, principal, pwd))
		return (FALSE);
	    continue;
	}
	/* clean up the rest of the line if necessary */
	if (!newline)
	    while (((gobble = getc(fp)) != EOF) && gobble != '\n');
    }
    free(princname);
    fclose(fp);
    return(isok);
}

/* Solaris Kerberos */
OM_uint32
krb5_gss_userok(OM_uint32 *minor,
		const gss_name_t pname,
		const char *user,
		int *user_ok)
{
	krb5_context ctxt;
	OM_uint32 kret;

	if (pname == NULL || user == NULL)
		return (GSS_S_CALL_INACCESSIBLE_READ);

	if (minor == NULL || user_ok == NULL)
		return (GSS_S_CALL_INACCESSIBLE_WRITE);

	*user_ok = 0;

	kret = krb5_gss_init_context(&ctxt);
	if (kret) {
		*minor = kret;
		return (GSS_S_FAILURE);
	}

	if (! kg_validate_name(pname)) {
		*minor = (OM_uint32) G_VALIDATE_FAILED;
		krb5_free_context(ctxt);
		return (GSS_S_CALL_BAD_STRUCTURE|GSS_S_BAD_NAME);
	}

	if (krb5_kuserok(ctxt, (krb5_principal) pname, user)) {
		*user_ok = 1;
	}

	krb5_free_context(ctxt);
	return (GSS_S_COMPLETE);
}

#else /* _WIN32 */

/*
 * If the given Kerberos name "server" translates to the same name as "luser"
 * (using * krb5_aname_to_lname()), returns TRUE.
 */
krb5_boolean KRB5_CALLCONV
krb5_kuserok(context, principal, luser)
    krb5_context context;
    krb5_principal principal;
    const char *luser;
{
    char kuser[50];

    if (krb5_aname_to_localname(context, principal, sizeof(kuser), kuser))
        return FALSE;

    if (strcmp(kuser, luser) == 0)
	    return TRUE;

    return FALSE;
}
#endif /* _WIN32 */