1 | /* $Id$ |
---|
2 | * |
---|
3 | * This generates the files necessary to load an nfs server. |
---|
4 | * |
---|
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>. |
---|
8 | */ |
---|
9 | |
---|
10 | #include <mit-copyright.h> |
---|
11 | #include <moira.h> |
---|
12 | #include <moira_site.h> |
---|
13 | |
---|
14 | #include <sys/stat.h> |
---|
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 | |
---|
24 | EXEC SQL INCLUDE sqlca; |
---|
25 | |
---|
26 | RCSID("$Header$"); |
---|
27 | |
---|
28 | #define min(x, y) ((x) < (y) ? (x) : (y)) |
---|
29 | |
---|
30 | char *whoami = "nfs.gen"; |
---|
31 | char *db = "moira/moira"; |
---|
32 | char nfs_dir[MAXPATHLEN]; |
---|
33 | struct hash *users, *groups; |
---|
34 | |
---|
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); |
---|
39 | |
---|
40 | int main(int argc, char **argv) |
---|
41 | { |
---|
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); |
---|
50 | } |
---|
51 | |
---|
52 | initialize_sms_error_table(); |
---|
53 | sprintf(nfs_dir, "%s/nfs", DCM_DIR); |
---|
54 | |
---|
55 | EXEC SQL CONNECT :db; |
---|
56 | |
---|
57 | changed = do_nfs(); |
---|
58 | |
---|
59 | EXEC SQL COMMIT; |
---|
60 | |
---|
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); |
---|
66 | } |
---|
67 | |
---|
68 | if (argc == 2) |
---|
69 | { |
---|
70 | sprintf(cmd, "cd %s; cp %s/nfs/* .; tar cf %s .", |
---|
71 | nfs_dir, MOIRA_DIR, argv[1]); |
---|
72 | if (system(cmd)) |
---|
73 | exit(MR_TAR_FAIL); |
---|
74 | } |
---|
75 | |
---|
76 | exit(MR_SUCCESS); |
---|
77 | } |
---|
78 | |
---|
79 | |
---|
80 | /* Generate the files. Returns zero if nothing changed, non-zero otherwise. */ |
---|
81 | |
---|
82 | int do_nfs(void) |
---|
83 | { |
---|
84 | EXEC SQL BEGIN DECLARE SECTION; |
---|
85 | char machname[MACHINE_NAME_SIZE], listname[SERVERHOSTS_VALUE3_SIZE]; |
---|
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; |
---|
109 | sq_save_unique_string(machs, strdup(strtrim(machname))); |
---|
110 | sq_save_unique_string(lists, strdup(strtrim(listname))); |
---|
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); |
---|
121 | } |
---|
122 | |
---|
123 | |
---|
124 | /* Make all of the credentials lists that will be needed. Returns 0 if |
---|
125 | * no files were actually changed |
---|
126 | */ |
---|
127 | |
---|
128 | int do_lists(struct save_queue *lists) |
---|
129 | { |
---|
130 | char file[MAXPATHLEN], *u; |
---|
131 | FILE *fd; |
---|
132 | EXEC SQL BEGIN DECLARE SECTION; |
---|
133 | char *listname; |
---|
134 | int id; |
---|
135 | EXEC SQL END DECLARE SECTION; |
---|
136 | |
---|
137 | sprintf(file, "%s/list-", nfs_dir); |
---|
138 | |
---|
139 | /* build the list of everyone, and store it in a file whose name |
---|
140 | * corresponds to the empty list. |
---|
141 | */ |
---|
142 | do_everyone(); |
---|
143 | |
---|
144 | fprintf(stderr, "Building specific lists\n"); |
---|
145 | /* now do each of the lists used by an NFS server */ |
---|
146 | |
---|
147 | while (sq_get_data(lists, &listname)) |
---|
148 | { |
---|
149 | if (strlen(listname) == 0) |
---|
150 | continue; |
---|
151 | sprintf(file, "%s/list-%s", nfs_dir, listname); |
---|
152 | fd = fopen(file, "w"); |
---|
153 | if (!fd) |
---|
154 | { |
---|
155 | fprintf(stderr, "cannot open %s for output\n", file); |
---|
156 | exit(MR_OCONFIG); |
---|
157 | } |
---|
158 | |
---|
159 | EXEC SQL DECLARE m_cursor CURSOR FOR |
---|
160 | SELECT m.member_id |
---|
161 | FROM imembers m, list l |
---|
162 | WHERE m.list_id = l.list_id AND l.name = :listname |
---|
163 | AND m.member_type = 'USER' |
---|
164 | ORDER BY member_id; |
---|
165 | EXEC SQL OPEN m_cursor; |
---|
166 | while (1) |
---|
167 | { |
---|
168 | EXEC SQL FETCH m_cursor INTO :id; |
---|
169 | if (sqlca.sqlcode) |
---|
170 | break; |
---|
171 | if ((u = hash_lookup(users, id))) |
---|
172 | fprintf(fd, "%s\n", u); |
---|
173 | } |
---|
174 | EXEC SQL CLOSE m_cursor; |
---|
175 | if (fclose(fd) == EOF) |
---|
176 | { |
---|
177 | fprintf(stderr, "error closing %s\n", file); |
---|
178 | exit(MR_CCONFIG); |
---|
179 | } |
---|
180 | } |
---|
181 | /* don't free here either |
---|
182 | sq_destroy(lists); |
---|
183 | */ |
---|
184 | return 1; |
---|
185 | sqlerr: |
---|
186 | db_error(sqlca.sqlcode); |
---|
187 | exit(MR_DBMS_ERR); |
---|
188 | } |
---|
189 | |
---|
190 | |
---|
191 | /* Build the list of everybody. */ |
---|
192 | struct grp { |
---|
193 | struct grp *next; |
---|
194 | char *lid; |
---|
195 | }; |
---|
196 | |
---|
197 | struct user { |
---|
198 | char name[USERS_LOGIN_SIZE]; |
---|
199 | int uid; |
---|
200 | struct grp *lists; |
---|
201 | }; |
---|
202 | |
---|
203 | |
---|
204 | void do_everyone(void) |
---|
205 | { |
---|
206 | const buflen = MAXPATHLEN; |
---|
207 | char buf[MAXPATHLEN], *l; |
---|
208 | struct user *u; |
---|
209 | struct grp *g; |
---|
210 | struct bucket *b, **p; |
---|
211 | EXEC SQL BEGIN DECLARE SECTION; |
---|
212 | char name[USERS_LOGIN_SIZE]; |
---|
213 | int gid, id, lid, uid; |
---|
214 | EXEC SQL END DECLARE SECTION; |
---|
215 | FILE *fd; |
---|
216 | |
---|
217 | fprintf(stderr, "Building the list of everybody\n"); |
---|
218 | sprintf(buf, "%s/list-", nfs_dir); |
---|
219 | fd = fopen(buf, "w"); |
---|
220 | if (!fd) |
---|
221 | { |
---|
222 | fprintf(stderr, "cannot open %s for output\n", buf); |
---|
223 | exit(MR_OCONFIG); |
---|
224 | } |
---|
225 | |
---|
226 | /* make space for group list */ |
---|
227 | groups = create_hash(15000); |
---|
228 | |
---|
229 | /* retrieve simple groups */ |
---|
230 | EXEC SQL DECLARE l_cursor CURSOR FOR |
---|
231 | SELECT gid, list_id |
---|
232 | FROM list |
---|
233 | WHERE nfsgroup != 0 AND grouplist != 0 AND active != 0 |
---|
234 | ORDER BY list_id; |
---|
235 | EXEC SQL OPEN l_cursor; |
---|
236 | while (1) |
---|
237 | { |
---|
238 | EXEC SQL FETCH l_cursor INTO :gid, :lid; |
---|
239 | if (sqlca.sqlcode) |
---|
240 | break; |
---|
241 | sprintf(buf, ":%d", gid); |
---|
242 | hash_store(groups, lid, strdup(buf)); |
---|
243 | } |
---|
244 | EXEC SQL CLOSE l_cursor; |
---|
245 | |
---|
246 | /* now do grplists */ |
---|
247 | users = create_hash(10000); |
---|
248 | EXEC SQL DECLARE u_cursor CURSOR FOR |
---|
249 | SELECT users_id, login, unix_uid |
---|
250 | FROM users |
---|
251 | WHERE status = 1 |
---|
252 | ORDER BY users_id; |
---|
253 | EXEC SQL OPEN u_cursor; |
---|
254 | while (1) |
---|
255 | { |
---|
256 | EXEC SQL FETCH u_cursor INTO :id, :name, :uid; |
---|
257 | if (sqlca.sqlcode) |
---|
258 | break; |
---|
259 | u = malloc(sizeof(struct user)); |
---|
260 | strcpy(u->name, strtrim(name)); |
---|
261 | u->uid = uid; |
---|
262 | u->lists = NULL; |
---|
263 | hash_store(users, id, u); |
---|
264 | } |
---|
265 | EXEC SQL CLOSE u_cursor; |
---|
266 | |
---|
267 | EXEC SQL DECLARE m_cursor2 CURSOR FOR |
---|
268 | SELECT list_id, member_id |
---|
269 | FROM imembers |
---|
270 | WHERE member_type = 'USER' |
---|
271 | ORDER BY list_id; |
---|
272 | EXEC SQL OPEN m_cursor2; |
---|
273 | while (1) |
---|
274 | { |
---|
275 | EXEC SQL FETCH m_cursor2 INTO :lid, :id; |
---|
276 | if (sqlca.sqlcode) |
---|
277 | break; |
---|
278 | if ((u = hash_lookup(users, id)) && (l = hash_lookup(groups, lid))) |
---|
279 | { |
---|
280 | g = malloc(sizeof(struct grp)); |
---|
281 | g->next = u->lists; |
---|
282 | u->lists = g; |
---|
283 | g->lid = l; |
---|
284 | } |
---|
285 | } |
---|
286 | EXEC SQL CLOSE m_cursor2; |
---|
287 | |
---|
288 | for (p = &(users->data[users->size - 1]); p >= users->data; p--) |
---|
289 | { |
---|
290 | for (b = *p; b; b = b->next) |
---|
291 | { |
---|
292 | u = (struct user *)b->data; |
---|
293 | sprintf(buf, "%s:%d:101", u->name, u->uid); |
---|
294 | for (g = u->lists; g; g = g->next) |
---|
295 | { |
---|
296 | if ((strlen(buf) + strlen(g->lid)) <= buflen) |
---|
297 | strcat(buf, g->lid); |
---|
298 | else |
---|
299 | { |
---|
300 | com_err(whoami, 0, "truncated server-side grp list for %s", |
---|
301 | u->name); |
---|
302 | break; |
---|
303 | } |
---|
304 | } |
---|
305 | b->data = strdup(buf); |
---|
306 | fprintf(fd, "%s\n", buf); |
---|
307 | } |
---|
308 | } |
---|
309 | |
---|
310 | fclose(fd); |
---|
311 | return; |
---|
312 | |
---|
313 | sqlerr: |
---|
314 | db_error(sqlca.sqlcode); |
---|
315 | exit(MR_DBMS_ERR); |
---|
316 | } |
---|
317 | |
---|
318 | |
---|
319 | /* Now do each of the servers, linking the credentials list file and |
---|
320 | * compiling the quota and dirs files. |
---|
321 | */ |
---|
322 | |
---|
323 | int do_machs(struct save_queue *machs) |
---|
324 | { |
---|
325 | EXEC SQL BEGIN DECLARE SECTION; |
---|
326 | char *machname, listname[SERVERHOSTS_VALUE3_SIZE]; |
---|
327 | char dev[NFSPHYS_DEVICE_SIZE], dir[FILESYS_NAME_SIZE]; |
---|
328 | char fstype[FILESYS_LOCKERTYPE_SIZE]; |
---|
329 | int uid, quota, id, gid, flag1, flag2, flag3; |
---|
330 | EXEC SQL END DECLARE SECTION; |
---|
331 | char file[MAXPATHLEN], f1[MAXPATHLEN], f2[MAXPATHLEN], *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); |
---|
354 | } |
---|
355 | hash_store(machines, id, machname); |
---|
356 | } |
---|
357 | |
---|
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; |
---|
383 | while ((cp = strchr(dev, '/'))) |
---|
384 | *cp = '@'; |
---|
385 | sprintf(file, "%s/%s.%s.quotas", nfs_dir, |
---|
386 | (char *)hash_lookup(machines, flag2), strtrim(dev)); |
---|
387 | fd = fopen(file, "w"); |
---|
388 | if (!fd) |
---|
389 | { |
---|
390 | fprintf(stderr, "cannot open %s for output\n", file); |
---|
391 | exit(MR_OCONFIG); |
---|
392 | } |
---|
393 | prevuid = -1; |
---|
394 | quotasum = 0; |
---|
395 | } |
---|
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; |
---|
405 | } |
---|
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); |
---|
419 | } |
---|
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; |
---|
447 | while ((cp = strchr(dev, '/'))) |
---|
448 | *cp = '@'; |
---|
449 | sprintf(file, "%s/%s.%s.quotas", nfs_dir, |
---|
450 | (char *)hash_lookup(machines, flag2), strtrim(dev)); |
---|
451 | fd = fopen(file, "w"); |
---|
452 | if (!fd) |
---|
453 | { |
---|
454 | fprintf(stderr, "cannot open %s for output\n", file); |
---|
455 | exit(MR_OCONFIG); |
---|
456 | } |
---|
457 | prevuid = -1; |
---|
458 | quotasum = 0; |
---|
459 | } |
---|
460 | if (gid != prevuid) |
---|
461 | { |
---|
462 | if ((cp = hash_lookup(groups, prevuid))) |
---|
463 | prevuid = atoi(cp + 1); |
---|
464 | if (quotasum) |
---|
465 | fprintf(fd, "%d %d\n", prevuid, quotasum); |
---|
466 | prevuid = gid; |
---|
467 | quotasum = quota; |
---|
468 | } |
---|
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); |
---|
481 | } |
---|
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; |
---|
508 | while ((cp = strchr(dev, '/'))) |
---|
509 | *cp = '@'; |
---|
510 | sprintf(file, "%s/%s.%s.dirs", nfs_dir, |
---|
511 | (char *)hash_lookup(machines, flag2), strtrim(dev)); |
---|
512 | fd = fopen(file, "w"); |
---|
513 | if (!fd) |
---|
514 | { |
---|
515 | fprintf(stderr, "cannot open %s for output\n", file); |
---|
516 | exit(MR_OCONFIG); |
---|
517 | } |
---|
518 | } |
---|
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); |
---|
526 | } |
---|
527 | return 1; |
---|
528 | sqlerr: |
---|
529 | db_error(sqlca.sqlcode); |
---|
530 | exit(MR_DBMS_ERR); |
---|
531 | } |
---|
532 | |
---|