322 lines
8.7 KiB
C
322 lines
8.7 KiB
C
#include <stdio.h>
|
|
#include <sys/stat.h>
|
|
#include <openssl/md5.h>
|
|
#include <openssl/sha.h>
|
|
#include <openssl/ripemd.h>
|
|
#include <errno.h>
|
|
#include <dirent.h>
|
|
#include <limits.h>
|
|
#include <stdlib.h>
|
|
#include <fcntl.h>
|
|
#include <limits.h>
|
|
|
|
typedef struct DEDUPE_STORE_CONTEXT {
|
|
char blob_dir[PATH_MAX];
|
|
FILE *output_manifest;
|
|
};
|
|
|
|
static void usage(char** argv) {
|
|
fprintf(stderr, "usage: %s c input_directory blob_dir output_manifest\n", argv[0]);
|
|
fprintf(stderr, "usage: %s x input_manifest blob_dir output_directory\n", argv[0]);
|
|
}
|
|
|
|
static int copy_file(const char *dst, const char *src) {
|
|
char buf[4096];
|
|
int dstfd, srcfd, bytes_read, bytes_written, total_read = 0;
|
|
if (src == NULL)
|
|
return 1;
|
|
if (dst == NULL)
|
|
return 2;
|
|
|
|
srcfd = open(src, O_RDONLY);
|
|
if (srcfd < 0)
|
|
return 3;
|
|
|
|
dstfd = open(dst, O_RDWR | O_CREAT | O_TRUNC, 0600);
|
|
if (dstfd < 0) {
|
|
close(srcfd);
|
|
return 4;
|
|
}
|
|
|
|
do {
|
|
total_read += bytes_read = read(srcfd, buf, 4096);
|
|
if (!bytes_read)
|
|
break;
|
|
if (bytes_read < 4096)
|
|
memset(&buf[bytes_read], 0, 4096 - bytes_read);
|
|
if (write(dstfd, buf, 4096) < 4096)
|
|
return 5;
|
|
} while(bytes_read == 4096);
|
|
|
|
close(dstfd);
|
|
close(srcfd);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void do_md5sum(FILE *mfile, unsigned char *rptr) {
|
|
char rdata[BUFSIZ];
|
|
int rsize;
|
|
MD5_CTX c;
|
|
|
|
MD5_Init(&c);
|
|
while(!feof(mfile)) {
|
|
rsize = fread(rdata, sizeof(char), BUFSIZ, mfile);
|
|
if(rsize > 0) {
|
|
MD5_Update(&c, rdata, rsize);
|
|
}
|
|
}
|
|
|
|
MD5_Final(rptr, &c);
|
|
}
|
|
|
|
static int do_md5sum_file(const char* filename, unsigned char *rptr) {
|
|
FILE *f = fopen(filename, "rb");
|
|
if (f == NULL) {
|
|
fprintf(stderr, "Unable to open file: %s\n", filename);
|
|
return 1;
|
|
}
|
|
do_md5sum(f, rptr);
|
|
fclose(f);
|
|
return 0;
|
|
}
|
|
|
|
static int store_st(struct DEDUPE_STORE_CONTEXT *context, struct stat st, const char* s);
|
|
|
|
void print_stat(struct DEDUPE_STORE_CONTEXT *context, char type, struct stat st, const char *f) {
|
|
fprintf(context->output_manifest, "%c\t%o\t%d\t%d\t%s\t", type, st.st_mode & (S_IRWXU | S_IRWXG | S_IRWXO | S_ISUID | S_ISGID), st.st_uid, st.st_gid, f);
|
|
}
|
|
|
|
static int store_file(struct DEDUPE_STORE_CONTEXT *context, struct stat st, const char* f) {
|
|
printf("%s\n", f);
|
|
unsigned char sumdata[SHA_DIGEST_LENGTH];
|
|
int ret;
|
|
if (ret = do_md5sum_file(f, sumdata)) {
|
|
fprintf(stderr, "Error calculating md5sum of %s\n", f);
|
|
return ret;
|
|
}
|
|
char psum[41];
|
|
int j;
|
|
for (j = 0; j < MD5_DIGEST_LENGTH; j++)
|
|
sprintf(&psum[(j*2)], "%02x", (int)sumdata[j]);
|
|
psum[(MD5_DIGEST_LENGTH * 2)] = '\0';
|
|
|
|
char out_blob[PATH_MAX];
|
|
sprintf(out_blob, "%s/%s", context->blob_dir, psum);
|
|
if (ret = copy_file(out_blob, f)) {
|
|
fprintf(stderr, "Error copying blob %s\n", f);
|
|
return ret;
|
|
}
|
|
|
|
fprintf(context->output_manifest, "%s\t\n", psum);
|
|
return 0;
|
|
}
|
|
|
|
static int store_dir(struct DEDUPE_STORE_CONTEXT *context, struct stat st, const char* d) {
|
|
printf("%s\n", d);
|
|
DIR *dp = opendir(d);
|
|
if (d == NULL) {
|
|
fprintf(stderr, "Error opening directory: %s\n", d);
|
|
return 1;
|
|
}
|
|
struct dirent *ep;
|
|
char full_path[PATH_MAX];
|
|
while (ep = readdir(dp)) {
|
|
if (strcmp(ep->d_name, ".") == 0)
|
|
continue;
|
|
if (strcmp(ep->d_name, "..") == 0)
|
|
continue;
|
|
struct stat cst;
|
|
int ret;
|
|
sprintf(full_path, "%s/%s", d, ep->d_name);
|
|
if (0 != (ret = lstat(full_path, &cst))) {
|
|
fprintf(stderr, "Error opening: %s\n", ep->d_name);
|
|
closedir(dp);
|
|
return ret;
|
|
}
|
|
|
|
if (ret = store_st(context->blob_dir, cst, full_path))
|
|
return ret;
|
|
}
|
|
closedir(dp);
|
|
return 0;
|
|
}
|
|
|
|
static int store_link(struct DEDUPE_STORE_CONTEXT *context, struct stat st, const char* l) {
|
|
printf("%s\n", l);
|
|
char link[PATH_MAX];
|
|
int ret = readlink(l, link, PATH_MAX);
|
|
if (ret < 0) {
|
|
fprintf(stderr, "Error reading symlink\n");
|
|
return errno;
|
|
}
|
|
fprintf(context->output_manifest, "%s\t\n", link);
|
|
return 0;
|
|
}
|
|
|
|
static int store_st(struct DEDUPE_STORE_CONTEXT *context, struct stat st, const char* s) {
|
|
if (S_ISREG(st.st_mode)) {
|
|
print_stat(context, 'f', st, s);
|
|
return store_file(context, st, s);
|
|
}
|
|
else if (S_ISDIR(st.st_mode)) {
|
|
print_stat(context, 'd', st, s);
|
|
fprintf(context->output_manifest, "\n");
|
|
return store_dir(context, st, s);
|
|
}
|
|
else if (S_ISLNK(st.st_mode)) {
|
|
print_stat(context, 'l', st, s);
|
|
return store_link(context, st, s);
|
|
}
|
|
else {
|
|
fprintf(stderr, "Skipping special: %s\n", s);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
void get_full_path(char *out_path, char *rel_path) {
|
|
char tmp[PATH_MAX];
|
|
getcwd(tmp, PATH_MAX);
|
|
chdir(rel_path);
|
|
getcwd(out_path, PATH_MAX);
|
|
chdir(tmp);
|
|
}
|
|
|
|
static char* tokenize(char *out, const char* line, const char sep) {
|
|
while (*line != sep) {
|
|
if (*line == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
*out = *line;
|
|
out++;
|
|
line++;
|
|
}
|
|
|
|
*out = NULL;
|
|
// resume at the next char
|
|
return line + 1;
|
|
}
|
|
|
|
static int dec_to_oct(int dec) {
|
|
int ret = 0;
|
|
int mult = 1;
|
|
while (dec != 0) {
|
|
int rem = dec % 10;
|
|
ret += (rem * mult);
|
|
dec /= 10;
|
|
mult *= 8;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
if (argc != 5) {
|
|
usage(argv);
|
|
return 1;
|
|
}
|
|
|
|
if (strcmp(argv[1], "c") == 0) {
|
|
struct stat st;
|
|
int ret;
|
|
if (0 != (ret = lstat(argv[2], &st))) {
|
|
fprintf(stderr, "Error opening input_file/input_directory.\n");
|
|
return ret;
|
|
}
|
|
|
|
if (!S_ISDIR(st.st_mode)) {
|
|
fprintf(stderr, "%s must be a directory.\n", argv[2]);
|
|
return;
|
|
}
|
|
|
|
char blob_dir[PATH_MAX];
|
|
struct DEDUPE_STORE_CONTEXT context;
|
|
context.output_manifest = fopen(argv[4], "wb");
|
|
if (context.output_manifest == NULL) {
|
|
fprintf(stderr, "Unable to open output file %s\n", argv[4]);
|
|
return 1;
|
|
}
|
|
get_full_path(context.blob_dir, argv[3]);
|
|
chdir(argv[2]);
|
|
|
|
return store_dir(&context, st, ".");
|
|
}
|
|
else if (strcmp(argv[1], "x") == 0) {
|
|
FILE *input_manifest = fopen(argv[2], "rb");
|
|
if (input_manifest == NULL) {
|
|
fprintf(stderr, "Unable to open input manifest %s\n", argv[2]);
|
|
return 1;
|
|
}
|
|
|
|
char blob_dir[PATH_MAX];
|
|
char *output_dir = argv[4];
|
|
get_full_path(blob_dir, argv[3]);
|
|
|
|
printf("%s\n" , output_dir);
|
|
chdir(output_dir);
|
|
|
|
char line[PATH_MAX];
|
|
while (fgets(line, PATH_MAX, input_manifest)) {
|
|
//printf("%s", line);
|
|
|
|
char type[4];
|
|
char mode[8];
|
|
char uid[32];
|
|
char gid[32];
|
|
char filename[PATH_MAX];
|
|
|
|
char *token = line;
|
|
token = tokenize(type, token, '\t');
|
|
token = tokenize(mode, token, '\t');
|
|
token = tokenize(uid, token, '\t');
|
|
token = tokenize(gid, token, '\t');
|
|
token = tokenize(filename, token, '\t');
|
|
|
|
int mode_oct = dec_to_oct(atoi(mode));
|
|
int ret;
|
|
printf("%s\t%s\t%s\t%s\t%s\t", type, mode, uid, gid, filename);
|
|
if (strcmp(type, "f") == 0) {
|
|
char md5[41];
|
|
token = tokenize(md5, token, '\t');
|
|
printf("%s\n", md5);
|
|
|
|
char blob_file[PATH_MAX];
|
|
sprintf(blob_file, "%s/%s", blob_dir, md5);
|
|
if (ret = copy_file(filename, blob_file)) {
|
|
fprintf(stderr, "Unable to copy file %s\n", filename);
|
|
return ret;
|
|
}
|
|
|
|
chmod(filename, mode_oct);
|
|
chown(filename, uid, gid);
|
|
}
|
|
else if (strcmp(type, "l") == 0) {
|
|
char link[41];
|
|
token = tokenize(link, token, '\t');
|
|
printf("%s\n", link);
|
|
|
|
symlink(link, filename);
|
|
|
|
chmod(filename, mode_oct);
|
|
lchown(filename, uid, gid);
|
|
}
|
|
else if (strcmp(type, "d") == 0) {
|
|
printf("\n");
|
|
|
|
mkdir(filename, mode_oct);
|
|
|
|
chmod(filename, mode_oct);
|
|
chown(filename, uid, gid);
|
|
}
|
|
}
|
|
|
|
fclose(input_manifest);
|
|
return 0;
|
|
}
|
|
else {
|
|
usage(argv);
|
|
return 1;
|
|
}
|
|
}
|