NSData с CCCrypt в многопоточной среде - PullRequest
Новая
       16

NSData с CCCrypt в многопоточной среде

2 голосов
/ 25 мая 2011

У меня есть файл, который был зашифрован с использованием AES. Я использую следующую категорию NSData: 

#import <CommonCrypto/CommonCryptor.h>

@implementation NSData (AES)

- (NSData *)AES256DecryptWithKey:(NSString *)key {

    // 'key' should be 32 bytes for AES256, will be null-padded otherwise
    char keyPtr[kCCKeySizeAES256+1]; // room for terminator (unused)
    bzero(keyPtr, sizeof(keyPtr)); // fill with zeroes (for padding)

    // fetch key data
    [key getCString:keyPtr maxLength:sizeof(keyPtr) encoding:NSUTF8StringEncoding];

    NSUInteger dataLength = [self length];

    //See the doc: For block ciphers, the output size will always be less than or 
    //equal to the input size plus the size of one block.
    //That's why we need to add the size of one block here
    size_t bufferSize = dataLength + kCCBlockSizeAES128;

    void *buffer = malloc(bufferSize);

    size_t numBytesDecrypted = 0;
    CCCryptorStatus cryptStatus = CCCrypt(kCCDecrypt,
                                          kCCAlgorithmAES128,
                                          kCCOptionPKCS7Padding,
                                          keyPtr,
                                          kCCKeySizeAES256,
                                          NULL /* initialization vector (optional) */,
                                          [self bytes], dataLength, /* input */
                                          buffer,       bufferSize, /* output */
                                          &numBytesDecrypted);

    NSLog(@"Bytes decrypted: %d",numBytesDecrypted);

    if (cryptStatus == kCCSuccess) {
        //the returned NSData takes ownership of the buffer and will free it on deallocation
        return [NSData dataWithBytesNoCopy:buffer length:numBytesDecrypted];
    }

    NSLog(@"Decrypt failed with error code %d",cryptStatus);
    free(buffer); //free the buffer;
    return nil;
}

@end

Процесс дескрипции работает нормально, когда я загружаю весь файл из файловой системы со следующим кодом: 

[NSData dataWithContentsOfFile:dataPath];

Проблема возникает, когда файл не читается с помощью предыдущего вызова, но когда внешний код разбивает файл на части и инициирует NSData с небольшим фрагментом данных и пытается расшифровать его, в частности, проблемы возникают, когда другой поток используйте этот код (или, по крайней мере, так я думаю): 

- (NSData *)readDataOfLength:(NSUInteger)length
{
    HTTPLogTrace2(@"%@[%p]: readDataOfLength:%lu", THIS_FILE, self, (unsigned long)length);

    if (![self openFileIfNeeded])
    {
        // File opening failed,
        // or response has been aborted due to another error.
        return nil;
    }

    // Determine how much data we should read.
    // 
    // It is OK if we ask to read more bytes than exist in the file.
    // It is NOT OK to over-allocate the buffer.

    UInt64 bytesLeftInFile = fileLength - fileOffset;

    NSUInteger bytesToRead = (NSUInteger)MIN(length, bytesLeftInFile);

    // Make sure buffer is big enough for read request.
    // Do not over-allocate.

    if (buffer == NULL || bufferSize < bytesToRead)
    {
        bufferSize = bytesToRead;
        buffer = reallocf(buffer, (size_t)bufferSize);

        if (buffer == NULL)
        {
            HTTPLogError(@"%@[%p]: Unable to allocate buffer", THIS_FILE, self);

            [self abort];
            return nil;
        }
    }

    // Perform the read

    HTTPLogVerbose(@"%@[%p]: Attempting to read %lu bytes from file", THIS_FILE, self, bytesToRead);

    ssize_t result = read(fileFD, buffer, bytesToRead);

    // Check the results

    if (result < 0)
    {
        HTTPLogError(@"%@: Error(%i) reading file(%@)", THIS_FILE, errno, filePath);

        [self abort];
        return nil;
    }
    else if (result == 0)
    {
        HTTPLogError(@"%@: Read EOF on file(%@)", THIS_FILE, filePath);

        [self abort];
        return nil;
    }
    else // (result > 0)
    {
        HTTPLogVerbose(@"%@[%p]: Read %d bytes from file", THIS_FILE, self, result);

        fileOffset += result;

        NSData *data = [NSData dataWithBytes:buffer length:result];
        return [data AES256DecryptWithKey:@"abcdefghijklmnopqrstuvwxyz123456"];
        //return data;
    }
}

В данном случае происходит сбой функции CCCrypt с кодом ошибки -4304 AKA "kCCDecodeError - Входные данные не были правильно декодированы или дешифрованы."

Более того, если в вызове CCCrypt вместо kCCOptionPKCS7Padding я передаю 0 -> Заполнение не выполняется, метод не расшифровывает первый фрагмент данных, но при переключении потока происходит сбой с -4300 AKA "kCCParamError - недопустимое значение параметра."

Со следующим сообщением в консоли: 

[Switching to process 13059 thread 0x0]
2011-05-25 18:00:03.631 Drm[1843:6e0b] Bytes decrypted: 131072
2011-05-25 18:00:03.647 Drm[1843:6e0b] Bytes decrypted: 68096
[Switching to process 11779 thread 0x0]
2011-05-25 18:00:04.547 Drm[1843:6e0b] Bytes decrypted: 0
2011-05-25 18:00:04.555 Drm[1843:6e0b] Decrypt failed with error code -4300

Кто-то может помочь?

1 Ответ

3 голосов
/ 25 мая 2011

AES - блочный шифр.Вы должны расшифровать его по одному блоку за раз.Блок AES имеет 128 бит (это не относится к «256» в AES256DecryptWithKey).Поэтому вы должны убедиться, что передаваемые данные кратны 16 байтам.

Я не пытался использовать CCCrypt() таким образом, это не совсем то, для чего оно.CCCrypt() - это удобная функция, когда вы хотите сделать расшифровку одним выстрелом.Когда вы хотите выполнить дешифрование «по ходу», вы используете CCCryptorCreate(), а затем несколько вызовов на CCCryptorUpdate() и, наконец, CCCryptorFinal() (или вы можете вызвать CCCryptorFinal(), а затем CCCryptorReset(), чтобы расшифровать больше материала с тем жеключ).Наконец, вы звоните CCCryptorRelease(), чтобы освободить свой криптор.

РЕДАКТИРОВАТЬ Я подумал еще об этом и понял, что CCCrypt() нельзя использовать таким образом, даже если вы рассталисьвход в 16-байтовые блоки.Каждый блок шифрования AES изменяет IV следующего блока, поэтому вы не можете просто запустить кого-то в середине потока.Вот почему вам нужен постоянный объект CCCryptor на протяжении всего сеанса.

...