use seteuid instead of setreuid

[unix-history] / usr / src / usr.bin / rdist / main.c

/*
* Copyright (c) 1983 Regents of the University of California.
* All rights reserved.
*
* %sccs.include.redist.c%
*/

#ifndef lint
char copyright[] =
"@(#) Copyright (c) 1983 Regents of the University of California.\n\
All rights reserved.\n";
#endif /* not lint */

#ifndef lint
static char sccsid[] = "@(#)main.c      5.8 (Berkeley) %G%";
#endif /* not lint */

#include "defs.h"

#define NHOSTS 100

/*
* Remote distribution program.
*/

char    *distfile = NULL;
#define _RDIST_TMP      "/rdistXXXXXX"
char    tempfile[sizeof _PATH_TMP + sizeof _RDIST_TMP + 1];
char    *tempname;

int     debug;          /* debugging flag */
int     nflag;          /* NOP flag, just print commands without executing */
int     qflag;          /* Quiet. Don't print messages */
int     options;        /* global options */
int     iamremote;      /* act as remote server for transfering files */

FILE    *fin = NULL;    /* input file pointer */
int     rem = -1;       /* file descriptor to remote source/sink process */
char    host[32];       /* host name */
int     nerrs;          /* number of errors while sending/receiving */
char    user[10];       /* user's name */
char    homedir[128];   /* user's home directory */
int     userid;         /* user's user ID */
int     groupid;        /* user's group ID */

struct  passwd *pw;     /* pointer to static area used by getpwent */
struct  group *gr;      /* pointer to static area used by getgrent */

static void usage __P((void));
static void docmdargs __P((int, char *[]));

int
main(argc, argv)
       int argc;
       char *argv[];
{
       register char *arg;
       int cmdargs = 0;
       char *dhosts[NHOSTS], **hp = dhosts;

       pw = getpwuid(userid = getuid());
       if (pw == NULL) {
               fprintf(stderr, "%s: Who are you?\n", argv[0]);
               exit(1);
       }
       strcpy(user, pw->pw_name);
       strcpy(homedir, pw->pw_dir);
       groupid = pw->pw_gid;
       gethostname(host, sizeof(host));
       strcpy(tempfile, _PATH_TMP);
       strcat(tempfile, _RDIST_TMP);
       if ((tempname = rindex(tempfile, '/')) != 0)
               tempname++;
       else
               tempname = tempfile;

       while (--argc > 0) {
               if ((arg = *++argv)[0] != '-')
                       break;
               if (!strcmp(arg, "-Server"))
                       iamremote++;
               else while (*++arg)
                       switch (*arg) {
                       case 'f':
                               if (--argc <= 0)
                                       usage();
                               distfile = *++argv;
                               if (distfile[0] == '-' && distfile[1] == '\0')
                                       fin = stdin;
                               break;

                       case 'm':
                               if (--argc <= 0)
                                       usage();
                               if (hp >= &dhosts[NHOSTS-2]) {
                                       fprintf(stderr, "rdist: too many destination hosts\n");
                                       exit(1);
                               }
                               *hp++ = *++argv;
                               break;

                       case 'd':
                               if (--argc <= 0)
                                       usage();
                               define(*++argv);
                               break;

                       case 'D':
                               debug++;
                               break;

                       case 'c':
                               cmdargs++;
                               break;

                       case 'n':
                               if (options & VERIFY) {
                                       printf("rdist: -n overrides -v\n");
                                       options &= ~VERIFY;
                               }
                               nflag++;
                               break;

                       case 'q':
                               qflag++;
                               break;

                       case 'b':
                               options |= COMPARE;
                               break;

                       case 'R':
                               options |= REMOVE;
                               break;

                       case 'v':
                               if (nflag) {
                                       printf("rdist: -n overrides -v\n");
                                       break;
                               }
                               options |= VERIFY;
                               break;

                       case 'w':
                               options |= WHOLE;
                               break;

                       case 'y':
                               options |= YOUNGER;
                               break;

                       case 'h':
                               options |= FOLLOW;
                               break;

                       case 'i':
                               options |= IGNLNKS;
                               break;

                       default:
                               usage();
                       }
       }
       *hp = NULL;

       seteuid(userid);
       mktemp(tempfile);

       if (iamremote) {
               server();
               exit(nerrs != 0);
       }

       if (cmdargs)
               docmdargs(argc, argv);
       else {
               if (fin == NULL) {
                       if(distfile == NULL) {
                               if((fin = fopen("distfile","r")) == NULL)
                                       fin = fopen("Distfile", "r");
                       } else
                               fin = fopen(distfile, "r");
                       if(fin == NULL) {
                               perror(distfile ? distfile : "distfile");
                               exit(1);
                       }
               }
               yyparse();
               if (nerrs == 0)
                       docmds(dhosts, argc, argv);
       }

       exit(nerrs != 0);
}

static void
usage()
{
       printf("Usage: rdist [-nqbhirvwyD] [-f distfile] [-d var=value] [-m host] [file ...]\n");
       printf("or: rdist [-nqbhirvwyD] -c source [...] machine[:dest]\n");
       exit(1);
}

/*
* rcp like interface for distributing files.
*/
static void
docmdargs(nargs, args)
       int nargs;
       char *args[];
{
       register struct namelist *nl, *prev;
       register char *cp;
       struct namelist *files, *hosts;
       struct subcmd *cmds;
       char *dest;
       static struct namelist tnl = { NULL, NULL };
       int i;

       if (nargs < 2)
               usage();

       prev = NULL;
       for (i = 0; i < nargs - 1; i++) {
               nl = makenl(args[i]);
               if (prev == NULL)
                       files = prev = nl;
               else {
                       prev->n_next = nl;
                       prev = nl;
               }
       }

       cp = args[i];
       if ((dest = index(cp, ':')) != NULL)
               *dest++ = '\0';
       tnl.n_name = cp;
       hosts = expand(&tnl, E_ALL);
       if (nerrs)
               exit(1);

       if (dest == NULL || *dest == '\0')
               cmds = NULL;
       else {
               cmds = makesubcmd(INSTALL);
               cmds->sc_options = options;
               cmds->sc_name = dest;
       }

       if (debug) {
               printf("docmdargs()\nfiles = ");
               prnames(files);
               printf("hosts = ");
               prnames(hosts);
       }
       insert(NULL, files, hosts, cmds);
       docmds(NULL, 0, NULL);
}

/*
* Print a list of NAME blocks (mostly for debugging).
*/
void
prnames(nl)
       register struct namelist *nl;
{
       printf("( ");
       while (nl != NULL) {
               printf("%s ", nl->n_name);
               nl = nl->n_next;
       }
       printf(")\n");
}

#if __STDC__
#include <stdarg.h>
#else
#include <varargs.h>
#endif

void
#if __STDC__
warn(const char *fmt, ...)
#else
warn(fmt, va_alist)
       char *fmt;
       va_dcl
#endif
{
       extern int yylineno;
       va_list ap;
#if __STDC__
       va_start(ap, fmt);
#else
       va_start(ap);
#endif
       (void)fprintf(stderr, "rdist: line %d: Warning: ", yylineno);
       (void)vfprintf(stderr, fmt, ap);
       (void)fprintf(stderr, "\n");
       va_end(ap);
}