| Phillip Lougher | 122edd1 | 2009-01-05 08:46:25 +0000 | [diff] [blame] | 1 | /* | 
 | 2 |  * Squashfs - a compressed read only filesystem for Linux | 
 | 3 |  * | 
 | 4 |  * Copyright (c) 2002, 2003, 2004, 2005, 2006, 2007, 2008 | 
| Phillip Lougher | d7f2ff6 | 2011-05-26 10:39:56 +0100 | [diff] [blame] | 5 |  * Phillip Lougher <phillip@squashfs.org.uk> | 
| Phillip Lougher | 122edd1 | 2009-01-05 08:46:25 +0000 | [diff] [blame] | 6 |  * | 
 | 7 |  * This program is free software; you can redistribute it and/or | 
 | 8 |  * modify it under the terms of the GNU General Public License | 
 | 9 |  * as published by the Free Software Foundation; either version 2, | 
 | 10 |  * or (at your option) any later version. | 
 | 11 |  * | 
 | 12 |  * This program is distributed in the hope that it will be useful, | 
 | 13 |  * but WITHOUT ANY WARRANTY; without even the implied warranty of | 
 | 14 |  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | 
 | 15 |  * GNU General Public License for more details. | 
 | 16 |  * | 
 | 17 |  * You should have received a copy of the GNU General Public License | 
 | 18 |  * along with this program; if not, write to the Free Software | 
 | 19 |  * Foundation, 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. | 
 | 20 |  * | 
 | 21 |  * fragment.c | 
 | 22 |  */ | 
 | 23 |  | 
 | 24 | /* | 
 | 25 |  * This file implements code to handle compressed fragments (tail-end packed | 
 | 26 |  * datablocks). | 
 | 27 |  * | 
 | 28 |  * Regular files contain a fragment index which is mapped to a fragment | 
 | 29 |  * location on disk and compressed size using a fragment lookup table. | 
 | 30 |  * Like everything in Squashfs this fragment lookup table is itself stored | 
 | 31 |  * compressed into metadata blocks.  A second index table is used to locate | 
 | 32 |  * these.  This second index table for speed of access (and because it | 
 | 33 |  * is small) is read at mount time and cached in memory. | 
 | 34 |  */ | 
 | 35 |  | 
 | 36 | #include <linux/fs.h> | 
 | 37 | #include <linux/vfs.h> | 
 | 38 | #include <linux/slab.h> | 
| Phillip Lougher | 122edd1 | 2009-01-05 08:46:25 +0000 | [diff] [blame] | 39 |  | 
 | 40 | #include "squashfs_fs.h" | 
 | 41 | #include "squashfs_fs_sb.h" | 
| Phillip Lougher | 122edd1 | 2009-01-05 08:46:25 +0000 | [diff] [blame] | 42 | #include "squashfs.h" | 
 | 43 |  | 
 | 44 | /* | 
 | 45 |  * Look-up fragment using the fragment index table.  Return the on disk | 
 | 46 |  * location of the fragment and its compressed size | 
 | 47 |  */ | 
 | 48 | int squashfs_frag_lookup(struct super_block *sb, unsigned int fragment, | 
 | 49 | 				u64 *fragment_block) | 
 | 50 | { | 
 | 51 | 	struct squashfs_sb_info *msblk = sb->s_fs_info; | 
 | 52 | 	int block = SQUASHFS_FRAGMENT_INDEX(fragment); | 
 | 53 | 	int offset = SQUASHFS_FRAGMENT_INDEX_OFFSET(fragment); | 
 | 54 | 	u64 start_block = le64_to_cpu(msblk->fragment_index[block]); | 
 | 55 | 	struct squashfs_fragment_entry fragment_entry; | 
 | 56 | 	int size; | 
 | 57 |  | 
 | 58 | 	size = squashfs_read_metadata(sb, &fragment_entry, &start_block, | 
 | 59 | 					&offset, sizeof(fragment_entry)); | 
 | 60 | 	if (size < 0) | 
 | 61 | 		return size; | 
 | 62 |  | 
 | 63 | 	*fragment_block = le64_to_cpu(fragment_entry.start_block); | 
 | 64 | 	size = le32_to_cpu(fragment_entry.size); | 
 | 65 |  | 
 | 66 | 	return size; | 
 | 67 | } | 
 | 68 |  | 
 | 69 |  | 
 | 70 | /* | 
 | 71 |  * Read the uncompressed fragment lookup table indexes off disk into memory | 
 | 72 |  */ | 
 | 73 | __le64 *squashfs_read_fragment_index_table(struct super_block *sb, | 
| Phillip Lougher | 1cac63c | 2011-05-24 04:33:34 +0100 | [diff] [blame] | 74 | 	u64 fragment_table_start, u64 next_table, unsigned int fragments) | 
| Phillip Lougher | 122edd1 | 2009-01-05 08:46:25 +0000 | [diff] [blame] | 75 | { | 
 | 76 | 	unsigned int length = SQUASHFS_FRAGMENT_INDEX_BYTES(fragments); | 
| Phillip Lougher | 1cac63c | 2011-05-24 04:33:34 +0100 | [diff] [blame] | 77 | 	__le64 *table; | 
| Phillip Lougher | 122edd1 | 2009-01-05 08:46:25 +0000 | [diff] [blame] | 78 |  | 
| Phillip Lougher | 1cac63c | 2011-05-24 04:33:34 +0100 | [diff] [blame] | 79 | 	/* | 
 | 80 | 	 * Sanity check, length bytes should not extend into the next table - | 
 | 81 | 	 * this check also traps instances where fragment_table_start is | 
 | 82 | 	 * incorrectly larger than the next table start | 
 | 83 | 	 */ | 
 | 84 | 	if (fragment_table_start + length > next_table) | 
 | 85 | 		return ERR_PTR(-EINVAL); | 
 | 86 |  | 
 | 87 | 	table = squashfs_read_table(sb, fragment_table_start, length); | 
 | 88 |  | 
 | 89 | 	/* | 
 | 90 | 	 * table[0] points to the first fragment table metadata block, this | 
 | 91 | 	 * should be less than fragment_table_start | 
 | 92 | 	 */ | 
| Phillip Lougher | d5b72ce | 2011-05-29 00:38:46 +0100 | [diff] [blame] | 93 | 	if (!IS_ERR(table) && le64_to_cpu(table[0]) >= fragment_table_start) { | 
| Phillip Lougher | 1cac63c | 2011-05-24 04:33:34 +0100 | [diff] [blame] | 94 | 		kfree(table); | 
 | 95 | 		return ERR_PTR(-EINVAL); | 
 | 96 | 	} | 
 | 97 |  | 
 | 98 | 	return table; | 
| Phillip Lougher | 122edd1 | 2009-01-05 08:46:25 +0000 | [diff] [blame] | 99 | } |