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
|
#import "XADLZMA2Handle.h"
#import "XADException.h"
static void *Alloc(void *p,size_t size) { return malloc(size); }
static void Free(void *p,void *address) { return free(address); }
static ISzAlloc allocator={Alloc,Free};
@implementation XADLZMA2Handle
-(id)initWithHandle:(CSHandle *)handle propertyData:(NSData *)propertydata
{
return [self initWithHandle:handle length:CSHandleMaxLength propertyData:propertydata];
}
-(id)initWithHandle:(CSHandle *)handle length:(off_t)length propertyData:(NSData *)propertydata
{
if((self=[super initWithName:[handle name] length:length]))
{
parent=[handle retain];
startoffs=[parent offsetInFile];
seekback=NO;
Lzma2Dec_Construct(&lzma);
if([propertydata length]>=1)
if(Lzma2Dec_Allocate(&lzma,((uint8_t *)[propertydata bytes])[0],&allocator)==SZ_OK)
{
return self;
}
}
[self release];
return nil;
}
-(void)dealloc
{
Lzma2Dec_Free(&lzma,&allocator);
[parent release];
[super dealloc];
}
-(void)setSeekBackAtEOF:(BOOL)seekateof { seekback=seekateof; }
-(void)resetStream
{
[parent seekToFileOffset:startoffs];
Lzma2Dec_Init(&lzma);
bufbytes=bufoffs=0;
}
-(int)streamAtMost:(int)num toBuffer:(void *)buffer
{
int total=0;
while(total<num)
{
size_t destlen=num-total;
size_t srclen=bufbytes-bufoffs;
ELzmaStatus status;
int res=Lzma2Dec_DecodeToBuf(&lzma,buffer+total,&destlen,inbuffer+bufoffs,&srclen,LZMA_FINISH_ANY,&status);
total+=destlen;
bufoffs+=srclen;
if(res!=SZ_OK) [XADException raiseDecrunchException];
if(status==LZMA_STATUS_NEEDS_MORE_INPUT)
{
bufbytes=[parent readAtMost:sizeof(inbuffer) toBuffer:inbuffer];
if(!bufbytes) [parent _raiseEOF];
bufoffs=0;
}
else if(status==LZMA_STATUS_FINISHED_WITH_MARK)
{
if(seekback) [parent skipBytes:-bufbytes+bufoffs];
[self endStream];
break;
}
}
return total;
}
@end
|
