ypdb.c revision 1.8 1 /* $NetBSD: ypdb.c,v 1.8 2002/07/06 21:39:25 wiz Exp $ */
2
3 /*
4 * Copyright (c) 1990, 1993
5 * The Regents of the University of California. All rights reserved.
6 * All rights reserved.
7 *
8 * This code is derived from software contributed to Berkeley by
9 * Margo Seltzer.
10 *
11 * This code is derived from ndbm module of BSD4.4 db (hash) by
12 * Mats O Jansson
13 *
14 * Redistribution and use in source and binary forms, with or without
15 * modification, are permitted provided that the following conditions
16 * are met:
17 * 1. Redistributions of source code must retain the above copyright
18 * notice, this list of conditions and the following disclaimer.
19 * 2. Redistributions in binary form must reproduce the above copyright
20 * notice, this list of conditions and the following disclaimer in the
21 * documentation and/or other materials provided with the distribution.
22 * 3. All advertising materials mentioning features or use of this software
23 * must display the following acknowledgement:
24 * This product includes software developed by the University of
25 * California, Berkeley and its contributors.
26 * 4. Neither the name of the University nor the names of its contributors
27 * may be used to endorse or promote products derived from this software
28 * without specific prior written permission.
29 *
30 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
31 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
32 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
33 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
34 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
35 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
36 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
37 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
38 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
39 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
40 * SUCH DAMAGE.
41 */
42
43 #include <sys/cdefs.h>
44 #ifndef lint
45 __RCSID("$NetBSD: ypdb.c,v 1.8 2002/07/06 21:39:25 wiz Exp $");
46 #endif
47
48 #include <sys/param.h>
49 #include <sys/types.h>
50
51 #include <db.h>
52 #include <errno.h>
53 #include <stdio.h>
54 #include <string.h>
55
56 #include <rpcsvc/yp.h>
57
58 #include "ypdb.h"
59
60 /*
61 * Returns:
62 * *DBM on success
63 * NULL on failure
64 */
65
66 DBM *
67 ypdb_open(const char *file, int flags, int mode)
68 {
69 char path[MAXPATHLEN], *cp;
70 DBM *db;
71 BTREEINFO info;
72
73 cp = strrchr(file, '.');
74 snprintf(path, sizeof(path), "%s%s", file,
75 (cp != NULL && strcmp(cp, ".db") == 0) ? "" : YPDB_SUFFIX);
76
77 /* try our btree format first */
78 info.flags = 0;
79 info.cachesize = 0;
80 info.maxkeypage = 0;
81 info.minkeypage = 0;
82 info.psize = 0;
83 info.compare = NULL;
84 info.prefix = NULL;
85 info.lorder = 0;
86 db = (DBM *)dbopen(path, flags, mode, DB_BTREE, (void *)&info);
87 if (db != NULL || errno != EFTYPE)
88 return (db);
89
90 /* fallback to standard hash (for sendmail's aliases.db) */
91 db = (DBM *)dbopen(path, flags, mode, DB_HASH, NULL);
92 return (db);
93 }
94
95 void
96 ypdb_close(DBM *db)
97 {
98 (void)(db->close)(db);
99 }
100
101 /*
102 * Returns:
103 * DATUM on success
104 * NULL on failure
105 */
106
107 datum
108 ypdb_fetch(DBM *db, datum key)
109 {
110 datum retkey;
111 DBT nk, nd;
112 int status;
113
114 nk.data = key.dptr;
115 nk.size = key.dsize;
116 status = (db->get)(db, &nk, &nd, 0);
117 if (status) {
118 retkey.dptr = NULL;
119 retkey.dsize = 0;
120 } else {
121 retkey.dptr = nd.data;
122 retkey.dsize = nd.size;
123 }
124 return (retkey);
125 }
126
127 /*
128 * Returns:
129 * DATUM on success
130 * NULL on failure
131 */
132
133 datum
134 ypdb_firstkey(DBM *db)
135 {
136 int status;
137 datum retkey;
138 DBT nk, nd;
139
140 status = (db->seq)(db, &nk, &nd, R_FIRST);
141 if (status) {
142 retkey.dptr = NULL;
143 retkey.dsize = 0;
144 } else {
145 retkey.dptr = nk.data;
146 retkey.dsize = nk.size;
147 }
148 return (retkey);
149 }
150
151 /*
152 * Returns:
153 * DATUM on success
154 * NULL on failure
155 */
156
157 datum
158 ypdb_nextkey(DBM *db)
159 {
160 int status;
161 datum retkey;
162 DBT nk, nd;
163
164 status = (db->seq)(db, &nk, &nd, R_NEXT);
165 if (status) {
166 retkey.dptr = NULL;
167 retkey.dsize = 0;
168 } else {
169 retkey.dptr = nk.data;
170 retkey.dsize = nk.size;
171 }
172 return (retkey);
173 }
174
175 /*
176 * Returns:
177 * DATUM on success
178 * NULL on failure
179 */
180
181 datum
182 ypdb_setkey(DBM *db, datum key)
183 {
184 int status;
185 DBT nk, nd;
186
187 nk.data = key.dptr;
188 nk.size = key.dsize;
189 status = (db->seq)(db, &nk, &nd, R_CURSOR);
190 if (status) {
191 key.dptr = NULL;
192 key.dsize = 0;
193 }
194 return (key);
195 }
196
197 /*
198 * Returns:
199 * 0 on success
200 * <0 failure
201 */
202
203 int
204 ypdb_delete(DBM *db, datum key)
205 {
206 int status;
207 DBT nk;
208
209 nk.data = key.dptr;
210 nk.size = key.dsize;
211 status = (db->del)(db, &nk, 0);
212 if (status)
213 return (-1);
214 else
215 return (0);
216 }
217
218 /*
219 * Returns:
220 * 0 on success
221 * <0 failure
222 * 1 if YPDB_INSERT and entry exists
223 */
224
225 int
226 ypdb_store(DBM *db, datum key, datum content, int flags)
227 {
228 DBT nk, nd;
229
230 if (key.dsize > YPMAXRECORD || content.dsize > YPMAXRECORD)
231 return -1;
232 nk.data = key.dptr;
233 nk.size = key.dsize;
234 nd.data = content.dptr;
235 nd.size = content.dsize;
236 return ((db->put)(db, &nk, &nd,
237 (flags == YPDB_INSERT) ? R_NOOVERWRITE : 0));
238 }
239