kuser/kswitch.c

   1 /*
   2  * Copyright (c) 2008 Kungliga Tekniska Högskolan
   3  * (Royal Institute of Technology, Stockholm, Sweden).
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  *
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  *
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  *
  17  * 3. Neither the name of the Institute nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  */
  33
  34 #include "kuser_locl.h"
  35
  36 /*
  37  *
  38  */
  39
  40 static int version_flag         = 0;
  41 static int help_flag            = 0;
  42 static char *cache;
  43 static char *principal;
  44 static char *type;
  45
  46 static struct getargs args[] = {
  47     { "type",                   't', arg_string, &type,
  48       NP_("type of credential cache", ""), "type" },
  49     { "cache",                  'c', arg_string, &cache,
  50       NP_("name of credential cache", ""), "cache" },
  51     { "principal",              'p', arg_string, &principal,
  52       NP_("name of principal", ""), "principal" },
  53     { "version",                0,   arg_flag, &version_flag,
  54       NP_("print version", ""), NULL },
  55     { "help",                   0,   arg_flag, &help_flag, NULL, NULL}
  56 };
  57
  58 static void
  59 usage (int ret) __attribute__((noreturn));
  60
  61 static void
  62 usage (int ret)
  63 {
  64     arg_printusage (args, sizeof(args)/sizeof(*args), NULL, "");
  65     exit (ret);
  66 }
  67
  68 int
  69 main (int argc, char **argv)
  70 {
  71     krb5_context context;
  72     krb5_error_code ret;
  73     krb5_ccache id = NULL;
  74     int optidx = 0;
  75
  76     setprogname (argv[0]);
  77
  78     setlocale (LC_ALL, "");
  79     bindtextdomain ("heimdal_kuser", HEIMDAL_LOCALEDIR);
  80     textdomain("heimdal_kuser");
  81
  82     ret = krb5_init_context (&context);
  83     if (ret == KRB5_CONFIG_BADFORMAT)
  84         errx (1, "krb5_init_context failed to parse configuration file");
  85     else if (ret)
  86         errx(1, "krb5_init_context failed: %d", ret);
  87
  88     if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx))
  89         usage(1);
  90
  91     if (help_flag)
  92         usage (0);
  93
  94     if(version_flag){
  95         print_version(NULL);
  96         exit(0);
  97     }
  98
  99     argc -= optidx;
 100     argv += optidx;
 101
 102     if (argc != 0)
 103         usage (1);
 104
 105     if (cache && principal)
 106         krb5_errx(context, 1,
 107                   N_("Both --cache and --principal given, choose one", ""));
 108
 109     if (principal) {
 110         krb5_principal p;
 111
 112         ret = krb5_parse_name(context, principal, &p);
 113         if (ret)
 114             krb5_err (context, 1, ret, "krb5_parse_name: %s", principal);
 115
 116         ret = krb5_cc_cache_match(context, p, &id);
 117         if (ret)
 118             krb5_err (context, 1, ret,
 119                       N_("Did not find principal: %s", ""), principal);
 120
 121         krb5_free_principal(context, p);
 122
 123     } else if (cache) {
 124         const krb5_cc_ops *ops;
 125         char *str;
 126
 127         ops = krb5_cc_get_prefix_ops(context, type);
 128         if (ops == NULL)
 129             krb5_err (context, 1, 0, "krb5_cc_get_prefix_ops");
 130
 131         asprintf(&str, "%s:%s", ops->prefix, cache);
 132         if (str == NULL)
 133             krb5_errx(context, 1, N_("out of memory", ""));
 134
 135         ret = krb5_cc_resolve(context, str, &id);
 136         if (ret)
 137             krb5_err (context, 1, ret, "krb5_cc_resolve: %s", str);
 138
 139         free(str);
 140     } else
 141         usage(1);
 142
 143     ret = krb5_cc_switch(context, id);
 144     if (ret)
 145         krb5_err (context, 1, ret, "krb5_cc_switch");
 146
 147     krb5_cc_close(context, id);
 148
 149     return 0;
 150 }