coreutils/fold.c

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