Home | History | Annotate | Line # | Download | only in common
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