/*
 * Copyright (c) 1995, 1996, 1997 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. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *      This product includes software developed by the Kungliga Tekniska
 *      Högskolan and its contributors.
 * 
 * 4. 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 "kafs_locl.h"

RCSID("$Id$");

#define AUTH_SUPERUSER "afs"

/*
 * Here only ASCII characters are relevant.
 */

#define IsAsciiLower(c) ('a' <= (c) && (c) <= 'z')

#define ToAsciiUpper(c) ((c) - 'a' + 'A')

static void
foldup(char *a, const char *b)
{
  for (; *b; a++, b++)
    if (IsAsciiLower(*b))
      *a = ToAsciiUpper(*b);
    else
      *a = *b;
  *a = '\0';
}

static krb5_error_code
get_cred(krb5_context context, krb5_ccache id,
	 const char *name, const char *inst, const char *krealm, 
	 CREDENTIALS *c)
{
    krb5_error_code ret;
    krb5_creds in_creds, *out_creds;

    if(krb_get_cred((char*)name, (char*)inst, (char*)krealm, c) == 0)
	return 0;

    memset(&in_creds, 0, sizeof(in_creds));
    ret = krb5_425_conv_principal(context, name, inst, krealm, 
				  &in_creds.server);
    if(ret)
	return ret;
    ret = krb5_cc_get_principal(context, id, &in_creds.client);
    if(ret){
	krb5_free_principal(context, in_creds.server);
	return ret;
    }
    ret = krb5_get_credentials(context, 0, id, &in_creds, &out_creds);
    krb5_free_principal(context, in_creds.server);
    krb5_free_principal(context, in_creds.client);
    if(ret)
	return ret;
    ret = krb524_convert_creds_kdc(context, out_creds, c);
    krb5_free_creds(context, out_creds);
    return ret;
}


/* Convert a string to a 32 bit ip number in network byte order. 
   Return 0 on error
   */

static u_int32_t
ip_aton(char *ip)
{
  u_int32_t addr;
  unsigned int a, b, c, d;

  if(sscanf(ip, "%u.%u.%u.%u", &a, &b, &c, &d) != 4)
      return 0;
  if((a | b | c | d) > 255)
      return 0;
  addr = (a << 24) | (b << 16) | (c << 8) | d;
  addr = htonl(addr);
  return addr;
}

#if 0
/* Try to get a db-server for an AFS cell from a AFSDB record */

static int
dns_find_cell(const char *cell, char *dbserver)
{
    struct dns_reply *r;
    int ok = -1;
    r = dns_lookup(cell, "afsdb");
    if(r){
	struct resource_record *rr = r->head;
	while(rr){
	    if(rr->type == T_AFSDB && rr->u.afsdb->preference == 1){
		strncpy(dbserver, rr->u.afsdb->domain, MaxHostNameLen);
		dbserver[MaxHostNameLen - 1] = 0;
		ok = 0;
		break;
	    }
	    rr = rr->next;
	}
	dns_free_data(r);
    }
    return ok;
}
#endif


/* Find the realm associated with cell. Do this by opening
   /usr/vice/etc/CellServDB and getting the realm-of-host for the
   first VL-server for the cell.

   This does not work when the VL-server is living in one realm, but
   the cell it is serving is living in another realm.
   */

static krb5_error_code
realm_of_cell(krb5_context context, const char *cell, krb5_realm *realm)
{
    FILE *F;
    char buf[1024];
    krb5_realm *realms;
    char *p;
    krb5_error_code ret = -1;

    if((F = fopen(_PATH_CELLSERVDB, "r"))){
	while(fgets(buf, sizeof(buf), F)){
	    if(buf[0] != '>')
		continue;
	    if(strncmp(buf + 1, cell, strlen(cell)) == 0){
		if(fgets(buf, sizeof(buf), F) == NULL)
		    break;
		p = strchr(buf, '#');
		if(p == NULL)
		    break;
		p++;
		if(buf[strlen(buf) - 1] == '\n')
		    buf[strlen(buf) - 1] = 0;
		ret = krb5_get_host_realm(context, p, &realms);
		*realm = strdup(realms[0]);
		krb5_free_host_realm(context, realms);
		break;
	    }
	}
	fclose(F);
    }
#if 0
    if(realm == NULL){
	if(dns_find_cell(cell, buf) == 0)
	    realm = krb_realmofhost(buf);
    }
#endif
    return ret;
}

/*
 * Get tokens for all cells[]
 */
static krb5_error_code
k5_afslog_cells(krb5_context context, krb5_ccache id,
		char *cells[], int max, krb5_const_realm realm, uid_t uid)
{
    krb5_error_code ret = 0;
    int i;
    for(i = 0; i < max; i++)
	ret = k5_afsklog_uid(context, id, cells[i], realm, uid);
    return ret;
}

/*
 * Try to find the cells we should try to klog to in "file".
 */
static void
k_find_cells(char *file, char *cells[], int size, int *index)
{
    FILE *f;
    char cell[64];
    int i;
    f = fopen(file, "r");
    if (f == NULL)
	return;
    while (*index < size && fgets(cell, sizeof(cell), f)) {
	char *nl = strchr(cell, '\n');
	if (nl) *nl = 0;
	for(i = 0; i < *index; i++)
	    if(strcmp(cells[i], cell) == 0)
		break;
	if(i == *index)
	    cells[(*index)++] = strdup(cell);
    }
    fclose(f);
}

static krb5_error_code
k5_afsklog_all_local_cells(krb5_context context, krb5_ccache id,
			   krb5_const_realm realm, uid_t uid)
{
    krb5_error_code ret;
    char *cells[32]; /* XXX */
    int num_cells = sizeof(cells) / sizeof(cells[0]);
    int index = 0;

    char *p;
    
    if ((p = getenv("HOME"))) {
	char home[MaxPathLen];

	if (k_concat(home, sizeof(home), p, "/.TheseCells", NULL) == 0)
	    k_find_cells(home, cells, num_cells, &index);
    }
    k_find_cells(_PATH_THESECELLS, cells, num_cells, &index);
    k_find_cells(_PATH_THISCELL, cells, num_cells, &index);
    
    ret = k5_afslog_cells(context, id, cells, index, realm, uid);
    while(index > 0)
	free(cells[--index]);
    return ret;
}

krb5_error_code
k5_afsklog_uid(krb5_context context, krb5_ccache id,
	       const char *cell, krb5_const_realm krealm, uid_t uid)
{
    int k_errno;
    krb5_error_code ret;
    CREDENTIALS c;
    krb5_realm vl_realm; /* realm of vl-server */
    krb5_realm lrealm; /* local realm */
    char CELL[64];

    if (cell == 0 || cell[0] == 0)
	return k5_afsklog_all_local_cells (context, id, krealm, uid);
    foldup(CELL, cell);

    ret = krb5_get_default_realm(context, &lrealm);
    if(ret || (krealm && strcmp(krealm, lrealm) == 0)){
	free(lrealm);
	lrealm = NULL;
    }

    /* We're about to find the the realm that holds the key for afs in
     * the specified cell. The problem is that null-instance
     * afs-principals are common and that hitting the wrong realm might
     * yield the wrong afs key. The following assumptions were made.
     *
     * Any realm passed to us is preferred.
     *
     * If there is a realm with the same name as the cell, it is most
     * likely the correct realm to talk to.
     *
     * In most (maybe even all) cases the database servers of the cell
     * will live in the realm we are looking for.
     *
     * Try the local realm, but if the previous cases fail, this is
     * really a long shot.
     *
     */
  
    /* comments on the ordering of these tests */

    /* If the user passes a realm, she probably knows something we don't
     * know and we should try afs@krealm (otherwise we're talking with a
     * blondino and she might as well have it.)
     */
  
    ret = -1;
    if(krealm){
	ret = get_cred(context, id, AUTH_SUPERUSER, cell, krealm, &c);
	if(ret)
	    ret = get_cred(context, id, AUTH_SUPERUSER, "", krealm, &c);
    }

    if(ret)
	ret = get_cred(context, id, AUTH_SUPERUSER, cell, CELL, &c);
    if(ret)
	ret = get_cred(context, id, AUTH_SUPERUSER, "", CELL, &c);
    
    /* this might work in some cases */
    if(ret){
	if(realm_of_cell(context, cell, &vl_realm) == 0){
	    ret = get_cred(context, id, AUTH_SUPERUSER, cell, vl_realm, &c);
	    if(ret)
		ret = get_cred(context, id, AUTH_SUPERUSER, "", vl_realm, &c);
	    free(vl_realm);
	}
    }
    
    if(ret && lrealm)
	ret = get_cred(context, id, AUTH_SUPERUSER, cell, lrealm, &c);
    if(lrealm)
	free(lrealm);
    
    if (ret == 0){
	struct ViceIoctl parms;
	struct ClearToken ct;
	int32_t sizeof_x;
	char buf[2048], *t;

	/*
	 * Build a struct ClearToken
	 */
	ct.AuthHandle = c.kvno;
	memcpy (ct.HandShakeKey, c.session, sizeof(c.session));
	ct.ViceId = uid;	/* is this always valid? */
	ct.BeginTimestamp = 1 + c.issue_date;
	ct.EndTimestamp = krb_life_to_time(c.issue_date, c.lifetime);

#define ODD(x) ((x) & 1)
	/* If we don't know the numerical ID lifetime should be even? */
	if (uid == 0 && ODD(ct.EndTimestamp - ct.BeginTimestamp))
	    ct.BeginTimestamp--;

	t = buf;
	/*
	 * length of secret token followed by secret token
	 */
	sizeof_x = c.ticket_st.length;
	memcpy(t, &sizeof_x, sizeof(sizeof_x));
	t += sizeof(sizeof_x);
	memcpy(t, c.ticket_st.dat, sizeof_x);
	t += sizeof_x;
	/*
	 * length of clear token followed by clear token
	 */
	sizeof_x = sizeof(ct);
	memcpy(t, &sizeof_x, sizeof(sizeof_x));
	t += sizeof(sizeof_x);
	memcpy(t, &ct, sizeof_x);
	t += sizeof_x;

	/*
	 * do *not* mark as primary cell
	 */
	sizeof_x = 0;
	memcpy(t, &sizeof_x, sizeof(sizeof_x));
	t += sizeof(sizeof_x);
	/*
	 * follow with cell name
	 */
	sizeof_x = strlen(cell) + 1;
	memcpy(t, cell, sizeof_x);
	t += sizeof_x;

	/*
	 * Build argument block
	 */
	parms.in = buf;
	parms.in_size = t - buf;
	parms.out = 0;
	parms.out_size = 0;
	k_pioctl(0, VIOCSETTOK, &parms, 0);
    }
    return k_errno;
}

krb5_error_code
k5_afsklog(krb5_context context, krb5_ccache id, 
	   const char *cell, krb5_const_realm realm)
{
    return k5_afsklog_uid (context, id, cell, realm, getuid());
}