lib/ext2fs/badblocks.c

   1 /*
   2  * badblocks.c --- routines to manipulate the bad block structure
   3  *
   4  * Copyright (C) 1994, 1995, 1996 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 <stdio.h>
  13 #include <string.h>
  14 #if HAVE_UNISTD_H
  15 #include <unistd.h>
  16 #endif
  17 #include <fcntl.h>
  18 #include <time.h>
  19 #if HAVE_SYS_STAT_H
  20 #include <sys/stat.h>
  21 #endif
  22 #if HAVE_SYS_TYPES_H
  23 #include <sys/types.h>
  24 #endif
  25
  26 #include <linux/ext2_fs.h>
  27
  28 #include "ext2fsP.h"
  29
  30 /*
  31  * Helper function for making a badblocks list
  32  */
  33 static errcode_t make_badblocks_list(int size, int num, blk_t *list,
  34                                      ext2_badblocks_list *ret)
  35 {
  36         ext2_badblocks_list     bb;
  37         errcode_t               retval;
  38
  39         retval = ext2fs_get_mem(sizeof(struct ext2_struct_badblocks_list),
  40                                 (void **) &bb);
  41         if (retval)
  42                 return retval;
  43         memset(bb, 0, sizeof(struct ext2_struct_badblocks_list));
  44         bb->magic = EXT2_ET_MAGIC_BADBLOCKS_LIST;
  45         bb->size = size ? size : 10;
  46         bb->num = num;
  47         retval = ext2fs_get_mem(bb->size * sizeof(blk_t), (void **) &bb->list);
  48         if (!bb->list) {
  49                 ext2fs_free_mem((void **) &bb);
  50                 return retval;
  51         }
  52         if (list)
  53                 memcpy(bb->list, list, bb->size * sizeof(blk_t));
  54         else
  55                 memset(bb->list, 0, bb->size * sizeof(blk_t));
  56         *ret = bb;
  57         return 0;
  58 }
  59
  60
  61 /*
  62  * This procedure creates an empty badblocks list.
  63  */
  64 errcode_t ext2fs_badblocks_list_create(ext2_badblocks_list *ret, int size)
  65 {
  66         return make_badblocks_list(size, 0, 0, ret);
  67 }
  68
  69 /*
  70  * This procedure copies a badblocks list
  71  */
  72 errcode_t ext2fs_badblocks_copy(ext2_badblocks_list src,
  73                                 ext2_badblocks_list *dest)
  74 {
  75         errcode_t       retval;
  76
  77         retval = make_badblocks_list(src->size, src->num, src->list,
  78                                      dest);
  79         if (retval)
  80                 return retval;
  81         (*dest)->badblocks_flags = src->badblocks_flags;
  82         return 0;
  83 }
  84
  85
  86 /*
  87  * This procedure frees a badblocks list.
  88  *
  89  * (note: moved to closefs.c)
  90  */
  91
  92
  93 /*
  94  * This procedure adds a block to a badblocks list.
  95  */
  96 errcode_t ext2fs_badblocks_list_add(ext2_badblocks_list bb, blk_t blk)
  97 {
  98         errcode_t       retval;
  99         int             i, j;
 100
 101         EXT2_CHECK_MAGIC(bb, EXT2_ET_MAGIC_BADBLOCKS_LIST);
 102
 103         if (bb->num >= bb->size) {
 104                 bb->size += 10;
 105                 retval = ext2fs_resize_mem(bb->size * sizeof(blk_t),
 106                                            (void **) &bb->list);
 107                 if (retval)
 108                         return retval;
 109         }
 110
 111         j = bb->num;
 112         for (i=0; i < bb->num; i++) {
 113                 if (bb->list[i] == blk)
 114                         return 0;
 115                 if (bb->list[i] > blk) {
 116                         j = i;
 117                         break;
 118                 }
 119         }
 120         for (i=bb->num; i > j; i--)
 121                 bb->list[i] = bb->list[i-1];
 122         bb->list[j] = blk;
 123         bb->num++;
 124         return 0;
 125 }
 126
 127 /*
 128  * This procedure tests to see if a particular block is on a badblocks
 129  * list.
 130  */
 131 int ext2fs_badblocks_list_test(ext2_badblocks_list bb, blk_t blk)
 132 {
 133         int     low, high, mid;
 134
 135         if (bb->magic != EXT2_ET_MAGIC_BADBLOCKS_LIST)
 136                 return 0;
 137
 138         if (bb->num == 0)
 139                 return 0;
 140
 141         low = 0;
 142         high = bb->num-1;
 143         if (blk == bb->list[low])
 144                 return 1;
 145         if (blk == bb->list[high])
 146                 return 1;
 147
 148         while (low < high) {
 149                 mid = (low+high)/2;
 150                 if (mid == low || mid == high)
 151                         break;
 152                 if (blk == bb->list[mid])
 153                         return 1;
 154                 if (blk < bb->list[mid])
 155                         high = mid;
 156                 else
 157                         low = mid;
 158         }
 159         return 0;
 160 }
 161
 162 errcode_t ext2fs_badblocks_list_iterate_begin(ext2_badblocks_list bb,
 163                                               ext2_badblocks_iterate *ret)
 164 {
 165         ext2_badblocks_iterate iter;
 166         errcode_t               retval;
 167
 168         EXT2_CHECK_MAGIC(bb, EXT2_ET_MAGIC_BADBLOCKS_LIST);
 169
 170         retval = ext2fs_get_mem(sizeof(struct ext2_struct_badblocks_iterate),
 171                               (void **) &iter);
 172         if (retval)
 173                 return retval;
 174
 175         iter->magic = EXT2_ET_MAGIC_BADBLOCKS_ITERATE;
 176         iter->bb = bb;
 177         iter->ptr = 0;
 178         *ret = iter;
 179         return 0;
 180 }
 181
 182 int ext2fs_badblocks_list_iterate(ext2_badblocks_iterate iter, blk_t *blk)
 183 {
 184         ext2_badblocks_list     bb;
 185
 186         if (iter->magic != EXT2_ET_MAGIC_BADBLOCKS_ITERATE)
 187                 return 0;
 188
 189         bb = iter->bb;
 190
 191         if (bb->magic != EXT2_ET_MAGIC_BADBLOCKS_LIST)
 192                 return 0;
 193
 194         if (iter->ptr < bb->num) {
 195                 *blk = bb->list[iter->ptr++];
 196                 return 1;
 197         }
 198         *blk = 0;
 199         return 0;
 200 }
 201
 202 void ext2fs_badblocks_list_iterate_end(ext2_badblocks_iterate iter)
 203 {
 204         if (!iter || (iter->magic != EXT2_ET_MAGIC_BADBLOCKS_ITERATE))
 205                 return;
 206
 207         iter->bb = 0;
 208         ext2fs_free_mem((void **) &iter);
 209 }
 210
 211
 212
 213
 214