]>
Commit | Line | Data |
---|---|---|
dd875c76 WD |
1 | /* |
2 | * uncompress.c | |
3 | * | |
4 | * Copyright (C) 1999 Linus Torvalds | |
5 | * Copyright (C) 2000-2002 Transmeta Corporation | |
6 | * | |
7 | * This program is free software; you can redistribute it and/or modify | |
8 | * it under the terms of the GNU General Public License (Version 2) as | |
9 | * published by the Free Software Foundation. | |
10 | * | |
11 | * cramfs interfaces to the uncompression library. There's really just | |
12 | * three entrypoints: | |
13 | * | |
14 | * - cramfs_uncompress_init() - called to initialize the thing. | |
15 | * - cramfs_uncompress_exit() - tell me when you're done | |
16 | * - cramfs_uncompress_block() - uncompress a block. | |
17 | * | |
18 | * NOTE NOTE NOTE! The uncompression is entirely single-threaded. We | |
19 | * only have one stream, and we'll initialize it only once even if it | |
20 | * then is used by multiple filesystems. | |
21 | */ | |
22 | ||
23 | #include <common.h> | |
24 | #include <malloc.h> | |
25 | #include <watchdog.h> | |
26 | #include <zlib.h> | |
27 | ||
28 | #if (CONFIG_COMMANDS & CFG_CMD_JFFS2) | |
29 | ||
30 | static z_stream stream; | |
31 | ||
32 | #define ZALLOC_ALIGNMENT 16 | |
33 | ||
34 | static void *zalloc (void *x, unsigned items, unsigned size) | |
35 | { | |
36 | void *p; | |
37 | ||
38 | size *= items; | |
39 | size = (size + ZALLOC_ALIGNMENT - 1) & ~(ZALLOC_ALIGNMENT - 1); | |
40 | ||
41 | p = malloc (size); | |
42 | ||
43 | return (p); | |
44 | } | |
45 | ||
46 | static void zfree (void *x, void *addr, unsigned nb) | |
47 | { | |
48 | free (addr); | |
49 | } | |
50 | ||
51 | /* Returns length of decompressed data. */ | |
52 | int cramfs_uncompress_block (void *dst, void *src, int srclen) | |
53 | { | |
54 | int err; | |
55 | ||
56 | inflateReset (&stream); | |
57 | ||
58 | stream.next_in = src; | |
59 | stream.avail_in = srclen; | |
60 | ||
61 | stream.next_out = dst; | |
62 | stream.avail_out = 4096 * 2; | |
63 | ||
64 | err = inflate (&stream, Z_FINISH); | |
65 | ||
66 | if (err != Z_STREAM_END) | |
67 | goto err; | |
68 | return stream.total_out; | |
69 | ||
70 | err: | |
71 | /*printf ("Error %d while decompressing!\n", err); */ | |
72 | /*printf ("%p(%d)->%p\n", src, srclen, dst); */ | |
73 | return -1; | |
74 | } | |
75 | ||
76 | int cramfs_uncompress_init (void) | |
77 | { | |
78 | int err; | |
79 | ||
80 | stream.zalloc = zalloc; | |
81 | stream.zfree = zfree; | |
82 | stream.next_in = 0; | |
83 | stream.avail_in = 0; | |
84 | ||
85 | #if defined(CONFIG_HW_WATCHDOG) || defined(CONFIG_WATCHDOG) | |
86 | stream.outcb = (cb_func) WATCHDOG_RESET; | |
87 | #else | |
88 | stream.outcb = Z_NULL; | |
89 | #endif /* CONFIG_HW_WATCHDOG */ | |
90 | ||
91 | err = inflateInit (&stream); | |
92 | if (err != Z_OK) { | |
93 | printf ("Error: inflateInit2() returned %d\n", err); | |
94 | return -1; | |
95 | } | |
96 | ||
97 | return 0; | |
98 | } | |
99 | ||
100 | int cramfs_uncompress_exit (void) | |
101 | { | |
102 | inflateEnd (&stream); | |
103 | return 0; | |
104 | } | |
105 | ||
106 | #endif /* CFG_FS_CRAMFS */ |