1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/* BSD-2-Clause license
*
* Copyright (c) 2018-2023 NST <www.newinfosec.ru>, sss <sss at dark-alexandr dot net>.
*
*/
#include <stddef.h>
#include <stdbool.h>
#include "webrdp_module_api.h"
#include "webrdp_core_api.h"
#include "rdp_backend_api.h"
#include "rdp_impl.h"
#include "rdp_ft.h"
#include "rdp_clipboard.h"
static bool
ft_api_handle_request(
const wrdp_backend_ft_file_request *request, void *backend_internals)
{
rdp_internals *i = backend_internals;
CLIPRDR_FILE_CONTENTS_REQUEST file_req;
uint32_t chunk_len = 512000;
size_t left = 0;
if (i->context->ft_to_client->is_runing)
{
const char *msg
= "rdp_module: clipboard: filetransfers: transfer "
"already runing, cannot start new one";
i->core->api_utils->log_msg(
(const uint8_t *)msg, strlen(msg), wrdp_log_level_error, 0);
return true;
}
memset(&file_req, 0, sizeof(CLIPRDR_FILE_CONTENTS_REQUEST));
i->context->ft_to_client->transfer_id = rand();
i->context->ft_to_client->file_size = request->req_size;
i->context->ft_to_client->file_id = request->file_id;
i->context->ft_to_client->transferred = request->file_offset;
i->context->ft_to_client->is_runing = true;
left = i->context->ft_to_client->file_size
- i->context->ft_to_client->transferred;
if (left < chunk_len)
{
chunk_len = left;
}
file_req.cbRequested = chunk_len;
file_req.listIndex = request->file_id;
file_req.streamId = i->context->ft_to_client->transfer_id;
file_req.dwFlags = FILECONTENTS_RANGE;
/* file_req.haveClipDataId = TRUE;
file_req.clipDataId =
i->context->ft_to_client->clipboard_data_id; */
/* TODO: check/change endianness */
memcpy(&(file_req.nPositionLow),
&(i->context->ft_to_client->transferred), 4);
memcpy(&(file_req.nPositionHigh),
(((uint8_t *)&(i->context->ft_to_client->transferred)) + 4), 4);
i->context->clipboard->clip_context->ClientFileContentsRequest(
i->context->clipboard->clip_context, &file_req);
return true;
}
static bool
ft_api_handle_chunk(const wrdp_backend_ft_chunk *chunk, const uint8_t *data,
void *backend_internals)
{
rdp_internals *i = backend_internals;
CLIPRDR_FILE_CONTENTS_RESPONSE resp;
memset(&resp, 0, sizeof(CLIPRDR_FILE_CONTENTS_RESPONSE));
resp.cbRequested = chunk->size;
resp.streamId = i->context->ft_to_server->transfer_id;
resp.common.msgFlags = CB_RESPONSE_OK;
resp.common.msgType = CB_FILECONTENTS_RESPONSE;
resp.requestedData = data;
i->context->clipboard->clip_context->ClientFileContentsResponse(
i->context->clipboard->clip_context, &resp);
i->context->ft_to_server->transferred += chunk->size;
if (i->context->ft_to_server->transferred
== i->context->ft_to_server->file_size)
{
i->context->ft_to_server->is_runing = false;
i->context->ft_to_server->transfer_id = 0;
i->context->ft_to_server->file_id = 0;
i->context->ft_to_server->file_size = 0;
i->context->ft_to_server->transferred = 0;
}
return true;
}
static bool
ft_api_handle_finish(
const wrdp_backend_ft_status *status, void *backend_internals)
{
/* TODO: handle errors */
rdp_internals *i = backend_internals;
if (status->status != ft_status_success)
{
/* TODO: test it */
CLIPRDR_FILE_CONTENTS_RESPONSE resp;
memset(&resp, 0, sizeof(CLIPRDR_FILE_CONTENTS_RESPONSE));
const char *msg = "filetransfer finished with error";
i->core->api_utils->log_msg(
(const uint8_t *)msg, strlen(msg), wrdp_log_level_error, 0);
resp.common.msgType = CB_FILECONTENTS_RESPONSE;
resp.common.msgFlags = CB_RESPONSE_FAIL;
resp.streamId = i->context->ft_to_server->transfer_id;
resp.cbRequested = 0;
resp.requestedData = 0;
i->context->clipboard->clip_context->ClientFileContentsResponse(
i->context->clipboard->clip_context, &resp);
}
i->context->ft_to_server->is_runing = false;
i->context->ft_to_server->transfer_id = 0;
i->context->ft_to_server->file_size
= i->context->ft_to_server->transferred = 0;
return true;
}
void
register_ft(wrdp_backend_module *backend)
{
backend->callbacks_ft->chunk = ft_api_handle_chunk;
backend->callbacks_ft->finish = ft_api_handle_finish;
backend->callbacks_ft->request = ft_api_handle_request;
}
|
