StdLib/BsdSocketLib/getservent.c

   1 /*
   2  * Copyright (c) 1983, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *      This product includes software developed by the University of
  16  *      California, Berkeley and its contributors.
  17  * 4. Neither the name of the University nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  */
  33
  34 #if defined(LIBC_SCCS) && !defined(lint)
  35 static char sccsid[] = "@(#)getservent.c        8.1 (Berkeley) 6/4/93";
  36 #endif /* LIBC_SCCS and not lint */
  37
  38 #include <sys/types.h>
  39 #include <sys/socket.h>
  40 #include <netdb.h>
  41 #include <stdio.h>
  42 #include <string.h>
  43 #include <stdlib.h>
  44 #ifdef YP
  45 #include <rpc/rpc.h>
  46 #include <rpcsvc/yp_prot.h>
  47 #include <rpcsvc/ypclnt.h>
  48 static int serv_stepping_yp = 0;
  49 extern int _yp_check __P(( char ** ));
  50 #endif
  51
  52
  53 #define MAXALIASES      35
  54
  55 static FILE *servf = NULL;
  56 static char line[BUFSIZ+1];
  57 static struct servent serv;
  58 static char *serv_aliases[MAXALIASES];
  59 int _serv_stayopen;
  60
  61 #ifdef YP
  62 char *___getservbyname_yp = NULL;
  63 char *___getservbyproto_yp = NULL;
  64 int ___getservbyport_yp = 0;
  65 static char *yp_domain = NULL;
  66
  67 static int
  68 _getservbyport_yp(line)
  69         char *line;
  70 {
  71         char *result;
  72         int resultlen;
  73         char buf[YPMAXRECORD + 2];
  74         int rv;
  75
  76         snprintf(buf, sizeof(buf), "%d/%s", ntohs(___getservbyport_yp),
  77                                                 ___getservbyproto_yp);
  78
  79         ___getservbyport_yp = 0;
  80         ___getservbyproto_yp = NULL;
  81
  82         if(!yp_domain) {
  83                 if(yp_get_default_domain(&yp_domain))
  84                         return (0);
  85         }
  86
  87         /*
  88          * We have to be a little flexible here. Ideally you're supposed
  89          * to have both a services.byname and a services.byport map, but
  90          * some systems have only services.byname. FreeBSD cheats a little
  91          * by putting the services.byport information in the same map as
  92          * services.byname so that either case will work. We allow for both
  93          * possibilities here: if there is no services.byport map, we try
  94          * services.byname instead.
  95          */
  96         if ((rv = yp_match(yp_domain, "services.byport", buf, strlen(buf),
  97                                                 &result, &resultlen))) {
  98                 if (rv == YPERR_MAP) {
  99                         if (yp_match(yp_domain, "services.byname", buf,
 100                                         strlen(buf), &result, &resultlen))
 101                         return(0);
 102                 } else
 103                         return(0);
 104         }
 105
 106         /* getservent() expects lines terminated with \n -- make it happy */
 107         snprintf(line, BUFSIZ, "%.*s\n", resultlen, result);
 108
 109         free(result);
 110         return(1);
 111 }
 112
 113 static int
 114 _getservbyname_yp(line)
 115         char *line;
 116 {
 117         char *result;
 118         int resultlen;
 119         char buf[YPMAXRECORD + 2];
 120
 121         if(!yp_domain) {
 122                 if(yp_get_default_domain(&yp_domain))
 123                         return (0);
 124         }
 125
 126         snprintf(buf, sizeof(buf), "%s/%s", ___getservbyname_yp,
 127                                                 ___getservbyproto_yp);
 128
 129         ___getservbyname_yp = 0;
 130         ___getservbyproto_yp = NULL;
 131
 132         if (yp_match(yp_domain, "services.byname", buf, strlen(buf),
 133                                                 &result, &resultlen)) {
 134                 return(0);
 135         }
 136
 137         /* getservent() expects lines terminated with \n -- make it happy */
 138         snprintf(line, BUFSIZ, "%.*s\n", resultlen, result);
 139
 140         free(result);
 141         return(1);
 142 }
 143
 144 static int
 145 _getservent_yp(line)
 146         char *line;
 147 {
 148         static char *key = NULL;
 149         static int keylen;
 150         char *lastkey, *result;
 151         int resultlen;
 152         int rv;
 153
 154         if(!yp_domain) {
 155                 if(yp_get_default_domain(&yp_domain))
 156                         return (0);
 157         }
 158
 159         if (!serv_stepping_yp) {
 160                 if (key)
 161                         free(key);
 162                 if ((rv = yp_first(yp_domain, "services.byname", &key, &keylen,
 163                              &result, &resultlen))) {
 164                         serv_stepping_yp = 0;
 165                         return(0);
 166                 }
 167                 serv_stepping_yp = 1;
 168         } else {
 169                 lastkey = key;
 170                 rv = yp_next(yp_domain, "services.byname", key, keylen, &key,
 171                              &keylen, &result, &resultlen);
 172                 free(lastkey);
 173                 if (rv) {
 174                         serv_stepping_yp = 0;
 175                         return (0);
 176                 }
 177         }
 178
 179         /* getservent() expects lines terminated with \n -- make it happy */
 180         snprintf(line, BUFSIZ, "%.*s\n", resultlen, result);
 181
 182         free(result);
 183
 184         return(1);
 185 }
 186 #endif
 187
 188 void
 189 setservent(int f)
 190 {
 191         if (servf == NULL)
 192                 servf = fopen(_PATH_SERVICES, "r" );
 193         else
 194                 rewind(servf);
 195         _serv_stayopen |= f;
 196 }
 197
 198 void
 199 endservent()
 200 {
 201         if (servf) {
 202                 fclose(servf);
 203                 servf = NULL;
 204         }
 205         _serv_stayopen = 0;
 206 }
 207
 208 struct servent *
 209 getservent()
 210 {
 211         char *p;
 212         register char *cp, **q;
 213
 214 #ifdef YP
 215         if (serv_stepping_yp && _getservent_yp(line)) {
 216                 p = (char *)&line;
 217                 goto unpack;
 218         }
 219 tryagain:
 220 #endif
 221         if (servf == NULL && (servf = fopen(_PATH_SERVICES, "r" )) == NULL)
 222                 return (NULL);
 223 again:
 224         if ((p = fgets(line, BUFSIZ, servf)) == NULL)
 225                 return (NULL);
 226 #ifdef YP
 227         if (*p == '+' && _yp_check(NULL)) {
 228                 if (___getservbyname_yp != NULL) {
 229                         if (!_getservbyname_yp(line))
 230                                 goto tryagain;
 231                 }
 232                 else if (___getservbyport_yp != 0) {
 233                         if (!_getservbyport_yp(line))
 234                                 goto tryagain;
 235                 }
 236                 else if (!_getservent_yp(line))
 237                         goto tryagain;
 238         }
 239 unpack:
 240 #endif
 241         if (*p == '#')
 242                 goto again;
 243         cp = strpbrk(p, "#\n");
 244         if (cp == NULL)
 245                 goto again;
 246         *cp = '\0';
 247         serv.s_name = p;
 248         p = strpbrk(p, " \t");
 249         if (p == NULL)
 250                 goto again;
 251         *p++ = '\0';
 252         while (*p == ' ' || *p == '\t')
 253                 p++;
 254         cp = strpbrk(p, ",/");
 255         if (cp == NULL)
 256                 goto again;
 257         *cp++ = '\0';
 258         serv.s_port = htons((u_short)atoi(p));
 259         serv.s_proto = cp;
 260         q = serv.s_aliases = serv_aliases;
 261         cp = strpbrk(cp, " \t");
 262         if (cp != NULL)
 263                 *cp++ = '\0';
 264         while (cp && *cp) {
 265                 if (*cp == ' ' || *cp == '\t') {
 266                         cp++;
 267                         continue;
 268                 }
 269                 if (q < &serv_aliases[MAXALIASES - 1])
 270                         *q++ = cp;
 271                 cp = strpbrk(cp, " \t");
 272                 if (cp != NULL)
 273                         *cp++ = '\0';
 274         }
 275         *q = NULL;
 276         return (&serv);
 277 }