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