1/*
2 * Copyright (c) 1997-2001 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: afslog.c,v 1.16 2001/05/16 22:10:15 assar Exp $");
37#endif
38#include <ctype.h>
39#include <krb5.h>
40#include <kafs.h>
41#include <roken.h>
42#include <getarg.h>
43#include <err.h>
44
45static int help_flag;
46static int version_flag;
47#if 0
48static int create_user;
49#endif
50static getarg_strings cells;
51static char *realm;
52static getarg_strings files;
53static int unlog_flag;
54static int verbose;
55
56struct getargs args[] = {
57 { "cell", 'c', arg_strings, &cells, "cells to get tokens for", "cells" },
58 { "file", 'p', arg_strings, &files, "files to get tokens for", "paths" },
59 { "realm", 'k', arg_string, &realm, "realm for afs cell", "realm" },
60 { "unlog", 'u', arg_flag, &unlog_flag, "remove tokens" },
61#if 0
62 { "create-user", 0, arg_flag, &create_user, "create user if not found" },
63#endif
64 { "verbose",'v', arg_flag, &verbose },
65 { "version", 0, arg_flag, &version_flag },
66 { "help", 'h', arg_flag, &help_flag },
67};
68
69static int num_args = sizeof(args) / sizeof(args[0]);
70
71static const char *
72expand_cell_name(const char *cell)
73{
74 FILE *f;
75 static char buf[128];
76 char *p;
77
78 f = fopen(_PATH_CELLSERVDB, "r");
79 if(f == NULL)
80 return cell;
81 while (fgets (buf, sizeof(buf), f) != NULL) {
82 if(buf[0] == '>'){
83 for(p=buf; *p && !isspace((unsigned char)*p) && *p != '#'; p++)
84 ;
85 *p = '\0';
86 if(strstr(buf, cell)){
87 fclose(f);
88 return buf + 1;
89 }
90 }
91 buf[0] = 0;
92 }
93 fclose(f);
94 return cell;
95}
96
97#if 0
98static int
99createuser (char *cell)
100{
101 char cellbuf[64];
102 char name[ANAME_SZ];
103 char instance[INST_SZ];
104 char realm[REALM_SZ];
105 char cmd[1024];
106
107 if (cell == NULL) {
108 FILE *f;
109 int len;
110
111 f = fopen (_PATH_THISCELL, "r");
112 if (f == NULL)
113 err (1, "open(%s)", _PATH_THISCELL);
114 if (fgets (cellbuf, sizeof(cellbuf), f) == NULL)
115 err (1, "read cellname from %s", _PATH_THISCELL);
116 len = strlen(cellbuf);
117 if (cellbuf[len-1] == '\n')
118 cellbuf[len-1] = '\0';
119 cell = cellbuf;
120 }
121
122 if(krb_get_default_principal(name, instance, realm))
123 errx (1, "Could not even figure out who you are");
124
125 snprintf (cmd, sizeof(cmd),
126 "pts createuser %s%s%s@%s -cell %s",
127 name, *instance ? "." : "", instance, strlwr(realm),
128 cell);
129 DEBUG("Executing %s", cmd);
130 return system(cmd);
131}
132#endif
133
134static void
135usage(int ecode)
136{
137 arg_printusage(args, num_args, NULL, "[cell]... [path]...");
138 exit(ecode);
139}
140
141static int
142afslog_cell(krb5_context context, krb5_ccache id,
143 const char *cell, int expand)
144{
145 const char *c = cell;
146 if(expand){
147 c = expand_cell_name(cell);
148 if(c == NULL){
149 krb5_warnx(context, "No cell matching \"%s\" found.", cell);
150 return -1;
151 }
152 if(verbose)
153 krb5_warnx(context, "Cell \"%s\" expanded to \"%s\"", cell, c);
154 }
155 return krb5_afslog(context, id, c, realm);
156}
157
158static int
159afslog_file(krb5_context context, krb5_ccache id,
160 const char *path)
161{
162 char cell[64];
163 if(k_afs_cell_of_file(path, cell, sizeof(cell))){
164 krb5_warnx(context, "No cell found for file \"%s\".", path);
165 return -1;
166 }
167 if(verbose)
168 krb5_warnx(context, "File \"%s\" lives in cell \"%s\"", path, cell);
169 return afslog_cell(context, id, cell, 0);
170}
171
172int
173main(int argc, char **argv)
174{
175 int optind = 0;
176 krb5_context context;
177 krb5_ccache id;
178 int i;
179 int num;
180 int ret = 0;
181
182 setprogname(argv[0]);
183
184 if(getarg(args, num_args, argc, argv, &optind))
185 usage(1);
186 if(help_flag)
187 usage(0);
188 if(version_flag) {
189 print_version(NULL);
190 exit(0);
191 }
192
193 ret = krb5_init_context(&context);
194 if (ret)
195 errx (1, "krb5_init_context failed: %d", ret);
196 if(!k_hasafs())
197 krb5_errx(context, 1,
198 "AFS doesn't seem to be present on this machine");
199
200 if(unlog_flag){
201 k_unlog();
202 exit(0);
203 }
204 krb5_cc_default(context, &id);
205 num = 0;
206 for(i = 0; i < files.num_strings; i++){
207 afslog_file(context, id, files.strings[i]);
208 num++;
209 free_getarg_strings (&files);
210 }
211 for(i = 0; i < cells.num_strings; i++){
212 afslog_cell(context, id, cells.strings[i], 1);
213 num++;
214 free_getarg_strings (&cells);
215 }
216 for(i = optind; i < argc; i++){
217 num++;
218 if(strcmp(argv[i], ".") == 0 ||
219 strcmp(argv[i], "..") == 0 ||
220 strchr(argv[i], '/') ||
221 access(argv[i], F_OK) == 0)
222 afslog_file(context, id, argv[i]);
223 else
224 afslog_cell(context, id, argv[i], 1);
225 }
226 if(num == 0) {
227 krb5_afslog(context, id, NULL, NULL);
228 }
229
230 return ret;
231}