yp_dbupdate.c revision 228679 
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#include <sys/cdefs.h>
34__FBSDID("$FreeBSD: head/usr.sbin/rpc.ypupdated/yp_dbupdate.c 228679 2011-12-18 00:44:59Z dim $");
35
36#include <sys/fcntl.h>
37
38#include <stdint.h>
39#include <stdio.h>
40#include <stdlib.h>
41#include <string.h>
42#include <errno.h>
43#include <limits.h>
44#include <db.h>
45#include <unistd.h>
46#include <rpcsvc/ypclnt.h>
47#include <rpcsvc/ypupdate_prot.h>
48#include "ypxfr_extern.h"
49#include "ypupdated_extern.h"
50
51static int
52yp_domake(char *map, char *domain)
53{
54	int pid;
55
56	switch ((pid = fork())) {
57	case 0:
58		execlp(MAP_UPDATE_PATH, MAP_UPDATE, map, domain, (char *)NULL);
59		yp_error("couldn't exec map update process: %s",
60						strerror(errno));
61		exit(1);
62		break;
63	case -1:
64		yp_error("fork() failed: %s", strerror(errno));
65		return(YPERR_YPERR);
66		break;
67	default:
68		children++;
69		break;
70	}
71
72	return(0);
73}
74
75int
76ypmap_update(char *netname, char *map, unsigned int op, unsigned int keylen,
77    char *keyval, unsigned int datlen, char *datval)
78{
79	DB *dbp;
80	DBT key = { NULL, 0 }, data = { NULL, 0 };
81	char *yp_last = "YP_LAST_MODIFIED";
82	char yplastbuf[YPMAXRECORD];
83	char *domptr;
84	int rval = 0;
85
86	if ((domptr = strchr(netname, '@')) == NULL)
87		return(ERR_ACCESS);
88	domptr++;
89
90
91	dbp = yp_open_db_rw(domptr, map, O_RDWR);
92	if (dbp == NULL)
93		return(ERR_DBASE);
94
95	key.data = keyval;
96	key.size = keylen;
97	data.data = datval;
98	data.size = datlen;
99
100	switch (op) {
101	case YPOP_DELETE: /* delete this entry */
102		rval = yp_del_record(dbp, &key);
103		if (rval == YP_TRUE)
104			rval = 0;
105		break;
106	case YPOP_INSERT: /* add, do not change */
107		rval = yp_put_record(dbp, &key, &data, 0);
108		if (rval == YP_TRUE)
109			rval = 0;
110		break;
111	case YPOP_STORE: /* add, or change */
112		rval = yp_put_record(dbp, &key, &data, 1);
113		if (rval == YP_TRUE)
114			rval = 0;
115		break;
116	case YPOP_CHANGE: /* change, do not add */
117		if (yp_get_record(domptr, map, &key, &data, 0) != YP_TRUE) {
118			rval = ERR_KEY;
119			break;
120		}
121		rval = yp_put_record(dbp, &key, &data, 1);
122		if (rval == YP_TRUE)
123			rval = 0;
124		break;
125	default:
126		yp_error("unknown update command: (%d)", op);
127	}
128
129	if (rval) {
130		(void)(dbp->close)(dbp);
131		return(rval);
132	}
133
134	snprintf(yplastbuf, sizeof(yplastbuf), "%ju", (intmax_t)time(NULL));
135	key.data = yp_last;
136	key.size = strlen(yp_last);
137	data.data = (char *)&yplastbuf;
138	data.size = strlen(yplastbuf);
139	if (yp_put_record(dbp, &key, &data, 1) != YP_TRUE) {
140		yp_error("failed to update timestamp in %s/%s", domptr, map);
141		(void)(dbp->close)(dbp);
142		return(ERR_DBASE);
143	}
144
145	(void)(dbp->close)(dbp);
146	return(yp_domake(map, domptr));
147}
148