3 files changed, 5 insertions, 4 deletions

diff --git a/fs/squashfs/dir.c b/fs/squashfs/dir.c
index 9dfe2ce0fb70..4a812bfcdd5e 100644
--- a/fs/squashfs/dir.c
+++ b/fs/squashfs/dir.c
@@ -173,8 +173,7 @@ static int squashfs_readdir(struct file *file, void *dirent, filldir_t filldir)
 
                 dir_count = le32_to_cpu(dirh.count) + 1;
 
-                /* dir_count should never be larger than 256 */
-                if (dir_count > 256)
+                if (dir_count > SQUASHFS_DIR_COUNT)
                         goto failed_read;
 
                 while (dir_count--) {
diff --git a/fs/squashfs/namei.c b/fs/squashfs/namei.c
index 0682b38d7e31..66c70fc8c80b 100644
--- a/fs/squashfs/namei.c
+++ b/fs/squashfs/namei.c
@@ -177,8 +177,7 @@ static struct dentry *squashfs_lookup(struct inode *dir, struct dentry *dentry,
 
                 dir_count = le32_to_cpu(dirh.count) + 1;
 
-                /* dir_count should never be larger than 256 */
-                if (dir_count > 256)
+                if (dir_count > SQUASHFS_DIR_COUNT)
                         goto data_error;
 
                 while (dir_count--) {
diff --git a/fs/squashfs/squashfs_fs.h b/fs/squashfs/squashfs_fs.h
index e8e14645de9a..b2f9f623ac98 100644
--- a/fs/squashfs/squashfs_fs.h
+++ b/fs/squashfs/squashfs_fs.h
@@ -52,6 +52,9 @@
 /* Max length of filename (not 255) */
 #define SQUASHFS_NAME_LEN               256
 
+/* Max value for directory header count*/
+#define SQUASHFS_DIR_COUNT              256
+
 #define SQUASHFS_INVALID_FRAG           (0xffffffffU)
 #define SQUASHFS_INVALID_XATTR          (0xffffffffU)
 #define SQUASHFS_INVALID_BLK            (-1LL)