ThePain/Swiftgram
1
0
Fork 0
mirror of https://github.com/Swiftgram/Telegram-iOS.git synced 2025-08-18 11:30:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'release/3.5.1'

Browse Source
This commit is contained in:
Andreas Linde 2014-01-14 19:07:48 +01:00
commit a561e5c3dd
1 changed files with 33 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
Classes/BITAuthenticator.m
View File

@ -48,6 +48,7 @@ static NSString* const kBITAuthenticatorAuthTokenKey = @"BITAuthenticatorAuthTok
static NSString* const kBITAuthenticatorAuthTokenTypeKey = @"BITAuthenticatorAuthTokenTypeKey"; static NSString* const kBITAuthenticatorAuthTokenTypeKey = @"BITAuthenticatorAuthTokenTypeKey";
typedef unsigned int bit_uint32; typedef unsigned int bit_uint32;
static unsigned char kBITPNGHeader[8] = {137, 80, 78, 71, 13, 10, 26, 10};
static unsigned char kBITPNGEndChunk[4] = {0x49, 0x45, 0x4e, 0x44}; static unsigned char kBITPNGEndChunk[4] = {0x49, 0x45, 0x4e, 0x44};
@implementation BITAuthenticator { @implementation BITAuthenticator {
@ -700,27 +701,36 @@ static unsigned char kBITPNGEndChunk[4] = {0x49, 0x45, 0x4e, 0x44};
return; return;
} }
if ((fs.st_size < 20) || (memcmp(source, kBITPNGHeader, 8))) {
// Not a PNG
free(source);
return;
}
buffer = source + 8;
NSString *result = nil;
bit_uint32 length; bit_uint32 length;
unsigned char *name; unsigned char *name;
unsigned char *data; unsigned char *data;
NSString *result = nil; int chunk_index = 0;
long long bytes_left = fs.st_size - 8;
buffer = source + 8;
int index = 0;
do { do {
memcpy(&length, buffer, 4); memcpy(&length, buffer, 4);
length = ntohl(length); length = ntohl(length);
name = (unsigned char *)malloc(4);
buffer += 4; buffer += 4;
name = (unsigned char *)malloc(4);
memcpy(name, buffer, 4); memcpy(name, buffer, 4);
buffer += 4; buffer += 4;
data = (unsigned char *)malloc(length + 1); data = (unsigned char *)malloc(length + 1);
if (bytes_left >= length) {
memcpy(data, buffer, length); memcpy(data, buffer, length);
buffer += length; buffer += length;
buffer += 4; buffer += 4;
if (!strcmp((const char *)name, "tEXt")) { if (!strcmp((const char *)name, "tEXt")) {
data[length] = 0; data[length] = 0;
NSString *key = [NSString stringWithCString:(char *)data encoding:NSUTF8StringEncoding]; NSString *key = [NSString stringWithCString:(char *)data encoding:NSUTF8StringEncoding];
@ -731,12 +741,15 @@ static unsigned char kBITPNGEndChunk[4] = {0x49, 0x45, 0x4e, 0x44};
} }
if (!memcmp(name, kBITPNGEndChunk, 4)){ if (!memcmp(name, kBITPNGEndChunk, 4)){
index = 128; chunk_index = 128;
}
} }
free(data); free(data);
free(name); free(name);
} while (index++ < 128);
bytes_left -= (length + 3 * 4);
} while ((chunk_index++ < 128) && (bytes_left > 8));
free(source); free(source);