X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;ds=inline;f=dus.c;h=88221f67b09dd67e3d48a290f3a656fec1ea9510;hb=e96b00b6716ceb5046d2e3592834cd167b05ab61;hp=a4b5d0acc1906b243786ee42c16604e3b535a596;hpb=4b8de32c8576d189eeef5ab22ee70f380d001613;p=dus.git diff --git a/dus.c b/dus.c index a4b5d0a..88221f6 100644 --- a/dus.c +++ b/dus.c @@ -34,12 +34,27 @@ #include #include #include +#include #define BUFFER_SIZE 4096 typedef int64_t size_sum_t; -int ignore_dotfiles = 1; +/* Yeah, global variables! */ + +int ignore_dotfiles = 0; /* 1 means ignore files and directories + starting with a dot */ + +int forced_height = 0; /* -1 means no height limit, strictly positive + means limit, 0 means not active */ + +int fancy_size_display = 0; /* 1 means to use floating values with K, M and G + as units */ + +int reverse_sorting = 0; /* 1 means to show the large ones first */ + +int show_top = 0; /* 1 means to show the top of the sorted list + instead of the bottom */ /********************************************************************/ @@ -50,6 +65,21 @@ int ignore_entry(const char *name) { (ignore_dotfiles && name[0] == '.'); } +void print_size_sum(size_sum_t s) { + char tmp[100]; + char *a = tmp + sizeof(tmp)/sizeof(char); + *(--a) = '\0'; + if(s) { + while(s) { + *(--a) = s%10 + '0'; + s /= 10; + } + } else { + *(--a) = '0'; + } + printf(a); +} + size_sum_t file_or_dir_size(const char *name) { DIR *dir; struct dirent *dir_e; @@ -61,7 +91,7 @@ size_sum_t file_or_dir_size(const char *name) { if(lstat(name, &dummy) != 0) { printf("Can not stat %s: %s\n", name, strerror(errno)); - exit (1); + exit(EXIT_FAILURE); } if(S_ISLNK(dummy.st_mode)) { @@ -80,7 +110,6 @@ size_sum_t file_or_dir_size(const char *name) { closedir(dir); } else { if(S_ISREG(dummy.st_mode)) { - /* printf("%d %s\n", dummy.st_size, name); */ result += dummy.st_size; } } @@ -123,19 +152,29 @@ int compare_files(const void *x1, const void *x2) { f1 = (const struct file_with_size **) x1; f2 = (const struct file_with_size **) x2; - if((*f1)->size < (*f2)->size) { - return -1; - } else if((*f1)->size > (*f2)->size) { - return 1; + if(reverse_sorting) { + if((*f1)->size < (*f2)->size) { + return 1; + } else if((*f1)->size > (*f2)->size) { + return -1; + } else { + return 0; + } } else { - return 0; + if((*f1)->size < (*f2)->size) { + return -1; + } else if((*f1)->size > (*f2)->size) { + return 1; + } else { + return 0; + } } } void print_sorted(struct file_with_size *root, int height) { struct file_with_size *node; struct file_with_size **nodes; - int nb, n, first; + int nb, n, first, last; nb = 0; for(node = root; node; node = node->next) { @@ -151,35 +190,47 @@ void print_sorted(struct file_with_size *root, int height) { qsort(nodes, nb, sizeof(struct file_with_size *), compare_files); - /* - for(n = 0; n < nb; n++) { - printf("%lld %s\n", - nodes[n]->size, - nodes[n]->filename); + first = 0; + last = nb; + + if(forced_height) { + height = forced_height; } - */ - first = 0; if(height > 0 && height < nb) { first = nb - height; } - for(n = first; n < nb; n++) { - if(nodes[n]->size < 1024) { - printf("% 7d %s\n", - ((int) nodes[n]->size), - nodes[n]->filename); - } else if(nodes[n]->size < 1024 * 1024) { - printf("% 6.1fK %s\n", - ((double) (nodes[n]->size))/(1024.0), - nodes[n]->filename); - } else if(nodes[n]->size < 1024 * 1024 * 1024) { - printf("% 6.1fM %s\n", - ((double) (nodes[n]->size))/(1024.0 * 1024), - nodes[n]->filename); - } else { - printf("% 6.1fG %s\n", - ((double) (nodes[n]->size))/(1024.0 * 1024.0 * 1024.0), - nodes[n]->filename); + + if(show_top) { + n = last; + last = nb - first; + first = nb - n; + } + + if(fancy_size_display) { + for(n = first; n < last; n++) { + if(nodes[n]->size < 1024) { + printf("% 7d %s\n", + ((int) nodes[n]->size), + nodes[n]->filename); + } else if(nodes[n]->size < 1024 * 1024) { + printf("% 6.1fK %s\n", + ((double) (nodes[n]->size))/(1024.0), + nodes[n]->filename); + } else if(nodes[n]->size < 1024 * 1024 * 1024) { + printf("% 6.1fM %s\n", + ((double) (nodes[n]->size))/(1024.0 * 1024), + nodes[n]->filename); + } else { + printf("% 6.1fG %s\n", + ((double) (nodes[n]->size))/(1024.0 * 1024.0 * 1024.0), + nodes[n]->filename); + } + } + } else { + for(n = first; n < last; n++) { + print_size_sum(nodes[n]->size); + printf(" %s\n", nodes[n]->filename); } } @@ -189,16 +240,67 @@ void print_sorted(struct file_with_size *root, int height) { /**********************************************************************/ int main(int argc, char **argv) { - int k; + int c; struct file_with_size *root; root = 0; setlocale (LC_ALL, ""); - if(argc > 1) { - for(k = 1; k < argc; k++) { - root = create(argv[k], root); + while (1) { + c = getopt(argc, argv, "dfrtl:hdu"); + if (c == -1) + break; + + switch (c) { + + case 'd': + ignore_dotfiles = 1; + break; + + case 'f': + fancy_size_display = 1; + break; + + case 'r': + reverse_sorting = 1; + break; + + case 't': + show_top = 1; + break; + + case 'l': + forced_height = atoi(optarg); + break; + + case 'h': + printf("Usage: dus [OPTION]... [FILE]...\n"); + printf("List files and directories sorted according to their size or content size (the content of the current directory by default).\n"); + printf("\n"); + printf(" -d ignore files and directories starting with a '.'\n"); + printf(" -f display size with float values and K, M and G units.\n"); + printf(" -r reverse the sorting order.\n"); + printf(" -t show the top of the list.\n"); + printf(" -l specificy the number of lines to display. The value -1\n"); + printf(" corresponds to all the lines. By default the command\n"); + printf(" shows one line less than the height the tty, or all the\n"); + printf(" lines if the standard output is not a tty.\n"); + printf(" -h show this help.\n"); + printf("\n"); + printf("Report bugs and comments to \n"); + exit(EXIT_SUCCESS); + + break; + + default: + exit(EXIT_FAILURE); + } + } + + if (optind < argc) { + while (optind < argc) { + root = create(argv[optind++], root); } } else { DIR *dir; @@ -218,7 +320,7 @@ int main(int argc, char **argv) { struct winsize win; if(ioctl (STDOUT_FILENO, TIOCGWINSZ, (char *) &win)) { printf("Can not get the tty size: %s\n", strerror(errno)); - exit (1); + exit (EXIT_FAILURE); } print_sorted(root, win.ws_row - 2); } else { @@ -227,5 +329,5 @@ int main(int argc, char **argv) { destroy(root); - exit(0); + exit(EXIT_SUCCESS); }