1 files changed, 6 insertions, 3 deletions

diff --git a/fs/xfs/libxfs/xfs_inode_fork.c b/fs/xfs/libxfs/xfs_inode_fork.c
index 222e103356c6..25c1e078aef6 100644
--- a/fs/xfs/libxfs/xfs_inode_fork.c
+++ b/fs/xfs/libxfs/xfs_inode_fork.c
@@ -26,6 +26,7 @@
 #include "xfs_inode.h"
 #include "xfs_trans.h"
 #include "xfs_inode_item.h"
+#include "xfs_btree.h"
 #include "xfs_bmap_btree.h"
 #include "xfs_bmap.h"
 #include "xfs_error.h"
@@ -429,11 +430,13 @@ xfs_iformat_btree(
         /* REFERENCED */
         int                     nrecs;
         int                     size;
+        int                     level;
 
         ifp = XFS_IFORK_PTR(ip, whichfork);
         dfp = (xfs_bmdr_block_t *)XFS_DFORK_PTR(dip, whichfork);
         size = XFS_BMAP_BROOT_SPACE(mp, dfp);
         nrecs = be16_to_cpu(dfp->bb_numrecs);
+        level = be16_to_cpu(dfp->bb_level);
 
         /*
          * blow out if -- fork has less extents than can fit in
@@ -446,7 +449,8 @@ xfs_iformat_btree(
                                         XFS_IFORK_MAXEXT(ip, whichfork) ||
                      XFS_BMDR_SPACE_CALC(nrecs) >
                                         XFS_DFORK_SIZE(dip, mp, whichfork) ||
-                     XFS_IFORK_NEXTENTS(ip, whichfork) > ip->i_d.di_nblocks)) {
+                     XFS_IFORK_NEXTENTS(ip, whichfork) > ip->i_d.di_nblocks) ||
+                     level == 0 || level > XFS_BTREE_MAXLEVELS) {
                 xfs_warn(mp, "corrupt inode %Lu (btree).",
                                         (unsigned long long) ip->i_ino);
                 XFS_CORRUPTION_ERROR("xfs_iformat_btree", XFS_ERRLEVEL_LOW,
@@ -497,15 +501,14 @@ xfs_iread_extents(
          * We know that the size is valid (it's checked in iformat_btree)
          */
         ifp->if_bytes = ifp->if_real_bytes = 0;
-        ifp->if_flags |= XFS_IFEXTENTS;
         xfs_iext_add(ifp, 0, nextents);
         error = xfs_bmap_read_extents(tp, ip, whichfork);
         if (error) {
                 xfs_iext_destroy(ifp);
-                ifp->if_flags &= ~XFS_IFEXTENTS;
                 return error;
         }
         xfs_validate_extents(ifp, nextents, XFS_EXTFMT_INODE(ip));
+        ifp->if_flags |= XFS_IFEXTENTS;
         return 0;
 }
 /*