/** * FreeRDP: A Remote Desktop Protocol Implementation * * Copyright 2014 Marc-Andre Moreau * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include #include #include #include #include #include #include "shadow.h" void shadow_client_context_new(freerdp_peer* peer, rdpShadowClient* client) { rdpShadowServer* server; server = (rdpShadowServer*) peer->ContextExtra; client->server = server; } void shadow_client_context_free(freerdp_peer* peer, rdpShadowClient* client) { } BOOL shadow_client_get_fds(freerdp_peer* peer, void** rfds, int* rcount) { return TRUE; } BOOL shadow_client_check_fds(freerdp_peer* peer) { return TRUE; } BOOL shadow_client_capabilities(freerdp_peer* peer) { return TRUE; } BOOL shadow_client_post_connect(freerdp_peer* peer) { fprintf(stderr, "Client %s is activated", peer->hostname); if (peer->settings->AutoLogonEnabled) { fprintf(stderr, " and wants to login automatically as %s\\%s", peer->settings->Domain ? peer->settings->Domain : "", peer->settings->Username); } fprintf(stderr, "\n"); fprintf(stderr, "Client requested desktop: %dx%dx%d\n", peer->settings->DesktopWidth, peer->settings->DesktopHeight, peer->settings->ColorDepth); if (!peer->settings->RemoteFxCodec) { fprintf(stderr, "Client does not support RemoteFX\n"); return FALSE; } //client->settings->DesktopWidth = 1024; //client->settings->DesktopHeight = 768; peer->update->DesktopResize(peer->update->context); return TRUE; } BOOL shadow_client_activate(freerdp_peer* peer) { return TRUE; } static const char* makecert_argv[4] = { "makecert", "-rdp", "-live", "-silent" }; static int makecert_argc = (sizeof(makecert_argv) / sizeof(char*)); int shadow_generate_certificate(rdpSettings* settings) { char* serverFilePath; MAKECERT_CONTEXT* context; serverFilePath = GetCombinedPath(settings->ConfigPath, "server"); if (!PathFileExistsA(serverFilePath)) CreateDirectoryA(serverFilePath, 0); settings->CertificateFile = GetCombinedPath(serverFilePath, "server.crt"); settings->PrivateKeyFile = GetCombinedPath(serverFilePath, "server.key"); if ((!PathFileExistsA(settings->CertificateFile)) || (!PathFileExistsA(settings->PrivateKeyFile))) { context = makecert_context_new(); makecert_context_process(context, makecert_argc, (char**) makecert_argv); makecert_context_set_output_file_name(context, "server"); if (!PathFileExistsA(settings->CertificateFile)) makecert_context_output_certificate_file(context, serverFilePath); if (!PathFileExistsA(settings->PrivateKeyFile)) makecert_context_output_private_key_file(context, serverFilePath); makecert_context_free(context); } free(serverFilePath); return 0; } void* shadow_client_thread(rdpShadowClient* client) { DWORD status; DWORD nCount; HANDLE events[32]; HANDLE ClientEvent; freerdp_peer* peer; rdpSettings* settings; peer = ((rdpContext*) client)->peer; settings = peer->settings; shadow_generate_certificate(settings); settings->RemoteFxCodec = TRUE; settings->ColorDepth = 32; settings->NlaSecurity = FALSE; settings->TlsSecurity = TRUE; settings->RdpSecurity = FALSE; peer->Capabilities = shadow_client_capabilities; peer->PostConnect = shadow_client_post_connect; peer->Activate = shadow_client_activate; shadow_input_register_callbacks(peer->input); peer->Initialize(peer); ClientEvent = peer->GetEventHandle(peer); while (1) { nCount = 0; events[nCount++] = ClientEvent; status = WaitForMultipleObjects(nCount, events, FALSE, INFINITE); if (WaitForSingleObject(ClientEvent, 0) == WAIT_OBJECT_0) { if (!peer->CheckFileDescriptor(peer)) { fprintf(stderr, "Failed to check FreeRDP file descriptor\n"); break; } } } peer->Disconnect(peer); freerdp_peer_context_free(peer); freerdp_peer_free(peer); ExitThread(0); return NULL; } void shadow_client_accepted(freerdp_listener* listener, freerdp_peer* peer) { HANDLE thread; rdpShadowClient* client; rdpShadowServer* server; server = (rdpShadowServer*) listener->info; peer->ContextExtra = (void*) server; peer->ContextSize = sizeof(rdpShadowClient); peer->ContextNew = (psPeerContextNew) shadow_client_context_new; peer->ContextFree = (psPeerContextFree) shadow_client_context_free; freerdp_peer_context_new(peer); client = (rdpShadowClient*) peer->context; thread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) shadow_client_thread, client, 0, NULL); }