1/*	$NetBSD$	*/
2
3/*-
4 * Copyright (c) 2011 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by UCHIYAMA Yasushi.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 *    notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 *    notice, this list of conditions and the following disclaimer in the
17 *    documentation and/or other materials provided with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31
32#include <sys/cdefs.h>
33#ifndef lint
34__RCSID("$NetBSD$");
35#endif /* not lint */
36
37#include <stdio.h>
38#include <string.h>
39#include <errno.h>
40#include <err.h>
41#include <time.h>
42
43#include "v7fs.h"
44#include "v7fs_impl.h"
45#include "v7fs_superblock.h"
46#include "v7fs_inode.h"
47#include "v7fs_file.h"
48
49#include "fsck_v7fs.h"
50#include "progress.h"
51
52static int check_filesystem(struct v7fs_self *, int);
53static int make_lost_and_found(struct v7fs_self *, struct v7fs_inode *);
54
55struct v7fs_inode lost_and_found;
56
57int
58v7fs_fsck(const struct v7fs_mount_device *mount, int flags)
59{
60	struct v7fs_self *fs;
61	int error;
62
63	if ((error = v7fs_io_init(&fs, mount, V7FS_BSIZE))) {
64		pfatal("I/O setup failed");
65		return FSCK_EXIT_CHECK_FAILED;
66	}
67
68	if ((error = v7fs_superblock_load(fs))) {
69		if ((error != EINVAL)) {
70			pfatal("Can't load superblock");
71			return FSCK_EXIT_CHECK_FAILED;
72		}
73		pwarn("Inavalid superblock");
74		if (!reply_trivial("CONTINUE?")) {
75			return FSCK_EXIT_CHECK_FAILED;
76		}
77	}
78
79	error = check_filesystem(fs, flags);
80
81	printf("write backing...(no progress report)"); fflush(stdout);
82	v7fs_io_fini(fs);
83	printf("done.\n");
84
85	return error;
86}
87
88int
89check_filesystem(struct v7fs_self *fs, int flags)
90{
91	int error;
92
93	/* Check superblock cached freeinode list. */
94	pwarn("[Superblock information]\n");
95	v7fs_superblock_dump(fs);
96	pwarn("[1] checking free inode in superblock...\n");
97	if ((error = freeinode_check(fs)))
98		return error;
99
100	/* Check free block linked list. */
101	pwarn("[2] checking free block link...\n");
102	if ((error = freeblock_check(fs)))
103		return error;
104
105	/* Check inode all. */
106	pwarn("[3] checking all ilist...\n");
107	if ((error = ilist_check(fs)))
108		return error;
109
110	/* Setup lost+found. */
111	pwarn("prepare lost+found\n");
112	if ((error = make_lost_and_found(fs, &lost_and_found)))
113		return FSCK_EXIT_CHECK_FAILED;
114
115	/* Check path(child and parent). Orphans are linked to lost+found. */
116	pwarn("[4] checking path name...\n");
117	if ((error = pathname_check(fs)))
118		return error;
119
120	if (flags & V7FS_FSCK_FREEBLOCK_DUP) {
121		/* Check duplicated block in freeblock. */
122		pwarn("[5] checking freeblock duplication...\n");
123		if ((error = freeblock_vs_freeblock_check(fs)))
124			return error;
125	}
126
127	if (flags & V7FS_FSCK_DATABLOCK_DUP) {
128		/* Check duplicated block in datablock. */
129		pwarn("[6] checking datablock duplication(vs datablock)...\n");
130		if ((error = datablock_vs_datablock_check(fs)))
131			return error;
132	}
133
134	if ((flags & V7FS_FSCK_DATABLOCK_DUP) && (flags &
135		V7FS_FSCK_FREEBLOCK_DUP)) {
136		/* Check off-diagonal. */
137		pwarn("[7] checking datablock duplication (vs freeblock)...\n");
138		if ((error = datablock_vs_freeblock_check(fs)))
139			return error;
140	}
141
142	return 0;
143}
144
145static int
146reply_subr(const char *str, bool trivial)
147{
148	int c;
149	char *line = NULL;
150	size_t linesize = 0;
151	ssize_t linelen;
152
153	printf("%s ", str);
154	switch (fsck_operate) {
155	case PREEN:
156		return trivial;
157	case ALWAYS_YES:
158		printf("[Y]\n");
159		return 1;
160	case ALWAYS_NO:
161		if (strstr(str, "CONTINUE")) {
162			printf("[Y]\n");
163			return 1;
164		}
165		printf("[N]\n");
166		return 0;
167	case ASK:
168		break;
169	}
170	fflush(stdout);
171
172	if ((linelen = getline(&line, &linesize, stdin)) == -1)	{
173		clearerr(stdin);
174		return 0;
175	}
176	c = line[0];
177
178	return c == 'y' || c == 'Y';
179}
180
181int
182reply(const char *str)
183{
184	return reply_subr(str, false);
185}
186
187int
188reply_trivial(const char *str)
189{
190	return reply_subr(str, true);
191}
192
193void
194progress(const struct progress_arg *p)
195{
196	static struct progress_arg Progress;
197	static char cdev[32];
198	static char label[32];
199
200	if (p) {
201		if (Progress.tick) {
202			progress_done();
203		}
204		Progress = *p;
205		if (p->cdev)
206			strcpy(cdev, p->cdev);
207		if (p->label)
208			strcpy(label, p->label);
209	}
210
211	if (fsck_operate == PREEN)
212		return;
213	if (!Progress.tick)
214		return;
215	if (++Progress.cnt > Progress.tick) {
216		Progress.cnt = 0;
217		Progress.total++;
218		progress_bar(cdev, label, Progress.total, PROGRESS_BAR_GRANULE);
219	}
220}
221
222int
223make_lost_and_found(struct v7fs_self *fs, struct v7fs_inode *p)
224{
225	struct v7fs_inode root_inode;
226	struct v7fs_fileattr attr;
227	v7fs_ino_t ino;
228	int error = 0;
229
230	if ((error = v7fs_inode_load(fs, &root_inode, V7FS_ROOT_INODE))) {
231		errno = error;
232		warn("(%s) / missing.", __func__);
233		return error;
234	}
235
236	memset(&attr, 0, sizeof(attr));
237	attr.uid = 0;
238	attr.gid = 0;
239	attr.mode = V7FS_IFDIR | 0755;
240	attr.device = 0;
241	attr.ctime = attr.mtime = attr.atime = (v7fs_time_t)time(NULL);
242
243	/* If lost+found already exists, returns EEXIST */
244	if (!(error = v7fs_file_allocate
245	      (fs, &root_inode, "lost+found", &attr, &ino)))
246		v7fs_superblock_writeback(fs);
247
248	if (error == EEXIST)
249		error = 0;
250
251	if (error) {
252		errno = error;
253		warn("(%s) Couldn't create lost+found", __func__);
254		return error;
255	}
256
257	if ((error = v7fs_inode_load(fs, p, ino))) {
258		errno = error;
259		warn("(%s:)lost+found is created, but missing.", __func__);
260	}
261
262	return error;
263}
264