]>
Commit | Line | Data |
---|---|---|
7ac48069 | 1 | /* $Id$ |
f852c398 | 2 | * |
3 | * This generates the files necessary to load an nfs server. | |
4 | * | |
7ac48069 | 5 | * Copyright (C) 1988-1998 by the Massachusetts Institute of Technology. |
6 | * For copying and distribution information, please see the file | |
7 | * <mit-copyright.h>. | |
f852c398 | 8 | */ |
9 | ||
10 | #include <mit-copyright.h> | |
f852c398 | 11 | #include <moira.h> |
12 | #include <moira_site.h> | |
7ac48069 | 13 | |
f852c398 | 14 | #include <sys/stat.h> |
7ac48069 | 15 | #include <sys/types.h> |
16 | ||
17 | #include <stdio.h> | |
18 | #include <stdlib.h> | |
19 | #include <string.h> | |
20 | #include <unistd.h> | |
21 | ||
22 | #include "util.h" | |
23 | ||
f852c398 | 24 | EXEC SQL INCLUDE sqlca; |
25 | ||
7ac48069 | 26 | RCSID("$Header$"); |
f852c398 | 27 | |
5eaef520 | 28 | #define min(x, y) ((x) < (y) ? (x) : (y)) |
f852c398 | 29 | |
30 | char *whoami = "nfs.gen"; | |
9c04b191 | 31 | char *db = "moira/moira"; |
f852c398 | 32 | char nfs_dir[64]; |
22bc6c27 | 33 | struct hash *users, *groups; |
34 | ||
7ac48069 | 35 | int do_nfs(void); |
36 | int do_lists(struct save_queue *lists); | |
37 | void do_everyone(void); | |
38 | int do_machs(struct save_queue *machs); | |
f852c398 | 39 | |
5eaef520 | 40 | int main(int argc, char **argv) |
f852c398 | 41 | { |
5eaef520 | 42 | char cmd[64]; |
43 | struct stat sb; | |
44 | int changed = 0; | |
45 | ||
46 | if (argc > 2) | |
47 | { | |
48 | fprintf(stderr, "usage: %s [outfile]\n", argv[0]); | |
49 | exit(MR_ARGS); | |
f852c398 | 50 | } |
51 | ||
5eaef520 | 52 | initialize_sms_error_table(); |
53 | sprintf(nfs_dir, "%s/nfs", DCM_DIR); | |
f852c398 | 54 | |
5eaef520 | 55 | EXEC SQL CONNECT :db; |
f852c398 | 56 | |
5eaef520 | 57 | changed = do_nfs(); |
f852c398 | 58 | |
5eaef520 | 59 | EXEC SQL COMMIT; |
f852c398 | 60 | |
5eaef520 | 61 | if (!changed) |
62 | { | |
63 | fprintf(stderr, "No files updated.\n"); | |
64 | if (argc == 2 && stat(argv[1], &sb) == 0) | |
65 | exit(MR_NO_CHANGE); | |
f852c398 | 66 | } |
67 | ||
5eaef520 | 68 | if (argc == 2) |
69 | { | |
70 | sprintf(cmd, "cd %s; cp %s/nfs/* .; tar cf %s .", | |
59ec8dae | 71 | nfs_dir, MOIRA_DIR, argv[1]); |
5eaef520 | 72 | if (system(cmd)) |
73 | exit(MR_TAR_FAIL); | |
f852c398 | 74 | } |
75 | ||
5eaef520 | 76 | exit(MR_SUCCESS); |
f852c398 | 77 | } |
78 | ||
79 | ||
80 | /* Generate the files. Returns zero if nothing changed, non-zero otherwise. */ | |
81 | ||
5eaef520 | 82 | int do_nfs(void) |
f852c398 | 83 | { |
5eaef520 | 84 | EXEC SQL BEGIN DECLARE SECTION; |
85 | char machname[41], listname[33]; | |
86 | EXEC SQL END DECLARE SECTION; | |
87 | struct save_queue *machs, *lists; | |
88 | int changed; | |
89 | ||
90 | machs = sq_create(); | |
91 | lists = sq_create(); | |
92 | ||
93 | /* The following is declarative, not executed, | |
94 | * and so is dependent on where it is in the file, | |
95 | * not in the order of execution of statements. | |
96 | */ | |
97 | EXEC SQL WHENEVER SQLERROR GOTO sqlerr; | |
98 | ||
99 | EXEC SQL DECLARE s_cursor CURSOR FOR | |
100 | SELECT m.name, s.value3 | |
101 | FROM machine m, serverhosts s | |
102 | WHERE m.mach_id = s.mach_id AND s.service = 'NFS' AND s.enable != 0; | |
103 | EXEC SQL OPEN s_cursor; | |
104 | while (1) | |
105 | { | |
106 | EXEC SQL FETCH s_cursor INTO :machname, :listname; | |
107 | if (sqlca.sqlcode) | |
108 | break; | |
7ac48069 | 109 | sq_save_unique_string(machs, strdup(strtrim(machname))); |
110 | sq_save_unique_string(lists, strdup(strtrim(listname))); | |
5eaef520 | 111 | } |
112 | EXEC SQL CLOSE s_cursor; | |
113 | ||
114 | changed = do_lists(lists); | |
115 | EXEC SQL COMMIT; | |
116 | changed += do_machs(machs); | |
117 | return changed; | |
118 | sqlerr: | |
119 | db_error(sqlca.sqlcode); | |
120 | exit(MR_DBMS_ERR); | |
f852c398 | 121 | } |
122 | ||
123 | ||
124 | /* Make all of the credentials lists that will be needed. Returns 0 if | |
5eaef520 | 125 | * no files were actually changed |
9c04b191 | 126 | */ |
f852c398 | 127 | |
5eaef520 | 128 | int do_lists(struct save_queue *lists) |
f852c398 | 129 | { |
5eaef520 | 130 | char file[64], *u; |
5eaef520 | 131 | FILE *fd; |
132 | EXEC SQL BEGIN DECLARE SECTION; | |
7ac48069 | 133 | char *listname; |
134 | int id; | |
5eaef520 | 135 | EXEC SQL END DECLARE SECTION; |
136 | ||
137 | sprintf(file, "%s/list-", nfs_dir); | |
138 | #if 0 | |
139 | if (stat(file, &sb) == 0) | |
140 | { | |
141 | if ((ModDiff (&flag1, "users", sb.st_mtime)) || | |
142 | (ModDiff (&flag2, "list", sb.st_mtime)) || | |
143 | (ModDiff (&flag3, "imembers", sb.st_mtime)) || | |
144 | (ModDiff (&flag4, "serverhosts", sb.st_mtime))) | |
145 | exit(MR_DATE); | |
146 | if (flag1 < 0 && flag2 < 0 && flag3 < 0 && flag4 < 0) | |
147 | { | |
148 | fprintf(stderr, "The lists do not need to be rebuilt.\n"); | |
149 | return 0; | |
f852c398 | 150 | } |
151 | } | |
5eaef520 | 152 | #endif |
153 | ||
154 | /* build the list of everyone, and store it in a file whose name | |
155 | * corresponds to the empty list. | |
156 | */ | |
157 | do_everyone(); | |
158 | ||
159 | fprintf(stderr, "Building specific lists\n"); | |
160 | /* now do each of the lists used by an NFS server */ | |
161 | ||
162 | while (sq_get_data(lists, &listname)) | |
163 | { | |
164 | if (strlen(listname) == 0) | |
165 | continue; | |
166 | sprintf(file, "%s/list-%s", nfs_dir, listname); | |
167 | fd = fopen(file, "w"); | |
168 | if (!fd) | |
169 | { | |
170 | fprintf(stderr, "cannot open %s for output\n", file); | |
171 | exit(MR_OCONFIG); | |
f852c398 | 172 | } |
173 | ||
5eaef520 | 174 | EXEC SQL DECLARE m_cursor CURSOR FOR |
175 | SELECT m.member_id | |
176 | FROM imembers m, list l | |
177 | WHERE m.list_id = l.list_id AND l.name = :listname | |
178 | AND m.member_type = 'USER' | |
179 | ORDER BY member_id; | |
180 | EXEC SQL OPEN m_cursor; | |
181 | while (1) | |
182 | { | |
183 | EXEC SQL FETCH m_cursor INTO :id; | |
184 | if (sqlca.sqlcode) | |
185 | break; | |
186 | if ((u = hash_lookup(users, id))) | |
187 | fprintf(fd, "%s\n", u); | |
f852c398 | 188 | } |
5eaef520 | 189 | EXEC SQL CLOSE m_cursor; |
190 | if (fclose(fd) == EOF) | |
191 | { | |
192 | fprintf(stderr, "error closing %s\n", file); | |
193 | exit(MR_CCONFIG); | |
f852c398 | 194 | } |
195 | } | |
196 | /* don't free here either | |
197 | sq_destroy(lists); | |
198 | */ | |
5eaef520 | 199 | return 1; |
200 | sqlerr: | |
201 | db_error(sqlca.sqlcode); | |
202 | exit(MR_DBMS_ERR); | |
f852c398 | 203 | } |
204 | ||
205 | ||
206 | /* Build the list of everybody. */ | |
207 | struct grp { | |
5eaef520 | 208 | struct grp *next; |
209 | char *lid; | |
f852c398 | 210 | }; |
211 | struct user { | |
5eaef520 | 212 | char name[9]; |
213 | int uid; | |
214 | struct grp *lists; | |
f852c398 | 215 | }; |
216 | ||
217 | ||
7ac48069 | 218 | void do_everyone(void) |
f852c398 | 219 | { |
5eaef520 | 220 | char buf[BUFSIZ], *l; |
221 | struct user *u; | |
222 | struct grp *g; | |
223 | struct bucket *b, **p; | |
224 | EXEC SQL BEGIN DECLARE SECTION; | |
225 | char name[33]; | |
7ac48069 | 226 | int gid, id, lid, uid; |
5eaef520 | 227 | EXEC SQL END DECLARE SECTION; |
228 | FILE *fd; | |
5eaef520 | 229 | |
230 | fprintf(stderr, "Building the list of everybody\n"); | |
231 | sprintf(buf, "%s/list-", nfs_dir); | |
232 | fd = fopen(buf, "w"); | |
233 | if (!fd) | |
234 | { | |
235 | fprintf(stderr, "cannot open %s for output\n", buf); | |
236 | exit(MR_OCONFIG); | |
f852c398 | 237 | } |
238 | ||
5eaef520 | 239 | /* make space for group list */ |
240 | groups = create_hash(15000); | |
241 | ||
242 | /* retrieve simple groups */ | |
243 | EXEC SQL DECLARE l_cursor CURSOR FOR | |
244 | SELECT gid, list_id | |
245 | FROM list | |
246 | WHERE grouplist != 0 AND active != 0 | |
247 | ORDER BY list_id; | |
248 | EXEC SQL OPEN l_cursor; | |
249 | while (1) | |
250 | { | |
251 | EXEC SQL FETCH l_cursor INTO :gid, :lid; | |
252 | if (sqlca.sqlcode) | |
253 | break; | |
254 | sprintf(buf, ":%d", gid); | |
7ac48069 | 255 | hash_store(groups, lid, strdup(buf)); |
5eaef520 | 256 | } |
257 | EXEC SQL CLOSE l_cursor; | |
258 | ||
259 | /* now do grplists */ | |
260 | users = create_hash(10000); | |
261 | EXEC SQL DECLARE u_cursor CURSOR FOR | |
262 | SELECT users_id, login, unix_uid | |
263 | FROM users | |
264 | WHERE status = 1 | |
265 | ORDER BY users_id; | |
266 | EXEC SQL OPEN u_cursor; | |
267 | while (1) | |
268 | { | |
269 | EXEC SQL FETCH u_cursor INTO :id, :name, :uid; | |
270 | if (sqlca.sqlcode) | |
271 | break; | |
272 | u = malloc(sizeof(struct user)); | |
273 | strcpy(u->name, strtrim(name)); | |
274 | u->uid = uid; | |
275 | u->lists = NULL; | |
276 | hash_store(users, id, u); | |
277 | } | |
278 | EXEC SQL CLOSE u_cursor; | |
279 | ||
280 | EXEC SQL DECLARE m_cursor2 CURSOR FOR | |
281 | SELECT list_id, member_id | |
282 | FROM imembers | |
283 | WHERE member_type = 'USER' | |
284 | ORDER BY list_id; | |
285 | EXEC SQL OPEN m_cursor2; | |
286 | while (1) | |
287 | { | |
288 | EXEC SQL FETCH m_cursor2 INTO :lid, :id; | |
289 | if (sqlca.sqlcode) | |
290 | break; | |
7ac48069 | 291 | if ((u = hash_lookup(users, id)) && (l = hash_lookup(groups, lid))) |
5eaef520 | 292 | { |
293 | g = malloc(sizeof(struct grp)); | |
294 | g->next = u->lists; | |
295 | u->lists = g; | |
296 | g->lid = l; | |
297 | } | |
298 | } | |
299 | EXEC SQL CLOSE m_cursor2; | |
300 | ||
301 | for (p = &(users->data[users->size - 1]); p >= users->data; p--) | |
302 | { | |
303 | for (b = *p; b; b = b->next) | |
304 | { | |
305 | u = (struct user *)b->data; | |
306 | sprintf(buf, "%s:%d", u->name, u->uid); | |
307 | for (g = u->lists; g; g = g->next) | |
308 | strcat(buf, g->lid); | |
7ac48069 | 309 | b->data = strdup(buf); |
5eaef520 | 310 | fprintf(fd, "%s\n", buf); |
f852c398 | 311 | } |
312 | } | |
313 | ||
5eaef520 | 314 | fclose(fd); |
5eaef520 | 315 | sqlerr: |
316 | db_error(sqlca.sqlcode); | |
317 | exit(MR_DBMS_ERR); | |
f852c398 | 318 | } |
319 | ||
320 | ||
5eaef520 | 321 | /* Now do each of the servers, linking the credentials list file and |
f852c398 | 322 | * compiling the quota and dirs files. |
323 | */ | |
324 | ||
5eaef520 | 325 | int do_machs(struct save_queue *machs) |
f852c398 | 326 | { |
5eaef520 | 327 | EXEC SQL BEGIN DECLARE SECTION; |
328 | char *machname, listname[33], dev[33], dir[81], fstype[9]; | |
7ac48069 | 329 | int uid, quota, id, gid, flag1, flag2, flag3; |
5eaef520 | 330 | EXEC SQL END DECLARE SECTION; |
331 | char file[64], f1[64], f2[64], *cp; | |
332 | int prevuid, quotasum, olddev, oldmach; | |
333 | FILE *fd; | |
334 | struct hash *machines; | |
335 | ||
336 | fprintf(stderr, "Building machine files\n"); | |
337 | ||
338 | machines = create_hash(100); | |
339 | while (sq_get_data(machs, &machname)) | |
340 | { | |
341 | EXEC SQL SELECT s.value3, m.mach_id | |
342 | INTO :listname, :id | |
343 | FROM serverhosts s, machine m | |
344 | WHERE s.mach_id = m.mach_id AND m.name = :machname AND | |
345 | s.service = 'NFS'; | |
346 | strtrim(machname); | |
347 | sprintf(f1, "%s/list-%s", nfs_dir, strtrim(listname)); | |
348 | sprintf(f2, "%s/%s.cred", nfs_dir, machname); | |
349 | unlink(f2); /* ignore errors on this unlink */ | |
350 | if (link(f1, f2)) | |
351 | { | |
352 | fprintf(stderr, "Cannot link %s to %s\n", f1, f2); | |
353 | exit(MR_OCONFIG); | |
f852c398 | 354 | } |
5eaef520 | 355 | hash_store(machines, id, machname); |
f852c398 | 356 | } |
357 | ||
5eaef520 | 358 | olddev = oldmach = -1; |
359 | fd = stdin; | |
360 | ||
361 | EXEC SQL DECLARE q_cursor CURSOR FOR | |
362 | SELECT DISTINCT q.quota, q.entity_id, q.phys_id, n.device, n.mach_id | |
363 | FROM quota q, nfsphys n | |
364 | WHERE n.nfsphys_id = q.phys_id AND q.phys_id != 0 AND | |
365 | n.status < 16 AND q.type = 'USER' | |
366 | ORDER BY n.mach_id, q.phys_id, q.entity_id; | |
367 | EXEC SQL OPEN q_cursor; | |
368 | while (1) | |
369 | { | |
370 | EXEC SQL FETCH q_cursor INTO :quota, :uid, :flag1, :dev, :flag2; | |
371 | if (sqlca.sqlcode) | |
372 | break; | |
373 | if (flag1 != olddev || flag2 != oldmach) | |
374 | { | |
375 | if (quotasum) | |
376 | fprintf(fd, "%d %d\n", prevuid, quotasum); | |
377 | if (flag2 == 0 || !hash_lookup(machines, flag2)) | |
378 | continue; | |
379 | if (fd != stdin) | |
380 | fclose(fd); | |
381 | olddev = flag1; | |
382 | oldmach = flag2; | |
7ac48069 | 383 | while ((cp = strchr(dev, '/'))) |
5eaef520 | 384 | *cp = '@'; |
385 | sprintf(file, "%s/%s.%s.quotas", nfs_dir, | |
7ac48069 | 386 | (char *)hash_lookup(machines, flag2), strtrim(dev)); |
5eaef520 | 387 | fd = fopen(file, "w"); |
388 | if (!fd) | |
389 | { | |
390 | fprintf(stderr, "cannot open %s for output\n", file); | |
391 | exit(MR_OCONFIG); | |
f852c398 | 392 | } |
5eaef520 | 393 | prevuid = -1; |
394 | quotasum = 0; | |
f852c398 | 395 | } |
5eaef520 | 396 | if (uid != prevuid) |
397 | { | |
398 | if ((cp = hash_lookup(users, prevuid)) && | |
399 | (cp = strchr(cp, ':'))) | |
400 | prevuid = atoi(cp + 1); | |
401 | if (quotasum) | |
402 | fprintf(fd, "%d %d\n", prevuid, quotasum); | |
403 | prevuid = uid; | |
404 | quotasum = quota; | |
f852c398 | 405 | } |
5eaef520 | 406 | else |
407 | quotasum += quota; | |
408 | } | |
409 | EXEC SQL CLOSE q_cursor; | |
410 | if ((cp = hash_lookup(users, prevuid)) && | |
411 | (cp = strchr(cp, ':'))) | |
412 | prevuid = atoi(cp + 1); | |
413 | if (quotasum) | |
414 | fprintf(fd, "%d %d\n", prevuid, quotasum); | |
415 | if (fd != stdin && fclose(fd) == EOF) | |
416 | { | |
417 | fprintf(stderr, "error closing %s", file); | |
418 | exit(MR_CCONFIG); | |
f852c398 | 419 | } |
5eaef520 | 420 | |
421 | olddev = oldmach = -1; | |
422 | fd = stdin; | |
423 | EXEC SQL DECLARE q_cursor2 CURSOR FOR | |
424 | SELECT DISTINCT q.quota, q.entity_id, q.phys_id, n.device, n.mach_id, | |
425 | n.status | |
426 | FROM quota q, nfsphys n | |
427 | WHERE n.nfsphys_id = q.phys_id AND q.phys_id != 0 AND | |
428 | n.status > 15 AND q.type = 'GROUP' | |
429 | ORDER BY mach_id, phys_id, entity_id; | |
430 | EXEC SQL OPEN q_cursor2; | |
431 | while (1) | |
432 | { | |
433 | EXEC SQL FETCH q_cursor2 INTO :quota, :gid, :flag1, :dev, | |
434 | :flag2, :flag3; | |
435 | if (sqlca.sqlcode) | |
436 | break; | |
437 | if (flag1 != olddev || flag2 != oldmach) | |
438 | { | |
439 | if (quotasum) | |
440 | fprintf(fd, "%d %d\n", prevuid, quotasum); | |
441 | if (flag2 == 0 || !hash_lookup(machines, flag2)) | |
442 | continue; | |
443 | if (fd != stdin) | |
444 | fclose(fd); | |
445 | olddev = flag1; | |
446 | oldmach = flag2; | |
7ac48069 | 447 | while ((cp = strchr(dev, '/'))) |
5eaef520 | 448 | *cp = '@'; |
449 | sprintf(file, "%s/%s.%s.quotas", nfs_dir, | |
7ac48069 | 450 | (char *)hash_lookup(machines, flag2), strtrim(dev)); |
5eaef520 | 451 | fd = fopen(file, "w"); |
452 | if (!fd) | |
453 | { | |
454 | fprintf(stderr, "cannot open %s for output\n", file); | |
455 | exit(MR_OCONFIG); | |
f852c398 | 456 | } |
5eaef520 | 457 | prevuid = -1; |
458 | quotasum = 0; | |
f852c398 | 459 | } |
5eaef520 | 460 | if (gid != prevuid) |
461 | { | |
7ac48069 | 462 | if ((cp = hash_lookup(groups, prevuid))) |
5eaef520 | 463 | prevuid = atoi(cp + 1); |
464 | if (quotasum) | |
465 | fprintf(fd, "%d %d\n", prevuid, quotasum); | |
466 | prevuid = gid; | |
467 | quotasum = quota; | |
f852c398 | 468 | } |
5eaef520 | 469 | else |
470 | quotasum += quota; | |
471 | } | |
472 | EXEC SQL CLOSE q_cursor2; | |
473 | if ((cp = hash_lookup(groups, prevuid))) | |
474 | prevuid = atoi(cp + 1); | |
475 | if (quotasum) | |
476 | fprintf(fd, "%d %d\n", prevuid, quotasum); | |
477 | if (fd != stdin && fclose(fd) == EOF) | |
478 | { | |
479 | fprintf(stderr, "error closing %s", file); | |
480 | exit(MR_CCONFIG); | |
f852c398 | 481 | } |
5eaef520 | 482 | |
483 | olddev = oldmach = -1; | |
484 | fd = stdin; | |
485 | ||
486 | EXEC SQL DECLARE q_cursor3 CURSOR FOR | |
487 | SELECT DISTINCT f.name, f.lockertype, u.unix_uid, l.gid, f.phys_id, | |
488 | f.mach_id, n.device | |
489 | FROM users u, list l, nfsphys n, filesys f | |
490 | WHERE u.users_id = f.owner AND l.list_id = f.owners | |
491 | AND f.createflg != 0 AND f.phys_id != 0 AND f.type = 'NFS' | |
492 | AND f.phys_id = n.nfsphys_id | |
493 | ORDER BY mach_id, phys_id; | |
494 | EXEC SQL OPEN q_cursor3; | |
495 | while (1) | |
496 | { | |
497 | EXEC SQL FETCH q_cursor3 INTO :dir, :fstype, :uid, :gid, :flag1, | |
498 | :flag2, :dev; | |
499 | if (sqlca.sqlcode) | |
500 | break; | |
501 | if ((flag1 != olddev || flag2 != oldmach) && | |
502 | hash_lookup(machines, flag2)) | |
503 | { | |
504 | if (fd != stdin) | |
505 | fclose(fd); | |
506 | olddev = flag1; | |
507 | oldmach = flag2; | |
7ac48069 | 508 | while ((cp = strchr(dev, '/'))) |
5eaef520 | 509 | *cp = '@'; |
510 | sprintf(file, "%s/%s.%s.dirs", nfs_dir, | |
7ac48069 | 511 | (char *)hash_lookup(machines, flag2), strtrim(dev)); |
5eaef520 | 512 | fd = fopen(file, "w"); |
513 | if (!fd) | |
514 | { | |
515 | fprintf(stderr, "cannot open %s for output\n", file); | |
516 | exit(MR_OCONFIG); | |
f852c398 | 517 | } |
518 | } | |
5eaef520 | 519 | fprintf(fd, "%s %d %d %s\n", strtrim(dir), uid, gid, strtrim(fstype)); |
520 | } | |
521 | EXEC SQL CLOSE q_cursor3; | |
522 | if (fclose(fd) == EOF) | |
523 | { | |
524 | fprintf(stderr, "error closing %s", file); | |
525 | exit(MR_CCONFIG); | |
f852c398 | 526 | } |
5eaef520 | 527 | return 1; |
528 | sqlerr: | |
529 | db_error(sqlca.sqlcode); | |
530 | exit(MR_DBMS_ERR); | |
f852c398 | 531 | } |