add wc

2018-10-05 20:04:10 +09:00 · 2018-10-05 20:04:10 +09:00 · ebf6e3f746
commit ebf6e3f746
parent 75a8a1751e
1 changed files with 131 additions and 0 deletions
--- a/apps/wc.c
+++ b/apps/wc.c
@ -0,0 +1,131 @@
+/* vim: tabstop=4 shiftwidth=4 noexpandtab
+ * This file is part of ToaruOS and is released under the terms
+ * of the NCSA / University of Illinois License - see LICENSE.md
+ * Copyright (C) 2018 K. Lange
+ *
+ * wc - count bytes, characters, words, lines...
+ */
+#include <stdio.h>
+#include <string.h>
+#include <unistd.h>
+#include <getopt.h>
+#include <errno.h>
+#include <toaru/decodeutf8.h>
+
+int main(int argc, char * argv[]) {
+	int show_lines = 0;
+	int show_words = 0;
+	int show_chars = 0;
+	int show_bytes = 0;
+
+	int opt;
+
+	while ((opt = getopt(argc,argv,"cmlw")) != -1) {
+		switch (opt) {
+			case 'c':
+				show_bytes = 1;
+				break;
+			case 'm':
+				show_chars = 1;
+				break;
+			case 'l':
+				show_lines = 1;
+				break;
+			case 'w':
+				show_words = 1;
+				break;
+		}
+	}
+
+	int retval = 0;
+	int total_lines = 0;
+	int total_chars = 0;
+	int total_words = 0;
+	int just_stdin = 0;
+
+	if (optind == argc) {
+		argv[optind] = "";
+		argc++;
+		just_stdin = 1;
+	}
+
+	for (int i = optind; i < argc; ++i) {
+		if (!*argv[i] && !just_stdin) {
+			fprintf(stderr, "%s: invalid zero-length file name\n", argv[0]);
+			retval = 1;
+			continue;
+		}
+		FILE * f = (!strcmp(argv[i], "-") || just_stdin) ? stdin : fopen(argv[i], "r");
+		if (!f) {
+			fprintf(stderr, "%s: %s: %s\n", argv[0], argv[i], strerror(errno));
+			retval = 1;
+			continue;
+		}
+
+		int lines = 0;
+		int chars = 0;
+		int words = 0;
+		int ch;
+		uint32_t state, c;
+		int last_was_whitespace = 0;
+
+		while (!feof(f)) {
+			ch = getc(f);
+			if (ch < 0) break;
+
+			if (show_chars) {
+				if (!decode(&state, &c, ch)) {
+				} else if (state == UTF8_REJECT) {
+					state = 0;
+				}
+			} else {
+				c = ch;
+			}
+
+			chars++;
+			if (c == '\n') {
+				last_was_whitespace = 1;
+				lines++;
+				words++;
+			} else if (c == ' ') {
+				if (last_was_whitespace) continue;
+				last_was_whitespace = 1;
+				words++;
+			} else {
+				last_was_whitespace = 0;
+			}
+		}
+
+		if (!last_was_whitespace && chars > 0) words++;
+
+		if (!show_words && !show_chars && !show_bytes && !show_lines) {
+			fprintf(stdout, "%d %d %d %s\n", lines, words, chars, argv[i]);
+		} else {
+			if (show_lines) fprintf(stdout, "%d ", lines);
+			if (show_words) fprintf(stdout, "%d ", words);
+			if (show_bytes|show_chars) fprintf(stdout, "%d ", chars);
+			fprintf(stdout, "%s\n", argv[i]);
+		}
+
+		total_lines += lines;
+		total_words += words;
+		total_chars += chars;
+
+		if (f != stdin) fclose(f);
+		if (just_stdin) return 0;
+	}
+
+	if (optind + 1 < argc) {
+		if (!show_words && !show_chars && !show_bytes && !show_lines) {
+			fprintf(stdout, "%d %d %d %s\n", total_lines, total_words, total_chars, "total");
+		} else {
+			if (show_lines) fprintf(stdout, "%d ", total_lines);
+			if (show_words) fprintf(stdout, "%d ", total_words);
+			if (show_bytes|show_chars) fprintf(stdout, "%d ", total_chars);
+			fprintf(stdout, "%s\n", "total");
+		}
+	}
+
+	return retval;
+}
+