summaryrefslogtreecommitdiff
path: root/coreutils/dd.c
diff options
context:
space:
mode:
Diffstat (limited to 'coreutils/dd.c')
-rw-r--r--coreutils/dd.c33
1 files changed, 25 insertions, 8 deletions
diff --git a/coreutils/dd.c b/coreutils/dd.c
index ce8bcc6..378e212 100644
--- a/coreutils/dd.c
+++ b/coreutils/dd.c
@@ -15,9 +15,9 @@
#include <unistd.h>
#include <string.h>
#include <fcntl.h>
+#include <signal.h> // For FEATURE_DD_SIGNAL_HANDLING
#include "busybox.h"
-
static const struct suffix_mult dd_suffixes[] = {
{ "c", 1 },
{ "w", 2 },
@@ -31,12 +31,20 @@ static const struct suffix_mult dd_suffixes[] = {
{ NULL, 0 }
};
+static size_t out_full;
+static size_t out_part;
+static size_t in_full;
+static size_t in_part;
+
+static void dd_output_status(int cur_signal)
+{
+ fprintf(stderr, "%ld+%ld records in\n%ld+%ld records out\n",
+ (long)in_full, (long)in_part,
+ (long)out_full, (long)out_part);
+}
+
int dd_main(int argc, char **argv)
{
- size_t out_full = 0;
- size_t out_part = 0;
- size_t in_full = 0;
- size_t in_part = 0;
size_t count = -1;
size_t bs = 512;
ssize_t n;
@@ -53,6 +61,17 @@ int dd_main(int argc, char **argv)
const char *outfile = NULL;
char *buf;
+ if (ENABLE_FEATURE_DD_SIGNAL_HANDLING)
+ {
+ struct sigaction sa;
+
+ memset(&sa, 0, sizeof(sa));
+ sa.sa_handler = dd_output_status;
+ sa.sa_flags = SA_RESTART;
+ sigemptyset(&sa.sa_mask);
+ sigaction(SIGUSR1, &sa, 0);
+ }
+
for (i = 1; i < argc; i++) {
if (strncmp("bs=", argv[i], 3) == 0)
bs = bb_xparse_number(argv[i]+3, dd_suffixes);
@@ -180,9 +199,7 @@ int dd_main(int argc, char **argv)
bb_perror_msg_and_die("%s", outfile);
}
- fprintf(stderr, "%ld+%ld records in\n%ld+%ld records out\n",
- (long)in_full, (long)in_part,
- (long)out_full, (long)out_part);
+ dd_output_status(0);
return EXIT_SUCCESS;
}