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       NP_("type of credential cache", ""), "type" },
  51     { "cache",                  'c', arg_string, &cache,
  52       NP_("name of credential cache", ""), "cache" },
  53     { "principal",              'p', arg_string, &principal,
  54       NP_("name of principal", ""), "principal" },
  55     { "version",                0,   arg_flag, &version_flag,
  56       NP_("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     setlocale (LC_ALL, "");
  78     bindtextdomain ("heimdal_kuser", HEIMDAL_LOCALEDIR);
  79     textdomain("heimdal_kuser");
  80
  81     ret = krb5_init_context (&context);
  82     if (ret == KRB5_CONFIG_BADFORMAT)
  83         errx (1, "krb5_init_context failed to parse configuration file");
  84     else if (ret)
  85         errx(1, "krb5_init_context failed: %d", ret);
  86
  87     if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx))
  88         usage(1);
  89
  90     if (help_flag)
  91         usage (0);
  92
  93     if(version_flag){
  94         print_version(NULL);
  95         exit(0);
  96     }
  97
  98     argc -= optidx;
  99     argv += optidx;
 100
 101     if (argc != 0)
 102         usage (1);
 103
 104     if (cache && principal)
 105         krb5_errx(context, 1,
 106                   N_("Both --cache and --principal given, choose one", ""));
 107
 108     if (principal) {
 109         krb5_principal p;
 110
 111         ret = krb5_parse_name(context, principal, &p);
 112         if (ret)
 113             krb5_err (context, 1, ret, "krb5_parse_name: %s", principal);
 114
 115         ret = krb5_cc_cache_match(context, p, &id);
 116         if (ret)
 117             krb5_err (context, 1, ret,
 118                       N_("Did not find principal: %s", ""), principal);
 119
 120         krb5_free_principal(context, p);
 121
 122     } else if (cache) {
 123         const krb5_cc_ops *ops;
 124         char *str;
 125
 126         ops = krb5_cc_get_prefix_ops(context, type);
 127         if (ops == NULL)
 128             krb5_err (context, 1, 0, "krb5_cc_get_prefix_ops");
 129
 130         asprintf(&str, "%s:%s", ops->prefix, cache);
 131         if (str == NULL)
 132             krb5_errx(context, 1, N_("out of memory", ""));
 133
 134         ret = krb5_cc_resolve(context, str, &id);
 135         if (ret)
 136             krb5_err (context, 1, ret, "krb5_cc_resolve: %s", str);
 137
 138         free(str);
 139     } else
 140         usage(1);
 141
 142     ret = krb5_cc_switch(context, id);
 143     if (ret)
 144         krb5_err (context, 1, ret, "krb5_cc_switch");
 145
 146     krb5_cc_close(context, id);
 147
 148     return 0;
 149 }