* to several directories, or files present in one directory and not
* in another one.
*
- * Copyright (c) 2010 Francois Fleuret
+ * Copyright (c) 2010, 2011 Francois Fleuret
* Written by Francois Fleuret <francois@fleuret.org>
*
* This file is part of finddup.
*
*/
-#define VERSION_NUMBER "1.1"
+#define VERSION_NUMBER "1.2"
#define _BSD_SOURCE
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/param.h>
+#include <sys/wait.h>
#include <dirent.h>
#include <stdlib.h>
#include <stdio.h>
int sort_by_time = 0; /* 1 means to sort files in each group according
to the modification time */
+char *command_to_exec = 0; /* the name of the command to exec for each
+ group of identical files */
+
+char *result_file_prefix = 0; /* The prefix to use to write result */
+
/********************************************************************/
/* malloc with error checking. */
void *safe_malloc(size_t n) {
void *p = malloc(n);
if (!p && n != 0) {
- printf("Can not allocate memory: %s\n", strerror(errno));
+ fprintf(stderr,
+ "finddup: Can not allocate memory: %s\n", strerror(errno));
exit(EXIT_FAILURE);
}
return p;
}
} else {
fprintf(stderr,
- "Different read size without error on files of same size.\n");
+ "finddup: Different read size without error on files of same size.\n");
exit(EXIT_FAILURE);
}
}
if(fd1 < 0) {
fprintf(stderr,
- "Can not open \"%s\" error: %s\n",
+ "finddup: Can not open \"%s\" error: %s\n",
f1->name,
strerror(errno));
}
if(fd2 < 0) {
fprintf(stderr,
- "Can not open \"%s\" error: %s\n",
+ "finddup: Can not open \"%s\" error: %s\n",
f2->name,
strerror(errno));
}
/**********************************************************************/
-struct file_node *scan_directory(struct file_node *tail, const char *name) {
+struct file_node *scan_directory_rec(struct file_node *tail, const char *name) {
DIR *dir;
struct dirent *dir_e;
struct stat sb;
char subname[PATH_MAX + 1];
if(lstat(name, &sb) != 0) {
- fprintf(stderr, "Can not stat \"%s\": %s\n", name, strerror(errno));
+ fprintf(stderr, "finddup: Can not stat \"%s\": %s\n", name, strerror(errno));
exit(EXIT_FAILURE);
}
while((dir_e = readdir(dir))) {
if(!ignore_entry(dir_e->d_name)) {
snprintf(subname, PATH_MAX, "%s/%s", name, dir_e->d_name);
- tail = scan_directory(tail, subname);
+ tail = scan_directory_rec(tail, subname);
}
}
closedir(dir);
return tail;
}
-void print_file(struct file_node *node) {
+struct file_node *scan_directory(struct file_node *tail, const char *name) {
+ struct file_node *result;
+ int length;
+
+ if(show_progress) {
+ fprintf(stderr, "Scanning '%s' ... ", name);
+ }
+
+ result = scan_directory_rec(tail, name);
+ length = file_list_length(result);
+
+ if(show_progress) {
+ fprintf(stderr, "done (%d file%s).\n",
+ length, (length > 1 ? "s" : ""));
+ }
+
+
+ return result;
+}
+
+void print_file(FILE *out, struct file_node *node) {
char tmp[PATH_MAX + 1];
if(show_realpaths) {
if(realpath(node->name, tmp)) {
if(show_groups) {
- printf("%d %s\n", node->group_id, tmp);
+ fprintf(out, "%d %s\n", node->group_id, tmp);
} else {
- printf("%s\n", tmp);
+ fprintf(out, "%s\n", tmp);
}
} else {
- printf("Can not get the realpath of \"%s\": %s\n",
- node->name,
- strerror(errno));
+ fprintf(stderr,
+ "finddup: Can not get the realpath of \"%s\": %s\n",
+ node->name,
+ strerror(errno));
exit(EXIT_FAILURE);
}
} else {
if(show_groups) {
- printf("%d %s\n", node->group_id, node->name);
+ fprintf(out, "%d %s\n", node->group_id, node->name);
} else {
- printf("%s\n", node->name);
+ fprintf(out, "%s\n", node->name);
}
}
}
}
}
+void exec_command(int nb, struct file_node **nodes) {
+ char **args;
+ int max_group_size = 0, group_size, m, n, status;
+ pid_t pid;
+
+ for(n = 0; n < nb; n++) {
+ if(n == 0 || nodes[n]->group_id != nodes[n-1]->group_id) {
+ group_size = 0;
+ }
+ group_size++;
+ if(group_size > max_group_size) {
+ max_group_size = group_size;
+ }
+ }
+
+ args = safe_malloc((max_group_size + 2) * sizeof(char *));
+ args[0] = command_to_exec;
+
+ n = 0;
+ while(n < nb) {
+ m = n;
+ while(n < nb && nodes[n]->group_id == nodes[m]->group_id) {
+ args[n - m + 1] = nodes[n]->name;
+ n++;
+ }
+ args[n - m + 1] = 0;
+ pid = fork();
+ if(pid < 0) {
+ } else if(pid == 0) {
+ if(execvp(command_to_exec, args) < 0) {
+ perror("execvp");
+ exit(EXIT_FAILURE);
+ }
+ } else {
+ while(wait(&status) != pid);
+ if(status > 0) {
+ exit(EXIT_FAILURE);
+ }
+ }
+ }
+
+ free(args);
+}
+
+void write_groups_in_files(int nb, struct file_node **nodes) {
+ FILE *file = 0;
+ int current_group = -1, n;
+ char filename[PATH_MAX + 1];
+
+ for(n = 0; n < nb; n++) {
+ if(nodes[n]->group_id != current_group) {
+ if(file) { fclose(file); }
+ sprintf(filename, "%s%06d", result_file_prefix, nodes[n]->group_id);
+ file = fopen(filename, "w");
+ current_group = nodes[n]->group_id;
+ printf("Writing %s.\n" , filename);
+ }
+ print_file(file, nodes[n]);
+ }
+
+ if(file) { fclose(file); }
+}
void print_result(struct file_node *list1, struct file_node *list2) {
struct file_node *node1, *node2;
qsort(nodes, nb, sizeof(struct file_node *), compare_nodes);
- for(n = 0; n < nb; n++) {
- if(!show_groups && n > 0 && nodes[n]->group_id != nodes[n-1]->group_id) {
- printf("\n");
+ if(command_to_exec) {
+ exec_command(nb, nodes);
+ } else if(result_file_prefix) {
+ write_groups_in_files(nb, nodes);
+ } else {
+ for(n = 0; n < nb; n++) {
+ if(!show_groups && n > 0 && nodes[n]->group_id != nodes[n-1]->group_id) {
+ printf("\n");
+ }
+ print_file(stdout, nodes[n]);
}
- print_file(nodes[n]);
}
free(nodes);
};
void print_progress(struct progress_state *state) {
- int position, k;
+ int position, k, normalizer;
struct winsize win;
char buffer[PROGRESS_BUFFER_SIZE];
char *s;
+ normalizer = (state->nb_values > 1 ? state->nb_values - 1 : 1);
+
if(show_progress) {
/* We use the previous bar_width to compute the position, so that
we avoid doing too many ioctls */
- position = (state->bar_width * state->value) / (state->nb_values - 1);
+ position = (state->bar_width * state->value) / normalizer;
if(state->bar_width <= 0 || position != state->last_position) {
if(!ioctl (STDERR_FILENO, TIOCGWINSZ, (char *) &win)) {
/* Something weird is going on if the previous test is wrong */
} else {
state->bar_width = win.ws_col - 7;
}
- position = (state->bar_width * state->value) / (state->nb_values - 1);
+ position = (state->bar_width * state->value) / normalizer;
state->last_position = position;
s = buffer;
for(k = 0; k < position; k++) {
}
/* We need four % because of the fprintf that follows */
- sprintf(s, " [%3d%%%%]\r",
- (100 * state->value) / (state->nb_values - 1));
+ sprintf(s, " [%3d%%]\r",
+ (100 * state->value) / normalizer);
- fprintf(stderr, buffer);
+ fprintf(stderr, "%s", buffer);
}
}
}
struct progress_state progress_state;
int not_in, found;
int nb_groups, nb_nodes;
- int list1_length, list2_length, previous_progress;
+ 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;
- if(show_progress) {
- fprintf(stderr, "Scanning '%s' ... ", dirname1);
- }
-
list1 = scan_directory(0, dirname1);
-
list1_length = file_list_length(list1);
- if(show_progress) {
- fprintf(stderr, "done (%d file%s).\n",
- list1_length, (list1_length > 1 ? "s" : ""));
- }
-
if(dirname2) {
if(strncmp(dirname2, "not:", 4) == 0) {
not_in = 1;
} else if(strncmp(dirname2, "and:", 4) == 0) {
dirname2 += 4;
}
- if(show_progress) {
- fprintf(stderr, "Scanning '%s' ... ", dirname2);
- }
list2 = scan_directory(0, dirname2);
- if(show_progress) {
- list2_length = file_list_length(list2);
- fprintf(stderr, "done (%d file%s).\n",
- list2_length, (list2_length > 1 ? "s" : ""));
- }
} else {
list2 = list1;
}
if(show_progress) {
- fprintf(stderr, "Now looking for identical files (this may take a while).\n");
+ fprintf(stderr,
+ "Now looking for identical files (this may take a while).\n");
}
nb_groups = 0;
fprintf(out, "Without DIR2, lists duplicated files found in DIR1, or the current directory if DIR1 is not provided. 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");
/* 01234567890123456789012345678901234567890123456789012345678901234567890123456789*/
+ fprintf(out, " -v, --version prints the version number and exit\n");
fprintf(out, " -h, --help show this help\n");
fprintf(out, " -d, --ignore-dots ignore dot files and directories\n");
fprintf(out, " -0, --ignore-empty ignore empty files\n");
fprintf(out, " -r, --real-paths 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, " -e <command>, --exec <command>\n");
+ fprintf(out, " execute the provided command for each group of\n");
+ fprintf(out, " identical files, with their names as arguments\n");
+ fprintf(out, " -f <string>, --result-prefix <string>\n");
+ fprintf(out, " for each group of identical files, write one\n");
+ fprintf(out, " result file whose name is the given prefix string\n");
+ fprintf(out, " followed by the group number, and containing\n");
+ fprintf(out, " one filename per line\n");
fprintf(out, "\n");
fprintf(out, "Report bugs and comments to <francois@fleuret.org>.\n");
}
/**********************************************************************/
static struct option long_options[] = {
+ { "version", no_argument, 0, 'v' },
{ "help", no_argument, 0, 'h' },
{ "same-inodes-are-different", no_argument, 0, 'i' },
{ "real-paths", no_argument, 0, 'r' },
{ "ignore-dots", no_argument, 0, 'd' },
{ "ignore-empty", no_argument, 0, '0' },
{ "show-progress", no_argument, 0, 'p' },
+ { "exec", 1, 0, 'e' },
+ { "result-prefix", 1, 0, 'f' },
{ 0, 0, 0, 0 }
};
setlocale (LC_ALL, "");
- while ((c = getopt_long(argc, argv, "hircgtd0pm",
+ while ((c = getopt_long(argc, argv, "vhircgtd0pme:f:",
long_options, NULL)) != -1) {
switch (c) {
+ case 'v':
+ printf("finddup version %s (%s)\n", VERSION_NUMBER, UNAME);
+ exit(EXIT_SUCCESS);
+ break;
+
case 'h':
usage(stdout);
exit(EXIT_SUCCESS);
show_hits = 0;
break;
+ case 'e':
+ if(command_to_exec != 0) {
+ free(command_to_exec);
+ }
+ command_to_exec = strdup(optarg);
+ break;
+
+ case 'f':
+ if(result_file_prefix != 0) {
+ free(result_file_prefix);
+ }
+ result_file_prefix = strdup(optarg);
+ show_groups = 0;
+ break;
+
default:
usage(stderr);
exit(EXIT_FAILURE);