Initial commit. v1.0
This commit is contained in:
commit
dd1c294798
347
backly.c
Normal file
347
backly.c
Normal file
@ -0,0 +1,347 @@
|
||||
#include <dirent.h>
|
||||
#include <errno.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <sys/stat.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/wait.h>
|
||||
#include <unistd.h>
|
||||
|
||||
void printUsage(FILE *stream)
|
||||
{
|
||||
fprintf(stream, "Usage: backly [options] <srcdir> <destdir>\n");
|
||||
}
|
||||
|
||||
void printHelp()
|
||||
{
|
||||
printUsage(stdout);
|
||||
printf("Turn destdir into a clone of srcdir.\n\n");
|
||||
|
||||
printf("Options:\n");
|
||||
printf(" --test\tRun in test mode.\n");
|
||||
printf(" --help\tPrint this help and quit.\n\n");
|
||||
|
||||
printf("Report bugs to me@bradleylaboon.com\n");
|
||||
printf("backly home page: <http://www.bradleylaboon.com/projects/backly>\n");
|
||||
}
|
||||
|
||||
void removeMissing(char *src, int srcPrefix, char *dest, int destPrefix, int testMode)
|
||||
{
|
||||
int srcLen = 0, destLen = 0;
|
||||
while (src[srcLen])
|
||||
srcLen++;
|
||||
while (dest[destLen])
|
||||
destLen++;
|
||||
|
||||
DIR *destDir = opendir(dest);
|
||||
if (destDir == NULL) {
|
||||
fprintf(stderr, "Could not open %s: %s\n", dest, strerror(errno));
|
||||
return;
|
||||
}
|
||||
|
||||
// If the entire directory doesn't exist in src, remove it from dest
|
||||
DIR *srcDir = opendir(src);
|
||||
if (srcDir == NULL) {
|
||||
if (errno == ENOENT) {
|
||||
printf("- %s", dest + destPrefix);
|
||||
fflush(stdout);
|
||||
if (testMode == 0) {
|
||||
int pid = fork();
|
||||
if (pid == -1) {
|
||||
fprintf(stderr, "Could not fork: %s\n", strerror(errno));
|
||||
closedir(destDir);
|
||||
return;
|
||||
} else if (pid == 0) {
|
||||
execlp("rm", "rm", "-rf", dest, (char *) NULL);
|
||||
} else {
|
||||
wait(NULL);
|
||||
}
|
||||
}
|
||||
printf("\n");
|
||||
closedir(destDir);
|
||||
return;
|
||||
} else {
|
||||
fprintf(stderr, "Could not open %s: %s\n", src, strerror(errno));
|
||||
closedir(destDir);
|
||||
return;
|
||||
}
|
||||
}
|
||||
closedir(srcDir);
|
||||
|
||||
// Look at each item in the folder
|
||||
struct dirent *item;
|
||||
while ((item = readdir(destDir)) != NULL) {
|
||||
// Ignore . and .. references
|
||||
if (strcmp(item->d_name, ".") == 0 || strcmp(item->d_name, "..") == 0)
|
||||
continue;
|
||||
|
||||
int itemLen = 0;
|
||||
while (item->d_name[itemLen])
|
||||
itemLen++;
|
||||
|
||||
// +2 because a trailing slash might be added
|
||||
char itemSrc[srcLen + itemLen + 2];
|
||||
char itemDest[destLen + itemLen + 2];
|
||||
sprintf(itemSrc, "%s%s", src, item->d_name);
|
||||
sprintf(itemDest, "%s%s", dest, item->d_name);
|
||||
|
||||
// Test if item is a directory
|
||||
int itemIsDir = 1;
|
||||
DIR *testDir = opendir(itemDest);
|
||||
if (testDir == NULL && errno == ENOTDIR)
|
||||
itemIsDir = 0;
|
||||
else if (testDir != NULL)
|
||||
closedir(testDir);
|
||||
|
||||
if (itemIsDir == 1) {
|
||||
// If it's a directory, append trailing slashes and recurse
|
||||
itemSrc[srcLen + itemLen] = '/';
|
||||
itemSrc[srcLen + itemLen + 1] = '\0';
|
||||
itemDest[destLen + itemLen] = '/';
|
||||
itemDest[destLen + itemLen + 1] = '\0';
|
||||
removeMissing(itemSrc, srcPrefix, itemDest, destPrefix, testMode);
|
||||
} else {
|
||||
// If the file doesn't exist in src, remove it from dest
|
||||
FILE *srcFile = fopen(itemSrc, "r");
|
||||
if (srcFile == NULL) {
|
||||
if (errno == ENOENT) {
|
||||
printf("- %s", itemDest + destPrefix);
|
||||
fflush(stdout);
|
||||
if (testMode == 0) {
|
||||
int pid = fork();
|
||||
if (pid == -1) {
|
||||
fprintf(stderr, "Could not fork: %s\n", strerror(errno));
|
||||
continue;
|
||||
} else if (pid == 0) {
|
||||
execlp("rm", "rm", "-f", itemDest, (char *) NULL);
|
||||
} else {
|
||||
wait(NULL);
|
||||
}
|
||||
}
|
||||
printf("\n");
|
||||
continue;
|
||||
} else {
|
||||
fprintf(stderr, "Could not open %s: %s\n", itemSrc, strerror(errno));
|
||||
continue;
|
||||
}
|
||||
} else {
|
||||
fclose(srcFile);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
closedir(destDir);
|
||||
return;
|
||||
}
|
||||
|
||||
void copyNew(char *src, int srcPrefix, char *dest, int destPrefix, int testMode)
|
||||
{
|
||||
int srcLen = 0, destLen = 0;
|
||||
while (src[srcLen])
|
||||
srcLen++;
|
||||
while(dest[destLen])
|
||||
destLen++;
|
||||
|
||||
DIR *srcDir = opendir(src);
|
||||
if (srcDir == NULL) {
|
||||
fprintf(stderr, "Could not open %s: %s\n", src, strerror(errno));
|
||||
return;
|
||||
}
|
||||
|
||||
// If the dest directory doesn't exist, create it
|
||||
DIR *destDir = opendir(dest);
|
||||
if (destDir == NULL) {
|
||||
if (errno == ENOENT) {
|
||||
printf("+ %s", dest + destPrefix);
|
||||
fflush(stdout);
|
||||
if (testMode == 0) {
|
||||
int pid = fork();
|
||||
if (pid == -1) {
|
||||
fprintf(stderr, "Could not fork: %s:\n", strerror(errno));
|
||||
closedir(srcDir);
|
||||
return;
|
||||
} else if (pid == 0) {
|
||||
execlp("mkdir", "mkdir", dest, (char *) NULL);
|
||||
} else {
|
||||
wait(NULL);
|
||||
}
|
||||
}
|
||||
printf("\n");
|
||||
} else {
|
||||
fprintf(stderr, "Could not open %s: %s\n", dest, strerror(errno));
|
||||
closedir(srcDir);
|
||||
return;
|
||||
}
|
||||
}
|
||||
closedir(destDir);
|
||||
|
||||
// Look at each item in the folder
|
||||
struct dirent *item;
|
||||
while ((item = readdir(srcDir)) != NULL) {
|
||||
if (strcmp(item->d_name, ".") == 0 || strcmp(item->d_name, "..") == 0)
|
||||
continue;
|
||||
|
||||
int itemLen = 0;
|
||||
while (item->d_name[itemLen])
|
||||
itemLen++;
|
||||
|
||||
char itemSrc[srcLen + itemLen + 2];
|
||||
char itemDest[destLen + itemLen + 2];
|
||||
sprintf(itemSrc, "%s%s", src, item->d_name);
|
||||
sprintf(itemDest, "%s%s", dest, item->d_name);
|
||||
|
||||
// Test if item is a directory
|
||||
int itemIsDir = 1;
|
||||
DIR *testDir = opendir(itemSrc);
|
||||
if (testDir == NULL && errno == ENOTDIR)
|
||||
itemIsDir = 0;
|
||||
else if (testDir != NULL)
|
||||
closedir(testDir);
|
||||
|
||||
if (itemIsDir == 1) {
|
||||
// If it's a directory, recurse
|
||||
itemSrc[srcLen + itemLen] = '/';
|
||||
itemSrc[srcLen + itemLen + 1] = '\0';
|
||||
itemDest[destLen + itemLen] = '/';
|
||||
itemDest[destLen + itemLen + 1] = '\0';
|
||||
copyNew(itemSrc, srcPrefix, itemDest, destPrefix, testMode);
|
||||
} else {
|
||||
// Check if file exists in dest
|
||||
int needToCopy = 0;
|
||||
FILE *destFile = fopen(itemDest, "r");
|
||||
if (destFile == NULL) {
|
||||
if (errno == ENOENT) {
|
||||
needToCopy = 1;
|
||||
} else {
|
||||
fprintf(stderr, "Could not open %s: %s\n", itemDest, strerror(errno));
|
||||
continue;
|
||||
}
|
||||
} else {
|
||||
fclose(destFile);
|
||||
|
||||
// Check if file size or modified time is different
|
||||
struct stat srcStat, destStat;
|
||||
if (stat(itemSrc, &srcStat) == -1) {
|
||||
fprintf(stderr, "Could not stat %s\n", itemSrc);
|
||||
continue;
|
||||
}
|
||||
if (stat(itemDest, &destStat) == -1) {
|
||||
fprintf(stderr, "Could not stat %s\n", itemDest);
|
||||
continue;
|
||||
}
|
||||
|
||||
if (srcStat.st_size != destStat.st_size)
|
||||
needToCopy = 1;
|
||||
else if (srcStat.st_mtime > destStat.st_mtime)
|
||||
needToCopy = 1;
|
||||
}
|
||||
|
||||
// Only copy file if it doesn't exist or has changed
|
||||
if (needToCopy == 1) {
|
||||
printf("+ %s ", itemSrc + srcPrefix);
|
||||
fflush(stdout);
|
||||
if (testMode == 0) {
|
||||
int pid = fork();
|
||||
if (pid == -1) {
|
||||
fprintf(stderr, "Could not fork: %s\n", strerror(errno));
|
||||
continue;
|
||||
} else if (pid == 0) {
|
||||
execlp("cp", "cp", "-fp", itemSrc, itemDest, (char *) NULL);
|
||||
} else {
|
||||
struct stat srcStat, destStat;
|
||||
double pDone = 0;
|
||||
int numPrinted = 0;
|
||||
if (stat(itemSrc, &srcStat) != -1) {
|
||||
while (waitpid(pid, NULL, WNOHANG) == 0) {
|
||||
if (stat(itemDest, &destStat) == -1)
|
||||
continue;
|
||||
|
||||
pDone = (double)destStat.st_size / (double)srcStat.st_size;
|
||||
|
||||
for (int i = 0; i < numPrinted; i++)
|
||||
printf("\b \b");
|
||||
numPrinted = printf("%.2lf%%", pDone * 100);
|
||||
fflush(stdout);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
printf("\n");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
closedir(srcDir);
|
||||
return;
|
||||
}
|
||||
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
int srcArg = 0, destArg = 0;
|
||||
int testMode = 0;
|
||||
|
||||
// Parse arguments
|
||||
for (int i = 1; i < argc; i++) {
|
||||
if (strcmp(argv[i], "--help") == 0) {
|
||||
printHelp();
|
||||
exit(0);
|
||||
} else if (strcmp(argv[i], "--test") == 0) {
|
||||
testMode = 1;
|
||||
printf("***Operating in test mode.***\n");
|
||||
} else if (srcArg == 0) {
|
||||
srcArg = i;
|
||||
} else {
|
||||
destArg = i;
|
||||
}
|
||||
}
|
||||
|
||||
if (srcArg == 0 || destArg == 0) {
|
||||
printUsage(stderr);
|
||||
fprintf(stderr, "Run 'backly --help' for more information.\n");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
// Determine argument lengths
|
||||
int srcLen = 0, destLen = 0;
|
||||
while (argv[srcArg][srcLen])
|
||||
srcLen++;
|
||||
while (argv[destArg][destLen])
|
||||
destLen++;
|
||||
|
||||
// Add trailing slashes to directory arguments if necessary
|
||||
char srcDir[srcLen + 2], destDir[destLen + 2];
|
||||
|
||||
if (argv[srcArg][srcLen - 1] == '/')
|
||||
sprintf(srcDir, "%s", argv[srcArg]);
|
||||
else
|
||||
sprintf(srcDir, "%s/", argv[srcArg]);
|
||||
|
||||
if (argv[destArg][destLen - 1] == '/')
|
||||
sprintf(destDir, "%s", argv[destArg]);
|
||||
else
|
||||
sprintf(destDir, "%s/", argv[destArg]);
|
||||
|
||||
// Check if directories exist and can be accessed
|
||||
DIR *dir = opendir(srcDir);
|
||||
if (dir == NULL) {
|
||||
fprintf(stderr, "Could not open source: %s\n", strerror(errno));
|
||||
exit(1);
|
||||
}
|
||||
closedir(dir);
|
||||
|
||||
dir = opendir(destDir);
|
||||
if (dir == NULL) {
|
||||
fprintf(stderr, "Could not open destination: %s\n", strerror(errno));
|
||||
exit(1);
|
||||
}
|
||||
closedir(dir);
|
||||
|
||||
// Remove files from the destination that don't exist in the source
|
||||
removeMissing(srcDir, strlen(srcDir), destDir, strlen(destDir), testMode);
|
||||
|
||||
// Copy new files and overwrite existing files if different
|
||||
copyNew(srcDir, strlen(srcDir), destDir, strlen(destDir), testMode);
|
||||
|
||||
return 0;
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user