int grokdir(char *dir, file_t **filelistp)
{
DIR *cd;
file_t *newfile;
struct dirent *dirinfo;
int lastchar;
int filecount = 0;
struct stat info;
struct stat linfo;
static int progress = 0;
static char indicator[] = "-\\|/";
cd = opendir(dir);
if (!cd) {
errormsg("could not chdir to %s\n", dir);
return 0;
}
while ((dirinfo = readdir(cd)) != NULL) {
if (strcmp(dirinfo->d_name, ".") && strcmp(dirinfo->d_name, "..")) {
if (!ISFLAG(flags, F_HIDEPROGRESS)) {
fprintf(stderr, "\rBuilding file list %c ", indicator[progress]);
progress = (progress + 1) % 4;
}
newfile = (file_t*) malloc(sizeof(file_t));
if (!newfile) {
errormsg("out of memory!\n");
closedir(cd);
exit(1);
} else newfile->next = *filelistp;
newfile->device = 0;
newfile->inode = 0;
newfile->crcsignature = NULL;
newfile->crcpartial = NULL;
newfile->duplicates = NULL;
newfile->hasdupes = 0;
newfile->d_name = (char*)malloc(strlen(dir)+strlen(dirinfo->d_name)+2);
if (!newfile->d_name) {
errormsg("out of memory!\n");
free(newfile);
closedir(cd);
exit(1);
}
strcpy(newfile->d_name, dir);
lastchar = strlen(dir) - 1;
if (lastchar >= 0 && dir[lastchar] != '/')
strcat(newfile->d_name, "/");
strcat(newfile->d_name, dirinfo->d_name);
if (filesize(newfile->d_name) == 0 && ISFLAG(flags, F_EXCLUDEEMPTY)) {
free(newfile->d_name);
free(newfile);
continue;
}
if (stat(newfile->d_name, &info) == -1) {
free(newfile->d_name);
free(newfile);
continue;
}
if (lstat(newfile->d_name, &linfo) == -1) {
free(newfile->d_name);
free(newfile);
continue;
}
if (S_ISDIR(info.st_mode)) {
if (ISFLAG(flags, F_RECURSE) && (ISFLAG(flags, F_FOLLOWLINKS) || !S_ISLNK(linfo.st_mode)))
filecount += grokdir(newfile->d_name, filelistp);
free(newfile->d_name);
free(newfile);
} else {
if (S_ISREG(linfo.st_mode) || (S_ISLNK(linfo.st_mode) && ISFLAG(flags, F_FOLLOWLINKS))) {
*filelistp = newfile;
filecount++;
} else {
free(newfile->d_name);
free(newfile);
}
}
}
}
closedir(cd);
return filecount;
}
int main(int argc, char **argv) {
int x;
int opt;
FILE *file1;
FILE *file2;
file_t *files = NULL;
file_t *curfile;
file_t **match = NULL;
filetree_t *checktree = NULL;
int filecount = 0;
int progress = 0;
char **oldargv;
int firstrecurse;
#ifndef OMIT_GETOPT_LONG
static struct option long_options[] =
{
{ "omitfirst", 0, 0, 'f' },
{ "recurse", 0, 0, 'r' },
{ "recursive", 0, 0, 'r' },
{ "recurse:", 0, 0, 'R' },
{ "recursive:", 0, 0, 'R' },
{ "quiet", 0, 0, 'q' },
{ "sameline", 0, 0, '1' },
{ "size", 0, 0, 'S' },
{ "symlinks", 0, 0, 's' },
{ "hardlinks", 0, 0, 'H' },
{ "relink", 0, 0, 'l' },
{ "noempty", 0, 0, 'n' },
{ "delete", 0, 0, 'd' },
{ "version", 0, 0, 'v' },
{ "help", 0, 0, 'h' },
{ "noprompt", 0, 0, 'N' },
{ "summarize", 0, 0, 'm'},
{ "summary", 0, 0, 'm' },
{ 0, 0, 0, 0 }
};
#define GETOPT getopt_long
#else
#define GETOPT getopt
#endif
program_name = argv[0];
oldargv = cloneargs(argc, argv);
while ((opt = GETOPT(argc, argv, "frRq1Ss::HlndvhNm"
#ifndef OMIT_GETOPT_LONG
, long_options, NULL
#endif
)) != EOF) {
switch (opt) {
case 'f':
SETFLAG(flags, F_OMITFIRST);
break;
case 'r':
SETFLAG(flags, F_RECURSE);
break;
case 'R':
SETFLAG(flags, F_RECURSEAFTER);
break;
case 'q':
SETFLAG(flags, F_HIDEPROGRESS);
break;
case '1':
SETFLAG(flags, F_DSAMELINE);
break;
case 'S':
SETFLAG(flags, F_SHOWSIZE);
break;
case 's':
SETFLAG(flags, F_FOLLOWLINKS);
break;
case 'H':
SETFLAG(flags, F_CONSIDERHARDLINKS);
break;
case 'n':
SETFLAG(flags, F_EXCLUDEEMPTY);
break;
case 'd':
SETFLAG(flags, F_DELETEFILES);
break;
case 'v':
printf("fdupes %s\n", VERSION);
exit(0);
case 'h':
help_text();
exit(1);
case 'N':
SETFLAG(flags, F_NOPROMPT);
break;
case 'm':
SETFLAG(flags, F_SUMMARIZEMATCHES);
break;
default:
fprintf(stderr, "Try `fdupes --help' for more information.\n");
exit(1);
}
}
if (optind >= argc) {
errormsg("no directories specified\n");
exit(1);
}
if (ISFLAG(flags, F_RECURSE) && ISFLAG(flags, F_RECURSEAFTER)) {
errormsg("options --recurse and --recurse: are not compatible\n");
exit(1);
}
if (ISFLAG(flags, F_SUMMARIZEMATCHES) && ISFLAG(flags, F_DELETEFILES)) {
errormsg("options --summarize and --delete are not compatible\n");
exit(1);
}
if (ISFLAG(flags, F_RECURSEAFTER)) {
firstrecurse = nonoptafter("--recurse:", argc, oldargv, argv, optind);
if (firstrecurse == argc)
firstrecurse = nonoptafter("-R", argc, oldargv, argv, optind);
if (firstrecurse == argc) {
errormsg("-R option must be isolated from other options\n");
exit(1);
}
/* F_RECURSE is not set for directories before --recurse: */
for (x = optind; x < firstrecurse; x++)
filecount += grokdir(argv[x], &files);
/* Set F_RECURSE for directories after --recurse: */
SETFLAG(flags, F_RECURSE);
for (x = firstrecurse; x < argc; x++)
filecount += grokdir(argv[x], &files);
} else {
for (x = optind; x < argc; x++)
filecount += grokdir(argv[x], &files);
}
if (!files) {
if (!ISFLAG(flags, F_HIDEPROGRESS)) fprintf(stderr, "\r%40s\r", " ");
exit(0);
}
curfile = files;
while (curfile) {
if (!checktree)
registerfile(&checktree, curfile);
else
match = checkmatch(&checktree, checktree, curfile);
if (match != NULL) {
file1 = fopen(curfile->d_name, "rb");
if (!file1) {
curfile = curfile->next;
continue;
}
file2 = fopen((*match)->d_name, "rb");
if (!file2) {
fclose(file1);
curfile = curfile->next;
continue;
}
if (confirmmatch(file1, file2)) {
registerpair(match, curfile, sort_pairs_by_mtime);
/*match->hasdupes = 1;
curfile->duplicates = match->duplicates;
match->duplicates = curfile;*/
}
fclose(file1);
fclose(file2);
}
curfile = curfile->next;
if (!ISFLAG(flags, F_HIDEPROGRESS)) {
fprintf(stderr, "\rProgress [%d/%d] %d%% ", progress, filecount,
(int)((float) progress / (float) filecount * 100.0));
progress++;
}
}
if (!ISFLAG(flags, F_HIDEPROGRESS)) fprintf(stderr, "\r%40s\r", " ");
if (ISFLAG(flags, F_DELETEFILES))
{
if (ISFLAG(flags, F_NOPROMPT))
{
deletefiles(files, 0, 0);
}
else
{
stdin = freopen("/dev/tty", "r", stdin);
deletefiles(files, 1, stdin);
}
}
else
if (ISFLAG(flags, F_SUMMARIZEMATCHES))
summarizematches(files);
else
printmatches(files);
while (files) {
curfile = files->next;
free(files->d_name);
free(files->crcsignature);
free(files->crcpartial);
free(files);
files = curfile;
}
for (x = 0; x < argc; x++)
free(oldargv[x]);
free(oldargv);
purgetree(checktree);
return 0;
}
int main(int argc, char **argv) {
int x;
int opt;
FILE *file1;
FILE *file2;
file_t *files = NULL;
file_t *curfile;
file_t *match = NULL;
filetree_t *checktree = NULL;
int filecount = 0;
int progress = 0;
static struct option long_options[] =
{
{ "omitfirst", 0, 0, 'f' },
{ "recurse", 0, 0, 'r' },
{ "quiet", 0, 0, 'q' },
{ "sameline", 0, 0, '1' },
{ "size", 0, 0, 'S' },
{ "symlinks", 0, 0, 's' },
{ "hardlinks", 0, 0, 'H' },
{ "noempty", 0, 0, 'n' },
{ "delete", 0, 0, 'd' },
{ "version", 0, 0, 'v' },
{ "help", 0, 0, 'h' },
{ 0, 0, 0, 0 }
};
program_name = argv[0];
while ((opt = getopt_long(argc, argv, "frq1SsHndvh", long_options, NULL)) != EOF) {
switch (opt) {
case 'f':
SETFLAG(flags, F_OMITFIRST);
break;
case 'r':
SETFLAG(flags, F_RECURSE);
break;
case 'q':
SETFLAG(flags, F_HIDEPROGRESS);
break;
case '1':
SETFLAG(flags, F_DSAMELINE);
break;
case 'S':
SETFLAG(flags, F_SHOWSIZE);
break;
case 's':
SETFLAG(flags, F_FOLLOWLINKS);
break;
case 'H':
SETFLAG(flags, F_CONSIDERHARDLINKS);
break;
case 'n':
SETFLAG(flags, F_EXCLUDEEMPTY);
break;
case 'd':
SETFLAG(flags, F_DELETEFILES);
break;
case 'v':
printf("fdupes %s\n", VERSION);
exit(0);
case 'h':
help_text();
exit(1);
default:
fprintf(stderr, "Try `fdupes --help' for more information\n");
exit(1);
}
}
if (optind >= argc) {
errormsg("no directories specified\n");
exit(1);
}
for (x = optind; x < argc; x++) filecount += grokdir(argv[x], &files);
if (!files) exit(0);
curfile = files;
while (curfile) {
if (!checktree)
#ifndef EXPERIMENTAL_RBTREE
registerfile(&checktree, curfile);
#else
registerfile(&checktree, NULL, TREE_ROOT, curfile);
#endif
else
match = checkmatch(&checktree, checktree, curfile);
if (match != NULL) {
file1 = fopen(curfile->d_name, "rb");
if (!file1) {
curfile = curfile->next;
continue;
}
file2 = fopen(match->d_name, "rb");
if (!file2) {
fclose(file1);
curfile = curfile->next;
continue;
}
if (confirmmatch(file1, file2)) {
match->hasdupes = 1;
curfile->duplicates = match->duplicates;
match->duplicates = curfile;
}
fclose(file1);
fclose(file2);
}
curfile = curfile->next;
if (!ISFLAG(flags, F_HIDEPROGRESS)) {
fprintf(stderr, "\rProgress [%d/%d] %d%% ", progress, filecount,
(int)((float) progress / (float) filecount * 100.0));
progress++;
}
}
