X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=finddup.c;h=9c713218902404e68789020a5ef2b6c25ab3cfd8;hb=f7bb05fb425e5ff6f1e9105cd36309786963825d;hp=f1e3cf0f007327d1d0ab56eac803f334348a0c73;hpb=f34775663d2b9b3a3cea95b0954e2e62dad1410e;p=finddup.git diff --git a/finddup.c b/finddup.c index f1e3cf0..9c71321 100644 --- a/finddup.c +++ b/finddup.c @@ -1,8 +1,8 @@ /* - * finddup is a simple utility find duplicated files, files common to - * several directories, or files present in one directory and not in - * another one. + * finddup is a simple utility to find duplicated files, files common + * to several directories, or files present in one directory and not + * in another one. * * Copyright (c) 2010 Francois Fleuret * Written by Francois Fleuret @@ -23,7 +23,7 @@ * */ -#define VERSION_NUMBER "0.6" +#define VERSION_NUMBER "0.7" #define _BSD_SOURCE @@ -41,8 +41,8 @@ #include #include -#define BUFFER_SIZE 4096 -#define LARGE_BUFFER_SIZE 65536 +/* 1M really helps compared to 64k */ +#define READ_BUFFER_SIZE (1024 * 1024) typedef int64_t size_sum_t; @@ -68,6 +68,9 @@ int show_groups = 1; /* 1 means to show the group IDs when printing int ignore_same_inode = 0; /* 1 means that comparison between two file with same inode will always be false */ +int tty_width = -1; /* Positive value means what width to use to show + the progress bar */ + /********************************************************************/ /* malloc with error checking. */ @@ -90,21 +93,6 @@ 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); -} - /**********************************************************************/ struct file_with_size { @@ -136,17 +124,17 @@ int file_list_length(struct file_with_size *head) { /**********************************************************************/ -int same_content(struct file_with_size *f1, struct file_with_size *f2) { +int same_content(struct file_with_size *f1, struct file_with_size *f2, + char *buffer1, char *buffer2) { int fd1, fd2, s1, s2; - char buffer1[LARGE_BUFFER_SIZE], buffer2[LARGE_BUFFER_SIZE]; fd1 = open(f1->filename, O_RDONLY); fd2 = open(f2->filename, O_RDONLY); if(fd1 >= 0 && fd2 >= 0) { while(1) { - s1 = read(fd1, buffer1, LARGE_BUFFER_SIZE); - s2 = read(fd2, buffer2, LARGE_BUFFER_SIZE); + s1 = read(fd1, buffer1, READ_BUFFER_SIZE); + s2 = read(fd2, buffer2, READ_BUFFER_SIZE); if(s1 < 0 || s2 < 0) { close(fd1); @@ -191,11 +179,12 @@ int same_content(struct file_with_size *f1, struct file_with_size *f2) { } } -int same_files(struct file_with_size *f1, struct file_with_size *f2) { +int same_files(struct file_with_size *f1, struct file_with_size *f2, + char *buffer1, char *buffer2) { if(ignore_same_inode && f1->inode == f2->inode) { return 0; } - return f1->size == f2->size && same_content(f1, f2); + return f1->size == f2->size && same_content(f1, f2, buffer1, buffer2); } /**********************************************************************/ @@ -250,12 +239,17 @@ struct file_with_size *scan_directory(struct file_with_size *tail, void print_file(struct file_with_size *node) { char tmp[PATH_MAX + 1]; if(show_realpaths) { - if(show_groups) { - realpath(node->filename, tmp); - printf("%d %s\n", node->group_id, tmp); + if(realpath(node->filename, tmp)) { + if(show_groups) { + printf("%d %s\n", node->group_id, tmp); + } else { + printf("%s\n", tmp); + } } else { - realpath(node->filename, tmp); - printf("%s\n", tmp); + printf("Can not get the realpath of \"%s\": %s\n", + node->filename, + strerror(errno)); + exit(EXIT_FAILURE); } } else { if(show_groups) { @@ -333,11 +327,35 @@ void print_result(struct file_with_size *list1, struct file_with_size *list2) { free(nodes); } +void print_progress(int max, int n, int *pp) { + int p, k; + int width; + if(show_progress && tty_width > 0) { + width = tty_width - 7; + p = (width * n) / (max - 1); + if(p > *pp) { + for(k = 0; k < p; k++) { + fprintf(stderr, "+"); + } + for(; k < width; k++) { + fprintf(stderr, "-"); + } + *pp = p; + p = (100 * n) / (max - 1); + fprintf(stderr, " [%3d%%]\r", p); + } + } +} + void start(const char *dirname1, const char *dirname2) { struct file_with_size *list1, *list2; struct file_with_size *node1, *node2; int not_in, found; - int k, p, pp, l1, n; + int nb_groups, nb_nodes; + int list1_length, previous_progress; + + char *buffer1 = safe_malloc(sizeof(char) * READ_BUFFER_SIZE); + char *buffer2 = safe_malloc(sizeof(char) * READ_BUFFER_SIZE); not_in = 0; @@ -350,10 +368,11 @@ void start(const char *dirname1, const char *dirname2) { if(dirname2) { if(strncmp(dirname2, "not:", 4) == 0) { not_in = 1; - /* We should show groups even in that mode. However they are not - properly calculated for now, so we force it off. */ + /* groups are not computed in the not: mode */ show_groups = 0; dirname2 += 4; + } else if(strncmp(dirname2, "and:", 4) == 0) { + dirname2 += 4; } if(show_progress) { fprintf(stderr, "%s ... ", dirname2); @@ -367,26 +386,20 @@ void start(const char *dirname1, const char *dirname2) { fprintf(stderr, "done.\n"); } - k = 0; - pp = -1; - n = 0; - l1 = file_list_length(list1); + nb_groups = 0; + previous_progress = -1; + nb_nodes = 0; + list1_length = file_list_length(list1); if(not_in) { for(node1 = list1; node1; node1 = node1->next) { - if(show_progress) { - p = (100 * n)/l1; - if(p > pp) { - fprintf(stderr, "%d%%\n", p); - pp = p; - } - n++; - } + print_progress(list1_length, nb_nodes, &previous_progress); + nb_nodes++; found = 0; for(node2 = list2; !found && node2; node2 = node2->next) { - if(same_files(node1, node2)) { + if(same_files(node1, node2, buffer1, buffer2)) { found = 1; } } @@ -402,25 +415,19 @@ void start(const char *dirname1, const char *dirname2) { } else { for(node1 = list1; node1; node1 = node1->next) { - if(show_progress) { - p = (100 * n)/l1; - if(p > pp) { - fprintf(stderr, "%d%%\n", p); - pp = p; - } - n++; - } + print_progress(list1_length, nb_nodes, &previous_progress); + nb_nodes++; for(node2 = list2; node2; node2 = node2->next) { if(node1->group_id < 0 || node2->group_id < 0) { - if(same_files(node1, node2)) { + if(same_files(node1, node2, buffer1, buffer2)) { if(node1->group_id < 0) { if(node2->group_id >= 0) { node1->group_id = node2->group_id; } else { - node1->group_id = k; + node1->group_id = nb_groups; node1->dir_id = 1; - k++; + nb_groups++; } } if(node2->group_id < 0) { @@ -433,6 +440,10 @@ void start(const char *dirname1, const char *dirname2) { } } + if(show_progress) { + fprintf(stderr, "\n"); + } + if(dirname2) { print_result(list1, list2); file_list_delete(list1); @@ -441,21 +452,32 @@ void start(const char *dirname1, const char *dirname2) { print_result(list1, 0); file_list_delete(list1); } + + free(buffer1); + free(buffer2); } void print_help(FILE *out) { - fprintf(out, "Usage: finddup [OPTION]... DIR1 [[not:]DIR2]\n"); + fprintf(out, "Usage: finddup [OPTION]... DIR1 [[and:|not:]DIR2]\n"); fprintf(out, "Version %s (%s)\n", VERSION_NUMBER, UNAME); - fprintf(out, "Without DIR2, lists duplicated files found in DIR1. With DIR2, lists files common to both directories. With the not: prefix, lists files found in DIR1 which do not exist in DIR2.\n"); + fprintf(out, "Without DIR2, lists duplicated files found in DIR1. With DIR2, lists files common to both directories. With the not: prefix, lists files found in DIR1 which do not exist in DIR2. The and: prefix is the default and should be used only if you have a directory starting with 'not:'\n"); fprintf(out, "\n"); - fprintf(out, " -h show this help\n"); - fprintf(out, " -d ignore dot files and directories\n"); - fprintf(out, " -0 ignore empty files\n"); - fprintf(out, " -c do not show which files in DIR2 corresponds to those in DIR1\n"); - fprintf(out, " -g do not show the file groups\n"); - fprintf(out, " -p show progress\n"); - fprintf(out, " -r show the real file paths\n"); - fprintf(out, " -i consider files with same inode as different\n"); + fprintf(out, " -h, --help\n"); + fprintf(out, " show this help\n"); + fprintf(out, " -d, --ignore-dots\n"); + fprintf(out, " ignore dot files and directories\n"); + fprintf(out, " -0, --ignore-empty\n"); + fprintf(out, " ignore empty files\n"); + fprintf(out, " -c, --hide-matchings\n"); + fprintf(out, " do not show which files in DIR2 corresponds to those in DIR1\n"); + fprintf(out, " -g, --no-group-ids\n"); + fprintf(out, " do not show the file groups\n"); + fprintf(out, " -p, --show-progress\n"); + fprintf(out, " show progress\n"); + fprintf(out, " -r, --real-paths\n"); + fprintf(out, " show the real file paths\n"); + fprintf(out, " -i, --same-inodes-are-different\n"); + fprintf(out, " consider files with same inode as different\n"); fprintf(out, "\n"); fprintf(out, "Report bugs and comments to \n"); } @@ -465,13 +487,28 @@ void print_help(FILE *out) { int main(int argc, char **argv) { int c; struct file_with_size *root; + struct winsize win; root = 0; setlocale (LC_ALL, ""); while (1) { - c = getopt(argc, argv, "hircgd0p"); + int option_index = 0; + static struct option long_options[] = { + { "help", no_argument, 0, 'h' }, + { "same-inodes-are-different", no_argument, 0, 'i' }, + { "real-paths", no_argument, 0, 'r' }, + { "hide-matchings", no_argument, 0, 'c' }, + { "no-group-ids", no_argument, 0, 'g' }, + { "ignore-dots", no_argument, 0, 'd' }, + { "ignore-empty", no_argument, 0, '0' }, + { "show-progress", no_argument, 0, 'p' }, + { 0, 0, 0, 0 } + }; + + c = getopt_long(argc, argv, "hircgd0p", + long_options, &option_index); if (c == -1) break; @@ -516,6 +553,12 @@ int main(int argc, char **argv) { } } + if(show_progress && + isatty(STDOUT_FILENO) && + !ioctl (STDOUT_FILENO, TIOCGWINSZ, (char *) &win)) { + tty_width = win.ws_col; + } + if(optind + 2 == argc) { start(argv[optind], argv[optind + 1]); } else if(optind + 1 == argc) {