xref: /openbsd/bin/dd/misc.c (revision 17df1aa7)
1 /*	$OpenBSD: misc.c,v 1.16 2009/10/27 23:59:21 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 %ld.%03ld secs (%0.0f bytes/sec)\n",
90 	    (long long)st.bytes, 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 }
117