dd: support conv=fsync

dd: support "dd --" if CONFIG_DESKTOP=y

function                                             old     new   delta
static.conv_words                                      -      28     +28
dd_main                                             1369    1373      +4
static.keywords                                      848     827     -21
------------------------------------------------------------------------------
(add/remove: 1/0 grow/shrink: 1/1 up/down: 32/-21)             Total: 11 bytes
This commit is contained in:
Denis Vlasenko 2008-04-02 21:20:35 +00:00
parent a7f4e4bbd8
commit 7b6d9d687d

View File

@ -79,23 +79,31 @@ static bool write_and_stats(const void *buf, size_t len, size_t obs,
#endif #endif
int dd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int dd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int dd_main(int argc, char **argv) int dd_main(int argc ATTRIBUTE_UNUSED, char **argv)
{ {
enum { enum {
FLAG_SYNC = 1 << 0, /* Must be in the same order as OP_conv_XXX! */
FLAG_NOERROR = 1 << 1, /* (see "flags |= (1 << what)" below) */
FLAG_NOTRUNC = 1 << 2, FLAG_NOTRUNC = 1 << 0,
FLAG_TWOBUFS = 1 << 3, FLAG_SYNC = 1 << 1,
FLAG_COUNT = 1 << 4, FLAG_NOERROR = 1 << 2,
FLAG_FSYNC = 1 << 3,
/* end of conv flags */
FLAG_TWOBUFS = 1 << 4,
FLAG_COUNT = 1 << 5,
}; };
static const char keywords[] ALIGN1 = static const char keywords[] ALIGN1 =
"bs=\0""count=\0""seek=\0""skip=\0""if=\0""of=\0" "bs=\0""count=\0""seek=\0""skip=\0""if=\0""of=\0"
#if ENABLE_FEATURE_DD_IBS_OBS #if ENABLE_FEATURE_DD_IBS_OBS
"ibs=\0""obs=\0""conv=\0""notrunc\0""sync\0""noerror\0" "ibs=\0""obs=\0""conv=\0"
#endif #endif
; ;
#if ENABLE_FEATURE_DD_IBS_OBS
static const char conv_words[] ALIGN1 =
"notrunc\0""sync\0""noerror\0""fsync\0";
#endif
enum { enum {
OP_bs = 1, OP_bs = 0,
OP_count, OP_count,
OP_seek, OP_seek,
OP_skip, OP_skip,
@ -105,9 +113,23 @@ int dd_main(int argc, char **argv)
OP_ibs, OP_ibs,
OP_obs, OP_obs,
OP_conv, OP_conv,
OP_conv_notrunc, /* Must be in the same order as FLAG_XXX! */
OP_conv_notrunc = 0,
OP_conv_sync, OP_conv_sync,
OP_conv_noerror, OP_conv_noerror,
OP_conv_fsync,
/* Unimplemented conv=XXX: */
//nocreat do not create the output file
//excl fail if the output file already exists
//fdatasync physically write output file data before finishing
//swab swap every pair of input bytes
//lcase change upper case to lower case
//ucase change lower case to upper case
//block pad newline-terminated records with spaces to cbs-size
//unblock replace trailing spaces in cbs-size records with newline
//ascii from EBCDIC to ASCII
//ebcdic from ASCII to EBCDIC
//ibm from ASCII to alternate EBCDIC
#endif #endif
}; };
int exitcode = EXIT_FAILURE; int exitcode = EXIT_FAILURE;
@ -138,85 +160,85 @@ int dd_main(int argc, char **argv)
signal_SA_RESTART_empty_mask(SIGUSR1, dd_output_status); signal_SA_RESTART_empty_mask(SIGUSR1, dd_output_status);
#endif #endif
for (n = 1; n < argc; n++) { for (n = 1; argv[n]; n++) {
smalluint key_len; int what;
smalluint what; char *val;
char *key;
char *arg = argv[n]; char *arg = argv[n];
//XXX:FIXME: we reject plain "dd --" This would cost ~20 bytes, so.. #if ENABLE_DESKTOP
//if (*arg == '-' && *++arg == '-' && !*++arg) continue; /* "dd --". NB: coreutils 6.9 will complain if they see
key = strstr(arg, "="); * more than one of them. We wouldn't. */
if (key == NULL) if (arg[0] == '-' && arg[1] == '-' && arg[2] == '\0')
continue;
#endif
val = strchr(arg, '=');
if (val == NULL)
bb_show_usage(); bb_show_usage();
key_len = key - arg + 1; *val = '\0';
key = xstrndup(arg, key_len); what = index_in_strings(keywords, arg);
what = index_in_strings(keywords, key) + 1; if (what < 0)
if (ENABLE_FEATURE_CLEAN_UP)
free(key);
if (what == 0)
bb_show_usage(); bb_show_usage();
arg += key_len; /* *val = '='; - to preserve ps listing? */
val++;
#if ENABLE_FEATURE_DD_IBS_OBS #if ENABLE_FEATURE_DD_IBS_OBS
if (what == OP_ibs) { if (what == OP_ibs) {
/* Must fit into positive ssize_t */ /* Must fit into positive ssize_t */
ibs = xatoul_range_sfx(arg, 1, ((size_t)-1L)/2, dd_suffixes); ibs = xatoul_range_sfx(val, 1, ((size_t)-1L)/2, dd_suffixes);
continue; /*continue;*/
} }
if (what == OP_obs) { if (what == OP_obs) {
obs = xatoul_range_sfx(arg, 1, ((size_t)-1L)/2, dd_suffixes); obs = xatoul_range_sfx(val, 1, ((size_t)-1L)/2, dd_suffixes);
continue; /*continue;*/
} }
if (what == OP_conv) { if (what == OP_conv) {
while (1) { while (1) {
/* find ',', replace them with NUL so we can use arg for /* find ',', replace them with NUL so we can use val for
* index_in_strings() without copying. * index_in_strings() without copying.
* We rely on arg being non-null, else strchr would fault. * We rely on val being non-null, else strchr would fault.
*/ */
key = strchr(arg, ','); arg = strchr(val, ',');
if (key) if (arg)
*key = '\0'; *arg = '\0';
what = index_in_strings(keywords, arg) + 1; what = index_in_strings(conv_words, val);
if (what < OP_conv_notrunc) if (what < 0)
bb_error_msg_and_die(bb_msg_invalid_arg, arg, "conv"); bb_error_msg_and_die(bb_msg_invalid_arg, val, "conv");
if (what == OP_conv_notrunc) flags |= (1 << what);
flags |= FLAG_NOTRUNC; if (!arg) /* no ',' left, so this was the last specifier */
if (what == OP_conv_sync)
flags |= FLAG_SYNC;
if (what == OP_conv_noerror)
flags |= FLAG_NOERROR;
if (!key) /* no ',' left, so this was the last specifier */
break; break;
arg = key + 1; /* skip this keyword and ',' */ /* *arg = ','; - to preserve ps listing? */
val = arg + 1; /* skip this keyword and ',' */
} }
continue; continue; /* we trashed 'what', can't fall through */
} }
#endif #endif
if (what == OP_bs) { if (what == OP_bs) {
ibs = obs = xatoul_range_sfx(arg, 1, ((size_t)-1L)/2, dd_suffixes); ibs = obs = xatoul_range_sfx(val, 1, ((size_t)-1L)/2, dd_suffixes);
continue; /*continue;*/
} }
/* These can be large: */ /* These can be large: */
if (what == OP_count) { if (what == OP_count) {
flags |= FLAG_COUNT; flags |= FLAG_COUNT;
count = XATOU_SFX(arg, dd_suffixes); count = XATOU_SFX(val, dd_suffixes);
continue; /*continue;*/
} }
if (what == OP_seek) { if (what == OP_seek) {
seek = XATOU_SFX(arg, dd_suffixes); seek = XATOU_SFX(val, dd_suffixes);
continue; /*continue;*/
} }
if (what == OP_skip) { if (what == OP_skip) {
skip = XATOU_SFX(arg, dd_suffixes); skip = XATOU_SFX(val, dd_suffixes);
continue; /*continue;*/
} }
if (what == OP_if) { if (what == OP_if) {
infile = arg; infile = val;
continue; /*continue;*/
} }
if (what == OP_of) if (what == OP_of) {
outfile = arg; outfile = val;
/*continue;*/
} }
} /* end of "for (argv[n])" */
//XXX:FIXME for huge ibs or obs, malloc'ing them isn't the brightest idea ever //XXX:FIXME for huge ibs or obs, malloc'ing them isn't the brightest idea ever
ibuf = obuf = xmalloc(ibs); ibuf = obuf = xmalloc(ibs);
if (ibs != obs) { if (ibs != obs) {
@ -304,13 +326,17 @@ int dd_main(int argc, char **argv)
} }
} else if (write_and_stats(ibuf, n, obs, outfile)) } else if (write_and_stats(ibuf, n, obs, outfile))
goto out_status; goto out_status;
if (flags & FLAG_FSYNC) {
if (fsync(ofd) < 0)
goto die_outfile;
}
} }
if (ENABLE_FEATURE_DD_IBS_OBS && oc) { if (ENABLE_FEATURE_DD_IBS_OBS && oc) {
w = full_write_or_warn(obuf, oc, outfile); w = full_write_or_warn(obuf, oc, outfile);
if (w < 0) goto out_status; if (w < 0) goto out_status;
if (w > 0) if (w > 0) G.out_part++;
G.out_part++;
} }
if (close(ifd) < 0) { if (close(ifd) < 0) {
die_infile: die_infile: