1/*
2 * Copyright (c) 1997-2003 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#ifdef HAVE_CONFIG_H
35#include <config.h>
36RCSID("$Id$");
37#endif
38#include <ctype.h>
39#ifdef KRB5
40#include <krb5.h>
41#endif
42#include <kafs.h>
43#include <roken.h>
44#include <getarg.h>
45#include <err.h>
46
47static int help_flag;
48static int version_flag;
49static getarg_strings cells;
50static char *realm;
51static getarg_strings files;
52static int unlog_flag;
53static int verbose;
54#ifdef KRB5
55static char *client_string;
56static char *cache_string;
57static int use_krb5 = 1;
58#endif
59
60struct getargs args[] = {
61    { "cell",	'c', arg_strings, &cells, "cells to get tokens for", "cell" },
62    { "file",	'p', arg_strings, &files, "files to get tokens for", "path" },
63    { "realm",	'k', arg_string, &realm, "realm for afs cell", "realm" },
64    { "unlog",	'u', arg_flag, &unlog_flag, "remove tokens" },
65#ifdef KRB5
66    { "principal",'P',arg_string,&client_string,"principal to use","principal"},
67    { "cache",   0,  arg_string, &cache_string, "ccache to use", "cache"},
68    { "v5",	 0,  arg_negative_flag, &use_krb5, "don't use Kerberos 5" },
69#endif
70    { "verbose",'v', arg_flag, &verbose },
71    { "version", 0,  arg_flag, &version_flag },
72    { "help",	'h', arg_flag, &help_flag },
73};
74
75static int num_args = sizeof(args) / sizeof(args[0]);
76
77#ifdef KRB5
78krb5_context context;
79krb5_ccache id;
80#endif
81
82static const char *
83expand_one_file(FILE *f, const char *cell)
84{
85    static char buf[1024];
86    char *p;
87
88    while (fgets (buf, sizeof(buf), f) != NULL) {
89	if(buf[0] == '>') {
90	    for(p = buf; *p && !isspace((unsigned char)*p) && *p != '#'; p++)
91		;
92	    *p = '\0';
93	    if(strncmp(buf + 1, cell, strlen(cell)) == 0)
94		return buf + 1;
95	}
96	buf[0] = '\0';
97    }
98    return NULL;
99}
100
101static const char *
102expand_cell_name(const char *cell)
103{
104    FILE *f;
105    const char *c;
106    const char **fn, *files[] = { _PATH_CELLSERVDB,
107				  _PATH_ARLA_CELLSERVDB,
108				  _PATH_OPENAFS_DEBIAN_CELLSERVDB,
109				  _PATH_ARLA_DEBIAN_CELLSERVDB,
110				  NULL };
111    for(fn = files; *fn; fn++) {
112	f = fopen(*fn, "r");
113	if(f == NULL)
114	    continue;
115	c = expand_one_file(f, cell);
116	fclose(f);
117	if(c)
118	    return c;
119    }
120    return cell;
121}
122
123static void
124usage(int ecode)
125{
126    arg_printusage(args, num_args, NULL, "[cell|path]...");
127    exit(ecode);
128}
129
130struct cell_list {
131    char *cell;
132    struct cell_list *next;
133} *cell_list;
134
135static int
136afslog_cell(const char *cell, int expand)
137{
138    struct cell_list *p, **q;
139    const char *c = cell;
140    if(expand){
141	c = expand_cell_name(cell);
142	if(c == NULL){
143	    warnx("No cell matching \"%s\" found.", cell);
144	    return -1;
145	}
146	if(verbose && strcmp(c, cell) != 0)
147	    warnx("Cell \"%s\" expanded to \"%s\"", cell, c);
148    }
149    /* add to list of cells to get tokens for, and also remove
150       duplicates; the actual afslog takes place later */
151    for(p = cell_list, q = &cell_list; p; q = &p->next, p = p->next)
152	if(strcmp(p->cell, c) == 0)
153	    return 0;
154    p = malloc(sizeof(*p));
155    if(p == NULL)
156	return -1;
157    p->cell = strdup(c);
158    if(p->cell == NULL) {
159	free(p);
160	return -1;
161    }
162    p->next = NULL;
163    *q = p;
164    return 0;
165}
166
167static int
168afslog_file(const char *path)
169{
170    char cell[64];
171    if(k_afs_cell_of_file(path, cell, sizeof(cell))){
172	warnx("No cell found for file \"%s\".", path);
173	return -1;
174    }
175    if(verbose)
176	warnx("File \"%s\" lives in cell \"%s\"", path, cell);
177    return afslog_cell(cell, 0);
178}
179
180static int
181do_afslog(const char *cell)
182{
183    int k5ret;
184
185    k5ret = 0;
186
187#ifdef KRB5
188    if(context != NULL && id != NULL && use_krb5) {
189	k5ret = krb5_afslog(context, id, cell, realm);
190	if(k5ret == 0)
191	    return 0;
192    }
193#endif
194    if (cell == NULL)
195	cell = "<default cell>";
196#ifdef KRB5
197    if (k5ret)
198	krb5_warn(context, k5ret, "krb5_afslog(%s)", cell);
199#endif
200    if (k5ret)
201	return 1;
202    return 0;
203}
204
205static void
206log_func(void *ctx, const char *str)
207{
208    fprintf(stderr, "%s\n", str);
209}
210
211int
212main(int argc, char **argv)
213{
214    int optind = 0;
215    int i;
216    int num;
217    int ret = 0;
218    int failed = 0;
219    struct cell_list *p;
220
221    setprogname(argv[0]);
222
223    if(getarg(args, num_args, argc, argv, &optind))
224	usage(1);
225    if(help_flag)
226	usage(0);
227    if(version_flag) {
228	print_version(NULL);
229	exit(0);
230    }
231
232    if(!k_hasafs())
233	errx(1, "AFS does not seem to be present on this machine");
234
235    if(unlog_flag){
236	k_unlog();
237	exit(0);
238    }
239#ifdef KRB5
240    ret = krb5_init_context(&context);
241    if (ret) {
242	context = NULL;
243    } else {
244	if (client_string) {
245	    krb5_principal client;
246
247	    ret = krb5_parse_name(context, client_string, &client);
248	    if (ret == 0)
249		ret = krb5_cc_cache_match(context, client, &id);
250	    if (ret)
251		id = NULL;
252	}
253	if (id == NULL && cache_string) {
254	    if(krb5_cc_resolve(context, cache_string, &id) != 0) {
255		krb5_warnx(context, "failed to open kerberos 5 cache '%s'",
256			   cache_string);
257		id = NULL;
258	    }
259	}
260	if (id == NULL)
261	    if(krb5_cc_default(context, &id) != 0)
262		id = NULL;
263    }
264#endif
265
266    if (verbose)
267	kafs_set_verbose(log_func, NULL);
268
269    num = 0;
270    for(i = 0; i < files.num_strings; i++){
271	afslog_file(files.strings[i]);
272	num++;
273    }
274    free_getarg_strings (&files);
275    for(i = 0; i < cells.num_strings; i++){
276	afslog_cell(cells.strings[i], 1);
277	num++;
278    }
279    free_getarg_strings (&cells);
280    for(i = optind; i < argc; i++){
281	num++;
282	if(strcmp(argv[i], ".") == 0 ||
283	   strcmp(argv[i], "..") == 0 ||
284	   strchr(argv[i], '/') ||
285	   access(argv[i], F_OK) == 0)
286	    afslog_file(argv[i]);
287	else
288	    afslog_cell(argv[i], 1);
289    }
290    if(num == 0) {
291	if(do_afslog(NULL))
292	    failed++;
293    } else
294	for(p = cell_list; p; p = p->next) {
295	    if(verbose)
296		warnx("Getting tokens for cell \"%s\"", p->cell);
297	    if(do_afslog(p->cell))
298		failed++;
299    }
300
301    return failed;
302}
303