1*80ee5cbfSDavid du Colombier #include <u.h>
2*80ee5cbfSDavid du Colombier #include <libc.h>
3*80ee5cbfSDavid du Colombier #include <flate.h>
4*80ee5cbfSDavid du Colombier #include "zlib.h"
5*80ee5cbfSDavid du Colombier
6*80ee5cbfSDavid du Colombier typedef struct ZRead ZRead;
7*80ee5cbfSDavid du Colombier
8*80ee5cbfSDavid du Colombier struct ZRead
9*80ee5cbfSDavid du Colombier {
10*80ee5cbfSDavid du Colombier ulong adler;
11*80ee5cbfSDavid du Colombier void *rr;
12*80ee5cbfSDavid du Colombier int (*r)(void*, void*, int);
13*80ee5cbfSDavid du Colombier };
14*80ee5cbfSDavid du Colombier
15*80ee5cbfSDavid du Colombier static int
zlread(void * vzr,void * buf,int n)16*80ee5cbfSDavid du Colombier zlread(void *vzr, void *buf, int n)
17*80ee5cbfSDavid du Colombier {
18*80ee5cbfSDavid du Colombier ZRead *zr;
19*80ee5cbfSDavid du Colombier
20*80ee5cbfSDavid du Colombier zr = vzr;
21*80ee5cbfSDavid du Colombier n = (*zr->r)(zr->rr, buf, n);
22*80ee5cbfSDavid du Colombier if(n <= 0)
23*80ee5cbfSDavid du Colombier return n;
24*80ee5cbfSDavid du Colombier zr->adler = adler32(zr->adler, buf, n);
25*80ee5cbfSDavid du Colombier return n;
26*80ee5cbfSDavid du Colombier }
27*80ee5cbfSDavid du Colombier
28*80ee5cbfSDavid du Colombier int
deflatezlib(void * wr,int (* w)(void *,void *,int),void * rr,int (* r)(void *,void *,int),int level,int debug)29*80ee5cbfSDavid du Colombier deflatezlib(void *wr, int (*w)(void*, void*, int), void *rr, int (*r)(void*, void*, int), int level, int debug)
30*80ee5cbfSDavid du Colombier {
31*80ee5cbfSDavid du Colombier ZRead zr;
32*80ee5cbfSDavid du Colombier uchar buf[4];
33*80ee5cbfSDavid du Colombier int ok;
34*80ee5cbfSDavid du Colombier
35*80ee5cbfSDavid du Colombier buf[0] = ZlibDeflate | ZlibWin32k;
36*80ee5cbfSDavid du Colombier
37*80ee5cbfSDavid du Colombier /* bogus zlib encoding of compression level */
38*80ee5cbfSDavid du Colombier buf[1] = ((level > 2) + (level > 5) + (level > 8)) << 6;
39*80ee5cbfSDavid du Colombier
40*80ee5cbfSDavid du Colombier /* header check field */
41*80ee5cbfSDavid du Colombier buf[1] |= 31 - ((buf[0] << 8) | buf[1]) % 31;
42*80ee5cbfSDavid du Colombier if((*w)(wr, buf, 2) != 2)
43*80ee5cbfSDavid du Colombier return FlateOutputFail;
44*80ee5cbfSDavid du Colombier
45*80ee5cbfSDavid du Colombier zr.rr = rr;
46*80ee5cbfSDavid du Colombier zr.r = r;
47*80ee5cbfSDavid du Colombier zr.adler = 1;
48*80ee5cbfSDavid du Colombier ok = deflate(wr, w, &zr, zlread, level, debug);
49*80ee5cbfSDavid du Colombier if(ok != FlateOk)
50*80ee5cbfSDavid du Colombier return ok;
51*80ee5cbfSDavid du Colombier
52*80ee5cbfSDavid du Colombier buf[0] = zr.adler >> 24;
53*80ee5cbfSDavid du Colombier buf[1] = zr.adler >> 16;
54*80ee5cbfSDavid du Colombier buf[2] = zr.adler >> 8;
55*80ee5cbfSDavid du Colombier buf[3] = zr.adler;
56*80ee5cbfSDavid du Colombier if((*w)(wr, buf, 4) != 4)
57*80ee5cbfSDavid du Colombier return FlateOutputFail;
58*80ee5cbfSDavid du Colombier
59*80ee5cbfSDavid du Colombier return FlateOk;
60*80ee5cbfSDavid du Colombier }
61