src/fold.c

   1 /* fold -- wrap each input line to fit in specified width.
   2    Copyright (C) 1991 Free Software Foundation, Inc.
   3
   4    This program is free software; you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation; either version 2, or (at your option)
   7    any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program; if not, write to the Free Software
  16    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  17
  18 /* Written by David MacKenzie, djm@gnu.ai.mit.edu. */
  19
  20 #ifdef HAVE_CONFIG_H
  21 #if defined (CONFIG_BROKETS)
  22 /* We use <config.h> instead of "config.h" so that a compilation
  23    using -I. -I$srcdir will use ./config.h rather than $srcdir/config.h
  24    (which it would do because it found this file in $srcdir).  */
  25 #include <config.h>
  26 #else
  27 #include "config.h"
  28 #endif
  29 #endif
  30
  31 /* Get isblank from GNU libc.  */
  32 #define _GNU_SOURCE
  33
  34 #include <stdio.h>
  35 #include <getopt.h>
  36 #include <sys/types.h>
  37 #include "system.h"
  38 #include "version.h"
  39
  40 char *xrealloc ();
  41 char *xmalloc ();
  42 void error ();
  43
  44 /* The name this program was run with. */
  45 char *program_name;
  46
  47 static int adjust_column ();
  48 static int fold_file ();
  49
  50 /* If nonzero, try to break on whitespace. */
  51 static int break_spaces;
  52
  53 /* If nonzero, count bytes, not column positions. */
  54 static int count_bytes;
  55
  56 /* If nonzero, at least one of the files we read was standard input. */
  57 static int have_read_stdin;
  58
  59 /* If non-zero, display usage information and exit.  */
  60 static int show_help;
  61
  62 /* If non-zero, print the version on standard output then exit.  */
  63 static int show_version;
  64
  65 static struct option const longopts[] =
  66 {
  67   {"bytes", no_argument, NULL, 'b'},
  68   {"spaces", no_argument, NULL, 's'},
  69   {"width", required_argument, NULL, 'w'},
  70   {"help", no_argument, &show_help, 1},
  71   {"version", no_argument, &show_version, 1},
  72   {NULL, 0, NULL, 0}
  73 };
  74 \f
  75 static void
  76 usage (status)
  77      int status;
  78 {
  79   if (status != 0)
  80     fprintf (stderr, "Try `%s --help' for more information.\n",
  81              program_name);
  82   else
  83     {
  84       printf ("\
  85 Usage: %s [OPTION]... [FILE]...\n\
  86 ",
  87               program_name);
  88       printf ("\
  89 \n\
  90   -b, --bytes         count bytes rather than columns\n\
  91   -s, --spaces        break at word boundaries\n\
  92   -w, --width=WIDTH   use WIDTH columns instead of 80\n\
  93 ");
  94     }
  95   exit (status);
  96 }
  97
  98 void
  99 main (argc, argv)
 100      int argc;
 101      char **argv;
 102 {
 103   int width = 80;
 104   int i;
 105   int optc;
 106   int errs = 0;
 107
 108   program_name = argv[0];
 109   break_spaces = count_bytes = have_read_stdin = 0;
 110
 111   /* Turn any numeric options into -w options.  */
 112   for (i = 1; i < argc; i++)
 113     {
 114       if (argv[i][0] == '-' && ISDIGIT (argv[i][1]))
 115         {
 116           char *s;
 117
 118           s = xmalloc (strlen (argv[i]) + 2);
 119           s[0] = '-';
 120           s[1] = 'w';
 121           strcpy (s + 2, argv[i] + 1);
 122           argv[i] = s;
 123         }
 124     }
 125
 126   while ((optc = getopt_long (argc, argv, "bsw:", longopts, (int *) 0))
 127          != EOF)
 128     {
 129       switch (optc)
 130         {
 131         case 0:
 132           break;
 133
 134         case 'b':               /* Count bytes rather than columns. */
 135           count_bytes = 1;
 136           break;
 137
 138         case 's':               /* Break at word boundaries. */
 139           break_spaces = 1;
 140           break;
 141
 142         case 'w':               /* Line width. */
 143           width = atoi (optarg);
 144           if (width < 1)
 145             error (1, 0, "%s: invalid line width", optarg);
 146           break;
 147
 148         default:
 149           usage (1);
 150         }
 151     }
 152
 153   if (show_version)
 154     {
 155       printf ("%s\n", version_string);
 156       exit (0);
 157     }
 158
 159   if (show_help)
 160     usage (0);
 161
 162   if (argc == optind)
 163     errs |= fold_file ("-", width);
 164   else
 165     for (i = optind; i < argc; i++)
 166       errs |= fold_file (argv[i], width);
 167
 168   if (have_read_stdin && fclose (stdin) == EOF)
 169     error (1, errno, "-");
 170   if (fclose (stdout) == EOF)
 171     error (1, errno, "write error");
 172
 173   exit (errs);
 174 }
 175
 176 /* Fold file FILENAME, or standard input if FILENAME is "-",
 177    to stdout, with maximum line length WIDTH.
 178    Return 0 if successful, 1 if an error occurs. */
 179
 180 static int
 181 fold_file (filename, width)
 182      char *filename;
 183      int width;
 184 {
 185   FILE *istream;
 186   register int c;
 187   int column = 0;               /* Screen column where next char will go. */
 188   int offset_out = 0;           /* Index in `line_out' for next char. */
 189   static char *line_out = NULL;
 190   static size_t allocated_out = 0;
 191
 192   if (!strcmp (filename, "-"))
 193     {
 194       istream = stdin;
 195       have_read_stdin = 1;
 196     }
 197   else
 198     istream = fopen (filename, "r");
 199
 200   if (istream == NULL)
 201     {
 202       error (0, errno, "%s", filename);
 203       return 1;
 204     }
 205
 206   while ((c = getc (istream)) != EOF)
 207     {
 208       if (offset_out + 1 >= allocated_out)
 209         {
 210           allocated_out += 1024;
 211           line_out = xrealloc (line_out, allocated_out);
 212         }
 213
 214       if (c == '\n')
 215         {
 216           line_out[offset_out++] = c;
 217           fwrite (line_out, sizeof (char), offset_out, stdout);
 218           column = offset_out = 0;
 219           continue;
 220         }
 221
 222     rescan:
 223       column = adjust_column (column, c);
 224
 225       if (column > width)
 226         {
 227           /* This character would make the line too long.
 228              Print the line plus a newline, and make this character
 229              start the next line. */
 230           if (break_spaces)
 231             {
 232               /* Look for the last blank. */
 233               int logical_end;
 234
 235               for (logical_end = offset_out - 1; logical_end >= 0;
 236                    logical_end--)
 237                 if (ISBLANK (line_out[logical_end]))
 238                   break;
 239               if (logical_end >= 0)
 240                 {
 241                   int i;
 242
 243                   /* Found a blank.  Don't output the part after it. */
 244                   logical_end++;
 245                   fwrite (line_out, sizeof (char), logical_end, stdout);
 246                   putchar ('\n');
 247                   /* Move the remainder to the beginning of the next line.
 248                      The areas being copied here might overlap. */
 249                   bcopy (line_out + logical_end, line_out,
 250                          offset_out - logical_end);
 251                   offset_out -= logical_end;
 252                   for (column = i = 0; i < offset_out; i++)
 253                     column = adjust_column (column, line_out[i]);
 254                   goto rescan;
 255                 }
 256             }
 257           line_out[offset_out++] = '\n';
 258           fwrite (line_out, sizeof (char), offset_out, stdout);
 259           column = offset_out = 0;
 260           goto rescan;
 261         }
 262
 263       line_out[offset_out++] = c;
 264     }
 265
 266   if (offset_out)
 267     fwrite (line_out, sizeof (char), offset_out, stdout);
 268
 269   if (ferror (istream))
 270     {
 271       error (0, errno, "%s", filename);
 272       if (strcmp (filename, "-"))
 273         fclose (istream);
 274       return 1;
 275     }
 276   if (strcmp (filename, "-") && fclose (istream) == EOF)
 277     {
 278       error (0, errno, "%s", filename);
 279       return 1;
 280     }
 281
 282   if (ferror (stdout))
 283     {
 284       error (0, errno, "write error");
 285       return 1;
 286     }
 287
 288   return 0;
 289 }
 290
 291 /* Assuming the current column is COLUMN, return the column that
 292    printing C will move the cursor to.
 293    The first column is 0. */
 294
 295 static int
 296 adjust_column (column, c)
 297      int column;
 298      char c;
 299 {
 300   if (!count_bytes)
 301     {
 302       if (c == '\b')
 303         {
 304           if (column > 0)
 305             column--;
 306         }
 307       else if (c == '\r')
 308         column = 0;
 309       else if (c == '\t')
 310         column = column + 8 - column % 8;
 311       else /* if (isprint (c)) */
 312         column++;
 313     }
 314   else
 315     column++;
 316   return column;
 317 }