11da177e4SLinus Torvalds /* 21da177e4SLinus Torvalds * uncompress.c 31da177e4SLinus Torvalds * 41da177e4SLinus Torvalds * (C) Copyright 1999 Linus Torvalds 51da177e4SLinus Torvalds * 61da177e4SLinus Torvalds * cramfs interfaces to the uncompression library. There's really just 71da177e4SLinus Torvalds * three entrypoints: 81da177e4SLinus Torvalds * 91da177e4SLinus Torvalds * - cramfs_uncompress_init() - called to initialize the thing. 101da177e4SLinus Torvalds * - cramfs_uncompress_exit() - tell me when you're done 111da177e4SLinus Torvalds * - cramfs_uncompress_block() - uncompress a block. 121da177e4SLinus Torvalds * 131da177e4SLinus Torvalds * NOTE NOTE NOTE! The uncompression is entirely single-threaded. We 141da177e4SLinus Torvalds * only have one stream, and we'll initialize it only once even if it 151da177e4SLinus Torvalds * then is used by multiple filesystems. 161da177e4SLinus Torvalds */ 171da177e4SLinus Torvalds 18*4f21e1eaSFabian Frederick #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt 19*4f21e1eaSFabian Frederick 201da177e4SLinus Torvalds #include <linux/kernel.h> 211da177e4SLinus Torvalds #include <linux/errno.h> 221da177e4SLinus Torvalds #include <linux/vmalloc.h> 231da177e4SLinus Torvalds #include <linux/zlib.h> 24f7f4f4ddSAl Viro #include "internal.h" 251da177e4SLinus Torvalds 261da177e4SLinus Torvalds static z_stream stream; 271da177e4SLinus Torvalds static int initialized; 281da177e4SLinus Torvalds 291da177e4SLinus Torvalds /* Returns length of decompressed data. */ 301da177e4SLinus Torvalds int cramfs_uncompress_block(void *dst, int dstlen, void *src, int srclen) 311da177e4SLinus Torvalds { 321da177e4SLinus Torvalds int err; 331da177e4SLinus Torvalds 341da177e4SLinus Torvalds stream.next_in = src; 351da177e4SLinus Torvalds stream.avail_in = srclen; 361da177e4SLinus Torvalds 371da177e4SLinus Torvalds stream.next_out = dst; 381da177e4SLinus Torvalds stream.avail_out = dstlen; 391da177e4SLinus Torvalds 401da177e4SLinus Torvalds err = zlib_inflateReset(&stream); 411da177e4SLinus Torvalds if (err != Z_OK) { 42f175ff81SFabian Frederick pr_err("zlib_inflateReset error %d\n", err); 431da177e4SLinus Torvalds zlib_inflateEnd(&stream); 441da177e4SLinus Torvalds zlib_inflateInit(&stream); 451da177e4SLinus Torvalds } 461da177e4SLinus Torvalds 471da177e4SLinus Torvalds err = zlib_inflate(&stream, Z_FINISH); 481da177e4SLinus Torvalds if (err != Z_STREAM_END) 491da177e4SLinus Torvalds goto err; 501da177e4SLinus Torvalds return stream.total_out; 511da177e4SLinus Torvalds 521da177e4SLinus Torvalds err: 53f175ff81SFabian Frederick pr_err("Error %d while decompressing!\n", err); 54f175ff81SFabian Frederick pr_err("%p(%d)->%p(%d)\n", src, srclen, dst, dstlen); 5598310e58SDavid VomLehn return -EIO; 561da177e4SLinus Torvalds } 571da177e4SLinus Torvalds 581da177e4SLinus Torvalds int cramfs_uncompress_init(void) 591da177e4SLinus Torvalds { 601da177e4SLinus Torvalds if (!initialized++) { 611da177e4SLinus Torvalds stream.workspace = vmalloc(zlib_inflate_workspacesize()); 621da177e4SLinus Torvalds if ( !stream.workspace ) { 631da177e4SLinus Torvalds initialized = 0; 641da177e4SLinus Torvalds return -ENOMEM; 651da177e4SLinus Torvalds } 661da177e4SLinus Torvalds stream.next_in = NULL; 671da177e4SLinus Torvalds stream.avail_in = 0; 681da177e4SLinus Torvalds zlib_inflateInit(&stream); 691da177e4SLinus Torvalds } 701da177e4SLinus Torvalds return 0; 711da177e4SLinus Torvalds } 721da177e4SLinus Torvalds 73368bdb3dSAlexey Dobriyan void cramfs_uncompress_exit(void) 741da177e4SLinus Torvalds { 751da177e4SLinus Torvalds if (!--initialized) { 761da177e4SLinus Torvalds zlib_inflateEnd(&stream); 771da177e4SLinus Torvalds vfree(stream.workspace); 781da177e4SLinus Torvalds } 791da177e4SLinus Torvalds } 80