wok-4.x annotate linux/stuff/linux-unlzma-2.6.33.2.u @ rev 5245

Up linux (2.6.33.2) With more modules and less patch
author Christophe Lincoln <pankso@slitaz.org>
date Sun Apr 11 19:23:48 2010 +0200 (2010-04-11)
parents
children
rev   line source
pankso@5245 1 --- linux-2.6.30.4/init/initramfs.c
pankso@5245 2 +++ linux-2.6.30.4/init/initramfs.c
pankso@5245 3 @@ -425,7 +425,8 @@
pankso@5245 4 return len - count;
pankso@5245 5 }
pankso@5245 6
pankso@5245 7 -static int __init flush_buffer(void *bufv, unsigned len)
pankso@5245 8 +#define flush_buffer cpio_flush_buffer
pankso@5245 9 +int __init flush_buffer(void *bufv, unsigned len)
pankso@5245 10 {
pankso@5245 11 char *buf = (char *) bufv;
pankso@5245 12 int written;
pankso@5245 13
pankso@5245 14 --- linux-2.6.30.4/lib/decompress_unlzma.c
pankso@5245 15 +++ linux-2.6.30.4/lib/decompress_unlzma.c
pankso@5245 16 @@ -278,6 +278,10 @@
pankso@5245 17 size_t global_pos;
pankso@5245 18 int(*flush)(void*, unsigned int);
pankso@5245 19 struct lzma_header *header;
pankso@5245 20 + int is_cpio_flush;
pankso@5245 21 + uint8_t **buffer_index;
pankso@5245 22 + int next_index;
pankso@5245 23 + int max_index;
pankso@5245 24 };
pankso@5245 25
pankso@5245 26 struct cstate {
pankso@5245 27 @@ -294,6 +298,14 @@
pankso@5245 28 static inline uint8_t INIT peek_old_byte(struct writer *wr,
pankso@5245 29 uint32_t offs)
pankso@5245 30 {
pankso@5245 31 + if (wr->is_cpio_flush) {
pankso@5245 32 + int32_t pos;
pankso@5245 33 + while (offs > wr->header->dict_size)
pankso@5245 34 + offs -= wr->header->dict_size;
pankso@5245 35 + pos = wr->buffer_pos - offs;
pankso@5245 36 + return wr->buffer_index[pos / LZMA_IOBUF_SIZE]
pankso@5245 37 + [pos % LZMA_IOBUF_SIZE];
pankso@5245 38 + }
pankso@5245 39 if (!wr->flush) {
pankso@5245 40 int32_t pos;
pankso@5245 41 while (offs > wr->header->dict_size)
pankso@5245 42 @@ -309,8 +321,41 @@
pankso@5245 43
pankso@5245 44 }
pankso@5245 45
pankso@5245 46 +static inline void INIT write_byte_if_cpio(struct writer *wr, uint8_t byte)
pankso@5245 47 +{
pankso@5245 48 + if (wr->buffer_pos % LZMA_IOBUF_SIZE == 0) {
pankso@5245 49 + // if the following large_malloc fails, the initramfs
pankso@5245 50 + // whould not be load with is_cpio_flush forced 0 too.
pankso@5245 51 + // Remember we do not allocate historic buffer.
pankso@5245 52 + // Let's assume it will never fail !
pankso@5245 53 + if (wr->next_index >= wr->max_index) {
pankso@5245 54 + // realloc wr->buffer_index
pankso@5245 55 + uint8_t **p = wr->buffer_index;
pankso@5245 56 + wr->buffer_index = (uint8_t **)
pankso@5245 57 + large_malloc(LZMA_IOBUF_SIZE +
pankso@5245 58 + sizeof(*p) * wr->max_index);
pankso@5245 59 + if (wr->max_index) {
pankso@5245 60 + memcpy(wr->buffer_index, p,
pankso@5245 61 + sizeof(*p) * wr->max_index);
pankso@5245 62 + free(p);
pankso@5245 63 + }
pankso@5245 64 + wr->max_index += LZMA_IOBUF_SIZE / sizeof(*p);
pankso@5245 65 + }
pankso@5245 66 + wr->buffer_index[wr->next_index++] =
pankso@5245 67 + (uint8_t *) large_malloc(LZMA_IOBUF_SIZE);
pankso@5245 68 + }
pankso@5245 69 + wr->buffer_index[wr->buffer_pos / LZMA_IOBUF_SIZE]
pankso@5245 70 + [wr->buffer_pos % LZMA_IOBUF_SIZE] =
pankso@5245 71 + wr->previous_byte = byte;
pankso@5245 72 + wr->buffer_pos++;
pankso@5245 73 +}
pankso@5245 74 +
pankso@5245 75 static inline void INIT write_byte(struct writer *wr, uint8_t byte)
pankso@5245 76 {
pankso@5245 77 + if (wr->is_cpio_flush) {
pankso@5245 78 + write_byte_if_cpio(wr, byte);
pankso@5245 79 + return;
pankso@5245 80 + }
pankso@5245 81 wr->buffer[wr->buffer_pos++] = wr->previous_byte = byte;
pankso@5245 82 if (wr->flush && wr->buffer_pos == wr->header->dict_size) {
pankso@5245 83 wr->buffer_pos = 0;
pankso@5245 84 @@ -328,7 +373,21 @@
pankso@5245 85 static inline void INIT copy_bytes(struct writer *wr,
pankso@5245 86 uint32_t rep0, int len)
pankso@5245 87 {
pankso@5245 88 - do {
pankso@5245 89 + if (wr->is_cpio_flush) {
pankso@5245 90 + int32_t pos;
pankso@5245 91 + uint32_t offs = rep0;
pankso@5245 92 + while (offs > wr->header->dict_size)
pankso@5245 93 + offs -= wr->header->dict_size;
pankso@5245 94 + pos = wr->buffer_pos - offs;
pankso@5245 95 + do {
pankso@5245 96 + write_byte_if_cpio(wr,
pankso@5245 97 + wr->buffer_index[pos / LZMA_IOBUF_SIZE]
pankso@5245 98 + [pos % LZMA_IOBUF_SIZE]);
pankso@5245 99 + pos++;
pankso@5245 100 + len--;
pankso@5245 101 + } while (len != 0 && wr->buffer_pos < wr->header->dst_size);
pankso@5245 102 + }
pankso@5245 103 + else do {
pankso@5245 104 copy_byte(wr, rep0);
pankso@5245 105 len--;
pankso@5245 106 } while (len != 0 && wr->buffer_pos < wr->header->dst_size);
pankso@5245 107 @@ -339,6 +398,9 @@
pankso@5245 108 int pos_state, uint16_t *prob,
pankso@5245 109 int lc, uint32_t literal_pos_mask) {
pankso@5245 110 int mi = 1;
pankso@5245 111 + static const int state[LZMA_NUM_STATES] =
pankso@5245 112 + { 0, 0, 0, 0, 1, 2, 3, 4, 5, 6, 4, 5 };
pankso@5245 113 +
pankso@5245 114 rc_update_bit_0(rc, prob);
pankso@5245 115 prob = (p + LZMA_LITERAL +
pankso@5245 116 (LZMA_LIT_SIZE
pankso@5245 117 @@ -369,18 +431,13 @@
pankso@5245 118 rc_get_bit(rc, prob_lit, &mi);
pankso@5245 119 }
pankso@5245 120 write_byte(wr, mi);
pankso@5245 121 - if (cst->state < 4)
pankso@5245 122 - cst->state = 0;
pankso@5245 123 - else if (cst->state < 10)
pankso@5245 124 - cst->state -= 3;
pankso@5245 125 - else
pankso@5245 126 - cst->state -= 6;
pankso@5245 127 + cst->state = state[cst->state];
pankso@5245 128 }
pankso@5245 129
pankso@5245 130 static inline void INIT process_bit1(struct writer *wr, struct rc *rc,
pankso@5245 131 struct cstate *cst, uint16_t *p,
pankso@5245 132 int pos_state, uint16_t *prob) {
pankso@5245 133 - int offset;
pankso@5245 134 + int offset;
pankso@5245 135 uint16_t *prob_len;
pankso@5245 136 int num_bits;
pankso@5245 137 int len;
pankso@5245 138 @@ -396,7 +453,7 @@
pankso@5245 139 prob = p + LZMA_LEN_CODER;
pankso@5245 140 } else {
pankso@5245 141 rc_update_bit_1(rc, prob);
pankso@5245 142 - prob = p + LZMA_IS_REP_G0 + cst->state;
pankso@5245 143 + prob += LZMA_IS_REP_G0 - LZMA_IS_REP;
pankso@5245 144 if (rc_is_bit_0(rc, prob)) {
pankso@5245 145 rc_update_bit_0(rc, prob);
pankso@5245 146 prob = (p + LZMA_IS_REP_0_LONG
pankso@5245 147 @@ -417,13 +474,13 @@
pankso@5245 148 uint32_t distance;
pankso@5245 149
pankso@5245 150 rc_update_bit_1(rc, prob);
pankso@5245 151 - prob = p + LZMA_IS_REP_G1 + cst->state;
pankso@5245 152 + prob += LZMA_IS_REP_G1 - LZMA_IS_REP_G0;
pankso@5245 153 if (rc_is_bit_0(rc, prob)) {
pankso@5245 154 rc_update_bit_0(rc, prob);
pankso@5245 155 distance = cst->rep1;
pankso@5245 156 } else {
pankso@5245 157 rc_update_bit_1(rc, prob);
pankso@5245 158 - prob = p + LZMA_IS_REP_G2 + cst->state;
pankso@5245 159 + prob += LZMA_IS_REP_G2 - LZMA_IS_REP_G1;
pankso@5245 160 if (rc_is_bit_0(rc, prob)) {
pankso@5245 161 rc_update_bit_0(rc, prob);
pankso@5245 162 distance = cst->rep2;
pankso@5245 163 @@ -444,24 +501,24 @@
pankso@5245 164 prob_len = prob + LZMA_LEN_CHOICE;
pankso@5245 165 if (rc_is_bit_0(rc, prob_len)) {
pankso@5245 166 rc_update_bit_0(rc, prob_len);
pankso@5245 167 - prob_len = (prob + LZMA_LEN_LOW
pankso@5245 168 + prob_len += LZMA_LEN_LOW - LZMA_LEN_CHOICE
pankso@5245 169 + (pos_state <<
pankso@5245 170 - LZMA_LEN_NUM_LOW_BITS));
pankso@5245 171 + LZMA_LEN_NUM_LOW_BITS);
pankso@5245 172 offset = 0;
pankso@5245 173 num_bits = LZMA_LEN_NUM_LOW_BITS;
pankso@5245 174 } else {
pankso@5245 175 rc_update_bit_1(rc, prob_len);
pankso@5245 176 - prob_len = prob + LZMA_LEN_CHOICE_2;
pankso@5245 177 + prob_len += LZMA_LEN_CHOICE_2 - LZMA_LEN_CHOICE;
pankso@5245 178 if (rc_is_bit_0(rc, prob_len)) {
pankso@5245 179 rc_update_bit_0(rc, prob_len);
pankso@5245 180 - prob_len = (prob + LZMA_LEN_MID
pankso@5245 181 + prob_len += LZMA_LEN_MID - LZMA_LEN_CHOICE_2
pankso@5245 182 + (pos_state <<
pankso@5245 183 - LZMA_LEN_NUM_MID_BITS));
pankso@5245 184 + LZMA_LEN_NUM_MID_BITS);
pankso@5245 185 offset = 1 << LZMA_LEN_NUM_LOW_BITS;
pankso@5245 186 num_bits = LZMA_LEN_NUM_MID_BITS;
pankso@5245 187 } else {
pankso@5245 188 rc_update_bit_1(rc, prob_len);
pankso@5245 189 - prob_len = prob + LZMA_LEN_HIGH;
pankso@5245 190 + prob_len += LZMA_LEN_HIGH - LZMA_LEN_CHOICE_2;
pankso@5245 191 offset = ((1 << LZMA_LEN_NUM_LOW_BITS)
pankso@5245 192 + (1 << LZMA_LEN_NUM_MID_BITS));
pankso@5245 193 num_bits = LZMA_LEN_NUM_HIGH_BITS;
pankso@5245 194 @@ -529,6 +586,7 @@
pankso@5245 195 void(*error_fn)(char *x)
pankso@5245 196 )
pankso@5245 197 {
pankso@5245 198 + extern int cpio_flush_buffer(void*, unsigned int);
pankso@5245 199 struct lzma_header header;
pankso@5245 200 int lc, pb, lp;
pankso@5245 201 uint32_t pos_state_mask;
pankso@5245 202 @@ -563,6 +621,10 @@
pankso@5245 203 wr.global_pos = 0;
pankso@5245 204 wr.previous_byte = 0;
pankso@5245 205 wr.buffer_pos = 0;
pankso@5245 206 + wr.is_cpio_flush = 0;
pankso@5245 207 + if (flush == cpio_flush_buffer)
pankso@5245 208 + wr.is_cpio_flush = 1;
pankso@5245 209 + wr.buffer_index = NULL;
pankso@5245 210
pankso@5245 211 rc_init(&rc, fill, inbuf, in_len);
pankso@5245 212
pankso@5245 213 @@ -596,23 +658,23 @@
pankso@5245 214 if (header.dict_size == 0)
pankso@5245 215 header.dict_size = 1;
pankso@5245 216
pankso@5245 217 - if (output)
pankso@5245 218 + if (output || wr.is_cpio_flush)
pankso@5245 219 wr.buffer = output;
pankso@5245 220 else {
pankso@5245 221 wr.bufsize = MIN(header.dst_size, header.dict_size);
pankso@5245 222 wr.buffer = large_malloc(wr.bufsize);
pankso@5245 223 }
pankso@5245 224 - if (wr.buffer == NULL)
pankso@5245 225 + if (wr.buffer == NULL && !wr.is_cpio_flush)
pankso@5245 226 goto exit_1;
pankso@5245 227
pankso@5245 228 num_probs = LZMA_BASE_SIZE + (LZMA_LIT_SIZE << (lc + lp));
pankso@5245 229 p = (uint16_t *) large_malloc(num_probs * sizeof(*p));
pankso@5245 230 if (p == 0)
pankso@5245 231 goto exit_2;
pankso@5245 232 - num_probs = LZMA_LITERAL + (LZMA_LIT_SIZE << (lc + lp));
pankso@5245 233 + num_probs += LZMA_LITERAL - LZMA_BASE_SIZE;
pankso@5245 234 for (i = 0; i < num_probs; i++)
pankso@5245 235 p[i] = (1 << RC_MODEL_TOTAL_BITS) >> 1;
pankso@5245 236 -
pankso@5245 237 + wr.max_index = wr.next_index = 0;
pankso@5245 238 rc_init_code(&rc);
pankso@5245 239
pankso@5245 240 while (get_pos(&wr) < header.dst_size) {
pankso@5245 241 @@ -631,12 +693,25 @@
pankso@5245 242
pankso@5245 243 if (posp)
pankso@5245 244 *posp = rc.ptr-rc.buffer;
pankso@5245 245 - if (wr.flush)
pankso@5245 246 + if (wr.is_cpio_flush) {
pankso@5245 247 + int i;
pankso@5245 248 + for (i = 0; i < wr.next_index -1; i++) {
pankso@5245 249 + wr.flush(wr.buffer_index[i], LZMA_IOBUF_SIZE);
pankso@5245 250 + large_free(wr.buffer_index[i]);
pankso@5245 251 + }
pankso@5245 252 + if (i < wr.next_index) {
pankso@5245 253 + wr.flush(wr.buffer_index[i],
pankso@5245 254 + wr.buffer_pos % LZMA_IOBUF_SIZE);
pankso@5245 255 + large_free(wr.buffer_index[i]);
pankso@5245 256 + }
pankso@5245 257 + large_free(wr.buffer_index);
pankso@5245 258 + }
pankso@5245 259 + else if (wr.flush)
pankso@5245 260 wr.flush(wr.buffer, wr.buffer_pos);
pankso@5245 261 ret = 0;
pankso@5245 262 large_free(p);
pankso@5245 263 exit_2:
pankso@5245 264 - if (!output)
pankso@5245 265 + if (!output && !wr.is_cpio_flush)
pankso@5245 266 large_free(wr.buffer);
pankso@5245 267 exit_1:
pankso@5245 268 if (!buf)