/
tsk_comparedir.cpp
413 lines (344 loc) · 11.7 KB
/
tsk_comparedir.cpp
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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
/*
** tsk_comparedir
** The Sleuth Kit
**
** Brian Carrier [carrier <at> sleuthkit [dot] org]
** Copyright (c) 2010-2011 Brian Carrier. All Rights reserved
**
** This software is distributed under the Common Public License 1.0
**
*/
#include "tsk3/tsk_tools_i.h"
#include "tsk_comparedir.h"
#include <locale.h>
#include <sys/stat.h>
#include <errno.h>
#include <set>
#ifdef WIN32
#include <windows.h>
#include "shlobj.h"
#else
#include <dirent.h>
#endif
/* The general concept of this procedure is to walk the image and load the file and dir names
* into a structure. Then, analyze the directory to see if the name is in there or not. If it
* was found, remove it. At the end, we'll have a list of names that were in either the image
* or dir, but not both. */
static TSK_TCHAR *progname;
#define TSK_CD_BUFSIZE 1024
static void
usage()
{
TFPRINTF(stderr,
_TSK_T
("usage: %s [-f fstype] [-i imgtype] [-b dev_sector_size] [-o sector_offset] [-n start_inum] [-vV] image [image] comparison_directory\n"),
progname);
tsk_fprintf(stderr,
"\t-i imgtype: The format of the image file (use '-i list' for supported types)\n");
tsk_fprintf(stderr,
"\t-b dev_sector_size: The size (in bytes) of the device sectors\n");
tsk_fprintf(stderr,
"\t-f fstype: The file system type (use '-f list' for supported types)\n");
tsk_fprintf(stderr,
"\t-o sector_offset: sector offset for file system to compare\n");
tsk_fprintf(stderr,
"\t-n start_inum: inum for directory in image file to start compare at\n");
tsk_fprintf(stderr, "\t-v: verbose output to stderr\n");
tsk_fprintf(stderr, "\t-V: Print version\n");
exit(1);
}
/**
* Process a local directory and compare its contents with the image.
* This will recursively call itself on subdirectories.
* @param a_dir Subdirectory of m_lclDir to process.
* @returns 1 on error
*/
uint8_t
TskCompareDir::processLclDir(const TSK_TCHAR * a_dir)
{
std::set < char *, ltstr >::iterator it;
#ifdef TSK_WIN32
WIN32_FIND_DATA ffd;
HANDLE hFind = INVALID_HANDLE_VALUE;
wchar_t fullpath[TSK_CD_BUFSIZE];
UTF16 *utf16;
UTF8 *utf8;
char file8[TSK_CD_BUFSIZE];
//create the full path (utf16)
wcsncpy(fullpath, (wchar_t *) m_lclDir, TSK_CD_BUFSIZE);
if (wcslen((wchar_t *) a_dir) > 0)
wcsncat(fullpath, a_dir, TSK_CD_BUFSIZE);
wcsncat(fullpath, L"\\*", TSK_CD_BUFSIZE);
//start the directory walk
hFind = FindFirstFile((LPCWSTR) fullpath, &ffd);
DWORD err = GetLastError();
if (hFind == INVALID_HANDLE_VALUE) {
fprintf(stderr, "Error opening directory: %S\n", fullpath);
wchar_t message[64];
FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_IGNORE_INSERTS, NULL, err, 0,
(LPWSTR) & message, 64, NULL);
fprintf(stderr, "error: %S", message);
return 1;
}
do {
wchar_t file[TSK_CD_BUFSIZE];
wcsncpy(file, a_dir, TSK_CD_BUFSIZE);
wcsncat(file, L"\\", TSK_CD_BUFSIZE);
wcsncat(file, ffd.cFileName, TSK_CD_BUFSIZE);
//if the file is a directory make recursive call
if (ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
// skip the '.' and '..' entries
if ((file[0] == L'.') && ((file[1] == '\0') || ((file[1] == L'.') && (file[2] == '\0')))) {
// do nothing
}
else if (processLclDir(file)) {
return 1;
}
}
else {
/* convert from utf16 to utf8 to try to find the file in the set
* of names that were found in the image file. */
utf8 = (UTF8 *) file8;
utf16 = (UTF16 *) file;
size_t ilen = wcslen(file);
TSKConversionResult retVal =
tsk_UTF16toUTF8_lclorder((const UTF16 **) &utf16,
&utf16[ilen], &utf8,
&utf8[TSK_CD_BUFSIZE], TSKlenientConversion);
*utf8 = '\0';
if (retVal != TSKconversionOK) {
fprintf(stderr, "Error Converting file name");
return 1;
}
//if the file is in the set, remove it and continue, if not print
it = m_filesInImg.find(file8);
if (it != m_filesInImg.end()) {
m_filesInImg.erase(it);
}
else {
printf("file: %s not found in image file\n", file8);
m_missDirFile = true;
}
}
} while (FindNextFile(hFind, &ffd) != 0);
FindClose(hFind);
#else
DIR *dp;
struct dirent *dirp;
char file[TSK_CD_BUFSIZE];
char fullPath[TSK_CD_BUFSIZE];
struct stat status;
strncpy(fullPath, m_lclDir, TSK_CD_BUFSIZE);
strncat(fullPath, a_dir, TSK_CD_BUFSIZE-strlen(fullPath));
if ((dp = opendir(fullPath)) == NULL) {
fprintf(stderr, "Error opening directory");
return 1;
}
while ((dirp = readdir(dp)) != NULL) {
strncpy(file, a_dir, TSK_CD_BUFSIZE);
strncat(file, "/", TSK_CD_BUFSIZE-strlen(file));
strncat(file, dirp->d_name, TSK_CD_BUFSIZE-strlen(file));
strncpy(fullPath, m_lclDir, TSK_CD_BUFSIZE);
strncat(fullPath, file, TSK_CD_BUFSIZE-strlen(fullPath));
stat(fullPath, &status);
if (S_ISDIR(status.st_mode)) {
// skip the '.' and '..' entries
if ((file[0] == '.') && ((file[1] == '\0') || ((file[1] == '.') && (file[2] == '\0')))) {
// do nothing
}
else if (processLclDir(file)) {
return 1;
}
}
else {
// see if we already saw this file in the image
it = m_filesInImg.find(file);
if (it != m_filesInImg.end()) {
m_filesInImg.erase(it);
}
else {
printf("file: %s not found in image file\n", file);
m_missDirFile = true;
}
}
}
#endif
return 0;
}
/********** Methods that load the internal list / set with info from the image **********/
TSK_RETVAL_ENUM
TskCompareDir::processFile(TSK_FS_FILE * a_fs_file, const char *a_path)
{
//exclude certain types
if (isDotDir(a_fs_file, a_path))
return TSK_OK;
if (isDir(a_fs_file))
return TSK_OK;
if ((isNtfsSystemFiles(a_fs_file, a_path)) || (isFATSystemFiles(a_fs_file)))
return TSK_OK;
if (!a_fs_file->meta)
return TSK_OK;
//create the full path
size_t len = strlen(a_fs_file->name->name) + strlen(a_path) + 1;
char *fullPath = (char *) tsk_malloc(len);
if (fullPath == NULL)
return TSK_ERR;
snprintf(fullPath, len, "/");
strncat(fullPath, a_path, len-strlen(fullPath));
strncat(fullPath, a_fs_file->name->name, len-strlen(fullPath));
//convert path for win32
#ifdef WIN32
for (size_t i = 0; i < strlen(fullPath); i++) {
if (fullPath[i] == '/')
fullPath[i] = '\\';
}
#endif
//add the path to the internal list/set
m_filesInImg.insert(fullPath);
return TSK_OK;
}
TSK_FILTER_ENUM
TskCompareDir::filterVol(const TSK_VS_PART_INFO * a_vs_part)
{
fprintf(stderr, "Error: volume system detected. You must specify a specific file system using '-o'\n");
return TSK_FILTER_STOP;
}
/*
* @param a_soffset Sector offset where file system to analyze is located
* @param a_inum is 0 if root directory should be processed
* @param a_lcl_dir Path of local directory to compare with image contents.
* @returns 1 on error
*/
uint8_t
TskCompareDir::compareDirs(TSK_OFF_T a_soffset, TSK_INUM_T a_inum,
TSK_FS_TYPE_ENUM a_fstype, const TSK_TCHAR * a_lcl_dir)
{
uint8_t retval;
// collect the file names that are in the disk image
if (a_inum != 0)
retval =
findFilesInFs(a_soffset * m_img_info->sector_size, a_fstype, a_inum);
else
retval = findFilesInFs(a_soffset * m_img_info->sector_size, a_fstype);
if (retval)
return 1;
m_missDirFile = false;
m_lclDir = a_lcl_dir;
// process the local directory
if (processLclDir(_TSK_T("")))
return 1;
if (!m_missDirFile)
printf("All files in directory found in image\n");
if (m_filesInImg.begin() == m_filesInImg.end()) {
printf("All files in image found in directory\n");
}
else {
std::set < char *, ltstr >::iterator it;
for (it = m_filesInImg.begin(); it != m_filesInImg.end(); it++)
printf("file: %s not found in directory\n",
(TSK_TCHAR *) * it);
}
return 0;
}
int
main(int argc, char **argv1)
{
TSK_TCHAR **argv;
TSK_IMG_TYPE_ENUM imgtype = TSK_IMG_TYPE_DETECT;
TSK_FS_TYPE_ENUM fstype = TSK_FS_TYPE_DETECT;
unsigned int ssize = 0;
#ifdef WIN32
argv = CommandLineToArgvW(GetCommandLineW(), &argc);
if (argv == NULL) {
fprintf(stderr, "Error getting wide arguments\n");
exit(1);
}
#else
argv = (TSK_TCHAR **) argv1;
#endif
TSK_OFF_T soffset = 0;
TSK_TCHAR *cp;
int ch;
TSK_INUM_T inum = 0;
progname = argv[0];
setlocale(LC_ALL, "");
while ((ch = GETOPT(argc, argv, _TSK_T("b:f:i:o:n:vV"))) > 0) {
switch (ch) {
case _TSK_T('?'):
default:
TFPRINTF(stderr, _TSK_T("Invalid argument: %s\n"),
argv[OPTIND]);
usage();
case _TSK_T('b'):
ssize = (unsigned int) TSTRTOUL(OPTARG, &cp, 0);
if (*cp || *cp == *OPTARG || ssize < 1) {
TFPRINTF(stderr,
_TSK_T
("invalid argument: sector size must be positive: %s\n"),
OPTARG);
usage();
}
break;
case _TSK_T('f'):
if (TSTRCMP(OPTARG, _TSK_T("list")) == 0) {
tsk_fs_type_print(stderr);
exit(1);
}
fstype = tsk_fs_type_toid(OPTARG);
if (fstype == TSK_FS_TYPE_UNSUPP) {
TFPRINTF(stderr,
_TSK_T("Unsupported file system type: %s\n"), OPTARG);
usage();
}
break;
case _TSK_T('i'):
if (TSTRCMP(OPTARG, _TSK_T("list")) == 0) {
tsk_img_type_print(stderr);
exit(1);
}
imgtype = tsk_img_type_toid(OPTARG);
if (imgtype == TSK_IMG_TYPE_UNSUPP) {
TFPRINTF(stderr, _TSK_T("Unsupported image type: %s\n"),
OPTARG);
usage();
}
break;
case _TSK_T('n'):
if (tsk_fs_parse_inum(OPTARG, &inum, NULL, NULL, NULL, NULL)) {
tsk_error_print(stderr);
usage();
}
break;
case _TSK_T('o'):
if ((soffset = tsk_parse_offset(OPTARG)) == -1) {
tsk_error_print(stderr);
usage();
}
break;
case _TSK_T('v'):
tsk_verbose++;
break;
case _TSK_T('V'):
tsk_version_print(stdout);
exit(0);
}
}
/* We need at least two more argument */
if (OPTIND + 1 >= argc) {
tsk_fprintf(stderr,
"Missing output directory and/or image name\n");
usage();
}
TskCompareDir tskCompareDir;
tskCompareDir.setFileFilterFlags(TSK_FS_DIR_WALK_FLAG_ALLOC);
if (tskCompareDir.openImage(argc - OPTIND - 1, &argv[OPTIND], imgtype, ssize)) {
tsk_error_print(stderr);
exit(1);
}
if (tskCompareDir.compareDirs(soffset, inum, fstype, argv[argc - 1])) {
tsk_error_print(stderr);
exit(1);
}
return 0;
}