xref: /openbsd/bin/dd/misc.c (revision ba080721)
1 /*	$OpenBSD: misc.c,v 1.26 2024/07/12 19:11:25 florian 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 
40 #include <errno.h>
41 #include <stdio.h>
42 #include <time.h>
43 #include <unistd.h>
44 
45 #include "dd.h"
46 #include "extern.h"
47 
48 /* SIGINFO handler */
49 void
sig_summary(int notused)50 sig_summary(int notused)
51 {
52 	int save_errno = errno;
53 	struct timespec elapsed, now;
54 	unsigned long long bps, msec;
55 
56 	if (ddflags & C_NOINFO)
57 		return;
58 
59 	clock_gettime(CLOCK_MONOTONIC, &now);
60 	timespecsub(&now, &st.start, &elapsed);
61 
62 	if (elapsed.tv_sec > 600)
63 		bps = st.bytes / elapsed.tv_sec;
64 	else if (elapsed.tv_sec > 0) {
65 		/* will overflow at ~ 30 exabytes / second */
66 		msec = elapsed.tv_sec * 1000 + elapsed.tv_nsec / 1000000;
67 		if (msec == 0)
68 			msec = 1;
69 		bps = st.bytes * 1000 / msec;
70 	} else if (elapsed.tv_nsec > 0)
71 		bps = st.bytes * 1000000000 / elapsed.tv_nsec;
72 	else
73 		bps = st.bytes;
74 
75 	/* Be async safe: use dprintf(3). */
76 	dprintf(STDERR_FILENO, "%zu+%zu records in\n%zu+%zu records out\n",
77 	    st.in_full, st.in_part, st.out_full, st.out_part);
78 
79 	if (st.swab) {
80 		dprintf(STDERR_FILENO, "%zu odd length swab %s\n",
81 		    st.swab, (st.swab == 1) ? "block" : "blocks");
82 	}
83 	if (st.trunc) {
84 		dprintf(STDERR_FILENO, "%zu truncated %s\n",
85 		    st.trunc, (st.trunc == 1) ? "block" : "blocks");
86 	}
87 	if (!(ddflags & C_NOXFER)) {
88 		dprintf(STDERR_FILENO,
89 		    "%lld bytes transferred in %lld.%03ld secs "
90 		    "(%llu bytes/sec)\n", (long long)st.bytes,
91 		    (long long)elapsed.tv_sec, elapsed.tv_nsec / 1000000, bps);
92 	}
93 	errno = save_errno;
94 }
95 
96 /* SIGINT handler */
97 void
sig_terminate(int signo)98 sig_terminate(int signo)
99 {
100 	sig_summary(0);
101 	_exit(128 + signo);
102 }
103 
104 /* atexit variation to summarize */
105 void
exit_summary(void)106 exit_summary(void)
107 {
108 	sig_summary(0);
109 }
110