FreeRDP/server/Windows/wf_peer.c

293 lines
7.8 KiB
C
Raw Normal View History

/**
* FreeRDP: A Remote Desktop Protocol Client
* FreeRDP Windows Server
*
* Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
*
* 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.
*/
2012-08-23 23:36:34 +04:00
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <winpr/tchar.h>
#include <winpr/windows.h>
2012-08-23 23:36:34 +04:00
#include <freerdp/listener.h>
#include <freerdp/utils/sleep.h>
#include <freerdp/codec/rfx.h>
#include <freerdp/utils/stream.h>
#include "wf_mirage.h"
#include "wf_peer.h"
void wf_peer_context_new(freerdp_peer* client, wfPeerContext* context)
{
2012-08-18 22:34:24 +04:00
wfInfoSingleton = wf_info_init(wfInfoSingleton);
wf_info_mirror_init(wfInfoSingleton, context);
}
void wf_peer_context_free(freerdp_peer* client, wfPeerContext* context)
2012-08-23 23:36:34 +04:00
{
wf_info_subscriber_release(wfInfoSingleton, context);
}
static DWORD WINAPI wf_peer_mirror_monitor(LPVOID lpParam)
{
2012-08-23 23:36:34 +04:00
DWORD start, end, diff;
DWORD rate;
freerdp_peer* client;
2012-08-15 23:27:03 +04:00
rate = 42;
client = (freerdp_peer*)lpParam;
2012-08-18 22:34:24 +04:00
//todo: make sure we dont encode after no clients
while(1)
{
2012-08-15 23:27:03 +04:00
2012-08-23 23:36:34 +04:00
start = GetTickCount();
if(wf_info_has_subscribers(wfInfoSingleton))
{
wf_info_update_changes(wfInfoSingleton);
if(wf_info_have_updates(wfInfoSingleton))
{
//wf_info_find_invalid_region(wfInfoSingleton);
//printf("Fake Encode!\n");
wf_rfx_encode(client);
2012-08-23 23:36:34 +04:00
}
2012-08-15 23:27:03 +04:00
}
end = GetTickCount();
2012-08-23 23:36:34 +04:00
diff = end - start;
2012-08-18 22:34:24 +04:00
if(diff < rate)
{
//printf("sleeping for %d ms...\n", rate - diff);
2012-08-23 23:36:34 +04:00
Sleep(rate - diff);
}
}
_tprintf(_T("monitor thread terminating...\n"));
2012-08-21 02:19:17 +04:00
wf_info_set_thread_count(wfInfoSingleton, wf_info_get_thread_count(wfInfoSingleton) - 1 );
return 0;
}
void wf_rfx_encode(freerdp_peer* client)
{
STREAM* s;
wfInfo* wfi;
RFX_RECT rect;
rdpUpdate* update;
wfPeerContext* wfp;
SURFACE_BITS_COMMAND* cmd;
GETCHANGESBUF* buf;
long height, width;
long offset;
int dRes;
wfp = (wfPeerContext*) client->context;
2012-08-20 19:45:48 +04:00
dRes = WaitForSingleObject(wfInfoSingleton->encodeMutex, INFINITE);
switch(dRes)
{
case WAIT_OBJECT_0:
wf_info_find_invalid_region(wfInfoSingleton);
2012-08-20 19:45:48 +04:00
if( (wfp->activated == false) ||
2012-08-21 02:19:17 +04:00
(wf_info_has_subscribers(wfInfoSingleton) == false) ||
2012-08-21 20:50:21 +04:00
!wf_info_have_invalid_region(wfInfoSingleton) ||
(wfInfoSingleton->enc_data == true) )
2012-08-20 19:45:48 +04:00
{
ReleaseMutex(wfInfoSingleton->encodeMutex);
break;
}
2012-08-21 02:19:17 +04:00
update = client->update;
cmd = &update->surface_bits_command;
wfi = wfp->wfInfo;
buf = (GETCHANGESBUF*)wfi->changeBuffer;
2012-08-18 22:34:24 +04:00
2012-08-21 02:19:17 +04:00
//printf("encode %d\n", wfi->nextUpdate - wfi->lastUpdate);
//printf("\tinvlaid region = (%d, %d), (%d, %d)\n", wfi->invalid_x1, wfi->invalid_y1, wfi->invalid_x2, wfi->invalid_y2);
width = wfi->invalid_x2 - wfi->invalid_x1;
height = wfi->invalid_y2 - wfi->invalid_y1;
stream_clear(wfp->s);
stream_set_pos(wfp->s, 0);
s = wfp->s;
rect.x = 0;
rect.y = 0;
2012-08-23 23:36:34 +04:00
rect.width = (uint16) width;
rect.height = (uint16) height;
offset = (4 * wfi->invalid_x1) + (wfi->invalid_y1 * wfi->width * 4);
//printf("width = %d, height = %d\n", width, height);
rfx_compose_message(wfp->rfx_context, s, &rect, 1,
((uint8*) (buf->Userbuffer)) + offset, width, height, wfi->width * 4);
cmd->destLeft = wfi->invalid_x1;
cmd->destTop = wfi->invalid_y1;
cmd->destRight = wfi->invalid_x1 + width;
cmd->destBottom = wfi->invalid_y1 + height;
cmd->bpp = 32;
cmd->codecID = client->settings->rfx_codec_id;
cmd->width = width;
cmd->height = height;
cmd->bitmapDataLength = stream_get_length(s);
cmd->bitmapData = stream_get_head(s);
2012-08-19 21:20:10 +04:00
2012-08-20 19:45:48 +04:00
wfi->enc_data = true;
ReleaseMutex(wfInfoSingleton->encodeMutex);
break;
case WAIT_TIMEOUT:
2012-08-19 21:20:10 +04:00
ReleaseMutex(wfInfoSingleton->encodeMutex);
break;
2012-08-21 02:19:17 +04:00
case WAIT_ABANDONED:
printf("\n\nwf_rfx_encode: Got ownership of abandoned mutex... releasing...\n", dRes);
ReleaseMutex(wfInfoSingleton->encodeMutex);
break;
default:
2012-08-21 02:19:17 +04:00
printf("\n\nwf_rfx_encode: Something else happened!!! dRes = %d\n", dRes);
}
}
void wf_peer_init(freerdp_peer* client)
{
client->context_size = sizeof(wfPeerContext);
client->ContextNew = (psPeerContextNew) wf_peer_context_new;
client->ContextFree = (psPeerContextFree) wf_peer_context_free;
freerdp_peer_context_new(client);
2012-08-23 23:36:34 +04:00
if (!wf_info_get_thread_count(wfInfoSingleton))
2012-08-21 02:19:17 +04:00
{
_tprintf(_T("Trying to create a monitor thread...\n"));
if (CreateThread(NULL, 0, wf_peer_mirror_monitor, client, 0, NULL) != 0)
_tprintf(_T("Created!\n"));
2012-08-21 02:19:17 +04:00
wf_info_set_thread_count(wfInfoSingleton, wf_info_get_thread_count(wfInfoSingleton) + 1 );
}
}
boolean wf_peer_post_connect(freerdp_peer* client)
{
wfPeerContext* context = (wfPeerContext*) client->context;
/**
* This callback is called when the entire connection sequence is done, i.e. we've received the
* Font List PDU from the client and sent out the Font Map PDU.
* The server may start sending graphics output and receiving keyboard/mouse input after this
* callback returns.
*/
printf("Client %s is activated (osMajorType %d osMinorType %d)", client->local ? "(local)" : client->hostname,
client->settings->os_major_type, client->settings->os_minor_type);
if (client->settings->autologon)
{
printf(" and wants to login automatically as %s\\%s",
client->settings->domain ? client->settings->domain : "",
client->settings->username);
/* A real server may perform OS login here if NLA is not executed previously. */
}
printf("\n");
printf("Client requested desktop: %dx%dx%d\n",
client->settings->width, client->settings->height, client->settings->color_depth);
printf("But we will try resizing to %dx%d\n",
wf_info_get_width(wfInfoSingleton),
wf_info_get_height(wfInfoSingleton)
);
client->settings->width = wf_info_get_width(wfInfoSingleton);
client->settings->height = wf_info_get_height(wfInfoSingleton);
client->update->DesktopResize(client->update->context);
return true;
}
boolean wf_peer_activate(freerdp_peer* client)
{
wfPeerContext* context = (wfPeerContext*) client->context;
context->activated = true;
return true;
}
void wf_peer_synchronize_event(rdpInput* input, uint32 flags)
{
}
2012-08-18 22:34:24 +04:00
void wf_peer_send_changes(rdpUpdate* update)
2012-08-18 22:34:24 +04:00
{
int dRes;
//are we currently encoding?
2012-08-19 21:20:10 +04:00
dRes = WaitForSingleObject(wfInfoSingleton->encodeMutex, 0);
switch(dRes)
{
case WAIT_OBJECT_0:
//are there changes to send?
2012-08-20 19:45:48 +04:00
if( !wf_info_have_updates(wfInfoSingleton) || !wf_info_have_invalid_region(wfInfoSingleton) || (wfInfoSingleton->enc_data == false) )
2012-08-19 21:20:10 +04:00
{
ReleaseMutex(wfInfoSingleton->encodeMutex);
break;
}
wf_info_updated(wfInfoSingleton);
/*
2012-08-20 19:45:48 +04:00
printf("\tSend...");
printf("\t(%d, %d), (%d, %d) [%dx%d]\n",
update->surface_bits_command.destLeft, update->surface_bits_command.destTop,
update->surface_bits_command.destRight, update->surface_bits_command.destBottom,
update->surface_bits_command.width, update->surface_bits_command.height);
*/
update->SurfaceBits(update->context, &update->surface_bits_command);
//wf_info_clear_invalid_region(wfInfoSingleton);
2012-08-20 19:45:48 +04:00
wfInfoSingleton->enc_data = false;
2012-08-19 21:20:10 +04:00
ReleaseMutex(wfInfoSingleton->encodeMutex);
break;
2012-08-19 21:20:10 +04:00
case WAIT_TIMEOUT:
break;
default:
2012-08-21 02:19:17 +04:00
printf("wf_peer_send_changes: Something else happened!!! dRes = %d\n", dRes);
}
2012-08-18 22:34:24 +04:00
2012-08-19 21:20:10 +04:00
2012-08-23 23:36:34 +04:00
}