blob: bad03f79152fafb2be918ee223ccbf3eed4d67c5 [file] [log] [blame]
Erik Andersene49d5ec2000-02-08 19:58:47 +00001/* vi: set sw=4 ts=4: */
Erik Andersen31638212000-01-15 22:28:50 +00002/*
3 * Mini wc implementation for busybox
4 *
5 * by Edward Betts <edward@debian.org>
6 *
7 * This program is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation; either version 2 of the License, or
10 * (at your option) any later version.
11 *
12 * This program is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with this program; if not, write to the Free Software
19 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 *
21 */
22
23#include "internal.h"
24#include <stdio.h>
25
Erik Andersen7ab9c7e2000-05-12 19:41:47 +000026static const char wc_usage[] = "wc [OPTION]... [FILE]...\n"
27#ifndef BB_FEATURE_TRIVIAL_HELP
28 "\nPrint line, word, and byte counts for each FILE, and a total line if\n"
Erik Andersene5b6c7d2000-04-17 16:16:10 +000029 "more than one FILE is specified. With no FILE, read standard input.\n\n"
30 "Options:\n"
Erik Andersene49d5ec2000-02-08 19:58:47 +000031 "\t-c\tprint the byte counts\n"
32 "\t-l\tprint the newline counts\n"
33
34 "\t-L\tprint the length of the longest line\n"
Erik Andersen7ab9c7e2000-05-12 19:41:47 +000035 "\t-w\tprint the word counts\n"
36#endif
37 ;
Erik Andersen31638212000-01-15 22:28:50 +000038
39static int total_lines, total_words, total_chars, max_length;
40static int print_lines, print_words, print_chars, print_length;
41
Erik Andersene49d5ec2000-02-08 19:58:47 +000042void print_counts(int lines, int words, int chars, int length,
43 const char *name)
44{
Erik Andersen31638212000-01-15 22:28:50 +000045 char const *space = "";
Erik Andersene49d5ec2000-02-08 19:58:47 +000046
Erik Andersen31638212000-01-15 22:28:50 +000047 if (print_lines) {
Erik Andersene49d5ec2000-02-08 19:58:47 +000048 printf("%7d", lines);
Erik Andersen31638212000-01-15 22:28:50 +000049 space = " ";
50 }
51 if (print_words) {
Erik Andersene49d5ec2000-02-08 19:58:47 +000052 printf("%s%7d", space, words);
Erik Andersen31638212000-01-15 22:28:50 +000053 space = " ";
54 }
55 if (print_chars) {
Erik Andersene49d5ec2000-02-08 19:58:47 +000056 printf("%s%7d", space, chars);
Erik Andersen31638212000-01-15 22:28:50 +000057 space = " ";
58 }
59 if (print_length)
Erik Andersene49d5ec2000-02-08 19:58:47 +000060 printf("%s%7d", space, length);
Erik Andersen31638212000-01-15 22:28:50 +000061 if (*name)
Erik Andersene49d5ec2000-02-08 19:58:47 +000062 printf(" %s", name);
63 putchar('\n');
Erik Andersen31638212000-01-15 22:28:50 +000064}
65
Erik Andersene49d5ec2000-02-08 19:58:47 +000066static void wc_file(FILE * file, const char *name)
Erik Andersen31638212000-01-15 22:28:50 +000067{
68 int lines, words, chars, length;
69 int in_word = 0, linepos = 0;
70 int c;
Erik Andersene49d5ec2000-02-08 19:58:47 +000071
Erik Andersen31638212000-01-15 22:28:50 +000072 lines = words = chars = length = 0;
73 while ((c = getc(file)) != EOF) {
74 chars++;
75 switch (c) {
Erik Andersene49d5ec2000-02-08 19:58:47 +000076 case '\n':
77 lines++;
78 case '\r':
79 case '\f':
80 if (linepos > length)
81 length = linepos;
82 linepos = 0;
83 goto word_separator;
84 case '\t':
85 linepos += 8 - (linepos % 8);
86 goto word_separator;
87 case ' ':
88 linepos++;
89 case '\v':
90 word_separator:
91 if (in_word) {
92 in_word = 0;
93 words++;
94 }
95 break;
96 default:
97 linepos++;
98 in_word = 1;
99 break;
Erik Andersen31638212000-01-15 22:28:50 +0000100 }
101 }
102 if (linepos > length)
103 length = linepos;
104 if (in_word)
105 words++;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000106 print_counts(lines, words, chars, length, name);
Erik Andersen31638212000-01-15 22:28:50 +0000107 total_lines += lines;
108 total_words += words;
109 total_chars += chars;
110 if (length > max_length)
111 max_length = length;
112 fclose(file);
113 fflush(stdout);
114}
115
Erik Andersene49d5ec2000-02-08 19:58:47 +0000116int wc_main(int argc, char **argv)
117{
Erik Andersen31638212000-01-15 22:28:50 +0000118 FILE *file;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000119
Erik Andersen31638212000-01-15 22:28:50 +0000120 total_lines = total_words = total_chars = max_length = 0;
121 print_lines = print_words = print_chars = print_length = 0;
122
123 while (--argc && **(++argv) == '-') {
124 while (*++(*argv))
125 switch (**argv) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000126 case 'c':
127 print_chars = 1;
128 break;
129 case 'l':
130 print_lines = 1;
131 break;
132 case 'L':
133 print_length = 1;
134 break;
135 case 'w':
136 print_words = 1;
137 break;
138 default:
139 usage(wc_usage);
Erik Andersen31638212000-01-15 22:28:50 +0000140 }
141 }
142
143 if (!print_lines && !print_words && !print_chars && !print_length)
144 print_lines = print_words = print_chars = 1;
145
146 if (argc == 0) {
147 wc_file(stdin, "");
148 exit(TRUE);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000149 } else if (argc == 1) {
Erik Andersen31638212000-01-15 22:28:50 +0000150 file = fopen(*argv, "r");
151 if (file == NULL) {
152 perror(*argv);
153 exit(FALSE);
154 }
155 wc_file(file, *argv);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000156 } else {
Erik Andersen31638212000-01-15 22:28:50 +0000157 while (argc-- > 0 && *argv != '\0' && strlen(*argv)) {
158 file = fopen(*argv, "r");
159 if (file == NULL) {
160 perror(*argv);
161 exit(FALSE);
162 }
163 wc_file(file, *argv);
164 argv++;
165 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000166 print_counts(total_lines, total_words, total_chars,
167 max_length, "total");
Erik Andersen31638212000-01-15 22:28:50 +0000168 }
Eric Andersenb6106152000-06-19 17:25:40 +0000169 return(TRUE);
Erik Andersen31638212000-01-15 22:28:50 +0000170}