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
66DBM *
67ypdb_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
95void
96ypdb_close(DBM *db)
97{
98	(void)(db->close)(db);
99}
100
101/*
102 * Returns:
103 *	DATUM on success
104 *	NULL on failure
105 */
106
107datum
108ypdb_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
133datum
134ypdb_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
157datum
158ypdb_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
181datum
182ypdb_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
203int
204ypdb_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
225int
226ypdb_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