1 /******************************************************************************
3 ** FILE NAME : LzmaWrapper.c
4 ** PROJECT : bootloader
9 ** DESCRIPTION : LZMA decoder support for U-boot 1.1.5
10 ** COPYRIGHT : Copyright (c) 2006
11 ** Infineon Technologies AG
12 ** Am Campeon 1-12, 85579 Neubiberg, Germany
14 ** This program is free software; you can redistribute it and/or modify
15 ** it under the terms of the GNU General Public License as published by
16 ** the Free Software Foundation; either version 2 of the License, or
17 ** (at your option) any later version.
20 ** $Date $Author $Comment
21 ** 2 Nov 2006 Lin Mars init version which derived from LzmaTest.c from
23 ** 24 May 2007 Lin Mars Fix issue for multiple lzma_inflate involved
24 *******************************************************************************/
29 #include "LzmaDecode.h"
30 #include "LzmaWrapper.h"
32 #if defined(DEBUG_ENABLE_BOOTSTRAP_PRINTF) || !defined(CFG_BOOTSTRAP_CODE)
33 static const char *kCantReadMessage
= "Can not read from source buffer";
34 static const char *kCantAllocateMessage
= "Not enough buffer for decompression";
37 static size_t rpos
=0, dpos
=0;
39 static int MyReadFileAndCheck(unsigned char *src
, void *dest
, size_t size
)
43 memcpy(dest
, src
+ rpos
, size
);
48 int lzma_inflate(unsigned char *source
, int s_len
, unsigned char *dest
, int *d_len
)
50 /* We use two 32-bit integers to construct 64-bit integer for file size.
51 You can remove outSizeHigh, if you don't need >= 4GB supporting,
52 or you can use UInt64 outSize, if your compiler supports 64-bit integers*/
54 UInt32 outSizeHigh
= 0;
56 unsigned char *outStream
;
59 /* waitEOS = 1, if there is no uncompressed size in headers,
60 so decoder will wait EOS (End of Stream Marker) in compressed stream */
63 unsigned char *inStream
;
65 CLzmaDecoderState state
; /* it's about 24-80 bytes structure, if int is 32-bit */
66 unsigned char properties
[LZMA_PROPERTIES_SIZE
];
72 if (sizeof(UInt32
) < 4)
74 #if defined(DEBUG_ENABLE_BOOTSTRAP_PRINTF) || !defined(CFG_BOOTSTRAP_CODE)
75 printf("LZMA decoder needs correct UInt32\n");
77 return LZMA_RESULT_DATA_ERROR
;
82 if ((long)(SizeT
)length
!= length
)
84 #if defined(DEBUG_ENABLE_BOOTSTRAP_PRINTF) || !defined(CFG_BOOTSTRAP_CODE)
85 printf("Too big compressed stream\n");
87 return LZMA_RESULT_DATA_ERROR
;
89 compressedSize
= (SizeT
)(length
- (LZMA_PROPERTIES_SIZE
+ 8));
92 /* Read LZMA properties for compressed stream */
94 if (!MyReadFileAndCheck(source
, properties
, sizeof(properties
)))
96 #if defined(DEBUG_ENABLE_BOOTSTRAP_PRINTF) || !defined(CFG_BOOTSTRAP_CODE)
97 printf("%s\n", kCantReadMessage
);
99 return LZMA_RESULT_DATA_ERROR
;
102 /* Read uncompressed size */
105 for (i
= 0; i
< 8; i
++)
108 if (!MyReadFileAndCheck(source
, &b
, 1))
110 #if defined(DEBUG_ENABLE_BOOTSTRAP_PRINTF) || !defined(CFG_BOOTSTRAP_CODE)
111 printf("%s\n", kCantReadMessage
);
113 return LZMA_RESULT_DATA_ERROR
;
118 outSize
+= (UInt32
)(b
) << (i
* 8);
120 outSizeHigh
+= (UInt32
)(b
) << ((i
- 4) * 8);
125 #if defined(DEBUG_ENABLE_BOOTSTRAP_PRINTF) || !defined(CFG_BOOTSTRAP_CODE)
126 printf("Stream with EOS marker is not supported");
128 return LZMA_RESULT_DATA_ERROR
;
130 outSizeFull
= (SizeT
)outSize
;
131 if (sizeof(SizeT
) >= 8)
132 outSizeFull
|= (((SizeT
)outSizeHigh
<< 16) << 16);
133 else if (outSizeHigh
!= 0 || (UInt32
)(SizeT
)outSize
!= outSize
)
135 #if defined(DEBUG_ENABLE_BOOTSTRAP_PRINTF) || !defined(CFG_BOOTSTRAP_CODE)
136 printf("Too big uncompressed stream");
138 return LZMA_RESULT_DATA_ERROR
;
142 /* Decode LZMA properties and allocate memory */
143 if (LzmaDecodeProperties(&state
.Properties
, properties
, LZMA_PROPERTIES_SIZE
) != LZMA_RESULT_OK
)
145 #if defined(DEBUG_ENABLE_BOOTSTRAP_PRINTF) || !defined(CFG_BOOTSTRAP_CODE)
146 printf("Incorrect stream properties");
148 return LZMA_RESULT_DATA_ERROR
;
150 state
.Probs
= (CProb
*)malloc(LzmaGetNumProbs(&state
.Properties
) * sizeof(CProb
));
152 if (outSizeFull
== 0)
156 if (outSizeFull
> d_len
)
162 if (compressedSize
== 0)
166 if ((compressedSize
+rpos
) > s_len
)
169 inStream
= source
+ rpos
;
173 || (outStream
== 0 && outSizeFull
!= 0)
174 || (inStream
== 0 && compressedSize
!= 0)
178 #if defined(DEBUG_ENABLE_BOOTSTRAP_PRINTF) || !defined(CFG_BOOTSTRAP_CODE)
179 printf("%s\n", kCantAllocateMessage
);
181 return LZMA_RESULT_DATA_ERROR
;
188 res
= LzmaDecode(&state
,
189 inStream
, compressedSize
, &inProcessed
,
190 outStream
, outSizeFull
, &outProcessed
);
193 #if defined(DEBUG_ENABLE_BOOTSTRAP_PRINTF) || !defined(CFG_BOOTSTRAP_CODE)
194 printf("\nDecoding error = %d\n", res
);
200 *d_len
= outProcessed
;