Merge pull request #111 from glyptodon/fake-merge-screen-sharing-003-common-ssh

GUAC-1389: Bring common-ssh up to date with screen sharing changes.
This commit is contained in:
James Muehlner 2016-03-01 22:05:41 -08:00
commit 873a7a3211
3 changed files with 239 additions and 173 deletions

View File

@ -41,7 +41,7 @@ SUBDIRS = \
tests tests
if ENABLE_COMMON_SSH if ENABLE_COMMON_SSH
#SUBDIRS += src/common-ssh SUBDIRS += src/common-ssh
endif endif
if ENABLE_TERMINAL if ENABLE_TERMINAL

View File

@ -27,6 +27,7 @@
#include <guacamole/object.h> #include <guacamole/object.h>
#include <guacamole/protocol.h> #include <guacamole/protocol.h>
#include <guacamole/socket.h> #include <guacamole/socket.h>
#include <guacamole/user.h>
#include <libssh2.h> #include <libssh2.h>
#include <fcntl.h> #include <fcntl.h>
@ -39,22 +40,20 @@
* session into a Guacamole protocol status code. * session into a Guacamole protocol status code.
* *
* @param filesystem * @param filesystem
* The Guacamole protocol object defining the filesystem associated with * The object (not guac_object) defining the filesystem associated with the
* the SFTP and SSH sessions. * SFTP and SSH sessions.
* *
* @return * @return
* The Guacamole protocol status code corresponding to the last reported * The Guacamole protocol status code corresponding to the last reported
* error of the SFTP layer, if nay, or GUAC_PROTOCOL_STATUS_SUCCESS if no * error of the SFTP layer, if nay, or GUAC_PROTOCOL_STATUS_SUCCESS if no
* error has occurred. * error has occurred.
*/ */
static guac_protocol_status guac_sftp_get_status(guac_object* filesystem) { static guac_protocol_status guac_sftp_get_status(
guac_common_ssh_sftp_filesystem* filesystem) {
guac_common_ssh_sftp_data* sftp_data =
(guac_common_ssh_sftp_data*) filesystem->data;
/* Get libssh2 objects */ /* Get libssh2 objects */
LIBSSH2_SFTP* sftp = sftp_data->sftp_session; LIBSSH2_SFTP* sftp = filesystem->sftp_session;
LIBSSH2_SESSION* session = sftp_data->ssh_session->session; LIBSSH2_SESSION* session = filesystem->ssh_session->session;
/* Return success code if no error occurred */ /* Return success code if no error occurred */
if (libssh2_session_last_errno(session) != LIBSSH2_ERROR_SFTP_PROTOCOL) if (libssh2_session_last_errno(session) != LIBSSH2_ERROR_SFTP_PROTOCOL)
@ -193,8 +192,8 @@ static int guac_ssh_append_filename(char* fullpath, const char* path,
* pointer to an open LIBSSH2_SFTP_HANDLE for the file to which the data * pointer to an open LIBSSH2_SFTP_HANDLE for the file to which the data
* should be written. * should be written.
* *
* @param client * @param user
* The client receiving the blob message. * The user receiving the blob message.
* *
* @param stream * @param stream
* The Guacamole protocol stream associated with the received blob message. * The Guacamole protocol stream associated with the received blob message.
@ -208,7 +207,7 @@ static int guac_ssh_append_filename(char* fullpath, const char* path,
* @return * @return
* Zero if the blob is handled successfully, or non-zero on error. * Zero if the blob is handled successfully, or non-zero on error.
*/ */
static int guac_common_ssh_sftp_blob_handler(guac_client* client, static int guac_common_ssh_sftp_blob_handler(guac_user* user,
guac_stream* stream, void* data, int length) { guac_stream* stream, void* data, int length) {
/* Pull file from stream */ /* Pull file from stream */
@ -216,18 +215,18 @@ static int guac_common_ssh_sftp_blob_handler(guac_client* client,
/* Attempt write */ /* Attempt write */
if (libssh2_sftp_write(file, data, length) == length) { if (libssh2_sftp_write(file, data, length) == length) {
guac_client_log(client, GUAC_LOG_DEBUG, "%i bytes written", length); guac_user_log(user, GUAC_LOG_DEBUG, "%i bytes written", length);
guac_protocol_send_ack(client->socket, stream, "SFTP: OK", guac_protocol_send_ack(user->socket, stream, "SFTP: OK",
GUAC_PROTOCOL_STATUS_SUCCESS); GUAC_PROTOCOL_STATUS_SUCCESS);
guac_socket_flush(client->socket); guac_socket_flush(user->socket);
} }
/* Inform of any errors */ /* Inform of any errors */
else { else {
guac_client_log(client, GUAC_LOG_INFO, "Unable to write to file"); guac_user_log(user, GUAC_LOG_INFO, "Unable to write to file");
guac_protocol_send_ack(client->socket, stream, "SFTP: Write failed", guac_protocol_send_ack(user->socket, stream, "SFTP: Write failed",
GUAC_PROTOCOL_STATUS_SERVER_ERROR); GUAC_PROTOCOL_STATUS_SERVER_ERROR);
guac_socket_flush(client->socket); guac_socket_flush(user->socket);
} }
return 0; return 0;
@ -240,8 +239,8 @@ static int guac_common_ssh_sftp_blob_handler(guac_client* client,
* pointer to an open LIBSSH2_SFTP_HANDLE for the file to which the data * pointer to an open LIBSSH2_SFTP_HANDLE for the file to which the data
* has been written and which should now be closed. * has been written and which should now be closed.
* *
* @param client * @param user
* The client receiving the end message. * The user receiving the end message.
* *
* @param stream * @param stream
* The Guacamole protocol stream associated with the received end message. * The Guacamole protocol stream associated with the received end message.
@ -249,7 +248,7 @@ static int guac_common_ssh_sftp_blob_handler(guac_client* client,
* @return * @return
* Zero if the file is closed successfully, or non-zero on error. * Zero if the file is closed successfully, or non-zero on error.
*/ */
static int guac_common_ssh_sftp_end_handler(guac_client* client, static int guac_common_ssh_sftp_end_handler(guac_user* user,
guac_stream* stream) { guac_stream* stream) {
/* Pull file from stream */ /* Pull file from stream */
@ -257,72 +256,68 @@ static int guac_common_ssh_sftp_end_handler(guac_client* client,
/* Attempt to close file */ /* Attempt to close file */
if (libssh2_sftp_close(file) == 0) { if (libssh2_sftp_close(file) == 0) {
guac_client_log(client, GUAC_LOG_DEBUG, "File closed"); guac_user_log(user, GUAC_LOG_DEBUG, "File closed");
guac_protocol_send_ack(client->socket, stream, "SFTP: OK", guac_protocol_send_ack(user->socket, stream, "SFTP: OK",
GUAC_PROTOCOL_STATUS_SUCCESS); GUAC_PROTOCOL_STATUS_SUCCESS);
guac_socket_flush(client->socket); guac_socket_flush(user->socket);
} }
else { else {
guac_client_log(client, GUAC_LOG_INFO, "Unable to close file"); guac_user_log(user, GUAC_LOG_INFO, "Unable to close file");
guac_protocol_send_ack(client->socket, stream, "SFTP: Close failed", guac_protocol_send_ack(user->socket, stream, "SFTP: Close failed",
GUAC_PROTOCOL_STATUS_SERVER_ERROR); GUAC_PROTOCOL_STATUS_SERVER_ERROR);
guac_socket_flush(client->socket); guac_socket_flush(user->socket);
} }
return 0; return 0;
} }
int guac_common_ssh_sftp_handle_file_stream(guac_object* filesystem, int guac_common_ssh_sftp_handle_file_stream(
guac_common_ssh_sftp_filesystem* filesystem, guac_user* user,
guac_stream* stream, char* mimetype, char* filename) { guac_stream* stream, char* mimetype, char* filename) {
guac_common_ssh_sftp_data* sftp_data =
(guac_common_ssh_sftp_data*) filesystem->data;
guac_client* client = sftp_data->ssh_session->client;
char fullpath[GUAC_COMMON_SSH_SFTP_MAX_PATH]; char fullpath[GUAC_COMMON_SSH_SFTP_MAX_PATH];
LIBSSH2_SFTP_HANDLE* file; LIBSSH2_SFTP_HANDLE* file;
/* Concatenate filename with path */ /* Concatenate filename with path */
if (!guac_ssh_append_filename(fullpath, sftp_data->upload_path, if (!guac_ssh_append_filename(fullpath, filesystem->upload_path,
filename)) { filename)) {
guac_client_log(client, GUAC_LOG_DEBUG, guac_user_log(user, GUAC_LOG_DEBUG,
"Filename \"%s\" is invalid or resulting path is too long", "Filename \"%s\" is invalid or resulting path is too long",
filename); filename);
/* Abort transfer - invalid filename */ /* Abort transfer - invalid filename */
guac_protocol_send_ack(client->socket, stream, guac_protocol_send_ack(user->socket, stream,
"SFTP: Illegal filename", "SFTP: Illegal filename",
GUAC_PROTOCOL_STATUS_CLIENT_BAD_REQUEST); GUAC_PROTOCOL_STATUS_CLIENT_BAD_REQUEST);
guac_socket_flush(client->socket); guac_socket_flush(user->socket);
return 0; return 0;
} }
/* Open file via SFTP */ /* Open file via SFTP */
file = libssh2_sftp_open(sftp_data->sftp_session, fullpath, file = libssh2_sftp_open(filesystem->sftp_session, fullpath,
LIBSSH2_FXF_WRITE | LIBSSH2_FXF_CREAT | LIBSSH2_FXF_TRUNC, LIBSSH2_FXF_WRITE | LIBSSH2_FXF_CREAT | LIBSSH2_FXF_TRUNC,
S_IRUSR | S_IWUSR); S_IRUSR | S_IWUSR);
/* Inform of status */ /* Inform of status */
if (file != NULL) { if (file != NULL) {
guac_client_log(client, GUAC_LOG_DEBUG, guac_user_log(user, GUAC_LOG_DEBUG,
"File \"%s\" opened", "File \"%s\" opened",
fullpath); fullpath);
guac_protocol_send_ack(client->socket, stream, "SFTP: File opened", guac_protocol_send_ack(user->socket, stream, "SFTP: File opened",
GUAC_PROTOCOL_STATUS_SUCCESS); GUAC_PROTOCOL_STATUS_SUCCESS);
guac_socket_flush(client->socket); guac_socket_flush(user->socket);
} }
else { else {
guac_client_log(client, GUAC_LOG_INFO, guac_user_log(user, GUAC_LOG_INFO,
"Unable to open file \"%s\"", fullpath); "Unable to open file \"%s\"", fullpath);
guac_protocol_send_ack(client->socket, stream, "SFTP: Open failed", guac_protocol_send_ack(user->socket, stream, "SFTP: Open failed",
guac_sftp_get_status(filesystem)); guac_sftp_get_status(filesystem));
guac_socket_flush(client->socket); guac_socket_flush(user->socket);
} }
/* Set handlers for file stream */ /* Set handlers for file stream */
@ -341,8 +336,8 @@ int guac_common_ssh_sftp_handle_file_stream(guac_object* filesystem,
* The data associated with the given stream is expected to be a pointer to an * The data associated with the given stream is expected to be a pointer to an
* open LIBSSH2_SFTP_HANDLE for the file from which the data is to be read. * open LIBSSH2_SFTP_HANDLE for the file from which the data is to be read.
* *
* @param client * @param user
* The client receiving the ack message. * The user receiving the ack message.
* *
* @param stream * @param stream
* The Guacamole protocol stream associated with the received ack message. * The Guacamole protocol stream associated with the received ack message.
@ -358,7 +353,7 @@ int guac_common_ssh_sftp_handle_file_stream(guac_object* filesystem,
* @return * @return
* Zero if the file is read from successfully, or non-zero on error. * Zero if the file is read from successfully, or non-zero on error.
*/ */
static int guac_common_ssh_sftp_ack_handler(guac_client* client, static int guac_common_ssh_sftp_ack_handler(guac_user* user,
guac_stream* stream, char* message, guac_protocol_status status) { guac_stream* stream, char* message, guac_protocol_status status) {
/* Pull file from stream */ /* Pull file from stream */
@ -373,82 +368,75 @@ static int guac_common_ssh_sftp_ack_handler(guac_client* client,
/* If bytes read, send as blob */ /* If bytes read, send as blob */
if (bytes_read > 0) { if (bytes_read > 0) {
guac_protocol_send_blob(client->socket, stream, guac_protocol_send_blob(user->socket, stream,
buffer, bytes_read); buffer, bytes_read);
guac_client_log(client, GUAC_LOG_DEBUG, "%i bytes sent to client", guac_user_log(user, GUAC_LOG_DEBUG, "%i bytes sent to user",
bytes_read); bytes_read);
} }
/* If EOF, send end */ /* If EOF, send end */
else if (bytes_read == 0) { else if (bytes_read == 0) {
guac_client_log(client, GUAC_LOG_DEBUG, "File sent"); guac_user_log(user, GUAC_LOG_DEBUG, "File sent");
guac_protocol_send_end(client->socket, stream); guac_protocol_send_end(user->socket, stream);
guac_client_free_stream(client, stream); guac_user_free_stream(user, stream);
} }
/* Otherwise, fail stream */ /* Otherwise, fail stream */
else { else {
guac_client_log(client, GUAC_LOG_INFO, "Error reading file"); guac_user_log(user, GUAC_LOG_INFO, "Error reading file");
guac_protocol_send_end(client->socket, stream); guac_protocol_send_end(user->socket, stream);
guac_client_free_stream(client, stream); guac_user_free_stream(user, stream);
} }
guac_socket_flush(client->socket); guac_socket_flush(user->socket);
} }
/* Otherwise, return stream to client */ /* Otherwise, return stream to user */
else else
guac_client_free_stream(client, stream); guac_user_free_stream(user, stream);
return 0; return 0;
} }
guac_stream* guac_common_ssh_sftp_download_file(guac_object* filesystem, guac_stream* guac_common_ssh_sftp_download_file(
guac_common_ssh_sftp_filesystem* filesystem, guac_user* user,
char* filename) { char* filename) {
guac_common_ssh_sftp_data* sftp_data =
(guac_common_ssh_sftp_data*) filesystem->data;
guac_client* client = sftp_data->ssh_session->client;
guac_stream* stream; guac_stream* stream;
LIBSSH2_SFTP_HANDLE* file; LIBSSH2_SFTP_HANDLE* file;
/* Attempt to open file for reading */ /* Attempt to open file for reading */
file = libssh2_sftp_open(sftp_data->sftp_session, filename, file = libssh2_sftp_open(filesystem->sftp_session, filename,
LIBSSH2_FXF_READ, 0); LIBSSH2_FXF_READ, 0);
if (file == NULL) { if (file == NULL) {
guac_client_log(client, GUAC_LOG_INFO, guac_user_log(user, GUAC_LOG_INFO,
"Unable to read file \"%s\"", filename); "Unable to read file \"%s\"", filename);
return NULL; return NULL;
} }
/* Allocate stream */ /* Allocate stream */
stream = guac_client_alloc_stream(client); stream = guac_user_alloc_stream(user);
stream->ack_handler = guac_common_ssh_sftp_ack_handler; stream->ack_handler = guac_common_ssh_sftp_ack_handler;
stream->data = file; stream->data = file;
/* Send stream start, strip name */ /* Send stream start, strip name */
filename = basename(filename); filename = basename(filename);
guac_protocol_send_file(client->socket, stream, guac_protocol_send_file(user->socket, stream,
"application/octet-stream", filename); "application/octet-stream", filename);
guac_socket_flush(client->socket); guac_socket_flush(user->socket);
guac_client_log(client, GUAC_LOG_DEBUG, "Sending file \"%s\"", filename); guac_user_log(user, GUAC_LOG_DEBUG, "Sending file \"%s\"", filename);
return stream; return stream;
} }
void guac_common_ssh_sftp_set_upload_path(guac_object* filesystem, void guac_common_ssh_sftp_set_upload_path(
const char* path) { guac_common_ssh_sftp_filesystem* filesystem, const char* path) {
guac_common_ssh_sftp_data* sftp_data = guac_client* client = filesystem->ssh_session->client;
(guac_common_ssh_sftp_data*) filesystem->data;
guac_client* client = sftp_data->ssh_session->client;
/* Ignore requests which exceed maximum-allowed path */ /* Ignore requests which exceed maximum-allowed path */
int length = strnlen(path, GUAC_COMMON_SSH_SFTP_MAX_PATH)+1; int length = strnlen(path, GUAC_COMMON_SSH_SFTP_MAX_PATH)+1;
@ -460,7 +448,7 @@ void guac_common_ssh_sftp_set_upload_path(guac_object* filesystem,
} }
/* Copy path */ /* Copy path */
memcpy(sftp_data->upload_path, path, length); memcpy(filesystem->upload_path, path, length);
guac_client_log(client, GUAC_LOG_DEBUG, "Upload path set to \"%s\"", path); guac_client_log(client, GUAC_LOG_DEBUG, "Upload path set to \"%s\"", path);
} }
@ -469,8 +457,8 @@ void guac_common_ssh_sftp_set_upload_path(guac_object* filesystem,
* Handler for ack messages received due to receipt of a "body" or "blob" * Handler for ack messages received due to receipt of a "body" or "blob"
* instruction associated with a SFTP directory list operation. * instruction associated with a SFTP directory list operation.
* *
* @param client * @param user
* The client receiving the ack message. * The user receiving the ack message.
* *
* @param stream * @param stream
* The Guacamole protocol stream associated with the received ack message. * The Guacamole protocol stream associated with the received ack message.
@ -486,7 +474,7 @@ void guac_common_ssh_sftp_set_upload_path(guac_object* filesystem,
* @return * @return
* Zero on success, non-zero on error. * Zero on success, non-zero on error.
*/ */
static int guac_common_ssh_sftp_ls_ack_handler(guac_client* client, static int guac_common_ssh_sftp_ls_ack_handler(guac_user* user,
guac_stream* stream, char* message, guac_protocol_status status) { guac_stream* stream, char* message, guac_protocol_status status) {
int bytes_read; int bytes_read;
@ -498,14 +486,14 @@ static int guac_common_ssh_sftp_ls_ack_handler(guac_client* client,
guac_common_ssh_sftp_ls_state* list_state = guac_common_ssh_sftp_ls_state* list_state =
(guac_common_ssh_sftp_ls_state*) stream->data; (guac_common_ssh_sftp_ls_state*) stream->data;
guac_common_ssh_sftp_data* sftp_data = list_state->sftp_data; guac_common_ssh_sftp_filesystem* filesystem = list_state->filesystem;
LIBSSH2_SFTP* sftp = sftp_data->sftp_session; LIBSSH2_SFTP* sftp = filesystem->sftp_session;
/* If unsuccessful, free stream and abort */ /* If unsuccessful, free stream and abort */
if (status != GUAC_PROTOCOL_STATUS_SUCCESS) { if (status != GUAC_PROTOCOL_STATUS_SUCCESS) {
libssh2_sftp_closedir(list_state->directory); libssh2_sftp_closedir(list_state->directory);
guac_client_free_stream(client, stream); guac_user_free_stream(user, stream);
free(list_state); free(list_state);
return 0; return 0;
} }
@ -525,7 +513,7 @@ static int guac_common_ssh_sftp_ls_ack_handler(guac_client* client,
if (!guac_ssh_append_filename(absolute_path, if (!guac_ssh_append_filename(absolute_path,
list_state->directory_name, filename)) { list_state->directory_name, filename)) {
guac_client_log(client, GUAC_LOG_DEBUG, guac_user_log(user, GUAC_LOG_DEBUG,
"Skipping filename \"%s\" - filename is invalid or " "Skipping filename \"%s\" - filename is invalid or "
"resulting path is too long", filename); "resulting path is too long", filename);
@ -539,12 +527,12 @@ static int guac_common_ssh_sftp_ls_ack_handler(guac_client* client,
/* Determine mimetype */ /* Determine mimetype */
const char* mimetype; const char* mimetype;
if (LIBSSH2_SFTP_S_ISDIR(attributes.permissions)) if (LIBSSH2_SFTP_S_ISDIR(attributes.permissions))
mimetype = GUAC_CLIENT_STREAM_INDEX_MIMETYPE; mimetype = GUAC_USER_STREAM_INDEX_MIMETYPE;
else else
mimetype = "application/octet-stream"; mimetype = "application/octet-stream";
/* Write entry */ /* Write entry */
blob_written |= guac_common_json_write_property(client, stream, blob_written |= guac_common_json_write_property(user, stream,
&list_state->json_state, absolute_path, mimetype); &list_state->json_state, absolute_path, mimetype);
} }
@ -553,20 +541,20 @@ static int guac_common_ssh_sftp_ls_ack_handler(guac_client* client,
if (bytes_read <= 0) { if (bytes_read <= 0) {
/* Complete JSON object */ /* Complete JSON object */
guac_common_json_end_object(client, stream, &list_state->json_state); guac_common_json_end_object(user, stream, &list_state->json_state);
guac_common_json_flush(client, stream, &list_state->json_state); guac_common_json_flush(user, stream, &list_state->json_state);
/* Clean up resources */ /* Clean up resources */
libssh2_sftp_closedir(list_state->directory); libssh2_sftp_closedir(list_state->directory);
free(list_state); free(list_state);
/* Signal of stream */ /* Signal of stream */
guac_protocol_send_end(client->socket, stream); guac_protocol_send_end(user->socket, stream);
guac_client_free_stream(client, stream); guac_user_free_stream(user, stream);
} }
guac_socket_flush(client->socket); guac_socket_flush(user->socket);
return 0; return 0;
} }
@ -576,8 +564,8 @@ static int guac_common_ssh_sftp_ls_ack_handler(guac_client* client,
* the Guacamole protocol, get messages request the body of a file within the * the Guacamole protocol, get messages request the body of a file within the
* filesystem. * filesystem.
* *
* @param client * @param user
* The client receiving the get message. * The user who sent the get message.
* *
* @param object * @param object
* The Guacamole protocol object associated with the get request itself. * The Guacamole protocol object associated with the get request itself.
@ -588,18 +576,18 @@ static int guac_common_ssh_sftp_ls_ack_handler(guac_client* client,
* @return * @return
* Zero on success, non-zero on error. * Zero on success, non-zero on error.
*/ */
static int guac_common_ssh_sftp_get_handler(guac_client* client, static int guac_common_ssh_sftp_get_handler(guac_user* user,
guac_object* object, char* name) { guac_object* object, char* name) {
guac_common_ssh_sftp_data* sftp_data = guac_common_ssh_sftp_filesystem* filesystem =
(guac_common_ssh_sftp_data*) object->data; (guac_common_ssh_sftp_filesystem*) object->data;
LIBSSH2_SFTP* sftp = sftp_data->sftp_session; LIBSSH2_SFTP* sftp = filesystem->sftp_session;
LIBSSH2_SFTP_ATTRIBUTES attributes; LIBSSH2_SFTP_ATTRIBUTES attributes;
/* Attempt to read file information */ /* Attempt to read file information */
if (libssh2_sftp_stat(sftp, name, &attributes)) { if (libssh2_sftp_stat(sftp, name, &attributes)) {
guac_client_log(client, GUAC_LOG_INFO, "Unable to read file \"%s\"", guac_user_log(user, GUAC_LOG_INFO, "Unable to read file \"%s\"",
name); name);
return 0; return 0;
} }
@ -610,7 +598,7 @@ static int guac_common_ssh_sftp_get_handler(guac_client* client,
/* Open as directory */ /* Open as directory */
LIBSSH2_SFTP_HANDLE* dir = libssh2_sftp_opendir(sftp, name); LIBSSH2_SFTP_HANDLE* dir = libssh2_sftp_opendir(sftp, name);
if (dir == NULL) { if (dir == NULL) {
guac_client_log(client, GUAC_LOG_INFO, guac_user_log(user, GUAC_LOG_INFO,
"Unable to read directory \"%s\"", name); "Unable to read directory \"%s\"", name);
return 0; return 0;
} }
@ -620,21 +608,21 @@ static int guac_common_ssh_sftp_get_handler(guac_client* client,
malloc(sizeof(guac_common_ssh_sftp_ls_state)); malloc(sizeof(guac_common_ssh_sftp_ls_state));
list_state->directory = dir; list_state->directory = dir;
list_state->sftp_data = sftp_data; list_state->filesystem = filesystem;
strncpy(list_state->directory_name, name, strncpy(list_state->directory_name, name,
sizeof(list_state->directory_name) - 1); sizeof(list_state->directory_name) - 1);
/* Allocate stream for body */ /* Allocate stream for body */
guac_stream* stream = guac_client_alloc_stream(client); guac_stream* stream = guac_user_alloc_stream(user);
stream->ack_handler = guac_common_ssh_sftp_ls_ack_handler; stream->ack_handler = guac_common_ssh_sftp_ls_ack_handler;
stream->data = list_state; stream->data = list_state;
/* Init JSON object state */ /* Init JSON object state */
guac_common_json_begin_object(client, stream, &list_state->json_state); guac_common_json_begin_object(user, stream, &list_state->json_state);
/* Associate new stream with get request */ /* Associate new stream with get request */
guac_protocol_send_body(client->socket, object, stream, guac_protocol_send_body(user->socket, object, stream,
GUAC_CLIENT_STREAM_INDEX_MIMETYPE, name); GUAC_USER_STREAM_INDEX_MIMETYPE, name);
} }
@ -645,23 +633,23 @@ static int guac_common_ssh_sftp_get_handler(guac_client* client,
LIBSSH2_SFTP_HANDLE* file = libssh2_sftp_open(sftp, name, LIBSSH2_SFTP_HANDLE* file = libssh2_sftp_open(sftp, name,
LIBSSH2_FXF_READ, 0); LIBSSH2_FXF_READ, 0);
if (file == NULL) { if (file == NULL) {
guac_client_log(client, GUAC_LOG_INFO, guac_user_log(user, GUAC_LOG_INFO,
"Unable to read file \"%s\"", name); "Unable to read file \"%s\"", name);
return 0; return 0;
} }
/* Allocate stream for body */ /* Allocate stream for body */
guac_stream* stream = guac_client_alloc_stream(client); guac_stream* stream = guac_user_alloc_stream(user);
stream->ack_handler = guac_common_ssh_sftp_ack_handler; stream->ack_handler = guac_common_ssh_sftp_ack_handler;
stream->data = file; stream->data = file;
/* Associate new stream with get request */ /* Associate new stream with get request */
guac_protocol_send_body(client->socket, object, stream, guac_protocol_send_body(user->socket, object, stream,
"application/octet-stream", name); "application/octet-stream", name);
} }
guac_socket_flush(client->socket); guac_socket_flush(user->socket);
return 0; return 0;
} }
@ -670,14 +658,14 @@ static int guac_common_ssh_sftp_get_handler(guac_client* client,
* the Guacamole protocol, put messages request write access to a file within * the Guacamole protocol, put messages request write access to a file within
* the filesystem. * the filesystem.
* *
* @param client * @param user
* The client receiving the put message. * The user who sent the put message.
* *
* @param object * @param object
* The Guacamole protocol object associated with the put request itself. * The Guacamole protocol object associated with the put request itself.
* *
* @param stream * @param stream
* The Guacamole protocol stream along which the client will be sending * The Guacamole protocol stream along which the user will be sending
* file data. * file data.
* *
* @param mimetype * @param mimetype
@ -689,13 +677,13 @@ static int guac_common_ssh_sftp_get_handler(guac_client* client,
* @return * @return
* Zero on success, non-zero on error. * Zero on success, non-zero on error.
*/ */
static int guac_common_ssh_sftp_put_handler(guac_client* client, static int guac_common_ssh_sftp_put_handler(guac_user* user,
guac_object* object, guac_stream* stream, char* mimetype, char* name) { guac_object* object, guac_stream* stream, char* mimetype, char* name) {
guac_common_ssh_sftp_data* sftp_data = guac_common_ssh_sftp_filesystem* filesystem =
(guac_common_ssh_sftp_data*) object->data; (guac_common_ssh_sftp_filesystem*) object->data;
LIBSSH2_SFTP* sftp = sftp_data->sftp_session; LIBSSH2_SFTP* sftp = filesystem->sftp_session;
/* Open file via SFTP */ /* Open file via SFTP */
LIBSSH2_SFTP_HANDLE* file = libssh2_sftp_open(sftp, name, LIBSSH2_SFTP_HANDLE* file = libssh2_sftp_open(sftp, name,
@ -704,17 +692,17 @@ static int guac_common_ssh_sftp_put_handler(guac_client* client,
/* Acknowledge stream if successful */ /* Acknowledge stream if successful */
if (file != NULL) { if (file != NULL) {
guac_client_log(client, GUAC_LOG_DEBUG, "File \"%s\" opened", name); guac_user_log(user, GUAC_LOG_DEBUG, "File \"%s\" opened", name);
guac_protocol_send_ack(client->socket, stream, "SFTP: File opened", guac_protocol_send_ack(user->socket, stream, "SFTP: File opened",
GUAC_PROTOCOL_STATUS_SUCCESS); GUAC_PROTOCOL_STATUS_SUCCESS);
} }
/* Abort on failure */ /* Abort on failure */
else { else {
guac_client_log(client, GUAC_LOG_INFO, guac_user_log(user, GUAC_LOG_INFO,
"Unable to open file \"%s\"", name); "Unable to open file \"%s\"", name);
guac_protocol_send_ack(client->socket, stream, "SFTP: Open failed", guac_protocol_send_ack(user->socket, stream, "SFTP: Open failed",
guac_sftp_get_status(object)); guac_sftp_get_status(filesystem));
} }
/* Set handlers for file stream */ /* Set handlers for file stream */
@ -724,60 +712,75 @@ static int guac_common_ssh_sftp_put_handler(guac_client* client,
/* Store file within stream */ /* Store file within stream */
stream->data = file; stream->data = file;
guac_socket_flush(client->socket); guac_socket_flush(user->socket);
return 0; return 0;
} }
guac_object* guac_common_ssh_create_sftp_filesystem( void* guac_common_ssh_expose_sftp_filesystem(guac_user* user, void* data) {
guac_common_ssh_session* session,
const char* name) {
guac_client* client = session->client; guac_common_ssh_sftp_filesystem* filesystem =
(guac_common_ssh_sftp_filesystem*) data;
/* No need to expose if there is no filesystem or the user has left */
if (user == NULL || filesystem == NULL)
return NULL;
/* Allocate and expose filesystem object for user */
return guac_common_ssh_alloc_sftp_filesystem_object(filesystem, user);
}
guac_object* guac_common_ssh_alloc_sftp_filesystem_object(
guac_common_ssh_sftp_filesystem* filesystem, guac_user* user) {
/* Init filesystem */
guac_object* fs_object = guac_user_alloc_object(user);
fs_object->get_handler = guac_common_ssh_sftp_get_handler;
fs_object->put_handler = guac_common_ssh_sftp_put_handler;
fs_object->data = filesystem;
/* Send filesystem to user */
guac_protocol_send_filesystem(user->socket, fs_object, filesystem->name);
guac_socket_flush(user->socket);
return fs_object;
}
guac_common_ssh_sftp_filesystem* guac_common_ssh_create_sftp_filesystem(
guac_common_ssh_session* session, const char* name) {
/* Request SFTP */ /* Request SFTP */
LIBSSH2_SFTP* sftp_session = libssh2_sftp_init(session->session); LIBSSH2_SFTP* sftp_session = libssh2_sftp_init(session->session);
if (sftp_session == NULL) { if (sftp_session == NULL)
guac_client_abort(client, GUAC_PROTOCOL_STATUS_UPSTREAM_ERROR,
"Unable to start SFTP session.");
return NULL; return NULL;
}
/* Allocate data for SFTP session */ /* Allocate data for SFTP session */
guac_common_ssh_sftp_data* sftp_data = guac_common_ssh_sftp_filesystem* filesystem =
malloc(sizeof(guac_common_ssh_sftp_data)); malloc(sizeof(guac_common_ssh_sftp_filesystem));
/* Associate SSH session with SFTP data */ /* Associate SSH session with SFTP data and user */
sftp_data->ssh_session = session; filesystem->name = strdup(name);
sftp_data->sftp_session = sftp_session; filesystem->ssh_session = session;
filesystem->sftp_session = sftp_session;
/* Initially upload files to current directory */ /* Initially upload files to current directory */
strcpy(sftp_data->upload_path, "."); strcpy(filesystem->upload_path, ".");
/* Init filesystem */
guac_object* filesystem = guac_client_alloc_object(client);
filesystem->get_handler = guac_common_ssh_sftp_get_handler;
filesystem->put_handler = guac_common_ssh_sftp_put_handler;
filesystem->data = sftp_data;
/* Send filesystem to client */
guac_protocol_send_filesystem(client->socket, filesystem, "/");
guac_socket_flush(client->socket);
/* Return allocated filesystem */ /* Return allocated filesystem */
return filesystem; return filesystem;
} }
void guac_common_ssh_destroy_sftp_filesystem(guac_object* filesystem) { void guac_common_ssh_destroy_sftp_filesystem(
guac_common_ssh_sftp_filesystem* filesystem) {
guac_common_ssh_sftp_data* sftp_data =
(guac_common_ssh_sftp_data*) filesystem->data;
/* Shutdown SFTP session */ /* Shutdown SFTP session */
libssh2_sftp_shutdown(sftp_data->sftp_session); libssh2_sftp_shutdown(filesystem->sftp_session);
/* Clean up the SFTP filesystem object */ /* Free associated memory */
guac_client_free_object(sftp_data->ssh_session->client, filesystem); free(filesystem->name);
free(filesystem);
} }

View File

@ -26,8 +26,8 @@
#include "guac_json.h" #include "guac_json.h"
#include "guac_ssh.h" #include "guac_ssh.h"
#include <guacamole/client.h>
#include <guacamole/object.h> #include <guacamole/object.h>
#include <guacamole/user.h>
#include <libssh2.h> #include <libssh2.h>
#include <libssh2_sftp.h> #include <libssh2_sftp.h>
@ -37,9 +37,15 @@
#define GUAC_COMMON_SSH_SFTP_MAX_PATH 2048 #define GUAC_COMMON_SSH_SFTP_MAX_PATH 2048
/** /**
* Data associated with an SFTP-driven filesystem object. * Representation of an SFTP-driven filesystem object. Unlike guac_object, this
* structure is not tied to any particular user.
*/ */
typedef struct guac_common_ssh_sftp_data { typedef struct guac_common_ssh_sftp_filesystem {
/**
* The human-readable display name of this filesystem.
*/
char* name;
/** /**
* The distinct SSH session used for SFTP. * The distinct SSH session used for SFTP.
@ -57,7 +63,7 @@ typedef struct guac_common_ssh_sftp_data {
*/ */
char upload_path[GUAC_COMMON_SSH_SFTP_MAX_PATH]; char upload_path[GUAC_COMMON_SSH_SFTP_MAX_PATH];
} guac_common_ssh_sftp_data; } guac_common_ssh_sftp_filesystem;
/** /**
* The current state of a directory listing operation. * The current state of a directory listing operation.
@ -65,9 +71,9 @@ typedef struct guac_common_ssh_sftp_data {
typedef struct guac_common_ssh_sftp_ls_state { typedef struct guac_common_ssh_sftp_ls_state {
/** /**
* Data associated with the current SFTP session. * The SFTP filesystem being listed.
*/ */
guac_common_ssh_sftp_data* sftp_data; guac_common_ssh_sftp_filesystem* filesystem;
/** /**
* Reference to the directory currently being listed over SFTP. This * Reference to the directory currently being listed over SFTP. This
@ -91,50 +97,102 @@ typedef struct guac_common_ssh_sftp_ls_state {
* Creates a new Guacamole filesystem object which provides access to files * Creates a new Guacamole filesystem object which provides access to files
* and directories via SFTP using the given SSH session. When the filesystem * and directories via SFTP using the given SSH session. When the filesystem
* will no longer be used, it must be explicitly destroyed with * will no longer be used, it must be explicitly destroyed with
* guac_common_ssh_destroy_sftp_filesystem(). * guac_common_ssh_destroy_sftp_filesystem(). The resulting object is not
* automatically exposed to users of the connection - filesystem operations
* must be mediated either through various handlers or through exposing a
* filesystem guac_object via guac_common_ssh_alloc_sftp_filesystem_object().
* *
* @param session * @param session
* The session to use to provide SFTP. This session will automatically be * The session to use to provide SFTP. This session will automatically be
* destroyed when this filesystem is destroyed. * destroyed when this filesystem is destroyed.
* *
* @param name * @param name
* The name to send as the name of the filesystem. * The name to send as the name of the filesystem whenever it is exposed
* to a user.
* *
* @return * @return
* A new Guacamole filesystem object, already configured to use SFTP for * A new SFTP filesystem object, not yet exposed to users.
* uploading and downloading files.
*/ */
guac_object* guac_common_ssh_create_sftp_filesystem( guac_common_ssh_sftp_filesystem* guac_common_ssh_create_sftp_filesystem(
guac_common_ssh_session* session, guac_common_ssh_session* session, const char* name);
const char* name);
/** /**
* Destroys the given filesystem object, disconnecting from SFTP and freeing * Destroys the given filesystem object, disconnecting from SFTP and freeing
* and associated resources. Any associated session or user objects must be * and associated resources. Any associated session or user objects must be
* explicitly destroyed. * explicitly destroyed.
* *
* @param object * @param filesystem
* The filesystem object to destroy. * The filesystem object to destroy.
*/ */
void guac_common_ssh_destroy_sftp_filesystem(guac_object* filesystem); void guac_common_ssh_destroy_sftp_filesystem(
guac_common_ssh_sftp_filesystem* filesystem);
/**
* Creates and exposes a new filesystem guac_object to the given user,
* providing access to the files within the given SFTP filesystem. The
* allocated guac_object must eventually be freed via guac_user_free_object().
*
* @param filesystem
* The filesystem object to expose.
*
* @param user
* The user that the SFTP filesystem should be exposed to.
*
* @return
* A new Guacamole filesystem object, configured to use SFTP for uploading
* and downloading files.
*/
guac_object* guac_common_ssh_alloc_sftp_filesystem_object(
guac_common_ssh_sftp_filesystem* filesystem, guac_user* user);
/**
* Allocates a new filesystem guac_object for the given user, returning the
* resulting guac_object. This function is provided for convenience, as it is
* can be used as the callback for guac_client_foreach_user() or
* guac_client_for_owner(). Note that this guac_object will be tracked
* internally by libguac, will be provided to us in the parameters of handlers
* related to that guac_object, and will automatically be freed when the
* associated guac_user is freed, so the return value of this function can
* safely be ignored.
*
* If either the given user or the given filesystem are NULL, then this
* function has no effect.
*
* @param user
* The use to expose the filesystem to, or NULL if nothing should be
* exposed.
*
* @param data
* A pointer to the guac_common_ssh_sftp_filesystem instance to expose
* to the given user, or NULL if nothing should be exposed.
*
* @return
* The guac_object allocated for the newly-exposed filesystem, or NULL if
* no filesystem object could be allocated.
*/
void* guac_common_ssh_expose_sftp_filesystem(guac_user* user, void* data);
/** /**
* Initiates an SFTP file download to the user via the Guacamole "file" * Initiates an SFTP file download to the user via the Guacamole "file"
* instruction. The download will be automatically monitored and continued * instruction. The download will be automatically monitored and continued
* after this function terminates in response to "ack" instructions received by * after this function terminates in response to "ack" instructions received by
* the client. * the user.
* *
* @param filesystem * @param filesystem
* The filesystem containing the file to be downloaded. * The filesystem containing the file to be downloaded.
* *
* @param user
* The user that should receive the file (via a "file" instruction).
*
* @param filename * @param filename
* The filename of the file to download, relative to the given filesystem. * The filename of the file to download, relative to the given filesystem.
* *
* @return * @return
* The file stream created for the file download, already configured to * The file stream created for the file download, already configured to
* properly handle "ack" responses, etc. from the client. * properly handle "ack" responses, etc. from the user.
*/ */
guac_stream* guac_common_ssh_sftp_download_file(guac_object* filesystem, guac_stream* guac_common_ssh_sftp_download_file(
guac_common_ssh_sftp_filesystem* filesystem, guac_user* user,
char* filename); char* filename);
/** /**
@ -145,6 +203,10 @@ guac_stream* guac_common_ssh_sftp_download_file(guac_object* filesystem,
* @param filesystem * @param filesystem
* The filesystem that should receive the uploaded file. * The filesystem that should receive the uploaded file.
* *
* @param user
* The user who is attempting to open the file stream (the user that sent
* the "file" instruction).
*
* @param stream * @param stream
* The stream through which the uploaded file data will be received. * The stream through which the uploaded file data will be received.
* *
@ -160,7 +222,8 @@ guac_stream* guac_common_ssh_sftp_download_file(guac_object* filesystem,
* Zero if the incoming stream has been handled successfully, non-zero on * Zero if the incoming stream has been handled successfully, non-zero on
* failure. * failure.
*/ */
int guac_common_ssh_sftp_handle_file_stream(guac_object* filesystem, int guac_common_ssh_sftp_handle_file_stream(
guac_common_ssh_sftp_filesystem* filesystem, guac_user* user,
guac_stream* stream, char* mimetype, char* filename); guac_stream* stream, char* mimetype, char* filename);
/** /**
@ -175,8 +238,8 @@ int guac_common_ssh_sftp_handle_file_stream(guac_object* filesystem,
* The path to use for future uploads submitted via the * The path to use for future uploads submitted via the
* guac_common_ssh_sftp_handle_file_stream() function. * guac_common_ssh_sftp_handle_file_stream() function.
*/ */
void guac_common_ssh_sftp_set_upload_path(guac_object* filesystem, void guac_common_ssh_sftp_set_upload_path(
const char* path); guac_common_ssh_sftp_filesystem* filesystem, const char* path);
#endif #endif