Blame


1 a0d146ed 2005-07-12 devnull #include "stdinc.h"
2 a0d146ed 2005-07-12 devnull #include "whack.h"
3 a0d146ed 2005-07-12 devnull
4 a0d146ed 2005-07-12 devnull enum
5 a0d146ed 2005-07-12 devnull {
6 a0d146ed 2005-07-12 devnull DMaxFastLen = 7,
7 a0d146ed 2005-07-12 devnull DBigLenCode = 0x3c, /* minimum code for large lenth encoding */
8 a0d146ed 2005-07-12 devnull DBigLenBits = 6,
9 a0d146ed 2005-07-12 devnull DBigLenBase = 1 /* starting items to encode for big lens */
10 a0d146ed 2005-07-12 devnull };
11 a0d146ed 2005-07-12 devnull
12 a0d146ed 2005-07-12 devnull static uchar lenval[1 << (DBigLenBits - 1)] =
13 a0d146ed 2005-07-12 devnull {
14 a0d146ed 2005-07-12 devnull 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
15 a0d146ed 2005-07-12 devnull 3, 3, 3, 3, 3, 3, 3, 3,
16 a0d146ed 2005-07-12 devnull 4, 4, 4, 4,
17 a0d146ed 2005-07-12 devnull 5,
18 a0d146ed 2005-07-12 devnull 6,
19 a0d146ed 2005-07-12 devnull 255,
20 a0d146ed 2005-07-12 devnull 255
21 a0d146ed 2005-07-12 devnull };
22 a0d146ed 2005-07-12 devnull
23 a0d146ed 2005-07-12 devnull static uchar lenbits[] =
24 a0d146ed 2005-07-12 devnull {
25 a0d146ed 2005-07-12 devnull 0, 0, 0,
26 28b49df3 2006-07-18 devnull 2, 3, 5, 5,
27 a0d146ed 2005-07-12 devnull };
28 a0d146ed 2005-07-12 devnull
29 a0d146ed 2005-07-12 devnull static uchar offbits[16] =
30 a0d146ed 2005-07-12 devnull {
31 a0d146ed 2005-07-12 devnull 5, 5, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 12, 13
32 a0d146ed 2005-07-12 devnull };
33 a0d146ed 2005-07-12 devnull
34 a0d146ed 2005-07-12 devnull static ushort offbase[16] =
35 a0d146ed 2005-07-12 devnull {
36 a0d146ed 2005-07-12 devnull 0, 0x20,
37 a0d146ed 2005-07-12 devnull 0x40, 0x60,
38 a0d146ed 2005-07-12 devnull 0x80, 0xc0,
39 a0d146ed 2005-07-12 devnull 0x100, 0x180,
40 a0d146ed 2005-07-12 devnull 0x200, 0x300,
41 a0d146ed 2005-07-12 devnull 0x400, 0x600,
42 a0d146ed 2005-07-12 devnull 0x800, 0xc00,
43 a0d146ed 2005-07-12 devnull 0x1000,
44 a0d146ed 2005-07-12 devnull 0x2000
45 a0d146ed 2005-07-12 devnull };
46 a0d146ed 2005-07-12 devnull
47 a0d146ed 2005-07-12 devnull void
48 a0d146ed 2005-07-12 devnull unwhackinit(Unwhack *uw)
49 a0d146ed 2005-07-12 devnull {
50 a0d146ed 2005-07-12 devnull uw->err[0] = '\0';
51 a0d146ed 2005-07-12 devnull }
52 a0d146ed 2005-07-12 devnull
53 a0d146ed 2005-07-12 devnull int
54 a0d146ed 2005-07-12 devnull unwhack(Unwhack *uw, uchar *dst, int ndst, uchar *src, int nsrc)
55 a0d146ed 2005-07-12 devnull {
56 a0d146ed 2005-07-12 devnull uchar *s, *d, *dmax, *smax, lit;
57 a0d146ed 2005-07-12 devnull ulong uwbits, lithist;
58 a0d146ed 2005-07-12 devnull int i, off, len, bits, use, code, uwnbits, overbits;
59 a0d146ed 2005-07-12 devnull
60 a0d146ed 2005-07-12 devnull d = dst;
61 a0d146ed 2005-07-12 devnull dmax = d + ndst;
62 a0d146ed 2005-07-12 devnull
63 a0d146ed 2005-07-12 devnull smax = src + nsrc;
64 a0d146ed 2005-07-12 devnull uwnbits = 0;
65 a0d146ed 2005-07-12 devnull uwbits = 0;
66 a0d146ed 2005-07-12 devnull overbits = 0;
67 a0d146ed 2005-07-12 devnull lithist = ~0;
68 a0d146ed 2005-07-12 devnull while(src < smax || uwnbits - overbits >= MinDecode){
69 a0d146ed 2005-07-12 devnull while(uwnbits <= 24){
70 a0d146ed 2005-07-12 devnull uwbits <<= 8;
71 a0d146ed 2005-07-12 devnull if(src < smax)
72 a0d146ed 2005-07-12 devnull uwbits |= *src++;
73 a0d146ed 2005-07-12 devnull else
74 a0d146ed 2005-07-12 devnull overbits += 8;
75 a0d146ed 2005-07-12 devnull uwnbits += 8;
76 a0d146ed 2005-07-12 devnull }
77 a0d146ed 2005-07-12 devnull
78 a0d146ed 2005-07-12 devnull /*
79 a0d146ed 2005-07-12 devnull * literal
80 a0d146ed 2005-07-12 devnull */
81 a0d146ed 2005-07-12 devnull len = lenval[(uwbits >> (uwnbits - 5)) & 0x1f];
82 a0d146ed 2005-07-12 devnull if(len == 0){
83 a0d146ed 2005-07-12 devnull if(lithist & 0xf){
84 a0d146ed 2005-07-12 devnull uwnbits -= 9;
85 a0d146ed 2005-07-12 devnull lit = (uwbits >> uwnbits) & 0xff;
86 a0d146ed 2005-07-12 devnull lit &= 255;
87 a0d146ed 2005-07-12 devnull }else{
88 a0d146ed 2005-07-12 devnull uwnbits -= 8;
89 a0d146ed 2005-07-12 devnull lit = (uwbits >> uwnbits) & 0x7f;
90 a0d146ed 2005-07-12 devnull if(lit < 32){
91 a0d146ed 2005-07-12 devnull if(lit < 24){
92 a0d146ed 2005-07-12 devnull uwnbits -= 2;
93 a0d146ed 2005-07-12 devnull lit = (lit << 2) | ((uwbits >> uwnbits) & 3);
94 a0d146ed 2005-07-12 devnull }else{
95 a0d146ed 2005-07-12 devnull uwnbits -= 3;
96 a0d146ed 2005-07-12 devnull lit = (lit << 3) | ((uwbits >> uwnbits) & 7);
97 a0d146ed 2005-07-12 devnull }
98 a0d146ed 2005-07-12 devnull lit = (lit - 64) & 0xff;
99 a0d146ed 2005-07-12 devnull }
100 a0d146ed 2005-07-12 devnull }
101 a0d146ed 2005-07-12 devnull if(d >= dmax){
102 a0d146ed 2005-07-12 devnull snprint(uw->err, WhackErrLen, "too much output");
103 a0d146ed 2005-07-12 devnull return -1;
104 a0d146ed 2005-07-12 devnull }
105 a0d146ed 2005-07-12 devnull *d++ = lit;
106 a0d146ed 2005-07-12 devnull lithist = (lithist << 1) | (lit < 32) | (lit > 127);
107 a0d146ed 2005-07-12 devnull continue;
108 a0d146ed 2005-07-12 devnull }
109 a0d146ed 2005-07-12 devnull
110 a0d146ed 2005-07-12 devnull /*
111 a0d146ed 2005-07-12 devnull * length
112 a0d146ed 2005-07-12 devnull */
113 a0d146ed 2005-07-12 devnull if(len < 255)
114 a0d146ed 2005-07-12 devnull uwnbits -= lenbits[len];
115 a0d146ed 2005-07-12 devnull else{
116 a0d146ed 2005-07-12 devnull uwnbits -= DBigLenBits;
117 a0d146ed 2005-07-12 devnull code = ((uwbits >> uwnbits) & ((1 << DBigLenBits) - 1)) - DBigLenCode;
118 a0d146ed 2005-07-12 devnull len = DMaxFastLen;
119 a0d146ed 2005-07-12 devnull use = DBigLenBase;
120 a0d146ed 2005-07-12 devnull bits = (DBigLenBits & 1) ^ 1;
121 a0d146ed 2005-07-12 devnull while(code >= use){
122 a0d146ed 2005-07-12 devnull len += use;
123 a0d146ed 2005-07-12 devnull code -= use;
124 a0d146ed 2005-07-12 devnull code <<= 1;
125 a0d146ed 2005-07-12 devnull uwnbits--;
126 a0d146ed 2005-07-12 devnull if(uwnbits < 0){
127 a0d146ed 2005-07-12 devnull snprint(uw->err, WhackErrLen, "len out of range");
128 a0d146ed 2005-07-12 devnull return -1;
129 a0d146ed 2005-07-12 devnull }
130 a0d146ed 2005-07-12 devnull code |= (uwbits >> uwnbits) & 1;
131 a0d146ed 2005-07-12 devnull use <<= bits;
132 a0d146ed 2005-07-12 devnull bits ^= 1;
133 a0d146ed 2005-07-12 devnull }
134 a0d146ed 2005-07-12 devnull len += code;
135 a0d146ed 2005-07-12 devnull
136 a0d146ed 2005-07-12 devnull while(uwnbits <= 24){
137 a0d146ed 2005-07-12 devnull uwbits <<= 8;
138 a0d146ed 2005-07-12 devnull if(src < smax)
139 a0d146ed 2005-07-12 devnull uwbits |= *src++;
140 a0d146ed 2005-07-12 devnull else
141 a0d146ed 2005-07-12 devnull overbits += 8;
142 a0d146ed 2005-07-12 devnull uwnbits += 8;
143 a0d146ed 2005-07-12 devnull }
144 a0d146ed 2005-07-12 devnull }
145 a0d146ed 2005-07-12 devnull
146 a0d146ed 2005-07-12 devnull /*
147 a0d146ed 2005-07-12 devnull * offset
148 a0d146ed 2005-07-12 devnull */
149 a0d146ed 2005-07-12 devnull uwnbits -= 4;
150 a0d146ed 2005-07-12 devnull bits = (uwbits >> uwnbits) & 0xf;
151 a0d146ed 2005-07-12 devnull off = offbase[bits];
152 a0d146ed 2005-07-12 devnull bits = offbits[bits];
153 a0d146ed 2005-07-12 devnull
154 a0d146ed 2005-07-12 devnull uwnbits -= bits;
155 a0d146ed 2005-07-12 devnull off |= (uwbits >> uwnbits) & ((1 << bits) - 1);
156 a0d146ed 2005-07-12 devnull off++;
157 a0d146ed 2005-07-12 devnull
158 a0d146ed 2005-07-12 devnull if(off > d - dst){
159 a0d146ed 2005-07-12 devnull snprint(uw->err, WhackErrLen, "offset out of range: off=%d d=%ld len=%d nbits=%d", off, d - dst, len, uwnbits);
160 a0d146ed 2005-07-12 devnull return -1;
161 a0d146ed 2005-07-12 devnull }
162 a0d146ed 2005-07-12 devnull if(d + len > dmax){
163 a0d146ed 2005-07-12 devnull snprint(uw->err, WhackErrLen, "len out of range");
164 a0d146ed 2005-07-12 devnull return -1;
165 a0d146ed 2005-07-12 devnull }
166 a0d146ed 2005-07-12 devnull s = d - off;
167 a0d146ed 2005-07-12 devnull for(i = 0; i < len; i++)
168 a0d146ed 2005-07-12 devnull d[i] = s[i];
169 a0d146ed 2005-07-12 devnull d += len;
170 a0d146ed 2005-07-12 devnull }
171 a0d146ed 2005-07-12 devnull if(uwnbits < overbits){
172 a0d146ed 2005-07-12 devnull snprint(uw->err, WhackErrLen, "compressed data overrun");
173 a0d146ed 2005-07-12 devnull return -1;
174 a0d146ed 2005-07-12 devnull }
175 a0d146ed 2005-07-12 devnull
176 a0d146ed 2005-07-12 devnull len = d - dst;
177 a0d146ed 2005-07-12 devnull
178 a0d146ed 2005-07-12 devnull return len;
179 a0d146ed 2005-07-12 devnull }