Home | History | Annotate | Download | only in ext2fs
      1 /*
      2  * brel_ma.c
      3  *
      4  * Copyright (C) 1996, 1997 Theodore Ts'o.
      5  *
      6  * TODO: rewrite to not use a direct array!!!  (Fortunately this
      7  * module isn't really used yet.)
      8  *
      9  * %Begin-Header%
     10  * This file may be redistributed under the terms of the GNU Library
     11  * General Public License, version 2.
     12  * %End-Header%
     13  */
     14 
     15 #include <fcntl.h>
     16 #include <stdio.h>
     17 #include <string.h>
     18 #if HAVE_UNISTD_H
     19 #include <unistd.h>
     20 #endif
     21 #if HAVE_ERRNO_H
     22 #include <errno.h>
     23 #endif
     24 
     25 #include "ext2_fs.h"
     26 #include "ext2fs.h"
     27 #include "brel.h"
     28 
     29 static errcode_t bma_put(ext2_brel brel, blk64_t old,
     30 			struct ext2_block_relocate_entry *ent);
     31 static errcode_t bma_get(ext2_brel brel, blk64_t old,
     32 			struct ext2_block_relocate_entry *ent);
     33 static errcode_t bma_start_iter(ext2_brel brel);
     34 static errcode_t bma_next(ext2_brel brel, blk64_t *old,
     35 			 struct ext2_block_relocate_entry *ent);
     36 static errcode_t bma_move(ext2_brel brel, blk64_t old, blk64_t new);
     37 static errcode_t bma_delete(ext2_brel brel, blk64_t old);
     38 static errcode_t bma_free(ext2_brel brel);
     39 
     40 struct brel_ma {
     41 	__u32 magic;
     42 	blk64_t max_block;
     43 	struct ext2_block_relocate_entry *entries;
     44 };
     45 
     46 errcode_t ext2fs_brel_memarray_create(char *name, blk64_t max_block,
     47 				      ext2_brel *new_brel)
     48 {
     49 	ext2_brel		brel = 0;
     50 	errcode_t	retval;
     51 	struct brel_ma 	*ma = 0;
     52 	size_t		size;
     53 
     54 	*new_brel = 0;
     55 
     56 	/*
     57 	 * Allocate memory structures
     58 	 */
     59 	retval = ext2fs_get_mem(sizeof(struct ext2_block_relocation_table),
     60 				&brel);
     61 	if (retval)
     62 		goto errout;
     63 	memset(brel, 0, sizeof(struct ext2_block_relocation_table));
     64 
     65 	retval = ext2fs_get_mem(strlen(name)+1, &brel->name);
     66 	if (retval)
     67 		goto errout;
     68 	strcpy(brel->name, name);
     69 
     70 	retval = ext2fs_get_mem(sizeof(struct brel_ma), &ma);
     71 	if (retval)
     72 		goto errout;
     73 	memset(ma, 0, sizeof(struct brel_ma));
     74 	brel->priv_data = ma;
     75 
     76 	size = (size_t) (sizeof(struct ext2_block_relocate_entry) *
     77 			 (max_block+1));
     78 	retval = ext2fs_get_array(max_block+1,
     79 		sizeof(struct ext2_block_relocate_entry), &ma->entries);
     80 	if (retval)
     81 		goto errout;
     82 	memset(ma->entries, 0, size);
     83 	ma->max_block = max_block;
     84 
     85 	/*
     86 	 * Fill in the brel data structure
     87 	 */
     88 	brel->put = bma_put;
     89 	brel->get = bma_get;
     90 	brel->start_iter = bma_start_iter;
     91 	brel->next = bma_next;
     92 	brel->move = bma_move;
     93 	brel->delete = bma_delete;
     94 	brel->free = bma_free;
     95 
     96 	*new_brel = brel;
     97 	return 0;
     98 
     99 errout:
    100 	bma_free(brel);
    101 	return retval;
    102 }
    103 
    104 static errcode_t bma_put(ext2_brel brel, blk64_t old,
    105 			struct ext2_block_relocate_entry *ent)
    106 {
    107 	struct brel_ma 	*ma;
    108 
    109 	ma = brel->priv_data;
    110 	if (old > ma->max_block)
    111 		return EXT2_ET_INVALID_ARGUMENT;
    112 	ma->entries[(unsigned)old] = *ent;
    113 	return 0;
    114 }
    115 
    116 static errcode_t bma_get(ext2_brel brel, blk64_t old,
    117 			struct ext2_block_relocate_entry *ent)
    118 {
    119 	struct brel_ma 	*ma;
    120 
    121 	ma = brel->priv_data;
    122 	if (old > ma->max_block)
    123 		return EXT2_ET_INVALID_ARGUMENT;
    124 	if (ma->entries[(unsigned)old].new == 0)
    125 		return ENOENT;
    126 	*ent = ma->entries[old];
    127 	return 0;
    128 }
    129 
    130 static errcode_t bma_start_iter(ext2_brel brel)
    131 {
    132 	brel->current = 0;
    133 	return 0;
    134 }
    135 
    136 static errcode_t bma_next(ext2_brel brel, blk64_t *old,
    137 			  struct ext2_block_relocate_entry *ent)
    138 {
    139 	struct brel_ma 	*ma;
    140 
    141 	ma = brel->priv_data;
    142 	while (++brel->current < ma->max_block) {
    143 		if (ma->entries[(unsigned)brel->current].new == 0)
    144 			continue;
    145 		*old = brel->current;
    146 		*ent = ma->entries[(unsigned)brel->current];
    147 		return 0;
    148 	}
    149 	*old = 0;
    150 	return 0;
    151 }
    152 
    153 static errcode_t bma_move(ext2_brel brel, blk64_t old, blk64_t new)
    154 {
    155 	struct brel_ma 	*ma;
    156 
    157 	ma = brel->priv_data;
    158 	if ((old > ma->max_block) || (new > ma->max_block))
    159 		return EXT2_ET_INVALID_ARGUMENT;
    160 	if (ma->entries[(unsigned)old].new == 0)
    161 		return ENOENT;
    162 	ma->entries[(unsigned)new] = ma->entries[old];
    163 	ma->entries[(unsigned)old].new = 0;
    164 	return 0;
    165 }
    166 
    167 static errcode_t bma_delete(ext2_brel brel, blk64_t old)
    168 {
    169 	struct brel_ma 	*ma;
    170 
    171 	ma = brel->priv_data;
    172 	if (old > ma->max_block)
    173 		return EXT2_ET_INVALID_ARGUMENT;
    174 	if (ma->entries[(unsigned)old].new == 0)
    175 		return ENOENT;
    176 	ma->entries[(unsigned)old].new = 0;
    177 	return 0;
    178 }
    179 
    180 static errcode_t bma_free(ext2_brel brel)
    181 {
    182 	struct brel_ma 	*ma;
    183 
    184 	if (!brel)
    185 		return 0;
    186 
    187 	ma = brel->priv_data;
    188 
    189 	if (ma) {
    190 		if (ma->entries)
    191 			ext2fs_free_mem(&ma->entries);
    192 		ext2fs_free_mem(&ma);
    193 	}
    194 	if (brel->name)
    195 		ext2fs_free_mem(&brel->name);
    196 	ext2fs_free_mem(&brel);
    197 	return 0;
    198 }
    199