maride/guacamole-spice-protocol
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Implement file truncation.

Browse Source
This commit is contained in:
Michael Jumper 2013-11-05 00:48:55 -08:00
parent 3ebed6a85c
commit 296836f65d
3 changed files with 57 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/protocols/rdp/guac_rdpdr/rdpdr_fs_messages_file_info.c
View File

@ -177,15 +177,24 @@ void guac_rdpdr_fs_process_set_rename_info(guac_rdpdr_device* device,
void guac_rdpdr_fs_process_set_allocation_info(guac_rdpdr_device* device,
wStream* input_stream, int file_id, int completion_id, int length) {
wStream* output_stream = guac_rdpdr_new_io_completion(device,
completion_id, STATUS_SUCCESS, 4);
int result;
UINT64 size;
wStream* output_stream;
/* Read new size */
Stream_Read_UINT64(input_stream, size); /* AllocationSize */
/* Truncate file */
result = guac_rdp_fs_truncate((guac_rdp_fs*) device->data, file_id, size);
if (result < 0)
output_stream = guac_rdpdr_new_io_completion(device,
completion_id, guac_rdp_fs_get_status(result), 4);
else
output_stream = guac_rdpdr_new_io_completion(device,
completion_id, STATUS_SUCCESS, 4);
/* Currently do nothing, just respond */
Stream_Write_UINT32(output_stream, length);
svc_plugin_send((rdpSvcPlugin*) device->rdpdr, output_stream);
GUAC_RDP_DEBUG(2, "Sent STATUS_SUCCESS for completion_id=%i",
completion_id);
}
@ -214,15 +223,24 @@ void guac_rdpdr_fs_process_set_disposition_info(guac_rdpdr_device* device,
void guac_rdpdr_fs_process_set_end_of_file_info(guac_rdpdr_device* device,
wStream* input_stream, int file_id, int completion_id, int length) {
wStream* output_stream = guac_rdpdr_new_io_completion(device,
completion_id, STATUS_SUCCESS, 4);
int result;
UINT64 size;
wStream* output_stream;
/* Read new size */
Stream_Read_UINT64(input_stream, size); /* AllocationSize */
/* Truncate file */
result = guac_rdp_fs_truncate((guac_rdp_fs*) device->data, file_id, size);
if (result < 0)
output_stream = guac_rdpdr_new_io_completion(device,
completion_id, guac_rdp_fs_get_status(result), 4);
else
output_stream = guac_rdpdr_new_io_completion(device,
completion_id, STATUS_SUCCESS, 4);
/* Currently do nothing, just respond */
Stream_Write_UINT32(output_stream, length);
svc_plugin_send((rdpSvcPlugin*) device->rdpdr, output_stream);
GUAC_RDP_DEBUG(2, "Sent STATUS_SUCCESS for completion_id=%i",
completion_id);
}

20
src/protocols/rdp/rdp_fs.c
View File

@ -429,6 +429,26 @@ int guac_rdp_fs_delete(guac_rdp_fs* fs, int file_id) {
}
int guac_rdp_fs_truncate(guac_rdp_fs* fs, int file_id, int length) {
/* Get file */
guac_rdp_fs_file* file = guac_rdp_fs_get_file(fs, file_id);
if (file == NULL) {
GUAC_RDP_DEBUG(1, "Delete of bad file_id: %i", file_id);
return GUAC_RDP_FS_EINVAL;
}
/* Attempt truncate */
if (ftruncate(file->fd, length)) {
GUAC_RDP_DEBUG(1, "ftruncate() to %i bytes failed: \"%s\"",
length, file->real_path);
return guac_rdp_fs_get_errorcode(errno);
}
return 0;
}
void guac_rdp_fs_close(guac_rdp_fs* fs, int file_id) {
guac_rdp_fs_file* file = guac_rdp_fs_get_file(fs, file_id);

8
src/protocols/rdp/rdp_fs.h
View File

@ -342,10 +342,16 @@ int guac_rdp_fs_rename(guac_rdp_fs* fs, int file_id,
const char* new_path);
/**
* Deletes the gile with the given ID.
* Deletes the file with the given ID.
*/
int guac_rdp_fs_delete(guac_rdp_fs* fs, int file_id);
/**
* Truncates the file with the given ID to the given length (in bytes), which
* may be larger.
*/
int guac_rdp_fs_truncate(guac_rdp_fs* fs, int file_id, int length);
/**
* Frees the given file ID, allowing future open operations to reuse it.
*/