path: root/src/ntfs-3g.probe.c (plain)
blob: cb73aee416b70a47ee65a42f2c0ac857c8c4fe87

1	/**
2	* ntfs-3g.probe - Probe NTFS volume mountability
3	*
4	* Copyright (c) 2007-2009 Szabolcs Szakacsits
5	*
6	* This program is free software; you can redistribute it and/or modify
7	* it under the terms of the GNU General Public License as published by
8	* the Free Software Foundation; either version 2 of the License, or
9	* (at your option) any later version.
10	*
11	* This program is distributed in the hope that it will be useful,
12	* but WITHOUT ANY WARRANTY; without even the implied warranty of
13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14	* GNU General Public License for more details.
15	*
16	* You should have received a copy of the GNU General Public License
17	* along with this program (in the main directory of the NTFS-3G
18	* distribution in the file COPYING); if not, write to the Free Software
19	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20	*/
21
22	#include "config.h"
23
24	#ifdef HAVE_STDLIB_H
25	#include <stdlib.h>
26	#endif
27	#ifdef HAVE_STRING_H
28	#include <string.h>
29	#endif
30	#ifdef HAVE_ERRNO_H
31	#include <errno.h>
32	#endif
33	#include <getopt.h>
34
35	#include "compat.h"
36	#include "volume.h"
37	#include "misc.h"
38
39	typedef enum {
40	PROBE_UNSET,
41	PROBE_READONLY,
42	PROBE_READWRITE
43	} probe_t;
44
45	static struct options {
46	probe_t probetype;
47	char *device;
48	} opts;
49
50	static const char *EXEC_NAME = "ntfs-3g.probe";
51
52	static const char *usage_msg =
53	"\n"
54	"%s %s - Probe NTFS volume mountability\n"
55	"\n"
56	"Copyright (C) 2007 Szabolcs Szakacsits\n"
57	"\n"
58	"Usage: %s <--readonly|--readwrite> <device|image_file>\n"
59	"\n"
60	"Example: ntfs-3g.probe --readwrite /dev/sda1\n"
61	"\n"
62	"%s";
63
64	static int ntfs_open(const char *device)
65	{
66	ntfs_volume *vol;
67	unsigned long flags = 0;
68	int ret = NTFS_VOLUME_OK;
69
70	if (opts.probetype == PROBE_READONLY)
71	flags |= NTFS_MNT_RDONLY;
72
73	vol = ntfs_mount(device, flags);
74	if (!vol)
75	ret = ntfs_volume_error(errno);
76
77	if (ret == 0 && ntfs_umount(vol, FALSE) == -1)
78	ret = ntfs_volume_error(errno);
79
80	return ret;
81	}
82
83	static void usage(void)
84	{
85	ntfs_log_info(usage_msg, EXEC_NAME, VERSION, EXEC_NAME, ntfs_home);
86	}
87
88	static int parse_options(int argc, char *argv[])
89	{
90	int c;
91
92	static const char *sopt = "-hrw";
93	static const struct option lopt[] = {
94	{ "readonly", no_argument, NULL, 'r' },
95	{ "readwrite", no_argument, NULL, 'w' },
96	{ "help", no_argument, NULL, 'h' },
97	{ NULL, 0, NULL, 0 }
98	};
99
100	opterr = 0; /* We handle errors. */
101	opts.probetype = PROBE_UNSET;
102
103	while ((c = getopt_long(argc, argv, sopt, lopt, NULL)) != -1) {
104	switch (c) {
105	case 1: /* A non-option argument */
106	if (!opts.device) {
107	opts.device = ntfs_malloc(PATH_MAX + 1);
108	if (!opts.device)
109	return -1;
110
111	strncpy(opts.device, optarg, PATH_MAX);
112	opts.device[PATH_MAX] = 0;
113	} else {
114	ntfs_log_error("%s: You must specify exactly "
115	"one device\n", EXEC_NAME);
116	return -1;
117	}
118	break;
119	case 'h':
120	usage();
121	exit(0);
122	case 'r':
123	opts.probetype = PROBE_READONLY;
124	break;
125	case 'w':
126	opts.probetype = PROBE_READWRITE;
127	break;
128	default:
129	ntfs_log_error("%s: Unknown option '%s'.\n", EXEC_NAME,
130	argv[optind - 1]);
131	return -1;
132	}
133	}
134
135	if (!opts.device) {
136	ntfs_log_error("ERROR: %s: Device is missing\n", EXEC_NAME);
137	return -1;
138	}
139
140	if (opts.probetype == PROBE_UNSET) {
141	ntfs_log_error("ERROR: %s: Probe type is missing\n", EXEC_NAME);
142	return -1;
143	}
144
145	return 0;
146	}
147
148	int main(int argc, char *argv[])
149	{
150	int err;
151
152	ntfs_log_set_handler(ntfs_log_handler_stderr);
153
154	if (parse_options(argc, argv)) {
155	usage();
156	exit(NTFS_VOLUME_SYNTAX_ERROR);
157	}
158
159	err = ntfs_open(opts.device);
160
161	free(opts.device);
162	if (err)
163	exit(err);
164	return (0);
165	}
166
167