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
45extern char *malloc();
46
47int
48main(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
180int
181match(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