2 * FreeRDP: A Remote Desktop Protocol client.
5 * Copyright 2011 Vic Lee
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
22 static boolean freerdp_peer_initialize(freerdp_peer* client)
24 client->context->rdp->settings->server_mode = true;
25 client->context->rdp->state = CONNECTION_STATE_INITIAL;
26 if (client->context->rdp->settings->rdp_key_file != NULL) {
27 client->context->rdp->settings->server_key =
28 key_new(client->context->rdp->settings->rdp_key_file);
34 static boolean freerdp_peer_get_fds(freerdp_peer* client, void** rfds, int* rcount)
36 rfds[*rcount] = (void*)(long)(client->context->rdp->transport->tcp->sockfd);
42 static boolean freerdp_peer_check_fds(freerdp_peer* client)
47 rdp = client->context->rdp;
49 status = rdp_check_fds(rdp);
56 static boolean peer_recv_data_pdu(freerdp_peer* client, STREAM* s)
61 uint8 compressed_type;
62 uint16 compressed_len;
65 if (!rdp_read_share_data_header(s, &length, &type, &share_id, &compressed_type, &compressed_len))
70 case DATA_PDU_TYPE_SYNCHRONIZE:
71 if (!rdp_recv_client_synchronize_pdu(client->context->rdp, s))
75 case DATA_PDU_TYPE_CONTROL:
76 if (!rdp_server_accept_client_control_pdu(client->context->rdp, s))
80 case DATA_PDU_TYPE_INPUT:
81 if (!input_recv(client->context->rdp->input, s))
85 case DATA_PDU_TYPE_BITMAP_CACHE_PERSISTENT_LIST:
86 /* TODO: notify server bitmap cache data */
89 case DATA_PDU_TYPE_FONT_LIST:
90 if (!rdp_server_accept_client_font_list_pdu(client->context->rdp, s))
92 if (client->PostConnect)
94 if (!client->PostConnect(client))
97 * PostConnect should only be called once and should not be called
98 * after a reactivation sequence.
100 client->PostConnect = NULL;
102 if (client->Activate)
104 /* Activate will be called everytime after the client is activated/reactivated. */
105 if (!client->Activate(client))
110 case DATA_PDU_TYPE_SHUTDOWN_REQUEST:
111 mcs_send_disconnect_provider_ultimatum(client->context->rdp->mcs);
115 printf("Data PDU type %d\n", type);
122 static boolean peer_recv_tpkt_pdu(freerdp_peer* client, STREAM* s)
130 uint16 securityFlags;
132 rdp = client->context->rdp;
134 if (!rdp_read_header(rdp, s, &length, &channelId))
136 printf("Incorrect RDP header.\n");
140 if (rdp->settings->encryption)
142 rdp_read_security_header(s, &securityFlags);
143 if (securityFlags & SEC_ENCRYPT)
145 if (!rdp_decrypt(rdp, s, length - 4, securityFlags))
147 printf("rdp_decrypt failed\n");
153 if (channelId != MCS_GLOBAL_CHANNEL_ID)
155 freerdp_channel_peer_process(client, s, channelId);
159 if (!rdp_read_share_control_header(s, &pduLength, &pduType, &pduSource))
162 client->settings->pdu_source = pduSource;
167 if (!peer_recv_data_pdu(client, s))
172 printf("Client sent pduType %d\n", pduType);
180 static boolean peer_recv_fastpath_pdu(freerdp_peer* client, STREAM* s)
184 rdpFastPath* fastpath;
186 rdp = client->context->rdp;
187 fastpath = rdp->fastpath;
188 length = fastpath_read_header_rdp(fastpath, s);
190 if (length == 0 || length > stream_get_left(s))
192 printf("incorrect FastPath PDU header length %d\n", length);
196 if (fastpath->encryptionFlags & FASTPATH_OUTPUT_ENCRYPTED)
198 rdp_decrypt(rdp, s, length, (fastpath->encryptionFlags & FASTPATH_OUTPUT_SECURE_CHECKSUM) ? SEC_SECURE_CHECKSUM : 0);
201 return fastpath_recv_inputs(fastpath, s);
204 static boolean peer_recv_pdu(freerdp_peer* client, STREAM* s)
206 if (tpkt_verify_header(s))
207 return peer_recv_tpkt_pdu(client, s);
209 return peer_recv_fastpath_pdu(client, s);
212 static boolean peer_recv_callback(rdpTransport* transport, STREAM* s, void* extra)
214 freerdp_peer* client = (freerdp_peer*) extra;
216 switch (client->context->rdp->state)
218 case CONNECTION_STATE_INITIAL:
219 if (!rdp_server_accept_nego(client->context->rdp, s))
223 case CONNECTION_STATE_NEGO:
224 if (!rdp_server_accept_mcs_connect_initial(client->context->rdp, s))
228 case CONNECTION_STATE_MCS_CONNECT:
229 if (!rdp_server_accept_mcs_erect_domain_request(client->context->rdp, s))
233 case CONNECTION_STATE_MCS_ERECT_DOMAIN:
234 if (!rdp_server_accept_mcs_attach_user_request(client->context->rdp, s))
238 case CONNECTION_STATE_MCS_ATTACH_USER:
239 if (!rdp_server_accept_mcs_channel_join_request(client->context->rdp, s))
243 case CONNECTION_STATE_MCS_CHANNEL_JOIN:
244 if (client->context->rdp->settings->encryption) {
245 if (!rdp_server_accept_client_keys(client->context->rdp, s))
249 client->context->rdp->state = CONNECTION_STATE_ESTABLISH_KEYS;
252 case CONNECTION_STATE_ESTABLISH_KEYS:
253 if (!rdp_server_accept_client_info(client->context->rdp, s))
255 IFCALL(client->Capabilities, client);
256 if (!rdp_send_demand_active(client->context->rdp))
260 case CONNECTION_STATE_LICENSE:
261 if (!rdp_server_accept_confirm_active(client->context->rdp, s))
264 * During reactivation sequence the client might sent some input or channel data
265 * before receiving the Deactivate All PDU. We need to process them as usual.
267 stream_set_pos(s, 0);
268 return peer_recv_pdu(client, s);
272 case CONNECTION_STATE_ACTIVE:
273 if (!peer_recv_pdu(client, s))
278 printf("Invalid state %d\n", client->context->rdp->state);
285 static void freerdp_peer_disconnect(freerdp_peer* client)
287 transport_disconnect(client->context->rdp->transport);
290 static int freerdp_peer_send_channel_data(freerdp_peer* client, int channelId, uint8* data, int size)
292 return rdp_send_channel_data(client->context->rdp, channelId, data, size);
295 void freerdp_peer_context_new(freerdp_peer* client)
300 client->input = rdp->input;
301 client->update = rdp->update;
302 client->settings = rdp->settings;
304 client->context = (rdpContext*) xzalloc(client->context_size);
305 client->context->rdp = rdp;
306 client->context->peer = client;
308 client->update->context = client->context;
309 client->input->context = client->context;
311 update_register_server_callbacks(client->update);
313 transport_attach(rdp->transport, client->sockfd);
315 rdp->transport->recv_callback = peer_recv_callback;
316 rdp->transport->recv_extra = client;
317 transport_set_blocking_mode(rdp->transport, false);
319 IFCALL(client->ContextNew, client, client->context);
322 void freerdp_peer_context_free(freerdp_peer* client)
324 IFCALL(client->ContextFree, client, client->context);
327 freerdp_peer* freerdp_peer_new(int sockfd)
329 freerdp_peer* client;
331 client = xnew(freerdp_peer);
335 client->sockfd = sockfd;
336 client->context_size = sizeof(rdpContext);
337 client->Initialize = freerdp_peer_initialize;
338 client->GetFileDescriptor = freerdp_peer_get_fds;
339 client->CheckFileDescriptor = freerdp_peer_check_fds;
340 client->Disconnect = freerdp_peer_disconnect;
341 client->SendChannelData = freerdp_peer_send_channel_data;
347 void freerdp_peer_free(freerdp_peer* client)
351 rdp_free(client->context->rdp);