usr/src/lib/libresolv2/common/irs/getservent_r.c

   1 /*
   2  * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
   3  * Copyright (c) 1998-1999 by Internet Software Consortium.
   4  *
   5  * Permission to use, copy, modify, and distribute this software for any
   6  * purpose with or without fee is hereby granted, provided that the above
   7  * copyright notice and this permission notice appear in all copies.
   8  *
   9  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
  10  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  11  * MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR
  12  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  13  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  14  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
  15  * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  16  */
  17
  18 #if defined(LIBC_SCCS) && !defined(lint)
  19 static const char rcsid[] = "$Id: getservent_r.c,v 1.6 2006/08/01 01:14:16 marka Exp $";
  20 #endif /* LIBC_SCCS and not lint */
  21
  22 #include <port_before.h>
  23 #if !defined(DO_PTHREADS)
  24         static int getservent_r_not_required = 0;
  25 #else
  26 #include <errno.h>
  27 #include <string.h>
  28 #include <stdio.h>
  29 #include <sys/types.h>
  30 #include <netinet/in.h>
  31 #include <netdb.h>
  32 #include <sys/param.h>
  33 #include <port_after.h>
  34
  35 #ifdef SERV_R_RETURN
  36
  37 static SERV_R_RETURN
  38 copy_servent(struct servent *, struct servent *, SERV_R_COPY_ARGS);
  39
  40 SERV_R_RETURN
  41 getservbyname_r(const char *name, const char *proto,
  42                 struct servent *sptr, SERV_R_ARGS) {
  43         struct servent *se = getservbyname(name, proto);
  44 #ifdef SERV_R_SETANSWER
  45         int n = 0;
  46
  47         if (se == NULL || (n = copy_servent(se, sptr, SERV_R_COPY)) != 0)
  48                 *answerp = NULL;
  49         else
  50                 *answerp = sptr;
  51
  52         return (n);
  53 #else
  54         if (se == NULL)
  55                 return (SERV_R_BAD);
  56
  57         return (copy_servent(se, sptr, SERV_R_COPY));
  58 #endif
  59 }
  60
  61 SERV_R_RETURN
  62 getservbyport_r(int port, const char *proto,
  63                 struct servent *sptr, SERV_R_ARGS) {
  64         struct servent *se = getservbyport(port, proto);
  65 #ifdef SERV_R_SETANSWER
  66         int n = 0;
  67
  68         if (se == NULL || (n = copy_servent(se, sptr, SERV_R_COPY)) != 0)
  69                 *answerp = NULL;
  70         else
  71                 *answerp = sptr;
  72
  73         return (n);
  74 #else
  75         if (se == NULL)
  76                 return (SERV_R_BAD);
  77
  78         return (copy_servent(se, sptr, SERV_R_COPY));
  79 #endif
  80 }
  81
  82 /*%
  83  *      These assume a single context is in operation per thread.
  84  *      If this is not the case we will need to call irs directly
  85  *      rather than through the base functions.
  86  */
  87
  88 SERV_R_RETURN
  89 getservent_r(struct servent *sptr, SERV_R_ARGS) {
  90         struct servent *se = getservent();
  91 #ifdef SERV_R_SETANSWER
  92         int n = 0;
  93
  94         if (se == NULL || (n = copy_servent(se, sptr, SERV_R_COPY)) != 0)
  95                 *answerp = NULL;
  96         else
  97                 *answerp = sptr;
  98
  99         return (n);
 100 #else
 101         if (se == NULL)
 102                 return (SERV_R_BAD);
 103
 104         return (copy_servent(se, sptr, SERV_R_COPY));
 105 #endif
 106 }
 107
 108 SERV_R_SET_RETURN
 109 #ifdef SERV_R_ENT_ARGS
 110 setservent_r(int stay_open, SERV_R_ENT_ARGS)
 111 #else
 112 setservent_r(int stay_open)
 113 #endif
 114 {
 115 #ifdef SERV_R_ENT_UNUSED
 116         SERV_R_ENT_UNUSED;
 117 #endif
 118         setservent(stay_open);
 119 #ifdef SERV_R_SET_RESULT
 120         return (SERV_R_SET_RESULT);
 121 #endif
 122 }
 123
 124 SERV_R_END_RETURN
 125 #ifdef SERV_R_ENT_ARGS
 126 endservent_r(SERV_R_ENT_ARGS)
 127 #else
 128 endservent_r()
 129 #endif
 130 {
 131 #ifdef SERV_R_ENT_UNUSED
 132         SERV_R_ENT_UNUSED;
 133 #endif
 134         endservent();
 135         SERV_R_END_RESULT(SERV_R_OK);
 136 }
 137
 138 /* Private */
 139
 140 #ifndef SERVENT_DATA
 141 static SERV_R_RETURN
 142 copy_servent(struct servent *se, struct servent *sptr, SERV_R_COPY_ARGS) {
 143         char *cp;
 144         int i, n;
 145         int numptr, len;
 146
 147         /* Find out the amount of space required to store the answer. */
 148         numptr = 1; /*%< NULL ptr */
 149         len = (char *)ALIGN(buf) - buf;
 150         for (i = 0; se->s_aliases[i]; i++, numptr++) {
 151                 len += strlen(se->s_aliases[i]) + 1;
 152         }
 153         len += strlen(se->s_name) + 1;
 154         len += strlen(se->s_proto) + 1;
 155         len += numptr * sizeof(char*);
 156
 157         if (len > (int)buflen) {
 158                 errno = ERANGE;
 159                 return (SERV_R_BAD);
 160         }
 161
 162         /* copy port value */
 163         sptr->s_port = se->s_port;
 164
 165         cp = (char *)ALIGN(buf) + numptr * sizeof(char *);
 166
 167         /* copy official name */
 168         n = strlen(se->s_name) + 1;
 169         strcpy(cp, se->s_name);
 170         sptr->s_name = cp;
 171         cp += n;
 172
 173         /* copy aliases */
 174         sptr->s_aliases = (char **)ALIGN(buf);
 175         for (i = 0 ; se->s_aliases[i]; i++) {
 176                 n = strlen(se->s_aliases[i]) + 1;
 177                 strcpy(cp, se->s_aliases[i]);
 178                 sptr->s_aliases[i] = cp;
 179                 cp += n;
 180         }
 181         sptr->s_aliases[i] = NULL;
 182
 183         /* copy proto */
 184         n = strlen(se->s_proto) + 1;
 185         strcpy(cp, se->s_proto);
 186         sptr->s_proto = cp;
 187         cp += n;
 188
 189         return (SERV_R_OK);
 190 }
 191 #else /* !SERVENT_DATA */
 192 static int
 193 copy_servent(struct servent *se, struct servent *sptr, SERV_R_COPY_ARGS) {
 194         char *cp, *eob;
 195         int i, n;
 196
 197         /* copy port value */
 198         sptr->s_port = se->s_port;
 199
 200         /* copy official name */
 201         cp = sdptr->line;
 202         eob = sdptr->line + sizeof(sdptr->line);
 203         if ((n = strlen(se->s_name) + 1) < (eob - cp)) {
 204                 strcpy(cp, se->s_name);
 205                 sptr->s_name = cp;
 206                 cp += n;
 207         } else {
 208                 return (-1);
 209         }
 210
 211         /* copy aliases */
 212         i = 0;
 213         sptr->s_aliases = sdptr->serv_aliases;
 214         while (se->s_aliases[i] && i < (_MAXALIASES-1)) {
 215                 if ((n = strlen(se->s_aliases[i]) + 1) < (eob - cp)) {
 216                         strcpy(cp, se->s_aliases[i]);
 217                         sptr->s_aliases[i] = cp;
 218                         cp += n;
 219                 } else {
 220                         break;
 221                 }
 222                 i++;
 223         }
 224         sptr->s_aliases[i] = NULL;
 225
 226         /* copy proto */
 227         if ((n = strlen(se->s_proto) + 1) < (eob - cp)) {
 228                 strcpy(cp, se->s_proto);
 229                 sptr->s_proto = cp;
 230                 cp += n;
 231         } else {
 232                 return (-1);
 233         }
 234
 235         return (SERV_R_OK);
 236 }
 237 #endif /* !SERVENT_DATA */
 238 #else /*SERV_R_RETURN */
 239         static int getservent_r_unknown_system = 0;
 240 #endif /*SERV_R_RETURN */
 241 #endif /* !defined(DO_PTHREADS) */
 242 /*! \file */