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 RCSID("$Id$");
  37
  38 /*
  39  *
  40  */
  41
  42 static int version_flag         = 0;
  43 static int help_flag            = 0;
  44 static char *cache;
  45 static char *principal;
  46 static char *type;
  47
  48 static struct getargs args[] = {
  49     { "type",                   't', arg_string, &type,
  50       "type of credential cache", "type" },
  51     { "cache",                  'c', arg_string, &cache,
  52       "name of credential cache", "cache" },
  53     { "principal",              'p', arg_string, &principal,
  54       "name of principal", "principal" },
  55     { "version",                0,   arg_flag, &version_flag,
  56       "print version", NULL },
  57     { "help",                   0,   arg_flag, &help_flag, NULL, NULL}
  58 };
  59
  60 static void
  61 usage (int ret)
  62 {
  63     arg_printusage (args, sizeof(args)/sizeof(*args), NULL, "");
  64     exit (ret);
  65 }
  66
  67 int
  68 main (int argc, char **argv)
  69 {
  70     krb5_context context;
  71     krb5_error_code ret;
  72     krb5_ccache id;
  73     int optidx = 0;
  74
  75     setprogname (argv[0]);
  76
  77     ret = krb5_init_context (&context);
  78     if (ret == KRB5_CONFIG_BADFORMAT)
  79         errx (1, "krb5_init_context failed to parse configuration file");
  80     else if (ret)
  81         errx(1, "krb5_init_context failed: %d", ret);
  82
  83     if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx))
  84         usage(1);
  85
  86     if (help_flag)
  87         usage (0);
  88
  89     if(version_flag){
  90         print_version(NULL);
  91         exit(0);
  92     }
  93
  94     argc -= optidx;
  95     argv += optidx;
  96
  97     if (argc != 0)
  98         usage (1);
  99
 100     if (cache && principal)
 101         krb5_errx(context, 1, "Both --cache and --principal given, choose one");
 102
 103     if (principal) {
 104         krb5_principal p;
 105
 106         ret = krb5_parse_name(context, principal, &p);
 107         if (ret)
 108             krb5_err (context, 1, ret, "krb5_parse_name: %s", principal);
 109
 110         ret = krb5_cc_cache_match(context, p, type, &id);
 111         if (ret)
 112             krb5_err (context, 1, ret,
 113                       "Did not find principal: %s", principal);
 114
 115         krb5_free_principal(context, p);
 116
 117     } else if (cache) {
 118         const krb5_cc_ops *ops;
 119         char *str;
 120
 121         ops = krb5_cc_get_prefix_ops(context, type);
 122         if (ops == NULL)
 123             krb5_err (context, 1, 0, "krb5_cc_get_prefix_ops");
 124
 125         asprintf(&str, "%s:%s", ops->prefix, cache);
 126         if (str == NULL)
 127             krb5_errx(context, 1, "out of memory");
 128
 129         ret = krb5_cc_resolve(context, str, &id);
 130         if (ret)
 131             krb5_err (context, 1, ret, "krb5_cc_resolve: %s", str);
 132
 133         free(str);
 134     } else
 135         usage(1);
 136
 137     ret = krb5_cc_switch(context, id);
 138     if (ret)
 139         krb5_err (context, 1, ret, "krb5_cc_switch");
 140
 141     krb5_cc_close(context, id);
 142
 143     return 0;
 144 }