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  * $FreeBSD: src/usr.sbin/rpc.ypupdated/yp_dbupdate.c,v 1.3.2.1 2002/02/15 00:46:57 des Exp $
33  * $DragonFly: src/usr.sbin/rpc.ypupdated/yp_dbupdate.c,v 1.3 2004/12/18 22:48:14 swildner Exp $
34  */
35 
36 #include <sys/fcntl.h>
37 
38 #include <stdio.h>
39 #include <string.h>
40 #include <errno.h>
41 #include <limits.h>
42 #include <db.h>
43 #include <unistd.h>
44 struct dom_binding {};
45 #include <rpcsvc/ypclnt.h>
46 #include <rpcsvc/ypupdate_prot.h>
47 #include "ypxfr_extern.h"
48 #include "ypupdated_extern.h"
49 
50 static int yp_domake(map, domain)
51 	char *map;
52 	char *domain;
53 {
54 	int pid;
55 
56 	switch ((pid = fork())) {
57 	case 0:
58 		execlp(MAP_UPDATE_PATH, MAP_UPDATE, map, domain, 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 
75 int ypmap_update(netname, map, op, keylen, keyval, datlen, datval)
76 	char *netname;
77 	char *map;
78 	unsigned int op;
79 	unsigned int keylen;
80 	char *keyval;
81 	unsigned int datlen;
82 	char *datval;
83 {
84 	DB *dbp;
85 	DBT key = { NULL, 0 }, data = { NULL, 0 };
86 	char *yp_last = "YP_LAST_MODIFIED";
87 	char yplastbuf[YPMAXRECORD];
88 	char *domptr;
89 	int rval = 0;
90 
91 	if ((domptr = strchr(netname, '@')) == NULL)
92 		return(ERR_ACCESS);
93 	domptr++;
94 
95 
96 	dbp = yp_open_db_rw(domptr, map, O_RDWR);
97 	if (dbp == NULL)
98 		return(ERR_DBASE);
99 
100 	key.data = keyval;
101 	key.size = keylen;
102 	data.data = datval;
103 	data.size = datlen;
104 
105 	switch (op) {
106 	case YPOP_DELETE: /* delete this entry */
107 		rval = yp_del_record(dbp, &key);
108 		if (rval == YP_TRUE)
109 			rval = 0;
110 		break;
111 	case YPOP_INSERT: /* add, do not change */
112 		rval = yp_put_record(dbp, &key, &data, 0);
113 		if (rval == YP_TRUE)
114 			rval = 0;
115 		break;
116 	case YPOP_STORE: /* add, or change */
117 		rval = yp_put_record(dbp, &key, &data, 1);
118 		if (rval == YP_TRUE)
119 			rval = 0;
120 		break;
121 	case YPOP_CHANGE: /* change, do not add */
122 		if (yp_get_record(domptr, map, &key, &data, 0) != YP_TRUE) {
123 			rval = ERR_KEY;
124 			break;
125 		}
126 		rval = yp_put_record(dbp, &key, &data, 1);
127 		if (rval == YP_TRUE)
128 			rval = 0;
129 		break;
130 	default:
131 		yp_error("unknown update command: (%d)", op);
132 	}
133 
134 	if (rval) {
135 		(dbp->close)(dbp);
136 		return(rval);
137 	}
138 
139 	snprintf(yplastbuf, sizeof(yplastbuf), "%lu", time(NULL));
140 	key.data = yp_last;
141 	key.size = strlen(yp_last);
142 	data.data = (char *)&yplastbuf;
143 	data.size = strlen(yplastbuf);
144 	if (yp_put_record(dbp, &key, &data, 1) != YP_TRUE) {
145 		yp_error("failed to update timestamp in %s/%s", domptr, map);
146 		(dbp->close)(dbp);
147 		return(ERR_DBASE);
148 	}
149 
150 	(dbp->close)(dbp);
151 	return(yp_domake(map, domptr));
152 }
153