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>
36 RCSID("$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
47 static int help_flag;
48 static int version_flag;
49 static getarg_strings cells;
50 static char *realm;
51 static getarg_strings files;
52 static int unlog_flag;
53 static int verbose;
54 #ifdef KRB5
55 static char *client_string;
56 static char *cache_string;
57 static int use_krb5 = 1;
58 #endif
59
60 struct 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
75 static int num_args = sizeof(args) / sizeof(args[0]);
76
77 #ifdef KRB5
78 krb5_context context;
79 krb5_ccache id;
80 #endif
81
82 static const char *
expand_one_file(FILE * f,const char * cell)83 expand_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
101 static const char *
expand_cell_name(const char * cell)102 expand_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
123 static void
usage(int ecode)124 usage(int ecode)
125 {
126 arg_printusage(args, num_args, NULL, "[cell|path]...");
127 exit(ecode);
128 }
129
130 struct cell_list {
131 char *cell;
132 struct cell_list *next;
133 } *cell_list;
134
135 static int
afslog_cell(const char * cell,int expand)136 afslog_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
167 static int
afslog_file(const char * path)168 afslog_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
180 static int
do_afslog(const char * cell)181 do_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
205 static void
log_func(void * ctx,const char * str)206 log_func(void *ctx, const char *str)
207 {
208 fprintf(stderr, "%s\n", str);
209 }
210
211 int
main(int argc,char ** argv)212 main(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