FreeRDP/client/iOS/Models/ConnectionParams.m

/*
 Connection Parameters abstraction
 
 Copyright 2013 Thinstuff Technologies GmbH, Author: Dorian Johnson
 
 This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0. 
 If a copy of the MPL was not distributed with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */


#import "ConnectionParams.h"
#import "GlobalDefaults.h"
#import "EncryptionController.h"
#import "Utils.h"
#import "TSXAdditions.h"

@interface ConnectionParams (Private)
- (id)initWithConnectionParams:(ConnectionParams*)params;
@end

@implementation ConnectionParams

// Designated initializer.
- (id)initWithDictionary:(NSDictionary*)dict
{
	if (!(self = [super init]))
		return nil;
	
	_connection_params = [dict mutableDeepCopy];
	
    [self decryptPasswordForKey:@"password"];
    [self decryptPasswordForKey:@"tsg_password"];
	
	return self;
}

- (void)decryptPasswordForKey:(NSString*)key
{
	if ([[_connection_params objectForKey:key] isKindOfClass:[NSData class]])
	{
		NSString* plaintext_password = [[[EncryptionController sharedEncryptionController] decryptor] decryptString:[_connection_params objectForKey:key]];
		[self setValue:plaintext_password forKey:key];
	}
}

- (id)initWithBaseDefaultParameters
{
	return [self initWithDictionary:[[NSUserDefaults standardUserDefaults] dictionaryForKey:@"TSXDefaultComputerBookmarkSettings"]];
}

- (id)init
{
	return [self initWithDictionary:[NSDictionary dictionary]];
}

- (id)initWithConnectionParams:(ConnectionParams*)params
{
	return [self initWithDictionary:params->_connection_params];
}
- (void)dealloc
{
	[_connection_params release]; _connection_params = nil; 
	[super dealloc];
}

- (id)copyWithZone:(NSZone *)zone
{
	return [[ConnectionParams alloc] initWithDictionary:_connection_params];
}

- (NSString*)description
{
	return [NSString stringWithFormat:@"ConnectionParams: %@", [_connection_params description]];
}

#pragma mark -
#pragma mark NSCoder

- (id)initWithCoder:(NSCoder *)decoder
{
	if ([decoder containsValueForKey:@"connectionParams"])
		return [self initWithDictionary:[decoder decodeObjectForKey:@"connectionParams"]];
	
	return [self init];
}

- (void)encodeWithCoder:(NSCoder *)coder
{	
	NSSet* unserializable_keys = [NSSet setWithObjects:@"view", nil];
	NSMutableDictionary* serializable_params = [[NSMutableDictionary alloc] initWithCapacity:[_connection_params count]];
	
	for (NSString* k in _connection_params)
		if ( ([k characterAtIndex:0] != '_') && ![unserializable_keys containsObject:k])
			[serializable_params setObject:[_connection_params objectForKey:k] forKey:k];
	
	if ([serializable_params objectForKey:@"password"] != nil)
        [self serializeDecryptedForKey:@"password" forParams:serializable_params];
	if ([serializable_params objectForKey:@"tsg_password"] != nil)
        [self serializeDecryptedForKey:@"tsg_password" forParams:serializable_params];
	
	[coder encodeObject:serializable_params forKey:@"connectionParams"];
	[serializable_params release];
}

- (void)serializeDecryptedForKey:(NSString*)key forParams:(NSMutableDictionary*)params
{
    NSData* encrypted_password = [[[EncryptionController sharedEncryptionController] encryptor] encryptString:[params objectForKey:key]];
    
    if (encrypted_password)
        [params setObject:encrypted_password forKey:key];
    else
        [params removeObjectForKey:key];
}

#pragma mark -
#pragma mark NSKeyValueCoding

- (void)setValue:(id)value forKey:(NSString *)key
{
	[self willChangeValueForKey:key];
	
	if (value == nil)
		[self setNilValueForKey:key];
	else
		[_connection_params setValue:value forKey:key];
	
	[self didChangeValueForKey:key];
}

- (void)setValue:(id)value forKeyPath:(NSString *)key
{
	[self willChangeValueForKey:key];
	
	if (value == nil)
		[self setNilValueForKey:key];
	else
		[_connection_params setValue:value forKeyPath:key];
	
	[self didChangeValueForKey:key];
}

- (void)setNilValueForKey:(NSString *)key
{
	[_connection_params removeObjectForKey:key];
}

- (id)valueForKey:(NSString*)key
{
	return [_connection_params valueForKey:key];
}

- (NSArray*)allKeys
{
	return [_connection_params allKeys];
}

#pragma mark -
#pragma mark KV convenience

- (BOOL)hasValueForKey:(NSString*)key
{
	return [_connection_params objectForKey:key] != nil;
}

- (void)setInt:(int)integer forKey:(NSString*)key
{
	[self setValue:[NSNumber numberWithInteger:integer] forKey:key];
}
- (int)intForKey:(NSString*)key
{
	return [[self valueForKey:key] intValue];
}


- (void)setBool:(BOOL)v forKey:(NSString*)key
{
	[self setValue:[NSNumber numberWithBool:v] forKey:key];
}
- (BOOL)boolForKey:(NSString*)key
{
	return [[_connection_params objectForKey:key] boolValue];
}


- (const char*)UTF8StringForKey:(NSString*)key
{
	id val = [self valueForKey:key];
	const char* str;
	
	if ([val respondsToSelector:@selector(UTF8String)] && (str = [val UTF8String]))
		return str;
	
	return "";
}

- (NSString*)StringForKey:(NSString*)key
{
	return [self valueForKey:key];
}

- (BOOL)hasValueForKeyPath:(NSString*)key
{
    return [_connection_params valueForKeyPath:key] != nil;    
}

- (void)setInt:(int)integer forKeyPath:(NSString*)key
{
	[self setValue:[NSNumber numberWithInteger:integer] forKeyPath:key];
}
- (int)intForKeyPath:(NSString*)key
{
	return [[self valueForKeyPath:key] intValue];
}

- (void)setBool:(BOOL)v forKeyPath:(NSString*)key
{
	[self setValue:[NSNumber numberWithBool:v] forKeyPath:key];
}

- (BOOL)boolForKeyPath:(NSString*)key
{
	return [[self valueForKeyPath:key] boolValue];
}

- (const char*)UTF8StringForKeyPath:(NSString*)key
{
	id val = [self valueForKeyPath:key];
	const char* str;
	
	if ([val respondsToSelector:@selector(UTF8String)] && (str = [val UTF8String]))
		return str;
	
	return "";
}

- (NSString*)StringForKeyPath:(NSString*)key
{
	return [self valueForKeyPath:key];
}

- (int)intForKey:(NSString*)key with3GEnabled:(BOOL)enabled
{
    if (enabled && [self boolForKey:@"enable_3g_settings"])
        return [self intForKeyPath:[NSString stringWithFormat:@"settings_3g.%@", key]];
    return [self intForKeyPath:key];
}

- (BOOL)boolForKey:(NSString*)key with3GEnabled:(BOOL)enabled
{
    if (enabled && [self boolForKey:@"enable_3g_settings"])
        return [self boolForKeyPath:[NSString stringWithFormat:@"settings_3g.%@", key]];
    return [self boolForKeyPath:key];    
}

@end