/* dd -- convert a file while copying it. Copyright (C) 85, 90, 91, 95, 96, 97, 1998 Free Software Foundation, Inc. This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ /* Written by Paul Rubin, David MacKenzie, and Stuart Kemp. */ #include #include #define SWAB_ALIGN_OFFSET 2 #if HAVE_INTTYPES_H # include #endif #include #include #include #include "human.h" #include "system.h" #include "closeout.h" #include "error.h" #include "safe-read.h" #ifndef SIGINFO # define SIGINFO SIGUSR1 #endif #define max(a, b) ((a) > (b) ? (a) : (b)) #define output_char(c) \ do { \ obuf[oc++] = (c); if (oc >= output_blocksize) write_output (); \ } while (0) /* Default input and output blocksize. */ #define DEFAULT_BLOCKSIZE 512 /* Conversions bit masks. */ #define C_ASCII 01 #define C_EBCDIC 02 #define C_IBM 04 #define C_BLOCK 010 #define C_UNBLOCK 020 #define C_LCASE 040 #define C_UCASE 0100 #define C_SWAB 0200 #define C_NOERROR 0400 #define C_NOTRUNC 01000 #define C_SYNC 02000 /* Use separate input and output buffers, and combine partial input blocks. */ #define C_TWOBUFS 04000 int full_write (); /* The name this program was run with. */ char *program_name; /* The name of the input file, or NULL for the standard input. */ static char *input_file = NULL; /* The input file descriptor. */ static int input_fd = 0; /* The name of the output file, or NULL for the standard output. */ static char *output_file = NULL; /* The output file descriptor. */ static int output_fd = 1; /* The number of bytes in which atomic reads are done. */ static size_t input_blocksize = 0; /* The number of bytes in which atomic writes are done. */ static size_t output_blocksize = 0; /* Conversion buffer size, in bytes. 0 prevents conversions. */ static size_t conversion_blocksize = 0; /* Skip this many records of `input_blocksize' bytes before input. */ static uintmax_t skip_records = 0; /* Skip this many records of `output_blocksize' bytes before output. */ static uintmax_t seek_record = 0; /* Copy only this many records. The default is effectively infinity. */ static uintmax_t max_records = (uintmax_t) -1; /* Bit vector of conversions to apply. */ static int conversions_mask = 0; /* If nonzero, filter characters through the translation table. */ static int translation_needed = 0; /* Number of partial blocks written. */ static uintmax_t w_partial = 0; /* Number of full blocks written. */ static uintmax_t w_full = 0; /* Number of partial blocks read. */ static uintmax_t r_partial = 0; /* Number of full blocks read. */ static uintmax_t r_full = 0; /* Records truncated by conv=block. */ static uintmax_t r_truncate = 0; /* Output representation of newline and space characters. They change if we're converting to EBCDIC. */ static unsigned char newline_character = '\n'; static unsigned char space_character = ' '; /* Output buffer. */ static unsigned char *obuf; /* Current index into `obuf'. */ static size_t oc = 0; /* Index into current line, for `conv=block' and `conv=unblock'. */ static size_t col = 0; struct conversion { char *convname; int conversion; }; static struct conversion conversions[] = { {"ascii", C_ASCII | C_TWOBUFS}, /* EBCDIC to ASCII. */ {"ebcdic", C_EBCDIC | C_TWOBUFS}, /* ASCII to EBCDIC. */ {"ibm", C_IBM | C_TWOBUFS}, /* Slightly different ASCII to EBCDIC. */ {"block", C_BLOCK | C_TWOBUFS}, /* Variable to fixed length records. */ {"unblock", C_UNBLOCK | C_TWOBUFS}, /* Fixed to variable length records. */ {"lcase", C_LCASE | C_TWOBUFS}, /* Translate upper to lower case. */ {"ucase", C_UCASE | C_TWOBUFS}, /* Translate lower to upper case. */ {"swab", C_SWAB | C_TWOBUFS}, /* Swap bytes of input. */ {"noerror", C_NOERROR}, /* Ignore i/o errors. */ {"notrunc", C_NOTRUNC}, /* Do not truncate output file. */ {"sync", C_SYNC}, /* Pad input records to ibs with NULs. */ {NULL, 0} }; /* Translation table formed by applying successive transformations. */ static unsigned char trans_table[256]; static unsigned char const ascii_to_ebcdic[] = { 0, 01, 02, 03, 067, 055, 056, 057, 026, 05, 045, 013, 014, 015, 016, 017, 020, 021, 022, 023, 074, 075, 062, 046, 030, 031, 077, 047, 034, 035, 036, 037, 0100, 0117, 0177, 0173, 0133, 0154, 0120, 0175, 0115, 0135, 0134, 0116, 0153, 0140, 0113, 0141, 0360, 0361, 0362, 0363, 0364, 0365, 0366, 0367, 0370, 0371, 0172, 0136, 0114, 0176, 0156, 0157, 0174, 0301, 0302, 0303, 0304, 0305, 0306, 0307, 0310, 0311, 0321, 0322, 0323, 0324, 0325, 0326, 0327, 0330, 0331, 0342, 0343, 0344, 0345, 0346, 0347, 0350, 0351, 0112, 0340, 0132, 0137, 0155, 0171, 0201, 0202, 0203, 0204, 0205, 0206, 0207, 0210, 0211, 0221, 0222, 0223, 0224, 0225, 0226, 0227, 0230, 0231, 0242, 0243, 0244, 0245, 0246, 0247, 0250, 0251, 0300, 0152, 0320, 0241, 07, 040, 041, 042, 043, 044, 025, 06, 027, 050, 051, 052, 053, 054, 011, 012, 033, 060, 061, 032, 063, 064, 065, 066, 010, 070, 071, 072, 073, 04, 024, 076, 0341, 0101, 0102, 0103, 0104, 0105, 0106, 0107, 0110, 0111, 0121, 0122, 0123, 0124, 0125, 0126, 0127, 0130, 0131, 0142, 0143, 0144, 0145, 0146, 0147, 0150, 0151, 0160, 0161, 0162, 0163, 0164, 0165, 0166, 0167, 0170, 0200, 0212, 0213, 0214, 0215, 0216, 0217, 0220, 0232, 0233, 0234, 0235, 0236, 0237, 0240, 0252, 0253, 0254, 0255, 0256, 0257, 0260, 0261, 0262, 0263, 0264, 0265, 0266, 0267, 0270, 0271, 0272, 0273, 0274, 0275, 0276, 0277, 0312, 0313, 0314, 0315, 0316, 0317, 0332, 0333, 0334, 0335, 0336, 0337, 0352, 0353, 0354, 0355, 0356, 0357, 0372, 0373, 0374, 0375, 0376, 0377 }; static unsigned char const ascii_to_ibm[] = { 0, 01, 02, 03, 067, 055, 056, 057, 026, 05, 045, 013, 014, 015, 016, 017, 020, 021, 022, 023, 074, 075, 062, 046, 030, 031, 077, 047, 034, 035, 036, 037, 0100, 0132, 0177, 0173, 0133, 0154, 0120, 0175, 0115, 0135, 0134, 0116, 0153, 0140, 0113, 0141, 0360, 0361, 0362, 0363, 0364, 0365, 0366, 0367, 0370, 0371, 0172, 0136, 0114, 0176, 0156, 0157, 0174, 0301, 0302, 0303, 0304, 0305, 0306, 0307, 0310, 0311, 0321, 0322, 0323, 0324, 0325, 0326, 0327, 0330, 0331, 0342, 0343, 0344, 0345, 0346, 0347, 0350, 0351, 0255, 0340, 0275, 0137, 0155, 0171, 0201, 0202, 0203, 0204, 0205, 0206, 0207, 0210, 0211, 0221, 0222, 0223, 0224, 0225, 0226, 0227, 0230, 0231, 0242, 0243, 0244, 0245, 0246, 0247, 0250, 0251, 0300, 0117, 0320, 0241, 07, 040, 041, 042, 043, 044, 025, 06, 027, 050, 051, 052, 053, 054, 011, 012, 033, 060, 061, 032, 063, 064, 065, 066, 010, 070, 071, 072, 073, 04, 024, 076, 0341, 0101, 0102, 0103, 0104, 0105, 0106, 0107, 0110, 0111, 0121, 0122, 0123, 0124, 0125, 0126, 0127, 0130, 0131, 0142, 0143, 0144, 0145, 0146, 0147, 0150, 0151, 0160, 0161, 0162, 0163, 0164, 0165, 0166, 0167, 0170, 0200, 0212, 0213, 0214, 0215, 0216, 0217, 0220, 0232, 0233, 0234, 0235, 0236, 0237, 0240, 0252, 0253, 0254, 0255, 0256, 0257, 0260, 0261, 0262, 0263, 0264, 0265, 0266, 0267, 0270, 0271, 0272, 0273, 0274, 0275, 0276, 0277, 0312, 0313, 0314, 0315, 0316, 0317, 0332, 0333, 0334, 0335, 0336, 0337, 0352, 0353, 0354, 0355, 0356, 0357, 0372, 0373, 0374, 0375, 0376, 0377 }; static unsigned char const ebcdic_to_ascii[] = { 0, 01, 02, 03, 0234, 011, 0206, 0177, 0227, 0215, 0216, 013, 014, 015, 016, 017, 020, 021, 022, 023, 0235, 0205, 010, 0207, 030, 031, 0222, 0217, 034, 035, 036, 037, 0200, 0201, 0202, 0203, 0204, 012, 027, 033, 0210, 0211, 0212, 0213, 0214, 05, 06, 07, 0220, 0221, 026, 0223, 0224, 0225, 0226, 04, 0230, 0231, 0232, 0233, 024, 025, 0236, 032, 040, 0240, 0241, 0242, 0243, 0244, 0245, 0246, 0247, 0250, 0133, 056, 074, 050, 053, 041, 046, 0251, 0252, 0253, 0254, 0255, 0256, 0257, 0260, 0261, 0135, 044, 052, 051, 073, 0136, 055, 057, 0262, 0263, 0264, 0265, 0266, 0267, 0270, 0271, 0174, 054, 045, 0137, 076, 077, 0272, 0273, 0274, 0275, 0276, 0277, 0300, 0301, 0302, 0140, 072, 043, 0100, 047, 075, 042, 0303, 0141, 0142, 0143, 0144, 0145, 0146, 0147, 0150, 0151, 0304, 0305, 0306, 0307, 0310, 0311, 0312, 0152, 0153, 0154, 0155, 0156, 0157, 0160, 0161, 0162, 0313, 0314, 0315, 0316, 0317, 0320, 0321, 0176, 0163, 0164, 0165, 0166, 0167, 0170, 0171, 0172, 0322, 0323, 0324, 0325, 0326, 0327, 0330, 0331, 0332, 0333, 0334, 0335, 0336, 0337, 0340, 0341, 0342, 0343, 0344, 0345, 0346, 0347, 0173, 0101, 0102, 0103, 0104, 0105, 0106, 0107, 0110, 0111, 0350, 0351, 0352, 0353, 0354, 0355, 0175, 0112, 0113, 0114, 0115, 0116, 0117, 0120, 0121, 0122, 0356, 0357, 0360, 0361, 0362, 0363, 0134, 0237, 0123, 0124, 0125, 0126, 0127, 0130, 0131, 0132, 0364, 0365, 0366, 0367, 0370, 0371, 060, 061, 062, 063, 064, 065, 066, 067, 070, 071, 0372, 0373, 0374, 0375, 0376, 0377 }; /* If nonzero, display usage information and exit. */ static int show_help; /* If nonzero, print the version on standard output and exit. */ static int show_version; static struct option const long_options[] = { {"help", no_argument, &show_help, 1}, {"version", no_argument, &show_version, 1}, {0, 0, 0, 0} }; static void usage (int status) { if (status != 0) fprintf (stderr, _("Try `%s --help' for more information.\n"), program_name); else { printf (_("Usage: %s [OPTION]...\n"), program_name); printf (_("\ Copy a file, converting and formatting according to the options.\n\ \n\ bs=BYTES force ibs=BYTES and obs=BYTES\n\ cbs=BYTES convert BYTES bytes at a time\n\ conv=KEYWORDS convert the file as per the comma separated keyword list\n\ count=BLOCKS copy only BLOCKS input blocks\n\ ibs=BYTES read BYTES bytes at a time\n\ if=FILE read from FILE instead of stdin\n\ obs=BYTES write BYTES bytes at a time\n\ of=FILE write to FILE instead of stdout\n\ seek=BLOCKS skip BLOCKS obs-sized blocks at start of output\n\ skip=BLOCKS skip BLOCKS ibs-sized blocks at start of input\n\ --help display this help and exit\n\ --version output version information and exit\n\ \n\ BYTES may be followed by the following multiplicative suffixes:\n\ xM M, c 1, w 2, b 512, kD 1000, k 1024, MD 1,000,000, M 1,048,576,\n\ GD 1,000,000,000, G 1,073,741,824, and so on for T, P, E, Z, Y.\n\ Each KEYWORD may be:\n\ \n\ ascii from EBCDIC to ASCII\n\ ebcdic from ASCII to EBCDIC\n\ ibm from ASCII to alternated EBCDIC\n\ block pad newline-terminated records with spaces to cbs-size\n\ unblock replace trailing spaces in cbs-size records with newline\n\ lcase change upper case to lower case\n\ notrunc do not truncate the output file\n\ ucase change lower case to upper case\n\ swab swap every pair of input bytes\n\ noerror continue after read errors\n\ sync pad every input block with NULs to ibs-size\n\ ")); puts (_("\nReport bugs to .")); close_stdout (); } exit (status); } static void translate_charset (const unsigned char *new_trans) { int i; for (i = 0; i < 256; i++) trans_table[i] = new_trans[trans_table[i]]; translation_needed = 1; } /* Return the number of 1 bits in `i'. */ static int bit_count (register unsigned int i) { register int set_bits; for (set_bits = 0; i != 0; set_bits++) i &= i - 1; return set_bits; } static void print_stats (void) { char buf[2][LONGEST_HUMAN_READABLE + 1]; fprintf (stderr, _("%s+%s records in\n"), human_readable (r_full, buf[0], 1, 1), human_readable (r_partial, buf[1], 1, 1)); fprintf (stderr, _("%s+%s records out\n"), human_readable (w_full, buf[0], 1, 1), human_readable (w_partial, buf[1], 1, 1)); if (r_truncate > 0) { fprintf (stderr, "%s %s\n", human_readable (r_truncate, buf[0], 1, 1), (r_truncate == 1 ? _("truncated record") : _("truncated records"))); } } static void cleanup (void) { print_stats (); if (close (input_fd) < 0) error (1, errno, "%s", input_file); if (close (output_fd) < 0) error (1, errno, "%s", output_file); } static inline void quit (int code) { cleanup (); exit (code); } static RETSIGTYPE interrupt_handler (int sig) { #ifdef SA_INTERRUPT struct sigaction sigact; sigact.sa_handler = SIG_DFL; sigemptyset (&sigact.sa_mask); sigact.sa_flags = 0; sigaction (sig, &sigact, NULL); #else /* !SA_INTERRUPT */ signal (sig, SIG_DFL); #endif /* SA_INTERRUPT */ cleanup (); kill (getpid (), sig); } static RETSIGTYPE siginfo_handler (int sig) { print_stats (); } /* Encapsulate portability mess of establishing signal handlers. */ static void install_handler (int sig_num, RETSIGTYPE (*sig_handler) (int sig)) { #ifdef _POSIX_VERSION struct sigaction sigact; sigaction (sig_num, NULL, &sigact); if (sigact.sa_handler != SIG_IGN) { sigact.sa_handler = sig_handler; sigemptyset (&sigact.sa_mask); sigact.sa_flags = 0; sigaction (sig_num, &sigact, NULL); } #else if (signal (sig_num, SIG_IGN) != SIG_IGN) signal (sig_num, sig_handler); #endif } /* Write, then empty, the output buffer `obuf'. */ static void write_output (void) { int nwritten = full_write (output_fd, obuf, output_blocksize); if (nwritten != output_blocksize) { error (0, errno, "%s", output_file); if (nwritten > 0) w_partial++; quit (1); } else w_full++; oc = 0; } /* Interpret one "conv=..." option. */ static void parse_conversion (char *str) { char *new; int i; do { new = strchr (str, ','); if (new != NULL) *new++ = '\0'; for (i = 0; conversions[i].convname != NULL; i++) if (STREQ (conversions[i].convname, str)) { conversions_mask |= conversions[i].conversion; break; } if (conversions[i].convname == NULL) { error (0, 0, _("%s: invalid conversion"), str); usage (1); } str = new; } while (new != NULL); } /* Return the value of STR, interpreted as a non-negative decimal integer, optionally multiplied by various values. Assign nonzero to *INVALID if STR does not represent a number in this format. */ /* FIXME: use xstrtou?[lq] */ static uintmax_t parse_integer (char *str, int *invalid) { register uintmax_t n = 0; register char *p = str; while (ISDIGIT (*p)) { uintmax_t n10 = n * 10; int digit = *p - '0'; if (! (n10 / 10 == n && n10 <= n10 + digit)) { *invalid = 1; return 0; } n = n10 + digit; p++; } for (;;) { uintmax_t multiplier; int power = 0; #ifdef lint /* Suppress `used before initialized' warning. */ multiplier = 0; #endif switch (*p++) { case '\0': return n; case 'b': multiplier = 512; break; case 'c': continue; case 'E': /* Exa */ power = 6; break; case 'G': /* Giga */ power = 3; break; case 'k': /* kilo */ power = 1; break; case 'M': /* Mega */ power = 2; break; case 'P': /* Peta */ power = 5; break; case 'T': /* Tera */ power = 4; break; case 'w': multiplier = 2; break; case 'x': multiplier = parse_integer (p, invalid); p = ""; break; case 'Y': /* Yotta */ power = 8; break; case 'Z': /* Zetta */ power = 7; break; default: { *invalid = 1; return 0; } } if (power) { int base = 1024; switch (*p) { case 'B': p++; break; case 'D': p++; base = 1000; break; } for (multiplier = base; --power; multiplier *= base) if (multiplier * base / base != multiplier) { *invalid = 1; return 0; } } if (multiplier != 0 && n * multiplier / multiplier != n) { *invalid = 1; return 0; } n *= multiplier; } } static void scanargs (int argc, char **argv) { int i; int c; while ((c = getopt_long (argc, argv, "", long_options, NULL)) != -1) { switch (c) { case 0: break; default: usage (1); } } for (i = optind; i < argc; i++) { char *name, *val; name = argv[i]; val = strchr (name, '='); if (val == NULL) { error (0, 0, _("unrecognized option `%s'"), name); usage (1); } *val++ = '\0'; if (STREQ (name, "if")) input_file = val; else if (STREQ (name, "of")) output_file = val; else if (STREQ (name, "conv")) parse_conversion (val); else { int invalid = 0; uintmax_t n = parse_integer (val, &invalid); if (STREQ (name, "ibs")) { input_blocksize = n; invalid |= input_blocksize != n || input_blocksize == 0; conversions_mask |= C_TWOBUFS; } else if (STREQ (name, "obs")) { output_blocksize = n; invalid |= output_blocksize != n || output_blocksize == 0; conversions_mask |= C_TWOBUFS; } else if (STREQ (name, "bs")) { output_blocksize = input_blocksize = n; invalid |= output_blocksize != n || output_blocksize == 0; } else if (STREQ (name, "cbs")) { conversion_blocksize = n; invalid |= (conversion_blocksize != n || conversion_blocksize == 0); } else if (STREQ (name, "skip")) skip_records = n; else if (STREQ (name, "seek")) seek_record = n; else if (STREQ (name, "count")) max_records = n; else { error (0, 0, _("unrecognized option `%s=%s'"), name, val); usage (1); } if (invalid) error (1, 0, _("invalid number `%s'"), val); } } /* If bs= was given, both `input_blocksize' and `output_blocksize' will have been set to positive values. If either has not been set, bs= was not given, so make sure two buffers are used. */ if (input_blocksize == 0 || output_blocksize == 0) conversions_mask |= C_TWOBUFS; if (input_blocksize == 0) input_blocksize = DEFAULT_BLOCKSIZE; if (output_blocksize == 0) output_blocksize = DEFAULT_BLOCKSIZE; if (conversion_blocksize == 0) conversions_mask &= ~(C_BLOCK | C_UNBLOCK); } /* Fix up translation table. */ static void apply_translations (void) { int i; #define MX(a) (bit_count (conversions_mask & (a))) if ((MX (C_ASCII | C_EBCDIC | C_IBM) > 1) || (MX (C_BLOCK | C_UNBLOCK) > 1) || (MX (C_LCASE | C_UCASE) > 1) || (MX (C_UNBLOCK | C_SYNC) > 1)) { error (1, 0, _("\ only one conv in {ascii,ebcdic,ibm}, {lcase,ucase}, {block,unblock}, {unblock,sync}")); } #undef MX if (conversions_mask & C_ASCII) translate_charset (ebcdic_to_ascii); if (conversions_mask & C_UCASE) { for (i = 0; i < 256; i++) if (ISLOWER (trans_table[i])) trans_table[i] = toupper (trans_table[i]); translation_needed = 1; } else if (conversions_mask & C_LCASE) { for (i = 0; i < 256; i++) if (ISUPPER (trans_table[i])) trans_table[i] = tolower (trans_table[i]); translation_needed = 1; } if (conversions_mask & C_EBCDIC) { translate_charset (ascii_to_ebcdic); newline_character = ascii_to_ebcdic['\n']; space_character = ascii_to_ebcdic[' ']; } else if (conversions_mask & C_IBM) { translate_charset (ascii_to_ibm); newline_character = ascii_to_ibm['\n']; space_character = ascii_to_ibm[' ']; } } /* Apply the character-set translations specified by the user to the NREAD bytes in BUF. */ static void translate_buffer (unsigned char *buf, int nread) { register unsigned char *cp; register int i; for (i = nread, cp = buf; i; i--, cp++) *cp = trans_table[*cp]; } /* If nonnzero, the last char from the previous call to `swab_buffer' is saved in `saved_char'. */ static int char_is_saved = 0; /* Odd char from previous call. */ static unsigned char saved_char; /* Swap NREAD bytes in BUF, plus possibly an initial char from the previous call. If NREAD is odd, save the last char for the next call. Return the new start of the BUF buffer. */ static unsigned char * swab_buffer (unsigned char *buf, int *nread) { unsigned char *bufstart = buf; register unsigned char *cp; register int i; /* Is a char left from last time? */ if (char_is_saved) { *--bufstart = saved_char; (*nread)++; char_is_saved = 0; } if (*nread & 1) { /* An odd number of chars are in the buffer. */ saved_char = bufstart[--*nread]; char_is_saved = 1; } /* Do the byte-swapping by moving every second character two positions toward the end, working from the end of the buffer toward the beginning. This way we only move half of the data. */ cp = bufstart + *nread; /* Start one char past the last. */ for (i = *nread / 2; i; i--, cp -= 2) *cp = *(cp - 2); return ++bufstart; } /* Throw away RECORDS blocks of BLOCKSIZE bytes on file descriptor FDESC, which is open with read permission for FILE. Store up to BLOCKSIZE bytes of the data at a time in BUF, if necessary. */ static void skip (int fdesc, char *file, uintmax_t records, size_t blocksize, unsigned char *buf) { struct stat stats; off_t o; /* Use fstat instead of checking for errno == ESPIPE because lseek doesn't work on some special files but doesn't return an error, either. */ /* FIXME: can this really happen? What system? */ if (fstat (fdesc, &stats)) { error (0, errno, "%s", file); quit (1); } /* Try lseek and if an error indicates it was an inappropriate operation, fall back on using read. */ o = records * blocksize; if (o / blocksize != records || lseek (fdesc, o, SEEK_SET) == -1) { while (records-- > 0) { int nread; nread = safe_read (fdesc, buf, blocksize); if (nread < 0) { error (0, errno, "%s", file); quit (1); } /* POSIX doesn't say what to do when dd detects it has been asked to skip past EOF, so I assume it's non-fatal. FIXME: maybe give a warning. */ if (nread == 0) break; } } } /* Copy NREAD bytes of BUF, with no conversions. */ static void copy_simple (unsigned char *buf, int nread) { int nfree; /* Number of unused bytes in `obuf'. */ unsigned char *start = buf; /* First uncopied char in BUF. */ do { nfree = output_blocksize - oc; if (nfree > nread) nfree = nread; memcpy ((char *) (obuf + oc), (char *) start, nfree); nread -= nfree; /* Update the number of bytes left to copy. */ start += nfree; oc += nfree; if (oc >= output_blocksize) write_output (); } while (nread > 0); } /* Copy NREAD bytes of BUF, doing conv=block (pad newline-terminated records to `conversion_blocksize', replacing the newline with trailing spaces). */ static void copy_with_block (unsigned char *buf, int nread) { register int i; for (i = nread; i; i--, buf++) { if (*buf == newline_character) { if (col < conversion_blocksize) { size_t j; for (j = col; j < conversion_blocksize; j++) output_char (space_character); } col = 0; } else { if (col == conversion_blocksize) r_truncate++; else if (col < conversion_blocksize) output_char (*buf); col++; } } } /* Copy NREAD bytes of BUF, doing conv=unblock (replace trailing spaces in `conversion_blocksize'-sized records with a newline). */ static void copy_with_unblock (unsigned char *buf, int nread) { register int i; register unsigned char c; static int pending_spaces = 0; for (i = 0; i < nread; i++) { c = buf[i]; if (col++ >= conversion_blocksize) { col = pending_spaces = 0; /* Wipe out any pending spaces. */ i--; /* Push the char back; get it later. */ output_char (newline_character); } else if (c == space_character) pending_spaces++; else { /* `c' is the character after a run of spaces that were not at the end of the conversion buffer. Output them. */ while (pending_spaces) { output_char (space_character); --pending_spaces; } output_char (c); } } } /* The main loop. */ static int copy (void) { unsigned char *ibuf, *bufstart; /* Input buffer. */ int nread; /* Bytes read in the current block. */ int exit_status = 0; /* Leave at least one extra byte at the beginning and end of `ibuf' for conv=swab, but keep the buffer address even. But some peculiar device drivers work only with word-aligned buffers, so leave an extra two bytes. */ ibuf = (unsigned char *) xmalloc (input_blocksize + 2 * SWAB_ALIGN_OFFSET); ibuf += SWAB_ALIGN_OFFSET; if (conversions_mask & C_TWOBUFS) obuf = (unsigned char *) xmalloc (output_blocksize); else obuf = ibuf; if (skip_records != 0) skip (input_fd, input_file, skip_records, input_blocksize, ibuf); if (seek_record != 0) { /* FIXME: this loses for % ./dd if=dd seek=1 |: ./dd: a1 standard output: Bad file number 0+0 records in 0+0 records out */ skip (output_fd, output_file, seek_record, output_blocksize, obuf); } if (max_records == 0) quit (exit_status); while (1) { if (r_partial + r_full >= max_records) break; /* Zero the buffer before reading, so that if we get a read error, whatever data we are able to read is followed by zeros. This minimizes data loss. */ if ((conversions_mask & C_SYNC) && (conversions_mask & C_NOERROR)) memset ((char *) ibuf, 0, input_blocksize); nread = safe_read (input_fd, ibuf, input_blocksize); if (nread == 0) break; /* EOF. */ if (nread < 0) { error (0, errno, "%s", input_file); if (conversions_mask & C_NOERROR) { print_stats (); /* Seek past the bad block if possible. */ lseek (input_fd, (off_t) input_blocksize, SEEK_CUR); if (conversions_mask & C_SYNC) /* Replace the missing input with null bytes and proceed normally. */ nread = 0; else continue; } else { /* Write any partial block. */ exit_status = 2; break; } } if (nread < input_blocksize) { r_partial++; if (conversions_mask & C_SYNC) { if (!(conversions_mask & C_NOERROR)) /* If C_NOERROR, we zeroed the block before reading. */ memset ((char *) (ibuf + nread), 0, input_blocksize - nread); nread = input_blocksize; } } else r_full++; if (ibuf == obuf) /* If not C_TWOBUFS. */ { int nwritten = full_write (output_fd, obuf, nread); if (nwritten < 0) { error (0, errno, "%s", output_file); quit (1); } else if (nread == input_blocksize) w_full++; else w_partial++; continue; } /* Do any translations on the whole buffer at once. */ if (translation_needed) translate_buffer (ibuf, nread); if (conversions_mask & C_SWAB) bufstart = swab_buffer (ibuf, &nread); else bufstart = ibuf; if (conversions_mask & C_BLOCK) copy_with_block (bufstart, nread); else if (conversions_mask & C_UNBLOCK) copy_with_unblock (bufstart, nread); else copy_simple (bufstart, nread); } /* If we have a char left as a result of conv=swab, output it. */ if (char_is_saved) { if (conversions_mask & C_BLOCK) copy_with_block (&saved_char, 1); else if (conversions_mask & C_UNBLOCK) copy_with_unblock (&saved_char, 1); else output_char (saved_char); } if ((conversions_mask & C_BLOCK) && col > 0) { /* If the final input line didn't end with a '\n', pad the output block to `conversion_blocksize' chars. */ size_t i; for (i = col; i < conversion_blocksize; i++) output_char (space_character); } if ((conversions_mask & C_UNBLOCK) && col == conversion_blocksize) /* Add a final '\n' if there are exactly `conversion_blocksize' characters in the final record. */ output_char (newline_character); /* Write out the last block. */ if (oc != 0) { int nwritten = full_write (output_fd, obuf, oc); if (nwritten > 0) w_partial++; if (nwritten < 0) { error (0, errno, "%s", output_file); quit (1); } } free (ibuf - SWAB_ALIGN_OFFSET); if (obuf != ibuf) free (obuf); return exit_status; } int main (int argc, char **argv) { int i; int exit_status; program_name = argv[0]; setlocale (LC_ALL, ""); bindtextdomain (PACKAGE, LOCALEDIR); textdomain (PACKAGE); /* Initialize translation table to identity translation. */ for (i = 0; i < 256; i++) trans_table[i] = i; /* Decode arguments. */ scanargs (argc, argv); if (show_version) { printf ("dd (%s) %s\n", GNU_PACKAGE, VERSION); close_stdout (); exit (0); } if (show_help) usage (0); apply_translations (); if (input_file != NULL) { input_fd = open (input_file, O_RDONLY); if (input_fd < 0) error (1, errno, "%s", input_file); } else input_file = _("standard input"); if (input_fd == output_fd) error (1, 0, _("%s is closed"), (input_fd == 0 ? _("standard input") : _("standard output"))); if (output_file != NULL) { int omode = O_RDWR | O_CREAT; if (seek_record == 0 && !(conversions_mask & C_NOTRUNC)) omode |= O_TRUNC; output_fd = open (output_file, omode, 0666); if (output_fd < 0) error (1, errno, "%s", output_file); #if HAVE_FTRUNCATE if (seek_record != 0 && !(conversions_mask & C_NOTRUNC)) { off_t o = seek_record * output_blocksize; if (o / output_blocksize != seek_record) error (1, 0, _("file offset out of range")); if (ftruncate (output_fd, o) < 0) error (0, errno, "%s", output_file); } #endif } else { output_file = _("standard output"); } install_handler (SIGINT, interrupt_handler); install_handler (SIGQUIT, interrupt_handler); install_handler (SIGPIPE, interrupt_handler); install_handler (SIGINFO, siginfo_handler); exit_status = copy (); quit (exit_status); }