yp_dbupdate.c revision 50479
1/*
2 * Copyright (c) 1996
3 *	Bill Paul <wpaul@ctr.columbia.edu>.  All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 *    must display the following acknowledgement:
15 *	This product includes software developed by Bill Paul.
16 * 4. Neither the name of the author nor the names of any co-contributors
17 *    may be used to endorse or promote products derived from this software
18 *    without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY Bill Paul AND CONTRIBUTORS ``AS IS'' AND
21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23 * ARE DISCLAIMED.  IN NO EVENT SHALL Bill Paul OR CONTRIBUTORS BE LIABLE
24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30 * SUCH DAMAGE.
31 */
32
33#ifndef lint
34static const char rcsid[] =
35  "$FreeBSD: head/usr.sbin/rpc.ypupdated/yp_dbupdate.c 50479 1999-08-28 01:35:59Z peter $";
36#endif /* not lint */
37
38#include <sys/fcntl.h>
39
40#include <stdio.h>
41#include <string.h>
42#include <errno.h>
43#include <limits.h>
44#include <db.h>
45#include <unistd.h>
46struct dom_binding {};
47#include <rpcsvc/ypclnt.h>
48#include <rpcsvc/ypupdate_prot.h>
49#include "ypxfr_extern.h"
50#include "ypupdated_extern.h"
51
52static int yp_domake(map, domain)
53	char *map;
54	char *domain;
55{
56	int pid;
57
58	switch((pid = fork())) {
59	case 0:
60		execlp(MAP_UPDATE_PATH, MAP_UPDATE, map, domain, NULL);
61		yp_error("couldn't exec map update process: %s",
62						strerror(errno));
63		exit(1);
64		break;
65	case -1:
66		yp_error("fork() failed: %s", strerror(errno));
67		return(YPERR_YPERR);
68		break;
69	default:
70		children++;
71		break;
72	}
73
74	return(0);
75}
76
77int ypmap_update(netname, map, op, keylen, keyval, datlen, datval)
78	char *netname;
79	char *map;
80	unsigned int op;
81	unsigned int keylen;
82	char *keyval;
83	unsigned int datlen;
84	char *datval;
85{
86	DB *dbp;
87	DBT key = { NULL, 0 }, data = { NULL, 0 };
88	char *yp_last = "YP_LAST_MODIFIED";
89	char yplastbuf[YPMAXRECORD];
90	char *domptr;
91	int rval = 0;
92
93	if ((domptr = strchr(netname, '@')) == NULL)
94		return(ERR_ACCESS);
95	domptr++;
96
97
98	dbp = yp_open_db_rw(domptr, map, O_RDWR);
99	if (dbp == NULL)
100		return(ERR_DBASE);
101
102	key.data = keyval;
103	key.size = keylen;
104	data.data = datval;
105	data.size = datlen;
106
107	switch(op) {
108	case YPOP_DELETE: /* delete this entry */
109		rval = yp_del_record(dbp, &key);
110		if (rval == YP_TRUE)
111			rval = 0;
112		break;
113	case YPOP_INSERT: /* add, do not change */
114		rval = yp_put_record(dbp, &key, &data, 0);
115		if (rval == YP_TRUE)
116			rval = 0;
117		break;
118	case YPOP_STORE: /* add, or change */
119		rval = yp_put_record(dbp, &key, &data, 1);
120		if (rval == YP_TRUE)
121			rval = 0;
122		break;
123	case YPOP_CHANGE: /* change, do not add */
124		if (yp_get_record(domptr, map, &key, &data, 0) != YP_TRUE) {
125			rval = ERR_KEY;
126			break;
127		}
128		rval = yp_put_record(dbp, &key, &data, 1);
129		if (rval == YP_TRUE)
130			rval = 0;
131		break;
132	default:
133		yp_error("unknown update command: (%d)", op);
134	}
135
136	if (rval) {
137		(void)(dbp->close)(dbp);
138		return(rval);
139	}
140
141	snprintf(yplastbuf, sizeof(yplastbuf), "%lu", time(NULL));
142	key.data = yp_last;
143	key.size = strlen(yp_last);
144	data.data = (char *)&yplastbuf;
145	data.size = strlen(yplastbuf);
146	if (yp_put_record(dbp, &key, &data, 1) != YP_TRUE) {
147		yp_error("failed to update timestamp in %s/%s", domptr, map);
148		(void)(dbp->close)(dbp);
149		return(ERR_DBASE);
150	}
151
152	(void)(dbp->close)(dbp);
153	return(yp_domake(map, domptr));
154}
155