Whamcloud - gitweb
libext2fs: use strcpy()/strcat() instead of sprintf() in bmap functions
[tools/e2fsprogs.git] / lib / ext2fs / gen_bitmap64.c
index 9dcca03..dd2773a 100644 (file)
@@ -25,6 +25,9 @@
 #if HAVE_SYS_TYPES_H
 #include <sys/types.h>
 #endif
+#ifdef HAVE_SYS_TIME_H
+#include <sys/time.h>
+#endif
 
 #include "ext2_fs.h"
 #include "ext2fsP.h"
@@ -77,6 +80,12 @@ static void warn_bitmap(ext2fs_generic_bitmap bitmap,
 #endif
 }
 
+#ifdef ENABLE_BMAP_STATS_OPS
+#define INC_STAT(map, name) map->stats.name
+#else
+#define INC_STAT(map, name) ;;
+#endif
+
 
 errcode_t ext2fs_alloc_generic_bmap(ext2_filsys fs, errcode_t magic,
                                    int type, __u64 start, __u64 end,
@@ -110,11 +119,20 @@ errcode_t ext2fs_alloc_generic_bmap(ext2_filsys fs, errcode_t magic,
                return EINVAL;
        }
 
-       retval = ext2fs_get_mem(sizeof(struct ext2fs_struct_generic_bitmap),
-                               &bitmap);
+       retval = ext2fs_get_memzero(sizeof(struct ext2fs_struct_generic_bitmap),
+                                   &bitmap);
        if (retval)
                return retval;
 
+#ifdef ENABLE_BMAP_STATS
+       if (gettimeofday(&bitmap->stats.created,
+                        (struct timezone *) NULL) == -1) {
+               perror("gettimeofday");
+               return 1;
+       }
+       bitmap->stats.type = type;
+#endif
+
        /* XXX factor out, repeated in copy_bmap */
        bitmap->magic = magic;
        bitmap->fs = fs;
@@ -155,6 +173,73 @@ errcode_t ext2fs_alloc_generic_bmap(ext2_filsys fs, errcode_t magic,
        return 0;
 }
 
+#ifdef ENABLE_BMAP_STATS
+void ext2fs_print_bmap_statistics(ext2fs_generic_bitmap bitmap)
+{
+       struct ext2_bmap_statistics *stats = &bitmap->stats;
+#ifdef ENABLE_BMAP_STATS_OPS
+       float mark_seq_perc = 0.0, test_seq_perc = 0.0;
+       float mark_back_perc = 0.0, test_back_perc = 0.0;
+#endif
+       double inuse;
+       struct timeval now;
+
+#ifdef ENABLE_BMAP_STATS_OPS
+       if (stats->test_count) {
+               test_seq_perc = ((float)stats->test_seq /
+                                stats->test_count) * 100;
+               test_back_perc = ((float)stats->test_back /
+                                 stats->test_count) * 100;
+       }
+
+       if (stats->mark_count) {
+               mark_seq_perc = ((float)stats->mark_seq /
+                                stats->mark_count) * 100;
+               mark_back_perc = ((float)stats->mark_back /
+                                 stats->mark_count) * 100;
+       }
+#endif
+
+       if (gettimeofday(&now, (struct timezone *) NULL) == -1) {
+               perror("gettimeofday");
+               return;
+       }
+
+       inuse = (double) now.tv_sec + \
+               (((double) now.tv_usec) * 0.000001);
+       inuse -= (double) stats->created.tv_sec + \
+               (((double) stats->created.tv_usec) * 0.000001);
+
+       fprintf(stderr, "\n[+] %s bitmap (type %d)\n", bitmap->description,
+               stats->type);
+       fprintf(stderr, "=================================================\n");
+#ifdef ENABLE_BMAP_STATS_OPS
+       fprintf(stderr, "%16llu bits long\n",
+               bitmap->real_end - bitmap->start);
+       fprintf(stderr, "%16lu copy_bmap\n%16lu resize_bmap\n",
+               stats->copy_count, stats->resize_count);
+       fprintf(stderr, "%16lu mark bmap\n%16lu unmark_bmap\n",
+               stats->mark_count, stats->unmark_count);
+       fprintf(stderr, "%16lu test_bmap\n%16lu mark_bmap_extent\n",
+               stats->test_count, stats->mark_ext_count);
+       fprintf(stderr, "%16lu unmark_bmap_extent\n"
+               "%16lu test_clear_bmap_extent\n",
+               stats->unmark_ext_count, stats->test_ext_count);
+       fprintf(stderr, "%16lu set_bmap_range\n%16lu set_bmap_range\n",
+               stats->set_range_count, stats->get_range_count);
+       fprintf(stderr, "%16lu clear_bmap\n%16lu contiguous bit test (%.2f%%)\n",
+               stats->clear_count, stats->test_seq, test_seq_perc);
+       fprintf(stderr, "%16lu contiguous bit mark (%.2f%%)\n"
+               "%16llu bits tested backwards (%.2f%%)\n",
+               stats->mark_seq, mark_seq_perc,
+               stats->test_back, test_back_perc);
+       fprintf(stderr, "%16llu bits marked backwards (%.2f%%)\n"
+               "%16.2f seconds in use\n",
+               stats->mark_back, mark_back_perc, inuse);
+#endif /* ENABLE_BMAP_STATS_OPS */
+}
+#endif
+
 void ext2fs_free_generic_bmap(ext2fs_generic_bitmap bmap)
 {
        if (!bmap)
@@ -168,6 +253,13 @@ void ext2fs_free_generic_bmap(ext2fs_generic_bitmap bmap)
        if (!EXT2FS_IS_64_BITMAP(bmap))
                return;
 
+#ifdef ENABLE_BMAP_STATS
+       if (getenv("E2FSPROGS_BITMAP_STATS")) {
+               ext2fs_print_bmap_statistics(bmap);
+               bmap->bitmap_ops->print_stats(bmap);
+       }
+#endif
+
        bmap->bitmap_ops->free_bmap(bmap);
 
        if (bmap->description) {
@@ -195,11 +287,24 @@ errcode_t ext2fs_copy_generic_bmap(ext2fs_generic_bitmap src,
                return EINVAL;
 
        /* Allocate a new bitmap struct */
-       retval = ext2fs_get_mem(sizeof(struct ext2fs_struct_generic_bitmap),
-                               &new_bmap);
+       retval = ext2fs_get_memzero(sizeof(struct ext2fs_struct_generic_bitmap),
+                                   &new_bmap);
        if (retval)
                return retval;
 
+
+#ifdef ENABLE_BMAP_STATS_OPS
+       src->stats.copy_count++;
+#endif
+#ifdef ENABLE_BMAP_STATS
+       if (gettimeofday(&new_bmap->stats.created,
+                        (struct timezone *) NULL) == -1) {
+               perror("gettimeofday");
+               return 1;
+       }
+       new_bmap->stats.type = src->stats.type;
+#endif
+
        /* Copy all the high-level parts over */
        new_bmap->magic = src->magic;
        new_bmap->fs = src->fs;
@@ -217,7 +322,8 @@ errcode_t ext2fs_copy_generic_bmap(ext2fs_generic_bitmap src,
                        ext2fs_free_mem(&new_bmap);
                        return retval;
                }
-               sprintf(new_descr, "copy of %s", descr);
+               strcpy(new_descr, "copy of ");
+               strcat(new_descr, descr);
                new_bmap->description = new_descr;
        }
 
@@ -247,6 +353,8 @@ errcode_t ext2fs_resize_generic_bmap(ext2fs_generic_bitmap bmap,
        if (!EXT2FS_IS_64_BITMAP(bmap))
                return EINVAL;
 
+       INC_STAT(bmap, resize_count);
+
        return bmap->bitmap_ops->resize_bmap(bmap, new_end, new_real_end);
 }
 
@@ -335,6 +443,15 @@ int ext2fs_mark_generic_bmap(ext2fs_generic_bitmap bitmap,
 
        arg >>= bitmap->cluster_bits;
 
+#ifdef ENABLE_BMAP_STATS_OPS
+       if (arg == bitmap->stats.last_marked + 1)
+               bitmap->stats.mark_seq++;
+       if (arg < bitmap->stats.last_marked)
+               bitmap->stats.mark_back++;
+       bitmap->stats.last_marked = arg;
+       bitmap->stats.mark_count++;
+#endif
+
        if ((arg < bitmap->start) || (arg > bitmap->end)) {
                warn_bitmap(bitmap, EXT2FS_MARK_ERROR, arg);
                return 0;
@@ -363,6 +480,8 @@ int ext2fs_unmark_generic_bmap(ext2fs_generic_bitmap bitmap,
 
        arg >>= bitmap->cluster_bits;
 
+       INC_STAT(bitmap, unmark_count);
+
        if ((arg < bitmap->start) || (arg > bitmap->end)) {
                warn_bitmap(bitmap, EXT2FS_UNMARK_ERROR, arg);
                return 0;
@@ -391,6 +510,15 @@ int ext2fs_test_generic_bmap(ext2fs_generic_bitmap bitmap,
 
        arg >>= bitmap->cluster_bits;
 
+#ifdef ENABLE_BMAP_STATS_OPS
+       bitmap->stats.test_count++;
+       if (arg == bitmap->stats.last_tested + 1)
+               bitmap->stats.test_seq++;
+       if (arg < bitmap->stats.last_tested)
+               bitmap->stats.test_back++;
+       bitmap->stats.last_tested = arg;
+#endif
+
        if ((arg < bitmap->start) || (arg > bitmap->end)) {
                warn_bitmap(bitmap, EXT2FS_TEST_ERROR, arg);
                return 0;
@@ -419,6 +547,8 @@ errcode_t ext2fs_set_generic_bmap_range(ext2fs_generic_bitmap bmap,
        if (!EXT2FS_IS_64_BITMAP(bmap))
                return EINVAL;
 
+       INC_STAT(bmap, set_range_count);
+
        return bmap->bitmap_ops->set_bmap_range(bmap, start, num, in);
 }
 
@@ -442,6 +572,8 @@ errcode_t ext2fs_get_generic_bmap_range(ext2fs_generic_bitmap bmap,
        if (!EXT2FS_IS_64_BITMAP(bmap))
                return EINVAL;
 
+       INC_STAT(bmap, get_range_count);
+
        return bmap->bitmap_ops->get_bmap_range(bmap, start, num, out);
 }
 
@@ -513,6 +645,8 @@ int ext2fs_test_block_bitmap_range2(ext2fs_block_bitmap bmap,
        if (!EXT2FS_IS_64_BITMAP(bmap))
                return EINVAL;
 
+       INC_STAT(bmap, test_ext_count);
+
        return bmap->bitmap_ops->test_clear_bmap_extent(bmap, block, num);
 }
 
@@ -535,6 +669,8 @@ void ext2fs_mark_block_bitmap_range2(ext2fs_block_bitmap bmap,
        if (!EXT2FS_IS_64_BITMAP(bmap))
                return;
 
+       INC_STAT(bmap, mark_ext_count);
+
        if ((block < bmap->start) || (block+num-1 > bmap->end)) {
                ext2fs_warn_bitmap(EXT2_ET_BAD_BLOCK_MARK, block,
                                   bmap->description);
@@ -563,6 +699,8 @@ void ext2fs_unmark_block_bitmap_range2(ext2fs_block_bitmap bmap,
        if (!EXT2FS_IS_64_BITMAP(bmap))
                return;
 
+       INC_STAT(bmap, unmark_ext_count);
+
        if ((block < bmap->start) || (block+num-1 > bmap->end)) {
                ext2fs_warn_bitmap(EXT2_ET_BAD_BLOCK_UNMARK, block,
                                   bmap->description);
@@ -627,3 +765,55 @@ errcode_t ext2fs_convert_subcluster_bitmap(ext2_filsys fs,
        *bitmap = cmap;
        return 0;
 }
+
+errcode_t ext2fs_find_first_zero_generic_bmap(ext2fs_generic_bitmap bitmap,
+                                             __u64 start, __u64 end, __u64 *out)
+{
+       int b;
+
+       if (!bitmap)
+               return EINVAL;
+
+       if (EXT2FS_IS_64_BITMAP(bitmap) && bitmap->bitmap_ops->find_first_zero)
+               return bitmap->bitmap_ops->find_first_zero(bitmap, start,
+                                                          end, out);
+
+       if (EXT2FS_IS_32_BITMAP(bitmap)) {
+               blk_t blk = 0;
+               errcode_t retval;
+
+               if (((start) & ~0xffffffffULL) ||
+                   ((end) & ~0xffffffffULL)) {
+                       ext2fs_warn_bitmap2(bitmap, EXT2FS_TEST_ERROR, start);
+                       return EINVAL;
+               }
+
+               retval = ext2fs_find_first_zero_generic_bitmap(bitmap, start,
+                                                              end, &blk);
+               if (retval == 0)
+                       *out = blk;
+               return retval;
+       }
+
+       if (!EXT2FS_IS_64_BITMAP(bitmap))
+               return EINVAL;
+
+       start >>= bitmap->cluster_bits;
+       end >>= bitmap->cluster_bits;
+
+       if (start < bitmap->start || end > bitmap->end || start > end) {
+               warn_bitmap(bitmap, EXT2FS_TEST_ERROR, start);
+               return EINVAL;
+       }
+
+       while (start <= end) {
+               b = bitmap->bitmap_ops->test_bmap(bitmap, start);
+               if (!b) {
+                       *out = start << bitmap->cluster_bits;
+                       return 0;
+               }
+               start++;
+       }
+
+       return ENOENT;
+}