386BSD 0.1 development

[unix-history] / usr / src / usr.bin / fold / fold.c

diff --git a/usr/src/usr.bin/fold/fold.c b/usr/src/usr.bin/fold/fold.c
index 307f119..44c8561 100644 (file)
--- a/usr/src/usr.bin/fold/fold.c
+++ b/usr/src/usr.bin/fold/fold.c
@@ -1,95 +1,152 @@
-static char *sccsid = "@(#)fold.c      4.1 (Berkeley) %G%";
-#include <stdio.h>
-/*
- * fold - fold long lines for finite output devices
+/*-
+ * Copyright (c) 1990 The Regents of the University of California.
+ * All rights reserved.
+ *
+ * This code is derived from software contributed to Berkeley by
+ * Kevin Ruddy.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ * 3. All advertising materials mentioning features or use of this software
+ *    must display the following acknowledgement:
+ *     This product includes software developed by the University of
+ *     California, Berkeley and its contributors.
+ * 4. Neither the name of the University nor the names of its contributors
+ *    may be used to endorse or promote products derived from this software
+ *    without specific prior written permission.

  *
  *

- * Bill Joy UCB June 28, 1977
+ * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.

  */
 
  */
 

-int    fold =  80;
+#ifndef lint
+char copyright[] =
+"@(#) Copyright (c) 1990 The Regents of the University of California.\n\
+ All rights reserved.\n";
+#endif /* not lint */
+
+#ifndef lint
+static char sccsid[] = "@(#)fold.c     5.5 (Berkeley) 6/1/90";
+#endif /* not lint */
+
+#include <stdio.h>
+#include <string.h>
+
+#define        DEFLINEWIDTH    80

 
 main(argc, argv)
        int argc;
 
 main(argc, argv)
        int argc;

-       char *argv[];
+       char **argv;

 {
 {

-       register c;
-       FILE *f;
-       char obuf[BUFSIZ];
+       extern int errno, optind;
+       extern char *optarg;
+       register int ch;
+       int width;
+       char *p;

 
 

-       argc--, argv++;
-       setbuf(stdout, obuf);
-       if (argc > 0 && argv[0][0] == '-') {
-               fold = 0;
-               argv[0]++;
-               while (*argv[0] >= '0' && *argv[0] <= '9')
-                       fold =* 10, fold =+ *argv[0]++ - '0';
-               if (*argv[0]) {
-                       printf("Bad number for fold\n");
-                       exit(1);
-               }
-               argc--, argv++;
-       }
-       do {
-               if (argc > 0) {
-                       if (freopen(argv[0], "r", stdin) == NULL) {
-                               perror(argv[0]);
+       width = -1;
+       while ((ch = getopt(argc, argv, "0123456789w:")) != EOF)
+               switch (ch) {
+               case 'w':
+                       if ((width = atoi(optarg)) <= 0) {
+                               (void)fprintf(stderr,
+                                   "fold: illegal width value.\n");

                                exit(1);
                        }
                                exit(1);
                        }

-                       argc--, argv++;
-               }
-               for (;;) {
-                       c = getc(stdin);
-                       if (c == -1)
-                               break;
-                       putch(c);
+                       break;
+               case '0': case '1': case '2': case '3': case '4':
+               case '5': case '6': case '7': case '8': case '9':
+                       if (width == -1) {
+                               p = argv[optind - 1];
+                               if (p[0] == '-' && p[1] == ch && !p[2])
+                                       width = atoi(++p);
+                               else
+                                       width = atoi(argv[optind] + 1);
+                       }
+                       break;
+               default:
+                       (void)fprintf(stderr,
+                           "usage: fold [-w width] [file ...]\n");
+                       exit(1);

                }
                }

-       } while (argc > 0);
+       argv += optind;
+       argc -= optind;
+
+       if (width == -1)
+               width = DEFLINEWIDTH;
+       if (!*argv)
+               fold(width);
+       else for (; *argv; ++argv)
+               if (!freopen(*argv, "r", stdin)) {
+                       (void)fprintf(stderr,
+                           "fold: %s: %s\n", *argv, strerror(errno));
+                       exit(1);
+               } else
+                       fold(width);

        exit(0);
 }
 
        exit(0);
 }
 

-int    col;
-
-putch(c)
-       register c;
+fold(width)
+       register int width;

 {
 {

-       register ncol;
+       register int ch, col, new;

 
 

-       switch (c) {
-               case '\n':
-                       ncol = 0;
-                       break;
-               case '\t':
-                       ncol = (col + 8) &~ 7;
-                       break;
+       for (col = 0;;) {
+               switch (ch = getchar()) {
+               case EOF:
+                       return;

                case '\b':
                case '\b':

-                       ncol = col ? col - 1 : 0;
-                       break;
-               case '\r':
-                       ncol = 0;
+                       new = col ? col - 1 : 0;

                        break;
                        break;

-               default:
-                       ncol = col + 1;
-       }
-       if (ncol > fold)
-               putchar('\n'), col = 0;
-       putchar(c);
-       switch (c) {

                case '\n':
                case '\n':

-                       col = 0;
+               case '\r':
+                       new = 0;

                        break;
                case '\t':
                        break;
                case '\t':

-                       col =+ 8;
-                       col =& ~7;
+                       new = (col + 8) & ~7;
+                       break;
+               default:
+                       new = col + 1;

                        break;
                        break;

+               }
+
+               if (new > width) {
+                       putchar('\n');
+                       col = 0;
+               }
+               putchar(ch);
+
+               switch (ch) {

                case '\b':
                case '\b':

-                       if (col)
-                               col--;
+                       if (col > 0)
+                               --col;

                        break;
                        break;

+               case '\n':

                case '\r':
                        col = 0;
                        break;
                case '\r':
                        col = 0;
                        break;

+               case '\t':
+                       col += 8;
+                       col &= ~7;
+                       break;

                default:
                default:

-                       col++;
+                       ++col;

                        break;
                        break;

+               }

        }
 }
        }
 }