1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
#import "CSZlibHandle.h"
NSString *CSZlibException=@"CSZlibException";
@implementation CSZlibHandle
+(CSZlibHandle *)zlibHandleWithHandle:(CSHandle *)handle
{
return [[[CSZlibHandle alloc] initWithHandle:handle length:CSHandleMaxLength header:YES name:[handle name]] autorelease];
}
+(CSZlibHandle *)zlibHandleWithHandle:(CSHandle *)handle length:(off_t)length
{
return [[[CSZlibHandle alloc] initWithHandle:handle length:length header:YES name:[handle name]] autorelease];
}
+(CSZlibHandle *)deflateHandleWithHandle:(CSHandle *)handle
{
return [[[CSZlibHandle alloc] initWithHandle:handle length:CSHandleMaxLength header:NO name:[handle name]] autorelease];
}
+(CSZlibHandle *)deflateHandleWithHandle:(CSHandle *)handle length:(off_t)length
{
return [[[CSZlibHandle alloc] initWithHandle:handle length:length header:NO name:[handle name]] autorelease];
}
-(id)initWithHandle:(CSHandle *)handle length:(off_t)length header:(BOOL)header name:(NSString *)descname
{
if((self=[super initWithName:descname length:length]))
{
parent=[handle retain];
startoffs=[parent offsetInFile];
inited=YES;
seekback=NO;
memset(&zs,0,sizeof(zs));
if(header) inflateInit(&zs);
else inflateInit2(&zs,-MAX_WBITS);
}
return self;
}
-(id)initAsCopyOf:(CSZlibHandle *)other
{
if((self=[super initAsCopyOf:other]))
{
parent=[other->parent copy];
startoffs=other->startoffs;
inited=NO;
seekback=other->seekback;
memset(&zs,0,sizeof(zs));
if(inflateCopy(&zs,&other->zs)==Z_OK)
{
zs.next_in=inbuffer;
memcpy(inbuffer,other->zs.next_in,zs.avail_in);
inited=YES;
return self;
}
[self release];
}
return nil;
}
-(void)dealloc
{
if(inited) inflateEnd(&zs);
[parent release];
[super dealloc];
}
-(void)setSeekBackAtEOF:(BOOL)seekateof { seekback=seekateof; }
-(void)setEndStreamAtInputEOF:(BOOL)endateof { endstreamateof=endateof; } // Hack for NSIS's broken zlib usage
-(void)resetStream
{
[parent seekToFileOffset:startoffs];
zs.avail_in=0;
inflateReset(&zs);
}
-(int)streamAtMost:(int)num toBuffer:(void *)buffer
{
zs.next_out=buffer;
zs.avail_out=num;
for(;;)
{
int err=inflate(&zs,0);
if(err==Z_STREAM_END)
{
if(seekback) [parent skipBytes:-(off_t)zs.avail_in];
[self endStream];
return num-zs.avail_out;
}
else if(err!=Z_OK && err!=Z_BUF_ERROR) [self _raiseZlib];
if(!zs.avail_out) return num;
if(!zs.avail_in)
{
zs.avail_in=[parent readAtMost:sizeof(inbuffer) toBuffer:inbuffer];
zs.next_in=(void *)inbuffer;
if(!zs.avail_in)
{
if(endstreamateof)
{
[self endStream];
return num-zs.avail_out;
}
else [parent _raiseEOF];
}
}
}
}
-(void)_raiseZlib
{
[NSException raise:CSZlibException
format:@"Zlib error while attepting to read from \"%@\": %s.",name,zs.msg];
}
@end
|
