1/* 2 * Copyright (c) 1997-2004 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 "ktutil_locl.h" 35#include <err.h> 36 37static void usage(int status) __attribute__((noreturn)); 38 39 40static int help_flag; 41static int version_flag; 42int verbose_flag; 43char *keytab_string; 44static char keytab_buf[256]; 45 46static struct getargs args[] = { 47 { 48 "version", 49 0, 50 arg_flag, 51 &version_flag, 52 NULL, 53 NULL 54 }, 55 { 56 "help", 57 'h', 58 arg_flag, 59 &help_flag, 60 NULL, 61 NULL 62 }, 63 { 64 "keytab", 65 'k', 66 arg_string, 67 &keytab_string, 68 "keytab", 69 "keytab to operate on" 70 }, 71 { 72 "verbose", 73 'v', 74 arg_flag, 75 &verbose_flag, 76 "verbose", 77 "run verbosely" 78 } 79}; 80 81static int num_args = sizeof(args) / sizeof(args[0]); 82 83krb5_context context; 84 85krb5_keytab 86ktutil_open_keytab(void) 87{ 88 krb5_error_code ret; 89 krb5_keytab keytab; 90 if (keytab_string == NULL) { 91 ret = krb5_kt_default_name (context, keytab_buf, sizeof(keytab_buf)); 92 if (ret) { 93 krb5_warn(context, ret, "krb5_kt_default_name"); 94 return NULL; 95 } 96 keytab_string = keytab_buf; 97 } 98 ret = krb5_kt_resolve(context, keytab_string, &keytab); 99 if (ret) { 100 krb5_warn(context, ret, "resolving keytab %s", keytab_string); 101 return NULL; 102 } 103 if (verbose_flag) 104 fprintf (stderr, "Using keytab %s\n", keytab_string); 105 106 return keytab; 107} 108 109int 110help(void *opt, int argc, char **argv) 111{ 112 if(argc == 0) { 113 sl_help(commands, 1, argv - 1 /* XXX */); 114 } else { 115 SL_cmd *c = sl_match (commands, argv[0], 0); 116 if(c == NULL) { 117 fprintf (stderr, "No such command: %s. " 118 "Try \"help\" for a list of commands\n", 119 argv[0]); 120 } else { 121 if(c->func) { 122 char shelp[] = "--help"; 123 char *fake[3]; 124 fake[0] = argv[0]; 125 fake[1] = shelp; 126 fake[2] = NULL; 127 (*c->func)(2, fake); 128 fprintf(stderr, "\n"); 129 } 130 if(c->help && *c->help) 131 fprintf (stderr, "%s\n", c->help); 132 if((++c)->name && c->func == NULL) { 133 int f = 0; 134 fprintf (stderr, "Synonyms:"); 135 while (c->name && c->func == NULL) { 136 fprintf (stderr, "%s%s", f ? ", " : " ", (c++)->name); 137 f = 1; 138 } 139 fprintf (stderr, "\n"); 140 } 141 } 142 } 143 return 0; 144} 145 146static void 147usage(int status) 148{ 149 arg_printusage(args, num_args, NULL, "command"); 150 exit(status); 151} 152 153int 154main(int argc, char **argv) 155{ 156 int optidx = 0; 157 krb5_error_code ret; 158 setprogname(argv[0]); 159 ret = krb5_init_context(&context); 160 if (ret) 161 errx (1, "krb5_init_context failed: %d", ret); 162 if(getarg(args, num_args, argc, argv, &optidx)) 163 usage(1); 164 if(help_flag) 165 usage(0); 166 if(version_flag) { 167 print_version(NULL); 168 exit(0); 169 } 170 argc -= optidx; 171 argv += optidx; 172 if(argc == 0) 173 usage(1); 174 ret = sl_command(commands, argc, argv); 175 if(ret == -1) 176 krb5_warnx (context, "unrecognized command: %s", argv[0]); 177 return ret; 178} 179