Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
John Beppu | 3157b1f | 1999-12-10 07:42:50 +0000 | [diff] [blame] | 2 | /* |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 3 | * head implementation for busybox |
John Beppu | 3157b1f | 1999-12-10 07:42:50 +0000 | [diff] [blame] | 4 | * |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 5 | * Copyright (C) 2003 Manuel Novoa III <mjn3@codepoet.org> |
John Beppu | 3157b1f | 1999-12-10 07:42:50 +0000 | [diff] [blame] | 6 | * |
Rob Landley | f8fd4db | 2006-01-30 01:30:39 +0000 | [diff] [blame] | 7 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. |
John Beppu | 3157b1f | 1999-12-10 07:42:50 +0000 | [diff] [blame] | 8 | */ |
| 9 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 10 | /* BB_AUDIT SUSv3 compliant */ |
| 11 | /* BB_AUDIT GNU compatible -c, -q, and -v options in 'fancy' configuration. */ |
| 12 | /* http://www.opengroup.org/onlinepubs/007904975/utilities/head.html */ |
| 13 | |
Eric Andersen | cbe31da | 2001-02-20 06:14:08 +0000 | [diff] [blame] | 14 | #include "busybox.h" |
John Beppu | 3157b1f | 1999-12-10 07:42:50 +0000 | [diff] [blame] | 15 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 16 | static const char head_opts[] = |
| 17 | "n:" |
Bernhard Reutner-Fischer | 5816ccb | 2005-12-13 10:48:45 +0000 | [diff] [blame] | 18 | #if ENABLE_FEATURE_FANCY_HEAD |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 19 | "c:qv" |
| 20 | #endif |
| 21 | ; |
John Beppu | 3157b1f | 1999-12-10 07:42:50 +0000 | [diff] [blame] | 22 | |
"Vladimir N. Oleynik" | e75b41d | 2006-01-30 11:15:11 +0000 | [diff] [blame] | 23 | #if ENABLE_FEATURE_FANCY_HEAD |
| 24 | static const struct suffix_mult head_suffixes[] = { |
| 25 | { "b", 512 }, |
| 26 | { "k", 1024 }, |
| 27 | { "m", 1024*1024 }, |
| 28 | { NULL, 0 } |
| 29 | }; |
| 30 | #endif |
Denis Vlasenko | 9213a9e | 2006-09-17 16:28:10 +0000 | [diff] [blame] | 31 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 32 | static const char header_fmt_str[] = "\n==> %s <==\n"; |
John Beppu | 3157b1f | 1999-12-10 07:42:50 +0000 | [diff] [blame] | 33 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 34 | int head_main(int argc, char **argv) |
John Beppu | 3157b1f | 1999-12-10 07:42:50 +0000 | [diff] [blame] | 35 | { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 36 | unsigned long count = 10; |
| 37 | unsigned long i; |
Bernhard Reutner-Fischer | 5816ccb | 2005-12-13 10:48:45 +0000 | [diff] [blame] | 38 | #if ENABLE_FEATURE_FANCY_HEAD |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 39 | int count_bytes = 0; |
| 40 | int header_threshhold = 1; |
| 41 | #endif |
| 42 | |
Matt Kraai | c0321f9 | 2000-09-27 04:09:22 +0000 | [diff] [blame] | 43 | FILE *fp; |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 44 | const char *fmt; |
| 45 | char *p; |
| 46 | int opt; |
| 47 | int c; |
| 48 | int retval = EXIT_SUCCESS; |
John Beppu | 3157b1f | 1999-12-10 07:42:50 +0000 | [diff] [blame] | 49 | |
Rob Landley | f8fd4db | 2006-01-30 01:30:39 +0000 | [diff] [blame] | 50 | #if !ENABLE_DEBUG_YANK_SUSv2 || ENABLE_FEATURE_FANCY_HEAD |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 51 | /* Allow legacy syntax of an initial numeric option without -n. */ |
| 52 | if ((argc > 1) && (argv[1][0] == '-') |
| 53 | /* && (isdigit)(argv[1][1]) */ |
| 54 | && (((unsigned int)(argv[1][1] - '0')) <= 9) |
| 55 | ) { |
| 56 | --argc; |
| 57 | ++argv; |
| 58 | p = (*argv) + 1; |
| 59 | goto GET_COUNT; |
Robert Griebl | 13c26fc | 2002-05-17 22:18:04 +0000 | [diff] [blame] | 60 | } |
Glenn L McGrath | 0bd0257 | 2005-12-11 03:09:05 +0000 | [diff] [blame] | 61 | #endif |
Robert Griebl | 13c26fc | 2002-05-17 22:18:04 +0000 | [diff] [blame] | 62 | |
Denis Vlasenko | 67b23e6 | 2006-10-03 21:00:06 +0000 | [diff] [blame^] | 63 | /* No size benefit in converting this to getopt32 */ |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 64 | while ((opt = getopt(argc, argv, head_opts)) > 0) { |
Denis Vlasenko | c16bd21 | 2006-09-27 19:51:06 +0000 | [diff] [blame] | 65 | switch (opt) { |
Bernhard Reutner-Fischer | 5816ccb | 2005-12-13 10:48:45 +0000 | [diff] [blame] | 66 | #if ENABLE_FEATURE_FANCY_HEAD |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 67 | case 'q': |
| 68 | header_threshhold = INT_MAX; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 69 | break; |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 70 | case 'v': |
| 71 | header_threshhold = -1; |
| 72 | break; |
| 73 | case 'c': |
| 74 | count_bytes = 1; |
| 75 | /* fall through */ |
| 76 | #endif |
| 77 | case 'n': |
| 78 | p = optarg; |
Rob Landley | f8fd4db | 2006-01-30 01:30:39 +0000 | [diff] [blame] | 79 | #if !ENABLE_DEBUG_YANK_SUSv2 || ENABLE_FEATURE_FANCY_HEAD |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 80 | GET_COUNT: |
Bernhard Reutner-Fischer | 5816ccb | 2005-12-13 10:48:45 +0000 | [diff] [blame] | 81 | #endif |
"Vladimir N. Oleynik" | e75b41d | 2006-01-30 11:15:11 +0000 | [diff] [blame] | 82 | |
| 83 | #if !ENABLE_FEATURE_FANCY_HEAD |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 84 | count = bb_xgetularg10(p); |
"Vladimir N. Oleynik" | e75b41d | 2006-01-30 11:15:11 +0000 | [diff] [blame] | 85 | #else |
Denis Vlasenko | 9213a9e | 2006-09-17 16:28:10 +0000 | [diff] [blame] | 86 | count = bb_xgetularg_bnd_sfx(p, 10, |
| 87 | 0, ULONG_MAX, |
"Vladimir N. Oleynik" | e75b41d | 2006-01-30 11:15:11 +0000 | [diff] [blame] | 88 | head_suffixes); |
| 89 | #endif |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 90 | break; |
| 91 | default: |
| 92 | bb_show_usage(); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 93 | } |
| 94 | } |
| 95 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 96 | argv += optind; |
| 97 | if (!*argv) { |
| 98 | *--argv = "-"; |
John Beppu | 3157b1f | 1999-12-10 07:42:50 +0000 | [diff] [blame] | 99 | } |
John Beppu | 3157b1f | 1999-12-10 07:42:50 +0000 | [diff] [blame] | 100 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 101 | fmt = header_fmt_str + 1; |
Bernhard Reutner-Fischer | 5816ccb | 2005-12-13 10:48:45 +0000 | [diff] [blame] | 102 | #if ENABLE_FEATURE_FANCY_HEAD |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 103 | if (argc - optind <= header_threshhold) { |
| 104 | header_threshhold = 0; |
| 105 | } |
| 106 | #else |
| 107 | if (argc <= optind + 1) { |
| 108 | fmt += 11; |
| 109 | } |
| 110 | /* Now define some things here to avoid #ifdefs in the code below. |
| 111 | * These should optimize out of the if conditions below. */ |
| 112 | #define header_threshhold 1 |
| 113 | #define count_bytes 0 |
| 114 | #endif |
| 115 | |
| 116 | do { |
| 117 | if ((fp = bb_wfopen_input(*argv)) != NULL) { |
| 118 | if (fp == stdin) { |
| 119 | *argv = (char *) bb_msg_standard_input; |
| 120 | } |
| 121 | if (header_threshhold) { |
| 122 | bb_printf(fmt, *argv); |
| 123 | } |
| 124 | i = count; |
| 125 | while (i && ((c = getc(fp)) != EOF)) { |
| 126 | if (count_bytes || (c == '\n')) { |
| 127 | --i; |
| 128 | } |
| 129 | putchar(c); |
| 130 | } |
| 131 | if (bb_fclose_nonstdin(fp)) { |
| 132 | bb_perror_msg("%s", *argv); /* Avoid multibyte problems. */ |
| 133 | retval = EXIT_FAILURE; |
| 134 | } |
Rob Landley | d921b2e | 2006-08-03 15:41:12 +0000 | [diff] [blame] | 135 | xferror_stdout(); |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 136 | } |
| 137 | fmt = header_fmt_str; |
| 138 | } while (*++argv); |
| 139 | |
| 140 | bb_fflush_stdout_and_exit(retval); |
Matt Kraai | c0321f9 | 2000-09-27 04:09:22 +0000 | [diff] [blame] | 141 | } |