mirror of https://github.com/oxen-io/session-ios
Merge branch 'charlesmchen/stress'
Matthew Chen
8 years ago
commit
29dae9bb99
@ -0,0 +1,15 @@
|
||||
//
|
||||
// Copyright (c) 2017 Open Whisper Systems. All rights reserved.
|
||||
//
|
||||
|
||||
#import "DebugUIPage.h"
|
||||
|
||||
NS_ASSUME_NONNULL_BEGIN
|
||||
|
||||
@class TSThread;
|
||||
|
||||
@interface DebugUIStress : DebugUIPage
|
||||
|
||||
@end
|
||||
|
||||
NS_ASSUME_NONNULL_END
|
||||
@ -0,0 +1,505 @@
|
||||
//
|
||||
// Copyright (c) 2017 Open Whisper Systems. All rights reserved.
|
||||
//
|
||||
|
||||
#import "DebugUIStress.h"
|
||||
#import "Environment.h"
|
||||
#import "OWSMessageSender.h"
|
||||
#import "OWSTableViewController.h"
|
||||
#import "ThreadUtil.h"
|
||||
#import <SignalServiceKit/Cryptography.h>
|
||||
#import <SignalServiceKit/NSDate+OWS.h>
|
||||
#import <SignalServiceKit/OWSDynamicOutgoingMessage.h>
|
||||
#import <SignalServiceKit/SecurityUtils.h>
|
||||
#import <SignalServiceKit/TSGroupThread.h>
|
||||
#import <SignalServiceKit/TSStorageManager.h>
|
||||
#import <SignalServiceKit/TSThread.h>
|
||||
|
||||
NS_ASSUME_NONNULL_BEGIN
|
||||
|
||||
@implementation DebugUIStress
|
||||
|
||||
#pragma mark - Logging
|
||||
|
||||
+ (NSString *)tag
|
||||
{
|
||||
return [NSString stringWithFormat:@"[%@]", self.class];
|
||||
}
|
||||
|
||||
- (NSString *)tag
|
||||
{
|
||||
return self.class.tag;
|
||||
}
|
||||
|
||||
#pragma mark - Factory Methods
|
||||
|
||||
- (NSString *)name
|
||||
{
|
||||
return @"Stress";
|
||||
}
|
||||
|
||||
- (nullable OWSTableSection *)sectionForThread:(nullable TSThread *)thread
|
||||
{
|
||||
OWSAssert(thread);
|
||||
|
||||
NSMutableArray<OWSTableItem *> *items = [NSMutableArray new];
|
||||
[items addObject:[OWSTableItem itemWithTitle:@"Send empty message"
|
||||
actionBlock:^{
|
||||
[DebugUIStress sendStressMessage:thread block:^(SignalRecipient *recipient) {
|
||||
return [NSData new];
|
||||
}];
|
||||
}]];
|
||||
[items addObject:[OWSTableItem itemWithTitle:@"Send random noise message"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
NSUInteger contentLength = arc4random_uniform(32);
|
||||
return [Cryptography generateRandomBytes:contentLength];
|
||||
}];
|
||||
}]];
|
||||
[items addObject:[OWSTableItem itemWithTitle:@"Send no payload message"
|
||||
actionBlock:^{
|
||||
[DebugUIStress sendStressMessage:thread block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder = [OWSSignalServiceProtosContentBuilder new];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items addObject:[OWSTableItem itemWithTitle:@"Send empty null message"
|
||||
actionBlock:^{
|
||||
[DebugUIStress sendStressMessage:thread block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder = [OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosNullMessageBuilder *nullMessageBuilder = [OWSSignalServiceProtosNullMessageBuilder new];
|
||||
contentBuilder.nullMessage = [nullMessageBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items
|
||||
addObject:[OWSTableItem itemWithTitle:@"Send random null message"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosNullMessageBuilder *nullMessageBuilder =
|
||||
[OWSSignalServiceProtosNullMessageBuilder new];
|
||||
NSUInteger contentLength = arc4random_uniform(32);
|
||||
nullMessageBuilder.padding =
|
||||
[Cryptography generateRandomBytes:contentLength];
|
||||
contentBuilder.nullMessage = [nullMessageBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items
|
||||
addObject:[OWSTableItem itemWithTitle:@"Send empty sync message"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageBuilder *syncMessageBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageBuilder new];
|
||||
contentBuilder.syncMessage = [syncMessageBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items
|
||||
addObject:[OWSTableItem itemWithTitle:@"Send empty sync sent message"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageBuilder *syncMessageBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageSentBuilder *sentBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageSentBuilder new];
|
||||
syncMessageBuilder.sent = [sentBuilder build];
|
||||
contentBuilder.syncMessage = [syncMessageBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items addObject:[OWSTableItem itemWithTitle:@"Send whitespace text data message"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
dataBuilder.body = @" ";
|
||||
[DebugUIStress ensureGroupOfDataBuilder:dataBuilder
|
||||
thread:thread];
|
||||
contentBuilder.dataMessage = [dataBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items addObject:[OWSTableItem
|
||||
itemWithTitle:@"Send bad attachment data message"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
OWSSignalServiceProtosAttachmentPointerBuilder *attachmentPointer =
|
||||
[OWSSignalServiceProtosAttachmentPointerBuilder new];
|
||||
[attachmentPointer setId:arc4random_uniform(32) + 1];
|
||||
[attachmentPointer setContentType:@"1"];
|
||||
[attachmentPointer setSize:arc4random_uniform(32) + 1];
|
||||
[attachmentPointer setDigest:[Cryptography generateRandomBytes:1]];
|
||||
[attachmentPointer setFileName:@" "];
|
||||
[DebugUIStress ensureGroupOfDataBuilder:dataBuilder thread:thread];
|
||||
contentBuilder.dataMessage = [dataBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items addObject:[OWSTableItem itemWithTitle:@"Send normal text data message"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
dataBuilder.body = @"alice";
|
||||
[DebugUIStress ensureGroupOfDataBuilder:dataBuilder
|
||||
thread:thread];
|
||||
contentBuilder.dataMessage = [dataBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items addObject:[OWSTableItem itemWithTitle:@"Send N text messages with same timestamp"
|
||||
actionBlock:^{
|
||||
uint64_t timestamp = [NSDate ows_millisecondTimeStamp];
|
||||
for (int i = 0; i < 3; i++) {
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
timestamp:timestamp
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
dataBuilder.body = [NSString stringWithFormat:@"%@ %d",
|
||||
[NSUUID UUID].UUIDString,
|
||||
i];
|
||||
[DebugUIStress ensureGroupOfDataBuilder:dataBuilder
|
||||
thread:thread];
|
||||
contentBuilder.dataMessage = [dataBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}
|
||||
}]];
|
||||
[items addObject:[OWSTableItem
|
||||
itemWithTitle:@"Send text message with current timestamp"
|
||||
actionBlock:^{
|
||||
uint64_t timestamp = [NSDate ows_millisecondTimeStamp];
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
timestamp:timestamp
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
dataBuilder.body =
|
||||
[[NSUUID UUID].UUIDString stringByAppendingString:@" now"];
|
||||
[DebugUIStress ensureGroupOfDataBuilder:dataBuilder thread:thread];
|
||||
contentBuilder.dataMessage = [dataBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items addObject:[OWSTableItem
|
||||
itemWithTitle:@"Send text message with future timestamp"
|
||||
actionBlock:^{
|
||||
uint64_t timestamp = [NSDate ows_millisecondTimeStamp];
|
||||
timestamp += kHourInMs;
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
timestamp:timestamp
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
dataBuilder.body =
|
||||
[[NSUUID UUID].UUIDString stringByAppendingString:@" now"];
|
||||
[DebugUIStress ensureGroupOfDataBuilder:dataBuilder thread:thread];
|
||||
contentBuilder.dataMessage = [dataBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items addObject:[OWSTableItem
|
||||
itemWithTitle:@"Send text message with past timestamp"
|
||||
actionBlock:^{
|
||||
uint64_t timestamp = [NSDate ows_millisecondTimeStamp];
|
||||
timestamp -= kHourInMs;
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
timestamp:timestamp
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
dataBuilder.body =
|
||||
[[NSUUID UUID].UUIDString stringByAppendingString:@" now"];
|
||||
[DebugUIStress ensureGroupOfDataBuilder:dataBuilder thread:thread];
|
||||
contentBuilder.dataMessage = [dataBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items addObject:[OWSTableItem itemWithTitle:@"Send N text messages with same timestamp"
|
||||
actionBlock:^{
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
dataBuilder.body = @"alice";
|
||||
contentBuilder.dataMessage = [dataBuilder build];
|
||||
[DebugUIStress ensureGroupOfDataBuilder:dataBuilder thread:thread];
|
||||
NSData *data = [[contentBuilder build] data];
|
||||
|
||||
uint64_t timestamp = [NSDate ows_millisecondTimeStamp];
|
||||
|
||||
for (int i = 0; i < 3; i++) {
|
||||
[DebugUIStress sendStressMessage:thread
|
||||
timestamp:timestamp
|
||||
block:^(SignalRecipient *recipient) {
|
||||
return data;
|
||||
}];
|
||||
}
|
||||
}]];
|
||||
[items
|
||||
addObject:[OWSTableItem itemWithTitle:@"Send malformed sync sent message 1"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageBuilder *syncMessageBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageSentBuilder *sentBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageSentBuilder new];
|
||||
sentBuilder.destination = @"abc";
|
||||
sentBuilder.timestamp = arc4random_uniform(32) + 1;
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
sentBuilder.message = [dataBuilder build];
|
||||
syncMessageBuilder.sent = [sentBuilder build];
|
||||
contentBuilder.syncMessage = [syncMessageBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items
|
||||
addObject:[OWSTableItem itemWithTitle:@"Send malformed sync sent message 2"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageBuilder *syncMessageBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageSentBuilder *sentBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageSentBuilder new];
|
||||
sentBuilder.destination = @"abc";
|
||||
sentBuilder.timestamp = 0;
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
sentBuilder.message = [dataBuilder build];
|
||||
syncMessageBuilder.sent = [sentBuilder build];
|
||||
contentBuilder.syncMessage = [syncMessageBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items
|
||||
addObject:[OWSTableItem itemWithTitle:@"Send malformed sync sent message 3"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageBuilder *syncMessageBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageSentBuilder *sentBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageSentBuilder new];
|
||||
sentBuilder.destination = @"abc";
|
||||
sentBuilder.timestamp = 0;
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
dataBuilder.body = @" ";
|
||||
sentBuilder.message = [dataBuilder build];
|
||||
syncMessageBuilder.sent = [sentBuilder build];
|
||||
contentBuilder.syncMessage = [syncMessageBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items
|
||||
addObject:[OWSTableItem itemWithTitle:@"Send malformed sync sent message 4"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageBuilder *syncMessageBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageSentBuilder *sentBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageSentBuilder new];
|
||||
sentBuilder.destination = @"abc";
|
||||
sentBuilder.timestamp = 0;
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
dataBuilder.body = @" ";
|
||||
OWSSignalServiceProtosGroupContextBuilder *groupBuilder =
|
||||
[OWSSignalServiceProtosGroupContextBuilder new];
|
||||
[groupBuilder setId:[Cryptography generateRandomBytes:1]];
|
||||
dataBuilder.group = [groupBuilder build];
|
||||
sentBuilder.message = [dataBuilder build];
|
||||
syncMessageBuilder.sent = [sentBuilder build];
|
||||
contentBuilder.syncMessage = [syncMessageBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items
|
||||
addObject:[OWSTableItem itemWithTitle:@"Send malformed sync sent message 5"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageBuilder *syncMessageBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageSentBuilder *sentBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageSentBuilder new];
|
||||
sentBuilder.destination = @"abc";
|
||||
sentBuilder.timestamp = 0;
|
||||
OWSSignalServiceProtosDataMessageBuilder *dataBuilder =
|
||||
[OWSSignalServiceProtosDataMessageBuilder new];
|
||||
dataBuilder.body = @" ";
|
||||
OWSSignalServiceProtosGroupContextBuilder *groupBuilder =
|
||||
[OWSSignalServiceProtosGroupContextBuilder new];
|
||||
[groupBuilder setId:[Cryptography generateRandomBytes:1]];
|
||||
dataBuilder.group = [groupBuilder build];
|
||||
sentBuilder.message = [dataBuilder build];
|
||||
syncMessageBuilder.sent = [sentBuilder build];
|
||||
contentBuilder.syncMessage = [syncMessageBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
[items
|
||||
addObject:[OWSTableItem itemWithTitle:@"Send empty sync sent message 6"
|
||||
actionBlock:^{
|
||||
[DebugUIStress
|
||||
sendStressMessage:thread
|
||||
block:^(SignalRecipient *recipient) {
|
||||
OWSSignalServiceProtosContentBuilder *contentBuilder =
|
||||
[OWSSignalServiceProtosContentBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageBuilder *syncMessageBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageBuilder new];
|
||||
OWSSignalServiceProtosSyncMessageSentBuilder *sentBuilder =
|
||||
[OWSSignalServiceProtosSyncMessageSentBuilder new];
|
||||
sentBuilder.destination = @"abc";
|
||||
syncMessageBuilder.sent = [sentBuilder build];
|
||||
contentBuilder.syncMessage = [syncMessageBuilder build];
|
||||
return [[contentBuilder build] data];
|
||||
}];
|
||||
}]];
|
||||
|
||||
if ([thread isKindOfClass:[TSGroupThread class]]) {
|
||||
TSGroupThread *groupThread = (TSGroupThread *)thread;
|
||||
[items addObject:[OWSTableItem itemWithTitle:@"Hallucinate twin group"
|
||||
actionBlock:^{
|
||||
[DebugUIStress hallucinateTwinGroup:groupThread];
|
||||
}]];
|
||||
}
|
||||
return [OWSTableSection sectionWithTitle:self.name items:items];
|
||||
}
|
||||
|
||||
+ (void)ensureGroupOfDataBuilder:(OWSSignalServiceProtosDataMessageBuilder *)dataBuilder thread:(TSThread *)thread
|
||||
{
|
||||
OWSAssert(dataBuilder);
|
||||
OWSAssert(thread);
|
||||
|
||||
if (![thread isKindOfClass:[TSGroupThread class]]) {
|
||||
return;
|
||||
}
|
||||
|
||||
TSGroupThread *groupThread = (TSGroupThread *)thread;
|
||||
OWSSignalServiceProtosGroupContextBuilder *groupBuilder = [OWSSignalServiceProtosGroupContextBuilder new];
|
||||
[groupBuilder setType:OWSSignalServiceProtosGroupContextTypeDeliver];
|
||||
[groupBuilder setId:groupThread.groupModel.groupId];
|
||||
[dataBuilder setGroup:groupBuilder.build];
|
||||
}
|
||||
|
||||
+ (void)sendStressMessage:(TSOutgoingMessage *)message
|
||||
{
|
||||
OWSAssert(message);
|
||||
|
||||
OWSMessageSender *messageSender = [Environment getCurrent].messageSender;
|
||||
[messageSender sendMessage:message
|
||||
success:^{
|
||||
DDLogInfo(@"%@ Successfully sent message.", self.tag);
|
||||
}
|
||||
failure:^(NSError *error) {
|
||||
DDLogWarn(@"%@ Failed to deliver message with error: %@", self.tag, error);
|
||||
}];
|
||||
}
|
||||
|
||||
+ (void)sendStressMessage:(TSThread *)thread
|
||||
block:(DynamicOutgoingMessageBlock)block
|
||||
{
|
||||
OWSAssert(thread);
|
||||
OWSAssert(block);
|
||||
|
||||
OWSDynamicOutgoingMessage *message =
|
||||
[[OWSDynamicOutgoingMessage alloc] initWithPlainTextDataBlock:block thread:thread];
|
||||
|
||||
[self sendStressMessage:message];
|
||||
}
|
||||
|
||||
+ (void)sendStressMessage:(TSThread *)thread timestamp:(uint64_t)timestamp block:(DynamicOutgoingMessageBlock)block
|
||||
{
|
||||
OWSAssert(thread);
|
||||
OWSAssert(block);
|
||||
|
||||
OWSDynamicOutgoingMessage *message =
|
||||
[[OWSDynamicOutgoingMessage alloc] initWithPlainTextDataBlock:block timestamp:timestamp thread:thread];
|
||||
|
||||
[self sendStressMessage:message];
|
||||
}
|
||||
|
||||
// Creates a new group (by cloning the current group) without informing the,
|
||||
// other members. This can be used to test "group info requests", etc.
|
||||
+ (void)hallucinateTwinGroup:(TSGroupThread *)groupThread
|
||||
{
|
||||
__block TSGroupThread *thread;
|
||||
[[TSStorageManager sharedManager].dbReadWriteConnection
|
||||
readWriteWithBlock:^(YapDatabaseReadWriteTransaction *_Nonnull transaction) {
|
||||
TSGroupModel *groupModel =
|
||||
[[TSGroupModel alloc] initWithTitle:[groupThread.groupModel.groupName stringByAppendingString:@" Copy"]
|
||||
memberIds:[groupThread.groupModel.groupMemberIds mutableCopy]
|
||||
image:groupThread.groupModel.groupImage
|
||||
groupId:[SecurityUtils generateRandomBytes:16]];
|
||||
thread = [TSGroupThread getOrCreateThreadWithGroupModel:groupModel transaction:transaction];
|
||||
}];
|
||||
OWSAssert(thread);
|
||||
|
||||
[Environment presentConversationForThread:thread];
|
||||
}
|
||||
|
||||
@end
|
||||
|
||||
NS_ASSUME_NONNULL_END
|
||||
@ -0,0 +1,23 @@
|
||||
//
|
||||
// Copyright (c) 2017 Open Whisper Systems. All rights reserved.
|
||||
//
|
||||
|
||||
#import "TSOutgoingMessage.h"
|
||||
|
||||
NS_ASSUME_NONNULL_BEGIN
|
||||
|
||||
@class OWSSignalServiceProtosDataMessageBuilder;
|
||||
@class SignalRecipient;
|
||||
|
||||
typedef NSData *_Nonnull (^DynamicOutgoingMessageBlock)(SignalRecipient *);
|
||||
|
||||
@interface OWSDynamicOutgoingMessage : TSOutgoingMessage
|
||||
|
||||
- (instancetype)initWithPlainTextDataBlock:(DynamicOutgoingMessageBlock)block thread:(nullable TSThread *)thread;
|
||||
- (instancetype)initWithPlainTextDataBlock:(DynamicOutgoingMessageBlock)block
|
||||
timestamp:(uint64_t)timestamp
|
||||
thread:(nullable TSThread *)thread;
|
||||
|
||||
@end
|
||||
|
||||
NS_ASSUME_NONNULL_END
|
||||
@ -0,0 +1,55 @@
|
||||
//
|
||||
// Copyright (c) 2017 Open Whisper Systems. All rights reserved.
|
||||
//
|
||||
|
||||
#import "OWSDynamicOutgoingMessage.h"
|
||||
#import "NSDate+OWS.h"
|
||||
#import "OWSSignalServiceProtos.pb.h"
|
||||
|
||||
NS_ASSUME_NONNULL_BEGIN
|
||||
|
||||
@interface OWSDynamicOutgoingMessage ()
|
||||
|
||||
@property (nonatomic, readonly) DynamicOutgoingMessageBlock block;
|
||||
|
||||
@end
|
||||
|
||||
#pragma mark -
|
||||
|
||||
@implementation OWSDynamicOutgoingMessage
|
||||
|
||||
- (instancetype)initWithPlainTextDataBlock:(DynamicOutgoingMessageBlock)block thread:(nullable TSThread *)thread
|
||||
{
|
||||
return [self initWithPlainTextDataBlock:block timestamp:[NSDate ows_millisecondTimeStamp] thread:thread];
|
||||
}
|
||||
|
||||
- (instancetype)initWithPlainTextDataBlock:(DynamicOutgoingMessageBlock)block
|
||||
timestamp:(uint64_t)timestamp
|
||||
thread:(nullable TSThread *)thread
|
||||
{
|
||||
self = [super initWithTimestamp:timestamp inThread:thread];
|
||||
|
||||
if (self) {
|
||||
_block = block;
|
||||
}
|
||||
|
||||
return self;
|
||||
}
|
||||
|
||||
- (void)saveWithTransaction:(YapDatabaseReadWriteTransaction *)transaction
|
||||
{
|
||||
// override superclass with no-op.
|
||||
//
|
||||
// There's no need to save this message, since it's not displayed to the user.
|
||||
}
|
||||
|
||||
- (NSData *)buildPlainTextData:(SignalRecipient *)recipient
|
||||
{
|
||||
NSData *plainTextData = self.block(recipient);
|
||||
OWSAssert(plainTextData);
|
||||
return plainTextData;
|
||||
}
|
||||
|
||||
@end
|
||||
|
||||
NS_ASSUME_NONNULL_END
|
||||
@ -1,14 +1,16 @@
|
||||
// Created by Frederic Jacobs on 25/11/14.
|
||||
// Copyright (c) 2014 Open Whisper Systems. All rights reserved.
|
||||
//
|
||||
// Copyright (c) 2017 Open Whisper Systems. All rights reserved.
|
||||
//
|
||||
|
||||
#import "NSDate+OWS.h"
|
||||
#import <chrono>
|
||||
#import "NSDate+millisecondTimeStamp.h"
|
||||
|
||||
@implementation NSDate (millisecondTimeStamp)
|
||||
|
||||
+ (uint64_t)ows_millisecondTimeStamp {
|
||||
uint64_t milliseconds =
|
||||
(uint64_t)(std::chrono::system_clock::now().time_since_epoch() / std::chrono::milliseconds(1));
|
||||
+ (uint64_t)ows_millisecondTimeStamp
|
||||
{
|
||||
uint64_t milliseconds
|
||||
= (uint64_t)(std::chrono::system_clock::now().time_since_epoch() / std::chrono::milliseconds(1));
|
||||
return milliseconds;
|
||||
}
|
||||
|
||||
@ -1,11 +0,0 @@
|
||||
// Created by Frederic Jacobs on 25/11/14.
|
||||
// Copyright (c) 2014 Open Whisper Systems. All rights reserved.
|
||||
|
||||
|
||||
@interface NSDate (millisecondTimeStamp)
|
||||
|
||||
+ (uint64_t)ows_millisecondTimeStamp;
|
||||
+ (NSDate *)ows_dateWithMillisecondsSince1970:(uint64_t)milliseconds;
|
||||
+ (uint64_t)ows_millisecondsSince1970ForDate:(NSDate *)date;
|
||||
|
||||
@end
|
||||
Loading…
Reference in New Issue