01518551eb
The only modifications are those necessary to build as part of SSK. The follow-up commits will clean that up a bit.
45 lines
1.1 KiB
Objective-C
45 lines
1.1 KiB
Objective-C
//
|
|
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "NSData+messagePadding.h"
|
|
#import <SignalCoreKit/Cryptography.h>
|
|
#import <XCTest/XCTest.h>
|
|
|
|
@interface MessagePaddingTests : XCTestCase
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@implementation MessagePaddingTests
|
|
|
|
- (void)testV3Padding
|
|
{
|
|
for (NSUInteger i = 0; i < 159; i++) {
|
|
NSData *data = [NSMutableData dataWithLength:i];
|
|
XCTAssertTrue([data paddedMessageBody].length == 159);
|
|
}
|
|
|
|
for (NSUInteger i = 159; i < 319; i++) {
|
|
NSData *data = [NSMutableData dataWithLength:i];
|
|
XCTAssertTrue([data paddedMessageBody].length == 319);
|
|
}
|
|
|
|
for (NSUInteger i = 319; i < 479; i++) {
|
|
NSData *data = [NSMutableData dataWithLength:i];
|
|
XCTAssertTrue([data paddedMessageBody].length == 479);
|
|
}
|
|
}
|
|
|
|
- (void)testV3RandomPadding
|
|
{
|
|
for (int i = 0; i < 1000; i++) {
|
|
NSData *randomMessage = [Cryptography generateRandomBytes:501];
|
|
NSData *paddedMessage = [randomMessage paddedMessageBody];
|
|
XCTAssertTrue([[paddedMessage removePadding] isEqualToData:randomMessage]);
|
|
}
|
|
}
|
|
|
|
@end
|