inetd: comment tweak. no code changes

[oweals/busybox.git] / coreutils / catv.c

diff --git a/coreutils/catv.c b/coreutils/catv.c
index dd4aa44e39b022abf0bbecc00ccabe31ffed91e6..ff3139c80672ab3f1db7ae325b7588ccdeb6a735 100644 (file)
--- a/coreutils/catv.c
+++ b/coreutils/catv.c
@@ -10,56 +10,66 @@
 /* See "Cat -v considered harmful" at
  * http://cm.bell-labs.com/cm/cs/doc/84/kp.ps.gz */
 
-#include "busybox.h"
-#include <unistd.h>
-#include <fcntl.h>
+#include "libbb.h"
 
-int catv_main(int argc, char **argv)
+int catv_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int catv_main(int argc UNUSED_PARAM, char **argv)
 {
-       int retval = EXIT_SUCCESS, fd, flags;
-
-       flags = bb_getopt_ulflags(argc, argv, "etv");
-       flags ^= 4;
-
-       // Loop through files.
+       int retval = EXIT_SUCCESS;
+       int fd;
+       unsigned flags;
 
+       flags = getopt32(argv, "etv");
+#define CATV_OPT_e (1<<0)
+#define CATV_OPT_t (1<<1)
+#define CATV_OPT_v (1<<2)
+       flags ^= CATV_OPT_v;
        argv += optind;
-       do {
-               // Read from stdin if there's nothing else to do.
 
-               fd = 0;
-               if (*argv && 0>(fd = bb_xopen(*argv, O_RDONLY))) retval = EXIT_FAILURE;
-               else for(;;) {
+       /* Read from stdin if there's nothing else to do. */
+       if (!argv[0])
+               *--argv = (char*)"-";
+       do {
+               fd = open_or_warn_stdin(*argv);
+               if (fd < 0) {
+                       retval = EXIT_FAILURE;
+                       continue;
+               }
+               for (;;) {
                        int i, res;
 
-                       res = read(fd, bb_common_bufsiz1, sizeof(bb_common_bufsiz1));
-                       if (res < 0) retval = EXIT_FAILURE;
-                       if (res <1) break;
-                       for (i=0; i<res; i++) {
-                               char c=bb_common_bufsiz1[i];
+#define read_buf bb_common_bufsiz1
+                       res = read(fd, read_buf, COMMON_BUFSIZE);
+                       if (res < 0)
+                               retval = EXIT_FAILURE;
+                       if (res < 1)
+                               break;
+                       for (i = 0; i < res; i++) {
+                               unsigned char c = read_buf[i];
 
-                               if (c > 126 && (flags & 4)) {
+                               if (c > 126 && (flags & CATV_OPT_v)) {
                                        if (c == 127) {
                                                printf("^?");
                                                continue;
-                                       } else {
-                                               printf("M-");
-                                               c -= 128;
                                        }
+                                       printf("M-");
+                                       c -= 128;
                                }
                                if (c < 32) {
                                        if (c == 10) {
-                                          if (flags & 1) putchar('$');
-                                       } else if (flags & (c==9 ? 2 : 4)) {
+                                               if (flags & CATV_OPT_e)
+                                                       bb_putchar('$');
+                                       } else if (flags & (c==9 ? CATV_OPT_t : CATV_OPT_v)) {
                                                printf("^%c", c+'@');
                                                continue;
                                        }
                                }
-                               putchar(c);
+                               bb_putchar(c);
                        }
                }
-               if (ENABLE_FEATURE_CLEAN_UP && fd) close(fd);
+               if (ENABLE_FEATURE_CLEAN_UP && fd)
+                       close(fd);
        } while (*++argv);
 
-       return retval;
+       fflush_stdout_and_exit(retval);
 }