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