busybox/coreutils/wc.c

/* 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)
{
	int output = 0;

	if (print_type & print_lines) {
		printf("%7d", lines);
		output++;
	}
	if (print_type & print_words) {
		if (output++)
			putchar(' ');
		printf("%7d", words);
	}
	if (print_type & print_chars) {
		if (output++)
			putchar(' ');
		printf("%7d", chars);
	}
	if (print_type & print_length) {
		if (output++)
			putchar(' ');
		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);
}