[BACK]Return to scp.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / ssh

Annotation of src/usr.bin/ssh/scp.c, Revision 1.96

1.1       deraadt     1: /*
1.38      deraadt     2:  * scp - secure remote copy.  This is basically patched BSD rcp which
                      3:  * uses ssh to do the data transfer (instead of using rcmd).
1.27      markus      4:  *
1.38      deraadt     5:  * NOTE: This version should NOT be suid root.  (This uses ssh to
                      6:  * do the transfer and ssh has the necessary privileges.)
1.27      markus      7:  *
1.38      deraadt     8:  * 1995 Timo Rinne <tri@iki.fi>, Tatu Ylonen <ylo@cs.hut.fi>
                      9:  *
                     10:  * As far as I am concerned, the code I have written for this software
                     11:  * can be used freely for any purpose.  Any derived versions of this
                     12:  * software must be clearly marked as such, and if the derived work is
                     13:  * incompatible with the protocol description in the RFC file, it must be
                     14:  * called by a name other than "ssh" or "Secure Shell".
1.39      markus     15:  */
1.38      deraadt    16: /*
1.60      deraadt    17:  * Copyright (c) 1999 Theo de Raadt.  All rights reserved.
                     18:  * Copyright (c) 1999 Aaron Campbell.  All rights reserved.
1.27      markus     19:  *
1.38      deraadt    20:  * Redistribution and use in source and binary forms, with or without
                     21:  * modification, are permitted provided that the following conditions
                     22:  * are met:
                     23:  * 1. Redistributions of source code must retain the above copyright
                     24:  *    notice, this list of conditions and the following disclaimer.
                     25:  * 2. Redistributions in binary form must reproduce the above copyright
                     26:  *    notice, this list of conditions and the following disclaimer in the
                     27:  *    documentation and/or other materials provided with the distribution.
1.27      markus     28:  *
1.38      deraadt    29:  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
                     30:  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
                     31:  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
                     32:  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
                     33:  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
                     34:  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
                     35:  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
                     36:  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
                     37:  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
                     38:  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
                     39:  */
1.1       deraadt    40:
                     41: /*
1.35      deraadt    42:  * Parts from:
                     43:  *
1.1       deraadt    44:  * Copyright (c) 1983, 1990, 1992, 1993, 1995
                     45:  *     The Regents of the University of California.  All rights reserved.
                     46:  *
                     47:  * Redistribution and use in source and binary forms, with or without
                     48:  * modification, are permitted provided that the following conditions
                     49:  * are met:
                     50:  * 1. Redistributions of source code must retain the above copyright
                     51:  *    notice, this list of conditions and the following disclaimer.
                     52:  * 2. Redistributions in binary form must reproduce the above copyright
                     53:  *    notice, this list of conditions and the following disclaimer in the
                     54:  *    documentation and/or other materials provided with the distribution.
                     55:  * 3. All advertising materials mentioning features or use of this software
                     56:  *    must display the following acknowledgement:
                     57:  *     This product includes software developed by the University of
                     58:  *     California, Berkeley and its contributors.
                     59:  * 4. Neither the name of the University nor the names of its contributors
                     60:  *    may be used to endorse or promote products derived from this software
                     61:  *    without specific prior written permission.
                     62:  *
                     63:  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
                     64:  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
                     65:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
                     66:  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
                     67:  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
                     68:  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
                     69:  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
                     70:  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
                     71:  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
                     72:  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
                     73:  * SUCH DAMAGE.
                     74:  *
                     75:  */
                     76:
                     77: #include "includes.h"
1.96    ! markus     78: RCSID("$OpenBSD: scp.c,v 1.95 2002/12/05 11:08:35 markus Exp $");
1.1       deraadt    79:
                     80: #include "xmalloc.h"
1.51      markus     81: #include "atomicio.h"
1.50      markus     82: #include "pathnames.h"
1.51      markus     83: #include "log.h"
1.69      mouring    84: #include "misc.h"
1.1       deraadt    85:
1.11      aaron      86: /* For progressmeter() -- number of seconds before xfer considered "stalled" */
1.4       aaron      87: #define STALLTIME      5
1.79      stevesk    88: /* alarm() interval for updating progress meter */
                     89: #define PROGRESSTIME   1
1.4       aaron      90:
1.11      aaron      91: /* Visual statistics about files as they are transferred. */
                     92: void progressmeter(int);
                     93:
                     94: /* Returns width of the terminal (for progress meter calculations). */
                     95: int getttywidth(void);
1.35      deraadt    96: int do_cmd(char *host, char *remuser, char *cmd, int *fdin, int *fdout, int argc);
1.11      aaron      97:
1.70      mouring    98: /* Struct for addargs */
                     99: arglist args;
1.41      markus    100:
1.11      aaron     101: /* Time a transfer started. */
1.4       aaron     102: static struct timeval start;
1.11      aaron     103:
                    104: /* Number of bytes of current file transferred so far. */
1.68      markus    105: volatile off_t statbytes;
1.11      aaron     106:
                    107: /* Total size of current file. */
1.15      markus    108: off_t totalbytes = 0;
1.11      aaron     109:
                    110: /* Name of current file being transferred. */
                    111: char *curfile;
1.4       aaron     112:
1.1       deraadt   113: /* This is set to non-zero to enable verbose mode. */
1.17      markus    114: int verbose_mode = 0;
1.1       deraadt   115:
1.6       aaron     116: /* This is set to zero if the progressmeter is not desired. */
                    117: int showprogress = 1;
                    118:
1.34      deraadt   119: /* This is the program to execute for the secured connection. ("ssh" or -S) */
1.50      markus    120: char *ssh_program = _PATH_SSH_PROGRAM;
1.34      deraadt   121:
1.92      markus    122: /* This is used to store the pid of ssh_program */
                    123: pid_t do_cmd_pid;
                    124:
1.20      markus    125: /*
                    126:  * This function executes the given command as the specified user on the
                    127:  * given host.  This returns < 0 if execution fails, and >= 0 otherwise. This
                    128:  * assigns the input and output file descriptors on success.
                    129:  */
1.1       deraadt   130:
1.27      markus    131: int
1.35      deraadt   132: do_cmd(char *host, char *remuser, char *cmd, int *fdin, int *fdout, int argc)
1.1       deraadt   133: {
1.18      markus    134:        int pin[2], pout[2], reserved[2];
                    135:
                    136:        if (verbose_mode)
1.78      deraadt   137:                fprintf(stderr,
                    138:                    "Executing: program %s host %s, user %s, command %s\n",
                    139:                    ssh_program, host,
                    140:                    remuser ? remuser : "(unspecified)", cmd);
1.18      markus    141:
1.20      markus    142:        /*
                    143:         * Reserve two descriptors so that the real pipes won't get
                    144:         * descriptors 0 and 1 because that will screw up dup2 below.
                    145:         */
1.18      markus    146:        pipe(reserved);
                    147:
                    148:        /* Create a socket pair for communicating with ssh. */
                    149:        if (pipe(pin) < 0)
                    150:                fatal("pipe: %s", strerror(errno));
                    151:        if (pipe(pout) < 0)
                    152:                fatal("pipe: %s", strerror(errno));
                    153:
                    154:        /* Free the reserved descriptors. */
                    155:        close(reserved[0]);
                    156:        close(reserved[1]);
                    157:
                    158:        /* For a child to execute the command on the remote host using ssh. */
1.92      markus    159:        do_cmd_pid = fork();
                    160:        if (do_cmd_pid == 0) {
1.18      markus    161:                /* Child. */
                    162:                close(pin[1]);
                    163:                close(pout[0]);
                    164:                dup2(pin[0], 0);
                    165:                dup2(pout[1], 1);
                    166:                close(pin[0]);
                    167:                close(pout[1]);
                    168:
1.41      markus    169:                args.list[0] = ssh_program;
                    170:                if (remuser != NULL)
1.70      mouring   171:                        addargs(&args, "-l%s", remuser);
                    172:                addargs(&args, "%s", host);
                    173:                addargs(&args, "%s", cmd);
1.18      markus    174:
1.41      markus    175:                execvp(ssh_program, args.list);
1.34      deraadt   176:                perror(ssh_program);
1.18      markus    177:                exit(1);
1.92      markus    178:        } else if (do_cmd_pid == -1) {
                    179:                fatal("fork: %s", strerror(errno));
1.18      markus    180:        }
                    181:        /* Parent.  Close the other side, and return the local side. */
                    182:        close(pin[0]);
                    183:        *fdout = pin[1];
                    184:        close(pout[1]);
                    185:        *fdin = pout[0];
                    186:        return 0;
1.1       deraadt   187: }
                    188:
                    189: typedef struct {
                    190:        int cnt;
                    191:        char *buf;
                    192: } BUF;
                    193:
1.18      markus    194: BUF *allocbuf(BUF *, int, int);
                    195: void lostconn(int);
                    196: void nospace(void);
                    197: int okname(char *);
                    198: void run_err(const char *,...);
                    199: void verifydir(char *);
1.1       deraadt   200:
                    201: struct passwd *pwd;
1.18      markus    202: uid_t userid;
1.1       deraadt   203: int errs, remin, remout;
                    204: int pflag, iamremote, iamrecursive, targetshouldbedirectory;
                    205:
                    206: #define        CMDNEEDS        64
                    207: char cmd[CMDNEEDS];            /* must hold "rcp -r -p -d\0" */
                    208:
1.18      markus    209: int response(void);
                    210: void rsource(char *, struct stat *);
                    211: void sink(int, char *[]);
                    212: void source(int, char *[]);
                    213: void tolocal(int, char *[]);
                    214: void toremote(char *, int, char *[]);
                    215: void usage(void);
1.1       deraadt   216:
                    217: int
                    218: main(argc, argv)
                    219:        int argc;
                    220:        char *argv[];
                    221: {
1.92      markus    222:        int ch, fflag, tflag, status;
1.1       deraadt   223:        char *targ;
                    224:        extern char *optarg;
                    225:        extern int optind;
                    226:
1.41      markus    227:        args.list = NULL;
1.91      deraadt   228:        addargs(&args, "ssh");          /* overwritten with ssh_program */
1.70      mouring   229:        addargs(&args, "-x");
1.82      stevesk   230:        addargs(&args, "-oForwardAgent no");
1.84      stevesk   231:        addargs(&args, "-oClearAllForwardings yes");
1.41      markus    232:
1.1       deraadt   233:        fflag = tflag = 0;
1.83      stevesk   234:        while ((ch = getopt(argc, argv, "dfprtvBCc:i:P:q46S:o:F:")) != -1)
1.18      markus    235:                switch (ch) {
                    236:                /* User-visible flags. */
1.23      markus    237:                case '4':
1.41      markus    238:                case '6':
                    239:                case 'C':
1.70      mouring   240:                        addargs(&args, "-%c", ch);
1.41      markus    241:                        break;
                    242:                case 'o':
                    243:                case 'c':
                    244:                case 'i':
1.83      stevesk   245:                case 'F':
1.70      mouring   246:                        addargs(&args, "-%c%s", ch, optarg);
1.23      markus    247:                        break;
1.41      markus    248:                case 'P':
1.70      mouring   249:                        addargs(&args, "-p%s", optarg);
1.41      markus    250:                        break;
                    251:                case 'B':
1.70      mouring   252:                        addargs(&args, "-oBatchmode yes");
1.23      markus    253:                        break;
1.1       deraadt   254:                case 'p':
                    255:                        pflag = 1;
                    256:                        break;
                    257:                case 'r':
                    258:                        iamrecursive = 1;
                    259:                        break;
1.34      deraadt   260:                case 'S':
1.41      markus    261:                        ssh_program = xstrdup(optarg);
                    262:                        break;
                    263:                case 'v':
1.72      markus    264:                        addargs(&args, "-v");
1.41      markus    265:                        verbose_mode = 1;
                    266:                        break;
                    267:                case 'q':
                    268:                        showprogress = 0;
1.34      deraadt   269:                        break;
                    270:
1.18      markus    271:                /* Server options. */
1.1       deraadt   272:                case 'd':
                    273:                        targetshouldbedirectory = 1;
                    274:                        break;
1.18      markus    275:                case 'f':       /* "from" */
1.1       deraadt   276:                        iamremote = 1;
                    277:                        fflag = 1;
                    278:                        break;
1.18      markus    279:                case 't':       /* "to" */
1.1       deraadt   280:                        iamremote = 1;
                    281:                        tflag = 1;
                    282:                        break;
                    283:                default:
                    284:                        usage();
                    285:                }
                    286:        argc -= optind;
                    287:        argv += optind;
                    288:
                    289:        if ((pwd = getpwuid(userid = getuid())) == NULL)
1.18      markus    290:                fatal("unknown user %d", (int) userid);
1.1       deraadt   291:
1.18      markus    292:        if (!isatty(STDERR_FILENO))
1.7       deraadt   293:                showprogress = 0;
                    294:
1.1       deraadt   295:        remin = STDIN_FILENO;
                    296:        remout = STDOUT_FILENO;
                    297:
1.52      stevesk   298:        if (fflag) {
1.18      markus    299:                /* Follow "protocol", send data. */
                    300:                (void) response();
1.1       deraadt   301:                source(argc, argv);
                    302:                exit(errs != 0);
                    303:        }
1.18      markus    304:        if (tflag) {
                    305:                /* Receive data. */
1.1       deraadt   306:                sink(argc, argv);
                    307:                exit(errs != 0);
                    308:        }
                    309:        if (argc < 2)
                    310:                usage();
                    311:        if (argc > 2)
                    312:                targetshouldbedirectory = 1;
                    313:
                    314:        remin = remout = -1;
1.92      markus    315:        do_cmd_pid = -1;
1.1       deraadt   316:        /* Command to be executed on remote system using "ssh". */
1.55      deraadt   317:        (void) snprintf(cmd, sizeof cmd, "scp%s%s%s%s",
                    318:            verbose_mode ? " -v" : "",
1.35      deraadt   319:            iamrecursive ? " -r" : "", pflag ? " -p" : "",
                    320:            targetshouldbedirectory ? " -d" : "");
1.1       deraadt   321:
1.18      markus    322:        (void) signal(SIGPIPE, lostconn);
1.1       deraadt   323:
                    324:        if ((targ = colon(argv[argc - 1])))     /* Dest is remote host. */
                    325:                toremote(targ, argc, argv);
                    326:        else {
1.18      markus    327:                tolocal(argc, argv);    /* Dest is local host. */
1.1       deraadt   328:                if (targetshouldbedirectory)
                    329:                        verifydir(argv[argc - 1]);
1.92      markus    330:        }
                    331:        /*
                    332:         * Finally check the exit status of the ssh process, if one was forked
                    333:         * and no error has occured yet
                    334:         */
                    335:        if (do_cmd_pid != -1 && errs == 0) {
                    336:                if (remin != -1)
                    337:                    (void) close(remin);
                    338:                if (remout != -1)
                    339:                    (void) close(remout);
                    340:                if (waitpid(do_cmd_pid, &status, 0) == -1)
                    341:                        errs = 1;
                    342:                else {
                    343:                        if (!WIFEXITED(status) || WEXITSTATUS(status) != 0)
                    344:                                errs = 1;
                    345:                }
1.1       deraadt   346:        }
                    347:        exit(errs != 0);
                    348: }
                    349:
                    350: void
                    351: toremote(targ, argc, argv)
                    352:        char *targ, *argv[];
                    353:        int argc;
                    354: {
                    355:        int i, len;
                    356:        char *bp, *host, *src, *suser, *thost, *tuser;
                    357:
                    358:        *targ++ = 0;
                    359:        if (*targ == 0)
                    360:                targ = ".";
                    361:
1.94      markus    362:        if ((thost = strrchr(argv[argc - 1], '@'))) {
1.1       deraadt   363:                /* user@host */
                    364:                *thost++ = 0;
                    365:                tuser = argv[argc - 1];
                    366:                if (*tuser == '\0')
                    367:                        tuser = NULL;
                    368:                else if (!okname(tuser))
                    369:                        exit(1);
                    370:        } else {
                    371:                thost = argv[argc - 1];
                    372:                tuser = NULL;
                    373:        }
                    374:
                    375:        for (i = 0; i < argc - 1; i++) {
                    376:                src = colon(argv[i]);
1.18      markus    377:                if (src) {      /* remote to remote */
1.84      stevesk   378:                        static char *ssh_options =
1.89      markus    379:                            "-x -o'ClearAllForwardings yes'";
1.1       deraadt   380:                        *src++ = 0;
                    381:                        if (*src == 0)
                    382:                                src = ".";
1.94      markus    383:                        host = strrchr(argv[i], '@');
1.34      deraadt   384:                        len = strlen(ssh_program) + strlen(argv[i]) +
1.35      deraadt   385:                            strlen(src) + (tuser ? strlen(tuser) : 0) +
1.84      stevesk   386:                            strlen(thost) + strlen(targ) +
                    387:                            strlen(ssh_options) + CMDNEEDS + 20;
1.18      markus    388:                        bp = xmalloc(len);
1.1       deraadt   389:                        if (host) {
                    390:                                *host++ = 0;
1.23      markus    391:                                host = cleanhostname(host);
1.1       deraadt   392:                                suser = argv[i];
                    393:                                if (*suser == '\0')
                    394:                                        suser = pwd->pw_name;
                    395:                                else if (!okname(suser))
                    396:                                        continue;
1.55      deraadt   397:                                snprintf(bp, len,
1.84      stevesk   398:                                    "%s%s %s -n "
1.55      deraadt   399:                                    "-l %s %s %s %s '%s%s%s:%s'",
1.44      deraadt   400:                                    ssh_program, verbose_mode ? " -v" : "",
1.84      stevesk   401:                                    ssh_options, suser, host, cmd, src,
1.44      deraadt   402:                                    tuser ? tuser : "", tuser ? "@" : "",
                    403:                                    thost, targ);
1.23      markus    404:                        } else {
                    405:                                host = cleanhostname(argv[i]);
1.55      deraadt   406:                                snprintf(bp, len,
1.84      stevesk   407:                                    "exec %s%s %s -n %s "
1.55      deraadt   408:                                    "%s %s '%s%s%s:%s'",
1.44      deraadt   409:                                    ssh_program, verbose_mode ? " -v" : "",
1.84      stevesk   410:                                    ssh_options, host, cmd, src,
1.44      deraadt   411:                                    tuser ? tuser : "", tuser ? "@" : "",
                    412:                                    thost, targ);
1.23      markus    413:                        }
1.18      markus    414:                        if (verbose_mode)
                    415:                                fprintf(stderr, "Executing: %s\n", bp);
                    416:                        (void) system(bp);
                    417:                        (void) xfree(bp);
                    418:                } else {        /* local to remote */
1.1       deraadt   419:                        if (remin == -1) {
                    420:                                len = strlen(targ) + CMDNEEDS + 20;
1.18      markus    421:                                bp = xmalloc(len);
1.55      deraadt   422:                                (void) snprintf(bp, len, "%s -t %s", cmd, targ);
1.23      markus    423:                                host = cleanhostname(thost);
1.35      deraadt   424:                                if (do_cmd(host, tuser, bp, &remin,
                    425:                                    &remout, argc) < 0)
1.18      markus    426:                                        exit(1);
1.1       deraadt   427:                                if (response() < 0)
                    428:                                        exit(1);
1.18      markus    429:                                (void) xfree(bp);
1.1       deraadt   430:                        }
1.18      markus    431:                        source(1, argv + i);
1.1       deraadt   432:                }
                    433:        }
                    434: }
                    435:
                    436: void
                    437: tolocal(argc, argv)
                    438:        int argc;
                    439:        char *argv[];
                    440: {
                    441:        int i, len;
                    442:        char *bp, *host, *src, *suser;
                    443:
                    444:        for (i = 0; i < argc - 1; i++) {
1.18      markus    445:                if (!(src = colon(argv[i]))) {  /* Local to local. */
1.1       deraadt   446:                        len = strlen(_PATH_CP) + strlen(argv[i]) +
1.35      deraadt   447:                            strlen(argv[argc - 1]) + 20;
1.1       deraadt   448:                        bp = xmalloc(len);
1.55      deraadt   449:                        (void) snprintf(bp, len, "exec %s%s%s %s %s", _PATH_CP,
1.35      deraadt   450:                            iamrecursive ? " -r" : "", pflag ? " -p" : "",
                    451:                            argv[i], argv[argc - 1]);
1.18      markus    452:                        if (verbose_mode)
                    453:                                fprintf(stderr, "Executing: %s\n", bp);
1.1       deraadt   454:                        if (system(bp))
                    455:                                ++errs;
1.18      markus    456:                        (void) xfree(bp);
1.1       deraadt   457:                        continue;
                    458:                }
                    459:                *src++ = 0;
                    460:                if (*src == 0)
                    461:                        src = ".";
1.94      markus    462:                if ((host = strrchr(argv[i], '@')) == NULL) {
1.1       deraadt   463:                        host = argv[i];
                    464:                        suser = NULL;
                    465:                } else {
                    466:                        *host++ = 0;
                    467:                        suser = argv[i];
                    468:                        if (*suser == '\0')
                    469:                                suser = pwd->pw_name;
                    470:                        else if (!okname(suser))
                    471:                                continue;
                    472:                }
1.23      markus    473:                host = cleanhostname(host);
1.1       deraadt   474:                len = strlen(src) + CMDNEEDS + 20;
1.18      markus    475:                bp = xmalloc(len);
1.55      deraadt   476:                (void) snprintf(bp, len, "%s -f %s", cmd, src);
1.35      deraadt   477:                if (do_cmd(host, suser, bp, &remin, &remout, argc) < 0) {
1.18      markus    478:                        (void) xfree(bp);
                    479:                        ++errs;
                    480:                        continue;
1.1       deraadt   481:                }
1.18      markus    482:                xfree(bp);
1.1       deraadt   483:                sink(1, argv + argc - 1);
1.18      markus    484:                (void) close(remin);
1.1       deraadt   485:                remin = remout = -1;
                    486:        }
                    487: }
                    488:
                    489: void
                    490: source(argc, argv)
                    491:        int argc;
                    492:        char *argv[];
                    493: {
                    494:        struct stat stb;
                    495:        static BUF buffer;
                    496:        BUF *bp;
1.68      markus    497:        off_t i, amt, result;
                    498:        int fd, haderr, indx;
1.1       deraadt   499:        char *last, *name, buf[2048];
1.65      deraadt   500:        int len;
1.1       deraadt   501:
                    502:        for (indx = 0; indx < argc; ++indx) {
1.18      markus    503:                name = argv[indx];
1.11      aaron     504:                statbytes = 0;
1.65      deraadt   505:                len = strlen(name);
                    506:                while (len > 1 && name[len-1] == '/')
                    507:                        name[--len] = '\0';
1.85      markus    508:                if (strchr(name, '\n') != NULL) {
                    509:                        run_err("%s: skipping, filename contains a newline",
                    510:                            name);
                    511:                        goto next;
                    512:                }
1.1       deraadt   513:                if ((fd = open(name, O_RDONLY, 0)) < 0)
                    514:                        goto syserr;
                    515:                if (fstat(fd, &stb) < 0) {
                    516: syserr:                        run_err("%s: %s", name, strerror(errno));
                    517:                        goto next;
                    518:                }
                    519:                switch (stb.st_mode & S_IFMT) {
                    520:                case S_IFREG:
                    521:                        break;
                    522:                case S_IFDIR:
                    523:                        if (iamrecursive) {
                    524:                                rsource(name, &stb);
                    525:                                goto next;
                    526:                        }
                    527:                        /* FALLTHROUGH */
                    528:                default:
                    529:                        run_err("%s: not a regular file", name);
                    530:                        goto next;
                    531:                }
                    532:                if ((last = strrchr(name, '/')) == NULL)
                    533:                        last = name;
                    534:                else
                    535:                        ++last;
1.11      aaron     536:                curfile = last;
1.1       deraadt   537:                if (pflag) {
                    538:                        /*
                    539:                         * Make it compatible with possible future
                    540:                         * versions expecting microseconds.
                    541:                         */
1.55      deraadt   542:                        (void) snprintf(buf, sizeof buf, "T%lu 0 %lu 0\n",
1.47      markus    543:                            (u_long) stb.st_mtime,
                    544:                            (u_long) stb.st_atime);
1.29      deraadt   545:                        (void) atomicio(write, remout, buf, strlen(buf));
1.1       deraadt   546:                        if (response() < 0)
                    547:                                goto next;
                    548:                }
                    549: #define        FILEMODEMASK    (S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO)
1.61      markus    550:                snprintf(buf, sizeof buf, "C%04o %lld %s\n",
1.47      markus    551:                    (u_int) (stb.st_mode & FILEMODEMASK),
1.62      markus    552:                    (long long)stb.st_size, last);
1.18      markus    553:                if (verbose_mode) {
                    554:                        fprintf(stderr, "Sending file modes: %s", buf);
                    555:                }
1.29      deraadt   556:                (void) atomicio(write, remout, buf, strlen(buf));
1.1       deraadt   557:                if (response() < 0)
                    558:                        goto next;
                    559:                if ((bp = allocbuf(&buffer, fd, 2048)) == NULL) {
1.18      markus    560: next:                  (void) close(fd);
1.1       deraadt   561:                        continue;
                    562:                }
1.12      aaron     563:                if (showprogress) {
                    564:                        totalbytes = stb.st_size;
1.6       aaron     565:                        progressmeter(-1);
1.12      aaron     566:                }
1.1       deraadt   567:                /* Keep writing after an error so that we stay sync'd up. */
                    568:                for (haderr = i = 0; i < stb.st_size; i += bp->cnt) {
                    569:                        amt = bp->cnt;
                    570:                        if (i + amt > stb.st_size)
                    571:                                amt = stb.st_size - i;
                    572:                        if (!haderr) {
1.30      deraadt   573:                                result = atomicio(read, fd, bp->buf, amt);
1.1       deraadt   574:                                if (result != amt)
                    575:                                        haderr = result >= 0 ? EIO : errno;
                    576:                        }
                    577:                        if (haderr)
1.29      deraadt   578:                                (void) atomicio(write, remout, bp->buf, amt);
1.1       deraadt   579:                        else {
1.29      deraadt   580:                                result = atomicio(write, remout, bp->buf, amt);
1.1       deraadt   581:                                if (result != amt)
                    582:                                        haderr = result >= 0 ? EIO : errno;
1.4       aaron     583:                                statbytes += result;
1.1       deraadt   584:                        }
                    585:                }
1.18      markus    586:                if (showprogress)
1.6       aaron     587:                        progressmeter(1);
1.4       aaron     588:
1.1       deraadt   589:                if (close(fd) < 0 && !haderr)
                    590:                        haderr = errno;
                    591:                if (!haderr)
1.29      deraadt   592:                        (void) atomicio(write, remout, "", 1);
1.1       deraadt   593:                else
                    594:                        run_err("%s: %s", name, strerror(haderr));
1.18      markus    595:                (void) response();
1.1       deraadt   596:        }
                    597: }
                    598:
                    599: void
                    600: rsource(name, statp)
                    601:        char *name;
                    602:        struct stat *statp;
                    603: {
                    604:        DIR *dirp;
                    605:        struct dirent *dp;
                    606:        char *last, *vect[1], path[1100];
                    607:
                    608:        if (!(dirp = opendir(name))) {
                    609:                run_err("%s: %s", name, strerror(errno));
                    610:                return;
                    611:        }
                    612:        last = strrchr(name, '/');
                    613:        if (last == 0)
                    614:                last = name;
                    615:        else
                    616:                last++;
                    617:        if (pflag) {
1.55      deraadt   618:                (void) snprintf(path, sizeof(path), "T%lu 0 %lu 0\n",
1.47      markus    619:                    (u_long) statp->st_mtime,
                    620:                    (u_long) statp->st_atime);
1.29      deraadt   621:                (void) atomicio(write, remout, path, strlen(path));
1.1       deraadt   622:                if (response() < 0) {
                    623:                        closedir(dirp);
                    624:                        return;
                    625:                }
                    626:        }
1.55      deraadt   627:        (void) snprintf(path, sizeof path, "D%04o %d %.1024s\n",
1.47      markus    628:            (u_int) (statp->st_mode & FILEMODEMASK), 0, last);
1.18      markus    629:        if (verbose_mode)
                    630:                fprintf(stderr, "Entering directory: %s", path);
1.29      deraadt   631:        (void) atomicio(write, remout, path, strlen(path));
1.1       deraadt   632:        if (response() < 0) {
                    633:                closedir(dirp);
                    634:                return;
                    635:        }
1.63      stevesk   636:        while ((dp = readdir(dirp)) != NULL) {
1.1       deraadt   637:                if (dp->d_ino == 0)
                    638:                        continue;
                    639:                if (!strcmp(dp->d_name, ".") || !strcmp(dp->d_name, ".."))
                    640:                        continue;
                    641:                if (strlen(name) + 1 + strlen(dp->d_name) >= sizeof(path) - 1) {
                    642:                        run_err("%s/%s: name too long", name, dp->d_name);
                    643:                        continue;
                    644:                }
1.55      deraadt   645:                (void) snprintf(path, sizeof path, "%s/%s", name, dp->d_name);
1.1       deraadt   646:                vect[0] = path;
                    647:                source(1, vect);
                    648:        }
1.18      markus    649:        (void) closedir(dirp);
1.29      deraadt   650:        (void) atomicio(write, remout, "E\n", 2);
1.18      markus    651:        (void) response();
1.1       deraadt   652: }
                    653:
                    654: void
                    655: sink(argc, argv)
                    656:        int argc;
                    657:        char *argv[];
                    658: {
                    659:        static BUF buffer;
                    660:        struct stat stb;
1.18      markus    661:        enum {
                    662:                YES, NO, DISPLAYED
                    663:        } wrerr;
1.1       deraadt   664:        BUF *bp;
                    665:        off_t i, j;
                    666:        int amt, count, exists, first, mask, mode, ofd, omode;
1.36      deraadt   667:        off_t size;
                    668:        int setimes, targisdir, wrerrno = 0;
1.1       deraadt   669:        char ch, *cp, *np, *targ, *why, *vect[1], buf[2048];
1.40      markus    670:        struct timeval tv[2];
1.1       deraadt   671:
1.66      stevesk   672: #define        atime   tv[0]
                    673: #define        mtime   tv[1]
1.75      deraadt   674: #define        SCREWUP(str)    do { why = str; goto screwup; } while (0)
1.1       deraadt   675:
                    676:        setimes = targisdir = 0;
                    677:        mask = umask(0);
                    678:        if (!pflag)
1.18      markus    679:                (void) umask(mask);
1.1       deraadt   680:        if (argc != 1) {
                    681:                run_err("ambiguous target");
                    682:                exit(1);
                    683:        }
                    684:        targ = *argv;
                    685:        if (targetshouldbedirectory)
                    686:                verifydir(targ);
1.18      markus    687:
1.29      deraadt   688:        (void) atomicio(write, remout, "", 1);
1.1       deraadt   689:        if (stat(targ, &stb) == 0 && S_ISDIR(stb.st_mode))
                    690:                targisdir = 1;
                    691:        for (first = 1;; first = 0) {
                    692:                cp = buf;
1.30      deraadt   693:                if (atomicio(read, remin, cp, 1) <= 0)
1.1       deraadt   694:                        return;
                    695:                if (*cp++ == '\n')
                    696:                        SCREWUP("unexpected <newline>");
                    697:                do {
1.30      deraadt   698:                        if (atomicio(read, remin, &ch, sizeof(ch)) != sizeof(ch))
1.1       deraadt   699:                                SCREWUP("lost connection");
                    700:                        *cp++ = ch;
                    701:                } while (cp < &buf[sizeof(buf) - 1] && ch != '\n');
                    702:                *cp = 0;
                    703:
                    704:                if (buf[0] == '\01' || buf[0] == '\02') {
                    705:                        if (iamremote == 0)
1.29      deraadt   706:                                (void) atomicio(write, STDERR_FILENO,
1.44      deraadt   707:                                    buf + 1, strlen(buf + 1));
1.1       deraadt   708:                        if (buf[0] == '\02')
                    709:                                exit(1);
                    710:                        ++errs;
                    711:                        continue;
                    712:                }
                    713:                if (buf[0] == 'E') {
1.29      deraadt   714:                        (void) atomicio(write, remout, "", 1);
1.1       deraadt   715:                        return;
                    716:                }
                    717:                if (ch == '\n')
                    718:                        *--cp = 0;
                    719:
                    720:                cp = buf;
                    721:                if (*cp == 'T') {
                    722:                        setimes++;
                    723:                        cp++;
1.66      stevesk   724:                        mtime.tv_sec = strtol(cp, &cp, 10);
                    725:                        if (!cp || *cp++ != ' ')
1.1       deraadt   726:                                SCREWUP("mtime.sec not delimited");
1.66      stevesk   727:                        mtime.tv_usec = strtol(cp, &cp, 10);
                    728:                        if (!cp || *cp++ != ' ')
1.1       deraadt   729:                                SCREWUP("mtime.usec not delimited");
1.66      stevesk   730:                        atime.tv_sec = strtol(cp, &cp, 10);
                    731:                        if (!cp || *cp++ != ' ')
1.1       deraadt   732:                                SCREWUP("atime.sec not delimited");
1.66      stevesk   733:                        atime.tv_usec = strtol(cp, &cp, 10);
                    734:                        if (!cp || *cp++ != '\0')
1.1       deraadt   735:                                SCREWUP("atime.usec not delimited");
1.29      deraadt   736:                        (void) atomicio(write, remout, "", 1);
1.1       deraadt   737:                        continue;
                    738:                }
                    739:                if (*cp != 'C' && *cp != 'D') {
                    740:                        /*
                    741:                         * Check for the case "rcp remote:foo\* local:bar".
                    742:                         * In this case, the line "No match." can be returned
                    743:                         * by the shell before the rcp command on the remote is
                    744:                         * executed so the ^Aerror_message convention isn't
                    745:                         * followed.
                    746:                         */
                    747:                        if (first) {
                    748:                                run_err("%s", cp);
                    749:                                exit(1);
                    750:                        }
                    751:                        SCREWUP("expected control record");
                    752:                }
                    753:                mode = 0;
                    754:                for (++cp; cp < buf + 5; cp++) {
                    755:                        if (*cp < '0' || *cp > '7')
                    756:                                SCREWUP("bad mode");
                    757:                        mode = (mode << 3) | (*cp - '0');
                    758:                }
                    759:                if (*cp++ != ' ')
                    760:                        SCREWUP("mode not delimited");
                    761:
1.63      stevesk   762:                for (size = 0; isdigit(*cp);)
1.1       deraadt   763:                        size = size * 10 + (*cp++ - '0');
                    764:                if (*cp++ != ' ')
                    765:                        SCREWUP("size not delimited");
                    766:                if (targisdir) {
                    767:                        static char *namebuf;
                    768:                        static int cursize;
                    769:                        size_t need;
                    770:
                    771:                        need = strlen(targ) + strlen(cp) + 250;
1.55      deraadt   772:                        if (need > cursize) {
1.58      danh      773:                                if (namebuf)
                    774:                                        xfree(namebuf);
1.18      markus    775:                                namebuf = xmalloc(need);
1.55      deraadt   776:                                cursize = need;
                    777:                        }
                    778:                        (void) snprintf(namebuf, need, "%s%s%s", targ,
1.88      mouring   779:                            strcmp(targ, "/") ? "/" : "", cp);
1.1       deraadt   780:                        np = namebuf;
                    781:                } else
                    782:                        np = targ;
1.12      aaron     783:                curfile = cp;
1.1       deraadt   784:                exists = stat(np, &stb) == 0;
                    785:                if (buf[0] == 'D') {
                    786:                        int mod_flag = pflag;
                    787:                        if (exists) {
                    788:                                if (!S_ISDIR(stb.st_mode)) {
                    789:                                        errno = ENOTDIR;
                    790:                                        goto bad;
                    791:                                }
                    792:                                if (pflag)
1.18      markus    793:                                        (void) chmod(np, mode);
1.1       deraadt   794:                        } else {
1.18      markus    795:                                /* Handle copying from a read-only
                    796:                                   directory */
1.1       deraadt   797:                                mod_flag = 1;
                    798:                                if (mkdir(np, mode | S_IRWXU) < 0)
                    799:                                        goto bad;
                    800:                        }
1.58      danh      801:                        vect[0] = xstrdup(np);
1.1       deraadt   802:                        sink(1, vect);
                    803:                        if (setimes) {
                    804:                                setimes = 0;
1.59      deraadt   805:                                if (utimes(vect[0], tv) < 0)
1.18      markus    806:                                        run_err("%s: set times: %s",
1.59      deraadt   807:                                            vect[0], strerror(errno));
1.1       deraadt   808:                        }
                    809:                        if (mod_flag)
1.59      deraadt   810:                                (void) chmod(vect[0], mode);
                    811:                        if (vect[0])
                    812:                                xfree(vect[0]);
1.1       deraadt   813:                        continue;
                    814:                }
                    815:                omode = mode;
                    816:                mode |= S_IWRITE;
1.71      markus    817:                if ((ofd = open(np, O_WRONLY|O_CREAT, mode)) < 0) {
1.1       deraadt   818: bad:                   run_err("%s: %s", np, strerror(errno));
                    819:                        continue;
                    820:                }
1.29      deraadt   821:                (void) atomicio(write, remout, "", 1);
1.1       deraadt   822:                if ((bp = allocbuf(&buffer, ofd, 4096)) == NULL) {
1.18      markus    823:                        (void) close(ofd);
1.1       deraadt   824:                        continue;
                    825:                }
                    826:                cp = bp->buf;
                    827:                wrerr = NO;
1.7       deraadt   828:
                    829:                if (showprogress) {
                    830:                        totalbytes = size;
                    831:                        progressmeter(-1);
                    832:                }
1.12      aaron     833:                statbytes = 0;
1.1       deraadt   834:                for (count = i = 0; i < size; i += 4096) {
                    835:                        amt = 4096;
                    836:                        if (i + amt > size)
                    837:                                amt = size - i;
                    838:                        count += amt;
                    839:                        do {
1.43      markus    840:                                j = read(remin, cp, amt);
1.78      deraadt   841:                                if (j == -1 && (errno == EINTR ||
                    842:                                    errno == EAGAIN)) {
1.43      markus    843:                                        continue;
                    844:                                } else if (j <= 0) {
1.1       deraadt   845:                                        run_err("%s", j ? strerror(errno) :
1.63      stevesk   846:                                            "dropped connection");
1.1       deraadt   847:                                        exit(1);
                    848:                                }
                    849:                                amt -= j;
                    850:                                cp += j;
1.18      markus    851:                                statbytes += j;
1.1       deraadt   852:                        } while (amt > 0);
                    853:                        if (count == bp->cnt) {
                    854:                                /* Keep reading so we stay sync'd up. */
                    855:                                if (wrerr == NO) {
1.30      deraadt   856:                                        j = atomicio(write, ofd, bp->buf, count);
1.1       deraadt   857:                                        if (j != count) {
                    858:                                                wrerr = YES;
1.18      markus    859:                                                wrerrno = j >= 0 ? EIO : errno;
1.1       deraadt   860:                                        }
                    861:                                }
                    862:                                count = 0;
                    863:                                cp = bp->buf;
                    864:                        }
                    865:                }
1.7       deraadt   866:                if (showprogress)
                    867:                        progressmeter(1);
1.1       deraadt   868:                if (count != 0 && wrerr == NO &&
1.30      deraadt   869:                    (j = atomicio(write, ofd, bp->buf, count)) != count) {
1.1       deraadt   870:                        wrerr = YES;
1.18      markus    871:                        wrerrno = j >= 0 ? EIO : errno;
1.1       deraadt   872:                }
                    873:                if (ftruncate(ofd, size)) {
                    874:                        run_err("%s: truncate: %s", np, strerror(errno));
                    875:                        wrerr = DISPLAYED;
                    876:                }
                    877:                if (pflag) {
                    878:                        if (exists || omode != mode)
                    879:                                if (fchmod(ofd, omode))
                    880:                                        run_err("%s: set mode: %s",
1.63      stevesk   881:                                            np, strerror(errno));
1.1       deraadt   882:                } else {
                    883:                        if (!exists && omode != mode)
                    884:                                if (fchmod(ofd, omode & ~mask))
                    885:                                        run_err("%s: set mode: %s",
1.63      stevesk   886:                                            np, strerror(errno));
1.1       deraadt   887:                }
1.33      provos    888:                if (close(ofd) == -1) {
                    889:                        wrerr = YES;
                    890:                        wrerrno = errno;
                    891:                }
1.18      markus    892:                (void) response();
1.1       deraadt   893:                if (setimes && wrerr == NO) {
                    894:                        setimes = 0;
1.40      markus    895:                        if (utimes(np, tv) < 0) {
1.1       deraadt   896:                                run_err("%s: set times: %s",
1.63      stevesk   897:                                    np, strerror(errno));
1.1       deraadt   898:                                wrerr = DISPLAYED;
                    899:                        }
                    900:                }
1.18      markus    901:                switch (wrerr) {
1.1       deraadt   902:                case YES:
                    903:                        run_err("%s: %s", np, strerror(wrerrno));
                    904:                        break;
                    905:                case NO:
1.29      deraadt   906:                        (void) atomicio(write, remout, "", 1);
1.1       deraadt   907:                        break;
                    908:                case DISPLAYED:
                    909:                        break;
                    910:                }
                    911:        }
                    912: screwup:
                    913:        run_err("protocol error: %s", why);
                    914:        exit(1);
                    915: }
                    916:
                    917: int
1.86      itojun    918: response(void)
1.1       deraadt   919: {
                    920:        char ch, *cp, resp, rbuf[2048];
                    921:
1.30      deraadt   922:        if (atomicio(read, remin, &resp, sizeof(resp)) != sizeof(resp))
1.1       deraadt   923:                lostconn(0);
                    924:
                    925:        cp = rbuf;
1.18      markus    926:        switch (resp) {
                    927:        case 0:         /* ok */
1.1       deraadt   928:                return (0);
                    929:        default:
                    930:                *cp++ = resp;
                    931:                /* FALLTHROUGH */
1.18      markus    932:        case 1:         /* error, followed by error msg */
                    933:        case 2:         /* fatal error, "" */
1.1       deraadt   934:                do {
1.30      deraadt   935:                        if (atomicio(read, remin, &ch, sizeof(ch)) != sizeof(ch))
1.1       deraadt   936:                                lostconn(0);
                    937:                        *cp++ = ch;
                    938:                } while (cp < &rbuf[sizeof(rbuf) - 1] && ch != '\n');
                    939:
                    940:                if (!iamremote)
1.29      deraadt   941:                        (void) atomicio(write, STDERR_FILENO, rbuf, cp - rbuf);
1.1       deraadt   942:                ++errs;
                    943:                if (resp == 1)
                    944:                        return (-1);
                    945:                exit(1);
                    946:        }
                    947:        /* NOTREACHED */
                    948: }
                    949:
                    950: void
1.86      itojun    951: usage(void)
1.1       deraadt   952: {
1.83      stevesk   953:        (void) fprintf(stderr,
1.90      deraadt   954:            "usage: scp [-pqrvBC46] [-F config] [-S program] [-P port]\n"
                    955:            "           [-c cipher] [-i identity] [-o option]\n"
                    956:            "           [[user@]host1:]file1 [...] [[user@]host2:]file2\n");
1.1       deraadt   957:        exit(1);
                    958: }
                    959:
                    960: void
1.18      markus    961: run_err(const char *fmt,...)
1.1       deraadt   962: {
                    963:        static FILE *fp;
                    964:        va_list ap;
                    965:
                    966:        ++errs;
                    967:        if (fp == NULL && !(fp = fdopen(remout, "w")))
                    968:                return;
1.18      markus    969:        (void) fprintf(fp, "%c", 0x01);
                    970:        (void) fprintf(fp, "scp: ");
1.73      markus    971:        va_start(ap, fmt);
1.18      markus    972:        (void) vfprintf(fp, fmt, ap);
1.73      markus    973:        va_end(ap);
1.18      markus    974:        (void) fprintf(fp, "\n");
                    975:        (void) fflush(fp);
                    976:
                    977:        if (!iamremote) {
1.73      markus    978:                va_start(ap, fmt);
1.18      markus    979:                vfprintf(stderr, fmt, ap);
1.73      markus    980:                va_end(ap);
1.18      markus    981:                fprintf(stderr, "\n");
                    982:        }
1.1       deraadt   983: }
                    984:
                    985: void
                    986: verifydir(cp)
                    987:        char *cp;
                    988: {
                    989:        struct stat stb;
                    990:
                    991:        if (!stat(cp, &stb)) {
                    992:                if (S_ISDIR(stb.st_mode))
                    993:                        return;
                    994:                errno = ENOTDIR;
                    995:        }
                    996:        run_err("%s: %s", cp, strerror(errno));
                    997:        exit(1);
                    998: }
                    999:
                   1000: int
                   1001: okname(cp0)
                   1002:        char *cp0;
                   1003: {
                   1004:        int c;
                   1005:        char *cp;
                   1006:
                   1007:        cp = cp0;
                   1008:        do {
1.75      deraadt  1009:                c = (int)*cp;
1.1       deraadt  1010:                if (c & 0200)
                   1011:                        goto bad;
1.46      markus   1012:                if (!isalpha(c) && !isdigit(c) &&
1.94      markus   1013:                    c != '@' && c != '_' && c != '-' && c != '.' && c != '+')
1.1       deraadt  1014:                        goto bad;
                   1015:        } while (*++cp);
                   1016:        return (1);
                   1017:
1.25      markus   1018: bad:   fprintf(stderr, "%s: invalid user name\n", cp0);
1.1       deraadt  1019:        return (0);
                   1020: }
                   1021:
                   1022: BUF *
                   1023: allocbuf(bp, fd, blksize)
                   1024:        BUF *bp;
                   1025:        int fd, blksize;
                   1026: {
                   1027:        size_t size;
                   1028:        struct stat stb;
                   1029:
                   1030:        if (fstat(fd, &stb) < 0) {
                   1031:                run_err("fstat: %s", strerror(errno));
                   1032:                return (0);
                   1033:        }
1.95      markus   1034:        size = roundup(stb.st_blksize, blksize);
                   1035:        if (size == 0)
1.18      markus   1036:                size = blksize;
1.1       deraadt  1037:        if (bp->cnt >= size)
                   1038:                return (bp);
1.18      markus   1039:        if (bp->buf == NULL)
                   1040:                bp->buf = xmalloc(size);
                   1041:        else
                   1042:                bp->buf = xrealloc(bp->buf, size);
1.81      markus   1043:        memset(bp->buf, 0, size);
1.1       deraadt  1044:        bp->cnt = size;
                   1045:        return (bp);
                   1046: }
                   1047:
                   1048: void
                   1049: lostconn(signo)
                   1050:        int signo;
                   1051: {
                   1052:        if (!iamremote)
1.74      markus   1053:                write(STDERR_FILENO, "lost connection\n", 16);
                   1054:        if (signo)
                   1055:                _exit(1);
                   1056:        else
                   1057:                exit(1);
1.1       deraadt  1058: }
1.4       aaron    1059:
1.76      itojun   1060: static void
1.26      markus   1061: updateprogressmeter(int ignore)
1.4       aaron    1062: {
1.9       deraadt  1063:        int save_errno = errno;
                   1064:
1.4       aaron    1065:        progressmeter(0);
1.79      stevesk  1066:        signal(SIGALRM, updateprogressmeter);
                   1067:        alarm(PROGRESSTIME);
1.9       deraadt  1068:        errno = save_errno;
1.4       aaron    1069: }
                   1070:
1.76      itojun   1071: static int
1.56      itojun   1072: foregroundproc(void)
1.16      deraadt  1073: {
                   1074:        static pid_t pgrp = -1;
                   1075:        int ctty_pgrp;
                   1076:
                   1077:        if (pgrp == -1)
                   1078:                pgrp = getpgrp();
                   1079:
1.18      markus   1080:        return ((ioctl(STDOUT_FILENO, TIOCGPGRP, &ctty_pgrp) != -1 &&
                   1081:                 ctty_pgrp == pgrp));
1.16      deraadt  1082: }
                   1083:
1.8       deraadt  1084: void
                   1085: progressmeter(int flag)
1.4       aaron    1086: {
1.96    ! markus   1087:        static const char spaces[] = "                          "
        !          1088:            "                                                   "
        !          1089:            "                                                   "
        !          1090:            "                                                   "
        !          1091:            "                                                   "
        !          1092:            "                                                   ";
1.4       aaron    1093:        static const char prefixes[] = " KMGTP";
                   1094:        static struct timeval lastupdate;
1.11      aaron    1095:        static off_t lastsize;
1.4       aaron    1096:        struct timeval now, td, wait;
1.96    ! markus   1097:        off_t cursize, abbrevsize, bytespersec;
1.4       aaron    1098:        double elapsed;
1.96    ! markus   1099:        int ratio, remaining, i, ai, bi, nspaces;
1.87      deraadt  1100:        char buf[512];
1.4       aaron    1101:
                   1102:        if (flag == -1) {
1.18      markus   1103:                (void) gettimeofday(&start, (struct timezone *) 0);
1.4       aaron    1104:                lastupdate = start;
1.11      aaron    1105:                lastsize = 0;
1.18      markus   1106:        }
1.16      deraadt  1107:        if (foregroundproc() == 0)
                   1108:                return;
                   1109:
1.18      markus   1110:        (void) gettimeofday(&now, (struct timezone *) 0);
1.4       aaron    1111:        cursize = statbytes;
1.13      aaron    1112:        if (totalbytes != 0) {
1.21      markus   1113:                ratio = 100.0 * cursize / totalbytes;
1.13      aaron    1114:                ratio = MAX(ratio, 0);
                   1115:                ratio = MIN(ratio, 100);
1.18      markus   1116:        } else
1.13      aaron    1117:                ratio = 100;
                   1118:
1.4       aaron    1119:        abbrevsize = cursize;
1.96    ! markus   1120:        for (ai = 0; abbrevsize >= 10000 && ai < sizeof(prefixes); ai++)
1.4       aaron    1121:                abbrevsize >>= 10;
                   1122:
                   1123:        timersub(&now, &lastupdate, &wait);
                   1124:        if (cursize > lastsize) {
                   1125:                lastupdate = now;
                   1126:                lastsize = cursize;
                   1127:                wait.tv_sec = 0;
                   1128:        }
                   1129:        timersub(&now, &start, &td);
1.6       aaron    1130:        elapsed = td.tv_sec + (td.tv_usec / 1000000.0);
1.96    ! markus   1131:
        !          1132:        bytespersec = 0;
        !          1133:        if (statbytes > 0) {
        !          1134:                bytespersec = statbytes;
        !          1135:                if (elapsed > 0.0)
        !          1136:                        bytespersec /= elapsed;
        !          1137:        }
        !          1138:        for (bi = 1; bytespersec >= 1024000 && bi < sizeof(prefixes); bi++)
        !          1139:                bytespersec >>= 10;
        !          1140:
        !          1141:        nspaces = MIN(getttywidth() - 79, sizeof(spaces) - 1);
        !          1142:
        !          1143:        snprintf(buf, sizeof(buf),
        !          1144:            "\r%-45.45s%.*s%3d%% %4lld%c%c %3lld.%01d%cB/s",
        !          1145:            curfile,
        !          1146:            nspaces,
        !          1147:            spaces,
        !          1148:            ratio,
        !          1149:            (long long)abbrevsize,
        !          1150:            prefixes[ai],
        !          1151:            ai == 0 ? ' ' : 'B',
        !          1152:            (long long)(bytespersec / 1024),
        !          1153:            (int)((bytespersec % 1024) * 10 / 1024),
        !          1154:            prefixes[bi]
        !          1155:        );
1.4       aaron    1156:
1.44      deraadt  1157:        if (flag != 1 &&
                   1158:            (statbytes <= 0 || elapsed <= 0.0 || cursize > totalbytes)) {
1.4       aaron    1159:                snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
1.44      deraadt  1160:                    "   --:-- ETA");
1.4       aaron    1161:        } else if (wait.tv_sec >= STALLTIME) {
                   1162:                snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
1.44      deraadt  1163:                    " - stalled -");
1.4       aaron    1164:        } else {
1.28      deraadt  1165:                if (flag != 1)
1.44      deraadt  1166:                        remaining = (int)(totalbytes / (statbytes / elapsed) -
                   1167:                            elapsed);
1.28      deraadt  1168:                else
                   1169:                        remaining = elapsed;
                   1170:
1.14      aaron    1171:                i = remaining / 3600;
1.4       aaron    1172:                if (i)
                   1173:                        snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
1.35      deraadt  1174:                            "%2d:", i);
1.4       aaron    1175:                else
                   1176:                        snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
1.35      deraadt  1177:                            "   ");
1.4       aaron    1178:                i = remaining % 3600;
                   1179:                snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
1.35      deraadt  1180:                    "%02d:%02d%s", i / 60, i % 60,
                   1181:                    (flag != 1) ? " ETA" : "    ");
1.4       aaron    1182:        }
1.10      deraadt  1183:        atomicio(write, fileno(stdout), buf, strlen(buf));
1.4       aaron    1184:
                   1185:        if (flag == -1) {
1.26      markus   1186:                signal(SIGALRM, updateprogressmeter);
1.79      stevesk  1187:                alarm(PROGRESSTIME);
1.4       aaron    1188:        } else if (flag == 1) {
1.79      stevesk  1189:                alarm(0);
1.29      deraadt  1190:                atomicio(write, fileno(stdout), "\n", 1);
1.12      aaron    1191:                statbytes = 0;
1.4       aaron    1192:        }
                   1193: }
                   1194:
1.8       deraadt  1195: int
                   1196: getttywidth(void)
1.4       aaron    1197: {
                   1198:        struct winsize winsize;
                   1199:
                   1200:        if (ioctl(fileno(stdout), TIOCGWINSZ, &winsize) != -1)
1.18      markus   1201:                return (winsize.ws_col ? winsize.ws_col : 80);
1.4       aaron    1202:        else
1.18      markus   1203:                return (80);
1.4       aaron    1204: }