XADMaster/XADRAR15CryptHandle.m
2018-03-21 14:24:50 +02:00

79 lines
2 KiB
Objective-C

/*
* XADRAR15CryptHandle.m
*
* Copyright (c) 2017-present, MacPaw Inc. All rights reserved.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
* MA 02110-1301 USA
*/
#import "XADRAR15CryptHandle.h"
#import "CRC.h"
static inline uint16_t ror16(uint16_t val,int n) { return (val>>n)|(val<<(16-n)); }
@implementation XADRAR15CryptHandle
-(id)initWithHandle:(CSHandle *)handle length:(off_t)length password:(NSData *)passdata
{
if((self=[super initWithInputBufferForHandle:handle length:length]))
{
password=[passdata retain];
}
return self;
}
-(void)dealloc
{
[password release];
[super dealloc];
}
-(void)resetByteStream
{
int passlength=[password length];
const uint8_t *passbytes=[password bytes];
uint32_t crc=XADCalculateCRC(0xffffffff,passbytes,passlength,XADCRCTable_edb88320);
key0=crc;
key1=crc>>16;
key2=key3=0;
for(int i=0;i<passlength;i++)
{
uint8_t c=passbytes[i];
key2^=c^XADCRCTable_edb88320[c];
key3+=c+(XADCRCTable_edb88320[c]>>16);
}
}
-(uint8_t)produceByteAtOffset:(off_t)pos
{
key0+=0x1234;
key1^=XADCRCTable_edb88320[(key0>>1)&0xff];
key2-=XADCRCTable_edb88320[(key0>>1)&0xff]>>16;
key0^=key2;
key3=ror16(key3&0xffff,1)^key1;
key3=ror16(key3&0xffff,1);
key0^=key3;
if(CSInputAtEOF(input)) CSByteStreamEOF(self);
return CSInputNextByte(input)^(key0>>8);
}
@end