1e6a6d379SPhillip Lougher /* 2e6a6d379SPhillip Lougher * Squashfs - a compressed read only filesystem for Linux 3e6a6d379SPhillip Lougher * 4e6a6d379SPhillip Lougher * Copyright (c) 2002, 2003, 2004, 2005, 2006, 2007, 2008, 2009 5e6a6d379SPhillip Lougher * Phillip Lougher <phillip@lougher.demon.co.uk> 6e6a6d379SPhillip Lougher * 7e6a6d379SPhillip Lougher * This program is free software; you can redistribute it and/or 8e6a6d379SPhillip Lougher * modify it under the terms of the GNU General Public License 9e6a6d379SPhillip Lougher * as published by the Free Software Foundation; either version 2, 10e6a6d379SPhillip Lougher * or (at your option) any later version. 11e6a6d379SPhillip Lougher * 12e6a6d379SPhillip Lougher * This program is distributed in the hope that it will be useful, 13e6a6d379SPhillip Lougher * but WITHOUT ANY WARRANTY; without even the implied warranty of 14e6a6d379SPhillip Lougher * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 15e6a6d379SPhillip Lougher * GNU General Public License for more details. 16e6a6d379SPhillip Lougher * 17e6a6d379SPhillip Lougher * You should have received a copy of the GNU General Public License 18e6a6d379SPhillip Lougher * along with this program; if not, write to the Free Software 19e6a6d379SPhillip Lougher * Foundation, 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. 20e6a6d379SPhillip Lougher * 21e6a6d379SPhillip Lougher * zlib_wrapper.c 22e6a6d379SPhillip Lougher */ 23e6a6d379SPhillip Lougher 24e6a6d379SPhillip Lougher 25e6a6d379SPhillip Lougher #include <linux/mutex.h> 26e6a6d379SPhillip Lougher #include <linux/buffer_head.h> 27e6a6d379SPhillip Lougher #include <linux/zlib.h> 28e6a6d379SPhillip Lougher 29e6a6d379SPhillip Lougher #include "squashfs_fs.h" 30e6a6d379SPhillip Lougher #include "squashfs_fs_sb.h" 31e6a6d379SPhillip Lougher #include "squashfs_fs_i.h" 32e6a6d379SPhillip Lougher #include "squashfs.h" 33e6a6d379SPhillip Lougher 34f1a40359SPhillip Lougher void *squashfs_zlib_init() 35f1a40359SPhillip Lougher { 36f1a40359SPhillip Lougher z_stream *stream = kmalloc(sizeof(z_stream), GFP_KERNEL); 37f1a40359SPhillip Lougher if (stream == NULL) 38f1a40359SPhillip Lougher goto failed; 39f1a40359SPhillip Lougher stream->workspace = kmalloc(zlib_inflate_workspacesize(), 40f1a40359SPhillip Lougher GFP_KERNEL); 41f1a40359SPhillip Lougher if (stream->workspace == NULL) 42f1a40359SPhillip Lougher goto failed; 43f1a40359SPhillip Lougher 44f1a40359SPhillip Lougher return stream; 45f1a40359SPhillip Lougher 46f1a40359SPhillip Lougher failed: 47f1a40359SPhillip Lougher ERROR("Failed to allocate zlib workspace\n"); 48f1a40359SPhillip Lougher kfree(stream); 49f1a40359SPhillip Lougher return NULL; 50f1a40359SPhillip Lougher } 51f1a40359SPhillip Lougher 52f1a40359SPhillip Lougher 53f1a40359SPhillip Lougher void squashfs_zlib_free(void *strm) 54f1a40359SPhillip Lougher { 55f1a40359SPhillip Lougher z_stream *stream = strm; 56f1a40359SPhillip Lougher 57f1a40359SPhillip Lougher if (stream) 58f1a40359SPhillip Lougher kfree(stream->workspace); 59f1a40359SPhillip Lougher kfree(stream); 60f1a40359SPhillip Lougher } 61f1a40359SPhillip Lougher 62f1a40359SPhillip Lougher 63e6a6d379SPhillip Lougher int squashfs_zlib_uncompress(struct squashfs_sb_info *msblk, void **buffer, 64e6a6d379SPhillip Lougher struct buffer_head **bh, int b, int offset, int length, int srclength, 65e6a6d379SPhillip Lougher int pages) 66e6a6d379SPhillip Lougher { 67e6a6d379SPhillip Lougher int zlib_err = 0, zlib_init = 0; 68e6a6d379SPhillip Lougher int avail, bytes, k = 0, page = 0; 69f1a40359SPhillip Lougher z_stream *stream = msblk->stream; 70e6a6d379SPhillip Lougher 71e6a6d379SPhillip Lougher mutex_lock(&msblk->read_data_mutex); 72e6a6d379SPhillip Lougher 73f1a40359SPhillip Lougher stream->avail_out = 0; 74f1a40359SPhillip Lougher stream->avail_in = 0; 75e6a6d379SPhillip Lougher 76e6a6d379SPhillip Lougher bytes = length; 77e6a6d379SPhillip Lougher do { 78f1a40359SPhillip Lougher if (stream->avail_in == 0 && k < b) { 79e6a6d379SPhillip Lougher avail = min(bytes, msblk->devblksize - offset); 80e6a6d379SPhillip Lougher bytes -= avail; 81e6a6d379SPhillip Lougher wait_on_buffer(bh[k]); 82e6a6d379SPhillip Lougher if (!buffer_uptodate(bh[k])) 83e6a6d379SPhillip Lougher goto release_mutex; 84e6a6d379SPhillip Lougher 85e6a6d379SPhillip Lougher if (avail == 0) { 86e6a6d379SPhillip Lougher offset = 0; 87e6a6d379SPhillip Lougher put_bh(bh[k++]); 88e6a6d379SPhillip Lougher continue; 89e6a6d379SPhillip Lougher } 90e6a6d379SPhillip Lougher 91f1a40359SPhillip Lougher stream->next_in = bh[k]->b_data + offset; 92f1a40359SPhillip Lougher stream->avail_in = avail; 93e6a6d379SPhillip Lougher offset = 0; 94e6a6d379SPhillip Lougher } 95e6a6d379SPhillip Lougher 96f1a40359SPhillip Lougher if (stream->avail_out == 0 && page < pages) { 97f1a40359SPhillip Lougher stream->next_out = buffer[page++]; 98f1a40359SPhillip Lougher stream->avail_out = PAGE_CACHE_SIZE; 99e6a6d379SPhillip Lougher } 100e6a6d379SPhillip Lougher 101e6a6d379SPhillip Lougher if (!zlib_init) { 102f1a40359SPhillip Lougher zlib_err = zlib_inflateInit(stream); 103e6a6d379SPhillip Lougher if (zlib_err != Z_OK) { 104e6a6d379SPhillip Lougher ERROR("zlib_inflateInit returned unexpected " 105e6a6d379SPhillip Lougher "result 0x%x, srclength %d\n", 106e6a6d379SPhillip Lougher zlib_err, srclength); 107e6a6d379SPhillip Lougher goto release_mutex; 108e6a6d379SPhillip Lougher } 109e6a6d379SPhillip Lougher zlib_init = 1; 110e6a6d379SPhillip Lougher } 111e6a6d379SPhillip Lougher 112f1a40359SPhillip Lougher zlib_err = zlib_inflate(stream, Z_SYNC_FLUSH); 113e6a6d379SPhillip Lougher 114f1a40359SPhillip Lougher if (stream->avail_in == 0 && k < b) 115e6a6d379SPhillip Lougher put_bh(bh[k++]); 116e6a6d379SPhillip Lougher } while (zlib_err == Z_OK); 117e6a6d379SPhillip Lougher 118e6a6d379SPhillip Lougher if (zlib_err != Z_STREAM_END) { 119e6a6d379SPhillip Lougher ERROR("zlib_inflate error, data probably corrupt\n"); 120e6a6d379SPhillip Lougher goto release_mutex; 121e6a6d379SPhillip Lougher } 122e6a6d379SPhillip Lougher 123f1a40359SPhillip Lougher zlib_err = zlib_inflateEnd(stream); 124e6a6d379SPhillip Lougher if (zlib_err != Z_OK) { 125e6a6d379SPhillip Lougher ERROR("zlib_inflate error, data probably corrupt\n"); 126e6a6d379SPhillip Lougher goto release_mutex; 127e6a6d379SPhillip Lougher } 128e6a6d379SPhillip Lougher 129e6a6d379SPhillip Lougher mutex_unlock(&msblk->read_data_mutex); 130f1a40359SPhillip Lougher return stream->total_out; 131e6a6d379SPhillip Lougher 132e6a6d379SPhillip Lougher release_mutex: 133e6a6d379SPhillip Lougher mutex_unlock(&msblk->read_data_mutex); 134e6a6d379SPhillip Lougher 135e6a6d379SPhillip Lougher for (; k < b; k++) 136e6a6d379SPhillip Lougher put_bh(bh[k]); 137e6a6d379SPhillip Lougher 138e6a6d379SPhillip Lougher return -EIO; 139e6a6d379SPhillip Lougher } 140