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.3.2.1 2002/02/15 00:46:57 des Exp $
  33  * $DragonFly: src/usr.sbin/rpc.ypupdated/yp_dbupdate.c,v 1.3 2004/12/18 22:48:14 swildner Exp $
  34  */
  35
  36 #include <sys/fcntl.h>
  37
  38 #include <stdio.h>
  39 #include <string.h>
  40 #include <errno.h>
  41 #include <limits.h>
  42 #include <db.h>
  43 #include <unistd.h>
  44 struct dom_binding {};
  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 yp_domake(map, domain)
  51         char *map;
  52         char *domain;
  53 {
  54         int pid;
  55
  56         switch ((pid = fork())) {
  57         case 0:
  58                 execlp(MAP_UPDATE_PATH, MAP_UPDATE, map, domain, NULL);
  59                 yp_error("couldn't exec map update process: %s",
  60                                                 strerror(errno));
  61                 exit(1);
  62                 break;
  63         case -1:
  64                 yp_error("fork() failed: %s", strerror(errno));
  65                 return(YPERR_YPERR);
  66                 break;
  67         default:
  68                 children++;
  69                 break;
  70         }
  71
  72         return(0);
  73 }
  74
  75 int ypmap_update(netname, map, op, keylen, keyval, datlen, datval)
  76         char *netname;
  77         char *map;
  78         unsigned int op;
  79         unsigned int keylen;
  80         char *keyval;
  81         unsigned int datlen;
  82         char *datval;
  83 {
  84         DB *dbp;
  85         DBT key = { NULL, 0 }, data = { NULL, 0 };
  86         char *yp_last = "YP_LAST_MODIFIED";
  87         char yplastbuf[YPMAXRECORD];
  88         char *domptr;
  89         int rval = 0;
  90
  91         if ((domptr = strchr(netname, '@')) == NULL)
  92                 return(ERR_ACCESS);
  93         domptr++;
  94
  95
  96         dbp = yp_open_db_rw(domptr, map, O_RDWR);
  97         if (dbp == NULL)
  98                 return(ERR_DBASE);
  99
 100         key.data = keyval;
 101         key.size = keylen;
 102         data.data = datval;
 103         data.size = datlen;
 104
 105         switch (op) {
 106         case YPOP_DELETE: /* delete this entry */
 107                 rval = yp_del_record(dbp, &key);
 108                 if (rval == YP_TRUE)
 109                         rval = 0;
 110                 break;
 111         case YPOP_INSERT: /* add, do not change */
 112                 rval = yp_put_record(dbp, &key, &data, 0);
 113                 if (rval == YP_TRUE)
 114                         rval = 0;
 115                 break;
 116         case YPOP_STORE: /* add, or change */
 117                 rval = yp_put_record(dbp, &key, &data, 1);
 118                 if (rval == YP_TRUE)
 119                         rval = 0;
 120                 break;
 121         case YPOP_CHANGE: /* change, do not add */
 122                 if (yp_get_record(domptr, map, &key, &data, 0) != YP_TRUE) {
 123                         rval = ERR_KEY;
 124                         break;
 125                 }
 126                 rval = yp_put_record(dbp, &key, &data, 1);
 127                 if (rval == YP_TRUE)
 128                         rval = 0;
 129                 break;
 130         default:
 131                 yp_error("unknown update command: (%d)", op);
 132         }
 133
 134         if (rval) {
 135                 (dbp->close)(dbp);
 136                 return(rval);
 137         }
 138
 139         snprintf(yplastbuf, sizeof(yplastbuf), "%lu", time(NULL));
 140         key.data = yp_last;
 141         key.size = strlen(yp_last);
 142         data.data = (char *)&yplastbuf;
 143         data.size = strlen(yplastbuf);
 144         if (yp_put_record(dbp, &key, &data, 1) != YP_TRUE) {
 145                 yp_error("failed to update timestamp in %s/%s", domptr, map);
 146                 (dbp->close)(dbp);
 147                 return(ERR_DBASE);
 148         }
 149
 150         (dbp->close)(dbp);
 151         return(yp_domake(map, domptr));
 152 }