coreutils/fold.c

   1 /* vi: set sw=4 ts=4: */
   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
   8    Copyright (C) 2003 Glenn McGrath <bug1@iinet.net.au>
   9
  10    Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
  11 */
  12
  13 #include "busybox.h"
  14
  15 static unsigned long flags;
  16 #define FLAG_COUNT_BYTES        1
  17 #define FLAG_BREAK_SPACES       2
  18 #define FLAG_WIDTH                      4
  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 {
  26         if (!(flags & FLAG_COUNT_BYTES)) {
  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
  41 int fold_main(int argc, char **argv);
  42 int fold_main(int argc, char **argv)
  43 {
  44         char *w_opt;
  45         int width = 80;
  46         int i;
  47         int errs = 0;
  48
  49         if (ENABLE_INCLUDE_SUSv2) {
  50                 /* Turn any numeric options into -w options.  */
  51                 for (i = 1; i < argc; i++) {
  52                         char const *a = argv[i];
  53
  54                         if (*a++ == '-') {
  55                                 if (*a == '-' && !a[1])
  56                                         break;
  57                                 if (isdigit(*a)) {
  58                                         argv[i] = xasprintf("-w%s", a);
  59                                 }
  60                         }
  61                 }
  62         }
  63
  64         flags = getopt32(argc, argv, "bsw:", &w_opt);
  65         if (flags & FLAG_WIDTH)
  66                 width = xatoul_range(w_opt, 1, 10000);
  67
  68         argv += optind;
  69         if (!*argv) {
  70                 *--argv = (char*)"-";
  71         }
  72
  73         do {
  74                 FILE *istream = fopen_or_warn_stdin(*argv);
  75                 int c;
  76                 int column = 0;         /* Screen column where next char will go. */
  77                 int offset_out = 0;     /* Index in `line_out' for next char. */
  78                 static char *line_out = NULL;
  79                 static int allocated_out = 0;
  80
  81                 if (istream == NULL) {
  82                         errs |= EXIT_FAILURE;
  83                         continue;
  84                 }
  85
  86                 while ((c = getc(istream)) != EOF) {
  87                         if (offset_out + 1 >= allocated_out) {
  88                                 allocated_out += 1024;
  89                                 line_out = xrealloc(line_out, allocated_out);
  90                         }
  91
  92                         if (c == '\n') {
  93                                 line_out[offset_out++] = c;
  94                                 fwrite(line_out, sizeof(char), (size_t) offset_out, stdout);
  95                                 column = offset_out = 0;
  96                                 continue;
  97                         }
  98
  99 rescan:
 100                         column = adjust_column(column, c);
 101
 102                         if (column > width) {
 103                                 /* This character would make the line too long.
 104                                    Print the line plus a newline, and make this character
 105                                    start the next line. */
 106                                 if (flags & FLAG_BREAK_SPACES) {
 107                                         /* Look for the last blank. */
 108                                         int logical_end;
 109
 110                                         for (logical_end = offset_out - 1; logical_end >= 0; logical_end--) {
 111                                                 if (isblank(line_out[logical_end])) {
 112                                                         break;
 113                                                 }
 114                                         }
 115                                         if (logical_end >= 0) {
 116                                                 /* Found a blank.  Don't output the part after it. */
 117                                                 logical_end++;
 118                                                 fwrite(line_out, sizeof(char), (size_t) logical_end, stdout);
 119                                                 putchar('\n');
 120                                                 /* Move the remainder to the beginning of the next line.
 121                                                    The areas being copied here might overlap. */
 122                                                 memmove(line_out, line_out + logical_end, offset_out - logical_end);
 123                                                 offset_out -= logical_end;
 124                                                 for (column = i = 0; i < offset_out; i++) {
 125                                                         column = adjust_column(column, line_out[i]);
 126                                                 }
 127                                                 goto rescan;
 128                                         }
 129                                 } else {
 130                                         if (offset_out == 0) {
 131                                                 line_out[offset_out++] = c;
 132                                                 continue;
 133                                         }
 134                                 }
 135                                 line_out[offset_out++] = '\n';
 136                                 fwrite(line_out, sizeof(char), (size_t) offset_out, stdout);
 137                                 column = offset_out = 0;
 138                                 goto rescan;
 139                         }
 140
 141                         line_out[offset_out++] = c;
 142                 }
 143
 144                 if (offset_out) {
 145                         fwrite(line_out, sizeof(char), (size_t) offset_out, stdout);
 146                 }
 147
 148                 if (ferror(istream) || fclose_if_not_stdin(istream)) {
 149                         bb_perror_msg("%s", *argv);     /* Avoid multibyte problems. */
 150                         errs |= EXIT_FAILURE;
 151                 }
 152         } while (*++argv);
 153
 154         fflush_stdout_and_exit(errs);
 155 }