etc/services - sync with NetBSD-8
[minix.git] / usr.sbin / services_mkdb / output_db.c
blob043f7630ed417e19d0056ceaf9c46a09a002f96a
1 /* $NetBSD: output_db.c,v 1.1 2010/04/25 00:54:46 joerg Exp $ */
3 /*-
4 * Copyright (c) 1999 The NetBSD Foundation, Inc.
5 * All rights reserved.
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Luke Mewburn and Christos Zoulas.
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
32 #include <sys/cdefs.h>
33 #ifndef lint
34 __RCSID("$NetBSD: output_db.c,v 1.1 2010/04/25 00:54:46 joerg Exp $");
35 #endif /* not lint */
37 #include <sys/param.h>
39 #include <assert.h>
40 #include <db.h>
41 #include <err.h>
42 #include <fcntl.h>
43 #include <netdb.h>
44 #include <stdio.h>
45 #include <stdlib.h>
46 #include <string.h>
47 #include <unistd.h>
48 #include <util.h>
49 #include <ctype.h>
50 #include <errno.h>
51 #include <stringlist.h>
53 #include "extern.h"
55 static DB *db;
57 static const HASHINFO hinfo = {
58 .bsize = 256,
59 .ffactor = 4,
60 .nelem = 32768,
61 .cachesize = 1024,
62 .hash = NULL,
63 .lorder = 0
66 static void store(DBT *, DBT *, int);
67 static void killproto(DBT *);
68 static const char *mkaliases(StringList *, char *, size_t);
70 int
71 db_open(const char *tname)
73 db = dbopen(tname, O_RDWR | O_CREAT | O_EXCL,
74 (S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH), DB_HASH, &hinfo);
76 return db != NULL ? 0 : -1;
79 int
80 db_close(void)
82 int rv;
84 rv = (db->close)(db);
85 db = NULL;
87 return rv;
90 void
91 db_add(StringList *sl, size_t port, const char *proto, size_t *cnt,
92 int warndup)
94 size_t i;
95 char keyb[BUFSIZ], datab[BUFSIZ], abuf[BUFSIZ];
96 DBT data, key;
97 key.data = keyb;
98 data.data = datab;
100 /* key `indirect key', data `full line' */
101 data.size = snprintf(datab, sizeof(datab), "%zu", (*cnt)++) + 1;
102 key.size = snprintf(keyb, sizeof(keyb), "%s %zu/%s %s",
103 sl->sl_str[0], port, proto, mkaliases(sl, abuf, sizeof(abuf))) + 1;
104 store(&data, &key, warndup);
106 /* key `\377port/proto', data = `indirect key' */
107 key.size = snprintf(keyb, sizeof(keyb), "\377%zu/%s",
108 port, proto) + 1;
109 store(&key, &data, warndup);
111 /* key `\377port', data = `indirect key' */
112 killproto(&key);
113 store(&key, &data, warndup);
115 /* add references for service and all aliases */
116 for (i = 0; i < sl->sl_cur; i++) {
117 /* key `\376service/proto', data = `indirect key' */
118 key.size = snprintf(keyb, sizeof(keyb), "\376%s/%s",
119 sl->sl_str[i], proto) + 1;
120 store(&key, &data, warndup);
122 /* key `\376service', data = `indirect key' */
123 killproto(&key);
124 store(&key, &data, warndup);
126 sl_free(sl, 1);
129 static void
130 killproto(DBT *key)
132 char *p, *d = key->data;
134 if ((p = strchr(d, '/')) == NULL)
135 abort();
136 *p++ = '\0';
137 key->size = p - d;
140 static void
141 store(DBT *key, DBT *data, int warndup)
143 #ifdef DEBUG
144 int k = key->size - 1;
145 int d = data->size - 1;
146 (void)printf("store [%*.*s] [%*.*s]\n",
147 k, k, (char *)key->data + 1,
148 d, d, (char *)data->data + 1);
149 #endif
150 switch ((db->put)(db, key, data, R_NOOVERWRITE)) {
151 case 0:
152 break;
153 case 1:
154 if (warndup)
155 warnx("duplicate service `%s'",
156 &((char *)key->data)[1]);
157 break;
158 case -1:
159 err(1, "put");
160 break;
161 default:
162 abort();
163 break;
167 static const char *
168 mkaliases(StringList *sl, char *buf, size_t len)
170 size_t nc, i, pos;
172 buf[0] = 0;
173 for (i = 1, pos = 0; i < sl->sl_cur; i++) {
174 nc = strlcpy(buf + pos, sl->sl_str[i], len);
175 if (nc >= len)
176 goto out;
177 pos += nc;
178 len -= nc;
179 nc = strlcpy(buf + pos, " ", len);
180 if (nc >= len)
181 goto out;
182 pos += nc;
183 len -= nc;
185 return buf;
186 out:
187 warn("aliases for `%s' truncated", sl->sl_str[0]);
188 return buf;