1 /*
2  * util.c --- miscellaneous utilities
3  *
4  * Copyright (C) 1993, 1994, 1995, 1996, 1997 Theodore Ts'o.
5  *
6  * %Begin-Header%
7  * This file may be redistributed under the terms of the GNU Public
8  * License.
9  * %End-Header%
10  */
11 
12 #include <stdlib.h>
13 #include <stdio.h>
14 #include <unistd.h>
15 #include <string.h>
16 #include <ctype.h>
17 #ifdef __linux__
18 #include <sys/utsname.h>
19 #endif
20 
21 #ifdef HAVE_CONIO_H
22 #undef HAVE_TERMIOS_H
23 #include <conio.h>
24 #define read_a_char()	getch()
25 #else
26 #ifdef HAVE_TERMIOS_H
27 #include <termios.h>
28 #endif
29 #endif
30 
31 #ifdef HAVE_MALLOC_H
32 #include <malloc.h>
33 #endif
34 
35 #ifdef HAVE_ERRNO_H
36 #include <errno.h>
37 #endif
38 
39 #include "e2fsck.h"
40 
41 extern e2fsck_t e2fsck_global_ctx;   /* Try your very best not to use this! */
42 
43 #include <stdarg.h>
44 #include <time.h>
45 #include <sys/time.h>
46 #include <sys/resource.h>
47 
fatal_error(e2fsck_t ctx,const char * msg)48 void fatal_error(e2fsck_t ctx, const char *msg)
49 {
50 	ext2_filsys fs = ctx->fs;
51 	int exit_value = FSCK_ERROR;
52 
53 	if (msg)
54 		fprintf (stderr, "e2fsck: %s\n", msg);
55 	if (!fs)
56 		goto out;
57 	if (fs->io && fs->super) {
58 		ext2fs_mmp_stop(ctx->fs);
59 		if (ctx->fs->io->magic == EXT2_ET_MAGIC_IO_CHANNEL)
60 			io_channel_flush(ctx->fs->io);
61 		else
62 			log_err(ctx, "e2fsck: io manager magic bad!\n");
63 	}
64 	if (ext2fs_test_changed(fs)) {
65 		exit_value |= FSCK_NONDESTRUCT;
66 		log_out(ctx, _("\n%s: ***** FILE SYSTEM WAS MODIFIED *****\n"),
67 			ctx->device_name);
68 		if (ctx->mount_flags & EXT2_MF_ISROOT)
69 			exit_value |= FSCK_REBOOT;
70 	}
71 	if (!ext2fs_test_valid(fs)) {
72 		log_out(ctx, _("\n%s: ********** WARNING: Filesystem still has "
73 			       "errors **********\n\n"), ctx->device_name);
74 		exit_value |= FSCK_UNCORRECTED;
75 		exit_value &= ~FSCK_NONDESTRUCT;
76 	}
77 out:
78 	ctx->flags |= E2F_FLAG_ABORT;
79 	if (ctx->flags & E2F_FLAG_SETJMP_OK)
80 		longjmp(ctx->abort_loc, 1);
81 	exit(exit_value);
82 }
83 
log_out(e2fsck_t ctx,const char * fmt,...)84 void log_out(e2fsck_t ctx, const char *fmt, ...)
85 {
86 	va_list pvar;
87 
88 	va_start(pvar, fmt);
89 	vprintf(fmt, pvar);
90 	va_end(pvar);
91 	if (ctx->logf) {
92 		va_start(pvar, fmt);
93 		vfprintf(ctx->logf, fmt, pvar);
94 		va_end(pvar);
95 	}
96 }
97 
log_err(e2fsck_t ctx,const char * fmt,...)98 void log_err(e2fsck_t ctx, const char *fmt, ...)
99 {
100 	va_list pvar;
101 
102 	va_start(pvar, fmt);
103 	vfprintf(stderr, fmt, pvar);
104 	va_end(pvar);
105 	if (ctx->logf) {
106 		va_start(pvar, fmt);
107 		vfprintf(ctx->logf, fmt, pvar);
108 		va_end(pvar);
109 	}
110 }
111 
e2fsck_allocate_memory(e2fsck_t ctx,unsigned int size,const char * description)112 void *e2fsck_allocate_memory(e2fsck_t ctx, unsigned int size,
113 			     const char *description)
114 {
115 	void *ret;
116 	char buf[256];
117 
118 #ifdef DEBUG_ALLOCATE_MEMORY
119 	printf("Allocating %u bytes for %s...\n", size, description);
120 #endif
121 	ret = malloc(size);
122 	if (!ret) {
123 		sprintf(buf, "Can't allocate %u bytes for %s\n",
124 			size, description);
125 		fatal_error(ctx, buf);
126 	}
127 	memset(ret, 0, size);
128 	return ret;
129 }
130 
string_copy(e2fsck_t ctx EXT2FS_ATTR ((unused)),const char * str,int len)131 char *string_copy(e2fsck_t ctx EXT2FS_ATTR((unused)),
132 		  const char *str, int len)
133 {
134 	char	*ret;
135 
136 	if (!str)
137 		return NULL;
138 	if (!len)
139 		len = strlen(str);
140 	ret = malloc(len+1);
141 	if (ret) {
142 		strncpy(ret, str, len);
143 		ret[len] = 0;
144 	}
145 	return ret;
146 }
147 
148 #ifndef HAVE_STRNLEN
149 /*
150  * Incredibly, libc5 doesn't appear to have strnlen.  So we have to
151  * provide our own.
152  */
e2fsck_strnlen(const char * s,int count)153 int e2fsck_strnlen(const char * s, int count)
154 {
155 	const char *cp = s;
156 
157 	while (count-- && *cp)
158 		cp++;
159 	return cp - s;
160 }
161 #endif
162 
163 #ifndef HAVE_CONIO_H
read_a_char(void)164 static int read_a_char(void)
165 {
166 	char	c;
167 	int	r;
168 	int	fail = 0;
169 
170 	while(1) {
171 		if (e2fsck_global_ctx &&
172 		    (e2fsck_global_ctx->flags & E2F_FLAG_CANCEL)) {
173 			return 3;
174 		}
175 		r = read(0, &c, 1);
176 		if (r == 1)
177 			return c;
178 		if (fail++ > 100)
179 			break;
180 	}
181 	return EOF;
182 }
183 #endif
184 
ask_yn(e2fsck_t ctx,const char * string,int def)185 int ask_yn(e2fsck_t ctx, const char * string, int def)
186 {
187 	int		c;
188 	const char	*defstr;
189 	const char	*short_yes = _("yY");
190 	const char	*short_no = _("nN");
191 
192 #ifdef HAVE_TERMIOS_H
193 	struct termios	termios, tmp;
194 
195 	tcgetattr (0, &termios);
196 	tmp = termios;
197 	tmp.c_lflag &= ~(ICANON | ECHO);
198 	tmp.c_cc[VMIN] = 1;
199 	tmp.c_cc[VTIME] = 0;
200 	tcsetattr (0, TCSANOW, &tmp);
201 #endif
202 
203 	if (def == 1)
204 		defstr = _(_("<y>"));
205 	else if (def == 0)
206 		defstr = _(_("<n>"));
207 	else
208 		defstr = _(" (y/n)");
209 	log_out(ctx, "%s%s? ", string, defstr);
210 	while (1) {
211 		fflush (stdout);
212 		if ((c = read_a_char()) == EOF)
213 			break;
214 		if (c == 3) {
215 #ifdef HAVE_TERMIOS_H
216 			tcsetattr (0, TCSANOW, &termios);
217 #endif
218 			if (ctx->flags & E2F_FLAG_SETJMP_OK) {
219 				log_out(ctx, "\n");
220 				longjmp(e2fsck_global_ctx->abort_loc, 1);
221 			}
222 			log_out(ctx, "%s", _("cancelled!\n"));
223 			return 0;
224 		}
225 		if (strchr(short_yes, (char) c)) {
226 			def = 1;
227 			break;
228 		}
229 		else if (strchr(short_no, (char) c)) {
230 			def = 0;
231 			break;
232 		}
233 		else if ((c == 27 || c == ' ' || c == '\n') && (def != -1))
234 			break;
235 	}
236 	if (def)
237 		log_out(ctx, "%s", _("yes\n"));
238 	else
239 		log_out(ctx, "%s", _("no\n"));
240 #ifdef HAVE_TERMIOS_H
241 	tcsetattr (0, TCSANOW, &termios);
242 #endif
243 	return def;
244 }
245 
ask(e2fsck_t ctx,const char * string,int def)246 int ask (e2fsck_t ctx, const char * string, int def)
247 {
248 	if (ctx->options & E2F_OPT_NO) {
249 		log_out(ctx, _("%s? no\n\n"), string);
250 		return 0;
251 	}
252 	if (ctx->options & E2F_OPT_YES) {
253 		log_out(ctx, _("%s? yes\n\n"), string);
254 		return 1;
255 	}
256 	if (ctx->options & E2F_OPT_PREEN) {
257 		log_out(ctx, "%s? %s\n\n", string, def ? _("yes") : _("no"));
258 		return def;
259 	}
260 	return ask_yn(ctx, string, def);
261 }
262 
e2fsck_read_bitmaps(e2fsck_t ctx)263 void e2fsck_read_bitmaps(e2fsck_t ctx)
264 {
265 	ext2_filsys fs = ctx->fs;
266 	errcode_t	retval;
267 	const char	*old_op;
268 	unsigned int	save_type;
269 
270 	if (ctx->invalid_bitmaps) {
271 		com_err(ctx->program_name, 0,
272 		    _("e2fsck_read_bitmaps: illegal bitmap block(s) for %s"),
273 			ctx->device_name);
274 		fatal_error(ctx, 0);
275 	}
276 
277 	old_op = ehandler_operation(_("reading inode and block bitmaps"));
278 	e2fsck_set_bitmap_type(fs, EXT2FS_BMAP64_RBTREE, "fs_bitmaps",
279 			       &save_type);
280 	retval = ext2fs_read_bitmaps(fs);
281 	fs->default_bitmap_type = save_type;
282 	ehandler_operation(old_op);
283 	if (retval) {
284 		com_err(ctx->program_name, retval,
285 			_("while retrying to read bitmaps for %s"),
286 			ctx->device_name);
287 		fatal_error(ctx, 0);
288 	}
289 }
290 
e2fsck_write_bitmaps(e2fsck_t ctx)291 void e2fsck_write_bitmaps(e2fsck_t ctx)
292 {
293 	ext2_filsys fs = ctx->fs;
294 	errcode_t	retval;
295 	const char	*old_op;
296 
297 	old_op = ehandler_operation(_("writing block and inode bitmaps"));
298 	retval = ext2fs_write_bitmaps(fs);
299 	ehandler_operation(old_op);
300 	if (retval) {
301 		com_err(ctx->program_name, retval,
302 			_("while rewriting block and inode bitmaps for %s"),
303 			ctx->device_name);
304 		fatal_error(ctx, 0);
305 	}
306 }
307 
preenhalt(e2fsck_t ctx)308 void preenhalt(e2fsck_t ctx)
309 {
310 	ext2_filsys fs = ctx->fs;
311 
312 	if (!(ctx->options & E2F_OPT_PREEN))
313 		return;
314 	log_err(ctx, _("\n\n%s: UNEXPECTED INCONSISTENCY; "
315 		"RUN fsck MANUALLY.\n\t(i.e., without -a or -p options)\n"),
316 	       ctx->device_name);
317 	ctx->flags |= E2F_FLAG_EXITING;
318 	if (fs != NULL) {
319 		fs->super->s_state |= EXT2_ERROR_FS;
320 		ext2fs_mark_super_dirty(fs);
321 		ext2fs_close(fs);
322 	}
323 	exit(FSCK_UNCORRECTED);
324 }
325 
326 #ifdef RESOURCE_TRACK
init_resource_track(struct resource_track * track,io_channel channel)327 void init_resource_track(struct resource_track *track, io_channel channel)
328 {
329 #ifdef HAVE_GETRUSAGE
330 	struct rusage r;
331 #endif
332 	io_stats io_start = 0;
333 
334 	track->brk_start = sbrk(0);
335 	gettimeofday(&track->time_start, 0);
336 #ifdef HAVE_GETRUSAGE
337 #ifdef sun
338 	memset(&r, 0, sizeof(struct rusage));
339 #endif
340 	getrusage(RUSAGE_SELF, &r);
341 	track->user_start = r.ru_utime;
342 	track->system_start = r.ru_stime;
343 #else
344 	track->user_start.tv_sec = track->user_start.tv_usec = 0;
345 	track->system_start.tv_sec = track->system_start.tv_usec = 0;
346 #endif
347 	track->bytes_read = 0;
348 	track->bytes_written = 0;
349 	if (channel && channel->manager && channel->manager->get_stats)
350 		channel->manager->get_stats(channel, &io_start);
351 	if (io_start) {
352 		track->bytes_read = io_start->bytes_read;
353 		track->bytes_written = io_start->bytes_written;
354 	}
355 }
356 
357 #ifdef __GNUC__
358 #define _INLINE_ __inline__
359 #else
360 #define _INLINE_
361 #endif
362 
timeval_subtract(struct timeval * tv1,struct timeval * tv2)363 static _INLINE_ float timeval_subtract(struct timeval *tv1,
364 				       struct timeval *tv2)
365 {
366 	return ((tv1->tv_sec - tv2->tv_sec) +
367 		((float) (tv1->tv_usec - tv2->tv_usec)) / 1000000);
368 }
369 
print_resource_track(e2fsck_t ctx,const char * desc,struct resource_track * track,io_channel channel)370 void print_resource_track(e2fsck_t ctx, const char *desc,
371 			  struct resource_track *track, io_channel channel)
372 {
373 #ifdef HAVE_GETRUSAGE
374 	struct rusage r;
375 #endif
376 #ifdef HAVE_MALLINFO
377 	struct mallinfo	malloc_info;
378 #endif
379 	struct timeval time_end;
380 
381 	if ((desc && !(ctx->options & E2F_OPT_TIME2)) ||
382 	    (!desc && !(ctx->options & E2F_OPT_TIME)))
383 		return;
384 
385 	e2fsck_clear_progbar(ctx);
386 	gettimeofday(&time_end, 0);
387 
388 	if (desc)
389 		log_out(ctx, "%s: ", desc);
390 
391 #ifdef HAVE_MALLINFO
392 #define kbytes(x)	(((unsigned long)(x) + 1023) / 1024)
393 
394 	malloc_info = mallinfo();
395 	log_out(ctx, _("Memory used: %luk/%luk (%luk/%luk), "),
396 		kbytes(malloc_info.arena), kbytes(malloc_info.hblkhd),
397 		kbytes(malloc_info.uordblks), kbytes(malloc_info.fordblks));
398 #else
399 	log_out(ctx, _("Memory used: %lu, "),
400 		(unsigned long) (((char *) sbrk(0)) -
401 				 ((char *) track->brk_start)));
402 #endif
403 #ifdef HAVE_GETRUSAGE
404 	getrusage(RUSAGE_SELF, &r);
405 
406 	log_out(ctx, _("time: %5.2f/%5.2f/%5.2f\n"),
407 		timeval_subtract(&time_end, &track->time_start),
408 		timeval_subtract(&r.ru_utime, &track->user_start),
409 		timeval_subtract(&r.ru_stime, &track->system_start));
410 #else
411 	log_out(ctx, _("elapsed time: %6.3f\n"),
412 		timeval_subtract(&time_end, &track->time_start));
413 #endif
414 #define mbytes(x)	(((x) + 1048575) / 1048576)
415 	if (channel && channel->manager && channel->manager->get_stats) {
416 		io_stats delta = 0;
417 		unsigned long long bytes_read = 0;
418 		unsigned long long bytes_written = 0;
419 
420 		if (desc)
421 			log_out(ctx, "%s: ", desc);
422 
423 		channel->manager->get_stats(channel, &delta);
424 		if (delta) {
425 			bytes_read = delta->bytes_read - track->bytes_read;
426 			bytes_written = delta->bytes_written -
427 				track->bytes_written;
428 		}
429 		log_out(ctx, "I/O read: %lluMB, write: %lluMB, "
430 			"rate: %.2fMB/s\n",
431 			mbytes(bytes_read), mbytes(bytes_written),
432 			(double)mbytes(bytes_read + bytes_written) /
433 			timeval_subtract(&time_end, &track->time_start));
434 	}
435 }
436 #endif /* RESOURCE_TRACK */
437 
e2fsck_read_inode(e2fsck_t ctx,unsigned long ino,struct ext2_inode * inode,const char * proc)438 void e2fsck_read_inode(e2fsck_t ctx, unsigned long ino,
439 			      struct ext2_inode * inode, const char *proc)
440 {
441 	errcode_t retval;
442 
443 	retval = ext2fs_read_inode(ctx->fs, ino, inode);
444 	if (retval) {
445 		com_err("ext2fs_read_inode", retval,
446 			_("while reading inode %lu in %s"), ino, proc);
447 		fatal_error(ctx, 0);
448 	}
449 }
450 
e2fsck_read_inode_full(e2fsck_t ctx,unsigned long ino,struct ext2_inode * inode,int bufsize,const char * proc)451 void e2fsck_read_inode_full(e2fsck_t ctx, unsigned long ino,
452 			    struct ext2_inode *inode, int bufsize,
453 			    const char *proc)
454 {
455 	errcode_t retval;
456 
457 	retval = ext2fs_read_inode_full(ctx->fs, ino, inode, bufsize);
458 	if (retval) {
459 		com_err("ext2fs_read_inode_full", retval,
460 			_("while reading inode %lu in %s"), ino, proc);
461 		fatal_error(ctx, 0);
462 	}
463 }
464 
e2fsck_write_inode_full(e2fsck_t ctx,unsigned long ino,struct ext2_inode * inode,int bufsize,const char * proc)465 void e2fsck_write_inode_full(e2fsck_t ctx, unsigned long ino,
466 			     struct ext2_inode * inode, int bufsize,
467 			     const char *proc)
468 {
469 	errcode_t retval;
470 
471 	retval = ext2fs_write_inode_full(ctx->fs, ino, inode, bufsize);
472 	if (retval) {
473 		com_err("ext2fs_write_inode", retval,
474 			_("while writing inode %lu in %s"), ino, proc);
475 		fatal_error(ctx, 0);
476 	}
477 }
478 
e2fsck_write_inode(e2fsck_t ctx,unsigned long ino,struct ext2_inode * inode,const char * proc)479 void e2fsck_write_inode(e2fsck_t ctx, unsigned long ino,
480 			struct ext2_inode * inode, const char *proc)
481 {
482 	errcode_t retval;
483 
484 	retval = ext2fs_write_inode(ctx->fs, ino, inode);
485 	if (retval) {
486 		com_err("ext2fs_write_inode", retval,
487 			_("while writing inode %lu in %s"), ino, proc);
488 		fatal_error(ctx, 0);
489 	}
490 }
491 
492 #ifdef MTRACE
mtrace_print(char * mesg)493 void mtrace_print(char *mesg)
494 {
495 	FILE	*malloc_get_mallstream();
496 	FILE	*f = malloc_get_mallstream();
497 
498 	if (f)
499 		fprintf(f, "============= %s\n", mesg);
500 }
501 #endif
502 
get_backup_sb(e2fsck_t ctx,ext2_filsys fs,const char * name,io_manager manager)503 blk64_t get_backup_sb(e2fsck_t ctx, ext2_filsys fs, const char *name,
504 		      io_manager manager)
505 {
506 	struct ext2_super_block *sb;
507 	io_channel		io = NULL;
508 	void			*buf = NULL;
509 	int			blocksize;
510 	blk64_t			superblock, ret_sb = 8193;
511 
512 	if (fs && fs->super) {
513 		ret_sb = (fs->super->s_blocks_per_group +
514 			  fs->super->s_first_data_block);
515 		if (ctx) {
516 			ctx->superblock = ret_sb;
517 			ctx->blocksize = fs->blocksize;
518 		}
519 		return ret_sb;
520 	}
521 
522 	if (ctx) {
523 		if (ctx->blocksize) {
524 			ret_sb = ctx->blocksize * 8;
525 			if (ctx->blocksize == 1024)
526 				ret_sb++;
527 			ctx->superblock = ret_sb;
528 			return ret_sb;
529 		}
530 		ctx->superblock = ret_sb;
531 		ctx->blocksize = 1024;
532 	}
533 
534 	if (!name || !manager)
535 		goto cleanup;
536 
537 	if (manager->open(name, 0, &io) != 0)
538 		goto cleanup;
539 
540 	if (ext2fs_get_mem(SUPERBLOCK_SIZE, &buf))
541 		goto cleanup;
542 	sb = (struct ext2_super_block *) buf;
543 
544 	for (blocksize = EXT2_MIN_BLOCK_SIZE;
545 	     blocksize <= EXT2_MAX_BLOCK_SIZE ; blocksize *= 2) {
546 		superblock = blocksize*8;
547 		if (blocksize == 1024)
548 			superblock++;
549 		io_channel_set_blksize(io, blocksize);
550 		if (io_channel_read_blk64(io, superblock,
551 					-SUPERBLOCK_SIZE, buf))
552 			continue;
553 #ifdef WORDS_BIGENDIAN
554 		if (sb->s_magic == ext2fs_swab16(EXT2_SUPER_MAGIC))
555 			ext2fs_swap_super(sb);
556 #endif
557 		if ((sb->s_magic == EXT2_SUPER_MAGIC) &&
558 		    (EXT2_BLOCK_SIZE(sb) == blocksize)) {
559 			ret_sb = superblock;
560 			if (ctx) {
561 				ctx->superblock = superblock;
562 				ctx->blocksize = blocksize;
563 			}
564 			break;
565 		}
566 	}
567 
568 cleanup:
569 	if (io)
570 		io_channel_close(io);
571 	if (buf)
572 		ext2fs_free_mem(&buf);
573 	return (ret_sb);
574 }
575 
576 /*
577  * Given a mode, return the ext2 file type
578  */
ext2_file_type(unsigned int mode)579 int ext2_file_type(unsigned int mode)
580 {
581 	if (LINUX_S_ISREG(mode))
582 		return EXT2_FT_REG_FILE;
583 
584 	if (LINUX_S_ISDIR(mode))
585 		return EXT2_FT_DIR;
586 
587 	if (LINUX_S_ISCHR(mode))
588 		return EXT2_FT_CHRDEV;
589 
590 	if (LINUX_S_ISBLK(mode))
591 		return EXT2_FT_BLKDEV;
592 
593 	if (LINUX_S_ISLNK(mode))
594 		return EXT2_FT_SYMLINK;
595 
596 	if (LINUX_S_ISFIFO(mode))
597 		return EXT2_FT_FIFO;
598 
599 	if (LINUX_S_ISSOCK(mode))
600 		return EXT2_FT_SOCK;
601 
602 	return 0;
603 }
604 
605 #define STRIDE_LENGTH 8
606 /*
607  * Helper function which zeros out _num_ blocks starting at _blk_.  In
608  * case of an error, the details of the error is returned via _ret_blk_
609  * and _ret_count_ if they are non-NULL pointers.  Returns 0 on
610  * success, and an error code on an error.
611  *
612  * As a special case, if the first argument is NULL, then it will
613  * attempt to free the static zeroizing buffer.  (This is to keep
614  * programs that check for memory leaks happy.)
615  */
e2fsck_zero_blocks(ext2_filsys fs,blk_t blk,int num,blk_t * ret_blk,int * ret_count)616 errcode_t e2fsck_zero_blocks(ext2_filsys fs, blk_t blk, int num,
617 			     blk_t *ret_blk, int *ret_count)
618 {
619 	int		j, count;
620 	static char	*buf;
621 	errcode_t	retval;
622 
623 	/* If fs is null, clean up the static buffer and return */
624 	if (!fs) {
625 		if (buf) {
626 			free(buf);
627 			buf = 0;
628 		}
629 		return 0;
630 	}
631 	/* Allocate the zeroizing buffer if necessary */
632 	if (!buf) {
633 		buf = malloc(fs->blocksize * STRIDE_LENGTH);
634 		if (!buf) {
635 			com_err("malloc", ENOMEM, "%s",
636 				_("while allocating zeroizing buffer"));
637 			exit(1);
638 		}
639 		memset(buf, 0, fs->blocksize * STRIDE_LENGTH);
640 	}
641 	/* OK, do the write loop */
642 	for (j = 0; j < num; j += STRIDE_LENGTH, blk += STRIDE_LENGTH) {
643 		count = num - j;
644 		if (count > STRIDE_LENGTH)
645 			count = STRIDE_LENGTH;
646 		retval = io_channel_write_blk64(fs->io, blk, count, buf);
647 		if (retval) {
648 			if (ret_count)
649 				*ret_count = count;
650 			if (ret_blk)
651 				*ret_blk = blk;
652 			return retval;
653 		}
654 	}
655 	return 0;
656 }
657 
658 /*
659  * Check to see if a filesystem is in /proc/filesystems.
660  * Returns 1 if found, 0 if not
661  */
fs_proc_check(const char * fs_name)662 int fs_proc_check(const char *fs_name)
663 {
664 	FILE	*f;
665 	char	buf[80], *cp, *t;
666 
667 	f = fopen("/proc/filesystems", "r");
668 	if (!f)
669 		return (0);
670 	while (!feof(f)) {
671 		if (!fgets(buf, sizeof(buf), f))
672 			break;
673 		cp = buf;
674 		if (!isspace(*cp)) {
675 			while (*cp && !isspace(*cp))
676 				cp++;
677 		}
678 		while (*cp && isspace(*cp))
679 			cp++;
680 		if ((t = strchr(cp, '\n')) != NULL)
681 			*t = 0;
682 		if ((t = strchr(cp, '\t')) != NULL)
683 			*t = 0;
684 		if ((t = strchr(cp, ' ')) != NULL)
685 			*t = 0;
686 		if (!strcmp(fs_name, cp)) {
687 			fclose(f);
688 			return (1);
689 		}
690 	}
691 	fclose(f);
692 	return (0);
693 }
694 
695 /*
696  * Check to see if a filesystem is available as a module
697  * Returns 1 if found, 0 if not
698  */
check_for_modules(const char * fs_name)699 int check_for_modules(const char *fs_name)
700 {
701 #ifdef __linux__
702 	struct utsname	uts;
703 	FILE		*f;
704 	char		buf[1024], *cp, *t;
705 	int		i;
706 
707 	if (uname(&uts))
708 		return (0);
709 	snprintf(buf, sizeof(buf), "/lib/modules/%s/modules.dep", uts.release);
710 
711 	f = fopen(buf, "r");
712 	if (!f)
713 		return (0);
714 	while (!feof(f)) {
715 		if (!fgets(buf, sizeof(buf), f))
716 			break;
717 		if ((cp = strchr(buf, ':')) != NULL)
718 			*cp = 0;
719 		else
720 			continue;
721 		if ((cp = strrchr(buf, '/')) != NULL)
722 			cp++;
723 		else
724 			cp = buf;
725 		i = strlen(cp);
726 		if (i > 3) {
727 			t = cp + i - 3;
728 			if (!strcmp(t, ".ko"))
729 				*t = 0;
730 		}
731 		if (!strcmp(cp, fs_name)) {
732 			fclose(f);
733 			return (1);
734 		}
735 	}
736 	fclose(f);
737 #endif /* __linux__ */
738 	return (0);
739 }
740 
741 /*
742  * Helper function that does the right thing if write returns a
743  * partial write, or an EGAIN/EINTR error.
744  */
write_all(int fd,char * buf,size_t count)745 int write_all(int fd, char *buf, size_t count)
746 {
747 	ssize_t ret;
748 	int c = 0;
749 
750 	while (count > 0) {
751 		ret = write(fd, buf, count);
752 		if (ret < 0) {
753 			if ((errno == EAGAIN) || (errno == EINTR))
754 				continue;
755 			return -1;
756 		}
757 		count -= ret;
758 		buf += ret;
759 		c += ret;
760 	}
761 	return c;
762 }
763 
dump_mmp_msg(struct mmp_struct * mmp,const char * msg)764 void dump_mmp_msg(struct mmp_struct *mmp, const char *msg)
765 {
766 
767 	if (msg)
768 		printf("MMP check failed: %s\n", msg);
769 	if (mmp) {
770 		time_t t = mmp->mmp_time;
771 
772 		printf("MMP error info: last update: %s node: %s device: %s\n",
773 		       ctime(&t), mmp->mmp_nodename, mmp->mmp_bdevname);
774 	}
775 }
776 
e2fsck_mmp_update(ext2_filsys fs)777 errcode_t e2fsck_mmp_update(ext2_filsys fs)
778 {
779 	errcode_t retval;
780 
781 	retval = ext2fs_mmp_update(fs);
782 	if (retval == EXT2_ET_MMP_CHANGE_ABORT)
783 		dump_mmp_msg(fs->mmp_cmp,
784 			     _("UNEXPECTED INCONSISTENCY: the filesystem is "
785 			       "being modified while fsck is running.\n"));
786 
787 	return retval;
788 }
789 
e2fsck_set_bitmap_type(ext2_filsys fs,unsigned int default_type,const char * profile_name,unsigned int * old_type)790 void e2fsck_set_bitmap_type(ext2_filsys fs, unsigned int default_type,
791 			    const char *profile_name, unsigned int *old_type)
792 {
793 	unsigned type;
794 
795 	if (old_type)
796 		*old_type = fs->default_bitmap_type;
797 #ifdef HAVE_SIGNAL_H
798 	profile_get_uint(e2fsck_global_ctx->profile, "bitmaps",
799 			 profile_name, 0, default_type, &type);
800 	profile_get_uint(e2fsck_global_ctx->profile, "bitmaps",
801 			 "all", 0, type, &type);
802 	fs->default_bitmap_type = type ? type : default_type;
803 #else
804 	fs->default_bitmap_type = default_type;
805 #endif
806 }
807 
e2fsck_allocate_inode_bitmap(ext2_filsys fs,const char * descr,int deftype,const char * name,ext2fs_inode_bitmap * ret)808 errcode_t e2fsck_allocate_inode_bitmap(ext2_filsys fs, const char *descr,
809 				       int deftype,
810 				       const char *name,
811 				       ext2fs_inode_bitmap *ret)
812 {
813 	errcode_t	retval;
814 	unsigned int	save_type;
815 
816 	e2fsck_set_bitmap_type(fs, deftype, name, &save_type);
817 	retval = ext2fs_allocate_inode_bitmap(fs, descr, ret);
818 	fs->default_bitmap_type = save_type;
819 	return retval;
820 }
821 
e2fsck_allocate_block_bitmap(ext2_filsys fs,const char * descr,int deftype,const char * name,ext2fs_block_bitmap * ret)822 errcode_t e2fsck_allocate_block_bitmap(ext2_filsys fs, const char *descr,
823 				       int deftype,
824 				       const char *name,
825 				       ext2fs_block_bitmap *ret)
826 {
827 	errcode_t	retval;
828 	unsigned int	save_type;
829 
830 	e2fsck_set_bitmap_type(fs, deftype, name, &save_type);
831 	retval = ext2fs_allocate_block_bitmap(fs, descr, ret);
832 	fs->default_bitmap_type = save_type;
833 	return retval;
834 }
835 
e2fsck_allocate_subcluster_bitmap(ext2_filsys fs,const char * descr,int deftype,const char * name,ext2fs_block_bitmap * ret)836 errcode_t e2fsck_allocate_subcluster_bitmap(ext2_filsys fs, const char *descr,
837 					    int deftype,
838 					    const char *name,
839 					    ext2fs_block_bitmap *ret)
840 {
841 	errcode_t	retval;
842 	unsigned int	save_type;
843 
844 	e2fsck_set_bitmap_type(fs, deftype, name, &save_type);
845 	retval = ext2fs_allocate_subcluster_bitmap(fs, descr, ret);
846 	fs->default_bitmap_type = save_type;
847 	return retval;
848 }
849