Whamcloud - gitweb
libext2fs: force DIO alignment FreeBSD when operating on a block device
[tools/e2fsprogs.git] / lib / ext2fs / badblocks.c
index 5e8cd43..0f23983 100644 (file)
 /*
  * badblocks.c --- routines to manipulate the bad block structure
- * 
- * Copyright (C) 1994 Theodore Ts'o.  This file may be redistributed
- * under the terms of the GNU Public License.
+ *
+ * Copyright (C) 1994, 1995, 1996 Theodore Ts'o.
+ *
+ * %Begin-Header%
+ * This file may be redistributed under the terms of the GNU Library
+ * General Public License, version 2.
+ * %End-Header%
  */
 
+#include "config.h"
 #include <stdio.h>
 #include <string.h>
+#if HAVE_UNISTD_H
 #include <unistd.h>
-#include <stdlib.h>
+#endif
 #include <fcntl.h>
 #include <time.h>
+#if HAVE_SYS_STAT_H
 #include <sys/stat.h>
+#endif
+#if HAVE_SYS_TYPES_H
 #include <sys/types.h>
+#endif
 
-#include <linux/fs.h>
-#include <linux/ext2_fs.h>
-
-#include "ext2fs.h"
+#include "ext2_fs.h"
+#include "ext2fsP.h"
 
 /*
- * This procedure create an empty badblocks list.
+ * Helper function for making a badblocks list
  */
-errcode_t badblocks_list_create(badblocks_list *ret, int size)
+static errcode_t make_u32_list(int size, int num, __u32 *list,
+                              ext2_u32_list *ret)
 {
-       badblocks_list  bb;
+       ext2_u32_list   bb;
+       errcode_t       retval;
 
-       bb = malloc(sizeof(struct struct_badblocks_list));
-       if (!bb)
-               return ENOMEM;
-       memset(bb, 0, sizeof(struct struct_badblocks_list));
+       retval = ext2fs_get_mem(sizeof(struct ext2_struct_u32_list), &bb);
+       if (retval)
+               return retval;
+       memset(bb, 0, sizeof(struct ext2_struct_u32_list));
+       bb->magic = EXT2_ET_MAGIC_BADBLOCKS_LIST;
        bb->size = size ? size : 10;
-       bb->list = malloc(bb->size * sizeof(blk_t));
-       if (!bb->list) {
-               free(bb);
-               return ENOMEM;
+       bb->num = num;
+       retval = ext2fs_get_array(bb->size, sizeof(blk_t), &bb->list);
+       if (retval) {
+               ext2fs_free_mem(&bb);
+               return retval;
        }
+       if (list)
+               memcpy(bb->list, list, bb->size * sizeof(blk_t));
+       else
+               memset(bb->list, 0, bb->size * sizeof(blk_t));
        *ret = bb;
        return 0;
 }
 
+
 /*
- * This procedure frees a badblocks list.
+ * This procedure creates an empty u32 list.
+ */
+errcode_t ext2fs_u32_list_create(ext2_u32_list *ret, int size)
+{
+       return make_u32_list(size, 0, 0, ret);
+}
+
+/*
+ * This procedure creates an empty badblocks list.
+ */
+errcode_t ext2fs_badblocks_list_create(ext2_badblocks_list *ret, int size)
+{
+       return make_u32_list(size, 0, 0, (ext2_badblocks_list *) ret);
+}
+
+
+/*
+ * This procedure copies a badblocks list
  */
-void badblocks_list_free(badblocks_list bb)
+errcode_t ext2fs_u32_copy(ext2_u32_list src, ext2_u32_list *dest)
 {
-       if (bb->list)
-               free(bb->list);
-       bb->list = 0;
-       free(bb);
+       errcode_t       retval;
+
+       retval = make_u32_list(src->size, src->num, src->list, dest);
+       if (retval)
+               return retval;
+       (*dest)->badblocks_flags = src->badblocks_flags;
+       return 0;
 }
 
+errcode_t ext2fs_badblocks_copy(ext2_badblocks_list src,
+                               ext2_badblocks_list *dest)
+{
+       return ext2fs_u32_copy((ext2_u32_list) src,
+                              (ext2_u32_list *) dest);
+}
+
+/*
+ * This procedure frees a badblocks list.
+ *
+ * (note: moved to closefs.c)
+ */
+
+
 /*
  * This procedure adds a block to a badblocks list.
  */
-errcode_t badblocks_list_add(badblocks_list bb, blk_t blk)
+errcode_t ext2fs_u32_list_add(ext2_u32_list bb, __u32 blk)
 {
-       int     i;
+       errcode_t       retval;
+       int             i, j;
+       unsigned long   old_size;
 
-       for (i=0; i < bb->num; i++)
-               if (bb->list[i] == blk)
-                       return 0;
+       EXT2_CHECK_MAGIC(bb, EXT2_ET_MAGIC_BADBLOCKS_LIST);
 
        if (bb->num >= bb->size) {
-               bb->size += 10;
-               bb->list = realloc(bb->list, bb->size * sizeof(blk_t));
-               if (!bb->list) {
-                       bb->size = 0;
-                       bb->num = 0;
-                       return ENOMEM;
+               old_size = bb->size * sizeof(__u32);
+               bb->size += 100;
+               retval = ext2fs_resize_mem(old_size, bb->size * sizeof(__u32),
+                                          &bb->list);
+               if (retval) {
+                       bb->size -= 100;
+                       return retval;
                }
        }
 
-       bb->list[bb->num++] = blk;
+       /*
+        * Add special case code for appending to the end of the list
+        */
+       i = bb->num-1;
+       if ((bb->num != 0) && (bb->list[i] == blk))
+               return 0;
+       if ((bb->num == 0) || (bb->list[i] < blk)) {
+               bb->list[bb->num++] = blk;
+               return 0;
+       }
+
+       j = bb->num;
+       for (i=0; i < bb->num; i++) {
+               if (bb->list[i] == blk)
+                       return 0;
+               if (bb->list[i] > blk) {
+                       j = i;
+                       break;
+               }
+       }
+       for (i=bb->num; i > j; i--)
+               bb->list[i] = bb->list[i-1];
+       bb->list[j] = blk;
+       bb->num++;
        return 0;
 }
 
+errcode_t ext2fs_badblocks_list_add(ext2_badblocks_list bb, blk_t blk)
+{
+       return ext2fs_u32_list_add((ext2_u32_list) bb, (__u32) blk);
+}
+
+/*
+ * This procedure finds a particular block is on a badblocks
+ * list.
+ */
+int ext2fs_u32_list_find(ext2_u32_list bb, __u32 blk)
+{
+       int     low, high, mid;
+
+       if (bb->magic != EXT2_ET_MAGIC_BADBLOCKS_LIST)
+               return -1;
+
+       if (bb->num == 0)
+               return -1;
+
+       low = 0;
+       high = bb->num-1;
+       if (blk == bb->list[low])
+               return low;
+       if (blk == bb->list[high])
+               return high;
+
+       while (low < high) {
+               mid = ((unsigned)low + (unsigned)high)/2;
+               if (mid == low || mid == high)
+                       break;
+               if (blk == bb->list[mid])
+                       return mid;
+               if (blk < bb->list[mid])
+                       high = mid;
+               else
+                       low = mid;
+       }
+       return -1;
+}
+
 /*
  * This procedure tests to see if a particular block is on a badblocks
  * list.
  */
-int badblocks_list_test(badblocks_list bb, blk_t blk)
+int ext2fs_u32_list_test(ext2_u32_list bb, __u32 blk)
 {
-       int     i;
+       if (ext2fs_u32_list_find(bb, blk) < 0)
+               return 0;
+       else
+               return 1;
+}
 
-       for (i=0; i < bb->num; i++)
-               if (bb->list[i] == blk)
-                       return 1;
+int ext2fs_badblocks_list_test(ext2_badblocks_list bb, blk_t blk)
+{
+       return ext2fs_u32_list_test((ext2_u32_list) bb, (__u32) blk);
+}
+
+
+/*
+ * Remove a block from the badblock list
+ */
+int ext2fs_u32_list_del(ext2_u32_list bb, __u32 blk)
+{
+       int     remloc, i;
+
+       if (bb->num == 0)
+               return -1;
+
+       remloc = ext2fs_u32_list_find(bb, blk);
+       if (remloc < 0)
+               return -1;
 
+       for (i = remloc ; i < bb->num-1; i++)
+               bb->list[i] = bb->list[i+1];
+       bb->num--;
        return 0;
 }
 
-errcode_t badblocks_list_iterate_begin(badblocks_list bb,
-                                      badblocks_iterate *ret)
+void ext2fs_badblocks_list_del(ext2_u32_list bb, __u32 blk)
+{
+       ext2fs_u32_list_del(bb, blk);
+}
+
+errcode_t ext2fs_u32_list_iterate_begin(ext2_u32_list bb,
+                                       ext2_u32_iterate *ret)
 {
-       badblocks_iterate iter;
+       ext2_u32_iterate iter;
+       errcode_t               retval;
 
-       iter = malloc(sizeof(struct struct_badblocks_iterate));
-       if (!iter)
-               return ENOMEM;
+       EXT2_CHECK_MAGIC(bb, EXT2_ET_MAGIC_BADBLOCKS_LIST);
 
+       retval = ext2fs_get_mem(sizeof(struct ext2_struct_u32_iterate), &iter);
+       if (retval)
+               return retval;
+
+       iter->magic = EXT2_ET_MAGIC_BADBLOCKS_ITERATE;
        iter->bb = bb;
        iter->ptr = 0;
        *ret = iter;
        return 0;
 }
 
-int badblocks_list_iterate(badblocks_iterate iter, blk_t *blk)
+errcode_t ext2fs_badblocks_list_iterate_begin(ext2_badblocks_list bb,
+                                             ext2_badblocks_iterate *ret)
+{
+       return ext2fs_u32_list_iterate_begin((ext2_u32_list) bb,
+                                             (ext2_u32_iterate *) ret);
+}
+
+
+int ext2fs_u32_list_iterate(ext2_u32_iterate iter, __u32 *blk)
 {
-       badblocks_list  bb = iter->bb;
-       
+       ext2_u32_list   bb;
+
+       if (iter->magic != EXT2_ET_MAGIC_BADBLOCKS_ITERATE)
+               return 0;
+
+       bb = iter->bb;
+
+       if (bb->magic != EXT2_ET_MAGIC_BADBLOCKS_LIST)
+               return 0;
+
        if (iter->ptr < bb->num) {
                *blk = bb->list[iter->ptr++];
                return 1;
-       } 
+       }
        *blk = 0;
        return 0;
 }
 
-void badblocks_list_iterate_end(badblocks_iterate iter)
+int ext2fs_badblocks_list_iterate(ext2_badblocks_iterate iter, blk_t *blk)
 {
+       return ext2fs_u32_list_iterate((ext2_u32_iterate) iter,
+                                      (__u32 *) blk);
+}
+
+
+void ext2fs_u32_list_iterate_end(ext2_u32_iterate iter)
+{
+       if (!iter || (iter->magic != EXT2_ET_MAGIC_BADBLOCKS_ITERATE))
+               return;
+
        iter->bb = 0;
-       free(iter);
+       ext2fs_free_mem(&iter);
 }
 
+void ext2fs_badblocks_list_iterate_end(ext2_badblocks_iterate iter)
+{
+       ext2fs_u32_list_iterate_end((ext2_u32_iterate) iter);
+}
+
+
+int ext2fs_u32_list_equal(ext2_u32_list bb1, ext2_u32_list bb2)
+{
+       EXT2_CHECK_MAGIC(bb1, EXT2_ET_MAGIC_BADBLOCKS_LIST);
+       EXT2_CHECK_MAGIC(bb2, EXT2_ET_MAGIC_BADBLOCKS_LIST);
+
+       if (bb1->num != bb2->num)
+               return 0;
 
+       if (memcmp(bb1->list, bb2->list, bb1->num * sizeof(blk_t)) != 0)
+               return 0;
+       return 1;
+}
 
+int ext2fs_badblocks_equal(ext2_badblocks_list bb1, ext2_badblocks_list bb2)
+{
+       return ext2fs_u32_list_equal((ext2_u32_list) bb1,
+                                    (ext2_u32_list) bb2);
+}
 
-               
+int ext2fs_u32_list_count(ext2_u32_list bb)
+{
+       return bb->num;
+}