xref: /openbsd-src/bin/dd/misc.c (revision f763167468dba5339ed4b14b7ecaca2a397ab0f6)
1 /*	$OpenBSD: misc.c,v 1.21 2017/08/13 02:06:42 tedu 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 timespec elapsed, now;
56 	char buf[4][100];
57 	struct iovec iov[4];
58 	double nanosecs;
59 	int i = 0;
60 
61 	if (ddflags & C_NOINFO)
62 		return;
63 
64 	clock_gettime(CLOCK_MONOTONIC, &now);
65 	timespecsub(&now, &st.start, &elapsed);
66 	nanosecs = ((double)elapsed.tv_sec * 1000000000) + elapsed.tv_nsec;
67 	if (nanosecs == 0)
68 		nanosecs = 1;
69 
70 	/* Use snprintf(3) so that we don't reenter stdio(3). */
71 	(void)snprintf(buf[0], sizeof(buf[0]),
72 	    "%zu+%zu records in\n%zu+%zu records out\n",
73 	    st.in_full, st.in_part, st.out_full, st.out_part);
74 	iov[i].iov_base = buf[0];
75 	iov[i++].iov_len = strlen(buf[0]);
76 
77 	if (st.swab) {
78 		(void)snprintf(buf[1], sizeof(buf[1]),
79 		    "%zu odd length swab %s\n",
80 		     st.swab, (st.swab == 1) ? "block" : "blocks");
81 		iov[i].iov_base = buf[1];
82 		iov[i++].iov_len = strlen(buf[1]);
83 	}
84 	if (st.trunc) {
85 		(void)snprintf(buf[2], sizeof(buf[2]),
86 		    "%zu truncated %s\n",
87 		     st.trunc, (st.trunc == 1) ? "block" : "blocks");
88 		iov[i].iov_base = buf[2];
89 		iov[i++].iov_len = strlen(buf[2]);
90 	}
91 	if (!(ddflags & C_NOXFER)) {
92 		(void)snprintf(buf[3], sizeof(buf[3]),
93 		    "%lld bytes transferred in %lld.%03ld secs "
94 		    "(%0.0f bytes/sec)\n", (long long)st.bytes,
95 		    (long long)elapsed.tv_sec, elapsed.tv_nsec / 1000000,
96 		    ((double)st.bytes * 1000000000) / nanosecs);
97 		iov[i].iov_base = buf[3];
98 		iov[i++].iov_len = strlen(buf[3]);
99 	}
100 
101 	(void)writev(STDERR_FILENO, iov, i);
102 }
103 
104 void
105 summaryx(int notused)
106 {
107 	int save_errno = errno;
108 
109 	summary();
110 	errno = save_errno;
111 }
112 
113 void
114 terminate(int notused)
115 {
116 
117 	summary();
118 	_exit(0);
119 }
120