flashcp: pad output to BUFSIZE. Hopefully closes 5882
[oweals/busybox.git] / miscutils / strings.c
index 7758907c7fba8b3aeef246d235d8d9cf1f32a00e..9f5018244445a2b9f6504794afa89e3d677897e1 100644 (file)
 /*
  * strings implementation for busybox
  *
- * Copyright (c) 1980, 1987
- *     The Regents of the University of California.  All rights reserved.
+ * Copyright 2003 Tito Ragusa <farmatito@tiscali.it>
  *
- * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
- *
- * Original copyright notice is retained at the end of this file.
- *
- * Modified for BusyBox by Erik Andersen <andersen@codepoet.org>
- * Badly hacked by Tito Ragusa <farmatito@tiscali.it>
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  */
 
-#include <stdio.h>
-#include <stdlib.h>
-#include <getopt.h>
-#include <ctype.h>
-#include "busybox.h"
+//usage:#define strings_trivial_usage
+//usage:       "[-afo] [-n LEN] [FILE]..."
+//usage:#define strings_full_usage "\n\n"
+//usage:       "Display printable strings in a binary file\n"
+//usage:     "\n       -a      Scan whole file (default)"
+//usage:     "\n       -f      Precede strings with filenames"
+//usage:     "\n       -n LEN  At least LEN characters form a string (default 4)"
+//usage:     "\n       -o      Precede strings with decimal offsets"
+
+#include "libbb.h"
 
-#define WHOLE_FILE             1
-#define PRINT_NAME             2
-#define PRINT_OFFSET   4
-#define SIZE                   8
+#define WHOLE_FILE    1
+#define PRINT_NAME    2
+#define PRINT_OFFSET  4
+#define SIZE          8
 
-int strings_main(int argc, char **argv)
+int strings_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int strings_main(int argc UNUSED_PARAM, char **argv)
 {
-       int n, c, i = 0, status = EXIT_SUCCESS;
-       unsigned long opt;
-       unsigned long count;
-       FILE *file = stdin;
+       int n, c, status = EXIT_SUCCESS;
+       unsigned count;
+       off_t offset;
+       FILE *file;
        char *string;
        const char *fmt = "%s: ";
-       char *n_arg = "4";
+       const char *n_arg = "4";
 
-       opt = bb_getopt_ulflags(argc, argv, "afon:", &n_arg);
+       getopt32(argv, "afon:", &n_arg);
        /* -a is our default behaviour */
-
-       argc -= optind;
+       /*argc -= optind;*/
        argv += optind;
 
-       n = bb_xgetlarg(n_arg, 10, 1, INT_MAX);
+       n = xatou_range(n_arg, 1, INT_MAX);
        string = xzalloc(n + 1);
        n--;
 
-       if (argc == 0) {
+       if (!*argv) {
                fmt = "{%s}: ";
-               *argv = (char *)bb_msg_standard_input;
-               goto PIPE;
+               *--argv = (char *)bb_msg_standard_input;
        }
 
        do {
-               if ((file = bb_wfopen(*argv, "r"))) {
-PIPE:
-                       count = 0;
-                       do {
-                               c = fgetc(file);
-                               if (isprint(c) || c == '\t') {
-                                       if (i <= n) {
-                                               string[i] = c;
-                                       } else {
-                                               putchar(c);
-                                       }
-                                       if (i == n) {
-                                               if (opt & PRINT_NAME) {
+               file = fopen_or_warn_stdin(*argv);
+               if (!file) {
+                       status = EXIT_FAILURE;
+                       continue;
+               }
+               offset = 0;
+               count = 0;
+               do {
+                       c = fgetc(file);
+                       if (isprint_asciionly(c) || c == '\t') {
+                               if (count > n) {
+                                       bb_putchar(c);
+                               } else {
+                                       string[count] = c;
+                                       if (count == n) {
+                                               if (option_mask32 & PRINT_NAME) {
                                                        printf(fmt, *argv);
                                                }
-                                               if (opt & PRINT_OFFSET) {
-                                                       printf("%7lo ", count - n);
+                                               if (option_mask32 & PRINT_OFFSET) {
+                                                       printf("%7"OFF_FMT"o ", offset - n);
                                                }
-                                               printf("%s", string);
-                                       }
-                                       i++;
-                               } else {
-                                       if (i > n) {
-                                               putchar('\n');
+                                               fputs(string, stdout);
                                        }
-                                       i = 0;
+                                       count++;
                                }
-                               count++;
-                       } while (c != EOF);
-                       bb_fclose_nonstdin(file);
-               } else {
-                       status = EXIT_FAILURE;
-               }
-       } while (--argc > 0);
+                       } else {
+                               if (count > n) {
+                                       bb_putchar('\n');
+                               }
+                               count = 0;
+                       }
+                       offset++;
+               } while (c != EOF);
+               fclose_if_not_stdin(file);
+       } while (*++argv);
 
        if (ENABLE_FEATURE_CLEAN_UP)
                free(string);
 
-       bb_fflush_stdout_and_exit(status);
+       fflush_stdout_and_exit(status);
 }
-
-/*
- * Copyright (c) 1980, 1987
- *     The Regents of the University of California.  All rights reserved.
- *
- * 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. <BSD Advertising Clause omitted per the July 22, 1999 licensing change
- *             ftp://ftp.cs.berkeley.edu/pub/4bsd/README.Impt.License.Change>
- *
- * 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.
- *
- * 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.
- */