lib/libc/net/getservbyname_r.c

   1 /*      $NetBSD: getservbyname_r.c,v 1.9 2012/03/13 21:13:41 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[] = "@(#)getservbyname.c     8.1 (Berkeley) 6/4/93";
  36 #else
  37 __RCSID("$NetBSD: getservbyname_r.c,v 1.9 2012/03/13 21:13:41 christos Exp $");
  38 #endif
  39 #endif /* LIBC_SCCS and not lint */
  40
  41 #include "namespace.h"
  42 #include <assert.h>
  43 #include <cdbr.h>
  44 #include <netdb.h>
  45 #include <stdlib.h>
  46 #include <string.h>
  47
  48 #include "servent.h"
  49
  50 #ifdef __weak_alias
  51 __weak_alias(getservbyname_r,_getservbyname_r)
  52 #endif
  53
  54 static struct servent *
  55 _servent_getbyname(struct servent_data *sd, struct servent *sp,
  56     const char *name, const char *proto)
  57 {
  58
  59         if ((sd->flags & (_SV_CDB | _SV_PLAINFILE)) == 0)
  60                 return NULL;
  61
  62         if (sd->flags & _SV_CDB) {
  63                 uint8_t buf[255 * 2 + 2];
  64                 size_t namelen, protolen;
  65                 const uint8_t *data, *data_end;
  66                 const void *data_ptr;
  67                 size_t datalen;
  68
  69                 namelen = strlen(name);
  70                 if (namelen == 0 || namelen > 255)
  71                         return NULL;
  72                 if (proto != NULL) {
  73                         protolen = strlen(proto);
  74                         if (protolen == 0 || protolen > 255)
  75                                 return NULL;
  76                 } else
  77                         protolen = 0;
  78                 if (namelen + protolen > 255)
  79                         return NULL;
  80
  81                 buf[0] = (uint8_t)namelen;
  82                 buf[1] = (uint8_t)protolen;
  83                 memcpy(buf + 2, name, namelen);
  84                 memcpy(buf + 2 + namelen, proto, protolen);
  85
  86                 if (cdbr_find(sd->cdb, buf, 2 + namelen + protolen,
  87                     &data_ptr, &datalen))
  88                         return NULL;
  89
  90                 if (datalen < namelen + protolen + 6)
  91                         return NULL;
  92
  93                 data = data_ptr;
  94                 data_end = data + datalen;
  95                 if (protolen) {
  96                         if (data[2] != protolen)
  97                                 return NULL;
  98                         if (memcmp(data + 3, proto, protolen + 1))
  99                                 return NULL;
 100                 }
 101                 data += 3 + data[2] + 1;
 102                 if (data > data_end)
 103                         return NULL;
 104                 while (data != data_end) {
 105                         if (*data == '\0')
 106                                 return NULL;
 107                         if (data + data[0] + 2 > data_end)
 108                                 return NULL;
 109                         if (data[0] == namelen &&
 110                             memcmp(data + 1, name, namelen + 1) == 0)
 111                                 return _servent_parsedb(sd, sp, data_ptr,
 112                                     datalen);
 113                         data += data[0] + 2;
 114                 }
 115                 return NULL;
 116         } else {
 117                 while (_servent_getline(sd) != -1) {
 118                         char **cp;
 119                         if (_servent_parseline(sd, sp) == NULL)
 120                                 continue;
 121
 122                         if (strcmp(name, sp->s_name) == 0)
 123                                 goto gotname;
 124
 125                         for (cp = sp->s_aliases; *cp; cp++)
 126                                 if (strcmp(name, *cp) == 0)
 127                                         goto gotname;
 128                         continue;
 129 gotname:
 130                         if (proto == NULL || strcmp(sp->s_proto, proto) == 0)
 131                                 return sp;
 132                 }
 133                 return NULL;
 134         }
 135 }
 136
 137 struct servent *
 138 getservbyname_r(const char *name, const char *proto, struct servent *sp,
 139     struct servent_data *sd)
 140 {
 141         _DIAGASSERT(name != NULL);
 142         /* proto may be NULL */
 143
 144         setservent_r(sd->flags & _SV_STAYOPEN, sd);
 145         sp = _servent_getbyname(sd, sp, name, proto);
 146         if (!(sd->flags & _SV_STAYOPEN))
 147                 _servent_close(sd);
 148         return sp;
 149 }