Commit 9508c6b9 authored by Phillip Lougher's avatar Phillip Lougher

Squashfs: Refactor decompressor interface and code

The decompressor interface and code was written from
the point of view of single-threaded operation.  In doing
so it mixed a lot of single-threaded implementation specific
aspects into the decompressor code and elsewhere which makes it
difficult to seamlessly support multiple different decompressor
implementations.

This patch does the following:

1.  It removes compressor_options parsing from the decompressor
    init() function.  This allows the decompressor init() function
    to be dynamically called to instantiate multiple decompressors,
    without the compressor options needing to be read and parsed each
    time.

2.  It moves threading and all sleeping operations out of the
    decompressors.  In doing so, it makes the decompressors
    non-blocking wrappers which only deal with interfacing with
    the decompressor implementation.

3. It splits decompressor.[ch] into decompressor generic functions
   in decompressor.[ch], and moves the single threaded
   decompressor implementation into decompressor_single.c.

The result of this patch is Squashfs should now be able to
support multiple decompressors by adding new decompressor_xxx.c
files with specialised implementations of the functions in
decompressor_single.c
Signed-off-by: default avatarPhillip Lougher <phillip@squashfs.org.uk>
Reviewed-by: default avatarMinchan Kim <minchan@kernel.org>
parent 959f5854
...@@ -4,7 +4,7 @@ ...@@ -4,7 +4,7 @@
obj-$(CONFIG_SQUASHFS) += squashfs.o obj-$(CONFIG_SQUASHFS) += squashfs.o
squashfs-y += block.o cache.o dir.o export.o file.o fragment.o id.o inode.o squashfs-y += block.o cache.o dir.o export.o file.o fragment.o id.o inode.o
squashfs-y += namei.o super.o symlink.o decompressor.o squashfs-y += namei.o super.o symlink.o decompressor.o decompressor_single.o
squashfs-$(CONFIG_SQUASHFS_XATTR) += xattr.o xattr_id.o squashfs-$(CONFIG_SQUASHFS_XATTR) += xattr.o xattr_id.o
squashfs-$(CONFIG_SQUASHFS_LZO) += lzo_wrapper.o squashfs-$(CONFIG_SQUASHFS_LZO) += lzo_wrapper.o
squashfs-$(CONFIG_SQUASHFS_XZ) += xz_wrapper.o squashfs-$(CONFIG_SQUASHFS_XZ) += xz_wrapper.o
......
...@@ -93,7 +93,7 @@ int squashfs_read_data(struct super_block *sb, void **buffer, u64 index, ...@@ -93,7 +93,7 @@ int squashfs_read_data(struct super_block *sb, void **buffer, u64 index,
struct buffer_head **bh; struct buffer_head **bh;
int offset = index & ((1 << msblk->devblksize_log2) - 1); int offset = index & ((1 << msblk->devblksize_log2) - 1);
u64 cur_index = index >> msblk->devblksize_log2; u64 cur_index = index >> msblk->devblksize_log2;
int bytes, compressed, b = 0, k = 0, page = 0, avail; int bytes, compressed, b = 0, k = 0, page = 0, avail, i;
bh = kcalloc(((srclength + msblk->devblksize - 1) bh = kcalloc(((srclength + msblk->devblksize - 1)
>> msblk->devblksize_log2) + 1, sizeof(*bh), GFP_KERNEL); >> msblk->devblksize_log2) + 1, sizeof(*bh), GFP_KERNEL);
...@@ -158,6 +158,12 @@ int squashfs_read_data(struct super_block *sb, void **buffer, u64 index, ...@@ -158,6 +158,12 @@ int squashfs_read_data(struct super_block *sb, void **buffer, u64 index,
ll_rw_block(READ, b - 1, bh + 1); ll_rw_block(READ, b - 1, bh + 1);
} }
for (i = 0; i < b; i++) {
wait_on_buffer(bh[i]);
if (!buffer_uptodate(bh[i]))
goto block_release;
}
if (compressed) { if (compressed) {
length = squashfs_decompress(msblk, buffer, bh, b, offset, length = squashfs_decompress(msblk, buffer, bh, b, offset,
length, srclength, pages); length, srclength, pages);
...@@ -172,9 +178,6 @@ int squashfs_read_data(struct super_block *sb, void **buffer, u64 index, ...@@ -172,9 +178,6 @@ int squashfs_read_data(struct super_block *sb, void **buffer, u64 index,
for (bytes = length; k < b; k++) { for (bytes = length; k < b; k++) {
in = min(bytes, msblk->devblksize - offset); in = min(bytes, msblk->devblksize - offset);
bytes -= in; bytes -= in;
wait_on_buffer(bh[k]);
if (!buffer_uptodate(bh[k]))
goto block_release;
while (in) { while (in) {
if (pg_offset == PAGE_CACHE_SIZE) { if (pg_offset == PAGE_CACHE_SIZE) {
page++; page++;
......
...@@ -37,29 +37,29 @@ ...@@ -37,29 +37,29 @@
*/ */
static const struct squashfs_decompressor squashfs_lzma_unsupported_comp_ops = { static const struct squashfs_decompressor squashfs_lzma_unsupported_comp_ops = {
NULL, NULL, NULL, LZMA_COMPRESSION, "lzma", 0 NULL, NULL, NULL, NULL, LZMA_COMPRESSION, "lzma", 0
}; };
#ifndef CONFIG_SQUASHFS_LZO #ifndef CONFIG_SQUASHFS_LZO
static const struct squashfs_decompressor squashfs_lzo_comp_ops = { static const struct squashfs_decompressor squashfs_lzo_comp_ops = {
NULL, NULL, NULL, LZO_COMPRESSION, "lzo", 0 NULL, NULL, NULL, NULL, LZO_COMPRESSION, "lzo", 0
}; };
#endif #endif
#ifndef CONFIG_SQUASHFS_XZ #ifndef CONFIG_SQUASHFS_XZ
static const struct squashfs_decompressor squashfs_xz_comp_ops = { static const struct squashfs_decompressor squashfs_xz_comp_ops = {
NULL, NULL, NULL, XZ_COMPRESSION, "xz", 0 NULL, NULL, NULL, NULL, XZ_COMPRESSION, "xz", 0
}; };
#endif #endif
#ifndef CONFIG_SQUASHFS_ZLIB #ifndef CONFIG_SQUASHFS_ZLIB
static const struct squashfs_decompressor squashfs_zlib_comp_ops = { static const struct squashfs_decompressor squashfs_zlib_comp_ops = {
NULL, NULL, NULL, ZLIB_COMPRESSION, "zlib", 0 NULL, NULL, NULL, NULL, ZLIB_COMPRESSION, "zlib", 0
}; };
#endif #endif
static const struct squashfs_decompressor squashfs_unknown_comp_ops = { static const struct squashfs_decompressor squashfs_unknown_comp_ops = {
NULL, NULL, NULL, 0, "unknown", 0 NULL, NULL, NULL, NULL, 0, "unknown", 0
}; };
static const struct squashfs_decompressor *decompressor[] = { static const struct squashfs_decompressor *decompressor[] = {
...@@ -83,10 +83,10 @@ const struct squashfs_decompressor *squashfs_lookup_decompressor(int id) ...@@ -83,10 +83,10 @@ const struct squashfs_decompressor *squashfs_lookup_decompressor(int id)
} }
void *squashfs_decompressor_init(struct super_block *sb, unsigned short flags) static void *get_comp_opts(struct super_block *sb, unsigned short flags)
{ {
struct squashfs_sb_info *msblk = sb->s_fs_info; struct squashfs_sb_info *msblk = sb->s_fs_info;
void *strm, *buffer = NULL; void *buffer = NULL, *comp_opts;
int length = 0; int length = 0;
/* /*
...@@ -94,23 +94,40 @@ void *squashfs_decompressor_init(struct super_block *sb, unsigned short flags) ...@@ -94,23 +94,40 @@ void *squashfs_decompressor_init(struct super_block *sb, unsigned short flags)
*/ */
if (SQUASHFS_COMP_OPTS(flags)) { if (SQUASHFS_COMP_OPTS(flags)) {
buffer = kmalloc(PAGE_CACHE_SIZE, GFP_KERNEL); buffer = kmalloc(PAGE_CACHE_SIZE, GFP_KERNEL);
if (buffer == NULL) if (buffer == NULL) {
return ERR_PTR(-ENOMEM); comp_opts = ERR_PTR(-ENOMEM);
goto out;
}
length = squashfs_read_data(sb, &buffer, length = squashfs_read_data(sb, &buffer,
sizeof(struct squashfs_super_block), 0, NULL, sizeof(struct squashfs_super_block), 0, NULL,
PAGE_CACHE_SIZE, 1); PAGE_CACHE_SIZE, 1);
if (length < 0) { if (length < 0) {
strm = ERR_PTR(length); comp_opts = ERR_PTR(length);
goto finished; goto out;
} }
} }
strm = msblk->decompressor->init(msblk, buffer, length); comp_opts = squashfs_comp_opts(msblk, buffer, length);
finished: out:
kfree(buffer); kfree(buffer);
return comp_opts;
}
void *squashfs_decompressor_setup(struct super_block *sb, unsigned short flags)
{
struct squashfs_sb_info *msblk = sb->s_fs_info;
void *stream, *comp_opts = get_comp_opts(sb, flags);
if (IS_ERR(comp_opts))
return comp_opts;
stream = squashfs_decompressor_create(msblk, comp_opts);
if (IS_ERR(stream))
kfree(comp_opts);
return strm; return stream;
} }
...@@ -24,28 +24,21 @@ ...@@ -24,28 +24,21 @@
*/ */
struct squashfs_decompressor { struct squashfs_decompressor {
void *(*init)(struct squashfs_sb_info *, void *, int); void *(*init)(struct squashfs_sb_info *, void *);
void *(*comp_opts)(struct squashfs_sb_info *, void *, int);
void (*free)(void *); void (*free)(void *);
int (*decompress)(struct squashfs_sb_info *, void **, int (*decompress)(struct squashfs_sb_info *, void *, void **,
struct buffer_head **, int, int, int, int, int); struct buffer_head **, int, int, int, int, int);
int id; int id;
char *name; char *name;
int supported; int supported;
}; };
static inline void squashfs_decompressor_free(struct squashfs_sb_info *msblk, static inline void *squashfs_comp_opts(struct squashfs_sb_info *msblk,
void *s) void *buff, int length)
{ {
if (msblk->decompressor) return msblk->decompressor->comp_opts ?
msblk->decompressor->free(s); msblk->decompressor->comp_opts(msblk, buff, length) : NULL;
}
static inline int squashfs_decompress(struct squashfs_sb_info *msblk,
void **buffer, struct buffer_head **bh, int b, int offset, int length,
int srclength, int pages)
{
return msblk->decompressor->decompress(msblk, buffer, bh, b, offset,
length, srclength, pages);
} }
#ifdef CONFIG_SQUASHFS_XZ #ifdef CONFIG_SQUASHFS_XZ
......
/*
* Copyright (c) 2013
* Phillip Lougher <phillip@squashfs.org.uk>
*
* This work is licensed under the terms of the GNU GPL, version 2. See
* the COPYING file in the top-level directory.
*/
#include <linux/types.h>
#include <linux/mutex.h>
#include <linux/slab.h>
#include <linux/buffer_head.h>
#include "squashfs_fs.h"
#include "squashfs_fs_sb.h"
#include "decompressor.h"
#include "squashfs.h"
/*
* This file implements single-threaded decompression in the
* decompressor framework
*/
struct squashfs_stream {
void *stream;
struct mutex mutex;
};
void *squashfs_decompressor_create(struct squashfs_sb_info *msblk,
void *comp_opts)
{
struct squashfs_stream *stream;
int err = -ENOMEM;
stream = kmalloc(sizeof(*stream), GFP_KERNEL);
if (stream == NULL)
goto out;
stream->stream = msblk->decompressor->init(msblk, comp_opts);
if (IS_ERR(stream->stream)) {
err = PTR_ERR(stream->stream);
goto out;
}
kfree(comp_opts);
mutex_init(&stream->mutex);
return stream;
out:
kfree(stream);
return ERR_PTR(err);
}
void squashfs_decompressor_destroy(struct squashfs_sb_info *msblk)
{
struct squashfs_stream *stream = msblk->stream;
if (stream) {
msblk->decompressor->free(stream->stream);
kfree(stream);
}
}
int squashfs_decompress(struct squashfs_sb_info *msblk,
void **buffer, struct buffer_head **bh, int b, int offset, int length,
int srclength, int pages)
{
int res;
struct squashfs_stream *stream = msblk->stream;
mutex_lock(&stream->mutex);
res = msblk->decompressor->decompress(msblk, stream->stream, buffer,
bh, b, offset, length, srclength, pages);
mutex_unlock(&stream->mutex);
if (res < 0)
ERROR("%s decompression failed, data probably corrupt\n",
msblk->decompressor->name);
return res;
}
int squashfs_max_decompressors(void)
{
return 1;
}
...@@ -37,7 +37,7 @@ struct squashfs_lzo { ...@@ -37,7 +37,7 @@ struct squashfs_lzo {
void *output; void *output;
}; };
static void *lzo_init(struct squashfs_sb_info *msblk, void *buff, int len) static void *lzo_init(struct squashfs_sb_info *msblk, void *buff)
{ {
int block_size = max_t(int, msblk->block_size, SQUASHFS_METADATA_SIZE); int block_size = max_t(int, msblk->block_size, SQUASHFS_METADATA_SIZE);
...@@ -74,22 +74,16 @@ static void lzo_free(void *strm) ...@@ -74,22 +74,16 @@ static void lzo_free(void *strm)
} }
static int lzo_uncompress(struct squashfs_sb_info *msblk, void **buffer, static int lzo_uncompress(struct squashfs_sb_info *msblk, void *strm,
struct buffer_head **bh, int b, int offset, int length, int srclength, void **buffer, struct buffer_head **bh, int b, int offset, int length,
int pages) int srclength, int pages)
{ {
struct squashfs_lzo *stream = msblk->stream; struct squashfs_lzo *stream = strm;
void *buff = stream->input; void *buff = stream->input;
int avail, i, bytes = length, res; int avail, i, bytes = length, res;
size_t out_len = srclength; size_t out_len = srclength;
mutex_lock(&msblk->read_data_mutex);
for (i = 0; i < b; i++) { for (i = 0; i < b; i++) {
wait_on_buffer(bh[i]);
if (!buffer_uptodate(bh[i]))
goto block_release;
avail = min(bytes, msblk->devblksize - offset); avail = min(bytes, msblk->devblksize - offset);
memcpy(buff, bh[i]->b_data + offset, avail); memcpy(buff, bh[i]->b_data + offset, avail);
buff += avail; buff += avail;
...@@ -111,17 +105,9 @@ static int lzo_uncompress(struct squashfs_sb_info *msblk, void **buffer, ...@@ -111,17 +105,9 @@ static int lzo_uncompress(struct squashfs_sb_info *msblk, void **buffer,
bytes -= avail; bytes -= avail;
} }
mutex_unlock(&msblk->read_data_mutex);
return res; return res;
block_release:
for (; i < b; i++)
put_bh(bh[i]);
failed: failed:
mutex_unlock(&msblk->read_data_mutex);
ERROR("lzo decompression failed, data probably corrupt\n");
return -EIO; return -EIO;
} }
......
...@@ -48,7 +48,14 @@ extern void *squashfs_read_table(struct super_block *, u64, int); ...@@ -48,7 +48,14 @@ extern void *squashfs_read_table(struct super_block *, u64, int);
/* decompressor.c */ /* decompressor.c */
extern const struct squashfs_decompressor *squashfs_lookup_decompressor(int); extern const struct squashfs_decompressor *squashfs_lookup_decompressor(int);
extern void *squashfs_decompressor_init(struct super_block *, unsigned short); extern void *squashfs_decompressor_setup(struct super_block *, unsigned short);
/* decompressor_xxx.c */
extern void *squashfs_decompressor_create(struct squashfs_sb_info *, void *);
extern void squashfs_decompressor_destroy(struct squashfs_sb_info *);
extern int squashfs_decompress(struct squashfs_sb_info *, void **,
struct buffer_head **, int, int, int, int, int);
extern int squashfs_max_decompressors(void);
/* export.c */ /* export.c */
extern __le64 *squashfs_read_inode_lookup_table(struct super_block *, u64, u64, extern __le64 *squashfs_read_inode_lookup_table(struct super_block *, u64, u64,
......
...@@ -63,10 +63,9 @@ struct squashfs_sb_info { ...@@ -63,10 +63,9 @@ struct squashfs_sb_info {
__le64 *id_table; __le64 *id_table;
__le64 *fragment_index; __le64 *fragment_index;
__le64 *xattr_id_table; __le64 *xattr_id_table;
struct mutex read_data_mutex;
struct mutex meta_index_mutex; struct mutex meta_index_mutex;
struct meta_index *meta_index; struct meta_index *meta_index;
void *stream; struct squashfs_stream *stream;
__le64 *inode_lookup_table; __le64 *inode_lookup_table;
u64 inode_table; u64 inode_table;
u64 directory_table; u64 directory_table;
......
...@@ -98,7 +98,6 @@ static int squashfs_fill_super(struct super_block *sb, void *data, int silent) ...@@ -98,7 +98,6 @@ static int squashfs_fill_super(struct super_block *sb, void *data, int silent)
msblk->devblksize = sb_min_blocksize(sb, SQUASHFS_DEVBLK_SIZE); msblk->devblksize = sb_min_blocksize(sb, SQUASHFS_DEVBLK_SIZE);
msblk->devblksize_log2 = ffz(~msblk->devblksize); msblk->devblksize_log2 = ffz(~msblk->devblksize);
mutex_init(&msblk->read_data_mutex);
mutex_init(&msblk->meta_index_mutex); mutex_init(&msblk->meta_index_mutex);
/* /*
...@@ -206,13 +205,14 @@ static int squashfs_fill_super(struct super_block *sb, void *data, int silent) ...@@ -206,13 +205,14 @@ static int squashfs_fill_super(struct super_block *sb, void *data, int silent)
goto failed_mount; goto failed_mount;
/* Allocate read_page block */ /* Allocate read_page block */
msblk->read_page = squashfs_cache_init("data", 1, msblk->block_size); msblk->read_page = squashfs_cache_init("data",
squashfs_max_decompressors(), msblk->block_size);
if (msblk->read_page == NULL) { if (msblk->read_page == NULL) {
ERROR("Failed to allocate read_page block\n"); ERROR("Failed to allocate read_page block\n");
goto failed_mount; goto failed_mount;
} }
msblk->stream = squashfs_decompressor_init(sb, flags); msblk->stream = squashfs_decompressor_setup(sb, flags);
if (IS_ERR(msblk->stream)) { if (IS_ERR(msblk->stream)) {
err = PTR_ERR(msblk->stream); err = PTR_ERR(msblk->stream);
msblk->stream = NULL; msblk->stream = NULL;
...@@ -336,7 +336,7 @@ static int squashfs_fill_super(struct super_block *sb, void *data, int silent) ...@@ -336,7 +336,7 @@ static int squashfs_fill_super(struct super_block *sb, void *data, int silent)
squashfs_cache_delete(msblk->block_cache); squashfs_cache_delete(msblk->block_cache);
squashfs_cache_delete(msblk->fragment_cache); squashfs_cache_delete(msblk->fragment_cache);
squashfs_cache_delete(msblk->read_page); squashfs_cache_delete(msblk->read_page);
squashfs_decompressor_free(msblk, msblk->stream); squashfs_decompressor_destroy(msblk);
kfree(msblk->inode_lookup_table); kfree(msblk->inode_lookup_table);
kfree(msblk->fragment_index); kfree(msblk->fragment_index);
kfree(msblk->id_table); kfree(msblk->id_table);
...@@ -383,7 +383,7 @@ static void squashfs_put_super(struct super_block *sb) ...@@ -383,7 +383,7 @@ static void squashfs_put_super(struct super_block *sb)
squashfs_cache_delete(sbi->block_cache); squashfs_cache_delete(sbi->block_cache);
squashfs_cache_delete(sbi->fragment_cache); squashfs_cache_delete(sbi->fragment_cache);
squashfs_cache_delete(sbi->read_page); squashfs_cache_delete(sbi->read_page);
squashfs_decompressor_free(sbi, sbi->stream); squashfs_decompressor_destroy(sbi);
kfree(sbi->id_table); kfree(sbi->id_table);
kfree(sbi->fragment_index); kfree(sbi->fragment_index);
kfree(sbi->meta_index); kfree(sbi->meta_index);
......
...@@ -38,38 +38,63 @@ struct squashfs_xz { ...@@ -38,38 +38,63 @@ struct squashfs_xz {
struct xz_buf buf; struct xz_buf buf;
}; };
struct comp_opts { struct disk_comp_opts {
__le32 dictionary_size; __le32 dictionary_size;
__le32 flags; __le32 flags;
}; };
static void *squashfs_xz_init(struct squashfs_sb_info *msblk, void *buff, struct comp_opts {
int len) int dict_size;
};
static void *squashfs_xz_comp_opts(struct squashfs_sb_info *msblk,
void *buff, int len)
{ {
struct comp_opts *comp_opts = buff; struct disk_comp_opts *comp_opts = buff;
struct squashfs_xz *stream; struct comp_opts *opts;
int dict_size = msblk->block_size; int err = 0, n;
int err, n;
opts = kmalloc(sizeof(*opts), GFP_KERNEL);
if (opts == NULL) {
err = -ENOMEM;
goto out2;
}
if (comp_opts) { if (comp_opts) {
/* check compressor options are the expected length */ /* check compressor options are the expected length */
if (len < sizeof(*comp_opts)) { if (len < sizeof(*comp_opts)) {
err = -EIO; err = -EIO;
goto failed; goto out;
} }
dict_size = le32_to_cpu(comp_opts->dictionary_size); opts->dict_size = le32_to_cpu(comp_opts->dictionary_size);
/* the dictionary size should be 2^n or 2^n+2^(n+1) */ /* the dictionary size should be 2^n or 2^n+2^(n+1) */
n = ffs(dict_size) - 1; n = ffs(opts->dict_size) - 1;
if (dict_size != (1 << n) && dict_size != (1 << n) + if (opts->dict_size != (1 << n) && opts->dict_size != (1 << n) +
(1 << (n + 1))) { (1 << (n + 1))) {
err = -EIO; err = -EIO;
goto failed; goto out;
} }
} } else
/* use defaults */
opts->dict_size = max_t(int, msblk->block_size,
SQUASHFS_METADATA_SIZE);
dict_size = max_t(int, dict_size, SQUASHFS_METADATA_SIZE); return opts;
out:
kfree(opts);
out2:
return ERR_PTR(err);
}
static void *squashfs_xz_init(struct squashfs_sb_info *msblk, void *buff)
{
struct comp_opts *comp_opts = buff;
struct squashfs_xz *stream;
int err;
stream = kmalloc(sizeof(*stream), GFP_KERNEL); stream = kmalloc(sizeof(*stream), GFP_KERNEL);
if (stream == NULL) { if (stream == NULL) {
...@@ -77,7 +102,7 @@ static void *squashfs_xz_init(struct squashfs_sb_info *msblk, void *buff, ...@@ -77,7 +102,7 @@ static void *squashfs_xz_init(struct squashfs_sb_info *msblk, void *buff,
goto failed; goto failed;
} }
stream->state = xz_dec_init(XZ_PREALLOC, dict_size); stream->state = xz_dec_init(XZ_PREALLOC, comp_opts->dict_size);
if (stream->state == NULL) { if (stream->state == NULL) {
kfree(stream); kfree(stream);
err = -ENOMEM; err = -ENOMEM;
...@@ -103,15 +128,13 @@ static void squashfs_xz_free(void *strm) ...@@ -103,15 +128,13 @@ static void squashfs_xz_free(void *strm)
} }
static int squashfs_xz_uncompress(struct squashfs_sb_info *msblk, void **buffer, static int squashfs_xz_uncompress(struct squashfs_sb_info *msblk, void *strm,
struct buffer_head **bh, int b, int offset, int length, int srclength, void **buffer, struct buffer_head **bh, int b, int offset, int length,
int pages) int srclength, int pages)
{ {
enum xz_ret xz_err; enum xz_ret xz_err;
int avail, total = 0, k = 0, page = 0; int avail, total = 0, k = 0, page = 0;
struct squashfs_xz *stream = msblk->stream; struct squashfs_xz *stream = strm;
mutex_lock(&msblk->read_data_mutex);
xz_dec_reset(stream->state); xz_dec_reset(stream->state);
stream->buf.in_pos = 0; stream->buf.in_pos = 0;
...@@ -124,10 +147,6 @@ static int squashfs_xz_uncompress(struct squashfs_sb_info *msblk, void **buffer, ...@@ -124,10 +147,6 @@ static int squashfs_xz_uncompress(struct squashfs_sb_info *msblk, void **buffer,
if (stream->buf.in_pos == stream->buf.in_size && k < b) { if (stream->buf.in_pos == stream->buf.in_size && k < b) {
avail = min(length, msblk->devblksize - offset); avail = min(length, msblk->devblksize - offset);
length -= avail; length -= avail;
wait_on_buffer(bh[k]);
if (!buffer_uptodate(bh[k]))
goto release_mutex;
stream->buf.in = bh[k]->b_data + offset; stream->buf.in = bh[k]->b_data + offset;
stream->buf.in_size = avail; stream->buf.in_size = avail;
stream->buf.in_pos = 0; stream->buf.in_pos = 0;
...@@ -147,23 +166,12 @@ static int squashfs_xz_uncompress(struct squashfs_sb_info *msblk, void **buffer, ...@@ -147,23 +166,12 @@ static int squashfs_xz_uncompress(struct squashfs_sb_info *msblk, void **buffer,
put_bh(bh[k++]); put_bh(bh[k++]);
} while (xz_err == XZ_OK); } while (xz_err == XZ_OK);
if (xz_err != XZ_STREAM_END) { if (xz_err != XZ_STREAM_END || k < b)
ERROR("xz_dec_run error, data probably corrupt\n"); goto out;
goto release_mutex;
}
if (k < b) {
ERROR("xz_uncompress error, input remaining\n");
goto release_mutex;
}
total += stream->buf.out_pos;
mutex_unlock(&msblk->read_data_mutex);
return total;
release_mutex: return total + stream->buf.out_pos;
mutex_unlock(&msblk->read_data_mutex);
out:
for (; k < b; k++) for (; k < b; k++)
put_bh(bh[k]); put_bh(bh[k]);
...@@ -172,6 +180,7 @@ static int squashfs_xz_uncompress(struct squashfs_sb_info *msblk, void **buffer, ...@@ -172,6 +180,7 @@ static int squashfs_xz_uncompress(struct squashfs_sb_info *msblk, void **buffer,
const struct squashfs_decompressor squashfs_xz_comp_ops = { const struct squashfs_decompressor squashfs_xz_comp_ops = {
.init = squashfs_xz_init, .init = squashfs_xz_init,
.comp_opts = squashfs_xz_comp_opts,
.free = squashfs_xz_free, .free = squashfs_xz_free,
.decompress = squashfs_xz_uncompress, .decompress = squashfs_xz_uncompress,
.id = XZ_COMPRESSION, .id = XZ_COMPRESSION,
......
...@@ -33,7 +33,7 @@ ...@@ -33,7 +33,7 @@
#include "squashfs.h" #include "squashfs.h"
#include "decompressor.h" #include "decompressor.h"
static void *zlib_init(struct squashfs_sb_info *dummy, void *buff, int len) static void *zlib_init(struct squashfs_sb_info *dummy, void *buff)
{ {
z_stream *stream = kmalloc(sizeof(z_stream), GFP_KERNEL); z_stream *stream = kmalloc(sizeof(z_stream), GFP_KERNEL);
if (stream == NULL) if (stream == NULL)
...@@ -61,15 +61,13 @@ static void zlib_free(void *strm) ...@@ -61,15 +61,13 @@ static void zlib_free(void *strm)
} }
static int zlib_uncompress(struct squashfs_sb_info *msblk, void **buffer, static int zlib_uncompress(struct squashfs_sb_info *msblk, void *strm,
struct buffer_head **bh, int b, int offset, int length, int srclength, void **buffer, struct buffer_head **bh, int b, int offset, int length,
int pages) int srclength, int pages)
{ {
int zlib_err, zlib_init = 0; int zlib_err, zlib_init = 0;
int k = 0, page = 0; int k = 0, page = 0;
z_stream *stream = msblk->stream; z_stream *stream = strm;
mutex_lock(&msblk->read_data_mutex);
stream->avail_out = 0; stream->avail_out = 0;
stream->avail_in = 0; stream->avail_in = 0;
...@@ -78,10 +76,6 @@ static int zlib_uncompress(struct squashfs_sb_info *msblk, void **buffer, ...@@ -78,10 +76,6 @@ static int zlib_uncompress(struct squashfs_sb_info *msblk, void **buffer,
if (stream->avail_in == 0 && k < b) { if (stream->avail_in == 0 && k < b) {
int avail = min(length, msblk->devblksize - offset); int avail = min(length, msblk->devblksize - offset);
length -= avail; length -= avail;
wait_on_buffer(bh[k]);
if (!buffer_uptodate(bh[k]))
goto release_mutex;
stream->next_in = bh[k]->b_data + offset; stream->next_in = bh[k]->b_data + offset;
stream->avail_in = avail; stream->avail_in = avail;
offset = 0; offset = 0;
...@@ -94,12 +88,8 @@ static int zlib_uncompress(struct squashfs_sb_info *msblk, void **buffer, ...@@ -94,12 +88,8 @@ static int zlib_uncompress(struct squashfs_sb_info *msblk, void **buffer,
if (!zlib_init) { if (!zlib_init) {
zlib_err = zlib_inflateInit(stream); zlib_err = zlib_inflateInit(stream);
if (zlib_err != Z_OK) { if (zlib_err != Z_OK)
ERROR("zlib_inflateInit returned unexpected " goto out;
"result 0x%x, srclength %d\n",
zlib_err, srclength);
goto release_mutex;
}
zlib_init = 1; zlib_init = 1;
} }
...@@ -109,29 +99,19 @@ static int zlib_uncompress(struct squashfs_sb_info *msblk, void **buffer, ...@@ -109,29 +99,19 @@ static int zlib_uncompress(struct squashfs_sb_info *msblk, void **buffer,
put_bh(bh[k++]); put_bh(bh[k++]);
} while (zlib_err == Z_OK); } while (zlib_err == Z_OK);
if (zlib_err != Z_STREAM_END) { if (zlib_err != Z_STREAM_END)
ERROR("zlib_inflate error, data probably corrupt\n"); goto out;
goto release_mutex;
}
zlib_err = zlib_inflateEnd(stream); zlib_err = zlib_inflateEnd(stream);
if (zlib_err != Z_OK) { if (zlib_err != Z_OK)
ERROR("zlib_inflate error, data probably corrupt\n"); goto out;
goto release_mutex;
}
if (k < b) {
ERROR("zlib_uncompress error, data remaining\n");
goto release_mutex;
}
length = stream->total_out; if (k < b)
mutex_unlock(&msblk->read_data_mutex); goto out;
return length;
release_mutex: return stream->total_out;
mutex_unlock(&msblk->read_data_mutex);
out:
for (; k < b; k++) for (; k < b; k++)
put_bh(bh[k]); put_bh(bh[k]);
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment