1 /* $NetBSD: getservent_r.c,v 1.9 2008/01/06 16:34:18 christos Exp $ */ 2 3 /* 4 * Copyright (c) 1983, 1993 5 * The Regents of the University of California. All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 3. Neither the name of the University nor the names of its contributors 16 * may be used to endorse or promote products derived from this software 17 * without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 29 * SUCH DAMAGE. 30 */ 31 32 #include <sys/cdefs.h> 33 #if defined(LIBC_SCCS) && !defined(lint) 34 #if 0 35 static char sccsid[] = "@(#)getservent.c 8.1 (Berkeley) 6/4/93"; 36 #else 37 __RCSID("$NetBSD: getservent_r.c,v 1.9 2008/01/06 16:34:18 christos Exp $"); 38 #endif 39 #endif /* LIBC_SCCS and not lint */ 40 41 #include "namespace.h" 42 #include <netdb.h> 43 #include <errno.h> 44 #include <stdio.h> 45 #include <string.h> 46 #include <stdlib.h> 47 #include <fcntl.h> 48 #include <db.h> 49 50 #include "servent.h" 51 52 #ifdef __weak_alias 53 __weak_alias(endservent_r,_endservent_r) 54 __weak_alias(getservent_r,_getservent_r) 55 __weak_alias(setservent_r,_setservent_r) 56 #endif 57 58 int 59 _servent_open(struct servent_data *sd) 60 { 61 sd->flags |= _SV_FIRST; 62 if (sd->db == NULL) { 63 if ((sd->db = dbopen(_PATH_SERVICES_DB, O_RDONLY, 0, 64 DB_HASH, NULL)) != NULL) 65 sd->flags |= _SV_DB; 66 else 67 sd->db = fopen(_PATH_SERVICES, "r"); 68 } 69 return sd->db ? 0 : -1; 70 } 71 72 void 73 _servent_close(struct servent_data *sd) 74 { 75 if (sd->db) { 76 if (sd->flags & _SV_DB) { 77 DB *db = sd->db; 78 (*db->close)(db); 79 } else 80 (void)fclose((FILE *)sd->db); 81 sd->db = NULL; 82 } 83 sd->flags &= ~_SV_STAYOPEN; 84 } 85 86 87 int 88 _servent_getline(struct servent_data *sd) 89 { 90 if (sd->line) { 91 free(sd->line); 92 sd->line = NULL; 93 } 94 95 if (sd->db == NULL) 96 return -1; 97 98 if (sd->flags & _SV_DB) { 99 DB *db = sd->db; 100 DBT key, data; 101 u_int flags = (sd->flags & _SV_FIRST) ? R_FIRST : R_NEXT; 102 103 while ((*db->seq)(db, &key, &data, flags) == 0) { 104 flags = R_NEXT; 105 switch (((u_char *)key.data)[0]) { 106 case (u_char)'\377': 107 case (u_char)'\376': 108 continue; 109 default: 110 break; 111 } 112 sd->line = strdup(data.data); 113 break; 114 } 115 } else { 116 if (sd->flags & _SV_FIRST) 117 (void)rewind((FILE *)sd->db); 118 sd->line = fparseln((FILE *)sd->db, NULL, NULL, NULL, 119 FPARSELN_UNESCALL); 120 } 121 sd->flags &= ~_SV_FIRST; 122 return sd->line == NULL ? -1 : 0; 123 } 124 125 126 struct servent * 127 _servent_parseline(struct servent_data *sd, struct servent *sp) 128 { 129 size_t i = 0; 130 int oerrno; 131 char *p, *cp, **q; 132 133 if (sd->line == NULL) 134 return NULL; 135 136 sp->s_name = p = sd->line; 137 p = strpbrk(p, " \t"); 138 if (p == NULL) 139 return NULL; 140 *p++ = '\0'; 141 while (*p == ' ' || *p == '\t') 142 p++; 143 cp = strpbrk(p, ",/"); 144 if (cp == NULL) 145 return NULL; 146 *cp++ = '\0'; 147 sp->s_port = htons((u_short)atoi(p)); 148 sp->s_proto = cp; 149 if (sd->aliases == NULL) { 150 sd->maxaliases = 10; 151 sd->aliases = malloc(sd->maxaliases * sizeof(char *)); 152 if (sd->aliases == NULL) { 153 oerrno = errno; 154 endservent_r(sd); 155 errno = oerrno; 156 return NULL; 157 } 158 } 159 q = sp->s_aliases = sd->aliases; 160 cp = strpbrk(cp, " \t"); 161 if (cp != NULL) 162 *cp++ = '\0'; 163 while (cp && *cp) { 164 if (*cp == ' ' || *cp == '\t') { 165 cp++; 166 continue; 167 } 168 if (i == sd->maxaliases - 2) { 169 sd->maxaliases *= 2; 170 q = realloc(q, 171 sd->maxaliases * sizeof(char *)); 172 if (q == NULL) { 173 oerrno = errno; 174 endservent_r(sd); 175 errno = oerrno; 176 return NULL; 177 } 178 sp->s_aliases = sd->aliases = q; 179 } 180 q[i++] = cp; 181 cp = strpbrk(cp, " \t"); 182 if (cp != NULL) 183 *cp++ = '\0'; 184 } 185 q[i] = NULL; 186 return sp; 187 } 188 189 void 190 setservent_r(int f, struct servent_data *sd) 191 { 192 (void)_servent_open(sd); 193 sd->flags |= f ? _SV_STAYOPEN : 0; 194 } 195 196 void 197 endservent_r(struct servent_data *sd) 198 { 199 _servent_close(sd); 200 if (sd->aliases) { 201 free(sd->aliases); 202 sd->aliases = NULL; 203 sd->maxaliases = 0; 204 } 205 if (sd->line) { 206 free(sd->line); 207 sd->line = NULL; 208 } 209 } 210 211 struct servent * 212 getservent_r(struct servent *sp, struct servent_data *sd) 213 { 214 if (sd->db == NULL && _servent_open(sd) == -1) 215 return NULL; 216 217 for (;;) { 218 if (_servent_getline(sd) == -1) 219 return NULL; 220 if (_servent_parseline(sd, sp) == NULL) 221 continue; 222 return sp; 223 } 224 } 225