wc/ww.c

#include <stdio.h>
#include <unistd.h> // read, isatty

// STDIN_FILENO
int process_fn(int fn)
{
    char c[1];
		int seen_word=0;
		int seen_sep_after_word=0;
		int num_words = 0;
    while (read(fn, c, sizeof(c)) > 0) {
				if(*c != '\n' && *c != ' ' && *c != '\t'){
					seen_word = 1;
				} else if(seen_word) {
					seen_sep_after_word = 1;
				} else {
					// see a separator, but haven't seen a word: do nothing
					// exercise: what happens if you only track seen_sep,
					// instead of seen_sep_after_word?
					// test with: $ echo " x x" | ./wc
				}
				if(seen_word && seen_sep_after_word){
					num_words++;
					seen_sep_after_word = seen_word = 0;
				}
    }
		if(seen_word){
			num_words++;
		}
		printf("%i\n",num_words);
    return 0;
}

void usage(){
        printf("Usage: ww file.txt\n");
				printf("   or: cat file.txt | ww\n");
}

int main(int argc, char** argv)
{
    if(argc > 1 && argv[1][0] == '-' && argv[1][1] == 'h'){
			usage();
			return 0;
		} else if (!isatty(STDIN_FILENO)) {
        return process_fn(STDIN_FILENO);
    } else if (argc > 1) {
        FILE* fp = fopen(argv[1], "r");
        if (!fp) {
            perror("Could not open file");
            return 1;
        }
        return process_fn(fileno(fp));
    } else {
			usage();
    }
    return 0;
}