| /* vi: set sw=4 ts=4: */ |
| /* |
| * Mini wc implementation for busybox |
| * |
| * Copyright (C) 2000 Edward Betts <edward@debian.org> |
| * |
| * This program is free software; you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License as published by |
| * the Free Software Foundation; either version 2 of the License, or |
| * (at your option) any later version. |
| * |
| * This program is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License |
| * along with this program; if not, write to the Free Software |
| * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
| * |
| */ |
| |
| #include <stdio.h> |
| #include <getopt.h> |
| #include <string.h> |
| #include <stdlib.h> |
| #include "busybox.h" |
| |
| enum print_e { |
| print_lines = 1, |
| print_words = 2, |
| print_chars = 4, |
| print_length = 8 |
| }; |
| |
| static unsigned int total_lines = 0; |
| static unsigned int total_words = 0; |
| static unsigned int total_chars = 0; |
| static unsigned int max_length = 0; |
| static char print_type = 0; |
| |
| static void print_counts(const unsigned int lines, const unsigned int words, |
| const unsigned int chars, const unsigned int length, const char *name) |
| { |
| if (print_type & print_lines) { |
| printf("%7d ", lines); |
| } |
| if (print_type & print_words) { |
| printf("%7d ", words); |
| } |
| if (print_type & print_chars) { |
| printf("%7d ", chars); |
| } |
| if (print_type & print_length) { |
| printf("%7d ", length); |
| } |
| if (*name) { |
| printf("%s", name); |
| } |
| putchar('\n'); |
| } |
| |
| static void wc_file(FILE * file, const char *name) |
| { |
| unsigned int lines = 0; |
| unsigned int words = 0; |
| unsigned int chars = 0; |
| unsigned int length = 0; |
| unsigned int linepos = 0; |
| char in_word = 0; |
| int c; |
| |
| while ((c = getc(file)) != EOF) { |
| chars++; |
| switch (c) { |
| case '\n': |
| lines++; |
| case '\r': |
| case '\f': |
| if (linepos > length) |
| length = linepos; |
| linepos = 0; |
| goto word_separator; |
| case '\t': |
| linepos += 8 - (linepos % 8); |
| goto word_separator; |
| case ' ': |
| linepos++; |
| case '\v': |
| word_separator: |
| if (in_word) { |
| in_word = 0; |
| words++; |
| } |
| break; |
| default: |
| linepos++; |
| in_word = 1; |
| break; |
| } |
| } |
| if (linepos > length) { |
| length = linepos; |
| } |
| if (in_word) { |
| words++; |
| } |
| print_counts(lines, words, chars, length, name); |
| total_lines += lines; |
| total_words += words; |
| total_chars += chars; |
| if (length > max_length) { |
| max_length = length; |
| } |
| } |
| |
| int wc_main(int argc, char **argv) |
| { |
| int opt; |
| |
| while ((opt = getopt(argc, argv, "clLw")) > 0) { |
| switch (opt) { |
| case 'c': |
| print_type |= print_chars; |
| break; |
| case 'l': |
| print_type |= print_lines; |
| break; |
| case 'L': |
| print_type |= print_length; |
| break; |
| case 'w': |
| print_type |= print_words; |
| break; |
| default: |
| show_usage(); |
| } |
| } |
| |
| if (print_type == 0) { |
| print_type = print_lines | print_words | print_chars; |
| } |
| |
| if (argv[optind] == NULL || strcmp(argv[optind], "-") == 0) { |
| wc_file(stdin, ""); |
| } else { |
| unsigned short num_files_counted = 0; |
| while (optind < argc) { |
| if (print_type == print_chars) { |
| struct stat statbuf; |
| stat(argv[optind], &statbuf); |
| print_counts(0, 0, statbuf.st_size, 0, argv[optind]); |
| total_chars += statbuf.st_size; |
| } else { |
| FILE *file; |
| file = xfopen(argv[optind], "r"); |
| wc_file(file, argv[optind]); |
| fclose(file); |
| } |
| optind++; |
| num_files_counted++; |
| } |
| if (num_files_counted > 1) { |
| print_counts(total_lines, total_words, total_chars, max_length, "total"); |
| } |
| } |
| |
| return(EXIT_SUCCESS); |
| } |