bytetool.mm

来自「lumaqq」· MM 代码 · 共 218 行

/*
 * LumaQQ - Cross platform QQ client, special edition for Mac
 *
 * Copyright (C) 2007 luma <[e-mail]>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

#import "ByteTool.h"
#import "QQConstants.h"
#import "NSData-BytesOperation.h"

@implementation ByteTool

+ (UInt32)getUInt32:(const char*)bytes offset:(int)offset {
	return [self getUInt32:bytes offset:offset length:4];
}

+ (UInt32)getUInt32:(const char*)bytes offset:(int)offset length:(int)length {
	UInt32 ret = 0;
	int end = 0;
	int i;
	
	if(length > 4)
		end = offset + 4;
	else
		end = offset + length;
	for(i = offset; i < end; i++) {
		ret <<= 8;
		ret |= bytes[i] & 0xFF;
	}
	return ret;
}

+ (void)writeUInt32:(char*)bytes value:(UInt32)value at:(int)offset {
	UInt32 u = EndianU32_NtoB(value);
	memcpy(bytes + offset, &u, 4);
}

+ (UInt16)getUInt16:(const char*)bytes offset:(int)offset {
	int i;
	UInt16 ret = 0;
	for(i = offset; i < offset + 2; i++) {
		ret <<= 8;
		ret |= bytes[i] & 0xFF;
	}
	return ret;
}

+ (void)writeUInt16:(char*)bytes value:(UInt16)value at:(int)offset {
	UInt16 u = EndianU16_NtoB(value);
	memcpy(bytes + offset, &u, 2);
}

+ (void)getBytes:(char*)bytes from:(const char*)buffer start:(int)offset length:(int)length {
	memcpy(bytes, buffer + offset, length);
}

+ (void)writeBytes:(const char*)bytes to:(char*)buffer start:(int)offset length:(int)length {
	memcpy(buffer + offset, bytes, length);
}

+ (NSData*)randomKey {
	NSMutableData* key = [NSMutableData dataWithLength:kQQKeyLength];
	char* bytes = (char*)[key mutableBytes];
	for(int i = 0; i < kQQKeyLength; i++)
		bytes[i] = random();
	return key;
}

+ (NSString*)getString:(NSData*)data encoding:(int)encoding {
	CFStringEncoding strEncoding;
	switch(encoding) {
		case kQQEncodingASCII:
		case kQQEncodingUTF8:
			strEncoding = kCFStringEncodingUTF8;
			break;
		default:
			strEncoding = kCFStringEncodingGB_18030_2000;
			break;
	}
	
	NSString* str = (NSString*)CFStringCreateFromExternalRepresentation(kCFAllocatorDefault,
																		(CFDataRef)data,
																		strEncoding);
	return [str autorelease];
}

+ (NSString*)getString:(NSData*)data {
	return [self getString:data encoding:kQQEncodingDefault];
}

+ (NSData*)getBytes:(NSString*)str {
	return [self getBytes:str encoding:kQQEncodingDefault];
}

+ (NSData*)getBytes:(NSString*)str encoding:(int)encoding {
	// map encoding
	CFStringEncoding strEncoding;
	switch(encoding) {
		case kQQEncodingASCII:
		case kQQEncodingUTF8:
			strEncoding = kCFStringEncodingUTF8;
			break;
		default:
			strEncoding = kCFStringEncodingGB_18030_2000;
			break;
	}
	
	// get bytes
	int strLen = [str length];
	int length = strLen * 4;
	NSMutableData* data = [NSMutableData dataWithLength:length];
	CFIndex used = 0;
	CFStringGetBytes((CFStringRef)str,
					 CFRangeMake(0, strLen),
					 strEncoding,
					 ' ',
					 YES,
					 (UInt8*)[data mutableBytes],
					 length,
					 &used);
	
	// initial return data
	NSData* ret = [NSData dataWithBytes:[data bytes] length:used];
	return ret;
}

+ (NSString*)ip2String:(const char*)ip {
	NSString* address = [NSString stringWithFormat:@"%d.%d.%d.%d", ip[0] & 0xFF, ip[1] & 0xFF, ip[2] & 0xFF, ip[3] & 0xFF];
	return address;
}

+ (NSString*)ipData2String:(NSData*)data {
	const char* ip = (const char*)[data bytes];
	return [ByteTool ip2String:ip];
}

+ (NSData*)string2IpData:(NSString*)string {
	NSArray* components = [string componentsSeparatedByString:@"."];
	NSMutableData* data = [NSMutableData dataWithLength:4];
	if([components count] == 4) {
		char* ip = (char*)[data mutableBytes];
		int value = [[components objectAtIndex:0] intValue];
		if(value > 255 || value < 0)
			return nil;
		ip[0] = value & 0xFF;
		value = [[components objectAtIndex:1] intValue];
		if(value > 255 || value < 0)
			return nil;
		ip[1] = value & 0xFF;
		value = [[components objectAtIndex:2] intValue];
		if(value > 255 || value < 0)
			return nil;
		ip[2] = value & 0xFF;
		value = [[components objectAtIndex:3] intValue];
		if(value > 255 || value < 0)
			return nil;
		ip[3] = value & 0xFF;
		
		return data;
	} else {
		NSLog(@"Error IP String Format");
		return nil;
	}
}

+ (void)reverseIp:(char*)ip {
	char tmp = ip[0];
	ip[0] = ip[3];
	ip[3] = tmp;
	tmp = ip[1];
	ip[1] = ip[2];
	ip[2] = tmp;
}

+ (CustomFaceList*)buildCustomFaceList:(NSData*)data owner:(UInt32)owner {
	CustomFaceList* list = nil;
	
	char* bytes = (char*)[data bytes];
	int length = [data length];
	
	int from = 0;
	int index = [data indexOfByte:kQQTagCustomFace from:from];
	if(index == -1)
		return nil;
	else
		list = [[CustomFaceList alloc] init];
	
	while(index != -1) {
		ByteBuffer* buf = [ByteBuffer bufferWithBytes:bytes from:index length:(length - index)];
		CustomFace* face = [[CustomFace alloc] init];
		[face read:buf];
		[face setOwner:owner];
		[list addCustomFace:face];
		from = index + [face length];
		[face release];
		
		index = [data indexOfByte:kQQTagCustomFace from:from];
	}
	
	return [list autorelease];
}

@end