xref: /illumos-gate/usr/src/cmd/ypcmd/udpublickey.c (revision 4de2612967d06c4fdbf524a62556a1e8118a006f)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License, Version 1.0 only
6  * (the "License").  You may not use this file except in compliance
7  * with the License.
8  *
9  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10  * or http://www.opensolaris.org/os/licensing.
11  * See the License for the specific language governing permissions
12  * and limitations under the License.
13  *
14  * When distributing Covered Code, include this CDDL HEADER in each
15  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16  * If applicable, add the following below this CDDL HEADER, with the
17  * fields enclosed by brackets "[]" replaced with your own identifying
18  * information: Portions Copyright [yyyy] [name of copyright owner]
19  *
20  * CDDL HEADER END
21  *
22  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
23  * Use is subject to license terms.
24  */
25 
26 /*	Copyright (c) 1983, 1984, 1985, 1986, 1987, 1988, 1989 AT&T */
27 /*	  All Rights Reserved   */
28 
29 /*
30  * Portions of this source code were derived from Berkeley
31  * under license from the Regents of the University of
32  * California.
33  */
34 
35 #pragma ident	"%Z%%M%	%I%	%E% SMI"
36 
37 /*
38  * YP updater for public key map
39  */
40 #include <stdio.h>
41 #include <rpc/rpc.h>
42 #include <rpcsvc/ypclnt.h>
43 #include <sys/file.h>
44 
45 extern char *malloc();
46 
47 int
48 main(argc, argv)
49 	int argc;
50 	char *argv[];
51 {
52 	unsigned op;
53 	char name[MAXNETNAMELEN + 1];
54 	char key[256];
55 	char data[256];
56 	char line[256];
57 	unsigned keylen;
58 	unsigned datalen;
59 	FILE *rf;
60 	FILE *wf;
61 	char *fname;
62 	char *tmpname;
63 	int err;
64 
65 
66 	if (argc !=  3) {
67 		exit(YPERR_YPERR);
68 	}
69 	fname = argv[1];
70 	tmpname = malloc(strlen(fname) + 4);
71 	if (tmpname == NULL) {
72 		exit(YPERR_YPERR);
73 	}
74 	sprintf(tmpname, "%s.tmp", fname);
75 
76 	/*
77 	 * Get input
78 	 */
79 	if (! scanf("%s\n", name)) {
80 		exit(YPERR_YPERR);
81 	}
82 	if (! scanf("%u\n", &op)) {
83 		exit(YPERR_YPERR);
84 	}
85 	if (! scanf("%u\n", &keylen)) {
86 		exit(YPERR_YPERR);
87 	}
88 	if (! fread(key, keylen, 1, stdin)) {
89 		exit(YPERR_YPERR);
90 	}
91 	key[keylen] = 0;
92 	if (! scanf("%u\n", &datalen)) {
93 		exit(YPERR_YPERR);
94 	}
95 	if (! fread(data, datalen, 1, stdin)) {
96 		exit(YPERR_YPERR);
97 	}
98 	data[datalen] = 0;
99 
100 	/*
101 	 * Check permission
102 	 */
103 	if (strcmp(name, key) != 0) {
104 		exit(YPERR_ACCESS);
105 	}
106 	if (strcmp(name, "nobody") == 0) {
107 		/*
108 		 * Can't change "nobody"s key.
109 		 */
110 		exit(YPERR_ACCESS);
111 	}
112 
113 	/*
114 	 * Open files
115 	 */
116 	rf = fopen(fname, "r");
117 	if (rf == NULL) {
118 		exit(YPERR_YPERR);
119 	}
120 	wf = fopen(tmpname, "w");
121 	if (wf == NULL) {
122 		exit(YPERR_YPERR);
123 	}
124 	err = -1;
125 	while (fgets(line, sizeof (line), rf)) {
126 		if (err < 0 && match(line, name)) {
127 			switch (op) {
128 			case YPOP_INSERT:
129 				err = YPERR_KEY;
130 				break;
131 			case YPOP_STORE:
132 			case YPOP_CHANGE:
133 				fprintf(wf, "%s %s\n", key, data);
134 				err = 0;
135 				break;
136 			case YPOP_DELETE:
137 				/* do nothing */
138 				err = 0;
139 				break;
140 			}
141 		} else {
142 			fputs(line, wf);
143 		}
144 	}
145 	if (err < 0) {
146 		switch (op) {
147 		case YPOP_CHANGE:
148 		case YPOP_DELETE:
149 			err = YPERR_KEY;
150 			break;
151 		case YPOP_INSERT:
152 		case YPOP_STORE:
153 			err = 0;
154 			fprintf(wf, "%s %s\n", key, data);
155 			break;
156 		}
157 	}
158 	fclose(wf);
159 	fclose(rf);
160 	if (err == 0) {
161 		if (rename(tmpname, fname) < 0) {
162 			exit(YPERR_YPERR);
163 		}
164 	} else {
165 		if (unlink(tmpname) < 0) {
166 			exit(YPERR_YPERR);
167 		}
168 	}
169 	if (fork() == 0) {
170 		close(0); close(1); close(2);
171 		open("/dev/null", O_RDWR, 0);
172 		dup(0); dup(0);
173 		execl("/bin/sh", "sh", "-c", argv[2], NULL);
174 	}
175 	return (err);
176 	/* NOTREACHED */
177 }
178 
179 
180 int
181 match(line, name)
182 	char *line;
183 	char *name;
184 {
185 	int len;
186 
187 	len = strlen(name);
188 	return (strncmp(line, name, len) == 0 &&
189 		(line[len] == ' ' || line[len] == '\t'));
190 }
191