nis.subproj/ypmatch_cache.c

   1 /*
   2  * Copyright (c) 1999 Apple Computer, Inc. All rights reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * Portions Copyright (c) 1999 Apple Computer, Inc.  All Rights
   7  * Reserved.  This file contains Original Code and/or Modifications of
   8  * Original Code as defined in and that are subject to the Apple Public
   9  * Source License Version 1.1 (the "License").  You may not use this file
  10  * except in compliance with the License.  Please obtain a copy of the
  11  * License at http://www.apple.com/publicsource and read it before using
  12  * this file.
  13  *
  14  * The Original Code and all software distributed under the License are
  15  * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  16  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  17  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE OR NON- INFRINGEMENT.  Please see the
  19  * License for the specific language governing rights and limitations
  20  * under the License.
  21  *
  22  * @APPLE_LICENSE_HEADER_END@
  23  */
  24 /*
  25  * Copyright (c) 1992, 1993 Theo de Raadt <deraadt@theos.com>
  26  * All rights reserved.
  27  *
  28  * Redistribution and use in source and binary forms, with or without
  29  * modification, are permitted provided that the following conditions
  30  * are met:
  31  * 1. Redistributions of source code must retain the above copyright
  32  *    notice, this list of conditions and the following disclaimer.
  33  * 2. Redistributions in binary form must reproduce the above copyright
  34  *    notice, this list of conditions and the following disclaimer in the
  35  *    documentation and/or other materials provided with the distribution.
  36  * 3. All advertising materials mentioning features or use of this software
  37  *    must display the following acknowledgement:
  38  *      This product includes software developed by Theo de Raadt.
  39  * 4. The name of the author may not be used to endorse or promote products
  40  *    derived from this software without specific prior written permission.
  41  *
  42  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
  43  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  44  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  45  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
  46  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  47  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  48  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  49  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  50  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  51  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  52  * SUCH DAMAGE.
  53  */
  54
  55 #if defined(LIBC_SCCS) && !defined(lint)
  56 static char *rcsid = "$OpenBSD: ypmatch_cache.c,v 1.6 1996/12/03 08:20:06 deraadt Exp $";
  57 #endif /* LIBC_SCCS and not lint */
  58
  59 #include <sys/param.h>
  60 #include <sys/types.h>
  61 #include <sys/socket.h>
  62 #include <sys/file.h>
  63 #include <sys/uio.h>
  64 #include <errno.h>
  65 #include <stdio.h>
  66 #include <stdlib.h>
  67 #include <string.h>
  68 #include <unistd.h>
  69 #include <rpc/rpc.h>
  70 #include <rpc/xdr.h>
  71 #include <rpcsvc/yp.h>
  72 #include <rpcsvc/ypclnt.h>
  73 /* #define YPMATCHCACHE */
  74 #include "ypinternal.h"
  75
  76 int _yplib_cache = 5;
  77
  78 #ifdef YPMATCHCACHE
  79 static bool_t
  80 ypmatch_add(map, key, keylen, val, vallen)
  81         const char     *map;
  82         const char     *key;
  83         u_int           keylen;
  84         char           *val;
  85         u_int           vallen;
  86 {
  87         struct ypmatch_ent *ep;
  88         time_t t;
  89
  90         (void)time(&t);
  91
  92         for (ep = ypmc; ep; ep = ep->next)
  93                 if (ep->expire_t < t)
  94                         break;
  95         if (ep == NULL) {
  96                 if ((ep = malloc(sizeof *ep)) == NULL)
  97                         return 0;
  98                 (void)memset(ep, 0, sizeof *ep);
  99                 if (ypmc)
 100                         ep->next = ypmc;
 101                 ypmc = ep;
 102         }
 103
 104         if (ep->key) {
 105                 free(ep->key);
 106                 ep->key = NULL;
 107         }
 108         if (ep->val) {
 109                 free(ep->val);
 110                 ep->val = NULL;
 111         }
 112
 113         if ((ep->key = malloc(keylen)) == NULL)
 114                 return 0;
 115
 116         if ((ep->val = malloc(vallen)) == NULL) {
 117                 free(ep->key);
 118                 ep->key = NULL;
 119                 return 0;
 120         }
 121
 122         ep->keylen = keylen;
 123         ep->vallen = vallen;
 124
 125         (void)memcpy(ep->key, key, ep->keylen);
 126         (void)memcpy(ep->val, val, ep->vallen);
 127
 128         if (ep->map) {
 129                 if (strcmp(ep->map, map)) {
 130                         free(ep->map);
 131                         if ((ep->map = strdup(map)) == NULL)
 132                                 return 0;
 133                 }
 134         } else {
 135                 if ((ep->map = strdup(map)) == NULL)
 136                         return 0;
 137         }
 138
 139         ep->expire_t = t + _yplib_cache;
 140         return 1;
 141 }
 142
 143 static bool_t
 144 ypmatch_find(map, key, keylen, val, vallen)
 145         const char     *map;
 146         const char     *key;
 147         u_int           keylen;
 148         char          **val;
 149         u_int          *vallen;
 150 {
 151         struct ypmatch_ent *ep;
 152         time_t          t;
 153
 154         if (ypmc == NULL)
 155                 return 0;
 156
 157         (void) time(&t);
 158
 159         for (ep = ypmc; ep; ep = ep->next) {
 160                 if (ep->keylen != keylen)
 161                         continue;
 162                 if (strcmp(ep->map, map))
 163                         continue;
 164                 if (memcmp(ep->key, key, keylen))
 165                         continue;
 166                 if (t > ep->expire_t)
 167                         continue;
 168
 169                 *val = ep->val;
 170                 *vallen = ep->vallen;
 171                 return 1;
 172         }
 173         return 0;
 174 }
 175 #endif
 176
 177 int
 178 yp_match(indomain, inmap, inkey, inkeylen, outval, outvallen)
 179         const char     *indomain;
 180         const char     *inmap;
 181         const char     *inkey;
 182         int             inkeylen;
 183         char          **outval;
 184         int            *outvallen;
 185 {
 186         struct dom_binding *ysd;
 187         struct ypresp_val yprv;
 188         struct timeval  tv;
 189         struct ypreq_key yprk;
 190         int tries = 0, r;
 191
 192         if (indomain == NULL || *indomain == '\0' ||
 193             strlen(indomain) > YPMAXDOMAIN || inmap == NULL ||
 194             *inmap == '\0' || strlen(inmap) > YPMAXMAP ||
 195             inkey == NULL || inkeylen == 0)
 196                 return YPERR_BADARGS;
 197
 198         *outval = NULL;
 199         *outvallen = 0;
 200
 201 again:
 202         if (_yp_dobind(indomain, &ysd) != 0)
 203                 return YPERR_DOMAIN;
 204
 205 #ifdef YPMATCHCACHE
 206         if (!strcmp(_yp_domain, indomain) && ypmatch_find(inmap, inkey,
 207             inkeylen, &yprv.val.valdat_val, &yprv.val.valdat_len)) {
 208                 *outvallen = yprv.val.valdat_len;
 209                 if ((*outval = malloc(*outvallen + 1)) == NULL) {
 210                         _yp_unbind(ysd);
 211                         return YPERR_YPERR;
 212                 }
 213                 (void)memcpy(*outval, yprv.val.valdat_val, *outvallen);
 214                 (*outval)[*outvallen] = '\0';
 215                 _yp_unbind(ysd);
 216                 return 0;
 217         }
 218 #endif
 219
 220         tv.tv_sec = _yplib_timeout;
 221         tv.tv_usec = 0;
 222
 223         yprk.domain = (char *)indomain;
 224         yprk.map = (char *)inmap;
 225         yprk.key.keydat_val = (char *) inkey;
 226         yprk.key.keydat_len = inkeylen;
 227
 228         memset(&yprv, 0, sizeof yprv);
 229
 230         r = clnt_call(ysd->dom_client, YPPROC_MATCH,
 231             xdr_ypreq_key, &yprk, xdr_ypresp_val, &yprv, tv);
 232         if (r != RPC_SUCCESS) {
 233                 if (tries++)
 234                         clnt_perror(ysd->dom_client, "yp_match: clnt_call");
 235                 ysd->dom_vers = -1;
 236                 goto again;
 237         }
 238         if (!(r = ypprot_err(yprv.stat))) {
 239                 *outvallen = yprv.val.valdat_len;
 240                 if ((*outval = malloc(*outvallen + 1)) == NULL) {
 241                         r = YPERR_YPERR;
 242                         goto out;
 243                 }
 244                 (void)memcpy(*outval, yprv.val.valdat_val, *outvallen);
 245                 (*outval)[*outvallen] = '\0';
 246 #ifdef YPMATCHCACHE
 247                 if (strcmp(_yp_domain, indomain) == 0)
 248                         if (!ypmatch_add(inmap, inkey, inkeylen,
 249                             *outval, *outvallen))
 250                                 r = YPERR_RESRC;
 251 #endif
 252         }
 253 out:
 254         xdr_free(xdr_ypresp_val, (char *) &yprv);
 255         _yp_unbind(ysd);
 256         return r;
 257 }
 258
 259 int
 260 yp_next(indomain, inmap, inkey, inkeylen, outkey, outkeylen, outval, outvallen)
 261         const char     *indomain;
 262         const char     *inmap;
 263         const char     *inkey;
 264         int             inkeylen;
 265         char          **outkey;
 266         int            *outkeylen;
 267         char          **outval;
 268         int            *outvallen;
 269 {
 270         struct ypresp_key_val yprkv;
 271         struct ypreq_key yprk;
 272         struct dom_binding *ysd;
 273         struct timeval  tv;
 274         int tries = 0, r;
 275
 276         if (indomain == NULL || *indomain == '\0' ||
 277             strlen(indomain) > YPMAXDOMAIN || inmap == NULL ||
 278             *inmap == '\0' || strlen(inmap) > YPMAXMAP)
 279                 return YPERR_BADARGS;
 280
 281         *outkey = *outval = NULL;
 282         *outkeylen = *outvallen = 0;
 283
 284 again:
 285         if (_yp_dobind(indomain, &ysd) != 0)
 286                 return YPERR_DOMAIN;
 287
 288         tv.tv_sec = _yplib_timeout;
 289         tv.tv_usec = 0;
 290
 291         yprk.domain = (char *)indomain;
 292         yprk.map = (char *)inmap;
 293         yprk.key.keydat_val = (char *)inkey;
 294         yprk.key.keydat_len = inkeylen;
 295         (void)memset(&yprkv, 0, sizeof yprkv);
 296
 297         r = clnt_call(ysd->dom_client, YPPROC_NEXT,
 298             xdr_ypreq_key, &yprk, xdr_ypresp_key_val, &yprkv, tv);
 299         if (r != RPC_SUCCESS) {
 300                 if (tries++)
 301                         clnt_perror(ysd->dom_client, "yp_next: clnt_call");
 302                 ysd->dom_vers = -1;
 303                 goto again;
 304         }
 305         if (!(r = ypprot_err(yprkv.stat))) {
 306                 *outkeylen = yprkv.key.keydat_len;
 307                 if ((*outkey = malloc(*outkeylen + 1)) == NULL)
 308                         r = YPERR_RESRC;
 309                 else {
 310                         (void)memcpy(*outkey, yprkv.key.keydat_val, *outkeylen);
 311                         (*outkey)[*outkeylen] = '\0';
 312                 }
 313                 *outvallen = yprkv.val.valdat_len;
 314                 if ((*outval = malloc(*outvallen + 1)) == NULL)
 315                         r = YPERR_RESRC;
 316                 else {
 317                         (void)memcpy(*outval, yprkv.val.valdat_val, *outvallen);
 318                         (*outval)[*outvallen] = '\0';
 319                 }
 320         }
 321         xdr_free(xdr_ypresp_key_val, (char *) &yprkv);
 322         _yp_unbind(ysd);
 323         return r;
 324 }