minix3/commands/rcp/rcp.c

850 lines
16 KiB
C

/*
* Copyright (c) 1983 Regents of the University of California.
* All rights reserved. The Berkeley software License Agreement
* specifies the terms and conditions for redistribution.
*/
#ifndef lint
char copyright[] =
"@(#) Copyright (c) 1983 Regents of the University of California.\n\
All rights reserved.\n";
#endif /* not lint */
#ifndef lint
#if 0
static char sccsid[] = "@(#)rcp.c 1.1 87/12/21 SMI"; /* from UCB 5.3 6/8/85"*/
#endif
#endif /* not lint */
/*
* rcp
*/
#define NAMESERVER
#include <ctype.h>
#include <errno.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <utime.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <dirent.h>
#include <fcntl.h>
#include <pwd.h>
#include <signal.h>
#include <unistd.h>
#include <netdb.h>
#include <net/netlib.h>
#if __STDC__
#define PROTO(func, args) func args
#else
#define PROTO(func, args) func ()
#endif /* __STDC__ */
PROTO (int main, (int argc, char *argv[]));
PROTO (void lostconn, (int sig));
PROTO (void error, (char *fmt, ...) );
PROTO (int response, (void) );
PROTO (void source, (int argc, char *argv[]) );
PROTO (void sink, (int argc, char *argv[]) );
PROTO (void usage, (void) );
PROTO (char *colon, (char *cp) );
PROTO (int okname, (char *cp0) );
PROTO (int susystem, (char *s) );
PROTO (void verifydir, (char *cp) );
PROTO (void rsource, (char *name, struct stat *statp) );
PROTO (struct buffer *allocbuf, (struct buffer *bp, int fd, int blksize) );
#define vfork fork
int rem;
int errs;
int errno;
int iamremote, targetshouldbedirectory;
int iamrecursive;
int myuid; /* uid of invoker */
int pflag;
struct passwd *pwd;
int userid;
int port;
struct buffer {
int cnt;
char *buf;
};
#define ga() (void) write(rem, "", 1)
int main(argc, argv)
int argc;
char **argv;
{
char *targ, *host, *src;
#ifndef NAMESERVER
char *suser, *tuser;
#else /* NAMESERVER */
char *suser, *tuser, *thost;
#endif /* NAMESERVER */
int i;
char buf[BUFSIZ], cmd[16];
struct servent *sp;
sp = getservbyname("shell", "tcp");
if (sp == NULL) {
fprintf(stderr, "rcp: shell/tcp: unknown service\n");
exit(1);
}
port = sp->s_port;
pwd = getpwuid(userid = getuid());
if (pwd == 0) {
fprintf(stderr, "who are you?\n");
exit(1);
}
#ifdef NOT_DEF
/*
* This is a kludge to allow seteuid to user before touching
* files and seteuid root before doing rcmd so we can open
* the socket.
*/
myuid = getuid();
if (setruid(0) < 0) {
perror("setruid root");
exit(1);
}
seteuid(myuid);
#endif
for (argc--, argv++; argc > 0 && **argv == '-'; argc--, argv++) {
(*argv)++;
while (**argv) switch (*(*argv)++) {
case 'r':
iamrecursive++;
break;
case 'p': /* preserve mtimes and atimes */
pflag++;
break;
/* The rest of these are not for users. */
case 'd':
targetshouldbedirectory = 1;
break;
case 'f': /* "from" */
iamremote = 1;
(void) response();
source(--argc, ++argv);
exit(errs);
case 't': /* "to" */
iamremote = 1;
sink(--argc, ++argv);
exit(errs);
default:
usage();
exit(1);
}
}
if (iamremote)
{
close(2);
open("/dev/tty", 2);
}
if (argc < 2) {
usage();
exit(1);
}
rem = -1;
if (argc > 2)
targetshouldbedirectory = 1;
(void) sprintf(cmd, "rcp%s%s%s",
iamrecursive ? " -r" : "", pflag ? " -p" : "",
targetshouldbedirectory ? " -d" : "");
(void) signal(SIGPIPE, lostconn);
targ = colon(argv[argc - 1]);
if (targ) { /* ... to remote */
*targ++ = 0;
if (*targ == 0)
targ = ".";
#ifndef NAMESERVER
tuser = strrchr(argv[argc - 1], '.');
if (tuser) {
*tuser++ = 0;
if (!okname(tuser))
exit(1);
} else
tuser = pwd->pw_name;
#else /* NAMESERVER */
thost = strchr(argv[argc - 1], '@');
if (thost) {
*thost++ = 0;
tuser = argv[argc - 1];
if (*tuser == '\0')
tuser = pwd->pw_name;
else if (!okname(tuser))
exit(1);
} else {
thost = argv[argc - 1];
tuser = pwd->pw_name;
}
#endif /* NAMESERVER */
for (i = 0; i < argc - 1; i++) {
src = colon(argv[i]);
if (src) { /* remote to remote */
*src++ = 0;
if (*src == 0)
src = ".";
#ifndef NAMESERVER
suser = strrchr(argv[i], '.');
if (suser) {
*suser++ = 0;
if (!okname(suser))
#else /* NAMESERVER */
host = strchr(argv[i], '@');
if (host) {
*host++ = 0;
suser = argv[i];
if (*suser == '\0')
suser = pwd->pw_name;
else if (!okname(suser))
#endif /* NAMESERVER */
continue;
#ifndef NAMESERVER
(void) sprintf(buf, "rsh %s -l %s -n %s %s '%s.%s:%s'",
argv[i], suser, cmd, src,
argv[argc - 1], tuser, targ);
} else
(void) sprintf(buf, "rsh %s -n %s %s '%s.%s:%s'",
argv[i], cmd, src,
argv[argc - 1], tuser, targ);
#else /* NAMESERVER */
(void) sprintf(buf, "rsh %s -l %s -n %s %s '%s@%s:%s'",
host, suser, cmd, src,
tuser, thost, targ);
} else
(void) sprintf(buf, "rsh %s -n %s %s '%s@%s:%s'",
argv[i], cmd, src,
tuser, thost, targ);
#endif /* NAMESERVER */
(void) susystem(buf);
} else { /* local to remote */
if (rem == -1) {
(void) sprintf(buf, "%s -t %s",
cmd, targ);
#ifndef NAMESERVER
host = argv[argc - 1];
#else /* NAMESERVER */
host = thost;
#endif /* NAMESERVER */
#ifdef NOT_DEF
if (seteuid(0) < 0) {
perror("seteuid root");
exit(1);
}
#endif
rem = rcmd(&host, port, pwd->pw_name,
tuser, buf, 0);
#ifdef NO_DEF
seteuid(myuid);
#endif
if (rem < 0)
exit(1);
if (response() < 0)
exit(1);
}
source(1, argv+i);
}
}
} else { /* ... to local */
if (targetshouldbedirectory)
verifydir(argv[argc - 1]);
for (i = 0; i < argc - 1; i++) {
src = colon(argv[i]);
if (src == 0) { /* local to local */
(void) sprintf(buf, "cp%s%s %s %s",
iamrecursive ? " -r" : "",
pflag ? " -p" : "",
argv[i], argv[argc - 1]);
(void) susystem(buf);
} else { /* remote to local */
*src++ = 0;
if (*src == 0)
src = ".";
#ifndef NAMESERVER
suser = strrchr(argv[i], '.');
if (suser) {
*suser++ = 0;
if (!okname(suser))
#else /* NAMESERVER */
host = strchr(argv[i], '@');
if (host) {
*host++ = 0;
suser = argv[i];
if (*suser == '\0')
suser = pwd->pw_name;
else if (!okname(suser))
#endif /* NAMESERVER */
continue;
#ifndef NAMESERVER
} else
#else /* NAMESERVER */
} else {
host = argv[i];
#endif /* NAMESERVER */
suser = pwd->pw_name;
#ifdef NAMESERVER
}
#endif /* NAMESERVER */
(void) sprintf(buf, "%s -f %s", cmd, src);
#ifndef NAMESERVER
host = argv[i];
#endif /* NAMESERVER */
#ifdef NOT_DEF
if (seteuid(0) < 0) {
perror("seteuid root");
exit(1);
}
#endif
rem = rcmd(&host, port, pwd->pw_name, suser,
buf, 0);
#ifdef NOT_DEF
seteuid(myuid);
#endif
if (rem < 0) {
errs++;
continue;
}
sink(1, argv+argc-1);
(void) close(rem);
rem = -1;
}
}
}
exit(errs);
}
void
verifydir(cp)
char *cp;
{
struct stat stb;
if (stat(cp, &stb) >= 0) {
if ((stb.st_mode & S_IFMT) == S_IFDIR)
return;
errno = ENOTDIR;
}
error("rcp: %s: %s.\n", cp, strerror(errno));
exit(1);
}
char *
colon(cp)
char *cp;
{
while (*cp) {
if (*cp == ':')
return (cp);
if (*cp == '/')
return (0);
cp++;
}
return (0);
}
int
okname(cp0)
char *cp0;
{
register char *cp = cp0;
register int c;
do {
c = *cp;
if (c & 0200)
goto bad;
if (!isalpha(c) && !isdigit(c) && c != '_' && c != '-')
goto bad;
cp++;
} while (*cp);
return (1);
bad:
fprintf(stderr, "rcp: invalid user name %s\n", cp0);
return (0);
}
int
susystem(s)
char *s;
{
int status, pid, w;
register void PROTO ((*istat), (int) ), PROTO ((*qstat), (int) );
if ((pid = vfork()) == 0) {
#ifdef NOT_DEF
(void) setruid(myuid);
#endif
execl("/bin/sh", "sh", "-c", s, (char *)0);
_exit(127);
}
istat = signal(SIGINT, SIG_IGN);
qstat = signal(SIGQUIT, SIG_IGN);
while ((w = wait(&status)) != pid && w != -1)
;
if (w == -1)
status = -1;
(void) signal(SIGINT, istat);
(void) signal(SIGQUIT, qstat);
return (status);
}
void
source(argc, argv)
int argc;
char **argv;
{
char *last, *name;
struct stat stb;
static struct buffer buffer;
struct buffer *bp;
int x, sizerr, f, amt;
off_t i;
char buf[BUFSIZ];
for (x = 0; x < argc; x++) {
name = argv[x];
if ((f = open(name, 0)) < 0) {
error("rcp: %s: %s\n", name, strerror(errno));
continue;
}
if (fstat(f, &stb) < 0)
goto notreg;
switch (stb.st_mode&S_IFMT) {
case S_IFREG:
break;
case S_IFDIR:
if (iamrecursive) {
(void) close(f);
rsource(name, &stb);
continue;
}
/* fall into ... */
default:
notreg:
(void) close(f);
error("rcp: %s: not a plain file\n", name);
continue;
}
last = strrchr(name, '/');
if (last == 0)
last = name;
else
last++;
if (pflag) {
/*
* Make it compatible with possible future
* versions expecting microseconds.
*/
(void) sprintf(buf, "T%d 0 %d 0\n",
stb.st_mtime, stb.st_atime);
(void) write(rem, buf, strlen(buf));
if (response() < 0) {
(void) close(f);
continue;
}
}
(void) sprintf(buf, "C%04o %lld %s\n",
stb.st_mode&07777, stb.st_size, last);
(void) write(rem, buf, strlen(buf));
if (response() < 0) {
(void) close(f);
continue;
}
if ((bp = allocbuf(&buffer, f, BUFSIZ)) == 0) {
(void) close(f);
continue;
}
sizerr = 0;
for (i = 0; i < stb.st_size; i += bp->cnt) {
amt = bp->cnt;
if (i + amt > stb.st_size)
amt = stb.st_size - i;
if (sizerr == 0 && read(f, bp->buf, amt) != amt)
sizerr = 1;
(void) write(rem, bp->buf, amt);
}
(void) close(f);
if (sizerr == 0)
ga();
else
error("rcp: %s: file changed size\n", name);
(void) response();
}
}
void
rsource(name, statp)
char *name;
struct stat *statp;
{
DIR *d = opendir(name);
char *last;
struct dirent *dp;
char buf[BUFSIZ];
char *bufv[1];
if (d == 0) {
error("rcp: %s: %s\n", name, strerror(errno));
return;
}
last = strrchr(name, '/');
if (last == 0)
last = name;
else
last++;
if (pflag) {
(void) sprintf(buf, "T%d 0 %d 0\n",
statp->st_mtime, statp->st_atime);
(void) write(rem, buf, strlen(buf));
if (response() < 0) {
closedir(d);
return;
}
}
(void) sprintf(buf, "D%04o %d %s\n", statp->st_mode&07777, 0, last);
(void) write(rem, buf, strlen(buf));
if (response() < 0) {
closedir(d);
return;
}
while ((dp = readdir(d))) {
if (dp->d_ino == 0)
continue;
if (!strcmp(dp->d_name, ".") || !strcmp(dp->d_name, ".."))
continue;
if (strlen(name) + 1 + strlen(dp->d_name) >= BUFSIZ - 1) {
error("%s/%s: Name too long.\n", name, dp->d_name);
continue;
}
(void) sprintf(buf, "%s/%s", name, dp->d_name);
bufv[0] = buf;
source(1, bufv);
}
closedir(d);
(void) write(rem, "E\n", 2);
(void) response();
}
int
response()
{
char resp, c, rbuf[BUFSIZ], *cp = rbuf;
if (read(rem, &resp, 1) != 1)
lostconn(0);
switch (resp) {
case 0: /* ok */
return (0);
default:
*cp++ = resp;
/* fall into... */
case 1: /* error, followed by err msg */
case 2: /* fatal error, "" */
do {
if (read(rem, &c, 1) != 1)
lostconn(0);
*cp++ = c;
} while (cp < &rbuf[BUFSIZ] && c != '\n');
if (iamremote == 0)
(void) write(2, rbuf, cp - rbuf);
errs++;
if (resp == 1)
return (-1);
exit(1);
}
/*NOTREACHED*/
}
void
lostconn(sig)
int sig;
{
if (iamremote == 0)
fprintf(stderr, "rcp: lost connection\n");
exit(1);
}
void
sink(argc, argv)
int argc;
char **argv;
{
off_t i, j, size;
char *targ, *whopp, *cp;
int of, mode, wrerr, exists, first, count, amt;
struct buffer *bp;
static struct buffer buffer;
struct stat stb;
int targisdir = 0;
int mask = umask(0);
char *myargv[1];
char cmdbuf[BUFSIZ], nambuf[BUFSIZ];
int setimes = 0;
struct utimbuf utimbuf;
#define atime utimbuf.actime
#define mtime utimbuf.modtime
time_t dummy;
#define SCREWUP(str) { whopp = str; goto screwup; }
#ifdef NOT_DEF
seteuid(pwd->pw_uid);
#endif
if (!pflag)
(void) umask(mask);
if (argc != 1) {
error("rcp: ambiguous target\n");
exit(1);
}
targ = *argv;
if (targetshouldbedirectory)
verifydir(targ);
ga();
if (stat(targ, &stb) == 0 && (stb.st_mode & S_IFMT) == S_IFDIR)
targisdir = 1;
for (first = 1; ; first = 0) {
cp = cmdbuf;
if (read(rem, cp, 1) <= 0)
return;
if (*cp++ == '\n')
SCREWUP("unexpected '\\n'");
do {
if (read(rem, cp, 1) != 1)
SCREWUP("lost connection");
} while (*cp++ != '\n');
*cp = 0;
if (cmdbuf[0] == '\01' || cmdbuf[0] == '\02') {
if (iamremote == 0)
(void) write(2, cmdbuf+1, strlen(cmdbuf+1));
if (cmdbuf[0] == '\02')
exit(1);
errs++;
continue;
}
*--cp = 0;
cp = cmdbuf;
if (*cp == 'E') {
ga();
return;
}
#define getnum(t) (t) = 0; while (isdigit(*cp)) (t) = (t) * 10 + (*cp++ - '0');
if (*cp == 'T') {
setimes++;
cp++;
getnum(mtime);
if (*cp++ != ' ')
SCREWUP("mtime.sec not delimited");
getnum(dummy);
if (*cp++ != ' ')
SCREWUP("mtime.usec not delimited");
getnum(atime);
if (*cp++ != ' ')
SCREWUP("atime.sec not delimited");
getnum(dummy);
if (*cp++ != '\0')
SCREWUP("atime.usec not delimited");
ga();
continue;
}
if (*cp != 'C' && *cp != 'D') {
/*
* Check for the case "rcp remote:foo\* local:bar".
* In this case, the line "No match." can be returned
* by the shell before the rcp command on the remote is
* executed so the ^Aerror_message convention isn't
* followed.
*/
if (first) {
error("%s\n", cp);
exit(1);
}
SCREWUP("expected control record");
}
cp++;
mode = 0;
for (; cp < cmdbuf+5; cp++) {
if (*cp < '0' || *cp > '7')
SCREWUP("bad mode");
mode = (mode << 3) | (*cp - '0');
}
if (*cp++ != ' ')
SCREWUP("mode not delimited");
size = 0;
while (isdigit(*cp))
size = size * 10 + (*cp++ - '0');
if (*cp++ != ' ')
SCREWUP("size not delimited");
if (targisdir)
(void) sprintf(nambuf, "%s%s%s", targ,
*targ ? "/" : "", cp);
else
(void) strcpy(nambuf, targ);
exists = stat(nambuf, &stb) == 0;
if (cmdbuf[0] == 'D') {
if (exists) {
if ((stb.st_mode&S_IFMT) != S_IFDIR) {
errno = ENOTDIR;
goto bad;
}
if (pflag)
(void) chmod(nambuf, mode);
} else if (mkdir(nambuf, mode) < 0)
goto bad;
myargv[0] = nambuf;
sink(1, myargv);
if (setimes) {
setimes = 0;
if (utime(nambuf, &utimbuf) < 0)
error("rcp: can't set times on %s: %s\n",
nambuf, strerror(errno));
}
continue;
}
if ((of = creat(nambuf, mode)) < 0) {
bad:
error("rcp: %s: %s\n", nambuf, strerror(errno));
continue;
}
if (exists && pflag)
(void) chmod(nambuf, mode);
ga();
if ((bp = allocbuf(&buffer, of, BUFSIZ)) == 0) {
(void) close(of);
continue;
}
cp = bp->buf;
count = 0;
wrerr = 0;
for (i = 0; i < size; i += BUFSIZ) {
amt = BUFSIZ;
if (i + amt > size)
amt = size - i;
count += amt;
do {
j = read(rem, cp, amt);
if (j <= 0)
exit(1);
amt -= j;
cp += j;
} while (amt > 0);
if (count == bp->cnt) {
if (wrerr == 0 &&
write(of, bp->buf, count) != count)
wrerr++;
count = 0;
cp = bp->buf;
}
}
if (count != 0 && wrerr == 0 &&
write(of, bp->buf, count) != count)
wrerr++;
(void) close(of);
(void) response();
if (setimes) {
setimes = 0;
if (utime(nambuf, &utimbuf) < 0)
error("rcp: can't set times on %s: %s\n",
nambuf, strerror(errno));
}
if (wrerr)
error("rcp: %s: %s\n", nambuf, strerror(errno));
else
ga();
}
screwup:
error("rcp: protocol screwup: %s\n", whopp);
exit(1);
}
struct buffer *
allocbuf(bp, fd, blksize)
struct buffer *bp;
int fd, blksize;
{
struct stat stb;
int size;
if (fstat(fd, &stb) < 0) {
error("rcp: fstat: %s\n", strerror(errno));
return ((struct buffer *)0);
}
size= 0;
#if NOT_DEF
size = roundup(stb.st_blksize, blksize);
#endif
if (size == 0)
size = blksize;
if (bp->cnt < size) {
if (bp->buf != 0)
free(bp->buf);
bp->buf = (char *)malloc((unsigned) size);
if (bp->buf == 0) {
error("rcp: malloc: out of memory\n");
return ((struct buffer *)0);
}
}
bp->cnt = size;
return (bp);
}
/*VARARGS1*/
#if __STDC__
void
error (char *fmt, ...)
#else
error(fmt)
char *fmt;
#endif
{
char buf[BUFSIZ], *cp = buf;
va_list ap;
va_start(ap, fmt);
errs++;
*cp++ = 1;
(void) vsprintf(cp, fmt, ap);
va_end(ap);
(void) write(rem, buf, strlen(buf));
if (iamremote == 0)
(void) write(2, buf+1, strlen(buf+1));
}
void
usage()
{
fprintf(stderr, "Usage: rcp [-p] f1 f2; or: rcp [-rp] f1 ... fn d2\n");
}