Sync sys/queue.h with FreeBSD:
[dragonfly.git] / usr.bin / newkey / update.c
blob64d35ebc66c7dae48dd0ea5fbd747ce4deba07ac
1 /*
2 * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
3 * unrestricted use provided that this legend is included on all tape
4 * media and as a part of the software program in whole or part. Users
5 * may copy or modify Sun RPC without charge, but are not authorized
6 * to license or distribute it to anyone else except as part of a product or
7 * program developed by the user or with the express written consent of
8 * Sun Microsystems, Inc.
10 * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
11 * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
12 * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
14 * Sun RPC is provided with no support and without any obligation on the
15 * part of Sun Microsystems, Inc. to assist in its use, correction,
16 * modification or enhancement.
18 * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
19 * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
20 * OR ANY PART THEREOF.
22 * In no event will Sun Microsystems, Inc. be liable for any lost revenue
23 * or profits or other special, indirect and consequential damages, even if
24 * Sun has been advised of the possibility of such damages.
26 * Sun Microsystems, Inc.
27 * 2550 Garcia Avenue
28 * Mountain View, California 94043
30 * @(#)update.c 1.2 91/03/11 Copyr 1986 Sun Micro
31 * $FreeBSD: src/usr.bin/newkey/update.c,v 1.5 1999/08/28 01:04:35 peter Exp $
32 * $DragonFly: src/usr.bin/newkey/update.c,v 1.4 2005/01/11 00:29:12 joerg Exp $
36 * Copyright (C) 1986, 1989, Sun Microsystems, Inc.
40 * Administrative tool to add a new user to the publickey database
42 #include <sys/types.h>
43 #include <sys/time.h>
44 #include <sys/resource.h>
45 #include <pwd.h>
46 #include <stdio.h>
47 #include <stdlib.h>
48 #include <string.h>
49 #include <unistd.h>
51 #include <rpc/rpc.h>
52 #include <rpc/key_prot.h>
53 #ifdef YP
54 #include <rpcsvc/yp_prot.h>
55 #include <rpcsvc/ypclnt.h>
56 #include <sys/wait.h>
57 #include <netdb.h>
58 #endif /* YP */
60 #include "externs.h"
62 #ifdef YP
63 #define MAXMAPNAMELEN 256
64 #else
65 #define YPOP_CHANGE 1 /* change, do not add */
66 #define YPOP_INSERT 2 /* add, do not change */
67 #define YPOP_DELETE 3 /* delete this entry */
68 #define YPOP_STORE 4 /* add, or change */
69 #endif
71 #ifdef YP
72 static char *basename(char *);
73 static char SHELL[] = "/bin/sh";
74 static char YPDBPATH[]="/var/yp"; /* This is defined but not used! */
75 static char UPDATEFILE[] = "updaters";
76 #endif /* YP */
78 #ifdef YP
79 static int _openchild(char *, FILE **, FILE ** );
82 * Determine if requester is allowed to update the given map,
83 * and update it if so. Returns the yp status, which is zero
84 * if there is no access violation.
86 int
87 mapupdate(char *requester, char *mapname, u_int op, u_int keylen, char *key,
88 u_int datalen, char *data)
90 char updater[MAXMAPNAMELEN + 40];
91 FILE *childargs;
92 FILE *childrslt;
93 #ifdef WEXITSTATUS
94 int status;
95 #else
96 union wait status;
97 #endif
98 pid_t pid;
99 u_int yperrno;
102 #ifdef DEBUG
103 printf("%s %s\n", key, data);
104 #endif
105 (void)sprintf(updater, "make -s -f %s/%s %s", YPDBPATH, /* !!! */
106 UPDATEFILE, mapname);
107 pid = _openchild(updater, &childargs, &childrslt);
108 if (pid < 0) {
109 return (YPERR_YPERR);
113 * Write to child
115 (void)fprintf(childargs, "%s\n", requester);
116 (void)fprintf(childargs, "%u\n", op);
117 (void)fprintf(childargs, "%u\n", keylen);
118 (void)fwrite(key, (int)keylen, 1, childargs);
119 (void)fprintf(childargs, "\n");
120 (void)fprintf(childargs, "%u\n", datalen);
121 (void)fwrite(data, (int)datalen, 1, childargs);
122 (void)fprintf(childargs, "\n");
123 (void)fclose(childargs);
126 * Read from child
128 (void)fscanf(childrslt, "%d", &yperrno);
129 (void)fclose(childrslt);
131 (void)wait(&status);
132 #ifdef WEXITSTATUS
133 if (WEXITSTATUS(status) != 0) {
134 #else
135 if (status.w_retcode != 0) {
136 #endif
137 return (YPERR_YPERR);
139 return (yperrno);
143 * returns pid, or -1 for failure
145 static int
146 _openchild(char *command, FILE **fto, FILE **ffrom)
148 int i;
149 pid_t pid;
150 int pdto[2];
151 int pdfrom[2];
152 char *com;
153 struct rlimit rl;
155 if (pipe(pdto) < 0) {
156 goto error1;
158 if (pipe(pdfrom) < 0) {
159 goto error2;
161 switch (pid = fork()) {
162 case -1:
163 goto error3;
165 case 0:
167 * child: read from pdto[0], write into pdfrom[1]
169 (void)close(0);
170 (void)dup(pdto[0]);
171 (void)close(1);
172 (void)dup(pdfrom[1]);
173 getrlimit(RLIMIT_NOFILE, &rl);
174 for (i = rl.rlim_max - 1; i >= 3; i--) {
175 (void) close(i);
177 com = malloc((unsigned) strlen(command) + 6);
178 if (com == NULL) {
179 _exit(~0);
181 (void)sprintf(com, "exec %s", command);
182 execl(SHELL, basename(SHELL), "-c", com, NULL);
183 _exit(~0);
185 default:
187 * parent: write into pdto[1], read from pdfrom[0]
189 *fto = fdopen(pdto[1], "w");
190 (void)close(pdto[0]);
191 *ffrom = fdopen(pdfrom[0], "r");
192 (void)close(pdfrom[1]);
193 break;
195 return (pid);
198 * error cleanup and return
200 error3:
201 (void)close(pdfrom[0]);
202 (void)close(pdfrom[1]);
203 error2:
204 (void)close(pdto[0]);
205 (void)close(pdto[1]);
206 error1:
207 return (-1);
210 static char *
211 basename(char *path)
213 char *p;
215 p = strrchr(path, '/');
216 if (p == NULL) {
217 return (path);
218 } else {
219 return (p + 1);
223 #else /* YP */
225 #define ERR_ACCESS 1
226 #define ERR_MALLOC 2
227 #define ERR_READ 3
228 #define ERR_WRITE 4
229 #define ERR_DBASE 5
230 #define ERR_KEY 6
232 static int match( char * , char * );
235 * Determine if requester is allowed to update the given map,
236 * and update it if so. Returns the status, which is zero
237 * if there is no access violation. This function updates
238 * the local file and then shuts up.
241 localupdate(char *name, char *filename, u_int op, char *key, char *data)
243 char line[256];
244 FILE *rf;
245 FILE *wf;
246 char *tmpname;
247 int err;
250 * Check permission
252 if (strcmp(name, key) != 0) {
253 return (ERR_ACCESS);
255 if (strcmp(name, "nobody") == 0) {
257 * Can't change "nobody"s key.
259 return (ERR_ACCESS);
263 * Open files
265 tmpname = malloc(strlen(filename) + 4);
266 if (tmpname == NULL) {
267 return (ERR_MALLOC);
269 sprintf(tmpname, "%s.tmp", filename);
270 rf = fopen(filename, "r");
271 if (rf == NULL) {
272 return (ERR_READ);
274 wf = fopen(tmpname, "w");
275 if (wf == NULL) {
276 return (ERR_WRITE);
278 err = -1;
279 while (fgets(line, sizeof (line), rf)) {
280 if (err < 0 && match(line, name)) {
281 switch (op) {
282 case YPOP_INSERT:
283 err = ERR_KEY;
284 break;
285 case YPOP_STORE:
286 case YPOP_CHANGE:
287 fprintf(wf, "%s %s\n", key, data);
288 err = 0;
289 break;
290 case YPOP_DELETE:
291 /* do nothing */
292 err = 0;
293 break;
295 } else {
296 fputs(line, wf);
299 if (err < 0) {
300 switch (op) {
301 case YPOP_CHANGE:
302 case YPOP_DELETE:
303 err = ERR_KEY;
304 break;
305 case YPOP_INSERT:
306 case YPOP_STORE:
307 err = 0;
308 fprintf(wf, "%s %s\n", key, data);
309 break;
312 fclose(wf);
313 fclose(rf);
314 if (err == 0) {
315 if (rename(tmpname, filename) < 0) {
316 return (ERR_DBASE);
318 } else {
319 if (unlink(tmpname) < 0) {
320 return (ERR_DBASE);
323 return (err);
326 static int
327 match(char *line, char *name)
329 int len;
331 len = strlen(name);
332 return (strncmp(line, name, len) == 0 &&
333 (line[len] == ' ' || line[len] == '\t'));
335 #endif /* !YP */