"Robert P. J. Day" | 63fc1a9 | 2006-07-02 19:47:05 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 2 | /* fold -- wrap each input line to fit in specified width. |
| 3 | |
| 4 | Written by David MacKenzie, djm@gnu.ai.mit.edu. |
| 5 | Copyright (C) 91, 1995-2002 Free Software Foundation, Inc. |
| 6 | |
| 7 | Modified for busybox based on coreutils v 5.0 |
Glenn L McGrath | c6992fe | 2004-04-25 05:11:19 +0000 | [diff] [blame] | 8 | Copyright (C) 2003 Glenn McGrath <bug1@iinet.net.au> |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 9 | |
Rob Landley | 746cfc8 | 2005-12-02 17:55:45 +0000 | [diff] [blame] | 10 | Licensed under the GPL v2 or later, see the file LICENSE in this tarball. |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 11 | */ |
| 12 | |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 13 | #include "busybox.h" |
| 14 | |
Rob Landley | 746cfc8 | 2005-12-02 17:55:45 +0000 | [diff] [blame] | 15 | static unsigned long flags; |
| 16 | #define FLAG_COUNT_BYTES 1 |
| 17 | #define FLAG_BREAK_SPACES 2 |
| 18 | #define FLAG_WIDTH 4 |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 19 | |
| 20 | /* Assuming the current column is COLUMN, return the column that |
| 21 | printing C will move the cursor to. |
| 22 | The first column is 0. */ |
| 23 | |
| 24 | static int adjust_column(int column, char c) |
| 25 | { |
Rob Landley | 746cfc8 | 2005-12-02 17:55:45 +0000 | [diff] [blame] | 26 | if (!(flags & FLAG_COUNT_BYTES)) { |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 27 | if (c == '\b') { |
| 28 | if (column > 0) |
| 29 | column--; |
| 30 | } else if (c == '\r') |
| 31 | column = 0; |
| 32 | else if (c == '\t') |
| 33 | column = column + 8 - column % 8; |
| 34 | else /* if (isprint (c)) */ |
| 35 | column++; |
| 36 | } else |
| 37 | column++; |
| 38 | return column; |
| 39 | } |
| 40 | |
Rob Landley | dfba741 | 2006-03-06 20:47:33 +0000 | [diff] [blame] | 41 | int fold_main(int argc, char **argv) |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 42 | { |
Paul Fox | 377bdaf | 2005-12-08 18:48:20 +0000 | [diff] [blame] | 43 | char *w_opt; |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 44 | int width = 80; |
| 45 | int i; |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 46 | int errs = 0; |
| 47 | |
Rob Landley | f8fd4db | 2006-01-30 01:30:39 +0000 | [diff] [blame] | 48 | if(!ENABLE_DEBUG_YANK_SUSv2) { |
| 49 | /* Turn any numeric options into -w options. */ |
| 50 | for (i = 1; i < argc; i++) { |
| 51 | char const *a = argv[i]; |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 52 | |
"Vladimir N. Oleynik" | b1fd52e | 2006-01-30 10:49:14 +0000 | [diff] [blame] | 53 | if (*a++ == '-') { |
| 54 | if (*a == '-' && !a[1]) |
Rob Landley | f8fd4db | 2006-01-30 01:30:39 +0000 | [diff] [blame] | 55 | break; |
"Vladimir N. Oleynik" | b1fd52e | 2006-01-30 10:49:14 +0000 | [diff] [blame] | 56 | if (isdigit(*a)) { |
Rob Landley | d921b2e | 2006-08-03 15:41:12 +0000 | [diff] [blame] | 57 | argv[i] = xasprintf("-w%s", a); |
Rob Landley | f8fd4db | 2006-01-30 01:30:39 +0000 | [diff] [blame] | 58 | } |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 59 | } |
| 60 | } |
| 61 | } |
| 62 | |
Denis Vlasenko | 67b23e6 | 2006-10-03 21:00:06 +0000 | [diff] [blame] | 63 | flags = getopt32(argc, argv, "bsw:", &w_opt); |
Bernhard Reutner-Fischer | d77b781 | 2005-12-10 20:13:54 +0000 | [diff] [blame] | 64 | if (flags & FLAG_WIDTH) |
Denis Vlasenko | 1385899 | 2006-10-08 12:49:22 +0000 | [diff] [blame] | 65 | width = xatoul_range(w_opt, 1, 10000); |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 66 | |
| 67 | argv += optind; |
| 68 | if (!*argv) { |
| 69 | *--argv = "-"; |
| 70 | } |
| 71 | |
| 72 | do { |
Denis Vlasenko | ddec5af | 2006-10-26 23:25:17 +0000 | [diff] [blame] | 73 | FILE *istream = fopen_or_warn_stdin(*argv); |
Bernhard Reutner-Fischer | 4bf3127 | 2006-01-30 17:41:06 +0000 | [diff] [blame] | 74 | int c; |
| 75 | int column = 0; /* Screen column where next char will go. */ |
| 76 | int offset_out = 0; /* Index in `line_out' for next char. */ |
| 77 | static char *line_out = NULL; |
| 78 | static int allocated_out = 0; |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 79 | |
Bernhard Reutner-Fischer | 4bf3127 | 2006-01-30 17:41:06 +0000 | [diff] [blame] | 80 | if (istream == NULL) { |
| 81 | errs |= EXIT_FAILURE; |
| 82 | continue; |
| 83 | } |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 84 | |
Bernhard Reutner-Fischer | 4bf3127 | 2006-01-30 17:41:06 +0000 | [diff] [blame] | 85 | while ((c = getc(istream)) != EOF) { |
| 86 | if (offset_out + 1 >= allocated_out) { |
| 87 | allocated_out += 1024; |
| 88 | line_out = xrealloc(line_out, allocated_out); |
| 89 | } |
| 90 | |
| 91 | if (c == '\n') { |
| 92 | line_out[offset_out++] = c; |
| 93 | fwrite(line_out, sizeof(char), (size_t) offset_out, stdout); |
| 94 | column = offset_out = 0; |
| 95 | continue; |
| 96 | } |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 97 | |
| 98 | rescan: |
Bernhard Reutner-Fischer | 4bf3127 | 2006-01-30 17:41:06 +0000 | [diff] [blame] | 99 | column = adjust_column(column, c); |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 100 | |
Bernhard Reutner-Fischer | 4bf3127 | 2006-01-30 17:41:06 +0000 | [diff] [blame] | 101 | if (column > width) { |
| 102 | /* This character would make the line too long. |
| 103 | Print the line plus a newline, and make this character |
| 104 | start the next line. */ |
| 105 | if (flags & FLAG_BREAK_SPACES) { |
| 106 | /* Look for the last blank. */ |
| 107 | int logical_end; |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 108 | |
Bernhard Reutner-Fischer | 4bf3127 | 2006-01-30 17:41:06 +0000 | [diff] [blame] | 109 | for (logical_end = offset_out - 1; logical_end >= 0; logical_end--) { |
| 110 | if (isblank(line_out[logical_end])) { |
| 111 | break; |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 112 | } |
| 113 | } |
Bernhard Reutner-Fischer | 4bf3127 | 2006-01-30 17:41:06 +0000 | [diff] [blame] | 114 | if (logical_end >= 0) { |
| 115 | /* Found a blank. Don't output the part after it. */ |
| 116 | logical_end++; |
| 117 | fwrite(line_out, sizeof(char), (size_t) logical_end, stdout); |
| 118 | putchar('\n'); |
| 119 | /* Move the remainder to the beginning of the next line. |
| 120 | The areas being copied here might overlap. */ |
| 121 | memmove(line_out, line_out + logical_end, offset_out - logical_end); |
| 122 | offset_out -= logical_end; |
| 123 | for (column = i = 0; i < offset_out; i++) { |
| 124 | column = adjust_column(column, line_out[i]); |
| 125 | } |
| 126 | goto rescan; |
| 127 | } |
| 128 | } else { |
| 129 | if (offset_out == 0) { |
| 130 | line_out[offset_out++] = c; |
| 131 | continue; |
| 132 | } |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 133 | } |
Bernhard Reutner-Fischer | 4bf3127 | 2006-01-30 17:41:06 +0000 | [diff] [blame] | 134 | line_out[offset_out++] = '\n'; |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 135 | fwrite(line_out, sizeof(char), (size_t) offset_out, stdout); |
Bernhard Reutner-Fischer | 4bf3127 | 2006-01-30 17:41:06 +0000 | [diff] [blame] | 136 | column = offset_out = 0; |
| 137 | goto rescan; |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 138 | } |
| 139 | |
Bernhard Reutner-Fischer | 4bf3127 | 2006-01-30 17:41:06 +0000 | [diff] [blame] | 140 | line_out[offset_out++] = c; |
| 141 | } |
| 142 | |
| 143 | if (offset_out) { |
| 144 | fwrite(line_out, sizeof(char), (size_t) offset_out, stdout); |
| 145 | } |
| 146 | |
Denis Vlasenko | ddec5af | 2006-10-26 23:25:17 +0000 | [diff] [blame] | 147 | if (ferror(istream) || fclose_if_not_stdin(istream)) { |
Bernhard Reutner-Fischer | 4bf3127 | 2006-01-30 17:41:06 +0000 | [diff] [blame] | 148 | bb_perror_msg("%s", *argv); /* Avoid multibyte problems. */ |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 149 | errs |= EXIT_FAILURE; |
| 150 | } |
| 151 | } while (*++argv); |
| 152 | |
Denis Vlasenko | f0ed376 | 2006-10-26 23:21:47 +0000 | [diff] [blame] | 153 | fflush_stdout_and_exit(errs); |
Glenn L McGrath | f01b805 | 2003-04-27 06:02:14 +0000 | [diff] [blame] | 154 | } |