usr.sbin/rpc.ypupdated/yp_dbupdate.c

   1 /*
   2  * Copyright (c) 1996
   3  *      Bill Paul <wpaul@ctr.columbia.edu>.  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 Bill Paul.
  16  * 4. Neither the name of the author nor the names of any co-contributors
  17  *    may be used to endorse or promote products derived from this software
  18  *    without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY Bill Paul AND CONTRIBUTORS ``AS IS'' AND
  21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  23  * ARE DISCLAIMED.  IN NO EVENT SHALL Bill Paul OR CONTRIBUTORS BE LIABLE
  24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  30  * SUCH DAMAGE.
  31  *
  32  * $FreeBSD: src/usr.sbin/rpc.ypupdated/yp_dbupdate.c,v 1.9 2004/10/17 19:33:32 stefanf Exp $
  33  * $DragonFly: src/usr.sbin/rpc.ypupdated/yp_dbupdate.c,v 1.5 2007/11/25 01:28:24 swildner Exp $
  34  */
  35
  36 #include <sys/fcntl.h>
  37
  38 #include <stdio.h>
  39 #include <stdlib.h>
  40 #include <string.h>
  41 #include <errno.h>
  42 #include <limits.h>
  43 #include <db.h>
  44 #include <unistd.h>
  45 #include <rpcsvc/ypclnt.h>
  46 #include <rpcsvc/ypupdate_prot.h>
  47 #include "ypxfr_extern.h"
  48 #include "ypupdated_extern.h"
  49
  50 static int
  51 yp_domake(char *map, char *domain)
  52 {
  53         int pid;
  54
  55         switch ((pid = fork())) {
  56         case 0:
  57                 execlp(MAP_UPDATE_PATH, MAP_UPDATE, map, domain, NULL);
  58                 yp_error("couldn't exec map update process: %s",
  59                                                 strerror(errno));
  60                 exit(1);
  61                 break;
  62         case -1:
  63                 yp_error("fork() failed: %s", strerror(errno));
  64                 return(YPERR_YPERR);
  65                 break;
  66         default:
  67                 children++;
  68                 break;
  69         }
  70
  71         return(0);
  72 }
  73
  74 int
  75 ypmap_update(char *netname, char *map, unsigned int op, unsigned int keylen,
  76              char *keyval, unsigned int datlen, char *datval)
  77 {
  78         DB *dbp;
  79         DBT key = { NULL, 0 }, data = { NULL, 0 };
  80         char *yp_last = "YP_LAST_MODIFIED";
  81         char yplastbuf[YPMAXRECORD];
  82         char *domptr;
  83         int rval = 0;
  84
  85         if ((domptr = strchr(netname, '@')) == NULL)
  86                 return(ERR_ACCESS);
  87         domptr++;
  88
  89
  90         dbp = yp_open_db_rw(domptr, map, O_RDWR);
  91         if (dbp == NULL)
  92                 return(ERR_DBASE);
  93
  94         key.data = keyval;
  95         key.size = keylen;
  96         data.data = datval;
  97         data.size = datlen;
  98
  99         switch (op) {
 100         case YPOP_DELETE: /* delete this entry */
 101                 rval = yp_del_record(dbp, &key);
 102                 if (rval == YP_TRUE)
 103                         rval = 0;
 104                 break;
 105         case YPOP_INSERT: /* add, do not change */
 106                 rval = yp_put_record(dbp, &key, &data, 0);
 107                 if (rval == YP_TRUE)
 108                         rval = 0;
 109                 break;
 110         case YPOP_STORE: /* add, or change */
 111                 rval = yp_put_record(dbp, &key, &data, 1);
 112                 if (rval == YP_TRUE)
 113                         rval = 0;
 114                 break;
 115         case YPOP_CHANGE: /* change, do not add */
 116                 if (yp_get_record(domptr, map, &key, &data, 0) != YP_TRUE) {
 117                         rval = ERR_KEY;
 118                         break;
 119                 }
 120                 rval = yp_put_record(dbp, &key, &data, 1);
 121                 if (rval == YP_TRUE)
 122                         rval = 0;
 123                 break;
 124         default:
 125                 yp_error("unknown update command: (%d)", op);
 126         }
 127
 128         if (rval) {
 129                 (dbp->close)(dbp);
 130                 return(rval);
 131         }
 132
 133         snprintf(yplastbuf, sizeof(yplastbuf), "%lu", time(NULL));
 134         key.data = yp_last;
 135         key.size = strlen(yp_last);
 136         data.data = (char *)&yplastbuf;
 137         data.size = strlen(yplastbuf);
 138         if (yp_put_record(dbp, &key, &data, 1) != YP_TRUE) {
 139                 yp_error("failed to update timestamp in %s/%s", domptr, map);
 140                 (dbp->close)(dbp);
 141                 return(ERR_DBASE);
 142         }
 143
 144         (dbp->close)(dbp);
 145         return(yp_domake(map, domptr));
 146 }