Free clipboard data properly.
[libguac-client-rdp.git] / src / rdp_cliprdr.c
1
2 /* ***** BEGIN LICENSE BLOCK *****
3  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
4  *
5  * The contents of this file are subject to the Mozilla Public License Version
6  * 1.1 (the "License"); you may not use this file except in compliance with
7  * the License. You may obtain a copy of the License at
8  * http://www.mozilla.org/MPL/
9  *
10  * Software distributed under the License is distributed on an "AS IS" basis,
11  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
12  * for the specific language governing rights and limitations under the
13  * License.
14  *
15  * The Original Code is libguac-client-rdp.
16  *
17  * The Initial Developer of the Original Code is
18  * Michael Jumper.
19  * Portions created by the Initial Developer are Copyright (C) 2011
20  * the Initial Developer. All Rights Reserved.
21  *
22  * Contributor(s):
23  *
24  * Alternatively, the contents of this file may be used under the terms of
25  * either the GNU General Public License Version 2 or later (the "GPL"), or
26  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
27  * in which case the provisions of the GPL or the LGPL are applicable instead
28  * of those above. If you wish to allow use of your version of this file only
29  * under the terms of either the GPL or the LGPL, and not to allow others to
30  * use your version of this file under the terms of the MPL, indicate your
31  * decision by deleting the provisions above and replace them with the notice
32  * and other provisions required by the GPL or the LGPL. If you do not delete
33  * the provisions above, a recipient may use your version of this file under
34  * the terms of any one of the MPL, the GPL or the LGPL.
35  *
36  * ***** END LICENSE BLOCK ***** */
37
38
39 #include <freerdp/freerdp.h>
40 #include <freerdp/channels/channels.h>
41 #include <freerdp/utils/event.h>
42 #include <freerdp/plugins/cliprdr.h>
43
44 #include <guacamole/client.h>
45
46 #include "client.h"
47 #include "rdp_cliprdr.h"
48
49 void guac_rdp_process_cliprdr_event(guac_client* client, RDP_EVENT* event) {
50
51         switch (event->event_type) {
52
53             case RDP_EVENT_TYPE_CB_MONITOR_READY:
54                 guac_rdp_process_cb_monitor_ready(client, event);
55                 break;
56
57             case RDP_EVENT_TYPE_CB_FORMAT_LIST:
58                 guac_rdp_process_cb_format_list(client,
59                         (RDP_CB_FORMAT_LIST_EVENT*) event);
60                 break;
61
62             case RDP_EVENT_TYPE_CB_DATA_REQUEST:
63                 guac_rdp_process_cb_data_request(client,
64                         (RDP_CB_DATA_REQUEST_EVENT*) event);
65                 break;
66
67             case RDP_EVENT_TYPE_CB_DATA_RESPONSE:
68                 guac_rdp_process_cb_data_response(client,
69                         (RDP_CB_DATA_RESPONSE_EVENT*) event);
70                 break;
71
72             default:
73                 guac_client_log_info(client,
74                         "Unknown cliprdr event type: 0x%x",
75                         event->event_type);
76         }
77
78 }
79
80 void guac_rdp_process_cb_monitor_ready(guac_client* client, RDP_EVENT* event) {
81
82     rdpChannels* channels = 
83         ((rdp_guac_client_data*) client->data)->rdp_inst->context->channels;
84
85     RDP_CB_FORMAT_LIST_EVENT* format_list =
86         (RDP_CB_FORMAT_LIST_EVENT*) freerdp_event_new(
87             RDP_EVENT_CLASS_CLIPRDR,
88             RDP_EVENT_TYPE_CB_FORMAT_LIST,
89             NULL, NULL);
90
91     /* Received notification of clipboard support. */
92
93     /* Respond with supported format list */
94     format_list->formats = (uint32*) malloc(sizeof(uint32));
95     format_list->formats[0] = CB_FORMAT_TEXT;
96     format_list->num_formats = 1;
97
98     freerdp_channels_send_event(channels, (RDP_EVENT*) format_list);
99
100 }
101
102 void guac_rdp_process_cb_format_list(guac_client* client,
103         RDP_CB_FORMAT_LIST_EVENT* event) {
104
105     rdpChannels* channels = 
106         ((rdp_guac_client_data*) client->data)->rdp_inst->context->channels;
107
108     /* Received notification of available data */
109
110     int i;
111     for (i=0; i<event->num_formats; i++) {
112
113         /* If plain text available, request it */
114         if (event->formats[i] == CB_FORMAT_TEXT) {
115
116             /* Create new data request */
117             RDP_CB_DATA_REQUEST_EVENT* data_request =
118                 (RDP_CB_DATA_REQUEST_EVENT*) freerdp_event_new(
119                         RDP_EVENT_CLASS_CLIPRDR,
120                         RDP_EVENT_TYPE_CB_DATA_REQUEST,
121                         NULL, NULL);
122
123             /* We want plain text */
124             data_request->format = CB_FORMAT_TEXT;
125
126             /* Send request */
127             freerdp_channels_send_event(channels, (RDP_EVENT*) data_request);
128             return;
129
130         }
131
132     }
133
134     /* Otherwise, no supported data available */
135     guac_client_log_info(client, "Ignoring unsupported clipboard data");
136
137 }
138
139 void guac_rdp_process_cb_data_request(guac_client* client,
140         RDP_CB_DATA_REQUEST_EVENT* event) {
141
142     rdpChannels* channels = 
143         ((rdp_guac_client_data*) client->data)->rdp_inst->context->channels;
144
145     /* If text requested, send clipboard text contents */
146     if (event->format == CB_FORMAT_TEXT) {
147
148         /* Get clipboard data */
149         const char* clipboard =
150             ((rdp_guac_client_data*) client->data)->clipboard;
151
152         /* Create new data response */
153         RDP_CB_DATA_RESPONSE_EVENT* data_response =
154             (RDP_CB_DATA_RESPONSE_EVENT*) freerdp_event_new(
155                     RDP_EVENT_CLASS_CLIPRDR,
156                     RDP_EVENT_TYPE_CB_DATA_RESPONSE,
157                     NULL, NULL);
158
159         /* Set data and length */
160         if (clipboard != NULL) {
161             data_response->data = (uint8*) strdup(clipboard);
162             data_response->size = strlen(clipboard) + 1;
163         }
164         else {
165             data_response->data = (uint8*) strdup("");
166             data_response->size = 1;
167         }
168
169         /* Send response */
170         freerdp_channels_send_event(channels, (RDP_EVENT*) data_response);
171
172     }
173
174     /* Otherwise ... failure */
175     else
176         guac_client_log_error(client, 
177                 "Server requested unsupported clipboard data type");
178
179 }
180
181 void guac_rdp_process_cb_data_response(guac_client* client,
182         RDP_CB_DATA_RESPONSE_EVENT* event) {
183
184     /* Received clipboard data */
185     if (event->data[event->size - 1] == '\0') {
186
187         /* Free existing data */
188         free(((rdp_guac_client_data*) client->data)->clipboard);
189
190         /* Store clipboard data */
191         ((rdp_guac_client_data*) client->data)->clipboard =
192             strdup((char*) event->data);
193
194         /* Send clipboard data */
195         guac_protocol_send_clipboard(client->socket, (char*) event->data);
196
197     }
198     else
199         guac_client_log_error(client,
200                 "Clipboard data missing null terminator");
201
202 }
203