Whamcloud - gitweb
ext2fs: check if Lustre filesystem is mounted
[tools/e2fsprogs.git] / misc / e2freefrag.c
index df41853..268fac9 100644 (file)
@@ -11,6 +11,7 @@
  * License version 2.
  * %End-Header%
  */
+#include "config.h"
 #include <stdio.h>
 #ifdef HAVE_UNISTD_H
 #include <unistd.h>
 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"
 
-void usage(const char *prog)
+#if defined(HAVE_EXT2_IOCTLS) && !defined(DEBUGFS)
+# ifdef HAVE_LINUX_FSMAP_H
+#  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)
@@ -48,13 +69,18 @@ static int ul_log2(unsigned long arg)
         return l;
 }
 
-void init_chunk_info(ext2_filsys fs, struct chunk_info *info)
+static void init_chunk_info(ext2_filsys fs, struct chunk_info *info)
 {
        int i;
 
-       info->chunkbits = ul_log2(info->chunkbytes);
        info->blocksize_bits = ul_log2((unsigned long)fs->blocksize);
-       info->blks_in_chunk = info->chunkbytes >> info->blocksize_bits;
+       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;
@@ -66,14 +92,34 @@ void init_chunk_info(ext2_filsys fs, struct chunk_info *info)
        }
 }
 
-void scan_block_bitmap(ext2_filsys fs, struct chunk_info *info)
+static void update_chunk_stats(struct chunk_info *info,
+                              unsigned long chunk_size)
 {
-       unsigned long long blocks_count = fs->super->s_blocks_count;
+       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;
@@ -90,30 +136,20 @@ void scan_block_bitmap(ext2_filsys fs, struct chunk_info *info)
                /* Initialize starting block for first chunk correctly else
                 * there is a segfault when blocksize = 1024 in which case
                 * block_map->start = 1 */
-               for (blk = (chunk_num == 0 ? fs->super->s_first_data_block : 0);
-                    blk < num_blks; blk++, chunk_start_blk++) {
-                       int used = ext2fs_fast_test_block_bitmap(fs->block_map,
-                                                              chunk_start_blk);
+               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) {
-                               unsigned long index;
-
-                               index = ul_log2(last_chunk_size) + 1;
-                               info->histogram.fc_chunks[index]++;
-                               info->histogram.fc_blocks[index] +=
-                                                       last_chunk_size;
-
-                               if (last_chunk_size > info->max)
-                                       info->max = last_chunk_size;
-                               if (last_chunk_size < info->min)
-                                       info->min = last_chunk_size;
-                               info->avg += last_chunk_size;
-
-                               info->real_free_chunks++;
+                               update_chunk_stats(info, last_chunk_size);
                                last_chunk_size = 0;
                        }
                }
@@ -121,56 +157,157 @@ void scan_block_bitmap(ext2_filsys fs, struct chunk_info *info)
                if (chunk_free == info->blks_in_chunk)
                        info->free_chunks++;
        }
+       if (last_chunk_size != 0)
+               update_chunk_stats(info, last_chunk_size);
 }
 
-errcode_t get_chunk_info(ext2_filsys fs, struct chunk_info *info)
+#if defined(HAVE_EXT2_IOCTLS) && !defined(DEBUGFS)
+# define FSMAP_EXTENTS 1024
+static int scan_online(ext2_filsys fs, struct chunk_info *info)
 {
-       unsigned long total_chunks;
-       char *unitp = "KMGTPEZY";
-       int units = 10;
-       unsigned long start = 0, end, cum;
-       int i, retval = 0;
+       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;
+
+       /* 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);
+               }
+
+               p = &fsmap->fmh_recs[fsmap->fmh_entries - 1];
+               if (p->fmr_flags & FMR_OF_LAST)
+                       break;
+               fsmap_advance(fsmap);
+       }
+
+       return 1;
+}
+#else
+# define scan_online(fs, info) (0)
+#endif /* HAVE_EXT2_IOCTLS */
+
+static errcode_t scan_offline(ext2_filsys fs, struct chunk_info *info)
+{
+       errcode_t retval;
+
+       retval = ext2fs_read_block_bitmap(fs);
+       if (retval)
+               return retval;
        scan_block_bitmap(fs, info);
+       return 0;
+}
 
-       printf("Total blocks: %u\nFree blocks: %u (%0.1f%%)\n",
-              fs->super->s_blocks_count, fs->super->s_free_blocks_count,
-              (double)fs->super->s_free_blocks_count * 100 /
-                                               fs->super->s_blocks_count);
+static errcode_t dump_chunk_info(ext2_filsys fs, struct chunk_info *info,
+                                FILE *f)
+{
+       unsigned long total_chunks;
+       const char *unitp = "KMGTPEZY";
+       int units = 10;
+       unsigned long start = 0, end;
+       int i, retval = 0;
 
-       printf("\nChunksize: %lu bytes (%u blocks)\n",
-              info->chunkbytes, info->blks_in_chunk);
-       total_chunks = (fs->super->s_blocks_count + info->blks_in_chunk) >>
-                               (info->chunkbits - info->blocksize_bits);
-       printf("Total chunks: %lu\nFree chunks: %lu (%0.1f%%)\n",
-              total_chunks, info->free_chunks,
-              (double)info->free_chunks * 100 / total_chunks);
+       fprintf(f, "Total blocks: %llu\nFree blocks: %llu (%0.1f%%)\n",
+               ext2fs_blocks_count(fs->super),
+               ext2fs_free_blocks_count(fs->super),
+               (double)ext2fs_free_blocks_count(fs->super) * 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) {
-               info->min = (info->min * fs->blocksize) >> 10;
-               info->max = (info->max * fs->blocksize) >> 10;
-               info->avg = (info->avg / info->real_free_chunks *
-                            fs->blocksize) >> 10;
+               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;
        }
 
-       printf("\nMin free chunk: %lu KB \nMax free chunk: %lu KB\n"
-              "Avg free chunk: %lu KB\n", info->min, info->max, info->avg);
+       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);
 
-       printf("\nHISTOGRAM OF FREE CHUNK SIZES:\n");
-       printf("%s :  %12s  %12s  %7s\n", "Chunk Size Range", "Free chunks",
-              "Free Blocks", "Percent");
+       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)
-                       printf("%5lu%c...%5lu%c- :  %12lu  %12lu  %6.2f%%\n",
-                              start, *unitp, end, *unitp,
-                              info->histogram.fc_chunks[i],
-                              info->histogram.fc_blocks[i],
-                              (double)info->histogram.fc_blocks[i] * 100 /
-                              fs->super->s_free_blocks_count);
+               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 /
+                               ext2fs_free_blocks_count(fs->super));
+               }
                start = end;
                if (start == 1<<10) {
                        start = 1;
@@ -182,61 +319,80 @@ errcode_t get_chunk_info(ext2_filsys fs, struct chunk_info *info)
        return retval;
 }
 
-void close_device(char *device_name, ext2_filsys fs)
+static void close_device(char *device_name, ext2_filsys fs)
 {
-       int retval = ext2fs_close(fs);
+       int retval = ext2fs_close_free(&fs);
 
        if (retval)
                com_err(device_name, retval, "while closing the filesystem.\n");
 }
 
-void collect_info(ext2_filsys fs, struct chunk_info *chunk_info)
+static void collect_info(ext2_filsys fs, struct chunk_info *chunk_info, FILE *f)
 {
-       unsigned int retval = 0, i, free_blks;
+       unsigned int retval = 0;
 
-       printf("Device: %s\n", fs->device_name);
-       printf("Blocksize: %u bytes\n", fs->blocksize);
+       fprintf(f, "Device: %s\n", fs->device_name);
+       fprintf(f, "Blocksize: %u bytes\n", fs->blocksize);
 
-       retval = ext2fs_read_block_bitmap(fs);
+       init_chunk_info(fs, chunk_info);
+       if (!scan_online(fs, chunk_info)) {
+               init_chunk_info(fs, chunk_info);
+               retval = scan_offline(fs, chunk_info);
+       }
        if (retval) {
                com_err(fs->device_name, retval, "while reading block bitmap");
                close_device(fs->device_name, fs);
                exit(1);
        }
 
-       init_chunk_info(fs, chunk_info);
-
-       retval = get_chunk_info(fs, chunk_info);
+       retval = dump_chunk_info(fs, chunk_info, f);
        if (retval) {
-               com_err(fs->device_name, retval, "while collecting chunk info");
+               com_err(fs->device_name, retval, "while dumping chunk info");
                 close_device(fs->device_name, fs);
                exit(1);
        }
 }
 
-void open_device(char *device_name, ext2_filsys *fs)
+#ifndef DEBUGFS
+static void open_device(char *device_name, ext2_filsys *fs)
 {
        int retval;
-       int flag = EXT2_FLAG_FORCE;
+       int flag = EXT2_FLAG_FORCE | EXT2_FLAG_64BITS;
 
        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, char **argv)
+#else
 int main(int argc, char *argv[])
+#endif
 {
-       struct chunk_info chunk_info = { .chunkbytes = DEFAULT_CHUNKSIZE };
-       errcode_t retval = 0;
+       struct chunk_info chunk_info;
        ext2_filsys fs = NULL;
-       char *device_name;
        char *progname;
-       char c, *end;
+       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) {
@@ -250,20 +406,19 @@ int main(int argc, char *argv[])
                        if (chunk_info.chunkbytes &
                            (chunk_info.chunkbytes - 1)) {
                                fprintf(stderr, "%s: chunk size must be a "
-                                       "power of 2.", argv[0]);
+                                       "power of 2.\n", argv[0]);
                                usage(progname);
                        }
                        chunk_info.chunkbytes *= 1024;
                        break;
-               default:
-                       fprintf(stderr, "%s: bad option '%c'\n",
-                               progname, c);
                case 'h':
+               default:
                        usage(progname);
                        break;
                }
        }
 
+#ifndef DEBUGFS
        if (optind == argc) {
                fprintf(stderr, "%s: missing device name.\n", progname);
                usage(progname);
@@ -272,14 +427,19 @@ int main(int argc, char *argv[])
        device_name = argv[optind];
 
        open_device(device_name, &fs);
+#else
+       fs = current_fs;
+#endif
 
-       if (chunk_info.chunkbytes < fs->blocksize) {
+       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);
+       collect_info(fs, &chunk_info, stdout);
+#ifndef DEBUGFS
        close_device(device_name, fs);
 
-       return retval;
+       return 0;
+#endif
 }