1987da915Sopenharmony_ci/**
2987da915Sopenharmony_ci * ntfsck - Part of the Linux-NTFS project.
3987da915Sopenharmony_ci *
4987da915Sopenharmony_ci * Copyright (c) 2006 Yuval Fledel
5987da915Sopenharmony_ci *
6987da915Sopenharmony_ci * This utility will check and fix errors on an NTFS volume.
7987da915Sopenharmony_ci *
8987da915Sopenharmony_ci * This program is free software; you can redistribute it and/or modify
9987da915Sopenharmony_ci * it under the terms of the GNU General Public License as published by
10987da915Sopenharmony_ci * the Free Software Foundation; either version 2 of the License, or
11987da915Sopenharmony_ci * (at your option) any later version.
12987da915Sopenharmony_ci *
13987da915Sopenharmony_ci * This program is distributed in the hope that it will be useful,
14987da915Sopenharmony_ci * but WITHOUT ANY WARRANTY; without even the implied warranty of
15987da915Sopenharmony_ci * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16987da915Sopenharmony_ci * GNU General Public License for more details.
17987da915Sopenharmony_ci *
18987da915Sopenharmony_ci * You should have received a copy of the GNU General Public License
19987da915Sopenharmony_ci * along with this program (in the main directory of the Linux-NTFS
20987da915Sopenharmony_ci * distribution in the file COPYING); if not, write to the Free Software
21987da915Sopenharmony_ci * Foundation,Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
22987da915Sopenharmony_ci */
23987da915Sopenharmony_ci
24987da915Sopenharmony_ci
25987da915Sopenharmony_ci#include "config.h"
26987da915Sopenharmony_ci
27987da915Sopenharmony_ci#ifdef HAVE_STDIO_H
28987da915Sopenharmony_ci#include <stdio.h>
29987da915Sopenharmony_ci#endif
30987da915Sopenharmony_ci#ifdef HAVE_STDLIB_H
31987da915Sopenharmony_ci#include <stdlib.h>
32987da915Sopenharmony_ci#endif
33987da915Sopenharmony_ci#ifdef HAVE_STRING_H
34987da915Sopenharmony_ci#include <string.h>
35987da915Sopenharmony_ci#endif
36987da915Sopenharmony_ci#ifdef HAVE_FCNTL_H
37987da915Sopenharmony_ci#include <fcntl.h>
38987da915Sopenharmony_ci#endif
39987da915Sopenharmony_ci
40987da915Sopenharmony_ci#include <layout.h>
41987da915Sopenharmony_ci#include <bitmap.h>
42987da915Sopenharmony_ci#include <endians.h>
43987da915Sopenharmony_ci#include <bootsect.h>
44987da915Sopenharmony_ci#include <misc.h>
45987da915Sopenharmony_ci
46987da915Sopenharmony_ci#include "cluster.h"
47987da915Sopenharmony_ci#include "utils.h"
48987da915Sopenharmony_ci
49987da915Sopenharmony_ci#define RETURN_FS_ERRORS_CORRECTED (1)
50987da915Sopenharmony_ci#define RETURN_SYSTEM_NEEDS_REBOOT (2)
51987da915Sopenharmony_ci#define RETURN_FS_ERRORS_LEFT_UNCORRECTED (4)
52987da915Sopenharmony_ci#define RETURN_OPERATIONAL_ERROR (8)
53987da915Sopenharmony_ci#define RETURN_USAGE_OR_SYNTAX_ERROR (16)
54987da915Sopenharmony_ci#define RETURN_CANCELLED_BY_USER (32)
55987da915Sopenharmony_ci/* Where did 64 go? */
56987da915Sopenharmony_ci#define RETURN_SHARED_LIBRARY_ERROR (128)
57987da915Sopenharmony_ci
58987da915Sopenharmony_ci/* todo: command line: (everything is optional)
59987da915Sopenharmony_ci *  fsck-frontend options:
60987da915Sopenharmony_ci *	-C [fd]	: display progress bar (send it to the file descriptor if specified)
61987da915Sopenharmony_ci *	-T	: don't show the title on startup
62987da915Sopenharmony_ci *  fsck-checker options:
63987da915Sopenharmony_ci *	-a	: auto-repair. no questions. (optional: if marked clean and -f not specified, just check if mounable)
64987da915Sopenharmony_ci *	-p	: auto-repair safe. no questions (optional: same)
65987da915Sopenharmony_ci *	-n	: only check. no repair.
66987da915Sopenharmony_ci *	-r	: interactively repair.
67987da915Sopenharmony_ci *	-y	: always yes.
68987da915Sopenharmony_ci *	-v	: verbose.
69987da915Sopenharmony_ci *	-V	: version.
70987da915Sopenharmony_ci *  taken from fsck.ext2
71987da915Sopenharmony_ci *	-b sb	: use the superblock from sb. For corrupted volumes. (do we want separete boot/mft options?)
72987da915Sopenharmony_ci *	-c	: use badblocks(8) to find bad blocks (R/O mode) and add the findings to $Bad.
73987da915Sopenharmony_ci *	-C fd	: write competion info to fd. If 0, print a completion bar.
74987da915Sopenharmony_ci *	-d	: debugging output.
75987da915Sopenharmony_ci *	-D	: rebalance indices.
76987da915Sopenharmony_ci *	-f	: force checking even if marked clean.
77987da915Sopenharmony_ci *	-F	: flush buffers before beginning. (for time-benchmarking)
78987da915Sopenharmony_ci *	-k	: When used with -c, don't erase previous $Bad items.
79987da915Sopenharmony_ci *	-n	: Open fs as readonly. assume always no. (why is it needed if -r is not specified?)
80987da915Sopenharmony_ci *	-t	: Print time statistics.
81987da915Sopenharmony_ci *  taken from fsck.reiserfs
82987da915Sopenharmony_ci *	--rebuild-sb	: try to find $MFT start and rebuild the boot sector.
83987da915Sopenharmony_ci *	--rebuild-tree	: scan for items and rebuild the indices that point to them (0x30, $SDS, etc.)
84987da915Sopenharmony_ci *	--clean-reserved: zero rezerved fields. (use with care!)
85987da915Sopenharmony_ci *	--adjust-size -z: insert a sparse hole if the data_size is larger than the size marked in the runlist.
86987da915Sopenharmony_ci *	--logfile file	: report corruptions (unlike other errors) to a file instead of stderr.
87987da915Sopenharmony_ci *	--nolog		: don't report corruptions at all.
88987da915Sopenharmony_ci *	--quiet -q	: no progress bar.
89987da915Sopenharmony_ci *  taken from fsck.msdos
90987da915Sopenharmony_ci *	-w	: flush after every write.
91987da915Sopenharmony_ci *	- do n passes. (only 2 in fsck.msdos. second should not report errors. Bonus: stop when error list does not change)
92987da915Sopenharmony_ci *  taken from fsck.jfs
93987da915Sopenharmony_ci *	--omit-journal-reply: self-descriptive (why would someone do that?)
94987da915Sopenharmony_ci *	--replay-journal-only: self-descriptive. don't check otherwise.
95987da915Sopenharmony_ci *  taken from fsck.xfs
96987da915Sopenharmony_ci *	-s	: only serious errors should be reported.
97987da915Sopenharmony_ci *	-i ino	: verbose behaviour only for inode ino.
98987da915Sopenharmony_ci *	-b bno	: verbose behaviour only for cluster bno.
99987da915Sopenharmony_ci *	-L	: zero log.
100987da915Sopenharmony_ci *  inspired by others
101987da915Sopenharmony_ci *	- don't do cluster accounting.
102987da915Sopenharmony_ci *	- don't do mft record accounting.
103987da915Sopenharmony_ci *	- don't do file names accounting.
104987da915Sopenharmony_ci *	- don't do security_id accounting.
105987da915Sopenharmony_ci *	- don't check acl inheritance problems.
106987da915Sopenharmony_ci *	- undelete unused mft records. (bonus: different options for 100% salvagable and less)
107987da915Sopenharmony_ci *	- error-level-report n: only report errors above this error level
108987da915Sopenharmony_ci *	- error-level-repair n: only repair errors below this error level
109987da915Sopenharmony_ci *	- don't fail on ntfsclone metadata pruning.
110987da915Sopenharmony_ci *  signals:
111987da915Sopenharmony_ci *	SIGUSR1	: start displaying progress bar
112987da915Sopenharmony_ci *	SIGUSR2	: stop displaying progress bar.
113987da915Sopenharmony_ci */
114987da915Sopenharmony_ci
115987da915Sopenharmony_ci/* Assuming NO_NTFS_DEVICE_DEFAULT_IO_OPS is not set */
116987da915Sopenharmony_ci
117987da915Sopenharmony_cistatic int errors = 0;
118987da915Sopenharmony_cistatic int unsupported = 0;
119987da915Sopenharmony_ci
120987da915Sopenharmony_cistatic short bytes_per_sector, sectors_per_cluster;
121987da915Sopenharmony_ci//static s64 mft_offset, mftmirr_offset;
122987da915Sopenharmony_cistatic s64 current_mft_record;
123987da915Sopenharmony_ci
124987da915Sopenharmony_ci/**
125987da915Sopenharmony_ci * This is just a preliminary volume.
126987da915Sopenharmony_ci * Filled while checking the boot sector and used in the preliminary MFT check.
127987da915Sopenharmony_ci */
128987da915Sopenharmony_ci//static ntfs_volume vol;
129987da915Sopenharmony_ci
130987da915Sopenharmony_cistatic runlist_element *mft_rl, *mft_bitmap_rl;
131987da915Sopenharmony_ci
132987da915Sopenharmony_ci#define check_failed(FORMAT, ARGS...) \
133987da915Sopenharmony_ci	do { \
134987da915Sopenharmony_ci		errors++; \
135987da915Sopenharmony_ci		ntfs_log_redirect(__FUNCTION__,__FILE__,__LINE__, \
136987da915Sopenharmony_ci				NTFS_LOG_LEVEL_ERROR,NULL,FORMAT,##ARGS); \
137987da915Sopenharmony_ci	} while (0);
138987da915Sopenharmony_ci
139987da915Sopenharmony_ci/**
140987da915Sopenharmony_ci * 0 success.
141987da915Sopenharmony_ci * 1 fail.
142987da915Sopenharmony_ci */
143987da915Sopenharmony_cistatic int assert_u32_equal(u32 val, u32 ok, const char *name)
144987da915Sopenharmony_ci{
145987da915Sopenharmony_ci	if (val!=ok) {
146987da915Sopenharmony_ci		check_failed("Assertion failed for '%lld:%s'. should be 0x%x, "
147987da915Sopenharmony_ci			"was 0x%x.\n", (long long)current_mft_record, name,
148987da915Sopenharmony_ci			(int)ok, (int)val);
149987da915Sopenharmony_ci		//errors++;
150987da915Sopenharmony_ci		return 1;
151987da915Sopenharmony_ci	}
152987da915Sopenharmony_ci	return 0;
153987da915Sopenharmony_ci}
154987da915Sopenharmony_ci
155987da915Sopenharmony_cistatic int assert_u32_noteq(u32 val, u32 wrong, const char *name)
156987da915Sopenharmony_ci{
157987da915Sopenharmony_ci	if (val==wrong) {
158987da915Sopenharmony_ci		check_failed("Assertion failed for '%lld:%s'. should not be "
159987da915Sopenharmony_ci			"0x%x.\n", (long long)current_mft_record, name,
160987da915Sopenharmony_ci			(int)wrong);
161987da915Sopenharmony_ci		return 1;
162987da915Sopenharmony_ci	}
163987da915Sopenharmony_ci	return 0;
164987da915Sopenharmony_ci}
165987da915Sopenharmony_ci
166987da915Sopenharmony_cistatic int assert_u32_lesseq(u32 val1, u32 val2, const char *name)
167987da915Sopenharmony_ci{
168987da915Sopenharmony_ci	if (val1 > val2) {
169987da915Sopenharmony_ci		check_failed("Assertion failed for '%s'. 0x%x > 0x%x\n",
170987da915Sopenharmony_ci			name, (int)val1, (int)val2);
171987da915Sopenharmony_ci		//errors++;
172987da915Sopenharmony_ci		return 1;
173987da915Sopenharmony_ci	}
174987da915Sopenharmony_ci	return 0;
175987da915Sopenharmony_ci}
176987da915Sopenharmony_ci
177987da915Sopenharmony_cistatic int assert_u32_less(u32 val1, u32 val2, const char *name)
178987da915Sopenharmony_ci{
179987da915Sopenharmony_ci	if (val1 >= val2) {
180987da915Sopenharmony_ci		check_failed("Assertion failed for '%s'. 0x%x >= 0x%x\n",
181987da915Sopenharmony_ci			name, (int)val1, (int)val2);
182987da915Sopenharmony_ci		//errors++;
183987da915Sopenharmony_ci		return 1;
184987da915Sopenharmony_ci	}
185987da915Sopenharmony_ci	return 0;
186987da915Sopenharmony_ci}
187987da915Sopenharmony_ci
188987da915Sopenharmony_ci/**
189987da915Sopenharmony_ci * Return: 0 ok, 1 error.
190987da915Sopenharmony_ci *
191987da915Sopenharmony_ci * todo: may we use ntfs_boot_sector_is_ntfs() instead?
192987da915Sopenharmony_ci *	It already does the checks but will not be able to fix anything.
193987da915Sopenharmony_ci */
194987da915Sopenharmony_cistatic BOOL verify_boot_sector(struct ntfs_device *dev, ntfs_volume *rawvol)
195987da915Sopenharmony_ci{
196987da915Sopenharmony_ci	u8 buf[512];
197987da915Sopenharmony_ci	NTFS_BOOT_SECTOR *ntfs_boot = (NTFS_BOOT_SECTOR *)&buf;
198987da915Sopenharmony_ci	//u32 bytes_per_cluster;
199987da915Sopenharmony_ci
200987da915Sopenharmony_ci	current_mft_record = 9;
201987da915Sopenharmony_ci
202987da915Sopenharmony_ci	if (ntfs_pread(dev, 0, sizeof(buf), buf) != sizeof(buf)) {
203987da915Sopenharmony_ci		check_failed("Failed to read boot sector.\n");
204987da915Sopenharmony_ci		return 1;
205987da915Sopenharmony_ci	}
206987da915Sopenharmony_ci
207987da915Sopenharmony_ci	if ((buf[0]!=0xeb) ||
208987da915Sopenharmony_ci			((buf[1]!=0x52) && (buf[1]!=0x5b)) ||
209987da915Sopenharmony_ci			(buf[2]!=0x90)) {
210987da915Sopenharmony_ci		check_failed("Boot sector: Bad jump.\n");
211987da915Sopenharmony_ci	}
212987da915Sopenharmony_ci	if (ntfs_boot->oem_id != magicNTFS) {
213987da915Sopenharmony_ci		check_failed("Boot sector: Bad NTFS magic.\n");
214987da915Sopenharmony_ci	}
215987da915Sopenharmony_ci	bytes_per_sector = le16_to_cpu(ntfs_boot->bpb.bytes_per_sector);
216987da915Sopenharmony_ci	if (!bytes_per_sector) {
217987da915Sopenharmony_ci		check_failed("Boot sector: Bytes per sector is 0.\n");
218987da915Sopenharmony_ci	}
219987da915Sopenharmony_ci	if (bytes_per_sector%512) {
220987da915Sopenharmony_ci		check_failed("Boot sector: Bytes per sector is not a multiple"
221987da915Sopenharmony_ci					" of 512.\n");
222987da915Sopenharmony_ci	}
223987da915Sopenharmony_ci	sectors_per_cluster = ntfs_boot->bpb.sectors_per_cluster;
224987da915Sopenharmony_ci
225987da915Sopenharmony_ci	// todo: if partition, query bios and match heads/tracks? */
226987da915Sopenharmony_ci
227987da915Sopenharmony_ci	// Initialize some values into rawvol. We will need those later.
228987da915Sopenharmony_ci	rawvol->dev = dev;
229987da915Sopenharmony_ci	ntfs_boot_sector_parse(rawvol, (NTFS_BOOT_SECTOR *)buf);
230987da915Sopenharmony_ci
231987da915Sopenharmony_ci	return 0;
232987da915Sopenharmony_ci}
233987da915Sopenharmony_ci
234987da915Sopenharmony_ci/**
235987da915Sopenharmony_ci * Load the runlist of the <attr_type> attribute.
236987da915Sopenharmony_ci *
237987da915Sopenharmony_ci * Return NULL if an error.
238987da915Sopenharmony_ci * The caller is responsible on freeing the allocated memory if the result is not NULL.
239987da915Sopenharmony_ci *
240987da915Sopenharmony_ci * Set size_of_file_record to some reasonable size when in doubt (the Windows default is 1024.)
241987da915Sopenharmony_ci *
242987da915Sopenharmony_ci * attr_type must be little endian.
243987da915Sopenharmony_ci *
244987da915Sopenharmony_ci * This function has code duplication with check_file_record() and
245987da915Sopenharmony_ci * check_attr_record() but its goal is to be less strict. Thus the
246987da915Sopenharmony_ci * duplicated checks are the minimal required for not crashing.
247987da915Sopenharmony_ci *
248987da915Sopenharmony_ci * Assumes dev is open.
249987da915Sopenharmony_ci */
250987da915Sopenharmony_cistatic runlist *load_runlist(ntfs_volume *rawvol, s64 offset_to_file_record, ATTR_TYPES attr_type, u32 size_of_file_record)
251987da915Sopenharmony_ci{
252987da915Sopenharmony_ci	u8 *buf;
253987da915Sopenharmony_ci	u16 attrs_offset;
254987da915Sopenharmony_ci	u32 length;
255987da915Sopenharmony_ci	ATTR_RECORD *attr_rec;
256987da915Sopenharmony_ci
257987da915Sopenharmony_ci	if (size_of_file_record<22) // offset to attrs_offset
258987da915Sopenharmony_ci		return NULL;
259987da915Sopenharmony_ci
260987da915Sopenharmony_ci	buf = (u8*)ntfs_malloc(size_of_file_record);
261987da915Sopenharmony_ci	if (!buf)
262987da915Sopenharmony_ci		return NULL;
263987da915Sopenharmony_ci
264987da915Sopenharmony_ci	if (ntfs_pread(rawvol->dev, offset_to_file_record, size_of_file_record, buf) !=
265987da915Sopenharmony_ci			size_of_file_record) {
266987da915Sopenharmony_ci		check_failed("Failed to read file record at offset %lld (0x%llx).\n",
267987da915Sopenharmony_ci					(long long)offset_to_file_record,
268987da915Sopenharmony_ci					(long long)offset_to_file_record);
269987da915Sopenharmony_ci		return NULL;
270987da915Sopenharmony_ci	}
271987da915Sopenharmony_ci
272987da915Sopenharmony_ci	attrs_offset = le16_to_cpu(((MFT_RECORD*)buf)->attrs_offset);
273987da915Sopenharmony_ci	// first attribute must be after the header.
274987da915Sopenharmony_ci	if (attrs_offset<42) {
275987da915Sopenharmony_ci		check_failed("First attribute must be after the header (%u).\n", (int)attrs_offset);
276987da915Sopenharmony_ci	}
277987da915Sopenharmony_ci	attr_rec = (ATTR_RECORD *)(buf + attrs_offset);
278987da915Sopenharmony_ci	//printf("uv1.\n");
279987da915Sopenharmony_ci
280987da915Sopenharmony_ci	while ((u8*)attr_rec<=buf+size_of_file_record-4) {
281987da915Sopenharmony_ci
282987da915Sopenharmony_ci		//printf("Attr type: 0x%x.\n", attr_rec->type);
283987da915Sopenharmony_ci		// Check attribute record. (Only what is in the buffer)
284987da915Sopenharmony_ci		if (attr_rec->type==AT_END) {
285987da915Sopenharmony_ci			check_failed("Attribute 0x%x not found in file record at offset %lld (0x%llx).\n", (int)le32_to_cpu(attr_rec->type),
286987da915Sopenharmony_ci					(long long)offset_to_file_record,
287987da915Sopenharmony_ci					(long long)offset_to_file_record);
288987da915Sopenharmony_ci			return NULL;
289987da915Sopenharmony_ci		}
290987da915Sopenharmony_ci		if ((u8*)attr_rec>buf+size_of_file_record-8) {
291987da915Sopenharmony_ci			// not AT_END yet no room for the length field.
292987da915Sopenharmony_ci			check_failed("Attribute 0x%x is not AT_END, yet no "
293987da915Sopenharmony_ci					"room for the length field.\n",
294987da915Sopenharmony_ci					(int)le32_to_cpu(attr_rec->type));
295987da915Sopenharmony_ci			return NULL;
296987da915Sopenharmony_ci		}
297987da915Sopenharmony_ci
298987da915Sopenharmony_ci		length = le32_to_cpu(attr_rec->length);
299987da915Sopenharmony_ci
300987da915Sopenharmony_ci		// Check that this attribute does not overflow the mft_record
301987da915Sopenharmony_ci		if ((u8*)attr_rec+length >= buf+size_of_file_record) {
302987da915Sopenharmony_ci			check_failed("Attribute (0x%x) is larger than FILE record at offset %lld (0x%llx).\n",
303987da915Sopenharmony_ci					(int)le32_to_cpu(attr_rec->type),
304987da915Sopenharmony_ci					(long long)offset_to_file_record,
305987da915Sopenharmony_ci					(long long)offset_to_file_record);
306987da915Sopenharmony_ci			return NULL;
307987da915Sopenharmony_ci		}
308987da915Sopenharmony_ci		// todo: what ATTRIBUTE_LIST (0x20)?
309987da915Sopenharmony_ci
310987da915Sopenharmony_ci		if (attr_rec->type==attr_type) {
311987da915Sopenharmony_ci			// Eurika!
312987da915Sopenharmony_ci
313987da915Sopenharmony_ci			// ntfs_mapping_pairs_decompress only use two values from vol. Just fake it.
314987da915Sopenharmony_ci			// todo: it will also use vol->major_ver if defined(DEBUG). But only for printing purposes.
315987da915Sopenharmony_ci
316987da915Sopenharmony_ci			// Assume ntfs_boot_sector_parse() was called.
317987da915Sopenharmony_ci			return ntfs_mapping_pairs_decompress(rawvol, attr_rec, NULL);
318987da915Sopenharmony_ci		}
319987da915Sopenharmony_ci
320987da915Sopenharmony_ci		attr_rec = (ATTR_RECORD*)((u8*)attr_rec+length);
321987da915Sopenharmony_ci	}
322987da915Sopenharmony_ci	// If we got here, there was an overflow.
323987da915Sopenharmony_ci	check_failed("file record corrupted at offset %lld (0x%llx).\n",
324987da915Sopenharmony_ci			(long long)offset_to_file_record,
325987da915Sopenharmony_ci			(long long)offset_to_file_record);
326987da915Sopenharmony_ci	return NULL;
327987da915Sopenharmony_ci}
328987da915Sopenharmony_ci
329987da915Sopenharmony_ci/**
330987da915Sopenharmony_ci * Return: >=0 last VCN
331987da915Sopenharmony_ci *	   LCN_EINVAL error.
332987da915Sopenharmony_ci */
333987da915Sopenharmony_cistatic VCN get_last_vcn(runlist *rl)
334987da915Sopenharmony_ci{
335987da915Sopenharmony_ci	VCN res;
336987da915Sopenharmony_ci
337987da915Sopenharmony_ci	if (!rl)
338987da915Sopenharmony_ci		return LCN_EINVAL;
339987da915Sopenharmony_ci
340987da915Sopenharmony_ci	res = LCN_EINVAL;
341987da915Sopenharmony_ci	while (rl->length) {
342987da915Sopenharmony_ci		ntfs_log_verbose("vcn: %lld, length: %lld.\n",
343987da915Sopenharmony_ci				(long long)rl->vcn, (long long)rl->length);
344987da915Sopenharmony_ci		if (rl->vcn<0)
345987da915Sopenharmony_ci			res = rl->vcn;
346987da915Sopenharmony_ci		else
347987da915Sopenharmony_ci			res = rl->vcn + rl->length;
348987da915Sopenharmony_ci		rl++;
349987da915Sopenharmony_ci	}
350987da915Sopenharmony_ci
351987da915Sopenharmony_ci	return res;
352987da915Sopenharmony_ci}
353987da915Sopenharmony_ci
354987da915Sopenharmony_cistatic u32 mft_bitmap_records;
355987da915Sopenharmony_cistatic u8 *mft_bitmap_buf;
356987da915Sopenharmony_ci
357987da915Sopenharmony_ci/**
358987da915Sopenharmony_ci * Assumes mft_bitmap_rl is initialized.
359987da915Sopenharmony_ci * return: 0 ok.
360987da915Sopenharmony_ci *	   RETURN_OPERATIONAL_ERROR on error.
361987da915Sopenharmony_ci */
362987da915Sopenharmony_cistatic int mft_bitmap_load(ntfs_volume *rawvol)
363987da915Sopenharmony_ci{
364987da915Sopenharmony_ci	VCN vcn;
365987da915Sopenharmony_ci	u32 mft_bitmap_length;
366987da915Sopenharmony_ci
367987da915Sopenharmony_ci	vcn = get_last_vcn(mft_bitmap_rl);
368987da915Sopenharmony_ci	if (vcn<=LCN_EINVAL) {
369987da915Sopenharmony_ci		mft_bitmap_buf = NULL;
370987da915Sopenharmony_ci		/* This case should not happen, not even with on-disk errors */
371987da915Sopenharmony_ci		goto error;
372987da915Sopenharmony_ci	}
373987da915Sopenharmony_ci
374987da915Sopenharmony_ci	mft_bitmap_length = vcn * rawvol->cluster_size;
375987da915Sopenharmony_ci	mft_bitmap_records = 8 * mft_bitmap_length * rawvol->cluster_size /
376987da915Sopenharmony_ci		rawvol->mft_record_size;
377987da915Sopenharmony_ci
378987da915Sopenharmony_ci	//printf("sizes: %d, %d.\n", mft_bitmap_length, mft_bitmap_records);
379987da915Sopenharmony_ci
380987da915Sopenharmony_ci	mft_bitmap_buf = (u8*)ntfs_malloc(mft_bitmap_length);
381987da915Sopenharmony_ci	if (!mft_bitmap_buf)
382987da915Sopenharmony_ci		goto error;
383987da915Sopenharmony_ci	if (ntfs_rl_pread(rawvol, mft_bitmap_rl, 0, mft_bitmap_length,
384987da915Sopenharmony_ci			mft_bitmap_buf)!=mft_bitmap_length)
385987da915Sopenharmony_ci		goto error;
386987da915Sopenharmony_ci	return 0;
387987da915Sopenharmony_cierror:
388987da915Sopenharmony_ci	mft_bitmap_records = 0;
389987da915Sopenharmony_ci	ntfs_log_error("Could not load $MFT/Bitmap.\n");
390987da915Sopenharmony_ci	return RETURN_OPERATIONAL_ERROR;
391987da915Sopenharmony_ci}
392987da915Sopenharmony_ci
393987da915Sopenharmony_ci/**
394987da915Sopenharmony_ci * -1 Error.
395987da915Sopenharmony_ci *  0 Unused record
396987da915Sopenharmony_ci *  1 Used record
397987da915Sopenharmony_ci *
398987da915Sopenharmony_ci * Assumes mft_bitmap_rl was initialized.
399987da915Sopenharmony_ci */
400987da915Sopenharmony_cistatic int mft_bitmap_get_bit(s64 mft_no)
401987da915Sopenharmony_ci{
402987da915Sopenharmony_ci	if (mft_no>=mft_bitmap_records)
403987da915Sopenharmony_ci		return -1;
404987da915Sopenharmony_ci	return ntfs_bit_get(mft_bitmap_buf, mft_no);
405987da915Sopenharmony_ci}
406987da915Sopenharmony_ci
407987da915Sopenharmony_ci/**
408987da915Sopenharmony_ci * @attr_rec: The attribute record to check
409987da915Sopenharmony_ci * @mft_rec: The parent FILE record.
410987da915Sopenharmony_ci * @buflen: The size of the FILE record.
411987da915Sopenharmony_ci *
412987da915Sopenharmony_ci * Return:
413987da915Sopenharmony_ci *  NULL: Fatal error occured. Not sure where is the next record.
414987da915Sopenharmony_ci *  otherwise: pointer to the next attribute record.
415987da915Sopenharmony_ci *
416987da915Sopenharmony_ci * The function only check fields that are inside this attr record.
417987da915Sopenharmony_ci *
418987da915Sopenharmony_ci * Assumes mft_rec is current_mft_record.
419987da915Sopenharmony_ci */
420987da915Sopenharmony_cistatic ATTR_REC *check_attr_record(ATTR_REC *attr_rec, MFT_RECORD *mft_rec,
421987da915Sopenharmony_ci			u16 buflen)
422987da915Sopenharmony_ci{
423987da915Sopenharmony_ci	u16 name_offset;
424987da915Sopenharmony_ci	u16 attrs_offset = le16_to_cpu(mft_rec->attrs_offset);
425987da915Sopenharmony_ci	u32 attr_type = le32_to_cpu(attr_rec->type);
426987da915Sopenharmony_ci	u32 length = le32_to_cpu(attr_rec->length);
427987da915Sopenharmony_ci
428987da915Sopenharmony_ci	// Check that this attribute does not overflow the mft_record
429987da915Sopenharmony_ci	if ((u8*)attr_rec+length >= ((u8*)mft_rec)+buflen) {
430987da915Sopenharmony_ci		check_failed("Attribute (0x%x) is larger than FILE record (%lld).\n",
431987da915Sopenharmony_ci				(int)attr_type, (long long)current_mft_record);
432987da915Sopenharmony_ci		return NULL;
433987da915Sopenharmony_ci	}
434987da915Sopenharmony_ci
435987da915Sopenharmony_ci	// Attr type must be a multiple of 0x10 and 0x10<=x<=0x100.
436987da915Sopenharmony_ci	if ((attr_type & ~0x0F0) && (attr_type != 0x100)) {
437987da915Sopenharmony_ci		check_failed("Unknown attribute type 0x%x.\n",
438987da915Sopenharmony_ci			(int)attr_type);
439987da915Sopenharmony_ci		goto check_attr_record_next_attr;
440987da915Sopenharmony_ci	}
441987da915Sopenharmony_ci
442987da915Sopenharmony_ci	if (length<24) {
443987da915Sopenharmony_ci		check_failed("Attribute %lld:0x%x Length too short (%u).\n",
444987da915Sopenharmony_ci			(long long)current_mft_record, (int)attr_type,
445987da915Sopenharmony_ci			(int)length);
446987da915Sopenharmony_ci		goto check_attr_record_next_attr;
447987da915Sopenharmony_ci	}
448987da915Sopenharmony_ci
449987da915Sopenharmony_ci	// If this is the first attribute:
450987da915Sopenharmony_ci	// todo: instance number must be smaller than next_instance.
451987da915Sopenharmony_ci	if ((u8*)attr_rec == ((u8*)mft_rec) + attrs_offset) {
452987da915Sopenharmony_ci		if (!mft_rec->base_mft_record)
453987da915Sopenharmony_ci			assert_u32_equal(attr_type, 0x10,
454987da915Sopenharmony_ci				"First attribute type");
455987da915Sopenharmony_ci		// The following not always holds.
456987da915Sopenharmony_ci		// attr 0x10 becomes instance 1 and attr 0x40 becomes 0.
457987da915Sopenharmony_ci		//assert_u32_equal(attr_rec->instance, 0,
458987da915Sopenharmony_ci		//	"First attribute instance number");
459987da915Sopenharmony_ci	} else {
460987da915Sopenharmony_ci		assert_u32_noteq(attr_type, 0x10,
461987da915Sopenharmony_ci			"Not-first attribute type");
462987da915Sopenharmony_ci		// The following not always holds.
463987da915Sopenharmony_ci		//assert_u32_noteq(attr_rec->instance, 0,
464987da915Sopenharmony_ci		//	"Not-first attribute instance number");
465987da915Sopenharmony_ci	}
466987da915Sopenharmony_ci	//if (current_mft_record==938 || current_mft_record==1683 || current_mft_record==3152 || current_mft_record==22410)
467987da915Sopenharmony_ci	//printf("Attribute %lld:0x%x instance: %u isbase:%d.\n",
468987da915Sopenharmony_ci	//		current_mft_record, (int)attr_type, (int)le16_to_cpu(attr_rec->instance), (int)mft_rec->base_mft_record);
469987da915Sopenharmony_ci	// todo: instance is unique.
470987da915Sopenharmony_ci
471987da915Sopenharmony_ci	// Check flags.
472987da915Sopenharmony_ci	if (attr_rec->flags & ~(const_cpu_to_le16(0xc0ff))) {
473987da915Sopenharmony_ci		check_failed("Attribute %lld:0x%x Unknown flags (0x%x).\n",
474987da915Sopenharmony_ci			(long long)current_mft_record, (int)attr_type,
475987da915Sopenharmony_ci			(int)le16_to_cpu(attr_rec->flags));
476987da915Sopenharmony_ci	}
477987da915Sopenharmony_ci
478987da915Sopenharmony_ci	if (attr_rec->non_resident>1) {
479987da915Sopenharmony_ci		check_failed("Attribute %lld:0x%x Unknown non-resident "
480987da915Sopenharmony_ci			"flag (0x%x).\n", (long long)current_mft_record,
481987da915Sopenharmony_ci			(int)attr_type, (int)attr_rec->non_resident);
482987da915Sopenharmony_ci		goto check_attr_record_next_attr;
483987da915Sopenharmony_ci	}
484987da915Sopenharmony_ci
485987da915Sopenharmony_ci	name_offset = le16_to_cpu(attr_rec->name_offset);
486987da915Sopenharmony_ci	/*
487987da915Sopenharmony_ci	 * todo: name must be legal unicode.
488987da915Sopenharmony_ci	 * Not really, information below in urls is about filenames, but I
489987da915Sopenharmony_ci	 * believe it also applies to attribute names.  (Yura)
490987da915Sopenharmony_ci	 *  http://blogs.msdn.com/michkap/archive/2006/09/24/769540.aspx
491987da915Sopenharmony_ci	 *  http://blogs.msdn.com/michkap/archive/2006/09/10/748699.aspx
492987da915Sopenharmony_ci	 */
493987da915Sopenharmony_ci
494987da915Sopenharmony_ci	if (attr_rec->non_resident) {
495987da915Sopenharmony_ci		// Non-Resident
496987da915Sopenharmony_ci
497987da915Sopenharmony_ci		// Make sure all the fields exist.
498987da915Sopenharmony_ci		if (length<64) {
499987da915Sopenharmony_ci			check_failed("Non-resident attribute %lld:0x%x too short (%u).\n",
500987da915Sopenharmony_ci				(long long)current_mft_record, (int)attr_type,
501987da915Sopenharmony_ci				(int)length);
502987da915Sopenharmony_ci			goto check_attr_record_next_attr;
503987da915Sopenharmony_ci		}
504987da915Sopenharmony_ci		if (attr_rec->compression_unit && (length<72)) {
505987da915Sopenharmony_ci			check_failed("Compressed attribute %lld:0x%x too short (%u).\n",
506987da915Sopenharmony_ci				(long long)current_mft_record, (int)attr_type,
507987da915Sopenharmony_ci				(int)length);
508987da915Sopenharmony_ci			goto check_attr_record_next_attr;
509987da915Sopenharmony_ci		}
510987da915Sopenharmony_ci
511987da915Sopenharmony_ci		// todo: name comes before mapping pairs, and after the header.
512987da915Sopenharmony_ci		// todo: length==mapping_pairs_offset+length of compressed mapping pairs.
513987da915Sopenharmony_ci		// todo: mapping_pairs_offset is 8-byte aligned.
514987da915Sopenharmony_ci
515987da915Sopenharmony_ci		// todo: lowest vcn <= highest_vcn
516987da915Sopenharmony_ci		// todo: if base record -> lowest vcn==0
517987da915Sopenharmony_ci		// todo: lowest_vcn!=0 -> attribute list is used.
518987da915Sopenharmony_ci		// todo: lowest_vcn & highest_vcn are in the drive (0<=x<total clusters)
519987da915Sopenharmony_ci		// todo: mapping pairs agree with highest_vcn.
520987da915Sopenharmony_ci		// todo: compression unit == 0 or 4.
521987da915Sopenharmony_ci		// todo: reserved1 == 0.
522987da915Sopenharmony_ci		// todo: if not compressed nor sparse, initialized_size <= allocated_size and data_size <= allocated_size.
523987da915Sopenharmony_ci		// todo: if compressed or sparse, allocated_size <= initialized_size and allocated_size <= data_size
524987da915Sopenharmony_ci		// todo: if mft_no!=0 and not compressed/sparse, data_size==initialized_size.
525987da915Sopenharmony_ci		// todo: if mft_no!=0 and compressed/sparse, allocated_size==initialized_size.
526987da915Sopenharmony_ci		// todo: what about compressed_size if compressed?
527987da915Sopenharmony_ci		// todo: attribute must not be 0x10, 0x30, 0x40, 0x60, 0x70, 0x90, 0xd0 (not sure about 0xb0, 0xe0, 0xf0)
528987da915Sopenharmony_ci	} else {
529987da915Sopenharmony_ci		u16 value_offset = le16_to_cpu(attr_rec->value_offset);
530987da915Sopenharmony_ci		u32 value_length = le32_to_cpu(attr_rec->value_length);
531987da915Sopenharmony_ci		// Resident
532987da915Sopenharmony_ci		if (attr_rec->name_length) {
533987da915Sopenharmony_ci			if (name_offset < 24)
534987da915Sopenharmony_ci				check_failed("Resident attribute with "
535987da915Sopenharmony_ci					"name intersecting header.\n");
536987da915Sopenharmony_ci			if (value_offset < name_offset +
537987da915Sopenharmony_ci					attr_rec->name_length)
538987da915Sopenharmony_ci				check_failed("Named resident attribute "
539987da915Sopenharmony_ci					"with value before name.\n");
540987da915Sopenharmony_ci		}
541987da915Sopenharmony_ci		// if resident, length==value_length+value_offset
542987da915Sopenharmony_ci		//assert_u32_equal(le32_to_cpu(attr_rec->value_length)+
543987da915Sopenharmony_ci		//	value_offset, length,
544987da915Sopenharmony_ci		//	"length==value_length+value_offset");
545987da915Sopenharmony_ci		// if resident, length==value_length+value_offset
546987da915Sopenharmony_ci		if (value_length+value_offset > length) {
547987da915Sopenharmony_ci			check_failed("value_length(%d)+value_offset(%d)>length(%d) for attribute 0x%x.\n", (int)value_length, (int)value_offset, (int)length, (int)attr_type);
548987da915Sopenharmony_ci			return NULL;
549987da915Sopenharmony_ci		}
550987da915Sopenharmony_ci
551987da915Sopenharmony_ci		// Check resident_flags.
552987da915Sopenharmony_ci		if (attr_rec->resident_flags>0x01) {
553987da915Sopenharmony_ci			check_failed("Unknown resident flags (0x%x) for attribute 0x%x.\n", (int)attr_rec->resident_flags, (int)attr_type);
554987da915Sopenharmony_ci		} else if (attr_rec->resident_flags && (attr_type!=0x30)) {
555987da915Sopenharmony_ci			check_failed("Resident flags mark attribute 0x%x as indexed.\n", (int)attr_type);
556987da915Sopenharmony_ci		}
557987da915Sopenharmony_ci
558987da915Sopenharmony_ci		// reservedR is 0.
559987da915Sopenharmony_ci		assert_u32_equal(attr_rec->reservedR, 0, "Resident Reserved");
560987da915Sopenharmony_ci
561987da915Sopenharmony_ci		// todo: attribute must not be 0xa0 (not sure about 0xb0, 0xe0, 0xf0)
562987da915Sopenharmony_ci		// todo: check content well-formness per attr_type.
563987da915Sopenharmony_ci	}
564987da915Sopenharmony_ci	return 0;
565987da915Sopenharmony_cicheck_attr_record_next_attr:
566987da915Sopenharmony_ci	return (ATTR_REC *)(((u8 *)attr_rec) + length);
567987da915Sopenharmony_ci}
568987da915Sopenharmony_ci
569987da915Sopenharmony_ci/**
570987da915Sopenharmony_ci * All checks that can be satisfied only by data from the buffer.
571987da915Sopenharmony_ci * No other [MFT records/metadata files] are required.
572987da915Sopenharmony_ci *
573987da915Sopenharmony_ci * The buffer is changed by removing the Update Sequence.
574987da915Sopenharmony_ci *
575987da915Sopenharmony_ci * Return:
576987da915Sopenharmony_ci *	0	Everything's cool.
577987da915Sopenharmony_ci *	else	Consider this record as damaged.
578987da915Sopenharmony_ci */
579987da915Sopenharmony_cistatic BOOL check_file_record(u8 *buffer, u16 buflen)
580987da915Sopenharmony_ci{
581987da915Sopenharmony_ci	u16 usa_count, usa_ofs, attrs_offset, usa;
582987da915Sopenharmony_ci	u32 bytes_in_use, bytes_allocated, i;
583987da915Sopenharmony_ci	MFT_RECORD *mft_rec = (MFT_RECORD *)buffer;
584987da915Sopenharmony_ci	ATTR_REC *attr_rec;
585987da915Sopenharmony_ci
586987da915Sopenharmony_ci	// check record magic
587987da915Sopenharmony_ci	assert_u32_equal(le32_to_cpu(mft_rec->magic), le32_to_cpu(magic_FILE), "FILE record magic");
588987da915Sopenharmony_ci	// todo: records 16-23 must be filled in order.
589987da915Sopenharmony_ci	// todo: what to do with magic_BAAD?
590987da915Sopenharmony_ci
591987da915Sopenharmony_ci	// check usa_count+offset to update seq <= attrs_offset <
592987da915Sopenharmony_ci	//	bytes_in_use <= bytes_allocated <= buflen.
593987da915Sopenharmony_ci	usa_ofs = le16_to_cpu(mft_rec->usa_ofs);
594987da915Sopenharmony_ci	usa_count = le16_to_cpu(mft_rec->usa_count);
595987da915Sopenharmony_ci	attrs_offset = le16_to_cpu(mft_rec->attrs_offset);
596987da915Sopenharmony_ci	bytes_in_use = le32_to_cpu(mft_rec->bytes_in_use);
597987da915Sopenharmony_ci	bytes_allocated = le32_to_cpu(mft_rec->bytes_allocated);
598987da915Sopenharmony_ci	if (assert_u32_lesseq(usa_ofs+usa_count, attrs_offset,
599987da915Sopenharmony_ci				"usa_ofs+usa_count <= attrs_offset") ||
600987da915Sopenharmony_ci			assert_u32_less(attrs_offset, bytes_in_use,
601987da915Sopenharmony_ci				"attrs_offset < bytes_in_use") ||
602987da915Sopenharmony_ci			assert_u32_lesseq(bytes_in_use, bytes_allocated,
603987da915Sopenharmony_ci				"bytes_in_use <= bytes_allocated") ||
604987da915Sopenharmony_ci			assert_u32_lesseq(bytes_allocated, buflen,
605987da915Sopenharmony_ci				"bytes_allocated <= max_record_size")) {
606987da915Sopenharmony_ci		return 1;
607987da915Sopenharmony_ci	}
608987da915Sopenharmony_ci
609987da915Sopenharmony_ci
610987da915Sopenharmony_ci	// We should know all the flags.
611987da915Sopenharmony_ci	if (le16_to_cpu(mft_rec->flags) > 0xf) {
612987da915Sopenharmony_ci		check_failed("Unknown MFT record flags (0x%x).\n",
613987da915Sopenharmony_ci			(unsigned int)le16_to_cpu(mft_rec->flags));
614987da915Sopenharmony_ci	}
615987da915Sopenharmony_ci	// todo: flag in_use must be on.
616987da915Sopenharmony_ci
617987da915Sopenharmony_ci	// Remove update seq & check it.
618987da915Sopenharmony_ci	usa = *(u16*)(buffer+usa_ofs); // The value that should be at the end of every sector.
619987da915Sopenharmony_ci	if (assert_u32_equal(usa_count-1, buflen/NTFS_BLOCK_SIZE, "USA length"))
620987da915Sopenharmony_ci		return (1);
621987da915Sopenharmony_ci	for (i=1;i<usa_count;i++) {
622987da915Sopenharmony_ci		u16 *fixup = (u16*)(buffer+NTFS_BLOCK_SIZE*i-2); // the value at the end of the sector.
623987da915Sopenharmony_ci		u16 saved_val = *(u16*)(buffer+usa_ofs+2*i); // the actual data value that was saved in the us array.
624987da915Sopenharmony_ci
625987da915Sopenharmony_ci		assert_u32_equal(*fixup, usa, "fixup");
626987da915Sopenharmony_ci		*fixup = saved_val; // remove it.
627987da915Sopenharmony_ci	}
628987da915Sopenharmony_ci
629987da915Sopenharmony_ci	attr_rec = (ATTR_REC *)(buffer + attrs_offset);
630987da915Sopenharmony_ci	while ((u8*)attr_rec<=buffer+buflen-4) {
631987da915Sopenharmony_ci
632987da915Sopenharmony_ci		// Check attribute record. (Only what is in the buffer)
633987da915Sopenharmony_ci		if (attr_rec->type==AT_END) {
634987da915Sopenharmony_ci			// Done.
635987da915Sopenharmony_ci			return 0;
636987da915Sopenharmony_ci		}
637987da915Sopenharmony_ci		if ((u8*)attr_rec>buffer+buflen-8) {
638987da915Sopenharmony_ci			// not AT_END yet no room for the length field.
639987da915Sopenharmony_ci			check_failed("Attribute 0x%x is not AT_END, yet no "
640987da915Sopenharmony_ci					"room for the length field.\n",
641987da915Sopenharmony_ci					(int)le32_to_cpu(attr_rec->type));
642987da915Sopenharmony_ci			return 1;
643987da915Sopenharmony_ci		}
644987da915Sopenharmony_ci
645987da915Sopenharmony_ci		attr_rec = check_attr_record(attr_rec, mft_rec, buflen);
646987da915Sopenharmony_ci		if (!attr_rec)
647987da915Sopenharmony_ci			return 1;
648987da915Sopenharmony_ci	}
649987da915Sopenharmony_ci	// If we got here, there was an overflow.
650987da915Sopenharmony_ci	return 1;
651987da915Sopenharmony_ci
652987da915Sopenharmony_ci	// todo: an attribute should be at the offset to first attribute, and the offset should be inside the buffer. It should have the value of "next attribute id".
653987da915Sopenharmony_ci	// todo: if base record, it should start with attribute 0x10.
654987da915Sopenharmony_ci
655987da915Sopenharmony_ci	// Highlevel check of attributes.
656987da915Sopenharmony_ci	//  todo: Attributes are well-formed.
657987da915Sopenharmony_ci	//  todo: Room for next attribute in the end of the previous record.
658987da915Sopenharmony_ci
659987da915Sopenharmony_ci	return FALSE;
660987da915Sopenharmony_ci}
661987da915Sopenharmony_ci
662987da915Sopenharmony_cistatic void replay_log(ntfs_volume *vol __attribute__((unused)))
663987da915Sopenharmony_ci{
664987da915Sopenharmony_ci	// At this time, only check that the log is fully replayed.
665987da915Sopenharmony_ci	ntfs_log_warning("Unsupported: replay_log()\n");
666987da915Sopenharmony_ci	// todo: if logfile is clean, return success.
667987da915Sopenharmony_ci	unsupported++;
668987da915Sopenharmony_ci}
669987da915Sopenharmony_ci
670987da915Sopenharmony_cistatic void verify_mft_record(ntfs_volume *vol, s64 mft_num)
671987da915Sopenharmony_ci{
672987da915Sopenharmony_ci	u8 *buffer;
673987da915Sopenharmony_ci	int is_used;
674987da915Sopenharmony_ci
675987da915Sopenharmony_ci	current_mft_record = mft_num;
676987da915Sopenharmony_ci
677987da915Sopenharmony_ci	is_used = mft_bitmap_get_bit(mft_num);
678987da915Sopenharmony_ci	if (is_used<0) {
679987da915Sopenharmony_ci		ntfs_log_error("Error getting bit value for record %lld.\n",
680987da915Sopenharmony_ci			(long long)mft_num);
681987da915Sopenharmony_ci	} else if (!is_used) {
682987da915Sopenharmony_ci		ntfs_log_verbose("Record %lld unused. Skipping.\n",
683987da915Sopenharmony_ci				(long long)mft_num);
684987da915Sopenharmony_ci		return;
685987da915Sopenharmony_ci	}
686987da915Sopenharmony_ci
687987da915Sopenharmony_ci	buffer = ntfs_malloc(vol->mft_record_size);
688987da915Sopenharmony_ci	if (!buffer)
689987da915Sopenharmony_ci		goto verify_mft_record_error;
690987da915Sopenharmony_ci
691987da915Sopenharmony_ci	ntfs_log_verbose("MFT record %lld\n", (long long)mft_num);
692987da915Sopenharmony_ci	if (ntfs_attr_pread(vol->mft_na, mft_num*vol->mft_record_size, vol->mft_record_size, buffer) < 0) {
693987da915Sopenharmony_ci		ntfs_log_perror("Couldn't read $MFT record %lld", (long long)mft_num);
694987da915Sopenharmony_ci		goto verify_mft_record_error;
695987da915Sopenharmony_ci	}
696987da915Sopenharmony_ci
697987da915Sopenharmony_ci	check_file_record(buffer, vol->mft_record_size);
698987da915Sopenharmony_ci	// todo: if offset to first attribute >= 0x30, number of mft record should match.
699987da915Sopenharmony_ci	// todo: Match the "record is used" with the mft bitmap.
700987da915Sopenharmony_ci	// todo: if this is not base, check that the parent is a base, and is in use, and pointing to this record.
701987da915Sopenharmony_ci
702987da915Sopenharmony_ci	// todo: if base record: for each extent record:
703987da915Sopenharmony_ci	//   todo: verify_file_record
704987da915Sopenharmony_ci	//   todo: hard link count should be the number of 0x30 attributes.
705987da915Sopenharmony_ci	//   todo: Order of attributes.
706987da915Sopenharmony_ci	//   todo: make sure compression_unit is the same.
707987da915Sopenharmony_ci
708987da915Sopenharmony_ci	return;
709987da915Sopenharmony_civerify_mft_record_error:
710987da915Sopenharmony_ci
711987da915Sopenharmony_ci	if (buffer)
712987da915Sopenharmony_ci		free(buffer);
713987da915Sopenharmony_ci	errors++;
714987da915Sopenharmony_ci}
715987da915Sopenharmony_ci
716987da915Sopenharmony_ci/**
717987da915Sopenharmony_ci * This function serves as bootstraping for the more comprehensive checks.
718987da915Sopenharmony_ci * It will load the MFT runlist and MFT/Bitmap runlist.
719987da915Sopenharmony_ci * It should not depend on other checks or we may have a circular dependancy.
720987da915Sopenharmony_ci * Also, this loadng must be forgiving, unlike the comprehensive checks.
721987da915Sopenharmony_ci */
722987da915Sopenharmony_cistatic int verify_mft_preliminary(ntfs_volume *rawvol)
723987da915Sopenharmony_ci{
724987da915Sopenharmony_ci	current_mft_record = 0;
725987da915Sopenharmony_ci	s64 mft_offset, mftmirr_offset;
726987da915Sopenharmony_ci	int res;
727987da915Sopenharmony_ci
728987da915Sopenharmony_ci	ntfs_log_trace("Entering verify_mft_preliminary().\n");
729987da915Sopenharmony_ci	// todo: get size_of_file_record from boot sector
730987da915Sopenharmony_ci	// Load the first segment of the $MFT/DATA runlist.
731987da915Sopenharmony_ci	mft_offset = rawvol->mft_lcn * rawvol->cluster_size;
732987da915Sopenharmony_ci	mftmirr_offset = rawvol->mftmirr_lcn * rawvol->cluster_size;
733987da915Sopenharmony_ci	mft_rl = load_runlist(rawvol, mft_offset, AT_DATA, 1024);
734987da915Sopenharmony_ci	if (!mft_rl) {
735987da915Sopenharmony_ci		check_failed("Loading $MFT runlist failed. Trying $MFTMirr.\n");
736987da915Sopenharmony_ci		mft_rl = load_runlist(rawvol, mftmirr_offset, AT_DATA, 1024);
737987da915Sopenharmony_ci	}
738987da915Sopenharmony_ci	if (!mft_rl) {
739987da915Sopenharmony_ci		check_failed("Loading $MFTMirr runlist failed too. Aborting.\n");
740987da915Sopenharmony_ci		return RETURN_FS_ERRORS_LEFT_UNCORRECTED | RETURN_OPERATIONAL_ERROR;
741987da915Sopenharmony_ci	}
742987da915Sopenharmony_ci	// TODO: else { recover $MFT } // Use $MFTMirr to recover $MFT.
743987da915Sopenharmony_ci	// todo: support loading the next runlist extents when ATTRIBUTE_LIST is used on $MFT.
744987da915Sopenharmony_ci	// If attribute list: Gradually load mft runlist. (parse runlist from first file record, check all referenced file records, continue with the next file record). If no attribute list, just load it.
745987da915Sopenharmony_ci
746987da915Sopenharmony_ci	// Load the runlist of $MFT/Bitmap.
747987da915Sopenharmony_ci	// todo: what about ATTRIBUTE_LIST? Can we reuse code?
748987da915Sopenharmony_ci	mft_bitmap_rl = load_runlist(rawvol, mft_offset, AT_BITMAP, 1024);
749987da915Sopenharmony_ci	if (!mft_bitmap_rl) {
750987da915Sopenharmony_ci		check_failed("Loading $MFT/Bitmap runlist failed. Trying $MFTMirr.\n");
751987da915Sopenharmony_ci		mft_bitmap_rl = load_runlist(rawvol, mftmirr_offset, AT_BITMAP, 1024);
752987da915Sopenharmony_ci	}
753987da915Sopenharmony_ci	if (!mft_bitmap_rl) {
754987da915Sopenharmony_ci		check_failed("Loading $MFTMirr/Bitmap runlist failed too. Aborting.\n");
755987da915Sopenharmony_ci		return RETURN_FS_ERRORS_LEFT_UNCORRECTED;
756987da915Sopenharmony_ci		// todo: rebuild the bitmap by using the "in_use" file record flag or by filling it with 1's.
757987da915Sopenharmony_ci	}
758987da915Sopenharmony_ci
759987da915Sopenharmony_ci	/* Load $MFT/Bitmap */
760987da915Sopenharmony_ci	if ((res = mft_bitmap_load(rawvol)))
761987da915Sopenharmony_ci		return res;
762987da915Sopenharmony_ci	return -1; /* FIXME: Just added to fix compiler warning without
763987da915Sopenharmony_ci			thinking about what should be here.  (Yura) */
764987da915Sopenharmony_ci}
765987da915Sopenharmony_ci
766987da915Sopenharmony_cistatic void check_volume(ntfs_volume *vol)
767987da915Sopenharmony_ci{
768987da915Sopenharmony_ci	s64 mft_num, nr_mft_records;
769987da915Sopenharmony_ci
770987da915Sopenharmony_ci	ntfs_log_warning("Unsupported: check_volume()\n");
771987da915Sopenharmony_ci	unsupported++;
772987da915Sopenharmony_ci
773987da915Sopenharmony_ci	// For each mft record, verify that it contains a valid file record.
774987da915Sopenharmony_ci	nr_mft_records = vol->mft_na->initialized_size >>
775987da915Sopenharmony_ci			vol->mft_record_size_bits;
776987da915Sopenharmony_ci	ntfs_log_info("Checking %lld MFT records.\n", (long long)nr_mft_records);
777987da915Sopenharmony_ci
778987da915Sopenharmony_ci	for (mft_num=0; mft_num < nr_mft_records; mft_num++) {
779987da915Sopenharmony_ci	 	verify_mft_record(vol, mft_num);
780987da915Sopenharmony_ci	}
781987da915Sopenharmony_ci
782987da915Sopenharmony_ci	// todo: Check metadata files.
783987da915Sopenharmony_ci
784987da915Sopenharmony_ci	// todo: Second pass on mft records. Now check the contents as well.
785987da915Sopenharmony_ci	// todo: When going through runlists, build a bitmap.
786987da915Sopenharmony_ci
787987da915Sopenharmony_ci	// todo: cluster accounting.
788987da915Sopenharmony_ci	return;
789987da915Sopenharmony_ci}
790987da915Sopenharmony_ci
791987da915Sopenharmony_cistatic int reset_dirty(ntfs_volume *vol)
792987da915Sopenharmony_ci{
793987da915Sopenharmony_ci	le16 flags;
794987da915Sopenharmony_ci
795987da915Sopenharmony_ci	if (!(vol->flags | VOLUME_IS_DIRTY))
796987da915Sopenharmony_ci		return 0;
797987da915Sopenharmony_ci
798987da915Sopenharmony_ci	ntfs_log_verbose("Resetting dirty flag.\n");
799987da915Sopenharmony_ci
800987da915Sopenharmony_ci	flags = vol->flags & ~VOLUME_IS_DIRTY;
801987da915Sopenharmony_ci
802987da915Sopenharmony_ci	if (ntfs_volume_write_flags(vol, flags)) {
803987da915Sopenharmony_ci		ntfs_log_error("Error setting volume flags.\n");
804987da915Sopenharmony_ci		return -1;
805987da915Sopenharmony_ci	}
806987da915Sopenharmony_ci	return 0;
807987da915Sopenharmony_ci}
808987da915Sopenharmony_ci
809987da915Sopenharmony_ci/**
810987da915Sopenharmony_ci * main - Does just what C99 claim it does.
811987da915Sopenharmony_ci *
812987da915Sopenharmony_ci * For more details on arguments and results, check the man page.
813987da915Sopenharmony_ci */
814987da915Sopenharmony_ciint main(int argc, char **argv)
815987da915Sopenharmony_ci{
816987da915Sopenharmony_ci	struct ntfs_device *dev;
817987da915Sopenharmony_ci	ntfs_volume rawvol;
818987da915Sopenharmony_ci	ntfs_volume *vol;
819987da915Sopenharmony_ci	const char *name;
820987da915Sopenharmony_ci	int ret;
821987da915Sopenharmony_ci
822987da915Sopenharmony_ci	if (argc != 2)
823987da915Sopenharmony_ci		return RETURN_USAGE_OR_SYNTAX_ERROR;
824987da915Sopenharmony_ci	name = argv[1];
825987da915Sopenharmony_ci
826987da915Sopenharmony_ci	ntfs_log_set_handler(ntfs_log_handler_outerr);
827987da915Sopenharmony_ci	//ntfs_log_set_levels(NTFS_LOG_LEVEL_DEBUG | NTFS_LOG_LEVEL_TRACE | NTFS_LOG_LEVEL_QUIET | NTFS_LOG_LEVEL_INFO | NTFS_LOG_LEVEL_VERBOSE | NTFS_LOG_LEVEL_PROGRESS);
828987da915Sopenharmony_ci
829987da915Sopenharmony_ci	/* Allocate an ntfs_device structure. */
830987da915Sopenharmony_ci	dev = ntfs_device_alloc(name, 0, &ntfs_device_default_io_ops, NULL);
831987da915Sopenharmony_ci	if (!dev)
832987da915Sopenharmony_ci		return RETURN_OPERATIONAL_ERROR;
833987da915Sopenharmony_ci	if (dev->d_ops->open(dev, O_RDONLY)) { //O_RDWR/O_RDONLY?
834987da915Sopenharmony_ci		ntfs_log_perror("Error opening partition device");
835987da915Sopenharmony_ci		ntfs_device_free(dev);
836987da915Sopenharmony_ci		return RETURN_OPERATIONAL_ERROR;
837987da915Sopenharmony_ci	}
838987da915Sopenharmony_ci
839987da915Sopenharmony_ci	if ((ret = verify_boot_sector(dev,&rawvol))) {
840987da915Sopenharmony_ci		dev->d_ops->close(dev);
841987da915Sopenharmony_ci		return ret;
842987da915Sopenharmony_ci	}
843987da915Sopenharmony_ci	ntfs_log_verbose("Boot sector verification complete. Proceeding to $MFT");
844987da915Sopenharmony_ci
845987da915Sopenharmony_ci	verify_mft_preliminary(&rawvol);
846987da915Sopenharmony_ci
847987da915Sopenharmony_ci	/* ntfs_device_mount() expects the device to be closed. */
848987da915Sopenharmony_ci	if (dev->d_ops->close(dev))
849987da915Sopenharmony_ci		ntfs_log_perror("Failed to close the device.");
850987da915Sopenharmony_ci
851987da915Sopenharmony_ci	// at this point we know that the volume is valid enough for mounting.
852987da915Sopenharmony_ci
853987da915Sopenharmony_ci	/* Call ntfs_device_mount() to do the actual mount. */
854987da915Sopenharmony_ci	vol = ntfs_device_mount(dev, NTFS_MNT_RDONLY);
855987da915Sopenharmony_ci	if (!vol) {
856987da915Sopenharmony_ci		ntfs_device_free(dev);
857987da915Sopenharmony_ci		return 2;
858987da915Sopenharmony_ci	}
859987da915Sopenharmony_ci
860987da915Sopenharmony_ci	replay_log(vol);
861987da915Sopenharmony_ci
862987da915Sopenharmony_ci	if (vol->flags & VOLUME_IS_DIRTY)
863987da915Sopenharmony_ci		ntfs_log_warning("Volume is dirty.\n");
864987da915Sopenharmony_ci
865987da915Sopenharmony_ci	check_volume(vol);
866987da915Sopenharmony_ci
867987da915Sopenharmony_ci	if (errors)
868987da915Sopenharmony_ci		ntfs_log_info("Errors found.\n");
869987da915Sopenharmony_ci	if (unsupported)
870987da915Sopenharmony_ci		ntfs_log_info("Unsupported cases found.\n");
871987da915Sopenharmony_ci
872987da915Sopenharmony_ci	if (!errors && !unsupported) {
873987da915Sopenharmony_ci		reset_dirty(vol);
874987da915Sopenharmony_ci	}
875987da915Sopenharmony_ci
876987da915Sopenharmony_ci	ntfs_umount(vol, FALSE);
877987da915Sopenharmony_ci
878987da915Sopenharmony_ci	if (errors)
879987da915Sopenharmony_ci		return 2;
880987da915Sopenharmony_ci	if (unsupported)
881987da915Sopenharmony_ci		return 1;
882987da915Sopenharmony_ci	return 0;
883987da915Sopenharmony_ci}
884987da915Sopenharmony_ci
885