GUACAMOLE-200: Refactor RDPDR printer such that the "ack" handler cannot block.

This commit is contained in:
Michael Jumper 2017-02-13 14:44:06 -08:00
parent 4da4ce7279
commit 3fc43fba37
5 changed files with 850 additions and 353 deletions

View File

@ -72,6 +72,7 @@ guacdr_sources = \
guac_rdpdr/rdpdr_fs_service.c \
guac_rdpdr/rdpdr_messages.c \
guac_rdpdr/rdpdr_printer.c \
guac_rdpdr/rdpdr_print_job.c \
guac_rdpdr/rdpdr_service.c \
rdp_fs.c \
rdp_stream.c \
@ -89,6 +90,7 @@ noinst_HEADERS = \
guac_rdpdr/rdpdr_fs_service.h \
guac_rdpdr/rdpdr_messages.h \
guac_rdpdr/rdpdr_printer.h \
guac_rdpdr/rdpdr_print_job.h \
guac_rdpdr/rdpdr_service.h \
guac_rdpsnd/rdpsnd_messages.h \
guac_rdpsnd/rdpsnd_service.h \

View File

@ -0,0 +1,581 @@
/*
* 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 "config.h"
#include "rdpdr_print_job.h"
#include <guacamole/client.h>
#include <guacamole/protocol.h>
#include <guacamole/socket.h>
#include <guacamole/stream.h>
#include <guacamole/user.h>
#include <errno.h>
#include <pthread.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
/**
* The command to run when filtering postscript to produce PDF. This must be
* a NULL-terminated array of arguments, where the first argument is the name
* of the file to run.
*/
char* const guac_rdpdr_pdf_filter_command[] = {
"gs",
"-q",
"-dNOPAUSE",
"-dBATCH",
"-dSAFER",
"-dPARANOIDSAFER",
"-sDEVICE=pdfwrite",
"-sOutputFile=-",
"-c",
".setpdfwrite",
"-sstdout=/dev/null",
"-f",
"-",
NULL
};
/**
* Updates the state of the given print job. Any threads currently blocked by a
* call to guac_rdpdr_print_job_wait_for_ack() will be unblocked.
*
* @param job
* The print job whose state should be updated.
*
* @param state
* The new state to assign to the given print job.
*/
static void guac_rdpdr_print_job_set_state(guac_rdpdr_print_job* job,
guac_rdpdr_print_job_state state) {
pthread_mutex_lock(&(job->state_lock));
/* Update stream state, signalling modification */
job->state = state;
pthread_cond_signal(&(job->state_modified));
pthread_mutex_unlock(&(job->state_lock));
}
/**
* Suspends execution of the current thread until the state of the given print
* job is not GUAC_RDPDR_PRINT_JOB_WAITING_FOR_ACK. If the state of the print
* job is GUAC_RDPDR_PRINT_JOB_ACK_RECEIVED, the state is automatically reset
* back to GUAC_RDPDR_PRINT_JOB_WAITING_FOR_ACK prior to returning.
*
* @param job
* The print job to wait for.
*
* @return
* Zero if the state of the print job is GUAC_RDPDR_PRINT_JOB_CLOSED,
* non-zero if the state was GUAC_RDPDR_PRINT_JOB_ACK_RECEIVED and has been
* automatically reset to GUAC_RDPDR_PRINT_JOB_WAITING_FOR_ACK.
*/
static int guac_rdpdr_print_job_wait_for_ack(guac_rdpdr_print_job* job) {
/* Wait for ack if stream open and not yet received */
pthread_mutex_lock(&(job->state_lock));
if (job->state == GUAC_RDPDR_PRINT_JOB_WAITING_FOR_ACK)
pthread_cond_wait(&job->state_modified, &job->state_lock);
/* Reset state if ack received */
int got_ack = (job->state == GUAC_RDPDR_PRINT_JOB_ACK_RECEIVED);
if (got_ack)
job->state = GUAC_RDPDR_PRINT_JOB_WAITING_FOR_ACK;
/* Return whether ack was successfully received */
pthread_mutex_unlock(&(job->state_lock));
return got_ack;
}
/**
* Sends a "file" instruction to the given user describing the PDF file that
* will be sent using the output of the given print job. If the given user no
* longer exists, the print stream will be automatically terminated.
*
* @param user
* The user receiving the "file" instruction.
*
* @param data
* A pointer to the guac_rdpdr_print_job representing the print job being
* streamed.
*
* @return
* Always NULL.
*/
static void* guac_rdpdr_print_job_begin_stream(guac_user* user, void* data) {
guac_rdpdr_print_job* job = (guac_rdpdr_print_job*) data;
guac_client_log(job->client, GUAC_LOG_DEBUG, "Beginning print stream: %s",
job->filename);
/* Kill job and do nothing if user no longer exists */
if (user == NULL) {
guac_rdpdr_print_job_kill(job);
return NULL;
}
/* Send document as a PDF file stream */
guac_protocol_send_file(user->socket, job->stream,
"application/pdf", job->filename);
guac_socket_flush(user->socket);
return NULL;
}
/**
* Sends a "blob" instruction to the given user containing the provided data
* along the stream associated with the provided print job. If the given user
* no longer exists, the print stream will be automatically terminated.
*
* @param user
* The user receiving the "blob" instruction.
*
* @param data
* A pointer to an guac_rdpdr_print_blob structure containing the data to
* be written, the number of bytes being written, and the print job being
* streamed.
*
* @return
* Always NULL.
*/
static void* guac_rdpdr_print_job_send_blob(guac_user* user, void* data) {
guac_rdpdr_print_blob* blob = (guac_rdpdr_print_blob*) data;
guac_rdpdr_print_job* job = blob->job;
guac_client_log(job->client, GUAC_LOG_DEBUG, "Sending %i byte(s) "
"of filtered output.", blob->length);
/* Kill job and do nothing if user no longer exists */
if (user == NULL) {
guac_rdpdr_print_job_kill(job);
return NULL;
}
/* Send single blob of print data */
guac_protocol_send_blob(user->socket, job->stream,
blob->buffer, blob->length);
guac_socket_flush(user->socket);
return NULL;
}
/**
* Sends an "end" instruction to the given user, closing the stream associated
* with the given print job. If the given user no longer exists, the print
* stream will be automatically terminated.
*
* @param user
* The user receiving the "end" instruction.
*
* @param data
* A pointer to the guac_rdpdr_print_job representing the print job being
* streamed.
*
* @return
* Always NULL.
*/
static void* guac_rdpdr_print_job_end_stream(guac_user* user, void* data) {
guac_rdpdr_print_job* job = (guac_rdpdr_print_job*) data;
guac_client_log(job->client, GUAC_LOG_DEBUG, "End of print stream.");
/* Kill job and do nothing if user no longer exists */
if (user == NULL) {
guac_rdpdr_print_job_kill(job);
return NULL;
}
/* Explicitly close down stream */
guac_protocol_send_end(user->socket, job->stream);
guac_socket_flush(user->socket);
/* Clean up our end of the stream */
guac_user_free_stream(job->user, job->stream);
return NULL;
}
/**
* Handler for "ack" messages received in response to printed data. Additional
* data will be sent as a result or, if no data remains, the stream will be
* terminated. It is required that the data pointer of the provided stream be
* set to the file descriptor from which the printed data should be read.
*
* @param user
* The user to whom the printed data is being sent.
*
* @param stream
* The stream along which the printed data is to be sent. The data pointer
* of this stream MUST be set to the file descriptor from which the data
* being sent is to be read.
*
* @param message
* An arbitrary, human-readable message describing the success/failure of
* the operation being acknowledged (either stream creation or receipt of
* a blob).
*
* @param status
* The status code describing the success/failure of the operation being
* acknowledged (either stream creation or receipt of a blob).
*
* @return
* Always zero.
*/
static int guac_rdpdr_print_filter_ack_handler(guac_user* user,
guac_stream* stream, char* message, guac_protocol_status status) {
guac_rdpdr_print_job* job = (guac_rdpdr_print_job*) stream->data;
/* Update state for successful acks */
if (status == GUAC_PROTOCOL_STATUS_SUCCESS)
guac_rdpdr_print_job_set_state(job, GUAC_RDPDR_PRINT_JOB_ACK_RECEIVED);
/* Terminate stream if ack signals an error */
else {
/* Note that the stream was aborted by the user */
guac_client_log(job->client, GUAC_LOG_INFO, "User explicitly aborted "
"print stream.");
/* Kill job (the results will no longer be received) */
guac_rdpdr_print_job_kill(job);
}
return 0;
}
/**
* Forks a new print filtering process which accepts PostScript input and
* produces PDF output. File descriptors for writing input and reading output
* will automatically be allocated and must be manually closed when processing
* is complete.
*
* @param client
* The guac_client associated with the print job for which this filter
* process is being created.
*
* @param input_fd
* A pointer to an int which should receive the input file descriptor of
* the filter process. PostScript input for the filter process should be
* written to this file descriptor.
*
* @param output_fd
* A pointer to an int which should receive the output file descriptor of
* the filter process. PDF output from the filter process must be
* continuously read from this file descriptor or the pipeline may block.
*
* @return
* The PID of the filter process, or -1 if the filter process could not be
* created. If the filter process could not be created, the values assigned
* through input_fd and output_fd are undefined.
*/
static pid_t guac_rdpdr_create_filter_process(guac_client* client,
int* input_fd, int* output_fd) {
int child_pid;
int stdin_pipe[2];
int stdout_pipe[2];
/* Create STDIN pipe */
if (pipe(stdin_pipe)) {
guac_client_log(client, GUAC_LOG_ERROR, "Unable to create STDIN "
"pipe for PDF filter process: %s", strerror(errno));
return -1;
}
/* Create STDOUT pipe */
if (pipe(stdout_pipe)) {
guac_client_log(client, GUAC_LOG_ERROR, "Unable to create STDOUT "
"pipe for PDF filter process: %s", strerror(errno));
close(stdin_pipe[0]);
close(stdin_pipe[1]);
return -1;
}
/* Store parent side of stdin/stdout */
*input_fd = stdin_pipe[1];
*output_fd = stdout_pipe[0];
/* Fork child process */
child_pid = fork();
/* Log fork errors */
if (child_pid == -1) {
guac_client_log(client, GUAC_LOG_ERROR, "Unable to fork PDF filter "
"process: %s", strerror(errno));
close(stdin_pipe[0]);
close(stdin_pipe[1]);
close(stdout_pipe[0]);
close(stdout_pipe[1]);
return -1;
}
/* Child process */
if (child_pid == 0) {
/* Close unneeded ends of pipe */
close(stdin_pipe[1]);
close(stdout_pipe[0]);
/* Reassign file descriptors as STDIN/STDOUT */
dup2(stdin_pipe[0], STDIN_FILENO);
dup2(stdout_pipe[1], STDOUT_FILENO);
/* Run PDF filter */
guac_client_log(client, GUAC_LOG_INFO, "Running %s",
guac_rdpdr_pdf_filter_command[0]);
if (execvp(guac_rdpdr_pdf_filter_command[0],
guac_rdpdr_pdf_filter_command) < 0)
guac_client_log(client, GUAC_LOG_ERROR, "Unable to execute PDF "
"filter command: %s", strerror(errno));
else
guac_client_log(client, GUAC_LOG_ERROR, "Unable to execute PDF "
"filter command, but no error given");
/* Terminate child process */
exit(1);
}
/* Log fork success */
guac_client_log(client, GUAC_LOG_INFO, "Created PDF filter process "
"PID=%i", child_pid);
/* Close unneeded ends of pipe */
close(stdin_pipe[0]);
close(stdout_pipe[1]);
return child_pid;
}
/**
* Thread which continuously reads from the output file descriptor associated
* with the given print job, writing filtered PDF output to the associated
* Guacamole stream, and terminating only after the print job has completed
* processing or the associated Guacamole stream has closed.
*
* @param data
* A pointer to the guac_rdpdr_print_job representing the print job that
* should be read.
*
* @return
* Always NULL.
*/
static void* guac_rdpdr_print_job_output_thread(void* data) {
int length;
char buffer[6048];
guac_rdpdr_print_job* job = (guac_rdpdr_print_job*) data;
guac_client_log(job->client, GUAC_LOG_DEBUG, "Reading output from filter "
"process...");
/* Read continuously while data remains */
while ((length = read(job->output_fd, buffer, sizeof(buffer))) > 0) {
/* Wait for client to be ready for blob */
if (guac_rdpdr_print_job_wait_for_ack(job)) {
guac_rdpdr_print_blob blob = {
.job = job,
.buffer = buffer,
.length = length
};
/* Write a single blob of output */
guac_client_for_user(job->client, job->user,
guac_rdpdr_print_job_send_blob, &blob);
}
/* Abort if stream is closed */
else {
guac_client_log(job->client, GUAC_LOG_DEBUG, "Print stream "
"explicitly aborted.");
break;
}
}
/* Warn of read errors */
if (length < 0)
guac_client_log(job->client, GUAC_LOG_ERROR,
"Error reading from filter: %s", strerror(errno));
/* Terminate stream */
guac_client_for_user(job->client, job->user,
guac_rdpdr_print_job_end_stream, job);
/* Ensure all associated file descriptors are closed */
close(job->input_fd);
close(job->output_fd);
guac_client_log(job->client, GUAC_LOG_DEBUG, "Print job completed.");
return NULL;
}
void* guac_rdpdr_print_job_alloc(guac_user* user, void* data) {
/* Allocate nothing if user does not exist */
if (user == NULL)
return NULL;
/* Allocate stream for print job output */
guac_stream* stream = guac_user_alloc_stream(user);
if (stream == NULL)
return NULL;
/* Bail early if allocation fails */
guac_rdpdr_print_job* job = malloc(sizeof(guac_rdpdr_print_job));
if (job == NULL)
return NULL;
/* Associate job with stream and dependent data */
job->client = user->client;
job->user = user;
job->stream = stream;
job->bytes_received = 0;
/* Set default filename for job */
strcpy(job->filename, GUAC_RDPDR_PRINT_JOB_DEFAULT_FILENAME);
/* Prepare stream for receipt of acks */
stream->ack_handler = guac_rdpdr_print_filter_ack_handler;
stream->data = job;
/* Create print filter process */
job->filter_pid = guac_rdpdr_create_filter_process(job->client,
&job->input_fd, &job->output_fd);
/* Abort if print filter process cannot be created */
if (job->filter_pid == -1) {
guac_user_free_stream(user, stream);
free(job);
return NULL;
}
/* Init stream state signal and lock */
job->state = GUAC_RDPDR_PRINT_JOB_WAITING_FOR_ACK;
pthread_cond_init(&job->state_modified, NULL);
pthread_mutex_init(&job->state_lock, NULL);
/* Start output thread */
pthread_create(&job->output_thread, NULL,
guac_rdpdr_print_job_output_thread, job);
/* Print job allocated successfully */
return job;
}
int guac_rdpdr_print_job_write(guac_rdpdr_print_job* job,
void* buffer, int length) {
/* Create print job, if not yet created */
if (job->bytes_received == 0) {
char* filename = job->filename;
unsigned char* search = buffer;
int i;
/* Search for filename within buffer */
for (i=0; i<length-9 && i < 2048; i++) {
/* If title. use as filename */
if (memcmp(search, "%%Title: ", 9) == 0) {
/* Skip past "%%Title: " */
search += 9;
/* Copy as much of title as reasonable */
int j;
for (j=0; j < GUAC_RDPDR_PRINT_JOB_FILENAME_MAX_LENGTH - 5 /* extension + 1 */ && i<length; i++, j++) {
/* Get character, stop at EOL */
char c = *(search++);
if (c == '\r' || c == '\n')
break;
/* Copy to filename */
filename[j] = c;
}
/* Append filename with extension */
strcpy(&(filename[j]), ".pdf");
break;
}
/* Next character */
search++;
}
/* Begin print stream */
guac_client_for_user(job->client, job->user,
guac_rdpdr_print_job_begin_stream, job);
} /* end if print job beginning */
/* Update counter of bytes received */
job->bytes_received += length;
/* Write data to filter process */
return write(job->input_fd, buffer, length);
}
void guac_rdpdr_print_job_free(guac_rdpdr_print_job* job) {
/* No more input will be provided */
close(job->input_fd);
/* Wait for job to terminate */
pthread_join(job->output_thread, NULL);
/* Free base structure */
free(job);
}
void guac_rdpdr_print_job_kill(guac_rdpdr_print_job* job) {
/* Stop all handling of I/O */
close(job->input_fd);
close(job->output_fd);
/* Mark stream as closed */
guac_rdpdr_print_job_set_state(job, GUAC_RDPDR_PRINT_JOB_CLOSED);
}

View File

@ -0,0 +1,226 @@
/*
* 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.
*/
#ifndef GUAC_RDPDR_PRINT_JOB_H
#define GUAC_RDPDR_PRINT_JOB_H
#include "config.h"
#include <guacamole/client.h>
#include <guacamole/stream.h>
#include <guacamole/user.h>
#include <pthread.h>
#include <unistd.h>
/**
* The maximum number of bytes in the filename of an RDPDR print job sent as a
* file over the Guacamole protocol, including NULL terminator.
*/
#define GUAC_RDPDR_PRINT_JOB_FILENAME_MAX_LENGTH 1024
/**
* The default filename to use for the PDF output of an RDPDR print job if no
* document title can be found within the printed data.
*/
#define GUAC_RDPDR_PRINT_JOB_DEFAULT_FILENAME "guacamole-print.pdf"
/**
* The current state of an RDPDR print job.
*/
typedef enum guac_rdpdr_print_job_state {
/**
* The print stream has been opened with the Guacamole client, but the
* client has not yet confirmed that it is ready to receive data.
*/
GUAC_RDPDR_PRINT_JOB_WAITING_FOR_ACK,
/**
* The print stream has been opened with the Guacamole client, and the
* client has responded with an "ack", confirming that it is ready to
* receive data (or that data has been received and it is ready to receive
* more).
*/
GUAC_RDPDR_PRINT_JOB_ACK_RECEIVED,
/**
* The print stream has been closed or the printer is terminating, and no
* further data should be sent to the client.
*/
GUAC_RDPDR_PRINT_JOB_CLOSED
} guac_rdpdr_print_job_state;
/**
* Data specific to an instance of the printer device.
*/
typedef struct guac_rdpdr_print_job {
guac_client* client;
/**
* The user receiving the output from the print job.
*/
guac_user* user;
/**
* The stream along which the print job output should be sent.
*/
guac_stream* stream;
/**
* The PID of the print filter process converting PostScript data into PDF.
*/
pid_t filter_pid;
/**
* The filename that should be used when the converted PDF output is
* streamed to the Guacamole user. This value will be automatically
* determined based on the contents of the printed document.
*/
char filename[GUAC_RDPDR_PRINT_JOB_FILENAME_MAX_LENGTH];
/**
* File descriptor that should be written to when sending documents to the
* printer.
*/
int input_fd;
/**
* File descriptor that should be read from when receiving output from the
* printer.
*/
int output_fd;
/**
* The current state of the print stream, dependent on whether the client
* has acknowledged creation of the stream, whether the client has
* acknowledged receipt of data along the steam, and whether the print
* stream itself has closed.
*/
guac_rdpdr_print_job_state state;
/**
* Lock which is acquired prior to modifying the state property or waiting
* on the state_modified conditional.
*/
pthread_mutex_t state_lock;
/**
* Conditional which signals modification to the state property of this
* structure.
*/
pthread_cond_t state_modified;
/**
* Thread which transfers data from the printer to the Guacamole client.
*/
pthread_t output_thread;
/**
* The number of bytes received in the current print job.
*/
int bytes_received;
} guac_rdpdr_print_job;
/**
* A blob of print data being sent to the Guacamole user.
*/
typedef struct guac_rdpdr_print_blob {
/**
* The print job which generated the data being sent.
*/
guac_rdpdr_print_job* job;
/**
* The data being sent.
*/
void* buffer;
/**
* The number of bytes of data being sent.
*/
int length;
} guac_rdpdr_print_blob;
/**
* Allocates a new print job for the given user. It is expected that this
* function will be invoked via a call to guac_client_for_user() or
* guac_client_for_owner().
*
* @param user
* The user that should receive the output from the print job.
*
* @param data
* An arbitrary data parameter required by guac_client_for_user() and
* guac_client_for_owner() but ignored by this function. This should
* always be NULL.
*
* @return
* A pointer to a newly-allocated guac_rdpdr_print_job, or NULL if the
* print job could not be created.
*/
void* guac_rdpdr_print_job_alloc(guac_user* user, void* data);
/**
* Writes PostScript print data to the given active print job. The print job
* will automatically convert this data to PDF, streaming the result to the
* Guacamole user associated with the print job. This function may block if
* the print job is not yet ready for more data.
*
* @param buffer
* The PostScript print data to write to the given print job.
*
* @param length
* The number of bytes of PostScript print data to write.
*
* @return
* The number of bytes written, or -1 if an error occurs which prevents
* further writes.
*/
int guac_rdpdr_print_job_write(guac_rdpdr_print_job* job,
void* buffer, int length);
/**
* Frees the memory associated with the given print job, closing all underlying
* file descriptors, and ending the file transfer to the associated Guacamole
* user. This function may block if the print filter process has not yet
* finished processing the received data.
*
* @param job
* The print job to free.
*/
void guac_rdpdr_print_job_free(guac_rdpdr_print_job* job);
/**
* Forcibly kills the given print job, stopping all associated processing and
* streaming. The memory associated with the print job will still need to be
* reclaimed via guac_rdpdr_print_job_free().
*
* @param job
* The print job to kill.
*/
void guac_rdpdr_print_job_kill(guac_rdpdr_print_job* job);
#endif

View File

@ -21,6 +21,7 @@
#include "rdpdr_messages.h"
#include "rdpdr_printer.h"
#include "rdpdr_print_job.h"
#include "rdpdr_service.h"
#include "rdp_status.h"
@ -38,330 +39,60 @@
#endif
#include <errno.h>
#include <pthread.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
/* Command to run GhostScript safely as a filter writing PDF */
char* const guac_rdpdr_pdf_filter_command[] = {
"gs",
"-q",
"-dNOPAUSE",
"-dBATCH",
"-dSAFER",
"-dPARANOIDSAFER",
"-sDEVICE=pdfwrite",
"-sOutputFile=-",
"-c",
".setpdfwrite",
"-sstdout=/dev/null",
"-f",
"-",
NULL
};
/**
* Handler for "ack" messages received in response to printed data. Additional
* data will be sent as a result or, if no data remains, the stream will be
* terminated. It is required that the data pointer of the provided stream be
* set to the file descriptor from which the printed data should be read.
*
* @param user
* The user to whom the printed data is being sent.
*
* @param stream
* The stream along which the printed data is to be sent. The data pointer
* of this stream MUST be set to the file descriptor from which the data
* being sent is to be read.
*
* @param message
* An arbitrary, human-readable message describing the success/failure of
* the operation being acknowledged (either stream creation or receipt of
* a blob).
*
* @param status
* The status code describing the success/failure of the operation being
* acknowledged (either stream creation or receipt of a blob).
*
* @return
* Always zero.
*/
static int guac_rdpdr_print_filter_ack_handler(guac_user* user,
guac_stream* stream, char* message, guac_protocol_status status) {
char buffer[6048];
/* Pull file descriptor from stream data */
int fd = (intptr_t) stream->data;
/* Reading only if ack reports success */
if (status == GUAC_PROTOCOL_STATUS_SUCCESS) {
/* Write a single blob of output */
int length = read(fd, buffer, sizeof(buffer));
if (length > 0) {
guac_protocol_send_blob(user->socket, stream, buffer, length);
guac_socket_flush(user->socket);
return 0;
}
/* Warn of read errors, fall through to termination */
else if (length < 0)
guac_user_log(user, GUAC_LOG_ERROR,
"Error reading from filter: %s", strerror(errno));
}
/* Note if stream aborted by user, fall through to termination */
else
guac_user_log(user, GUAC_LOG_INFO, "Print stream aborted.");
/* Explicitly close down stream */
guac_protocol_send_end(user->socket, stream);
guac_socket_flush(user->socket);
/* Clean up our end of the stream */
guac_user_free_stream(user, stream);
close(fd);
return 0;
}
static int guac_rdpdr_create_print_process(guac_rdpdr_device* device) {
guac_rdpdr_printer_data* printer_data = (guac_rdpdr_printer_data*) device->data;
int child_pid;
int stdin_pipe[2];
int stdout_pipe[2];
/* Create STDIN pipe */
if (pipe(stdin_pipe)) {
guac_client_log(device->rdpdr->client, GUAC_LOG_ERROR,
"Unable to create STDIN pipe for PDF filter process: %s", strerror(errno));
return 1;
}
/* Create STDOUT pipe */
if (pipe(stdout_pipe)) {
guac_client_log(device->rdpdr->client, GUAC_LOG_ERROR,
"Unable to create STDIN pipe for PDF filter process: %s", strerror(errno));
close(stdin_pipe[0]);
close(stdin_pipe[1]);
return 1;
}
/* Store our side of stdin/stdout */
printer_data->printer_input = stdin_pipe[1];
printer_data->printer_output = stdout_pipe[0];
/* Fork child process */
child_pid = fork();
/* Log fork errors */
if (child_pid == -1) {
guac_client_log(device->rdpdr->client, GUAC_LOG_ERROR,
"Unable to fork PDF filter process: %s", strerror(errno));
close(stdin_pipe[0]);
close(stdin_pipe[1]);
close(stdout_pipe[0]);
close(stdout_pipe[1]);
return 1;
}
/* Child process */
if (child_pid == 0) {
/* Close unneeded ends of pipe */
close(stdin_pipe[1]);
close(stdout_pipe[0]);
/* Reassign file descriptors as STDIN/STDOUT */
dup2(stdin_pipe[0], STDIN_FILENO);
dup2(stdout_pipe[1], STDOUT_FILENO);
/* Run PDF filter */
guac_client_log(device->rdpdr->client, GUAC_LOG_INFO, "Running %s", guac_rdpdr_pdf_filter_command[0]);
if (execvp(guac_rdpdr_pdf_filter_command[0], guac_rdpdr_pdf_filter_command) < 0)
guac_client_log(device->rdpdr->client, GUAC_LOG_ERROR, "Unable to execute PDF filter command: %s", strerror(errno));
else
guac_client_log(device->rdpdr->client, GUAC_LOG_ERROR, "Unable to execute PDF filter command, but no error given");
/* Terminate child process */
exit(1);
}
/* Log fork success */
guac_client_log(device->rdpdr->client, GUAC_LOG_INFO, "Created PDF filter process PID=%i", child_pid);
/* Close unneeded ends of pipe */
close(stdin_pipe[0]);
close(stdout_pipe[1]);
return 0;
}
void guac_rdpdr_process_print_job_create(guac_rdpdr_device* device,
wStream* input_stream, int completion_id) {
guac_rdpdr_printer_data* printer_data =
(guac_rdpdr_printer_data*) device->data;
/* Log creation of print job */
guac_client_log(device->rdpdr->client, GUAC_LOG_INFO, "Print job created");
/* Create print job */
device->data = guac_client_for_owner(device->rdpdr->client,
guac_rdpdr_print_job_alloc, NULL);
/* Respond with success */
wStream* output_stream = guac_rdpdr_new_io_completion(device,
completion_id, STATUS_SUCCESS, 4);
/* No bytes received yet */
printer_data->bytes_received = 0;
Stream_Write_UINT32(output_stream, 0); /* fileId */
svc_plugin_send((rdpSvcPlugin*) device->rdpdr, output_stream);
}
/**
* Given data representing a print device with a pending pring job, allocates a
* new stream for the given user, associating it with the provided data and
* returning the resulting guac_stream. The stream will be pre-configured to
* send blobs of print data in response to "ack" messages received from the
* user. If the given user is NULL, no stream will be allocated, and the
* print job will be immediately aborted.
*
* @param user
* The user to whom the print job is being sent, or NULL if no stream
* should be allocated.
*
* @param data
* A pointer to the guac_rdpdr_device instance associated with the new
* print job.
*
* @return
* The guac_stream allocated for the new print job, or NULL if no stream
* could be allocated.
*/
static void* guac_rdpdr_alloc_printer_stream(guac_user* owner, void* data) {
guac_rdpdr_device* device = (guac_rdpdr_device*) data;
guac_rdpdr_printer_data* printer_data =
(guac_rdpdr_printer_data*) device->data;
/* Abort immediately if there is no owner */
if (owner == NULL) {
close(printer_data->printer_output);
close(printer_data->printer_input);
printer_data->printer_output = -1;
printer_data->printer_input = -1;
return NULL;
}
/* Allocate stream for owner */
guac_stream* stream = guac_user_alloc_stream(owner);
stream->ack_handler = guac_rdpdr_print_filter_ack_handler;
stream->data = (void*) (intptr_t) printer_data->printer_output;
return stream;
}
void guac_rdpdr_process_print_job_write(guac_rdpdr_device* device,
wStream* input_stream, int completion_id) {
guac_rdpdr_printer_data* printer_data = (guac_rdpdr_printer_data*) device->data;
int status=0, length;
guac_rdpdr_print_job* job = (guac_rdpdr_print_job*) device->data;
unsigned char* buffer;
int length;
int status;
wStream* output_stream;
/* Read buffer of print data */
Stream_Read_UINT32(input_stream, length);
Stream_Seek(input_stream, 8); /* Offset */
Stream_Seek(input_stream, 20); /* Padding */
buffer = Stream_Pointer(input_stream);
/* Create print job, if not yet created */
if (printer_data->bytes_received == 0) {
char filename[1024] = "guacamole-print.pdf";
unsigned char* search = buffer;
int i;
/* Search for filename within buffer */
for (i=0; i<length-9 && i < 2048; i++) {
/* If title. use as filename */
if (memcmp(search, "%%Title: ", 9) == 0) {
/* Skip past "%%Title: " */
search += 9;
/* Copy as much of title as reasonable */
int j;
for (j=0; j<sizeof(filename) - 5 /* extension + 1 */ && i<length; i++, j++) {
/* Get character, stop at EOL */
char c = *(search++);
if (c == '\r' || c == '\n')
break;
/* Copy to filename */
filename[j] = c;
}
/* Append filename with extension */
strcpy(&(filename[j]), ".pdf");
break;
}
/* Next character */
search++;
}
/* Start print process */
if (guac_rdpdr_create_print_process(device) != 0) {
status = STATUS_DEVICE_OFF_LINE;
length = 0;
}
/* If print started successfully, create outbound stream */
else {
guac_client_log(device->rdpdr->client, GUAC_LOG_INFO,
"Print job created");
/* Allocate stream */
guac_stream* stream = (guac_stream*) guac_client_for_owner(
device->rdpdr->client, guac_rdpdr_alloc_printer_stream,
device);
/* Begin file if stream allocation was successful */
if (stream != NULL)
guac_protocol_send_file(device->rdpdr->client->socket, stream,
"application/pdf", filename);
}
} /* end if print job beginning */
printer_data->bytes_received += length;
/* If not yet failed, write received data */
if (status == 0) {
/* Write data to printer, translate output for RDP */
length = write(printer_data->printer_input, buffer, length);
if (length == -1) {
guac_client_log(device->rdpdr->client, GUAC_LOG_ERROR, "Error writing to printer: %s", strerror(errno));
status = STATUS_DEVICE_OFF_LINE;
length = 0;
}
/* Write data only if job exists, translating status for RDP */
if (job != NULL && (length = guac_rdpdr_print_job_write(job,
buffer, length)) >= 0) {
status = STATUS_SUCCESS;
}
output_stream = guac_rdpdr_new_io_completion(device, completion_id,
status, 5);
/* Report device offline if write fails */
else {
status = STATUS_DEVICE_OFF_LINE;
length = 0;
}
wStream* output_stream = guac_rdpdr_new_io_completion(device,
completion_id, status, 5);
Stream_Write_UINT32(output_stream, length);
Stream_Write_UINT8(output_stream, 0); /* Padding */
@ -373,21 +104,22 @@ void guac_rdpdr_process_print_job_write(guac_rdpdr_device* device,
void guac_rdpdr_process_print_job_close(guac_rdpdr_device* device,
wStream* input_stream, int completion_id) {
guac_rdpdr_printer_data* printer_data =
(guac_rdpdr_printer_data*) device->data;
/* End print job */
guac_rdpdr_print_job* job = (guac_rdpdr_print_job*) device->data;
if (job != NULL) {
guac_rdpdr_print_job_free(job);
device->data = NULL;
}
wStream* output_stream = guac_rdpdr_new_io_completion(device,
completion_id, STATUS_SUCCESS, 4);
Stream_Write_UINT32(output_stream, 0); /* Padding */
/* Close input - the Guacamole stream will continue while output remains */
close(printer_data->printer_input);
printer_data->printer_input = -1;
guac_client_log(device->rdpdr->client, GUAC_LOG_INFO, "Print job closed");
svc_plugin_send((rdpSvcPlugin*) device->rdpdr, output_stream);
/* Log end of print job */
guac_client_log(device->rdpdr->client, GUAC_LOG_INFO, "Print job closed");
}
static void guac_rdpdr_device_printer_announce_handler(guac_rdpdr_device* device,
@ -447,19 +179,10 @@ static void guac_rdpdr_device_printer_iorequest_handler(guac_rdpdr_device* devic
static void guac_rdpdr_device_printer_free_handler(guac_rdpdr_device* device) {
guac_rdpdr_printer_data* printer_data =
(guac_rdpdr_printer_data*) device->data;
/* Close print job input (STDIN for filter process) if open */
if (printer_data->printer_input != -1)
close(printer_data->printer_input);
/* Close print job output (STDOUT for filter process) if open */
if (printer_data->printer_output != -1)
close(printer_data->printer_output);
/* Free underlying data */
free(device->data);
/* Close print job if open */
guac_rdpdr_print_job* job = (guac_rdpdr_print_job*) device->data;
if (job != NULL)
guac_rdpdr_print_job_free(job);
}
@ -469,7 +192,6 @@ void guac_rdpdr_register_printer(guac_rdpdrPlugin* rdpdr) {
/* Get new device */
guac_rdpdr_device* device = &(rdpdr->devices[id]);
guac_rdpdr_printer_data* printer_data;
/* Init device */
device->rdpdr = rdpdr;
@ -481,11 +203,8 @@ void guac_rdpdr_register_printer(guac_rdpdrPlugin* rdpdr) {
device->iorequest_handler = guac_rdpdr_device_printer_iorequest_handler;
device->free_handler = guac_rdpdr_device_printer_free_handler;
/* Init data */
printer_data = malloc(sizeof(guac_rdpdr_printer_data));
printer_data->printer_input = -1;
printer_data->printer_output = -1;
device->data = printer_data;
/* No active print job yet */
device->data = NULL;
}

View File

@ -31,42 +31,11 @@
#include "compat/winpr-stream.h"
#endif
/**
* Data specific to an instance of the printer device.
*/
typedef struct guac_rdpdr_printer_data {
/**
* File descriptor that should be written to when sending documents to the
* printer. If no print job is in progress, this will be -1.
*/
int printer_input;
/**
* File descriptor that should be read from when receiving output from the
* printer. If no print job is in progress, this will be -1.
*/
int printer_output;
/**
* The number of bytes received in the current print job.
*/
int bytes_received;
} guac_rdpdr_printer_data;
/**
* Registers a new printer device within the RDPDR plugin. This must be done
* before RDPDR connection finishes.
*/
void guac_rdpdr_register_printer(guac_rdpdrPlugin* rdpdr);
/**
* The command to run when filtering postscript to produce PDF. This must be
* a NULL-terminated array of arguments, where the first argument is the name
* of the file to run.
*/
extern char* const guac_rdpdr_pdf_filter_command[];
#endif