add du (not block based)

apps/du.c

@@ -0,0 +1,115 @@
+/* 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
+ *
+ * du - calculate file size usage
+ *
+ * TODO: Should use st_blocks, but we don't set that in the kernel yet?
+ *
+ */
+#include <stdio.h>
+#include <string.h>
+#include <unistd.h>
+#include <errno.h>
+#include <dirent.h>
+
+#include <fcntl.h>
+#include <sys/stat.h>
+#include <sys/ioctl.h>
+
+static int human = 0;
+static int all = 1;
+
+static uint64_t count_thing(char * tmp);
+
+static int print_human_readable_size(char * _out, size_t s) {
+	if (s >= 1<<20) {
+		size_t t = s / (1 << 20);
+		return sprintf(_out, "%d.%1dM", (int)t, (int)(s - t * (1 << 20)) / ((1 << 20) / 10));
+	} else if (s >= 1<<10) {
+		size_t t = s / (1 << 10);
+		return sprintf(_out, "%d.%1dK", (int)t, (int)(s - t * (1 << 10)) / ((1 << 10) / 10));
+	} else {
+		return sprintf(_out, "%d", (int)s);
+	}
+}
+
+static void print_size(uint64_t size, char * name) {
+	char sizes[8];
+	if (!human) {
+		sprintf(sizes, "%7llu", size/1024LLU);
+	} else {
+		print_human_readable_size(sizes, size);
+	}
+	fprintf(stdout, "%7s %s\n", sizes, name);
+}
+
+static uint64_t count_directory(char * source) {
+	DIR * dirp = opendir(source);
+	if (dirp == NULL) {
+		//fprintf(stderr, "could not open %s\n", source);
+		return 0;
+	}
+
+	uint64_t total = 0;
+
+	struct dirent * ent = readdir(dirp);
+	while (ent != NULL) {
+		if (!strcmp(ent->d_name,".") || !strcmp(ent->d_name,"..")) {
+			ent = readdir(dirp);
+			continue;
+		}
+		char tmp[strlen(source)+strlen(ent->d_name)+2];
+		sprintf(tmp, "%s/%s", source, ent->d_name);
+		total += count_thing(tmp);
+		ent = readdir(dirp);
+	}
+	closedir(dirp);
+
+	if (all) {
+		print_size(total, source);
+	}
+
+	return total;
+}
+
+static uint64_t count_thing(char * tmp) {
+	struct stat statbuf;
+	lstat(tmp,&statbuf);
+	if (S_ISDIR(statbuf.st_mode)) {
+		return count_directory(tmp);
+	} else {
+		return statbuf.st_size;
+	}
+}
+
+
+int main(int argc, char * argv[]) {
+	int opt;
+	while ((opt = getopt(argc, argv, "hs")) != -1) {
+		switch (opt) {
+			case 'h': /* human readable */
+				human = 1;
+				break;
+			case 's': /* summary */
+				all = 0;
+				break;
+			default:
+				fprintf(stderr, "rm: unrecognized option '%c'\n", opt);
+				break;
+		}
+	}
+
+	int ret = 0;
+
+	for (int i = optind; i < argc; ++i) {
+		uint64_t total = count_thing(argv[i]);
+		if (!all) {
+			print_size(total, argv[i]);
+		}
+	}
+
+	return ret;
+}
+