1 /* vi: set sw=4 ts=4: */
3 * Mini dd implementation for busybox
6 * Copyright (C) 2000,2001 Matt Kraai
8 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
11 #include <signal.h> /* For FEATURE_DD_SIGNAL_HANDLING */
14 /* This is a NOEXEC applet. Be very careful! */
17 static const struct suffix_mult dd_suffixes[] = {
23 { "K", 1024 }, /* compat with coreutils dd */
32 off_t out_full, out_part, in_full, in_part;
34 #define G (*(struct globals*)&bb_common_bufsiz1)
36 static void dd_output_status(int ATTRIBUTE_UNUSED cur_signal)
38 fprintf(stderr, "%"OFF_FMT"d+%"OFF_FMT"d records in\n"
39 "%"OFF_FMT"d+%"OFF_FMT"d records out\n",
41 G.out_full, G.out_part);
44 static ssize_t full_write_or_warn(int fd, const void *buf, size_t len,
45 const char * const filename)
47 ssize_t n = full_write(fd, buf, len);
49 bb_perror_msg("writing '%s'", filename);
53 static bool write_and_stats(int fd, const void *buf, size_t len, size_t obs,
56 ssize_t n = full_write_or_warn(fd, buf, len, filename);
67 #define XATOU_SFX xatoull_sfx
69 #define XATOU_SFX xatoul_sfx
72 int dd_main(int argc, char **argv);
73 int dd_main(int argc, char **argv)
79 TWOBUFS_FLAG = 1 << 3,
81 static const char keywords[] =
82 "bs=\0""count=\0""seek=\0""skip=\0""if=\0""of=\0"
83 #if ENABLE_FEATURE_DD_IBS_OBS
84 "ibs=\0""obs=\0""conv=\0""notrunc\0""sync\0""noerror\0"
94 #if ENABLE_FEATURE_DD_IBS_OBS
103 int flags = TRUNC_FLAG;
104 size_t oc = 0, ibs = 512, obs = 512;
106 off_t seek = 0, skip = 0, count = OFF_T_MAX;
108 const char *infile = NULL, *outfile = NULL;
111 memset(&G, 0, sizeof(G)); /* because of NOEXEC */
113 if (ENABLE_FEATURE_DD_SIGNAL_HANDLING) {
116 memset(&sa, 0, sizeof(sa));
117 sa.sa_handler = dd_output_status;
118 sa.sa_flags = SA_RESTART;
119 sigemptyset(&sa.sa_mask);
120 sigaction(SIGUSR1, &sa, 0);
123 for (n = 1; n < argc; n++) {
129 //XXX:FIXME: we reject plain "dd --" This would cost ~20 bytes, so..
130 //if (*arg == '-' && *++arg == '-' && !*++arg) continue;
131 key = strstr(arg, "=");
134 key_len = key - arg + 1;
135 key = xstrndup(arg, key_len);
136 what = index_in_strings(keywords, key) + 1;
137 if (ENABLE_FEATURE_CLEAN_UP)
142 /* Must fit into positive ssize_t */
143 #if ENABLE_FEATURE_DD_IBS_OBS
144 if (what == OP_ibs) {
145 ibs = xatoul_range_sfx(arg, 1, ((size_t)-1L)/2, dd_suffixes);
148 if (what == OP_obs) {
149 obs = xatoul_range_sfx(arg, 1, ((size_t)-1L)/2, dd_suffixes);
152 if (what == OP_conv) {
154 /* find ',', replace them with nil so we can use arg for
155 * index_in_strings() without copying.
156 * We rely on arg being non-null, else strchr would fault.
158 key = strchr(arg, ',');
161 what = index_in_strings(keywords, arg) + 1;
162 if (what < OP_conv_notrunc)
163 bb_error_msg_and_die(bb_msg_invalid_arg, arg, "conv");
164 if (what == OP_conv_notrunc)
165 flags &= ~TRUNC_FLAG;
166 if (what == OP_conv_sync)
168 if (what == OP_conv_noerror)
170 if (!key) /* no ',' left, so this was the last specifier */
172 arg = key + 1; /* skip this keyword and ',' */
178 ibs = obs = xatoul_range_sfx(arg, 1, ((size_t)-1L)/2, dd_suffixes);
181 /* These can be large: */
182 if (what == OP_count) {
183 count = XATOU_SFX(arg, dd_suffixes);
186 if (what == OP_seek) {
187 seek = XATOU_SFX(arg, dd_suffixes);
190 if (what == OP_skip) {
191 skip = XATOU_SFX(arg, dd_suffixes);
201 //XXX:FIXME for huge ibs or obs, malloc'ing them isn't the brightest idea ever
202 ibuf = obuf = xmalloc(ibs);
204 flags |= TWOBUFS_FLAG;
208 ifd = xopen(infile, O_RDONLY);
211 infile = bb_msg_standard_input;
213 if (outfile != NULL) {
214 int oflag = O_WRONLY | O_CREAT;
216 if (!seek && (flags & TRUNC_FLAG))
219 ofd = xopen(outfile, oflag);
221 if (seek && (flags & TRUNC_FLAG)) {
222 if (ftruncate(ofd, seek * obs) < 0) {
225 if (fstat(ofd, &st) < 0 || S_ISREG(st.st_mode) ||
232 outfile = bb_msg_standard_output;
235 if (lseek(ifd, skip * ibs, SEEK_CUR) < 0) {
237 n = safe_read(ifd, ibuf, ibs);
246 if (lseek(ofd, seek * obs, SEEK_CUR) < 0)
250 while (G.in_full + G.in_part != count) {
251 if (flags & NOERROR) /* Pre-zero the buffer when for NOERROR */
252 memset(ibuf, '\0', ibs);
253 n = safe_read(ifd, ibuf, ibs);
257 if (flags & NOERROR) {
259 bb_perror_msg("%s", infile);
263 if ((size_t)n == ibs)
267 if (flags & SYNC_FLAG) {
268 memset(ibuf + n, '\0', ibs - n);
272 if (flags & TWOBUFS_FLAG) {
279 memcpy(obuf + oc, tmp, d);
284 if (write_and_stats(ofd, obuf, obs, obs, outfile))
289 } else if (write_and_stats(ofd, ibuf, n, obs, outfile))
293 if (ENABLE_FEATURE_DD_IBS_OBS && oc) {
294 w = full_write_or_warn(ofd, obuf, oc, outfile);
295 if (w < 0) goto out_status;
299 if (close(ifd) < 0) {
301 bb_perror_msg_and_die("%s", infile);
304 if (close(ofd) < 0) {
306 bb_perror_msg_and_die("%s", outfile);