(Logical change 1.558)
cantab.net!aia21
parent
5d703b9636
commit
9c86b21e87
|
|
@ -0,0 +1,392 @@
|
|||
/**
|
||||
* ntfsmftalloc - Part of the Linux-NTFS project.
|
||||
*
|
||||
* Copyright (c) 2002-2003 Anton Altaparmakov
|
||||
*
|
||||
* This utility will allocate and initialize an mft record.
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation; either version 2 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program (in the main directory of the Linux-NTFS source
|
||||
* in the file COPYING); if not, write to the Free Software Foundation,
|
||||
* Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
*/
|
||||
|
||||
#include "config.h"
|
||||
|
||||
#ifdef HAVE_UNISTD_H
|
||||
# include <unistd.h>
|
||||
#endif
|
||||
#ifdef HAVE_STDLIB_H
|
||||
# include <stdlib.h>
|
||||
#endif
|
||||
#ifdef HAVE_STDIO_H
|
||||
# include <stdio.h>
|
||||
#endif
|
||||
#ifdef HAVE_STDARG_H
|
||||
# include <stdarg.h>
|
||||
#endif
|
||||
#ifdef HAVE_STRING_H
|
||||
# include <string.h>
|
||||
#endif
|
||||
#ifdef HAVE_ERRNO_H
|
||||
# include <errno.h>
|
||||
#endif
|
||||
#include <time.h>
|
||||
#ifdef HAVE_GETOPT_H
|
||||
# include <getopt.h>
|
||||
#else
|
||||
extern int optind;
|
||||
#endif
|
||||
#include <limits.h>
|
||||
#ifndef LLONG_MAX
|
||||
# define LLONG_MAX 9223372036854775807LL
|
||||
#endif
|
||||
|
||||
#include "types.h"
|
||||
#include "attrib.h"
|
||||
#include "inode.h"
|
||||
#include "layout.h"
|
||||
#include "volume.h"
|
||||
#include "mft.h"
|
||||
#include "utils.h"
|
||||
|
||||
static const char *EXEC_NAME = "ntfsmftalloc";
|
||||
|
||||
/* Need these global so ntfsmftalloc_exit can access them. */
|
||||
static BOOL success = FALSE;
|
||||
|
||||
static char *dev_name;
|
||||
|
||||
static ntfs_volume *vol;
|
||||
static ntfs_inode *ni = NULL;
|
||||
static ntfs_inode *base_ni = NULL;
|
||||
static s64 base_mft_no = -1;
|
||||
|
||||
static struct {
|
||||
/* -h, print usage and exit. */
|
||||
int no_action; /* -n, do not write to device, only display
|
||||
what would be done. */
|
||||
int quiet; /* -q, quiet execution. */
|
||||
int verbose; /* -v, verbose execution, given twice, really
|
||||
verbose execution (debug mode). */
|
||||
int force; /* -f, force allocation. */
|
||||
/* -V, print version and exit. */
|
||||
} opts;
|
||||
|
||||
/**
|
||||
* Dprintf - debugging output (-vv); overriden by quiet (-q)
|
||||
*/
|
||||
static void Dprintf(const char *fmt, ...) __attribute__((format(printf, 1, 2)));
|
||||
static void Dprintf(const char *fmt, ...)
|
||||
{
|
||||
va_list ap;
|
||||
|
||||
if (!opts.quiet && opts.verbose > 1) {
|
||||
printf("DEBUG: ");
|
||||
va_start(ap, fmt);
|
||||
vprintf(fmt, ap);
|
||||
va_end(ap);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Eprintf - error output; ignores quiet (-q)
|
||||
*/
|
||||
int Eprintf(const char *fmt, ...)
|
||||
{
|
||||
va_list ap;
|
||||
|
||||
fprintf(stderr, "ERROR: ");
|
||||
va_start(ap, fmt);
|
||||
vfprintf(stderr, fmt, ap);
|
||||
va_end(ap);
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* Generate code for Vprintf() function: Verbose output (-v). */
|
||||
GEN_PRINTF(Vprintf, stdout, &opts.verbose, TRUE)
|
||||
|
||||
/* Generate code for Qprintf() function: Quietable output (if not -q). */
|
||||
GEN_PRINTF(Qprintf, stdout, &opts.quiet, FALSE)
|
||||
|
||||
/**
|
||||
* err_exit - error output and terminate; ignores quiet (-q)
|
||||
*/
|
||||
static void err_exit(const char *fmt, ...) __attribute__((noreturn))
|
||||
__attribute__((format(printf, 1, 2)));
|
||||
static void err_exit(const char *fmt, ...)
|
||||
{
|
||||
va_list ap;
|
||||
|
||||
fprintf(stderr, "ERROR: ");
|
||||
va_start(ap, fmt);
|
||||
vfprintf(stderr, fmt, ap);
|
||||
va_end(ap);
|
||||
fprintf(stderr, "Aborting...\n");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
/**
|
||||
* copyright - print copyright statements
|
||||
*/
|
||||
static void copyright(void)
|
||||
{
|
||||
fprintf(stderr, "Copyright (c) 2004 Anton Altaparmakov\n"
|
||||
"Allocate and initialize a base or an extent mft "
|
||||
"record.\n"
|
||||
"If a base mft record is not specified, a base record "
|
||||
"is allocated and initialized.\n"
|
||||
"If a base mft record is specified, an extent mft "
|
||||
"record is allocated and\n"
|
||||
"initialized to point to the specified base mft "
|
||||
"record.\n");
|
||||
}
|
||||
|
||||
/**
|
||||
* license - print licese statement
|
||||
*/
|
||||
static void license(void)
|
||||
{
|
||||
fprintf(stderr, "%s", ntfs_gpl);
|
||||
}
|
||||
|
||||
/**
|
||||
* usage - print a list of the parameters to the program
|
||||
*/
|
||||
void usage(void) __attribute__ ((noreturn));
|
||||
void usage (void)
|
||||
{
|
||||
copyright();
|
||||
fprintf(stderr, "Usage: %s [options] device [base-mft-record]\n"
|
||||
" -n Do not write to disk\n"
|
||||
" -f Force execution despite errors\n"
|
||||
" -q Quiet execution\n"
|
||||
" -v Verbose execution\n"
|
||||
" -vv Very verbose execution\n"
|
||||
" -V Display version information\n"
|
||||
" -l Display licensing information\n"
|
||||
" -h Display this help\n", EXEC_NAME);
|
||||
fprintf(stderr, "%s%s", ntfs_bugs, ntfs_home);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
/**
|
||||
* parse_options
|
||||
*/
|
||||
static void parse_options(int argc, char *argv[])
|
||||
{
|
||||
long long ll;
|
||||
char *s;
|
||||
int c;
|
||||
|
||||
if (argc && *argv)
|
||||
EXEC_NAME = *argv;
|
||||
fprintf(stderr, "%s v%s\n", EXEC_NAME, VERSION);
|
||||
while ((c = getopt(argc, argv, "fh?nqvVl")) != EOF)
|
||||
switch (c) {
|
||||
case 'f':
|
||||
opts.force = 1;
|
||||
break;
|
||||
case 'n':
|
||||
opts.no_action = 1;
|
||||
break;
|
||||
case 'q':
|
||||
opts.quiet = 1;
|
||||
break;
|
||||
case 'v':
|
||||
opts.verbose++;
|
||||
break;
|
||||
case 'V':
|
||||
/* Version number already printed, so just exit. */
|
||||
exit(0);
|
||||
case 'l':
|
||||
copyright();
|
||||
license();
|
||||
exit(0);
|
||||
case 'h':
|
||||
case '?':
|
||||
default:
|
||||
usage();
|
||||
}
|
||||
if (optind == argc)
|
||||
usage();
|
||||
/* Get the device. */
|
||||
dev_name = argv[optind++];
|
||||
Dprintf("device name = %s\n", dev_name);
|
||||
if (optind != argc) {
|
||||
/* Get the base mft record number. */
|
||||
ll = strtoll(argv[optind++], &s, 0);
|
||||
if (*s || !ll || (ll >= LLONG_MAX && errno == ERANGE))
|
||||
err_exit("Invalid base mft record number: %s\n",
|
||||
argv[optind - 1]);
|
||||
base_mft_no = ll;
|
||||
Dprintf("base mft record number = 0x%llx\n", (long long)ll);
|
||||
}
|
||||
if (optind != argc)
|
||||
usage();
|
||||
}
|
||||
|
||||
/**
|
||||
* dump_mft_record
|
||||
*/
|
||||
static void dump_mft_record(MFT_RECORD *m)
|
||||
{
|
||||
ATTR_RECORD *a;
|
||||
unsigned int u;
|
||||
MFT_REF r;
|
||||
|
||||
printf("-- Beginning dump of mft record. --\n");
|
||||
u = le32_to_cpu(m->magic);
|
||||
printf("Mft record signature (magic) = %c%c%c%c\n", u & 0xff,
|
||||
u >> 8 & 0xff, u >> 16 & 0xff, u >> 24 & 0xff);
|
||||
u = le16_to_cpu(m->usa_ofs);
|
||||
printf("Update sequence array offset = %u (0x%x)\n", u, u);
|
||||
printf("Update sequence array size = %u\n", le16_to_cpu(m->usa_count));
|
||||
printf("$LogFile sequence number (lsn) = %llu\n",
|
||||
(unsigned long long)le64_to_cpu(m->lsn));
|
||||
printf("Sequence number = %u\n", le16_to_cpu(m->sequence_number));
|
||||
printf("Reference (hard link) count = %u\n",
|
||||
le16_to_cpu(m->link_count));
|
||||
u = le16_to_cpu(m->attrs_offset);
|
||||
printf("First attribute offset = %u (0x%x)\n", u, u);
|
||||
printf("Flags = %u: ", le16_to_cpu(m->flags));
|
||||
if (m->flags & MFT_RECORD_IN_USE)
|
||||
printf("MFT_RECORD_IN_USE");
|
||||
else
|
||||
printf("MFT_RECORD_NOT_IN_USE");
|
||||
if (m->flags & MFT_RECORD_IS_DIRECTORY)
|
||||
printf(" | MFT_RECORD_IS_DIRECTORY");
|
||||
printf("\n");
|
||||
u = le32_to_cpu(m->bytes_in_use);
|
||||
printf("Bytes in use = %u (0x%x)\n", u, u);
|
||||
u = le32_to_cpu(m->bytes_allocated);
|
||||
printf("Bytes allocated = %u (0x%x)\n", u, u);
|
||||
r = le64_to_cpu(m->base_mft_record);
|
||||
printf("Base mft record reference:\n\tMft record number = %llu\n\t"
|
||||
"Sequence number = %u\n",
|
||||
(unsigned long long)MREF(r), MSEQNO(r));
|
||||
printf("Next attribute instance = %u\n",
|
||||
le16_to_cpu(m->next_attr_instance));
|
||||
a = (ATTR_RECORD*)((char*)m + le16_to_cpu(m->attrs_offset));
|
||||
printf("-- Beginning dump of attributes within mft record. --\n");
|
||||
while ((char*)a < (char*)m + le32_to_cpu(m->bytes_in_use)) {
|
||||
if (a->type == AT_END)
|
||||
break;
|
||||
a = (ATTR_RECORD*)((char*)a + le32_to_cpu(a->length));
|
||||
};
|
||||
printf("-- End of attributes. --\n");
|
||||
}
|
||||
|
||||
/**
|
||||
* ntfsmftalloc_exit
|
||||
*/
|
||||
static void ntfsmftalloc_exit(void)
|
||||
{
|
||||
if (success)
|
||||
return;
|
||||
/* If there is a base inode, close that instead of the extent inode. */
|
||||
if (base_ni)
|
||||
ni = base_ni;
|
||||
/* Close the inode. */
|
||||
if (ni && ntfs_inode_close(ni)) {
|
||||
fprintf(stderr, "Warning: Failed to close inode 0x%llx: %s\n",
|
||||
(long long)ni->mft_no, strerror(errno));
|
||||
}
|
||||
/* Unmount the volume. */
|
||||
if (ntfs_umount(vol, 0) == -1)
|
||||
fprintf(stderr, "Warning: Could not umount %s: %s\n", dev_name,
|
||||
strerror(errno));
|
||||
}
|
||||
|
||||
/**
|
||||
* main
|
||||
*/
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
unsigned long mnt_flags, ul;
|
||||
int err;
|
||||
|
||||
/* Initialize opts to zero / required values. */
|
||||
memset(&opts, 0, sizeof(opts));
|
||||
/* Parse command line options. */
|
||||
parse_options(argc, argv);
|
||||
utils_set_locale();
|
||||
/* Make sure the file system is not mounted. */
|
||||
if (ntfs_check_if_mounted(dev_name, &mnt_flags))
|
||||
Eprintf("Failed to determine whether %s is mounted: %s\n",
|
||||
dev_name, strerror(errno));
|
||||
else if (mnt_flags & NTFS_MF_MOUNTED) {
|
||||
Eprintf("%s is mounted.\n", dev_name);
|
||||
if (!opts.force)
|
||||
err_exit("Refusing to run!\n");
|
||||
fprintf(stderr, "ntfsmftalloc forced anyway. Hope /etc/mtab "
|
||||
"is incorrect.\n");
|
||||
}
|
||||
/* Mount the device. */
|
||||
if (opts.no_action) {
|
||||
Qprintf("Running in READ-ONLY mode!\n");
|
||||
ul = MS_RDONLY;
|
||||
} else
|
||||
ul = 0;
|
||||
vol = ntfs_mount(dev_name, ul);
|
||||
if (!vol)
|
||||
err_exit("Failed to mount %s: %s\n", dev_name, strerror(errno));
|
||||
/* Register our exit function which will unlock and close the device. */
|
||||
err = atexit(&ntfsmftalloc_exit);
|
||||
if (err == -1) {
|
||||
Eprintf("Could not set up exit() function because atexit() "
|
||||
"failed: %s Aborting...\n", strerror(errno));
|
||||
ntfsmftalloc_exit();
|
||||
exit(1);
|
||||
}
|
||||
if (base_mft_no != -1) {
|
||||
base_ni = ntfs_inode_open(vol, base_mft_no);
|
||||
if (!base_ni)
|
||||
err_exit("Failed to open base inode 0x%llx: %s\n",
|
||||
(long long)base_mft_no,
|
||||
strerror(errno));
|
||||
}
|
||||
/* Open the specified inode. */
|
||||
ni = ntfs_mft_record_alloc(vol, base_ni);
|
||||
if (!ni)
|
||||
err_exit("Failed to allocate mft record: %s\n",
|
||||
strerror(errno));
|
||||
printf("Allocated %s mft record 0x%llx", base_ni ? "extent" : "base",
|
||||
(long long)ni->mft_no);
|
||||
if (base_ni)
|
||||
printf(" with base mft record 0x%llx",
|
||||
(long long)base_mft_no);
|
||||
printf(".\n");
|
||||
if (!opts.quiet && opts.verbose > 1) {
|
||||
Dprintf("Dumping allocated mft record 0x%llx:\n",
|
||||
(long long)ni->mft_no);
|
||||
dump_mft_record(ni->mrec);
|
||||
}
|
||||
/* Close the (base) inode. */
|
||||
if (base_ni)
|
||||
ni = base_ni;
|
||||
err = ntfs_inode_close(ni);
|
||||
if (err)
|
||||
err_exit("Failed to close inode 0x%llx: %s\n",
|
||||
(long long)ni->mft_no, strerror(errno));
|
||||
/* Unmount the volume. */
|
||||
err = ntfs_umount(vol, 0);
|
||||
/* Disable our ntfsmftalloc_exit() handler. */
|
||||
success = TRUE;
|
||||
if (err == -1)
|
||||
fprintf(stderr, "Warning: Failed to umount %s: %s\n", dev_name,
|
||||
strerror(errno));
|
||||
else
|
||||
Qprintf("ntfsmftalloc completed successfully.\n");
|
||||
return 0;
|
||||
}
|
||||
Loading…
Reference in New Issue