util.c revision 99110
1/*- 2 * Copyright (c) 1992, 1993 3 * The Regents of the University of California. 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 the University of 16 * California, Berkeley and its contributors. 17 * 4. Neither the name of the University nor the names of its contributors 18 * may be used to endorse or promote products derived from this software 19 * without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31 * SUCH DAMAGE. 32 */ 33 34#ifndef lint 35#if 0 36static char sccsid[] = "@(#)util.c 8.2 (Berkeley) 4/2/94"; 37#endif 38#endif /* not lint */ 39#include <sys/cdefs.h> 40__FBSDID("$FreeBSD: head/bin/rcp/util.c 99110 2002-06-30 05:15:05Z obrien $"); 41 42#include <sys/param.h> 43#include <sys/stat.h> 44#include <sys/wait.h> 45 46#include <ctype.h> 47#include <err.h> 48#include <errno.h> 49#include <paths.h> 50#include <signal.h> 51#include <stdio.h> 52#include <stdlib.h> 53#include <string.h> 54#include <unistd.h> 55 56#include "extern.h" 57 58char * 59colon(char *cp) 60{ 61 if (*cp == ':') /* Leading colon is part of file name. */ 62 return (0); 63 64 for (; *cp; ++cp) { 65 if (*cp == ':') 66 return (cp); 67 if (*cp == '/') 68 return (0); 69 } 70 return (0); 71} 72 73void 74verifydir(char *cp) 75{ 76 struct stat stb; 77 78 if (!stat(cp, &stb)) { 79 if (S_ISDIR(stb.st_mode)) 80 return; 81 errno = ENOTDIR; 82 } 83 run_err("%s: %s", cp, strerror(errno)); 84 exit(1); 85} 86 87int 88okname(char *cp0) 89{ 90 int c; 91 char *cp; 92 93 cp = cp0; 94 do { 95 c = *cp; 96 if (c & 0200) 97 goto bad; 98 if (!isalpha(c) && !isdigit(c) && c != '_' && c != '-' && c != '.') 99 goto bad; 100 } while (*++cp); 101 return (1); 102 103bad: warnx("%s: invalid user name", cp0); 104 return (0); 105} 106 107int 108susystem(char *s, int userid) 109{ 110 sig_t istat, qstat; 111 int status; 112 pid_t pid; 113 114 pid = vfork(); 115 switch (pid) { 116 case -1: 117 return (127); 118 119 case 0: 120 (void)setuid(userid); 121 execl(_PATH_BSHELL, "sh", "-c", s, (char *)NULL); 122 _exit(127); 123 } 124 istat = signal(SIGINT, SIG_IGN); 125 qstat = signal(SIGQUIT, SIG_IGN); 126 if (waitpid(pid, &status, 0) < 0) 127 status = -1; 128 (void)signal(SIGINT, istat); 129 (void)signal(SIGQUIT, qstat); 130 return (status); 131} 132 133BUF * 134allocbuf(BUF *bp, int fd, int blksize) 135{ 136 struct stat stb; 137 size_t size; 138 139 if (fstat(fd, &stb) < 0) { 140 run_err("fstat: %s", strerror(errno)); 141 return (0); 142 } 143 size = roundup(stb.st_blksize, blksize); 144 if (size == 0) 145 size = blksize; 146 if (bp->cnt >= size) 147 return (bp); 148 if ((bp->buf = realloc(bp->buf, size)) == NULL) { 149 bp->cnt = 0; 150 run_err("%s", strerror(errno)); 151 return (0); 152 } 153 bp->cnt = size; 154 return (bp); 155} 156 157void 158lostconn(int signo) 159{ 160 if (!iamremote) 161 warnx("lost connection"); 162 exit(1); 163} 164