bin/dd/misc.c

   1 /*      $OpenBSD: misc.c,v 1.17 2013/04/16 22:13:43 deraadt Exp $       */
   2 /*      $NetBSD: misc.c,v 1.4 1995/03/21 09:04:10 cgd Exp $     */
   3
   4 /*-
   5  * Copyright (c) 1991, 1993, 1994
   6  *      The Regents of the University of California.  All rights reserved.
   7  *
   8  * This code is derived from software contributed to Berkeley by
   9  * Keith Muller of the University of California, San Diego and Lance
  10  * Visser of Convex Computer Corporation.
  11  *
  12  * Redistribution and use in source and binary forms, with or without
  13  * modification, are permitted provided that the following conditions
  14  * are met:
  15  * 1. Redistributions of source code must retain the above copyright
  16  *    notice, this list of conditions and the following disclaimer.
  17  * 2. Redistributions in binary form must reproduce the above copyright
  18  *    notice, this list of conditions and the following disclaimer in the
  19  *    documentation and/or other materials provided with the distribution.
  20  * 3. Neither the name of the University nor the names of its contributors
  21  *    may be used to endorse or promote products derived from this software
  22  *    without specific prior written permission.
  23  *
  24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  34  * SUCH DAMAGE.
  35  */
  36
  37 #include <sys/types.h>
  38 #include <sys/time.h>
  39 #include <sys/uio.h>
  40
  41 #include <err.h>
  42 #include <stdio.h>
  43 #include <stdlib.h>
  44 #include <string.h>
  45 #include <errno.h>
  46 #include <time.h>
  47 #include <unistd.h>
  48
  49 #include "dd.h"
  50 #include "extern.h"
  51
  52 void
  53 summary(void)
  54 {
  55         struct timeval nowtv;
  56         char buf[4][100];
  57         struct iovec iov[4];
  58         double microsecs;
  59         int i = 0;
  60
  61         (void)gettimeofday(&nowtv, (struct timezone *)NULL);
  62         timersub(&nowtv, &st.startv, &nowtv);
  63         microsecs = ((double)nowtv.tv_sec * 1000000) + nowtv.tv_usec;
  64         if (microsecs == 0)
  65                 microsecs = 1;
  66
  67         /* Use snprintf(3) so that we don't reenter stdio(3). */
  68         (void)snprintf(buf[0], sizeof(buf[0]),
  69             "%zu+%zu records in\n%zu+%zu records out\n",
  70             st.in_full, st.in_part, st.out_full, st.out_part);
  71         iov[i].iov_base = buf[0];
  72         iov[i++].iov_len = strlen(buf[0]);
  73
  74         if (st.swab) {
  75                 (void)snprintf(buf[1], sizeof(buf[1]),
  76                     "%zu odd length swab %s\n",
  77                      st.swab, (st.swab == 1) ? "block" : "blocks");
  78                 iov[i].iov_base = buf[1];
  79                 iov[i++].iov_len = strlen(buf[1]);
  80         }
  81         if (st.trunc) {
  82                 (void)snprintf(buf[2], sizeof(buf[2]),
  83                     "%zu truncated %s\n",
  84                      st.trunc, (st.trunc == 1) ? "block" : "blocks");
  85                 iov[i].iov_base = buf[2];
  86                 iov[i++].iov_len = strlen(buf[2]);
  87         }
  88         (void)snprintf(buf[3], sizeof(buf[3]),
  89             "%qd bytes transferred in %lld.%03ld secs (%0.0f bytes/sec)\n",
  90             (long long)st.bytes, (long long)nowtv.tv_sec, nowtv.tv_usec / 1000,
  91             ((double)st.bytes * 1000000) / microsecs);
  92
  93         iov[i].iov_base = buf[3];
  94         iov[i++].iov_len = strlen(buf[3]);
  95
  96         (void)writev(STDERR_FILENO, iov, i);
  97 }
  98
  99 /* ARGSUSED */
 100 void
 101 summaryx(int notused)
 102 {
 103         int save_errno = errno;
 104
 105         summary();
 106         errno = save_errno;
 107 }
 108
 109 /* ARGSUSED */
 110 void
 111 terminate(int notused)
 112 {
 113
 114         summary();
 115         _exit(0);
 116 }