/** * FreeRDP: A Remote Desktop Protocol Implementation * Activation Sequence * * Copyright 2011 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 "activation.h" /* static const char* const CTRLACTION_STRINGS[] = { "", "CTRLACTION_REQUEST_CONTROL", "CTRLACTION_GRANTED_CONTROL", "CTRLACTION_DETACH", "CTRLACTION_COOPERATE" }; */ void rdp_write_synchronize_pdu(STREAM* s, rdpSettings* settings) { stream_write_UINT16(s, SYNCMSGTYPE_SYNC); /* messageType (2 bytes) */ stream_write_UINT16(s, settings->PduSource); /* targetUser (2 bytes) */ } BOOL rdp_recv_synchronize_pdu(rdpRdp* rdp, STREAM* s) { if (rdp->settings->ServerMode) return rdp_recv_server_synchronize_pdu(rdp, s); else return rdp_recv_client_synchronize_pdu(rdp, s); } BOOL rdp_recv_server_synchronize_pdu(rdpRdp* rdp, STREAM* s) { rdp->finalize_sc_pdus |= FINALIZE_SC_SYNCHRONIZE_PDU; return TRUE; } BOOL rdp_send_server_synchronize_pdu(rdpRdp* rdp) { STREAM* s; s = rdp_data_pdu_init(rdp); rdp_write_synchronize_pdu(s, rdp->settings); return rdp_send_data_pdu(rdp, s, DATA_PDU_TYPE_SYNCHRONIZE, rdp->mcs->user_id); } BOOL rdp_recv_client_synchronize_pdu(rdpRdp* rdp, STREAM* s) { UINT16 messageType; rdp->finalize_sc_pdus |= FINALIZE_SC_SYNCHRONIZE_PDU; if (stream_get_left(s) < 4) return FALSE; stream_read_UINT16(s, messageType); /* messageType (2 bytes) */ if (messageType != SYNCMSGTYPE_SYNC) return FALSE; /* targetUser (2 bytes) */ stream_seek_UINT16(s); return TRUE; } BOOL rdp_send_client_synchronize_pdu(rdpRdp* rdp) { STREAM* s; s = rdp_data_pdu_init(rdp); rdp_write_synchronize_pdu(s, rdp->settings); return rdp_send_data_pdu(rdp, s, DATA_PDU_TYPE_SYNCHRONIZE, rdp->mcs->user_id); } BOOL rdp_recv_control_pdu(STREAM* s, UINT16* action) { if (stream_get_left(s) < 8) return FALSE; stream_read_UINT16(s, *action); /* action (2 bytes) */ stream_seek_UINT16(s); /* grantId (2 bytes) */ stream_seek_UINT32(s); /* controlId (4 bytes) */ return TRUE; } void rdp_write_client_control_pdu(STREAM* s, UINT16 action) { stream_write_UINT16(s, action); /* action (2 bytes) */ stream_write_UINT16(s, 0); /* grantId (2 bytes) */ stream_write_UINT32(s, 0); /* controlId (4 bytes) */ } BOOL rdp_recv_server_control_pdu(rdpRdp* rdp, STREAM* s) { UINT16 action; if(rdp_recv_control_pdu(s, &action) == FALSE) return FALSE; switch (action) { case CTRLACTION_COOPERATE: rdp->finalize_sc_pdus |= FINALIZE_SC_CONTROL_COOPERATE_PDU; break; case CTRLACTION_GRANTED_CONTROL: rdp->finalize_sc_pdus |= FINALIZE_SC_CONTROL_GRANTED_PDU; break; } return TRUE; } BOOL rdp_send_server_control_cooperate_pdu(rdpRdp* rdp) { STREAM* s; s = rdp_data_pdu_init(rdp); stream_write_UINT16(s, CTRLACTION_COOPERATE); /* action (2 bytes) */ stream_write_UINT16(s, 0); /* grantId (2 bytes) */ stream_write_UINT32(s, 0); /* controlId (4 bytes) */ return rdp_send_data_pdu(rdp, s, DATA_PDU_TYPE_CONTROL, rdp->mcs->user_id); } BOOL rdp_send_server_control_granted_pdu(rdpRdp* rdp) { STREAM* s; s = rdp_data_pdu_init(rdp); stream_write_UINT16(s, CTRLACTION_GRANTED_CONTROL); /* action (2 bytes) */ stream_write_UINT16(s, rdp->mcs->user_id); /* grantId (2 bytes) */ stream_write_UINT32(s, 0x03EA); /* controlId (4 bytes) */ return rdp_send_data_pdu(rdp, s, DATA_PDU_TYPE_CONTROL, rdp->mcs->user_id); } BOOL rdp_send_client_control_pdu(rdpRdp* rdp, UINT16 action) { STREAM* s; s = rdp_data_pdu_init(rdp); rdp_write_client_control_pdu(s, action); return rdp_send_data_pdu(rdp, s, DATA_PDU_TYPE_CONTROL, rdp->mcs->user_id); } void rdp_write_persistent_list_entry(STREAM* s, UINT32 key1, UINT32 key2) { stream_write_UINT32(s, key1); /* key1 (4 bytes) */ stream_write_UINT32(s, key2); /* key2 (4 bytes) */ } void rdp_write_client_persistent_key_list_pdu(STREAM* s, rdpSettings* settings) { stream_write_UINT16(s, 0); /* numEntriesCache0 (2 bytes) */ stream_write_UINT16(s, 0); /* numEntriesCache1 (2 bytes) */ stream_write_UINT16(s, 0); /* numEntriesCache2 (2 bytes) */ stream_write_UINT16(s, 0); /* numEntriesCache3 (2 bytes) */ stream_write_UINT16(s, 0); /* numEntriesCache4 (2 bytes) */ stream_write_UINT16(s, 0); /* totalEntriesCache0 (2 bytes) */ stream_write_UINT16(s, 0); /* totalEntriesCache1 (2 bytes) */ stream_write_UINT16(s, 0); /* totalEntriesCache2 (2 bytes) */ stream_write_UINT16(s, 0); /* totalEntriesCache3 (2 bytes) */ stream_write_UINT16(s, 0); /* totalEntriesCache4 (2 bytes) */ stream_write_BYTE(s, PERSIST_FIRST_PDU | PERSIST_LAST_PDU); /* bBitMask (1 byte) */ stream_write_BYTE(s, 0); /* pad1 (1 byte) */ stream_write_UINT16(s, 0); /* pad3 (2 bytes) */ /* entries */ } BOOL rdp_send_client_persistent_key_list_pdu(rdpRdp* rdp) { STREAM* s; s = rdp_data_pdu_init(rdp); rdp_write_client_persistent_key_list_pdu(s, rdp->settings); return rdp_send_data_pdu(rdp, s, DATA_PDU_TYPE_BITMAP_CACHE_PERSISTENT_LIST, rdp->mcs->user_id); } BOOL rdp_recv_client_font_list_pdu(STREAM* s) { if (stream_get_left(s) < 8) return FALSE; return TRUE; } void rdp_write_client_font_list_pdu(STREAM* s, UINT16 flags) { stream_write_UINT16(s, 0); /* numberFonts (2 bytes) */ stream_write_UINT16(s, 0); /* totalNumFonts (2 bytes) */ stream_write_UINT16(s, flags); /* listFlags (2 bytes) */ stream_write_UINT16(s, 50); /* entrySize (2 bytes) */ } BOOL rdp_send_client_font_list_pdu(rdpRdp* rdp, UINT16 flags) { STREAM* s; s = rdp_data_pdu_init(rdp); rdp_write_client_font_list_pdu(s, flags); return rdp_send_data_pdu(rdp, s, DATA_PDU_TYPE_FONT_LIST, rdp->mcs->user_id); } BOOL rdp_recv_font_map_pdu(rdpRdp* rdp, STREAM* s) { if (rdp->settings->ServerMode) return rdp_recv_server_font_map_pdu(rdp, s); else return rdp_recv_client_font_map_pdu(rdp, s); } BOOL rdp_recv_server_font_map_pdu(rdpRdp* rdp, STREAM* s) { rdp->finalize_sc_pdus |= FINALIZE_SC_FONT_MAP_PDU; return TRUE; } BOOL rdp_recv_client_font_map_pdu(rdpRdp* rdp, STREAM* s) { if(stream_get_left(s) < 8) return FALSE; rdp->finalize_sc_pdus |= FINALIZE_SC_FONT_MAP_PDU; stream_seek_UINT16(s); /* numberEntries (2 bytes) */ stream_seek_UINT16(s); /* totalNumEntries (2 bytes) */ stream_seek_UINT16(s); /* mapFlags (2 bytes) */ stream_seek_UINT16(s); /* entrySize (2 bytes) */ return TRUE; } BOOL rdp_send_server_font_map_pdu(rdpRdp* rdp) { STREAM* s; s = rdp_data_pdu_init(rdp); stream_write_UINT16(s, 0); /* numberEntries (2 bytes) */ stream_write_UINT16(s, 0); /* totalNumEntries (2 bytes) */ stream_write_UINT16(s, FONTLIST_FIRST | FONTLIST_LAST); /* mapFlags (2 bytes) */ stream_write_UINT16(s, 4); /* entrySize (2 bytes) */ return rdp_send_data_pdu(rdp, s, DATA_PDU_TYPE_FONT_MAP, rdp->mcs->user_id); } BOOL rdp_recv_deactivate_all(rdpRdp* rdp, STREAM* s) { UINT16 lengthSourceDescriptor; /* * Windows XP can send short DEACTIVATE_ALL PDU that doesn't contain * the following fields. */ if (stream_get_left(s) > 0) { do { if(stream_get_left(s) < 4) break; stream_read_UINT32(s, rdp->settings->ShareId); /* shareId (4 bytes) */ if(stream_get_left(s) < 2) break; stream_read_UINT16(s, lengthSourceDescriptor); /* lengthSourceDescriptor (2 bytes) */ if(stream_get_left(s) < lengthSourceDescriptor) break; stream_seek(s, lengthSourceDescriptor); /* sourceDescriptor (should be 0x00) */ } while(0); } rdp->state = CONNECTION_STATE_CAPABILITY; while (rdp->state != CONNECTION_STATE_ACTIVE) { if (rdp_check_fds(rdp) < 0) return FALSE; if (rdp->disconnect) break; } return TRUE; } BOOL rdp_send_deactivate_all(rdpRdp* rdp) { STREAM* s; s = rdp_pdu_init(rdp); stream_write_UINT32(s, rdp->settings->ShareId); /* shareId (4 bytes) */ stream_write_UINT16(s, 1); /* lengthSourceDescriptor (2 bytes) */ stream_write_BYTE(s, 0); /* sourceDescriptor (should be 0x00) */ return rdp_send_pdu(rdp, s, PDU_TYPE_DEACTIVATE_ALL, rdp->mcs->user_id); } BOOL rdp_server_accept_client_control_pdu(rdpRdp* rdp, STREAM* s) { UINT16 action; if (!rdp_recv_control_pdu(s, &action)) return FALSE; if (action == CTRLACTION_REQUEST_CONTROL) { if (!rdp_send_server_control_granted_pdu(rdp)) return FALSE; } return TRUE; } BOOL rdp_server_accept_client_font_list_pdu(rdpRdp* rdp, STREAM* s) { if (!rdp_recv_client_font_list_pdu(s)) return FALSE; if (!rdp_send_server_font_map_pdu(rdp)) return FALSE; return TRUE; }