wok annotate linux/stuff/linux-unlzma-2.6.33.4.u @ rev 5578

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