xref: /onnv-gate/usr/src/cmd/ntfsprogs/ntfsfix.c (revision 10465:f9789e1a1626)
19663SMark.Logan@Sun.COM /**
29663SMark.Logan@Sun.COM  * ntfsfix - Part of the Linux-NTFS project.
39663SMark.Logan@Sun.COM  *
49663SMark.Logan@Sun.COM  * Copyright (c) 2000-2006 Anton Altaparmakov
59663SMark.Logan@Sun.COM  * Copyright (c) 2002-2006 Szabolcs Szakacsits
69663SMark.Logan@Sun.COM  * Copyright (c) 2007      Yura Pakhuchiy
79663SMark.Logan@Sun.COM  *
89663SMark.Logan@Sun.COM  * This utility fixes some common NTFS problems, resets the NTFS journal file
99663SMark.Logan@Sun.COM  * and schedules an NTFS consistency check for the first boot into Windows.
109663SMark.Logan@Sun.COM  *
119663SMark.Logan@Sun.COM  *	Anton Altaparmakov <aia21@cantab.net>
129663SMark.Logan@Sun.COM  *
139663SMark.Logan@Sun.COM  * This program is free software; you can redistribute it and/or modify
149663SMark.Logan@Sun.COM  * it under the terms of the GNU General Public License as published by
159663SMark.Logan@Sun.COM  * the Free Software Foundation; either version 2 of the License, or
169663SMark.Logan@Sun.COM  * (at your option) any later version.
179663SMark.Logan@Sun.COM  *
189663SMark.Logan@Sun.COM  * This program is distributed in the hope that it will be useful,
199663SMark.Logan@Sun.COM  * but WITHOUT ANY WARRANTY; without even the implied warranty of
209663SMark.Logan@Sun.COM  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
219663SMark.Logan@Sun.COM  * GNU General Public License for more details.
229663SMark.Logan@Sun.COM  *
239663SMark.Logan@Sun.COM  * You should have received a copy of the GNU General Public License
249663SMark.Logan@Sun.COM  * along with this program (in the main directory of the Linux-NTFS source
259663SMark.Logan@Sun.COM  * in the file COPYING); if not, write to the Free Software Foundation,
269663SMark.Logan@Sun.COM  * Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
279663SMark.Logan@Sun.COM  */
289663SMark.Logan@Sun.COM 
299663SMark.Logan@Sun.COM /*
309663SMark.Logan@Sun.COM  * WARNING: This program might not work on architectures which do not allow
319663SMark.Logan@Sun.COM  * unaligned access. For those, the program would need to start using
329663SMark.Logan@Sun.COM  * get/put_unaligned macros (#include <asm/unaligned.h>), but not doing it yet,
339663SMark.Logan@Sun.COM  * since NTFS really mostly applies to ia32 only, which does allow unaligned
349663SMark.Logan@Sun.COM  * accesses. We might not actually have a problem though, since the structs are
359663SMark.Logan@Sun.COM  * defined as being packed so that might be enough for gcc to insert the
369663SMark.Logan@Sun.COM  * correct code.
379663SMark.Logan@Sun.COM  *
389663SMark.Logan@Sun.COM  * If anyone using a non-little endian and/or an aligned access only CPU tries
399663SMark.Logan@Sun.COM  * this program please let me know whether it works or not!
409663SMark.Logan@Sun.COM  *
419663SMark.Logan@Sun.COM  *	Anton Altaparmakov <aia21@cantab.net>
429663SMark.Logan@Sun.COM  */
439663SMark.Logan@Sun.COM 
449663SMark.Logan@Sun.COM #include "config.h"
459663SMark.Logan@Sun.COM 
469663SMark.Logan@Sun.COM #ifdef HAVE_UNISTD_H
479663SMark.Logan@Sun.COM #include <unistd.h>
489663SMark.Logan@Sun.COM #endif
499663SMark.Logan@Sun.COM #ifdef HAVE_STDLIB_H
509663SMark.Logan@Sun.COM #include <stdlib.h>
519663SMark.Logan@Sun.COM #endif
529663SMark.Logan@Sun.COM #ifdef HAVE_STDIO_H
539663SMark.Logan@Sun.COM #include <stdio.h>
549663SMark.Logan@Sun.COM #endif
559663SMark.Logan@Sun.COM #ifdef HAVE_FCNTL_H
569663SMark.Logan@Sun.COM #include <fcntl.h>
579663SMark.Logan@Sun.COM #endif
589663SMark.Logan@Sun.COM #ifdef HAVE_ERRNO_H
599663SMark.Logan@Sun.COM #include <errno.h>
609663SMark.Logan@Sun.COM #endif
619663SMark.Logan@Sun.COM #ifdef HAVE_STRING_H
629663SMark.Logan@Sun.COM #include <string.h>
639663SMark.Logan@Sun.COM #endif
649663SMark.Logan@Sun.COM #ifdef HAVE_GETOPT_H
659663SMark.Logan@Sun.COM #include <getopt.h>
669663SMark.Logan@Sun.COM #endif
679663SMark.Logan@Sun.COM 
68*10465SMark.Logan@Sun.COM #include "compat.h"
699663SMark.Logan@Sun.COM #include "types.h"
709663SMark.Logan@Sun.COM #include "attrib.h"
719663SMark.Logan@Sun.COM #include "mft.h"
729663SMark.Logan@Sun.COM #include "device.h"
739663SMark.Logan@Sun.COM #include "logfile.h"
749663SMark.Logan@Sun.COM #include "utils.h"
759663SMark.Logan@Sun.COM #include "version.h"
769663SMark.Logan@Sun.COM #include "logging.h"
779663SMark.Logan@Sun.COM 
789663SMark.Logan@Sun.COM #ifdef NO_NTFS_DEVICE_DEFAULT_IO_OPS
799663SMark.Logan@Sun.COM #	error "No default device io operations!  Cannot build ntfsfix.  \
809663SMark.Logan@Sun.COM You need to run ./configure without the --disable-default-device-io-ops \
819663SMark.Logan@Sun.COM switch if you want to be able to build the NTFS utilities."
829663SMark.Logan@Sun.COM #endif
839663SMark.Logan@Sun.COM 
849663SMark.Logan@Sun.COM static const char *EXEC_NAME = "ntfsfix";
859663SMark.Logan@Sun.COM static const char *OK        = "OK\n";
869663SMark.Logan@Sun.COM static const char *FAILED    = "FAILED\n";
879663SMark.Logan@Sun.COM 
889663SMark.Logan@Sun.COM static struct {
899663SMark.Logan@Sun.COM 	char *volume;
909663SMark.Logan@Sun.COM } opt;
919663SMark.Logan@Sun.COM 
929663SMark.Logan@Sun.COM /**
939663SMark.Logan@Sun.COM  * usage
949663SMark.Logan@Sun.COM  */
959663SMark.Logan@Sun.COM __attribute__((noreturn))
usage(void)969663SMark.Logan@Sun.COM static int usage(void)
979663SMark.Logan@Sun.COM {
989663SMark.Logan@Sun.COM 	ntfs_log_info("%s v%s (libntfs %s)\n"
999663SMark.Logan@Sun.COM 		   "\n"
1009663SMark.Logan@Sun.COM 		   "Usage: %s [options] device\n"
1019663SMark.Logan@Sun.COM 		   "    Attempt to fix an NTFS partition.\n"
1029663SMark.Logan@Sun.COM 		   "\n"
1039663SMark.Logan@Sun.COM 		   "    -h, --help             Display this help\n"
1049663SMark.Logan@Sun.COM 		   "    -V, --version          Display version information\n"
1059663SMark.Logan@Sun.COM 		   "\n"
1069663SMark.Logan@Sun.COM 		   "For example: %s /dev/hda6\n\n",
1079663SMark.Logan@Sun.COM 		   EXEC_NAME, VERSION, ntfs_libntfs_version(), EXEC_NAME,
1089663SMark.Logan@Sun.COM 		   EXEC_NAME);
1099663SMark.Logan@Sun.COM 	ntfs_log_info("%s%s", ntfs_bugs, ntfs_home);
1109663SMark.Logan@Sun.COM 	exit(1);
1119663SMark.Logan@Sun.COM }
1129663SMark.Logan@Sun.COM 
1139663SMark.Logan@Sun.COM /**
1149663SMark.Logan@Sun.COM  * version
1159663SMark.Logan@Sun.COM  */
1169663SMark.Logan@Sun.COM __attribute__((noreturn))
version(void)1179663SMark.Logan@Sun.COM static void version(void)
1189663SMark.Logan@Sun.COM {
1199663SMark.Logan@Sun.COM 	ntfs_log_info("%s v%s\n\n"
1209663SMark.Logan@Sun.COM 		   "Attempt to fix an NTFS partition.\n\n"
1219663SMark.Logan@Sun.COM 		   "Copyright (c) 2000-2006 Anton Altaparmakov\n"
1229663SMark.Logan@Sun.COM 		   "Copyright (c) 2002-2006 Szabolcs Szakacsits\n"
1239663SMark.Logan@Sun.COM 		   "Copyright (c) 2007      Yura Pakhuchiy\n\n",
1249663SMark.Logan@Sun.COM 		   EXEC_NAME, VERSION);
1259663SMark.Logan@Sun.COM 	ntfs_log_info("%s\n%s%s", ntfs_gpl, ntfs_bugs, ntfs_home);
1269663SMark.Logan@Sun.COM 	exit(1);
1279663SMark.Logan@Sun.COM }
1289663SMark.Logan@Sun.COM 
1299663SMark.Logan@Sun.COM /**
1309663SMark.Logan@Sun.COM  * parse_options
1319663SMark.Logan@Sun.COM  */
parse_options(int argc,char ** argv)1329663SMark.Logan@Sun.COM static void parse_options(int argc, char **argv)
1339663SMark.Logan@Sun.COM {
1349663SMark.Logan@Sun.COM 	int c;
1359663SMark.Logan@Sun.COM 	static const char *sopt = "-hV";
1369663SMark.Logan@Sun.COM 	static const struct option lopt[] = {
1379663SMark.Logan@Sun.COM 		{ "help",	no_argument,	NULL, 'h' },
1389663SMark.Logan@Sun.COM 		{ "version",	no_argument,	NULL, 'V' },
1399663SMark.Logan@Sun.COM 		{ NULL, 0, NULL, 0 }
1409663SMark.Logan@Sun.COM 	};
1419663SMark.Logan@Sun.COM 
1429663SMark.Logan@Sun.COM 	memset(&opt, 0, sizeof(opt));
1439663SMark.Logan@Sun.COM 
1449663SMark.Logan@Sun.COM 	while ((c = getopt_long(argc, argv, sopt, lopt, NULL)) != -1) {
1459663SMark.Logan@Sun.COM 		switch (c) {
1469663SMark.Logan@Sun.COM 		case 1:	/* A non-option argument */
1479663SMark.Logan@Sun.COM 			if (!opt.volume)
1489663SMark.Logan@Sun.COM 				opt.volume = argv[optind - 1];
1499663SMark.Logan@Sun.COM 			else {
1509663SMark.Logan@Sun.COM 				ntfs_log_info("ERROR: Too many arguments.\n");
1519663SMark.Logan@Sun.COM 				usage();
1529663SMark.Logan@Sun.COM 			}
1539663SMark.Logan@Sun.COM 			break;
1549663SMark.Logan@Sun.COM 		case 'h':
1559663SMark.Logan@Sun.COM 		case '?':
1569663SMark.Logan@Sun.COM 			usage();
1579663SMark.Logan@Sun.COM 		case 'V':
1589663SMark.Logan@Sun.COM 			version();
1599663SMark.Logan@Sun.COM 		default:
1609663SMark.Logan@Sun.COM 			ntfs_log_info("ERROR: Unknown option '%s'.\n", argv[optind - 1]);
1619663SMark.Logan@Sun.COM 			usage();
1629663SMark.Logan@Sun.COM 		}
1639663SMark.Logan@Sun.COM 	}
1649663SMark.Logan@Sun.COM 
1659663SMark.Logan@Sun.COM 	if (opt.volume == NULL) {
1669663SMark.Logan@Sun.COM 		ntfs_log_info("ERROR: You must specify a device.\n");
1679663SMark.Logan@Sun.COM 		usage();
1689663SMark.Logan@Sun.COM 	}
1699663SMark.Logan@Sun.COM }
1709663SMark.Logan@Sun.COM 
1719663SMark.Logan@Sun.COM /**
1729663SMark.Logan@Sun.COM  * OLD_ntfs_volume_set_flags
1739663SMark.Logan@Sun.COM  */
OLD_ntfs_volume_set_flags(ntfs_volume * vol,const le16 flags)1749663SMark.Logan@Sun.COM static int OLD_ntfs_volume_set_flags(ntfs_volume *vol, const le16 flags)
1759663SMark.Logan@Sun.COM {
1769663SMark.Logan@Sun.COM 	MFT_RECORD *m = NULL;
1779663SMark.Logan@Sun.COM 	ATTR_RECORD *a;
1789663SMark.Logan@Sun.COM 	VOLUME_INFORMATION *c;
1799663SMark.Logan@Sun.COM 	ntfs_attr_search_ctx *ctx;
1809663SMark.Logan@Sun.COM 	int ret = -1;   /* failure */
1819663SMark.Logan@Sun.COM 
1829663SMark.Logan@Sun.COM 	if (!vol) {
1839663SMark.Logan@Sun.COM 		errno = EINVAL;
1849663SMark.Logan@Sun.COM 		return -1;
1859663SMark.Logan@Sun.COM 	}
1869663SMark.Logan@Sun.COM 	if (ntfs_file_record_read(vol, FILE_Volume, &m, NULL)) {
1879663SMark.Logan@Sun.COM 		ntfs_log_perror("Failed to read $Volume");
1889663SMark.Logan@Sun.COM 		return -1;
1899663SMark.Logan@Sun.COM 	}
1909663SMark.Logan@Sun.COM 	/* Sanity check */
1919663SMark.Logan@Sun.COM 	if (!(m->flags & MFT_RECORD_IN_USE)) {
1929663SMark.Logan@Sun.COM 		ntfs_log_error("$Volume has been deleted. Cannot handle this "
1939663SMark.Logan@Sun.COM 				"yet. Run chkdsk to fix this.\n");
1949663SMark.Logan@Sun.COM 		errno = EIO;
1959663SMark.Logan@Sun.COM 		goto err_exit;
1969663SMark.Logan@Sun.COM 	}
1979663SMark.Logan@Sun.COM 	/* Get a pointer to the volume information attribute. */
1989663SMark.Logan@Sun.COM 	ctx = ntfs_attr_get_search_ctx(NULL, m);
1999663SMark.Logan@Sun.COM 	if (!ctx) {
2009663SMark.Logan@Sun.COM 		ntfs_log_debug("Failed to allocate attribute search "
2019663SMark.Logan@Sun.COM 				"context.\n");
2029663SMark.Logan@Sun.COM 		goto err_exit;
2039663SMark.Logan@Sun.COM 	}
2049663SMark.Logan@Sun.COM 	if (ntfs_attr_lookup(AT_VOLUME_INFORMATION, AT_UNNAMED, 0, 0, 0, NULL,
2059663SMark.Logan@Sun.COM 			0, ctx)) {
2069663SMark.Logan@Sun.COM 		ntfs_log_error("Attribute $VOLUME_INFORMATION was not found in "
2079663SMark.Logan@Sun.COM 				"$Volume!\n");
2089663SMark.Logan@Sun.COM 		goto err_out;
2099663SMark.Logan@Sun.COM 	}
2109663SMark.Logan@Sun.COM 	a = ctx->attr;
2119663SMark.Logan@Sun.COM 	/* Sanity check. */
2129663SMark.Logan@Sun.COM 	if (a->non_resident) {
2139663SMark.Logan@Sun.COM 		ntfs_log_error("Attribute $VOLUME_INFORMATION must be resident "
2149663SMark.Logan@Sun.COM 				"(and it isn't)!\n");
2159663SMark.Logan@Sun.COM 		errno = EIO;
2169663SMark.Logan@Sun.COM 		goto err_out;
2179663SMark.Logan@Sun.COM 	}
2189663SMark.Logan@Sun.COM 	/* Get a pointer to the value of the attribute. */
2199663SMark.Logan@Sun.COM 	c = (VOLUME_INFORMATION*)(le16_to_cpu(a->u.res.value_offset) + (char*)a);
2209663SMark.Logan@Sun.COM 	/* Sanity checks. */
2219663SMark.Logan@Sun.COM 	if ((char*)c + le32_to_cpu(a->u.res.value_length) >
2229663SMark.Logan@Sun.COM 			(char*)m + le32_to_cpu(m->bytes_in_use) ||
2239663SMark.Logan@Sun.COM 			le16_to_cpu(a->u.res.value_offset) +
2249663SMark.Logan@Sun.COM 			le32_to_cpu(a->u.res.value_length) > le32_to_cpu(a->length)) {
2259663SMark.Logan@Sun.COM 		ntfs_log_error("Attribute $VOLUME_INFORMATION in $Volume is "
2269663SMark.Logan@Sun.COM 				"corrupt!\n");
2279663SMark.Logan@Sun.COM 		errno = EIO;
2289663SMark.Logan@Sun.COM 		goto err_out;
2299663SMark.Logan@Sun.COM 	}
2309663SMark.Logan@Sun.COM 	/* Set the volume flags. */
2319663SMark.Logan@Sun.COM 	vol->flags = c->flags = flags;
2329663SMark.Logan@Sun.COM 	if (ntfs_mft_record_write(vol, FILE_Volume, m)) {
2339663SMark.Logan@Sun.COM 		ntfs_log_perror("Error writing $Volume");
2349663SMark.Logan@Sun.COM 		goto err_out;
2359663SMark.Logan@Sun.COM 	}
2369663SMark.Logan@Sun.COM 	ret = 0; /* success */
2379663SMark.Logan@Sun.COM err_out:
2389663SMark.Logan@Sun.COM 	ntfs_attr_put_search_ctx(ctx);
2399663SMark.Logan@Sun.COM err_exit:
2409663SMark.Logan@Sun.COM 	free(m);
2419663SMark.Logan@Sun.COM 	return ret;
2429663SMark.Logan@Sun.COM }
2439663SMark.Logan@Sun.COM 
2449663SMark.Logan@Sun.COM /**
2459663SMark.Logan@Sun.COM  * set_dirty_flag
2469663SMark.Logan@Sun.COM  */
set_dirty_flag(ntfs_volume * vol)2479663SMark.Logan@Sun.COM static int set_dirty_flag(ntfs_volume *vol)
2489663SMark.Logan@Sun.COM {
2499663SMark.Logan@Sun.COM 	le16 flags;
2509663SMark.Logan@Sun.COM 
2519663SMark.Logan@Sun.COM 	if (NVolWasDirty(vol))
2529663SMark.Logan@Sun.COM 		return 0;
2539663SMark.Logan@Sun.COM 	ntfs_log_info("Setting required flags on partition... ");
2549663SMark.Logan@Sun.COM 	/*
2559663SMark.Logan@Sun.COM 	 * Set chkdsk flag, i.e. mark the partition dirty so chkdsk will run
2569663SMark.Logan@Sun.COM 	 * and fix it for us.
2579663SMark.Logan@Sun.COM 	 */
2589663SMark.Logan@Sun.COM 	flags = vol->flags | VOLUME_IS_DIRTY;
2599663SMark.Logan@Sun.COM 	if (OLD_ntfs_volume_set_flags(vol, flags)) {
2609663SMark.Logan@Sun.COM 		ntfs_log_info(FAILED);
2619663SMark.Logan@Sun.COM 		ntfs_log_error("Error setting volume flags.\n");
2629663SMark.Logan@Sun.COM 		return -1;
2639663SMark.Logan@Sun.COM 	}
2649663SMark.Logan@Sun.COM 	vol->flags = flags;
2659663SMark.Logan@Sun.COM 	NVolSetWasDirty(vol);
2669663SMark.Logan@Sun.COM 	ntfs_log_info(OK);
2679663SMark.Logan@Sun.COM 	return 0;
2689663SMark.Logan@Sun.COM }
2699663SMark.Logan@Sun.COM 
2709663SMark.Logan@Sun.COM /**
2719663SMark.Logan@Sun.COM  * empty_journal
2729663SMark.Logan@Sun.COM  */
empty_journal(ntfs_volume * vol)2739663SMark.Logan@Sun.COM static int empty_journal(ntfs_volume *vol)
2749663SMark.Logan@Sun.COM {
2759663SMark.Logan@Sun.COM 	if (NVolLogFileEmpty(vol))
2769663SMark.Logan@Sun.COM 		return 0;
2779663SMark.Logan@Sun.COM 	ntfs_log_info("Going to empty the journal ($LogFile)... ");
2789663SMark.Logan@Sun.COM 	if (ntfs_logfile_reset(vol)) {
2799663SMark.Logan@Sun.COM 		ntfs_log_info(FAILED);
2809663SMark.Logan@Sun.COM 		ntfs_log_perror("Failed to reset $LogFile");
2819663SMark.Logan@Sun.COM 		return -1;
2829663SMark.Logan@Sun.COM 	}
2839663SMark.Logan@Sun.COM 	ntfs_log_info(OK);
2849663SMark.Logan@Sun.COM 	return 0;
2859663SMark.Logan@Sun.COM }
2869663SMark.Logan@Sun.COM 
2879663SMark.Logan@Sun.COM /**
2889663SMark.Logan@Sun.COM  * fix_mftmirr
2899663SMark.Logan@Sun.COM  */
fix_mftmirr(ntfs_volume * vol)2909663SMark.Logan@Sun.COM static int fix_mftmirr(ntfs_volume *vol)
2919663SMark.Logan@Sun.COM {
2929663SMark.Logan@Sun.COM 	s64 l, br;
2939663SMark.Logan@Sun.COM 	unsigned char *m, *m2;
2949663SMark.Logan@Sun.COM 	int i, ret = -1; /* failure */
2959663SMark.Logan@Sun.COM 	BOOL done;
2969663SMark.Logan@Sun.COM 
2979663SMark.Logan@Sun.COM 	ntfs_log_info("\nProcessing $MFT and $MFTMirr...\n");
2989663SMark.Logan@Sun.COM 
2999663SMark.Logan@Sun.COM 	/* Load data from $MFT and $MFTMirr and compare the contents. */
3009663SMark.Logan@Sun.COM 	m = (u8*)malloc(vol->mftmirr_size << vol->mft_record_size_bits);
3019663SMark.Logan@Sun.COM 	if (!m) {
3029663SMark.Logan@Sun.COM 		ntfs_log_perror("Failed to allocate memory");
3039663SMark.Logan@Sun.COM 		return -1;
3049663SMark.Logan@Sun.COM 	}
3059663SMark.Logan@Sun.COM 	m2 = (u8*)malloc(vol->mftmirr_size << vol->mft_record_size_bits);
3069663SMark.Logan@Sun.COM 	if (!m2) {
3079663SMark.Logan@Sun.COM 		ntfs_log_perror("Failed to allocate memory");
3089663SMark.Logan@Sun.COM 		free(m);
3099663SMark.Logan@Sun.COM 		return -1;
3109663SMark.Logan@Sun.COM 	}
3119663SMark.Logan@Sun.COM 
3129663SMark.Logan@Sun.COM 	ntfs_log_info("Reading $MFT... ");
3139663SMark.Logan@Sun.COM 	l = ntfs_attr_mst_pread(vol->mft_na, 0, vol->mftmirr_size,
3149663SMark.Logan@Sun.COM 			vol->mft_record_size, m);
3159663SMark.Logan@Sun.COM 	if (l != vol->mftmirr_size) {
3169663SMark.Logan@Sun.COM 		ntfs_log_info(FAILED);
3179663SMark.Logan@Sun.COM 		if (l != -1)
3189663SMark.Logan@Sun.COM 			errno = EIO;
3199663SMark.Logan@Sun.COM 		ntfs_log_perror("Failed to read $MFT");
3209663SMark.Logan@Sun.COM 		goto error_exit;
3219663SMark.Logan@Sun.COM 	}
3229663SMark.Logan@Sun.COM 	ntfs_log_info(OK);
3239663SMark.Logan@Sun.COM 
3249663SMark.Logan@Sun.COM 	ntfs_log_info("Reading $MFTMirr... ");
3259663SMark.Logan@Sun.COM 	l = ntfs_attr_mst_pread(vol->mftmirr_na, 0, vol->mftmirr_size,
3269663SMark.Logan@Sun.COM 			vol->mft_record_size, m2);
3279663SMark.Logan@Sun.COM 	if (l != vol->mftmirr_size) {
3289663SMark.Logan@Sun.COM 		ntfs_log_info(FAILED);
3299663SMark.Logan@Sun.COM 		if (l != -1)
3309663SMark.Logan@Sun.COM 			errno = EIO;
3319663SMark.Logan@Sun.COM 		ntfs_log_perror("Failed to read $MFTMirr");
3329663SMark.Logan@Sun.COM 		goto error_exit;
3339663SMark.Logan@Sun.COM 	}
3349663SMark.Logan@Sun.COM 	ntfs_log_info(OK);
3359663SMark.Logan@Sun.COM 
3369663SMark.Logan@Sun.COM 	/*
3379663SMark.Logan@Sun.COM 	 * FIXME: Need to actually check the $MFTMirr for being real. Otherwise
3389663SMark.Logan@Sun.COM 	 * we might corrupt the partition if someone is experimenting with
3399663SMark.Logan@Sun.COM 	 * software RAID and the $MFTMirr is not actually in the position we
3409663SMark.Logan@Sun.COM 	 * expect it to be... )-:
3419663SMark.Logan@Sun.COM 	 * FIXME: We should emit a warning it $MFTMirr is damaged and ask
3429663SMark.Logan@Sun.COM 	 * user whether to recreate it from $MFT or whether to abort. - The
3439663SMark.Logan@Sun.COM 	 * warning needs to include the danger of software RAID arrays.
3449663SMark.Logan@Sun.COM 	 * Maybe we should go as far as to detect whether we are running on a
3459663SMark.Logan@Sun.COM 	 * MD disk and if yes then bomb out right at the start of the program?
3469663SMark.Logan@Sun.COM 	 */
3479663SMark.Logan@Sun.COM 
3489663SMark.Logan@Sun.COM 	ntfs_log_info("Comparing $MFTMirr to $MFT... ");
3499663SMark.Logan@Sun.COM 	done = FALSE;
3509663SMark.Logan@Sun.COM 	for (i = 0; i < vol->mftmirr_size; ++i) {
3519663SMark.Logan@Sun.COM 		MFT_RECORD *mrec, *mrec2;
3529663SMark.Logan@Sun.COM 		const char *ESTR[12] = { "$MFT", "$MFTMirr", "$LogFile",
3539663SMark.Logan@Sun.COM 			"$Volume", "$AttrDef", "root directory", "$Bitmap",
3549663SMark.Logan@Sun.COM 			"$Boot", "$BadClus", "$Secure", "$UpCase", "$Extend" };
3559663SMark.Logan@Sun.COM 		const char *s;
3569663SMark.Logan@Sun.COM 		BOOL use_mirr;
3579663SMark.Logan@Sun.COM 
3589663SMark.Logan@Sun.COM 		if (i < 12)
3599663SMark.Logan@Sun.COM 			s = ESTR[i];
3609663SMark.Logan@Sun.COM 		else if (i < 16)
3619663SMark.Logan@Sun.COM 			s = "system file";
3629663SMark.Logan@Sun.COM 		else
3639663SMark.Logan@Sun.COM 			s = "mft record";
3649663SMark.Logan@Sun.COM 
3659663SMark.Logan@Sun.COM 		use_mirr = FALSE;
3669663SMark.Logan@Sun.COM 		mrec = (MFT_RECORD*)(m + i * vol->mft_record_size);
3679663SMark.Logan@Sun.COM 		if (mrec->flags & MFT_RECORD_IN_USE) {
3689663SMark.Logan@Sun.COM 			if (ntfs_is_baad_record(mrec->magic)) {
3699663SMark.Logan@Sun.COM 				ntfs_log_info(FAILED);
3709663SMark.Logan@Sun.COM 				ntfs_log_error("$MFT error: Incomplete multi "
3719663SMark.Logan@Sun.COM 						"sector transfer detected in "
3729663SMark.Logan@Sun.COM 						"%s.\nCannot handle this yet. "
3739663SMark.Logan@Sun.COM 						")-:\n", s);
3749663SMark.Logan@Sun.COM 				goto error_exit;
3759663SMark.Logan@Sun.COM 			}
3769663SMark.Logan@Sun.COM 			if (!ntfs_is_mft_record(mrec->magic)) {
3779663SMark.Logan@Sun.COM 				ntfs_log_info(FAILED);
3789663SMark.Logan@Sun.COM 				ntfs_log_error("$MFT error: Invalid mft "
3799663SMark.Logan@Sun.COM 						"record for %s.\nCannot "
3809663SMark.Logan@Sun.COM 						"handle this yet. )-:\n", s);
3819663SMark.Logan@Sun.COM 				goto error_exit;
3829663SMark.Logan@Sun.COM 			}
3839663SMark.Logan@Sun.COM 		}
3849663SMark.Logan@Sun.COM 		mrec2 = (MFT_RECORD*)(m2 + i * vol->mft_record_size);
3859663SMark.Logan@Sun.COM 		if (mrec2->flags & MFT_RECORD_IN_USE) {
3869663SMark.Logan@Sun.COM 			if (ntfs_is_baad_record(mrec2->magic)) {
3879663SMark.Logan@Sun.COM 				ntfs_log_info(FAILED);
3889663SMark.Logan@Sun.COM 				ntfs_log_error("$MFTMirr error: Incomplete "
3899663SMark.Logan@Sun.COM 						"multi sector transfer "
3909663SMark.Logan@Sun.COM 						"detected in %s.\n", s);
3919663SMark.Logan@Sun.COM 				goto error_exit;
3929663SMark.Logan@Sun.COM 			}
3939663SMark.Logan@Sun.COM 			if (!ntfs_is_mft_record(mrec2->magic)) {
3949663SMark.Logan@Sun.COM 				ntfs_log_info(FAILED);
3959663SMark.Logan@Sun.COM 				ntfs_log_error("$MFTMirr error: Invalid mft "
3969663SMark.Logan@Sun.COM 						"record for %s.\n", s);
3979663SMark.Logan@Sun.COM 				goto error_exit;
3989663SMark.Logan@Sun.COM 			}
3999663SMark.Logan@Sun.COM 			/* $MFT is corrupt but $MFTMirr is ok, use $MFTMirr. */
4009663SMark.Logan@Sun.COM 			if (!(mrec->flags & MFT_RECORD_IN_USE) &&
4019663SMark.Logan@Sun.COM 					!ntfs_is_mft_record(mrec->magic))
4029663SMark.Logan@Sun.COM 				use_mirr = TRUE;
4039663SMark.Logan@Sun.COM 		}
4049663SMark.Logan@Sun.COM 		if (memcmp(mrec, mrec2, ntfs_mft_record_get_data_size(mrec))) {
4059663SMark.Logan@Sun.COM 			if (!done) {
4069663SMark.Logan@Sun.COM 				done = TRUE;
4079663SMark.Logan@Sun.COM 				ntfs_log_info(FAILED);
4089663SMark.Logan@Sun.COM 			}
4099663SMark.Logan@Sun.COM 			ntfs_log_info("Correcting differences in $MFT%s "
4109663SMark.Logan@Sun.COM 					"record %d...", use_mirr ? "" : "Mirr",
4119663SMark.Logan@Sun.COM 					i);
4129663SMark.Logan@Sun.COM 			br = ntfs_mft_record_write(vol, i,
4139663SMark.Logan@Sun.COM 					use_mirr ? mrec2 : mrec);
4149663SMark.Logan@Sun.COM 			if (br) {
4159663SMark.Logan@Sun.COM 				ntfs_log_info(FAILED);
4169663SMark.Logan@Sun.COM 				ntfs_log_perror("Error correcting $MFT%s",
4179663SMark.Logan@Sun.COM 						use_mirr ? "" : "Mirr");
4189663SMark.Logan@Sun.COM 				goto error_exit;
4199663SMark.Logan@Sun.COM 			}
4209663SMark.Logan@Sun.COM 			ntfs_log_info(OK);
4219663SMark.Logan@Sun.COM 		}
4229663SMark.Logan@Sun.COM 	}
4239663SMark.Logan@Sun.COM 	if (!done)
4249663SMark.Logan@Sun.COM 		ntfs_log_info(OK);
4259663SMark.Logan@Sun.COM 	ntfs_log_info("Processing of $MFT and $MFTMirr completed "
4269663SMark.Logan@Sun.COM 			"successfully.\n");
4279663SMark.Logan@Sun.COM 	ret = 0;
4289663SMark.Logan@Sun.COM error_exit:
4299663SMark.Logan@Sun.COM 	free(m);
4309663SMark.Logan@Sun.COM 	free(m2);
4319663SMark.Logan@Sun.COM 	return ret;
4329663SMark.Logan@Sun.COM }
4339663SMark.Logan@Sun.COM 
4349663SMark.Logan@Sun.COM /**
4359663SMark.Logan@Sun.COM  * fix_mount
4369663SMark.Logan@Sun.COM  */
fix_mount(void)4379663SMark.Logan@Sun.COM static int fix_mount(void)
4389663SMark.Logan@Sun.COM {
4399663SMark.Logan@Sun.COM 	int ret = -1; /* failure */
4409663SMark.Logan@Sun.COM 	ntfs_volume *vol;
4419663SMark.Logan@Sun.COM 	struct ntfs_device *dev;
4429663SMark.Logan@Sun.COM 
4439663SMark.Logan@Sun.COM 	ntfs_log_info("Attempting to correct errors... ");
4449663SMark.Logan@Sun.COM 
4459663SMark.Logan@Sun.COM 	dev = ntfs_device_alloc(opt.volume, 0, &ntfs_device_default_io_ops,
4469663SMark.Logan@Sun.COM 			NULL);
4479663SMark.Logan@Sun.COM 	if (!dev) {
4489663SMark.Logan@Sun.COM 		ntfs_log_info(FAILED);
4499663SMark.Logan@Sun.COM 		ntfs_log_perror("Failed to allocate device");
4509663SMark.Logan@Sun.COM 		return -1;
4519663SMark.Logan@Sun.COM 	}
4529663SMark.Logan@Sun.COM 	vol = ntfs_volume_startup(dev, 0);
4539663SMark.Logan@Sun.COM 	if (!vol) {
4549663SMark.Logan@Sun.COM 		ntfs_log_info(FAILED);
4559663SMark.Logan@Sun.COM 		ntfs_log_perror("Failed to startup volume");
4569663SMark.Logan@Sun.COM 		ntfs_log_error("Volume is corrupt. You should run chkdsk.\n");
4579663SMark.Logan@Sun.COM 		ntfs_device_free(dev);
4589663SMark.Logan@Sun.COM 		return -1;
4599663SMark.Logan@Sun.COM 	}
4609663SMark.Logan@Sun.COM 	if (fix_mftmirr(vol) < 0)
4619663SMark.Logan@Sun.COM 		goto error_exit;
4629663SMark.Logan@Sun.COM 	if (set_dirty_flag(vol) < 0)
4639663SMark.Logan@Sun.COM 		goto error_exit;
4649663SMark.Logan@Sun.COM 	if (empty_journal(vol) < 0)
4659663SMark.Logan@Sun.COM 		goto error_exit;
4669663SMark.Logan@Sun.COM 	ret = 0;
4679663SMark.Logan@Sun.COM error_exit:
4689663SMark.Logan@Sun.COM 	/* ntfs_umount() will invoke ntfs_device_free() for us. */
4699663SMark.Logan@Sun.COM 	if (ntfs_umount(vol, 0))
4709663SMark.Logan@Sun.COM 		ntfs_umount(vol, 1);
4719663SMark.Logan@Sun.COM 	return ret;
4729663SMark.Logan@Sun.COM }
4739663SMark.Logan@Sun.COM 
4749663SMark.Logan@Sun.COM /**
4759663SMark.Logan@Sun.COM  * main
4769663SMark.Logan@Sun.COM  */
main(int argc,char ** argv)4779663SMark.Logan@Sun.COM int main(int argc, char **argv)
4789663SMark.Logan@Sun.COM {
4799663SMark.Logan@Sun.COM 	ntfs_volume *vol;
4809663SMark.Logan@Sun.COM 	unsigned long mnt_flags;
4819663SMark.Logan@Sun.COM 	int ret = 1; /* failure */
4829663SMark.Logan@Sun.COM 	BOOL force = FALSE;
4839663SMark.Logan@Sun.COM 
4849663SMark.Logan@Sun.COM 	ntfs_log_set_handler(ntfs_log_handler_outerr);
4859663SMark.Logan@Sun.COM 
4869663SMark.Logan@Sun.COM 	parse_options(argc, argv);
4879663SMark.Logan@Sun.COM 
4889663SMark.Logan@Sun.COM 	if (!ntfs_check_if_mounted(opt.volume, &mnt_flags)) {
4899663SMark.Logan@Sun.COM 		if ((mnt_flags & NTFS_MF_MOUNTED) &&
4909663SMark.Logan@Sun.COM 				!(mnt_flags & NTFS_MF_READONLY) && !force) {
4919663SMark.Logan@Sun.COM 			ntfs_log_error("Refusing to operate on read-write "
4929663SMark.Logan@Sun.COM 					"mounted device %s.\n", opt.volume);
4939663SMark.Logan@Sun.COM 			exit(1);
4949663SMark.Logan@Sun.COM 		}
4959663SMark.Logan@Sun.COM 	} else
4969663SMark.Logan@Sun.COM 		ntfs_log_perror("Failed to determine whether %s is mounted",
4979663SMark.Logan@Sun.COM 				opt.volume);
4989663SMark.Logan@Sun.COM 	/* Attempt a full mount first. */
4999663SMark.Logan@Sun.COM 	ntfs_log_info("Mounting volume... ");
5009663SMark.Logan@Sun.COM 	vol = ntfs_mount(opt.volume, 0);
5019663SMark.Logan@Sun.COM 	if (vol) {
5029663SMark.Logan@Sun.COM 		ntfs_log_info(OK);
5039663SMark.Logan@Sun.COM 		ntfs_log_info("Processing of $MFT and $MFTMirr completed "
5049663SMark.Logan@Sun.COM 				"successfully.\n");
5059663SMark.Logan@Sun.COM 	} else {
5069663SMark.Logan@Sun.COM 		ntfs_log_info(FAILED);
5079663SMark.Logan@Sun.COM 		exit(1); /* XXX remove before use */
5089663SMark.Logan@Sun.COM 		if (fix_mount() < 0)
5099663SMark.Logan@Sun.COM 			exit(1);
5109663SMark.Logan@Sun.COM 		vol = ntfs_mount(opt.volume, 0);
5119663SMark.Logan@Sun.COM 		if (!vol) {
5129663SMark.Logan@Sun.COM 			ntfs_log_perror("Remount failed");
5139663SMark.Logan@Sun.COM 			exit(1);
5149663SMark.Logan@Sun.COM 		}
5159663SMark.Logan@Sun.COM 	}
5169663SMark.Logan@Sun.COM 	/* So the unmount does not clear it again. */
5179663SMark.Logan@Sun.COM 	NVolSetWasDirty(vol);
5189663SMark.Logan@Sun.COM 	/* Check NTFS version is ok for us (in $Volume) */
5199663SMark.Logan@Sun.COM 	ntfs_log_info("NTFS volume version is %i.%i.\n", vol->major_ver,
5209663SMark.Logan@Sun.COM 			vol->minor_ver);
5219663SMark.Logan@Sun.COM 	if (ntfs_version_is_supported(vol)) {
5229663SMark.Logan@Sun.COM 		ntfs_log_error("Error: Unknown NTFS version.\n");
5239663SMark.Logan@Sun.COM 		goto error_exit;
5249663SMark.Logan@Sun.COM 	}
5259663SMark.Logan@Sun.COM 	if (vol->major_ver >= 3) {
5269663SMark.Logan@Sun.COM 		/*
5279663SMark.Logan@Sun.COM 		 * FIXME: If on NTFS 3.0+, check for presence of the usn
5289663SMark.Logan@Sun.COM 		 * journal and stamp it if present.
5299663SMark.Logan@Sun.COM 		 */
5309663SMark.Logan@Sun.COM 	}
5319663SMark.Logan@Sun.COM 	/* FIXME: We should be marking the quota out of date, too. */
5329663SMark.Logan@Sun.COM 	/* That's all for now! */
5339663SMark.Logan@Sun.COM 	ntfs_log_info("NTFS partition %s was processed successfully.\n",
5349663SMark.Logan@Sun.COM 			vol->u.dev->d_name);
5359663SMark.Logan@Sun.COM 	/* Set return code to 0. */
5369663SMark.Logan@Sun.COM 	ret = 0;
5379663SMark.Logan@Sun.COM error_exit:
5389663SMark.Logan@Sun.COM 	if (ntfs_umount(vol, 0))
5399663SMark.Logan@Sun.COM 		ntfs_umount(vol, 1);
5409663SMark.Logan@Sun.COM 	return ret;
5419663SMark.Logan@Sun.COM }
5429663SMark.Logan@Sun.COM 
543