1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdint.h>
#include <dirent.h>
#include <string.h>
#define hash_init 5381
uint32_t hash_djb2(const uint8_t * str, uint32_t hash) {
int c;
while ((c = *str++))
hash = ((hash << 5) + hash) ^ c;
return hash;
}
void usage(const char * binname) {
printf("Usage: %s [-d <dir>] [outfile]\n", binname);
exit(-1);
}
void processdir(DIR * dirp, const char * curpath, FILE * outfile) {
char fullpath[1024];
char buf[16 * 1024];
struct dirent * ent;
DIR * rec_dirp;
uint32_t cur_hash = hash_djb2(curpath, hash_init);
uint32_t size, w, hash;
uint8_t b;
FILE * infile;
while ((ent = readdir(dirp))) {
strcpy(fullpath, curpath);
strcat(fullpath, ent->d_name);
switch(ent->d_type) {
case DT_DIR:
if (strcmp(ent->d_name, ".") == 0)
continue;
if (strcmp(ent->d_name, "..") == 0)
continue;
strcat(fullpath, "/");
rec_dirp = opendir(fullpath);
processdir(rec_dirp, fullpath, outfile);
closedir(rec_dirp);
break;
case DT_REG:
case DT_LNK:
hash = hash_djb2(ent->d_name, cur_hash);
infile = fopen(fullpath, "rb");
if (!infile) {
perror("opening file");
exit(-1);
}
b = (hash >> 0) & 0xff; fwrite(&b, 1, 1, outfile);
b = (hash >> 8) & 0xff; fwrite(&b, 1, 1, outfile);
b = (hash >> 16) & 0xff; fwrite(&b, 1, 1, outfile);
b = (hash >> 24) & 0xff; fwrite(&b, 1, 1, outfile);
fseek(infile, 0, SEEK_END);
size = ftell(infile);
fseek(infile, 0, SEEK_SET);
b = (size >> 0) & 0xff; fwrite(&b, 1, 1, outfile);
b = (size >> 8) & 0xff; fwrite(&b, 1, 1, outfile);
b = (size >> 16) & 0xff; fwrite(&b, 1, 1, outfile);
b = (size >> 24) & 0xff; fwrite(&b, 1, 1, outfile);
while (size) {
w = size > 16 * 1024 ? 16 * 1024 : size;
fread(buf, 1, w, infile);
fwrite(buf, 1, w, outfile);
size -= w;
}
fclose(infile);
break;
}
}
}
int main(int argc, char ** argv) {
char * binname = *argv++;
char * o;
char * outname = NULL;
char * dirname = ".";
uint64_t z = 0;
FILE * outfile;
DIR * dirp;
while ((o = *argv++)) {
if (*o == '-') {
o++;
switch (*o) {
case 'd':
dirname = o;
break;
default:
usage(binname);
break;
}
} else {
if (outname)
usage(binname);
outname = o;
}
}
if (!outname)
outfile = stdout;
else
outfile = fopen(outname, "wb");
if (!outfile) {
perror("opening file");
exit(-1);
}
dirp = opendir(dirname);
if (!dirp) {
perror("opening directory");
exit(-1);
}
processdir(dirp, "", outfile);
fwrite(&z, 1, 8, outfile);
if (outname)
fclose(outfile);
closedir(dirp);
}
|
