[thirdparty/e2fsprogs.git] / misc / e2freefrag.c

/*
 * e2freefrag - report filesystem free-space fragmentation
 *
 * Copyright (C) 2009 Sun Microsystems, Inc.
 *
 * Author: Rupesh Thakare <rupesh@sun.com>
 *         Andreas Dilger <adilger@sun.com>
 *
 * %Begin-Header%
 * This file may be redistributed under the terms of the GNU Public
 * License version 2.
 * %End-Header%
 */
#include "config.h"
#include <stdio.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#ifdef HAVE_STDLIB_H
#include <stdlib.h>
#endif
#ifdef HAVE_GETOPT_H
#include <getopt.h>
#else
extern char *optarg;
extern int optind;
#endif
#if defined(HAVE_EXT2_IOCTLS) && !defined(DEBUGFS)
# include <sys/ioctl.h>
# include <sys/types.h>
# include <sys/stat.h>
# include <fcntl.h>
# include <limits.h>
#endif

#include "ext2fs/ext2_fs.h"
#include "ext2fs/ext2fs.h"
#include "e2freefrag.h"

#if defined(HAVE_EXT2_IOCTLS) && !defined(DEBUGFS)
# if defined(HAVE_LINUX_FSMAP_H) && defined(HAVE_FSMAP_SIZEOF)
#  include <linux/fsmap.h>
# endif
# include "fsmap.h"
#endif

#ifndef PATH_MAX
#define PATH_MAX 4096
#endif

static void usage(const char *prog)
{
	fprintf(stderr, "usage: %s [-c chunksize in kb] [-h] "
		"device_name\n", prog);
#ifndef DEBUGFS
	exit(1);
#endif
}

static int ul_log2(unsigned long arg)
{
        int     l = 0;

        arg >>= 1;
        while (arg) {
                l++;
                arg >>= 1;
        }
        return l;
}

static void init_chunk_info(ext2_filsys fs, struct chunk_info *info)
{
	int i;

	info->blocksize_bits = ul_log2((unsigned long)fs->blocksize);
	if (info->chunkbytes) {
		info->chunkbits = ul_log2(info->chunkbytes);
		info->blks_in_chunk = info->chunkbytes >> info->blocksize_bits;
	} else {
		info->chunkbits = ul_log2(DEFAULT_CHUNKSIZE);
		info->blks_in_chunk = DEFAULT_CHUNKSIZE >> info->blocksize_bits;
	}

	info->min = ~0UL;
	info->max = info->avg = 0;
	info->real_free_chunks = 0;

	for (i = 0; i < MAX_HIST; i++) {
		info->histogram.fc_chunks[i] = 0;
		info->histogram.fc_blocks[i] = 0;
	}
}

static void update_chunk_stats(struct chunk_info *info,
			       unsigned long chunk_size)
{
	unsigned long idx;

	idx = ul_log2(chunk_size) + 1;
	if (idx >= MAX_HIST)
		idx = MAX_HIST-1;
	info->histogram.fc_chunks[idx]++;
	info->histogram.fc_blocks[idx] += chunk_size;

	if (chunk_size > info->max)
		info->max = chunk_size;
	if (chunk_size < info->min)
		info->min = chunk_size;
	info->avg += chunk_size;
	info->real_free_chunks++;
}

static void scan_block_bitmap(ext2_filsys fs, struct chunk_info *info)
{
	unsigned long long blocks_count = ext2fs_blocks_count(fs->super);
	unsigned long long chunks = (blocks_count + info->blks_in_chunk) >>
				(info->chunkbits - info->blocksize_bits);
	unsigned long long chunk_num;
	unsigned long last_chunk_size = 0;
	unsigned long long chunk_start_blk = 0;
	int used;

	for (chunk_num = 0; chunk_num < chunks; chunk_num++) {
		unsigned long long blk, num_blks;
		int chunk_free;

		/* Last chunk may be smaller */
		if (chunk_start_blk + info->blks_in_chunk > blocks_count)
			num_blks = blocks_count - chunk_start_blk;
		else
			num_blks = info->blks_in_chunk;

		chunk_free = 0;

		/* Initialize starting block for first chunk correctly else
		 * there is a segfault when blocksize = 1024 in which case
		 * block_map->start = 1 */
		for (blk = 0; blk < num_blks; blk++, chunk_start_blk++) {
			if (chunk_num == 0 && blk == 0) {
				blk = fs->super->s_first_data_block;
				chunk_start_blk = blk;
			}
			used = ext2fs_fast_test_block_bitmap2(fs->block_map,
				chunk_start_blk >> fs->cluster_ratio_bits);
			if (!used) {
				last_chunk_size++;
				chunk_free++;
			}

			if (used && last_chunk_size != 0) {
				update_chunk_stats(info, last_chunk_size);
				last_chunk_size = 0;
			}
		}

		if (chunk_free == info->blks_in_chunk)
			info->free_chunks++;
	}
	if (last_chunk_size != 0)
		update_chunk_stats(info, last_chunk_size);
}

#if defined(HAVE_EXT2_IOCTLS) && !defined(DEBUGFS)
# define FSMAP_EXTENTS	1024
static int scan_online(ext2_filsys fs, struct chunk_info *info,
		       blk64_t *free_blks)
{
	struct fsmap_head *fsmap;
	struct fsmap *extent;
	struct fsmap *p;
	char mntpoint[PATH_MAX + 1];
	errcode_t retval;
	int mount_flags;
	int fd;
	int ret;
	unsigned int i;

	/* Try to open the mountpoint for a live query. */
	retval = ext2fs_check_mount_point(fs->device_name, &mount_flags,
					  mntpoint, PATH_MAX);
	if (retval) {
		com_err(fs->device_name, retval, "while checking mount status");
		return 0;
	}
	if (!(mount_flags & EXT2_MF_MOUNTED))
		return 0;
	fd = open(mntpoint, O_RDONLY);
	if (fd < 0) {
		com_err(mntpoint, errno, "while opening mount point");
		return 0;
	}

	fsmap = malloc(fsmap_sizeof(FSMAP_EXTENTS));
	if (!fsmap) {
		com_err(fs->device_name, errno, "while allocating memory");
		return 0;
	}

	memset(fsmap, 0, sizeof(*fsmap));
	fsmap->fmh_count = FSMAP_EXTENTS;
	fsmap->fmh_keys[1].fmr_device = UINT_MAX;
	fsmap->fmh_keys[1].fmr_physical = ULLONG_MAX;
	fsmap->fmh_keys[1].fmr_owner = ULLONG_MAX;
	fsmap->fmh_keys[1].fmr_offset = ULLONG_MAX;
	fsmap->fmh_keys[1].fmr_flags = UINT_MAX;

	*free_blks = 0;
	/* Fill the extent histogram with live data */
	while (1) {
		ret = ioctl(fd, FS_IOC_GETFSMAP, fsmap);
		if (ret < 0) {
			com_err(fs->device_name, errno, "while calling fsmap");
			free(fsmap);
			return 0;
		}

		/* No more extents to map, exit */
		if (!fsmap->fmh_entries)
			break;

		for (i = 0, extent = fsmap->fmh_recs;
		     i < fsmap->fmh_entries;
		     i++, extent++) {
			if (!(extent->fmr_flags & FMR_OF_SPECIAL_OWNER) ||
			    extent->fmr_owner != FMR_OWN_FREE)
				continue;
			update_chunk_stats(info,
					   extent->fmr_length / fs->blocksize);
			*free_blks += (extent->fmr_length / fs->blocksize);
		}

		p = &fsmap->fmh_recs[fsmap->fmh_entries - 1];
		if (p->fmr_flags & FMR_OF_LAST)
			break;
		fsmap_advance(fsmap);
	}
	free(fsmap);
	return 1;
}
#else
# define scan_online(fs, info, free_blks)	(0)
#endif /* HAVE_EXT2_IOCTLS */

static errcode_t scan_offline(ext2_filsys fs, struct chunk_info *info,
			      blk64_t *free_blks)
{
	errcode_t retval;

	*free_blks = ext2fs_free_blocks_count(fs->super);
	retval = ext2fs_read_block_bitmap(fs);
	if (retval)
		return retval;
	scan_block_bitmap(fs, info);
	return 0;
}

static errcode_t dump_chunk_info(ext2_filsys fs, struct chunk_info *info,
				 FILE *f, blk64_t free_blks)
{
	unsigned long total_chunks;
	const char *unitp = "KMGTPEZY";
	int units = 10;
	unsigned long start = 0, end;
	int i, retval = 0;

	fprintf(f, "Total blocks: %llu\nFree blocks: %llu (%0.1f%%)\n",
		(unsigned long long) ext2fs_blocks_count(fs->super),
		(unsigned long long) free_blks,
		(double)free_blks * 100 /
		ext2fs_blocks_count(fs->super));

	if (info->chunkbytes) {
		fprintf(f, "\nChunksize: %lu bytes (%u blocks)\n",
			info->chunkbytes, info->blks_in_chunk);
		total_chunks = (ext2fs_blocks_count(fs->super) +
				info->blks_in_chunk) >>
			(info->chunkbits - info->blocksize_bits);
		fprintf(f, "Total chunks: %lu\nFree chunks: %lu (%0.1f%%)\n",
			total_chunks, info->free_chunks,
			(double)info->free_chunks * 100 / total_chunks);
	}

	/* Display chunk information in KB */
	if (info->real_free_chunks) {
		unsigned int scale = fs->blocksize >> 10;
		info->min = info->min * scale;
		info->max = info->max * scale;
		info->avg = info->avg / info->real_free_chunks * scale;
	} else {
		info->min = 0;
	}

	fprintf(f, "\nMin. free extent: %lu KB \nMax. free extent: %lu KB\n"
		"Avg. free extent: %lu KB\n", info->min, info->max, info->avg);
	fprintf(f, "Num. free extent: %lu\n", info->real_free_chunks);

	fprintf(f, "\nHISTOGRAM OF FREE EXTENT SIZES:\n");
	fprintf(f, "%s :  %12s  %12s  %7s\n", "Extent Size Range",
		"Free extents", "Free Blocks", "Percent");
	for (i = 0; i < MAX_HIST; i++) {
		end = 1 << (i + info->blocksize_bits - units);
		if (info->histogram.fc_chunks[i] != 0) {
			char end_str[32];

			sprintf(end_str, "%5lu%c-", end, *unitp);
			if (i == MAX_HIST-1)
				strcpy(end_str, "max ");
			fprintf(f, "%5lu%c...%7s  :  %12lu  %12lu  %6.2f%%\n",
				start, *unitp, end_str,
				info->histogram.fc_chunks[i],
				info->histogram.fc_blocks[i],
				(double)info->histogram.fc_blocks[i] * 100 /
				free_blks);
		}
		start = end;
		if (start == 1<<10) {
			start = 1;
			units += 10;
			unitp++;
		}
	}

	return retval;
}

static void close_device(char *device_name, ext2_filsys fs)
{
	int retval = ext2fs_close_free(&fs);

	if (retval)
		com_err(device_name, retval, "while closing the filesystem.\n");
}

static void collect_info(ext2_filsys fs, struct chunk_info *chunk_info, FILE *f)
{
	unsigned int retval = 0;
	blk64_t free_blks = 0;

	fprintf(f, "Device: %s\n", fs->device_name);
	fprintf(f, "Blocksize: %u bytes\n", fs->blocksize);

	init_chunk_info(fs, chunk_info);
	if (!scan_online(fs, chunk_info, &free_blks)) {
		init_chunk_info(fs, chunk_info);
		retval = scan_offline(fs, chunk_info, &free_blks);
	}
	if (retval) {
		com_err(fs->device_name, retval, "while reading block bitmap");
		close_device(fs->device_name, fs);
		exit(1);
	}

	retval = dump_chunk_info(fs, chunk_info, f, free_blks);
	if (retval) {
		com_err(fs->device_name, retval, "while dumping chunk info");
                close_device(fs->device_name, fs);
		exit(1);
	}
}

#ifndef DEBUGFS
static void open_device(char *device_name, ext2_filsys *fs)
{
	int retval;
	int flag = EXT2_FLAG_FORCE | EXT2_FLAG_64BITS | EXT2_FLAG_THREADS;

	retval = ext2fs_open(device_name, flag, 0, 0, unix_io_manager, fs);
	if (retval) {
		com_err(device_name, retval, "while opening filesystem");
		exit(1);
	}
	(*fs)->default_bitmap_type = EXT2FS_BMAP64_RBTREE;
}
#endif

#ifdef DEBUGFS
#include "debugfs.h"

void do_freefrag(int argc, ss_argv_t argv, int sci_idx EXT2FS_ATTR((unused)),
		 void *infop EXT2FS_ATTR((unused)))
#else
int main(int argc, char *argv[])
#endif
{
	struct chunk_info chunk_info;
	ext2_filsys fs = NULL;
	char *progname;
	char *end;
	int c;

#ifdef DEBUGFS
	if (check_fs_open(argv[0]))
		return;
	reset_getopt();
#else
	char *device_name;

	add_error_table(&et_ext2_error_table);
#endif
	progname = argv[0];
	memset(&chunk_info, 0, sizeof(chunk_info));

	while ((c = getopt(argc, argv, "c:h")) != EOF) {
		switch (c) {
		case 'c':
			chunk_info.chunkbytes = strtoull(optarg, &end, 0);
			if (*end != '\0') {
				fprintf(stderr, "%s: bad chunk size '%s'\n",
					progname, optarg);
				usage(progname);
			}
			if (chunk_info.chunkbytes &
			    (chunk_info.chunkbytes - 1)) {
				fprintf(stderr, "%s: chunk size must be a "
					"power of 2.\n", argv[0]);
				usage(progname);
			}
			chunk_info.chunkbytes *= 1024;
			break;
		case 'h':
		default:
			usage(progname);
			break;
		}
	}

#ifndef DEBUGFS
	if (optind == argc) {
		fprintf(stderr, "%s: missing device name.\n", progname);
		usage(progname);
	}

	device_name = argv[optind];

	open_device(device_name, &fs);
#else
	fs = current_fs;
#endif

	if (chunk_info.chunkbytes && (chunk_info.chunkbytes < fs->blocksize)) {
		fprintf(stderr, "%s: chunksize must be greater than or equal "
			"to filesystem blocksize.\n", progname);
		exit(1);
	}
	collect_info(fs, &chunk_info, stdout);
#ifndef DEBUGFS
	close_device(device_name, fs);

	return 0;
#endif
}
Commit	Line	Data
0b2681f4 TT	1	/*
	2	* e2freefrag - report filesystem free-space fragmentation
	3	*
	4	* Copyright (C) 2009 Sun Microsystems, Inc.
	5	*
	6	* Author: Rupesh Thakare <rupesh@sun.com>
	7	* Andreas Dilger <adilger@sun.com>
	8	*
	9	* %Begin-Header%
	10	* This file may be redistributed under the terms of the GNU Public
	11	* License version 2.
	12	* %End-Header%
	13	*/
d1154eb4	14	#include "config.h"
0b2681f4 TT	15	#include <stdio.h>
	16	#ifdef HAVE_UNISTD_H
	17	#include <unistd.h>
	18	#endif
	19	#ifdef HAVE_STDLIB_H
	20	#include <stdlib.h>
	21	#endif
	22	#ifdef HAVE_GETOPT_H
	23	#include <getopt.h>
	24	#else
	25	extern char *optarg;
	26	extern int optind;
	27	#endif
82424fc6 DW	28	#if defined(HAVE_EXT2_IOCTLS) && !defined(DEBUGFS)
	29	# include <sys/ioctl.h>
	30	# include <sys/types.h>
	31	# include <sys/stat.h>
	32	# include <fcntl.h>
	33	# include <limits.h>
	34	#endif
0b2681f4 TT	35
	36	#include "ext2fs/ext2_fs.h"
	37	#include "ext2fs/ext2fs.h"
	38	#include "e2freefrag.h"
	39
82424fc6	40	#if defined(HAVE_EXT2_IOCTLS) && !defined(DEBUGFS)
5598a968	41	# if defined(HAVE_LINUX_FSMAP_H) && defined(HAVE_FSMAP_SIZEOF)
82424fc6 DW	42	# include <linux/fsmap.h>
	43	# endif
	44	# include "fsmap.h"
	45	#endif
	46
97e4f235 TT	47	#ifndef PATH_MAX
	48	#define PATH_MAX 4096
	49	#endif
	50
3cebf9c1	51	static void usage(const char *prog)
0b2681f4 TT	52	{
	53	fprintf(stderr, "usage: %s [-c chunksize in kb] [-h] "
	54	"device_name\n", prog);
5e96c572	55	#ifndef DEBUGFS
0b2681f4	56	exit(1);
5e96c572	57	#endif
0b2681f4 TT	58	}
	59
	60	static int ul_log2(unsigned long arg)
	61	{
	62	int l = 0;
	63
	64	arg >>= 1;
	65	while (arg) {
	66	l++;
	67	arg >>= 1;
	68	}
	69	return l;
	70	}
	71
3cebf9c1	72	static void init_chunk_info(ext2_filsys fs, struct chunk_info *info)
0b2681f4 TT	73	{
	74	int i;
	75
0b2681f4	76	info->blocksize_bits = ul_log2((unsigned long)fs->blocksize);
cba91c4b TT	77	if (info->chunkbytes) {
	78	info->chunkbits = ul_log2(info->chunkbytes);
	79	info->blks_in_chunk = info->chunkbytes >> info->blocksize_bits;
	80	} else {
	81	info->chunkbits = ul_log2(DEFAULT_CHUNKSIZE);
	82	info->blks_in_chunk = DEFAULT_CHUNKSIZE >> info->blocksize_bits;
	83	}
0b2681f4 TT	84
	85	info->min = ~0UL;
	86	info->max = info->avg = 0;
	87	info->real_free_chunks = 0;
	88
ad751f11 AD	89	for (i = 0; i < MAX_HIST; i++) {
	90	info->histogram.fc_chunks[i] = 0;
	91	info->histogram.fc_blocks[i] = 0;
	92	}
0b2681f4 TT	93	}
0b2681f4 TT	94
3cebf9c1 TT	95	static void update_chunk_stats(struct chunk_info *info,
3cebf9c1 TT	96	unsigned long chunk_size)
1e48a456	97	{
3cebf9c1	98	unsigned long idx;
1e48a456	99
3cebf9c1 TT	100	idx = ul_log2(chunk_size) + 1;
	101	if (idx >= MAX_HIST)
	102	idx = MAX_HIST-1;
	103	info->histogram.fc_chunks[idx]++;
	104	info->histogram.fc_blocks[idx] += chunk_size;
1e48a456 TT	105
	106	if (chunk_size > info->max)
	107	info->max = chunk_size;
	108	if (chunk_size < info->min)
	109	info->min = chunk_size;
	110	info->avg += chunk_size;
	111	info->real_free_chunks++;
	112	}
	113
3cebf9c1	114	static void scan_block_bitmap(ext2_filsys fs, struct chunk_info *info)
0b2681f4	115	{
4efbac6f	116	unsigned long long blocks_count = ext2fs_blocks_count(fs->super);
0b2681f4 TT	117	unsigned long long chunks = (blocks_count + info->blks_in_chunk) >>
	118	(info->chunkbits - info->blocksize_bits);
	119	unsigned long long chunk_num;
	120	unsigned long last_chunk_size = 0;
	121	unsigned long long chunk_start_blk = 0;
3e343b8d	122	int used;
0b2681f4 TT	123
	124	for (chunk_num = 0; chunk_num < chunks; chunk_num++) {
	125	unsigned long long blk, num_blks;
	126	int chunk_free;
	127
	128	/* Last chunk may be smaller */
	129	if (chunk_start_blk + info->blks_in_chunk > blocks_count)
	130	num_blks = blocks_count - chunk_start_blk;
	131	else
	132	num_blks = info->blks_in_chunk;
	133
	134	chunk_free = 0;
	135
	136	/* Initialize starting block for first chunk correctly else
	137	* there is a segfault when blocksize = 1024 in which case
	138	* block_map->start = 1 */
3e343b8d TT	139	for (blk = 0; blk < num_blks; blk++, chunk_start_blk++) {
	140	if (chunk_num == 0 && blk == 0) {
	141	blk = fs->super->s_first_data_block;
	142	chunk_start_blk = blk;
	143	}
3c041a51	144	used = ext2fs_fast_test_block_bitmap2(fs->block_map,
554bc091	145	chunk_start_blk >> fs->cluster_ratio_bits);
0b2681f4 TT	146	if (!used) {
	147	last_chunk_size++;
	148	chunk_free++;
	149	}
	150
	151	if (used && last_chunk_size != 0) {
1e48a456	152	update_chunk_stats(info, last_chunk_size);
0b2681f4 TT	153	last_chunk_size = 0;
	154	}
	155	}
	156
	157	if (chunk_free == info->blks_in_chunk)
	158	info->free_chunks++;
	159	}
1e48a456 TT	160	if (last_chunk_size != 0)
1e48a456 TT	161	update_chunk_stats(info, last_chunk_size);
0b2681f4 TT	162	}
0b2681f4 TT	163
82424fc6 DW	164	#if defined(HAVE_EXT2_IOCTLS) && !defined(DEBUGFS)
82424fc6 DW	165	# define FSMAP_EXTENTS 1024
93accdce DW	166	static int scan_online(ext2_filsys fs, struct chunk_info *info,
93accdce DW	167	blk64_t *free_blks)
82424fc6 DW	168	{
	169	struct fsmap_head *fsmap;
	170	struct fsmap *extent;
	171	struct fsmap *p;
	172	char mntpoint[PATH_MAX + 1];
	173	errcode_t retval;
	174	int mount_flags;
	175	int fd;
	176	int ret;
ac3256fd	177	unsigned int i;
82424fc6 DW	178
	179	/* Try to open the mountpoint for a live query. */
	180	retval = ext2fs_check_mount_point(fs->device_name, &mount_flags,
	181	mntpoint, PATH_MAX);
	182	if (retval) {
	183	com_err(fs->device_name, retval, "while checking mount status");
	184	return 0;
	185	}
ac3256fd	186	if (!(mount_flags & EXT2_MF_MOUNTED))
82424fc6 DW	187	return 0;
	188	fd = open(mntpoint, O_RDONLY);
	189	if (fd < 0) {
	190	com_err(mntpoint, errno, "while opening mount point");
	191	return 0;
	192	}
	193
	194	fsmap = malloc(fsmap_sizeof(FSMAP_EXTENTS));
	195	if (!fsmap) {
	196	com_err(fs->device_name, errno, "while allocating memory");
	197	return 0;
	198	}
	199
	200	memset(fsmap, 0, sizeof(*fsmap));
	201	fsmap->fmh_count = FSMAP_EXTENTS;
	202	fsmap->fmh_keys[1].fmr_device = UINT_MAX;
	203	fsmap->fmh_keys[1].fmr_physical = ULLONG_MAX;
	204	fsmap->fmh_keys[1].fmr_owner = ULLONG_MAX;
	205	fsmap->fmh_keys[1].fmr_offset = ULLONG_MAX;
	206	fsmap->fmh_keys[1].fmr_flags = UINT_MAX;
	207
93accdce	208	*free_blks = 0;
82424fc6 DW	209	/* Fill the extent histogram with live data */
	210	while (1) {
	211	ret = ioctl(fd, FS_IOC_GETFSMAP, fsmap);
	212	if (ret < 0) {
	213	com_err(fs->device_name, errno, "while calling fsmap");
	214	free(fsmap);
	215	return 0;
	216	}
	217
	218	/* No more extents to map, exit */
	219	if (!fsmap->fmh_entries)
	220	break;
	221
	222	for (i = 0, extent = fsmap->fmh_recs;
	223	i < fsmap->fmh_entries;
	224	i++, extent++) {
	225	if (!(extent->fmr_flags & FMR_OF_SPECIAL_OWNER) \|\|
	226	extent->fmr_owner != FMR_OWN_FREE)
	227	continue;
	228	update_chunk_stats(info,
	229	extent->fmr_length / fs->blocksize);
93accdce	230	*free_blks += (extent->fmr_length / fs->blocksize);
82424fc6 DW	231	}
	232
	233	p = &fsmap->fmh_recs[fsmap->fmh_entries - 1];
	234	if (p->fmr_flags & FMR_OF_LAST)
	235	break;
	236	fsmap_advance(fsmap);
	237	}
7d671e66	238	free(fsmap);
82424fc6 DW	239	return 1;
	240	}
	241	#else
93accdce	242	# define scan_online(fs, info, free_blks) (0)
82424fc6 DW	243	#endif /* HAVE_EXT2_IOCTLS */
82424fc6 DW	244
93accdce DW	245	static errcode_t scan_offline(ext2_filsys fs, struct chunk_info *info,
93accdce DW	246	blk64_t *free_blks)
82424fc6 DW	247	{
	248	errcode_t retval;
	249
93accdce	250	*free_blks = ext2fs_free_blocks_count(fs->super);
82424fc6 DW	251	retval = ext2fs_read_block_bitmap(fs);
	252	if (retval)
	253	return retval;
	254	scan_block_bitmap(fs, info);
	255	return 0;
	256	}
	257
	258	static errcode_t dump_chunk_info(ext2_filsys fs, struct chunk_info *info,
93accdce	259	FILE *f, blk64_t free_blks)
0b2681f4 TT	260	{
0b2681f4 TT	261	unsigned long total_chunks;
3cebf9c1	262	const char *unitp = "KMGTPEZY";
0b2681f4	263	int units = 10;
cf5301d7	264	unsigned long start = 0, end;
0b2681f4 TT	265	int i, retval = 0;
0b2681f4 TT	266
f222de72	267	fprintf(f, "Total blocks: %llu\nFree blocks: %llu (%0.1f%%)\n",
33b9a60c TT	268	(unsigned long long) ext2fs_blocks_count(fs->super),
33b9a60c TT	269	(unsigned long long) free_blks,
93accdce	270	(double)free_blks * 100 /
5e96c572	271	ext2fs_blocks_count(fs->super));
0b2681f4	272
cba91c4b	273	if (info->chunkbytes) {
5e96c572 TT	274	fprintf(f, "\nChunksize: %lu bytes (%u blocks)\n",
5e96c572 TT	275	info->chunkbytes, info->blks_in_chunk);
4efbac6f	276	total_chunks = (ext2fs_blocks_count(fs->super) +
cba91c4b TT	277	info->blks_in_chunk) >>
cba91c4b TT	278	(info->chunkbits - info->blocksize_bits);
5e96c572 TT	279	fprintf(f, "Total chunks: %lu\nFree chunks: %lu (%0.1f%%)\n",
	280	total_chunks, info->free_chunks,
	281	(double)info->free_chunks * 100 / total_chunks);
cba91c4b	282	}
0b2681f4 TT	283
	284	/* Display chunk information in KB */
	285	if (info->real_free_chunks) {
d0550895 TT	286	unsigned int scale = fs->blocksize >> 10;
	287	info->min = info->min * scale;
	288	info->max = info->max * scale;
	289	info->avg = info->avg / info->real_free_chunks * scale;
0b2681f4 TT	290	} else {
	291	info->min = 0;
	292	}
	293
5e96c572 TT	294	fprintf(f, "\nMin. free extent: %lu KB \nMax. free extent: %lu KB\n"
	295	"Avg. free extent: %lu KB\n", info->min, info->max, info->avg);
	296	fprintf(f, "Num. free extent: %lu\n", info->real_free_chunks);
0b2681f4	297
5e96c572 TT	298	fprintf(f, "\nHISTOGRAM OF FREE EXTENT SIZES:\n");
	299	fprintf(f, "%s : %12s %12s %7s\n", "Extent Size Range",
	300	"Free extents", "Free Blocks", "Percent");
0b2681f4 TT	301	for (i = 0; i < MAX_HIST; i++) {
0b2681f4 TT	302	end = 1 << (i + info->blocksize_bits - units);
1e48a456 TT	303	if (info->histogram.fc_chunks[i] != 0) {
	304	char end_str[32];
	305
	306	sprintf(end_str, "%5lu%c-", end, *unitp);
	307	if (i == MAX_HIST-1)
	308	strcpy(end_str, "max ");
5e96c572 TT	309	fprintf(f, "%5lu%c...%7s : %12lu %12lu %6.2f%%\n",
	310	start, *unitp, end_str,
	311	info->histogram.fc_chunks[i],
	312	info->histogram.fc_blocks[i],
	313	(double)info->histogram.fc_blocks[i] * 100 /
93accdce	314	free_blks);
1e48a456	315	}
0b2681f4 TT	316	start = end;
	317	if (start == 1<<10) {
	318	start = 1;
	319	units += 10;
	320	unitp++;
	321	}
	322	}
	323
	324	return retval;
	325	}
	326
3cebf9c1	327	static void close_device(char *device_name, ext2_filsys fs)
0b2681f4	328	{
47fee2ef	329	int retval = ext2fs_close_free(&fs);
0b2681f4 TT	330
	331	if (retval)
	332	com_err(device_name, retval, "while closing the filesystem.\n");
	333	}
	334
3cebf9c1	335	static void collect_info(ext2_filsys fs, struct chunk_info chunk_info, FILE f)
0b2681f4	336	{
cf5301d7	337	unsigned int retval = 0;
93accdce	338	blk64_t free_blks = 0;
0b2681f4	339
5e96c572 TT	340	fprintf(f, "Device: %s\n", fs->device_name);
5e96c572 TT	341	fprintf(f, "Blocksize: %u bytes\n", fs->blocksize);
0b2681f4	342
82424fc6	343	init_chunk_info(fs, chunk_info);
93accdce	344	if (!scan_online(fs, chunk_info, &free_blks)) {
82424fc6	345	init_chunk_info(fs, chunk_info);
93accdce	346	retval = scan_offline(fs, chunk_info, &free_blks);
82424fc6	347	}
0b2681f4 TT	348	if (retval) {
	349	com_err(fs->device_name, retval, "while reading block bitmap");
	350	close_device(fs->device_name, fs);
	351	exit(1);
	352	}
	353
93accdce	354	retval = dump_chunk_info(fs, chunk_info, f, free_blks);
0b2681f4	355	if (retval) {
82424fc6	356	com_err(fs->device_name, retval, "while dumping chunk info");
0b2681f4 TT	357	close_device(fs->device_name, fs);
	358	exit(1);
	359	}
	360	}
	361
3cebf9c1 TT	362	#ifndef DEBUGFS
3cebf9c1 TT	363	static void open_device(char device_name, ext2_filsys fs)
0b2681f4 TT	364	{
0b2681f4 TT	365	int retval;
37d56864	366	int flag = EXT2_FLAG_FORCE \| EXT2_FLAG_64BITS \| EXT2_FLAG_THREADS;
0b2681f4 TT	367
	368	retval = ext2fs_open(device_name, flag, 0, 0, unix_io_manager, fs);
	369	if (retval) {
	370	com_err(device_name, retval, "while opening filesystem");
	371	exit(1);
	372	}
ea2d3788	373	(*fs)->default_bitmap_type = EXT2FS_BMAP64_RBTREE;
0b2681f4	374	}
3cebf9c1	375	#endif
0b2681f4	376
5e96c572	377	#ifdef DEBUGFS
5e96c572 TT	378	#include "debugfs.h"
5e96c572 TT	379
49125d40	380	void do_freefrag(int argc, ss_argv_t argv, int sci_idx EXT2FS_ATTR((unused)),
2fcbcb1b	381	void *infop EXT2FS_ATTR((unused)))
5e96c572	382	#else
0b2681f4	383	int main(int argc, char *argv[])
5e96c572	384	#endif
0b2681f4	385	{
3cebf9c1	386	struct chunk_info chunk_info;
0b2681f4	387	ext2_filsys fs = NULL;
0b2681f4	388	char *progname;
b887f08f MF	389	char *end;
b887f08f MF	390	int c;
0b2681f4	391
5e96c572 TT	392	#ifdef DEBUGFS
	393	if (check_fs_open(argv[0]))
	394	return;
bf140bf2	395	reset_getopt();
5e96c572	396	#else
3cebf9c1 TT	397	char *device_name;
3cebf9c1 TT	398
137a7dc0	399	add_error_table(&et_ext2_error_table);
5e96c572	400	#endif
0b2681f4	401	progname = argv[0];
3cebf9c1	402	memset(&chunk_info, 0, sizeof(chunk_info));
0b2681f4 TT	403
	404	while ((c = getopt(argc, argv, "c:h")) != EOF) {
	405	switch (c) {
	406	case 'c':
	407	chunk_info.chunkbytes = strtoull(optarg, &end, 0);
	408	if (*end != '\0') {
	409	fprintf(stderr, "%s: bad chunk size '%s'\n",
	410	progname, optarg);
	411	usage(progname);
	412	}
	413	if (chunk_info.chunkbytes &
	414	(chunk_info.chunkbytes - 1)) {
	415	fprintf(stderr, "%s: chunk size must be a "
aff2cf8e	416	"power of 2.\n", argv[0]);
0b2681f4 TT	417	usage(progname);
	418	}
	419	chunk_info.chunkbytes *= 1024;
	420	break;
0b2681f4	421	case 'h':
4ebbc0a3	422	default:
0b2681f4 TT	423	usage(progname);
	424	break;
	425	}
	426	}
	427
5e96c572	428	#ifndef DEBUGFS
0b2681f4 TT	429	if (optind == argc) {
	430	fprintf(stderr, "%s: missing device name.\n", progname);
	431	usage(progname);
	432	}
	433
	434	device_name = argv[optind];
	435
	436	open_device(device_name, &fs);
5e96c572 TT	437	#else
	438	fs = current_fs;
	439	#endif
0b2681f4	440
cba91c4b	441	if (chunk_info.chunkbytes && (chunk_info.chunkbytes < fs->blocksize)) {
0b2681f4 TT	442	fprintf(stderr, "%s: chunksize must be greater than or equal "
	443	"to filesystem blocksize.\n", progname);
	444	exit(1);
	445	}
5e96c572 TT	446	collect_info(fs, &chunk_info, stdout);
5e96c572 TT	447	#ifndef DEBUGFS
0b2681f4 TT	448	close_device(device_name, fs);
0b2681f4 TT	449
3cebf9c1	450	return 0;
5e96c572	451	#endif
0b2681f4	452	}