oxen-io/session-ios
1
0
Fork 0
mirror of https://github.com/oxen-io/session-ios.git synced 2023-12-13 21:30:14 +01:00
Code Issues Projects Releases Wiki Activity

Send database action in debug UI.

Browse source
This commit is contained in:
Matthew Chen 2018-01-04 10:50:23 -05:00
parent eefd66e4a6
commit 3a9886bb2b
1 changed files with 85 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

87
Signal/src/ViewControllers/DebugUI/DebugUIMisc.m
View file

@ -1,5 +1,5 @@
//
// Copyright (c) 2017 Open Whisper Systems. All rights reserved.
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
//
#import "DebugUIMisc.h"
@ -22,12 +22,22 @@
NS_ASSUME_NONNULL_BEGIN
@interface TSAccountManager (Debug)
@interface TSAccountManager (DebugUI)
- (void)resetForRegistration;
@end
#pragma mark -
@interface OWSStorage (DebugUI)
- (NSData *)databasePassword;
@end
#pragma mark -
@implementation DebugUIMisc
#pragma mark - Factory Methods
@ -72,6 +82,18 @@ NS_ASSUME_NONNULL_BEGIN
}];
}]];
if (thread) {
[items addObject:[OWSTableItem itemWithTitle:@"Send Encrypted Database"
actionBlock:^{
[DebugUIMisc sendEncryptedDatabase:thread];
}]];
[items addObject:[OWSTableItem itemWithTitle:@"Send Unencrypted Database"
actionBlock:^{
[DebugUIMisc sendUnencryptedDatabase:thread];
}]];
}
return [OWSTableSection sectionWithTitle:self.name items:items];
}
@ -140,6 +162,67 @@ NS_ASSUME_NONNULL_BEGIN
}];
}
+ (void)sendEncryptedDatabase:(TSThread *)thread
{
NSString *temporaryDirectory = NSTemporaryDirectory();
NSString *fileName = [[NSUUID UUID].UUIDString stringByAppendingString:@".sqlite"];
NSString *filePath = [temporaryDirectory stringByAppendingPathComponent:fileName];
__block BOOL success;
[TSStorageManager.sharedManager.newDatabaseConnection
readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction) {
NSError *error;
success = [[NSFileManager defaultManager] copyItemAtPath:TSStorageManager.databaseFilePath
toPath:filePath
error:&error];
if (!success || error) {
OWSFail(@"%@ Could not copy database file: %@.", self.logTag, error);
success = NO;
}
}];
if (!success) {
return;
}
OWSMessageSender *messageSender = [Environment current].messageSender;
NSString *utiType = [MIMETypeUtil utiTypeForFileExtension:fileName.pathExtension];
DataSource *_Nullable dataSource = [DataSourcePath dataSourceWithFilePath:filePath];
[dataSource setSourceFilename:fileName];
SignalAttachment *attachment = [SignalAttachment attachmentWithDataSource:dataSource dataUTI:utiType];
NSData *databasePassword = [TSStorageManager.sharedManager databasePassword];
attachment.captionText = [databasePassword hexadecimalString];
if (!attachment || [attachment hasError]) {
OWSFail(@"%@ attachment[%@]: %@", self.logTag, [attachment sourceFilename], [attachment errorName]);
return;
}
[ThreadUtil sendMessageWithAttachment:attachment inThread:thread messageSender:messageSender completion:nil];
}
+ (void)sendUnencryptedDatabase:(TSThread *)thread
{
NSString *temporaryDirectory = NSTemporaryDirectory();
NSString *fileName = [[NSUUID UUID].UUIDString stringByAppendingString:@".sqlite"];
NSString *filePath = [temporaryDirectory stringByAppendingPathComponent:fileName];
NSError *error = [TSStorageManager.sharedManager.newDatabaseConnection backupToPath:filePath];
if (error) {
OWSFail(@"%@ Could not copy database file: %@.", self.logTag, error);
return;
}
OWSMessageSender *messageSender = [Environment current].messageSender;
NSString *utiType = [MIMETypeUtil utiTypeForFileExtension:fileName.pathExtension];
DataSource *_Nullable dataSource = [DataSourcePath dataSourceWithFilePath:filePath];
[dataSource setSourceFilename:fileName];
SignalAttachment *attachment = [SignalAttachment attachmentWithDataSource:dataSource dataUTI:utiType];
if (!attachment || [attachment hasError]) {
OWSFail(@"%@ attachment[%@]: %@", self.logTag, [attachment sourceFilename], [attachment errorName]);
return;
}
[ThreadUtil sendMessageWithAttachment:attachment inThread:thread messageSender:messageSender completion:nil];
}
@end
NS_ASSUME_NONNULL_END