/*

 RDP ui callbacks

 Copyright 2013 Thincast Technologies GmbH, Authors: Martin Fleisz, 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 <Foundation/Foundation.h>

#import <freerdp/gdi/gdi.h>

#import "ios_freerdp_ui.h"

#import "RDPSession.h"

#pragma mark -

#pragma mark Certificate authentication

static void ios_resize_display_buffer(mfInfo *mfi);

static BOOL ios_ui_authenticate_raw(freerdp *instance, char **username, char **password,

                                    char **domain, const char *title)

{

	mfInfo *mfi = MFI_FROM_INSTANCE(instance);

	NSMutableDictionary *params = [NSMutableDictionary

	    dictionaryWithObjectsAndKeys:(*username) ? [NSString stringWithUTF8String:*username] : @"",

	                                 @"username",

	                                 (*password) ? [NSString stringWithUTF8String:*password] : @"",

	                                 @"password",

	                                 (*domain) ? [NSString stringWithUTF8String:*domain] : @"",

	                                 @"domain",

	                                 [NSString

	                                     stringWithUTF8String:instance->settings->ServerHostname],

	                                 @"hostname", // used for the auth prompt message; not changed

	                                 nil];

	// request auth UI

	[mfi->session performSelectorOnMainThread:@selector(sessionRequestsAuthenticationWithParams:)

	                               withObject:params

	                            waitUntilDone:YES];

	// wait for UI request to be completed

	[[mfi->session uiRequestCompleted] lock];

	[[mfi->session uiRequestCompleted] wait];

	[[mfi->session uiRequestCompleted] unlock];

	if (![[params valueForKey:@"result"] boolValue])

	{

		mfi->unwanted = YES;

		return FALSE;

	}

	// Free old values

	free(*username);

	free(*password);

	free(*domain);

	// set values back

	*username = strdup([[params objectForKey:@"username"] UTF8String]);

	*password = strdup([[params objectForKey:@"password"] UTF8String]);

	*domain = strdup([[params objectForKey:@"domain"] UTF8String]);

	if (!(*username) || !(*password) || !(*domain))

	{

		free(*username);

		free(*password);

		free(*domain);

		return FALSE;

	}

	return TRUE;

}

BOOL ios_ui_authenticate(freerdp *instance, char **username, char **password, char **domain)

{

	return ios_ui_authenticate_raw(instance, username, password, domain, "");

}

BOOL ios_ui_gw_authenticate(freerdp *instance, char **username, char **password, char **domain)

{

	return ios_ui_authenticate_raw(instance, username, password, domain, "gateway");

}

DWORD ios_ui_verify_certificate(freerdp *instance, const char *common_name, const char *subject,

                                const char *issuer, const char *fingerprint, BOOL host_mismatch)

{

	// check whether we accept all certificates

	if ([[NSUserDefaults standardUserDefaults] boolForKey:@"security.accept_certificates"] == YES)

		return 2;

	mfInfo *mfi = MFI_FROM_INSTANCE(instance);

	NSMutableDictionary *params = [NSMutableDictionary

	    dictionaryWithObjectsAndKeys:(subject) ? [NSString stringWithUTF8String:subject] : @"",

	                                 @"subject",

	                                 (issuer) ? [NSString stringWithUTF8String:issuer] : @"",

	                                 @"issuer",

	                                 (fingerprint) ? [NSString stringWithUTF8String:subject] : @"",

	                                 @"fingerprint", nil];

	// request certificate verification UI

	[mfi->session performSelectorOnMainThread:@selector(sessionVerifyCertificateWithParams:)

	                               withObject:params

	                            waitUntilDone:YES];

	// wait for UI request to be completed

	[[mfi->session uiRequestCompleted] lock];

	[[mfi->session uiRequestCompleted] wait];

	[[mfi->session uiRequestCompleted] unlock];

	if (![[params valueForKey:@"result"] boolValue])

	{

		mfi->unwanted = YES;

		return 0;

	}

	return 1;

}

DWORD ios_ui_verify_changed_certificate(freerdp *instance, const char *common_name,

                                        const char *subject, const char *issuer,

                                        const char *new_fingerprint, const char *old_subject,

                                        const char *old_issuer, const char *old_fingerprint)

{

	return ios_ui_verify_certificate(instance, common_name, subject, issuer, new_fingerprint,

	                                 FALSE);

}

#pragma mark -

#pragma mark Graphics updates

BOOL ios_ui_begin_paint(rdpContext *context)

{

	rdpGdi *gdi = context->gdi;

	gdi->primary->hdc->hwnd->invalid->null = TRUE;

	return TRUE;

}

BOOL ios_ui_end_paint(rdpContext *context)

{

	mfInfo *mfi = MFI_FROM_INSTANCE(context->instance);

	rdpGdi *gdi = context->gdi;

	CGRect dirty_rect =

	    CGRectMake(gdi->primary->hdc->hwnd->invalid->x, gdi->primary->hdc->hwnd->invalid->y,

	               gdi->primary->hdc->hwnd->invalid->w, gdi->primary->hdc->hwnd->invalid->h);

	if (!gdi->primary->hdc->hwnd->invalid->null)

		[mfi->session performSelectorOnMainThread:@selector(setNeedsDisplayInRectAsValue:)

		                               withObject:[NSValue valueWithCGRect:dirty_rect]

		                            waitUntilDone:NO];

	return TRUE;

}

BOOL ios_ui_resize_window(rdpContext *context)

{

	rdpSettings *settings;

	rdpGdi *gdi;

	if (!context || !context->settings)

		return FALSE;

	settings = context->settings;

	gdi = context->gdi;

	if (!gdi_resize(gdi, settings->DesktopWidth, settings->DesktopHeight))

		return FALSE;

	ios_resize_display_buffer(MFI_FROM_INSTANCE(context->instance));

	return TRUE;

}

#pragma mark -

#pragma mark Exported

static void ios_create_bitmap_context(mfInfo *mfi)

{

	[mfi->session performSelectorOnMainThread:@selector(sessionBitmapContextWillChange)

	                               withObject:nil

	                            waitUntilDone:YES];

	rdpGdi *gdi = mfi->instance->context->gdi;

	CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();

	if (GetBytesPerPixel(gdi->dstFormat) == 2)

		mfi->bitmap_context = CGBitmapContextCreate(

		    gdi->primary_buffer, gdi->width, gdi->height, 5, gdi->stride, colorSpace,

		    kCGBitmapByteOrder16Little | kCGImageAlphaNoneSkipFirst);

	else

		mfi->bitmap_context = CGBitmapContextCreate(

		    gdi->primary_buffer, gdi->width, gdi->height, 8, gdi->stride, colorSpace,

		    kCGBitmapByteOrder32Little | kCGImageAlphaNoneSkipFirst);

	CGColorSpaceRelease(colorSpace);

	[mfi->session performSelectorOnMainThread:@selector(sessionBitmapContextDidChange)

	                               withObject:nil

	                            waitUntilDone:YES];

}

void ios_allocate_display_buffer(mfInfo *mfi)

{

	ios_create_bitmap_context(mfi);

}

void ios_resize_display_buffer(mfInfo *mfi)

{

	// Release the old context in a thread-safe manner

	CGContextRef old_context = mfi->bitmap_context;

	mfi->bitmap_context = NULL;

	CGContextRelease(old_context);

	// Create the new context

	ios_create_bitmap_context(mfi);

}