1 /* 2 * CDDL HEADER START 3 * 4 * The contents of this file are subject to the terms of the 5 * Common Development and Distribution License, Version 1.0 only 6 * (the "License"). You may not use this file except in compliance 7 * with the License. 8 * 9 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 10 * or http://www.opensolaris.org/os/licensing. 11 * See the License for the specific language governing permissions 12 * and limitations under the License. 13 * 14 * When distributing Covered Code, include this CDDL HEADER in each 15 * file and include the License file at usr/src/OPENSOLARIS.LICENSE. 16 * If applicable, add the following below this CDDL HEADER, with the 17 * fields enclosed by brackets "[]" replaced with your own identifying 18 * information: Portions Copyright [yyyy] [name of copyright owner] 19 * 20 * CDDL HEADER END 21 * 22 * Copyright 2005 Sun Microsystems, Inc. All rights reserved. 23 * Use is subject to license terms. 24 */ 25 26 /* Copyright (c) 1983, 1984, 1985, 1986, 1987, 1988, 1989 AT&T */ 27 /* All Rights Reserved */ 28 29 /* 30 * Portions of this source code were derived from Berkeley 31 * under license from the Regents of the University of 32 * California. 33 */ 34 35 #pragma ident "%Z%%M% %I% %E% SMI" 36 37 /* 38 * YP updater for public key map 39 */ 40 #include <stdio.h> 41 #include <rpc/rpc.h> 42 #include <rpcsvc/ypclnt.h> 43 #include <sys/file.h> 44 45 extern char *malloc(); 46 47 int 48 main(argc, argv) 49 int argc; 50 char *argv[]; 51 { 52 unsigned op; 53 char name[MAXNETNAMELEN + 1]; 54 char key[256]; 55 char data[256]; 56 char line[256]; 57 unsigned keylen; 58 unsigned datalen; 59 FILE *rf; 60 FILE *wf; 61 char *fname; 62 char *tmpname; 63 int err; 64 65 66 if (argc != 3) { 67 exit(YPERR_YPERR); 68 } 69 fname = argv[1]; 70 tmpname = malloc(strlen(fname) + 4); 71 if (tmpname == NULL) { 72 exit(YPERR_YPERR); 73 } 74 sprintf(tmpname, "%s.tmp", fname); 75 76 /* 77 * Get input 78 */ 79 if (! scanf("%s\n", name)) { 80 exit(YPERR_YPERR); 81 } 82 if (! scanf("%u\n", &op)) { 83 exit(YPERR_YPERR); 84 } 85 if (! scanf("%u\n", &keylen)) { 86 exit(YPERR_YPERR); 87 } 88 if (! fread(key, keylen, 1, stdin)) { 89 exit(YPERR_YPERR); 90 } 91 key[keylen] = 0; 92 if (! scanf("%u\n", &datalen)) { 93 exit(YPERR_YPERR); 94 } 95 if (! fread(data, datalen, 1, stdin)) { 96 exit(YPERR_YPERR); 97 } 98 data[datalen] = 0; 99 100 /* 101 * Check permission 102 */ 103 if (strcmp(name, key) != 0) { 104 exit(YPERR_ACCESS); 105 } 106 if (strcmp(name, "nobody") == 0) { 107 /* 108 * Can't change "nobody"s key. 109 */ 110 exit(YPERR_ACCESS); 111 } 112 113 /* 114 * Open files 115 */ 116 rf = fopen(fname, "r"); 117 if (rf == NULL) { 118 exit(YPERR_YPERR); 119 } 120 wf = fopen(tmpname, "w"); 121 if (wf == NULL) { 122 exit(YPERR_YPERR); 123 } 124 err = -1; 125 while (fgets(line, sizeof (line), rf)) { 126 if (err < 0 && match(line, name)) { 127 switch (op) { 128 case YPOP_INSERT: 129 err = YPERR_KEY; 130 break; 131 case YPOP_STORE: 132 case YPOP_CHANGE: 133 fprintf(wf, "%s %s\n", key, data); 134 err = 0; 135 break; 136 case YPOP_DELETE: 137 /* do nothing */ 138 err = 0; 139 break; 140 } 141 } else { 142 fputs(line, wf); 143 } 144 } 145 if (err < 0) { 146 switch (op) { 147 case YPOP_CHANGE: 148 case YPOP_DELETE: 149 err = YPERR_KEY; 150 break; 151 case YPOP_INSERT: 152 case YPOP_STORE: 153 err = 0; 154 fprintf(wf, "%s %s\n", key, data); 155 break; 156 } 157 } 158 fclose(wf); 159 fclose(rf); 160 if (err == 0) { 161 if (rename(tmpname, fname) < 0) { 162 exit(YPERR_YPERR); 163 } 164 } else { 165 if (unlink(tmpname) < 0) { 166 exit(YPERR_YPERR); 167 } 168 } 169 if (fork() == 0) { 170 close(0); close(1); close(2); 171 open("/dev/null", O_RDWR, 0); 172 dup(0); dup(0); 173 execl("/bin/sh", "sh", "-c", argv[2], NULL); 174 } 175 return (err); 176 /* NOTREACHED */ 177 } 178 179 180 int 181 match(line, name) 182 char *line; 183 char *name; 184 { 185 int len; 186 187 len = strlen(name); 188 return (strncmp(line, name, len) == 0 && 189 (line[len] == ' ' || line[len] == '\t')); 190 } 191