Hi, On Fri, May 27, 2016 at 10:54:15AM +0200, Pavel Grunt wrote: > Hi Victor, > > On Mon, 2016-05-23 at 13:50 +0200, Victor Toso wrote: > > By introducing a flush_callback such as SpiceFileTransferTaskFlushCb > > SpiceFileTransferTask becomes agnostic on how channel-main flushes > > the data. > > The spice_file_transfer_task_flush_done() function is now introduced > > to tell SpiceFileTransferTask that flushing is over and we can read > > more data if no error has happened. > > > > Helper functions spice_file_transfer_task_get_channel() and > > spice_file_transfer_task_get_cancellable() were created to retrieve > > properties from SpiceFileTransferTask without the need to access its > > private structure. > > > > Can it be done in more steps please - _get_channel() replacing the > access to the private structure in a separate patch ? Sure! > > > > This change is related to split SpiceFileTransferTask from > > channel-main. > > --- > > src/channel-main.c | 157 +++++++++++++++++++++++++++++++++++----------------- > > - > > 1 file changed, 105 insertions(+), 52 deletions(-) > > > > diff --git a/src/channel-main.c b/src/channel-main.c > > index 4fb514e..f4e8c2b 100644 > > --- a/src/channel-main.c > > +++ b/src/channel-main.c > > @@ -77,6 +77,16 @@ G_DEFINE_TYPE(SpiceFileTransferTask, > > spice_file_transfer_task, G_TYPE_OBJECT) > > (G_TYPE_INSTANCE_GET_PRIVATE((o), SPICE_TYPE_FILE_TRANSFER_TASK, > > SpiceFileTransferTaskPrivate)) > > > > #define FILE_XFER_CHUNK_SIZE (VD_AGENT_MAX_DATA_SIZE * 32) > > + > > +typedef void (*SpiceFileTransferTaskFlushCb)(SpiceFileTransferTask > > *xfer_task, > > + void *buffer, > > + gssize count, > > + gpointer user_data); > > + > > +static SpiceMainChannel > > *spice_file_transfer_task_get_channel(SpiceFileTransferTask *self); > > +static GCancellable > > *spice_file_transfer_task_get_cancellable(SpiceFileTransferTask *self); > > +static void spice_file_transfer_task_flush_done(SpiceFileTransferTask *self, > > GError *error); > > + > > struct _SpiceFileTransferTaskPrivate > > > > /* private */ > > @@ -90,6 +100,8 @@ struct _SpiceFileTransferTaskPrivate > > GCancellable *cancellable; > > GFileProgressCallback progress_callback; > > gpointer progress_callback_data; > > + SpiceFileTransferTaskFlushCb flush_callback; > > + gpointer flush_callback_data; > > GAsyncReadyCallback callback; > > gpointer user_data; > > char *buffer; > > @@ -1851,70 +1863,46 @@ static void file_xfer_data_flushed_cb(GObject > > *source_object, > > SpiceMainChannel *channel = (SpiceMainChannel *)source_object; > > GError *error = NULL; > > > > - self->priv->pending = FALSE; > > file_xfer_flush_finish(channel, res, &error); > > - if (error || self->priv->error) { > > - spice_file_transfer_task_completed(self, error); > > - return; > > - } > > - > > - if (spice_util_get_debug()) { > > - const GTimeSpan interval = 20 * G_TIME_SPAN_SECOND; > > - gint64 now = g_get_monotonic_time(); > > - > > - if (interval < now - self->priv->last_update) { > > - gchar *basename = g_file_get_basename(self->priv->file); > > - self->priv->last_update = now; > > - SPICE_DEBUG("transferred %.2f%% of the file %s", > > - 100.0 * self->priv->read_bytes / self->priv- > > >file_size, basename); > > - g_free(basename); > > - } > > - } > > - > > - if (self->priv->progress_callback) { > > - goffset read = 0; > > - goffset total = 0; > > - SpiceMainChannel *main_channel = self->priv->channel; > > - GHashTableIter iter; > > - gpointer key, value; > > - > > - /* since the progress_callback does not have a parameter to indicate > > - * which file the progress is associated with, report progress on all > > - * current transfers */ > > - g_hash_table_iter_init(&iter, main_channel->priv->file_xfer_tasks); > > - while (g_hash_table_iter_next(&iter, &key, &value)) { > > - SpiceFileTransferTask *t = (SpiceFileTransferTask *)value; > > - read += t->priv->read_bytes; > > - total += t->priv->file_size; > > - } > > - > > - self->priv->progress_callback(read, total, self->priv- > > >progress_callback_data); > > - } > > - > > - /* Read more data */ > > - file_xfer_continue_read(self); > > + spice_file_transfer_task_flush_done(self, error); > > } > > > > -static void file_xfer_queue(SpiceFileTransferTask *self, int data_size) > > +static void file_xfer_queue(SpiceFileTransferTask *xfer_task, void *buffer, > > int data_size) > > { > > VDAgentFileXferDataMessage msg; > > - SpiceMainChannel *channel = SPICE_MAIN_CHANNEL(self->priv->channel); > > + SpiceMainChannel *channel = > > spice_file_transfer_task_get_channel(xfer_task); > > what if the channel is NULL (not possible now?), but it is worth > checking Right, this was not considered before so I did not really think about it. In case channel could be NULL due any kind of problem, the transfer-task should be cancelled and no flush should really happen. I'll include the check. > > > > - msg.id = self->priv->id; > > + g_object_get(xfer_task, "id", &msg.id, NULL); > > msg.size = data_size; > > agent_msg_queue_many(channel, VD_AGENT_FILE_XFER_DATA, > > &msg, sizeof(msg), > > - self->priv->buffer, data_size, NULL); > > + buffer, data_size, NULL); > > spice_channel_wakeup(SPICE_CHANNEL(channel), FALSE); > > } > > > > +static void file_xfer_flush_callback(SpiceFileTransferTask *xfer_task, > > + void *buffer, > > + gssize count, > > + gpointer user_data) > > +{ > > + SpiceMainChannel *main_channel; > > + GCancellable *cancellable; > > + > > + file_xfer_queue(xfer_task, buffer, count); > > + if (count == 0) > > + return; > > + > > + main_channel = spice_file_transfer_task_get_channel(xfer_task); > > + cancellable = spice_file_transfer_task_get_cancellable(xfer_task); > > + file_xfer_flush_async(main_channel, cancellable, > > file_xfer_data_flushed_cb, xfer_task); > > +} > > + > > /* main context */ > > static void file_xfer_read_cb(GObject *source_object, > > GAsyncResult *res, > > gpointer user_data) > > { > > SpiceFileTransferTask *self = user_data; > > - SpiceMainChannel *channel = self->priv->channel; > > gssize count; > > GError *error = NULL; > > > > @@ -1930,12 +1918,11 @@ static void file_xfer_read_cb(GObject *source_object, > > if (count > 0 || self->priv->file_size == 0) { > > self->priv->read_bytes += count; > > g_object_notify(G_OBJECT(self), "progress"); > > - file_xfer_queue(self, count); > > - if (count == 0) > > - return; > > - file_xfer_flush_async(channel, self->priv->cancellable, > > - file_xfer_data_flushed_cb, self); > > - self->priv->pending = TRUE; > > + > > + if (self->priv->flush_callback) { > > + self->priv->pending = TRUE; > > + self->priv->flush_callback(self, self->priv->buffer, count, self- > > >priv->flush_callback_data); > > + } > > } else if (error) { > > spice_channel_wakeup(SPICE_CHANNEL(self->priv->channel), FALSE); > > spice_file_transfer_task_completed(self, error); > > @@ -3083,6 +3070,8 @@ static void > > file_xfer_send_start_msg_async(SpiceMainChannel *channel, > > GCancellable *cancellable, > > GFileProgressCallback > > progress_callback, > > gpointer progress_callback_data, > > + SpiceFileTransferTaskFlushCb > > flush_callback, > > + gpointer flush_callback_data, > > GAsyncReadyCallback callback, > > gpointer user_data) > > { > > @@ -3102,6 +3091,8 @@ static void > > file_xfer_send_start_msg_async(SpiceMainChannel *channel, > > task->priv->flags = flags; > > task->priv->progress_callback = progress_callback; > > task->priv->progress_callback_data = progress_callback_data; > > + task->priv->flush_callback = flush_callback; > > + task->priv->flush_callback_data = flush_callback_data; > > task->priv->callback = callback; > > task->priv->user_data = user_data; > > > > @@ -3191,6 +3182,8 @@ void spice_main_file_copy_async(SpiceMainChannel > > *channel, > > cancellable, > > progress_callback, > > progress_callback_data, > > + file_xfer_flush_callback, > > + NULL, > > callback, > > user_data); > > } > > @@ -3218,6 +3211,66 @@ gboolean spice_main_file_copy_finish(SpiceMainChannel > > *channel, > > return g_task_propagate_boolean(task, error); > > } > > > > +static SpiceMainChannel > > *spice_file_transfer_task_get_channel(SpiceFileTransferTask *self) > > +{ > > + g_assert_nonnull(self); > please, don't assert because of transfer task, use g_return Although the discussion was more server-side, I would agree that internal api could assert in the situation above in spice-gtk too. But I can change it if you don't agree. > > + return self->priv->channel; > > +} > > + > > +static GCancellable > > *spice_file_transfer_task_get_cancellable(SpiceFileTransferTask *self) > > +{ > > + g_assert_nonnull(self); > > + return self->priv->cancellable; > > +} > > + > > +static void spice_file_transfer_task_flush_done(SpiceFileTransferTask *self, > > GError *error) > > +{ > > + g_assert_nonnull(self); > > + g_return_if_fail(self->priv->pending); > > + > > + self->priv->pending = FALSE; > > + > > + if (error || self->priv->error) { > > + spice_file_transfer_task_completed(self, error); > > + return; > > + } > > + > > + if (spice_util_get_debug()) { > > + const GTimeSpan interval = 20 * G_TIME_SPAN_SECOND; > > + gint64 now = g_get_monotonic_time(); > > + > > + if (interval < now - self->priv->last_update) { > > + gchar *basename = g_file_get_basename(self->priv->file); > > + self->priv->last_update = now; > > + SPICE_DEBUG("transferred %.2f%% of the file %s", > > + 100.0 * self->priv->read_bytes / self->priv- > > >file_size, basename); > > + g_free(basename); > > + } > > + } > > + > > + if (self->priv->progress_callback) { > > + goffset read = 0; > > + goffset total = 0; > > + SpiceMainChannel *main_channel = self->priv->channel; > > + GHashTableIter iter; > > + gpointer key, value; > > + > > + /* since the progress_callback does not have a parameter to indicate > > + * which file the progress is associated with, report progress on all > > + * current transfers */ > > + g_hash_table_iter_init(&iter, main_channel->priv->file_xfer_tasks); > > + while (g_hash_table_iter_next(&iter, &key, &value)) { > > + SpiceFileTransferTask *t = (SpiceFileTransferTask *)value; > > + read += t->priv->read_bytes; > > + total += t->priv->file_size; > > + } > > + > > + self->priv->progress_callback(read, total, self->priv- > > >progress_callback_data); > > + } > > + > > + /* Read more data */ > > + file_xfer_continue_read(self); > > +} > > > > > > static void > > Reviewed-by: Pavel Grunt <pgrunt@xxxxxxxxxx> Thanks! toso _______________________________________________ Spice-devel mailing list Spice-devel@xxxxxxxxxxxxxxxxxxxxx https://lists.freedesktop.org/mailman/listinfo/spice-devel