added webrdp public code

1 files changed, 129 insertions, 0 deletions

diff --git a/src/rdp/rdp_ft.c b/src/rdp/rdp_ft.c
new file mode 100644
index 0000000..0c5575e
--- /dev/null
+++ b/src/rdp/rdp_ft.c
@@ -0,0 +1,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;
+}