Initial import from FreeBSD RELENG_4:
[games.git] / usr.sbin / rpc.ypupdated / yp_dbupdate.c
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
34 static const char rcsid[] =
35   "$FreeBSD: src/usr.sbin/rpc.ypupdated/yp_dbupdate.c,v 1.3.2.1 2002/02/15 00:46:57 des Exp $";
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>
46 struct dom_binding {};
47 #include <rpcsvc/ypclnt.h>
48 #include <rpcsvc/ypupdate_prot.h>
49 #include "ypxfr_extern.h"
50 #include "ypupdated_extern.h"
51
52 static 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
77 int 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 }