Whamcloud - gitweb
Add support for the DIR_NLINK feature.
[tools/e2fsprogs.git] / lib / ext2fs / bmap.c
index 43f9f31..24f05fd 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * bmap.c --- logical to phiscal block mapping
+ * bmap.c --- logical to physical block mapping
  *
  * Copyright (C) 1997 Theodore Ts'o.
  *
 #if HAVE_UNISTD_H
 #include <unistd.h>
 #endif
-#include <stdlib.h>
-
-#include <linux/ext2_fs.h>
+#include <errno.h>
 
+#include "ext2_fs.h"
 #include "ext2fs.h"
 
 #if defined(__GNUC__) && !defined(NO_INLINE_FUNCS)
 #define _BMAP_INLINE_
 #endif
 
-extern errcode_t ext2fs_bmap(ext2_filsys fs, ino_t ino,
+extern errcode_t ext2fs_bmap(ext2_filsys fs, ext2_ino_t ino,
                             struct ext2_inode *inode, 
                             char *block_buf, int bmap_flags,
                             blk_t block, blk_t *phys_blk);
 
-#define BMAP_ALLOC     1
-
 #define inode_bmap(inode, nr) ((inode)->i_block[(nr)])
 
-static blk_t _BMAP_INLINE_ block_bmap(ext2_filsys fs, char *buf, blk_t nr)
-{
-       blk_t tmp;
-
-       tmp = ((blk_t *) buf)[nr];
-
-       if ((fs->flags & EXT2_FLAG_SWAP_BYTES) ||
-           (fs->flags & EXT2_FLAG_SWAP_BYTES_READ))
-               return ext2fs_swab32(tmp);
-
-       return tmp;
-}
-
-static errcode_t _BMAP_INLINE_ block_ind_bmap(ext2_filsys fs, int flags, 
+static _BMAP_INLINE_ errcode_t block_ind_bmap(ext2_filsys fs, int flags, 
                                              blk_t ind, char *block_buf, 
                                              int *blocks_alloc,
                                              blk_t nr, blk_t *ret_blk)
@@ -57,6 +41,8 @@ static errcode_t _BMAP_INLINE_ block_ind_bmap(ext2_filsys fs, int flags,
        blk_t           b;
 
        if (!ind) {
+               if (flags & BMAP_SET)
+                       return EXT2_ET_SET_BMAP_NO_IND;
                *ret_blk = 0;
                return 0;
        }
@@ -64,11 +50,20 @@ static errcode_t _BMAP_INLINE_ block_ind_bmap(ext2_filsys fs, int flags,
        if (retval)
                return retval;
 
+       if (flags & BMAP_SET) {
+               b = *ret_blk;
+#ifdef WORDS_BIGENDIAN
+               b = ext2fs_swab32(b);
+#endif
+               ((blk_t *) block_buf)[nr] = b;
+               return io_channel_write_blk(fs->io, ind, 1, block_buf);
+       }
+
        b = ((blk_t *) block_buf)[nr];
 
-       if ((fs->flags & EXT2_FLAG_SWAP_BYTES) ||
-           (fs->flags & EXT2_FLAG_SWAP_BYTES_READ))
-               b = ext2fs_swab32(b);
+#ifdef WORDS_BIGENDIAN
+       b = ext2fs_swab32(b);
+#endif
 
        if (!b && (flags & BMAP_ALLOC)) {
                b = nr ? ((blk_t *) block_buf)[nr-1] : 0;
@@ -77,11 +72,11 @@ static errcode_t _BMAP_INLINE_ block_ind_bmap(ext2_filsys fs, int flags,
                if (retval)
                        return retval;
 
-               if ((fs->flags & EXT2_FLAG_SWAP_BYTES) ||
-                   (fs->flags & EXT2_FLAG_SWAP_BYTES_WRITE))
-                       ((blk_t *) block_buf)[nr] = ext2fs_swab32(b);
-               else
-                       ((blk_t *) block_buf)[nr] = b;
+#ifdef WORDS_BIGENDIAN
+               ((blk_t *) block_buf)[nr] = ext2fs_swab32(b);
+#else
+               ((blk_t *) block_buf)[nr] = b;
+#endif
 
                retval = io_channel_write_blk(fs->io, ind, 1, block_buf);
                if (retval)
@@ -94,19 +89,19 @@ static errcode_t _BMAP_INLINE_ block_ind_bmap(ext2_filsys fs, int flags,
        return 0;
 }
 
-static errcode_t _BMAP_INLINE_ block_dind_bmap(ext2_filsys fs, int flags,
+static _BMAP_INLINE_ errcode_t block_dind_bmap(ext2_filsys fs, int flags,
                                               blk_t dind, char *block_buf, 
                                               int *blocks_alloc,
                                               blk_t nr, blk_t *ret_blk)
 {
        blk_t           b;
        errcode_t       retval;
-       int             addr_per_block;
+       blk_t           addr_per_block;
        
-       addr_per_block = fs->blocksize >> 2;
+       addr_per_block = (blk_t) fs->blocksize >> 2;
 
-       retval = block_ind_bmap(fs, flags, dind, block_buf, blocks_alloc,
-                               nr / addr_per_block, &b);
+       retval = block_ind_bmap(fs, flags & ~BMAP_SET, dind, block_buf, 
+                               blocks_alloc, nr / addr_per_block, &b);
        if (retval)
                return retval;
        retval = block_ind_bmap(fs, flags, b, block_buf, blocks_alloc,
@@ -114,19 +109,19 @@ static errcode_t _BMAP_INLINE_ block_dind_bmap(ext2_filsys fs, int flags,
        return retval;
 }
 
-static errcode_t _BMAP_INLINE_ block_tind_bmap(ext2_filsys fs, int flags,
+static _BMAP_INLINE_ errcode_t block_tind_bmap(ext2_filsys fs, int flags,
                                               blk_t tind, char *block_buf, 
                                               int *blocks_alloc,
                                               blk_t nr, blk_t *ret_blk)
 {
        blk_t           b;
        errcode_t       retval;
-       int             addr_per_block;
+       blk_t           addr_per_block;
        
-       addr_per_block = fs->blocksize >> 2;
+       addr_per_block = (blk_t) fs->blocksize >> 2;
 
-       retval = block_dind_bmap(fs, flags, tind, block_buf, blocks_alloc,
-                                nr / addr_per_block, &b);
+       retval = block_dind_bmap(fs, flags & ~BMAP_SET, tind, block_buf, 
+                                blocks_alloc, nr / addr_per_block, &b);
        if (retval)
                return retval;
        retval = block_ind_bmap(fs, flags, b, block_buf, blocks_alloc,
@@ -134,36 +129,82 @@ static errcode_t _BMAP_INLINE_ block_tind_bmap(ext2_filsys fs, int flags,
        return retval;
 }
 
-errcode_t ext2fs_bmap(ext2_filsys fs, ino_t ino, struct ext2_inode *inode,
-                     char *block_buf, int bmap_flags, blk_t block,
-                     blk_t *phys_blk)
+errcode_t ext2fs_bmap2(ext2_filsys fs, ext2_ino_t ino, struct ext2_inode *inode,
+                      char *block_buf, int bmap_flags, blk64_t block,
+                      int *ret_flags, blk64_t *phys_blk)
 {
        struct ext2_inode inode_buf;
-       int addr_per_block;
-       blk_t   b;
+       ext2_extent_handle_t handle = 0;
+       blk_t addr_per_block;
+       blk_t   b, blk32;
        char    *buf = 0;
        errcode_t       retval = 0;
-       int             blocks_alloc = 0;
+       int             blocks_alloc = 0, inode_dirty = 0;
 
-       *phys_blk = 0;
+       if (!(bmap_flags & BMAP_SET))
+               *phys_blk = 0;
+
+       if (ret_flags)
+               *ret_flags = 0;
 
        /* Read inode structure if necessary */
        if (!inode) {
                retval = ext2fs_read_inode(fs, ino, &inode_buf);
-               if (!retval)
+               if (retval)
                        return retval;
                inode = &inode_buf;
        }
-       addr_per_block = fs->blocksize >> 2;
+       addr_per_block = (blk_t) fs->blocksize >> 2;
+
+       if (inode->i_flags & EXT4_EXTENTS_FL) {
+               struct ext2fs_extent    extent;
+               int                     offset;
+
+               if (bmap_flags & BMAP_SET) {
+                       retval = EXT2_ET_EXTENT_NOT_SUPPORTED;
+                       goto done;
+               }
+               retval = ext2fs_extent_open(fs, ino, &handle);
+               if (retval)
+                       goto done;
+               retval = ext2fs_extent_goto(handle, block);
+               if (retval) {
+                       /* If the extent is not found, return phys_blk = 0 */
+                       if (retval == EXT2_ET_EXTENT_NOT_FOUND)
+                               retval = 0;
+                       goto done;
+               }
+               retval = ext2fs_extent_get(handle, EXT2_EXTENT_CURRENT, &extent);
+               if (retval)
+                       goto done;
+               offset = block - extent.e_lblk;
+               if (block >= extent.e_lblk && (offset <= extent.e_len)) {
+                       *phys_blk = extent.e_pblk + offset;
+                       if (ret_flags && extent.e_flags & EXT2_EXTENT_FLAGS_UNINIT)
+                               *ret_flags |= BMAP_RET_UNINIT;
+               }
+               retval = 0;
+               goto done;
+       }
 
        if (!block_buf) {
-               retval = ext2fs_get_mem(fs->blocksize * 2, (void **) &buf);
+               retval = ext2fs_get_array(2, fs->blocksize, &buf);
                if (retval)
                        return retval;
                block_buf = buf;
        }
 
        if (block < EXT2_NDIR_BLOCKS) {
+               if (bmap_flags & BMAP_SET) {
+                       b = *phys_blk;
+#ifdef WORDS_BIGENDIAN
+                       b = ext2fs_swab32(b);
+#endif
+                       inode_bmap(inode, block) = b;
+                       inode_dirty++;
+                       goto done;
+               }
+
                *phys_blk = inode_bmap(inode, block);
                b = block ? inode_bmap(inode, block-1) : 0;
                
@@ -180,11 +221,15 @@ errcode_t ext2fs_bmap(ext2_filsys fs, ino_t ino, struct ext2_inode *inode,
        
        /* Indirect block */
        block -= EXT2_NDIR_BLOCKS;
+       blk32 = *phys_blk;
        if (block < addr_per_block) {
                b = inode_bmap(inode, EXT2_IND_BLOCK);
                if (!b) {
-                       if (!(bmap_flags & BMAP_ALLOC))
-                           goto done;
+                       if (!(bmap_flags & BMAP_ALLOC)) {
+                               if (bmap_flags & BMAP_SET)
+                                       retval = EXT2_ET_SET_BMAP_NO_IND;
+                               goto done;
+                       }
 
                        b = inode_bmap(inode, EXT2_IND_BLOCK-1);
                        retval = ext2fs_alloc_block(fs, b, block_buf, &b);
@@ -194,7 +239,9 @@ errcode_t ext2fs_bmap(ext2_filsys fs, ino_t ino, struct ext2_inode *inode,
                        blocks_alloc++;
                }
                retval = block_ind_bmap(fs, bmap_flags, b, block_buf, 
-                                       &blocks_alloc, block, phys_blk);
+                                       &blocks_alloc, block, &blk32);
+               if (retval == 0)
+                       *phys_blk = blk32;
                goto done;
        }
        
@@ -203,8 +250,11 @@ errcode_t ext2fs_bmap(ext2_filsys fs, ino_t ino, struct ext2_inode *inode,
        if (block < addr_per_block * addr_per_block) {
                b = inode_bmap(inode, EXT2_DIND_BLOCK);
                if (!b) {
-                       if (!(bmap_flags & BMAP_ALLOC))
-                           goto done;
+                       if (!(bmap_flags & BMAP_ALLOC)) {
+                               if (bmap_flags & BMAP_SET)
+                                       retval = EXT2_ET_SET_BMAP_NO_IND;
+                               goto done;
+                       }
 
                        b = inode_bmap(inode, EXT2_IND_BLOCK);
                        retval = ext2fs_alloc_block(fs, b, block_buf, &b);
@@ -214,7 +264,9 @@ errcode_t ext2fs_bmap(ext2_filsys fs, ino_t ino, struct ext2_inode *inode,
                        blocks_alloc++;
                }
                retval = block_dind_bmap(fs, bmap_flags, b, block_buf, 
-                                        &blocks_alloc, block, phys_blk);
+                                        &blocks_alloc, block, &blk32);
+               if (retval == 0)
+                       *phys_blk = blk32;
                goto done;
        }
 
@@ -222,8 +274,11 @@ errcode_t ext2fs_bmap(ext2_filsys fs, ino_t ino, struct ext2_inode *inode,
        block -= addr_per_block * addr_per_block;
        b = inode_bmap(inode, EXT2_TIND_BLOCK);
        if (!b) {
-               if (!(bmap_flags & BMAP_ALLOC))
+               if (!(bmap_flags & BMAP_ALLOC)) {
+                       if (bmap_flags & BMAP_SET)
+                               retval = EXT2_ET_SET_BMAP_NO_IND;
                        goto done;
+               }
 
                b = inode_bmap(inode, EXT2_DIND_BLOCK);
                retval = ext2fs_alloc_block(fs, b, block_buf, &b);
@@ -233,16 +288,34 @@ errcode_t ext2fs_bmap(ext2_filsys fs, ino_t ino, struct ext2_inode *inode,
                blocks_alloc++;
        }
        retval = block_tind_bmap(fs, bmap_flags, b, block_buf, 
-                                &blocks_alloc, block, phys_blk);
+                                &blocks_alloc, block, &blk32);
+       if (retval == 0)
+               *phys_blk = blk32;
 done:
        if (buf)
-               ext2fs_free_mem((void **) &buf);
-       if ((retval == 0) && blocks_alloc) {
+               ext2fs_free_mem(&buf);
+       if (handle)
+               ext2fs_extent_free(handle);
+       if ((retval == 0) && (blocks_alloc || inode_dirty)) {
                inode->i_blocks += (blocks_alloc * fs->blocksize) / 512;
                retval = ext2fs_write_inode(fs, ino, inode);
        }
        return retval;
 }
 
-
-
+errcode_t ext2fs_bmap(ext2_filsys fs, ext2_ino_t ino, struct ext2_inode *inode,
+                     char *block_buf, int bmap_flags, blk_t block,
+                     blk_t *phys_blk)
+{
+       errcode_t ret;
+       blk64_t ret_blk;
+
+       ret = ext2fs_bmap2(fs, ino, inode, block_buf, bmap_flags, block,
+                           0, &ret_blk);
+       if (ret)
+               return ret;
+       if (ret_blk >= ((long long) 1 << 32))
+               return EOVERFLOW;
+       *phys_blk = ret_blk;
+       return 0;
+}