133 lines
2.7 KiB
C
133 lines
2.7 KiB
C
/* 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 show_total = 0;
|
|
static int human = 0;
|
|
static int all = 1;
|
|
static int is_arg = 0;
|
|
|
|
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);
|
|
}
|
|
if (strlen(name) > 2 && name[0] == '/' && name[1] == '/') {
|
|
name = &name[1];
|
|
}
|
|
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;
|
|
}
|
|
|
|
int was_arg = is_arg;
|
|
is_arg = 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 || was_arg) {
|
|
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 {
|
|
if (is_arg) {
|
|
print_size(statbuf.st_size, tmp);
|
|
}
|
|
return statbuf.st_size;
|
|
}
|
|
}
|
|
|
|
|
|
int main(int argc, char * argv[]) {
|
|
int opt;
|
|
while ((opt = getopt(argc, argv, "hsc")) != -1) {
|
|
switch (opt) {
|
|
case 'h': /* human readable */
|
|
human = 1;
|
|
break;
|
|
case 'c':
|
|
show_total = 1;
|
|
break;
|
|
case 's': /* summary */
|
|
all = 0;
|
|
break;
|
|
default:
|
|
fprintf(stderr, "rm: unrecognized option '%c'\n", opt);
|
|
break;
|
|
}
|
|
}
|
|
|
|
int ret = 0;
|
|
uint64_t total = 0;
|
|
|
|
for (int i = optind; i < argc; ++i) {
|
|
is_arg = 1;
|
|
total += count_thing(argv[i]);
|
|
}
|
|
|
|
if (show_total) {
|
|
print_size(total, "total");
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|