320 lines
11 KiB
C
320 lines
11 KiB
C
/*
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you 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.
|
|
*/
|
|
|
|
#include "channels/rdpdr/rdpdr-fs-messages-file-info.h"
|
|
#include "channels/rdpdr/rdpdr.h"
|
|
#include "download.h"
|
|
#include "fs.h"
|
|
#include "unicode.h"
|
|
|
|
#include <guacamole/client.h>
|
|
#include <winpr/file.h>
|
|
#include <winpr/nt.h>
|
|
#include <winpr/stream.h>
|
|
#include <winpr/wtypes.h>
|
|
|
|
#include <stdint.h>
|
|
#include <string.h>
|
|
|
|
void guac_rdpdr_fs_process_query_basic_info(guac_rdp_common_svc* svc,
|
|
guac_rdpdr_device* device, guac_rdpdr_iorequest* iorequest,
|
|
wStream* input_stream) {
|
|
|
|
wStream* output_stream;
|
|
guac_rdp_fs_file* file;
|
|
|
|
/* Get file */
|
|
file = guac_rdp_fs_get_file((guac_rdp_fs*) device->data, iorequest->file_id);
|
|
if (file == NULL)
|
|
return;
|
|
|
|
guac_client_log(svc->client, GUAC_LOG_DEBUG, "%s: [file_id=%i]", __func__,
|
|
iorequest->file_id);
|
|
|
|
output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, STATUS_SUCCESS, 40);
|
|
|
|
Stream_Write_UINT32(output_stream, 36);
|
|
Stream_Write_UINT64(output_stream, file->ctime); /* CreationTime */
|
|
Stream_Write_UINT64(output_stream, file->atime); /* LastAccessTime */
|
|
Stream_Write_UINT64(output_stream, file->mtime); /* LastWriteTime */
|
|
Stream_Write_UINT64(output_stream, file->mtime); /* ChangeTime */
|
|
Stream_Write_UINT32(output_stream, file->attributes); /* FileAttributes */
|
|
|
|
/* Reserved field must not be sent */
|
|
|
|
guac_rdp_common_svc_write(svc, output_stream);
|
|
|
|
}
|
|
|
|
void guac_rdpdr_fs_process_query_standard_info(guac_rdp_common_svc* svc,
|
|
guac_rdpdr_device* device, guac_rdpdr_iorequest* iorequest,
|
|
wStream* input_stream) {
|
|
|
|
wStream* output_stream;
|
|
guac_rdp_fs_file* file;
|
|
BOOL is_directory = FALSE;
|
|
|
|
/* Get file */
|
|
file = guac_rdp_fs_get_file((guac_rdp_fs*) device->data, iorequest->file_id);
|
|
if (file == NULL)
|
|
return;
|
|
|
|
guac_client_log(svc->client, GUAC_LOG_DEBUG, "%s: [file_id=%i]", __func__,
|
|
iorequest->file_id);
|
|
|
|
if (file->attributes & FILE_ATTRIBUTE_DIRECTORY)
|
|
is_directory = TRUE;
|
|
|
|
output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, STATUS_SUCCESS, 26);
|
|
|
|
Stream_Write_UINT32(output_stream, 22);
|
|
Stream_Write_UINT64(output_stream, file->size); /* AllocationSize */
|
|
Stream_Write_UINT64(output_stream, file->size); /* EndOfFile */
|
|
Stream_Write_UINT32(output_stream, 1); /* NumberOfLinks */
|
|
Stream_Write_UINT8(output_stream, 0); /* DeletePending */
|
|
Stream_Write_UINT8(output_stream, is_directory); /* Directory */
|
|
|
|
/* Reserved field must not be sent */
|
|
|
|
guac_rdp_common_svc_write(svc, output_stream);
|
|
|
|
}
|
|
|
|
void guac_rdpdr_fs_process_query_attribute_tag_info(guac_rdp_common_svc* svc,
|
|
guac_rdpdr_device* device, guac_rdpdr_iorequest* iorequest,
|
|
wStream* input_stream) {
|
|
|
|
wStream* output_stream;
|
|
guac_rdp_fs_file* file;
|
|
|
|
/* Get file */
|
|
file = guac_rdp_fs_get_file((guac_rdp_fs*) device->data, iorequest->file_id);
|
|
if (file == NULL)
|
|
return;
|
|
|
|
guac_client_log(svc->client, GUAC_LOG_DEBUG, "%s: [file_id=%i]", __func__,
|
|
iorequest->file_id);
|
|
|
|
output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, STATUS_SUCCESS, 12);
|
|
|
|
Stream_Write_UINT32(output_stream, 8);
|
|
Stream_Write_UINT32(output_stream, file->attributes); /* FileAttributes */
|
|
Stream_Write_UINT32(output_stream, 0); /* ReparseTag */
|
|
|
|
/* Reserved field must not be sent */
|
|
|
|
guac_rdp_common_svc_write(svc, output_stream);
|
|
|
|
}
|
|
|
|
void guac_rdpdr_fs_process_set_rename_info(guac_rdp_common_svc* svc,
|
|
guac_rdpdr_device* device, guac_rdpdr_iorequest* iorequest,
|
|
int length, wStream* input_stream) {
|
|
|
|
int result;
|
|
int filename_length;
|
|
wStream* output_stream;
|
|
char destination_path[GUAC_RDP_FS_MAX_PATH];
|
|
|
|
/* Check stream size prior to reading. */
|
|
if (Stream_GetRemainingLength(input_stream) < 6) {
|
|
guac_client_log(svc->client, GUAC_LOG_WARNING, "Server Drive Set "
|
|
"Information Request (FileRenameInformation) PDU does not "
|
|
"contain the expected number of bytes. File redirection "
|
|
"may not work as expected.");
|
|
return;
|
|
}
|
|
|
|
/* Read structure */
|
|
Stream_Seek_UINT8(input_stream); /* ReplaceIfExists */
|
|
Stream_Seek_UINT8(input_stream); /* RootDirectory */
|
|
Stream_Read_UINT32(input_stream, filename_length); /* FileNameLength */
|
|
|
|
if (Stream_GetRemainingLength(input_stream) < filename_length) {
|
|
guac_client_log(svc->client, GUAC_LOG_WARNING, "Server Drive Set "
|
|
"Information Request (FileRenameInformation) PDU does not "
|
|
"contain the expected number of bytes. File redirection "
|
|
"may not work as expected.");
|
|
return;
|
|
}
|
|
|
|
/* Convert name to UTF-8 */
|
|
guac_rdp_utf16_to_utf8(Stream_Pointer(input_stream), filename_length/2,
|
|
destination_path, sizeof(destination_path));
|
|
|
|
guac_client_log(svc->client, GUAC_LOG_DEBUG, "%s: [file_id=%i]"
|
|
"destination_path=\"%s\"", __func__, iorequest->file_id,
|
|
destination_path);
|
|
|
|
/* If file moving to \Download folder, start stream, do not move */
|
|
if (strncmp(destination_path, "\\Download\\", 10) == 0
|
|
&& !((guac_rdp_fs*) device->data)->disable_download) {
|
|
|
|
guac_rdp_fs_file* file;
|
|
|
|
/* Get file */
|
|
file = guac_rdp_fs_get_file((guac_rdp_fs*) device->data, iorequest->file_id);
|
|
if (file == NULL)
|
|
return;
|
|
|
|
/* Initiate download, pretend move succeeded */
|
|
guac_client_for_owner(svc->client, guac_rdp_download_to_user, file->absolute_path);
|
|
output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, STATUS_SUCCESS, 4);
|
|
|
|
}
|
|
|
|
/* Otherwise, rename as requested */
|
|
else {
|
|
|
|
result = guac_rdp_fs_rename((guac_rdp_fs*) device->data,
|
|
iorequest->file_id, destination_path);
|
|
if (result < 0)
|
|
output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, guac_rdp_fs_get_status(result), 4);
|
|
else
|
|
output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, STATUS_SUCCESS, 4);
|
|
|
|
}
|
|
|
|
Stream_Write_UINT32(output_stream, length);
|
|
guac_rdp_common_svc_write(svc, output_stream);
|
|
|
|
}
|
|
|
|
void guac_rdpdr_fs_process_set_allocation_info(guac_rdp_common_svc* svc,
|
|
guac_rdpdr_device* device, guac_rdpdr_iorequest* iorequest,
|
|
int length, wStream* input_stream) {
|
|
|
|
int result;
|
|
UINT64 size;
|
|
wStream* output_stream;
|
|
|
|
/* Check to make sure the stream has at least 8 bytes (UINT64) */
|
|
if (Stream_GetRemainingLength(input_stream) < 8) {
|
|
guac_client_log(svc->client, GUAC_LOG_WARNING, "Server Drive Set "
|
|
"Information Request (FileAllocationInformation) PDU does not "
|
|
"contain the expected number of bytes. File redirection "
|
|
"may not work as expected.");
|
|
return;
|
|
}
|
|
|
|
/* Read new size */
|
|
Stream_Read_UINT64(input_stream, size); /* AllocationSize */
|
|
|
|
guac_client_log(svc->client, GUAC_LOG_DEBUG, "%s: [file_id=%i] "
|
|
"size=%" PRIu64, __func__, iorequest->file_id, (uint64_t) size);
|
|
|
|
/* Truncate file */
|
|
result = guac_rdp_fs_truncate((guac_rdp_fs*) device->data, iorequest->file_id, size);
|
|
if (result < 0)
|
|
output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, guac_rdp_fs_get_status(result), 4);
|
|
else
|
|
output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, STATUS_SUCCESS, 4);
|
|
|
|
Stream_Write_UINT32(output_stream, length);
|
|
guac_rdp_common_svc_write(svc, output_stream);
|
|
|
|
}
|
|
|
|
void guac_rdpdr_fs_process_set_disposition_info(guac_rdp_common_svc* svc,
|
|
guac_rdpdr_device* device, guac_rdpdr_iorequest* iorequest,
|
|
int length, wStream* input_stream) {
|
|
|
|
wStream* output_stream;
|
|
|
|
/* Delete file */
|
|
int result = guac_rdp_fs_delete((guac_rdp_fs*) device->data, iorequest->file_id);
|
|
if (result < 0)
|
|
output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, guac_rdp_fs_get_status(result), 4);
|
|
else
|
|
output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, STATUS_SUCCESS, 4);
|
|
|
|
guac_client_log(svc->client, GUAC_LOG_DEBUG, "%s: [file_id=%i]", __func__,
|
|
iorequest->file_id);
|
|
|
|
Stream_Write_UINT32(output_stream, length);
|
|
|
|
guac_rdp_common_svc_write(svc, output_stream);
|
|
|
|
}
|
|
|
|
void guac_rdpdr_fs_process_set_end_of_file_info(guac_rdp_common_svc* svc,
|
|
guac_rdpdr_device* device, guac_rdpdr_iorequest* iorequest,
|
|
int length, wStream* input_stream) {
|
|
|
|
int result;
|
|
UINT64 size;
|
|
wStream* output_stream;
|
|
|
|
/* Check to make sure stream contains at least 8 bytes (UINT64) */
|
|
if (Stream_GetRemainingLength(input_stream) < 8) {
|
|
guac_client_log(svc->client, GUAC_LOG_WARNING, "Server Drive Set "
|
|
"Information Request (FileEndOfFileInformation) PDU does not "
|
|
"contain the expected number of bytes. File redirection "
|
|
"may not work as expected.");
|
|
return;
|
|
}
|
|
|
|
/* Read new size */
|
|
Stream_Read_UINT64(input_stream, size); /* AllocationSize */
|
|
|
|
guac_client_log(svc->client, GUAC_LOG_DEBUG, "%s: [file_id=%i] "
|
|
"size=%" PRIu64, __func__, iorequest->file_id, (uint64_t) size);
|
|
|
|
/* Truncate file */
|
|
result = guac_rdp_fs_truncate((guac_rdp_fs*) device->data, iorequest->file_id, size);
|
|
if (result < 0)
|
|
output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, guac_rdp_fs_get_status(result), 4);
|
|
else
|
|
output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, STATUS_SUCCESS, 4);
|
|
|
|
Stream_Write_UINT32(output_stream, length);
|
|
guac_rdp_common_svc_write(svc, output_stream);
|
|
|
|
}
|
|
|
|
void guac_rdpdr_fs_process_set_basic_info(guac_rdp_common_svc* svc,
|
|
guac_rdpdr_device* device, guac_rdpdr_iorequest* iorequest,
|
|
int length, wStream* input_stream) {
|
|
|
|
wStream* output_stream = guac_rdpdr_new_io_completion(device,
|
|
iorequest->completion_id, STATUS_SUCCESS, 4);
|
|
|
|
/* Currently do nothing, just respond */
|
|
Stream_Write_UINT32(output_stream, length);
|
|
|
|
guac_client_log(svc->client, GUAC_LOG_DEBUG, "%s: [file_id=%i] IGNORED",
|
|
__func__, iorequest->file_id);
|
|
|
|
guac_rdp_common_svc_write(svc, output_stream);
|
|
|
|
}
|
|
|