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.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 struct dom_binding {};
  46 #include <rpcsvc/ypclnt.h>
  47 #include <rpcsvc/ypupdate_prot.h>
  48 #include "ypxfr_extern.h"
  49 #include "ypupdated_extern.h"
  50
  51 static int
  52 yp_domake(char *map, 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
  76 ypmap_update(char *netname, char *map, unsigned int op, unsigned int keylen,
  77              char *keyval, unsigned int datlen, char *datval)
  78 {
  79         DB *dbp;
  80         DBT key = { NULL, 0 }, data = { NULL, 0 };
  81         char *yp_last = "YP_LAST_MODIFIED";
  82         char yplastbuf[YPMAXRECORD];
  83         char *domptr;
  84         int rval = 0;
  85
  86         if ((domptr = strchr(netname, '@')) == NULL)
  87                 return(ERR_ACCESS);
  88         domptr++;
  89
  90
  91         dbp = yp_open_db_rw(domptr, map, O_RDWR);
  92         if (dbp == NULL)
  93                 return(ERR_DBASE);
  94
  95         key.data = keyval;
  96         key.size = keylen;
  97         data.data = datval;
  98         data.size = datlen;
  99
 100         switch (op) {
 101         case YPOP_DELETE: /* delete this entry */
 102                 rval = yp_del_record(dbp, &key);
 103                 if (rval == YP_TRUE)
 104                         rval = 0;
 105                 break;
 106         case YPOP_INSERT: /* add, do not change */
 107                 rval = yp_put_record(dbp, &key, &data, 0);
 108                 if (rval == YP_TRUE)
 109                         rval = 0;
 110                 break;
 111         case YPOP_STORE: /* add, or change */
 112                 rval = yp_put_record(dbp, &key, &data, 1);
 113                 if (rval == YP_TRUE)
 114                         rval = 0;
 115                 break;
 116         case YPOP_CHANGE: /* change, do not add */
 117                 if (yp_get_record(domptr, map, &key, &data, 0) != YP_TRUE) {
 118                         rval = ERR_KEY;
 119                         break;
 120                 }
 121                 rval = yp_put_record(dbp, &key, &data, 1);
 122                 if (rval == YP_TRUE)
 123                         rval = 0;
 124                 break;
 125         default:
 126                 yp_error("unknown update command: (%d)", op);
 127         }
 128
 129         if (rval) {
 130                 (dbp->close)(dbp);
 131                 return(rval);
 132         }
 133
 134         snprintf(yplastbuf, sizeof(yplastbuf), "%lu", time(NULL));
 135         key.data = yp_last;
 136         key.size = strlen(yp_last);
 137         data.data = (char *)&yplastbuf;
 138         data.size = strlen(yplastbuf);
 139         if (yp_put_record(dbp, &key, &data, 1) != YP_TRUE) {
 140                 yp_error("failed to update timestamp in %s/%s", domptr, map);
 141                 (dbp->close)(dbp);
 142                 return(ERR_DBASE);
 143         }
 144
 145         (dbp->close)(dbp);
 146         return(yp_domake(map, domptr));
 147 }