kswitch.c

/* -*- mode: c; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/* clients/kswitch/kswitch.c - Switch primary credential cache */
/*
 * Copyright 2011 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.
 */

#include "k5-int.h"
#include <locale.h>

#ifndef _WIN32
#define GET_PROGNAME(x) (strrchr((x), '/') ? strrchr((x), '/')+1 : (x))
#else
#define GET_PROGNAME(x) max(max(strrchr((x), '/'), strrchr((x), '\\')) + 1,(x))
#endif

static char *progname;

static void
usage(void)
{
    fprintf(stderr, _("Usage: %s {-c cache_name | -p principal}\n"), progname);
    fprintf(stderr, _("\t-c specify name of credentials cache\n"));
    fprintf(stderr, _("\t-p specify name of principal\n"));
    exit(2);
}

int
main(int argc, char **argv)
{
    krb5_context context;
    krb5_error_code ret;
    int c;
    krb5_ccache cache = NULL;
    krb5_principal princ = NULL;
    const char *cache_name = NULL, *princ_name = NULL;
    krb5_boolean errflag = FALSE;

    setlocale(LC_ALL, "");
    progname = GET_PROGNAME(argv[0]);

    while ((c = getopt(argc, argv, "c:p:")) != -1) {
        switch (c) {
        case 'c':
        case 'p':
            if (cache_name || princ_name) {
                fprintf(stderr, _("Only one -c or -p option allowed\n"));
                errflag = TRUE;
            } else if (c == 'c') {
                cache_name = optarg;
            } else {
                princ_name = optarg;
            }
            break;
        case '?':
        default:
            errflag = TRUE;
            break;
        }
    }

    if (optind != argc)
        errflag = TRUE;

    if (!cache_name && !princ_name) {
        fprintf(stderr, _("One of -c or -p must be specified\n"));
        errflag = TRUE;
    }

    if (errflag)
        usage();

    ret = krb5_init_context(&context);
    if (ret) {
        com_err(progname, ret, _("while initializing krb5"));
        exit(1);
    }

    if (cache_name) {
        ret = krb5_cc_resolve(context, cache_name, &cache);
        if (ret != 0) {
            com_err(progname, ret, _("while resolving %s"), cache_name);
            exit(1);
        }
    } else {
        ret = krb5_parse_name(context, princ_name, &princ);
        if (ret) {
            com_err(progname, ret, _("while parsing principal name %s"),
                    princ_name);
            exit(1);
        }
        ret = krb5_cc_cache_match(context, princ, &cache);
        if (ret) {
            com_err(progname, ret, _("while searching for ccache for %s"),
                    princ_name);
            exit(1);
        }
        krb5_free_principal(context, princ);
    }

    ret = krb5_cc_switch(context, cache);
    if (ret != 0) {
        com_err(progname, ret, _("while switching to credential cache"));
        exit(1);
    }

    krb5_cc_close(context, cache);
    krb5_free_context(context);
    return 0;
}