Blame libvirt-gobject/libvirt-gobject-output-stream.c

Packit a07778
/*
Packit a07778
 * libvirt-gobject-output-stream.h: libvirt gobject integration
Packit a07778
 *
Packit a07778
 * Copyright (C) 2011 Red Hat, Inc.
Packit a07778
 *
Packit a07778
 * This library is free software; you can redistribute it and/or
Packit a07778
 * modify it under the terms of the GNU Lesser General Public
Packit a07778
 * License as published by the Free Software Foundation; either
Packit a07778
 * version 2.1 of the License, or (at your option) any later version.
Packit a07778
 *
Packit a07778
 * This library is distributed in the hope that it will be useful,
Packit a07778
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
Packit a07778
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Packit a07778
 * Lesser General Public License for more details.
Packit a07778
 *
Packit a07778
 * You should have received a copy of the GNU Lesser General Public
Packit a07778
 * License along with this library. If not, see
Packit a07778
 * <http://www.gnu.org/licenses/>.
Packit a07778
 *
Packit a07778
 * Authors: Daniel P. Berrange <berrange@redhat.com>
Packit a07778
 *          Marc-André Lureau <marcandre.lureau@redhat.com>
Packit a07778
 */
Packit a07778
Packit a07778
#include <config.h>
Packit a07778
Packit a07778
#include <libvirt/virterror.h>
Packit a07778
#include <string.h>
Packit a07778
Packit a07778
#include "libvirt-glib/libvirt-glib.h"
Packit a07778
#include "libvirt-gobject/libvirt-gobject.h"
Packit a07778
#include "libvirt-gobject-output-stream.h"
Packit a07778
#include "libvirt-gobject-compat.h"
Packit a07778
Packit a07778
enum
Packit a07778
{
Packit a07778
    PROP_0,
Packit a07778
    PROP_STREAM
Packit a07778
};
Packit a07778
Packit a07778
struct _GVirOutputStreamPrivate
Packit a07778
{
Packit a07778
    GVirStream *stream;
Packit a07778
Packit a07778
    /* pending operation metadata */
Packit a07778
    GTask *task;
Packit a07778
    const void * buffer;
Packit a07778
    gsize count;
Packit a07778
};
Packit a07778
Packit a07778
#define gvir_output_stream_get_type _gvir_output_stream_get_type
Packit a07778
G_DEFINE_TYPE_WITH_PRIVATE(GVirOutputStream, gvir_output_stream, G_TYPE_OUTPUT_STREAM);
Packit a07778
Packit a07778
static void gvir_output_stream_get_property(GObject *object,
Packit a07778
                                            guint prop_id,
Packit a07778
                                            GValue *value,
Packit a07778
                                            GParamSpec *pspec)
Packit a07778
{
Packit a07778
    GVirOutputStream *stream = GVIR_OUTPUT_STREAM(object);
Packit a07778
Packit a07778
    switch (prop_id) {
Packit a07778
    case PROP_STREAM:
Packit a07778
        g_value_set_object(value, stream->priv->stream);
Packit a07778
        break;
Packit a07778
Packit a07778
    default:
Packit a07778
        G_OBJECT_WARN_INVALID_PROPERTY_ID(object, prop_id, pspec);
Packit a07778
    }
Packit a07778
}
Packit a07778
Packit a07778
static void gvir_output_stream_set_property(GObject *object,
Packit a07778
                                            guint prop_id,
Packit a07778
                                            const GValue *value,
Packit a07778
                                            GParamSpec *pspec)
Packit a07778
{
Packit a07778
    GVirOutputStream *stream = GVIR_OUTPUT_STREAM(object);
Packit a07778
Packit a07778
    switch (prop_id) {
Packit a07778
    case PROP_STREAM:
Packit a07778
        stream->priv->stream = g_value_get_object(value);
Packit a07778
        break;
Packit a07778
Packit a07778
    default:
Packit a07778
        G_OBJECT_WARN_INVALID_PROPERTY_ID(object, prop_id, pspec);
Packit a07778
    }
Packit a07778
}
Packit a07778
Packit a07778
static void gvir_output_stream_finalize(GObject *object)
Packit a07778
{
Packit a07778
    GVirOutputStream *stream = GVIR_OUTPUT_STREAM(object);
Packit a07778
Packit a07778
    stream->priv->stream = NULL; // unowned
Packit a07778
Packit a07778
    if (G_OBJECT_CLASS(gvir_output_stream_parent_class)->finalize)
Packit a07778
        (*G_OBJECT_CLASS(gvir_output_stream_parent_class)->finalize)(object);
Packit a07778
}
Packit a07778
Packit a07778
static gboolean
Packit a07778
gvir_output_stream_write_ready(GVirStream *stream,
Packit a07778
                               GVirStreamIOCondition cond,
Packit a07778
                               void *opaque)
Packit a07778
{
Packit a07778
    GVirOutputStream *output_stream = GVIR_OUTPUT_STREAM(opaque);
Packit a07778
    GVirOutputStreamPrivate *priv = output_stream->priv;
Packit a07778
    GTask *task = priv->task;
Packit a07778
    GCancellable *cancellable = g_task_get_cancellable(task);
Packit a07778
    GError *error = NULL;
Packit a07778
    gssize result;
Packit a07778
Packit a07778
    if (!(cond & GVIR_STREAM_IO_CONDITION_WRITABLE)) {
Packit a07778
        g_warn_if_reached();
Packit a07778
        g_task_return_new_error(task,
Packit a07778
                                G_IO_ERROR,
Packit a07778
                                G_IO_ERROR_INVALID_ARGUMENT,
Packit a07778
                                "%s",
Packit a07778
                                "Expected stream to be readable");
Packit a07778
        goto cleanup;
Packit a07778
    }
Packit a07778
Packit a07778
    result  = gvir_stream_send(stream, priv->buffer, priv->count,
Packit a07778
                               cancellable, &error);
Packit a07778
Packit a07778
    if (error != NULL) {
Packit a07778
        if (g_error_matches(error, G_IO_ERROR, G_IO_ERROR_WOULD_BLOCK)) {
Packit a07778
            g_warn_if_reached();
Packit a07778
            g_task_return_new_error(task,
Packit a07778
                                    G_IO_ERROR,
Packit a07778
                                    G_IO_ERROR_INVALID_ARGUMENT,
Packit a07778
                                    "%s",
Packit a07778
                                    "Expected stream to be writable");
Packit a07778
            g_error_free (error);
Packit a07778
        } else {
Packit a07778
            g_task_return_error(task, error);
Packit a07778
        }
Packit a07778
Packit a07778
        goto cleanup;
Packit a07778
    }
Packit a07778
Packit a07778
    g_task_return_int(task, result);
Packit a07778
Packit a07778
cleanup:
Packit a07778
    priv->task = NULL;
Packit a07778
    g_object_unref(task);
Packit a07778
    return FALSE;
Packit a07778
}
Packit a07778
Packit a07778
static void gvir_output_stream_write_async(GOutputStream *stream,
Packit a07778
                                           const void *buffer,
Packit a07778
                                           gsize count,
Packit a07778
                                           int io_priority G_GNUC_UNUSED,
Packit a07778
                                           GCancellable *cancellable,
Packit a07778
                                           GAsyncReadyCallback callback,
Packit a07778
                                           gpointer user_data)
Packit a07778
{
Packit a07778
    GVirOutputStream *output_stream = GVIR_OUTPUT_STREAM(stream);
Packit a07778
Packit a07778
    g_return_if_fail(GVIR_IS_OUTPUT_STREAM(stream));
Packit a07778
    g_return_if_fail(output_stream->priv->task == NULL);
Packit a07778
Packit a07778
    gvir_stream_add_watch_full(output_stream->priv->stream,
Packit a07778
                               G_PRIORITY_DEFAULT,
Packit a07778
                               GVIR_STREAM_IO_CONDITION_WRITABLE,
Packit a07778
                               gvir_output_stream_write_ready,
Packit a07778
                               g_object_ref(stream),
Packit a07778
                               (GDestroyNotify)g_object_unref);
Packit a07778
Packit a07778
    output_stream->priv->task =
Packit a07778
        g_task_new(stream, cancellable, callback, user_data);
Packit a07778
    output_stream->priv->buffer = buffer;
Packit a07778
    output_stream->priv->count = count;
Packit a07778
}
Packit a07778
Packit a07778
Packit a07778
static gssize gvir_output_stream_write_finish(GOutputStream *stream,
Packit a07778
                                              GAsyncResult *result,
Packit a07778
                                              GError **error)
Packit a07778
{
Packit a07778
    GVirOutputStream *output_stream = GVIR_OUTPUT_STREAM(stream);
Packit a07778
    virStreamPtr handle;
Packit a07778
    gssize count;
Packit a07778
Packit a07778
    g_return_val_if_fail(GVIR_IS_OUTPUT_STREAM(stream), -1);
Packit a07778
    g_return_val_if_fail(g_task_is_valid(result, stream), -1);
Packit a07778
    g_return_val_if_fail(error == NULL || *error == NULL, -1);
Packit a07778
    g_object_get(output_stream->priv->stream, "handle", &handle, NULL);
Packit a07778
Packit a07778
    count = g_task_propagate_int(G_TASK(result), error);
Packit a07778
Packit a07778
    virStreamFree(handle);
Packit a07778
Packit a07778
    return count;
Packit a07778
}
Packit a07778
Packit a07778
Packit a07778
static void gvir_output_stream_class_init(GVirOutputStreamClass *klass)
Packit a07778
{
Packit a07778
    GObjectClass *gobject_class = G_OBJECT_CLASS(klass);
Packit a07778
    GOutputStreamClass *goutputstream_class = G_OUTPUT_STREAM_CLASS(klass);
Packit a07778
Packit a07778
    gobject_class->finalize = gvir_output_stream_finalize;
Packit a07778
    gobject_class->get_property = gvir_output_stream_get_property;
Packit a07778
    gobject_class->set_property = gvir_output_stream_set_property;
Packit a07778
Packit a07778
    goutputstream_class->write_fn = NULL;
Packit a07778
    goutputstream_class->write_async = gvir_output_stream_write_async;
Packit a07778
    goutputstream_class->write_finish = gvir_output_stream_write_finish;
Packit a07778
Packit a07778
    g_object_class_install_property(gobject_class, PROP_STREAM,
Packit a07778
                                    g_param_spec_object("stream",
Packit a07778
                                                        "stream",
Packit a07778
                                                        "GVirStream",
Packit a07778
                                                        GVIR_TYPE_STREAM, G_PARAM_CONSTRUCT_ONLY |
Packit a07778
                                                        G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
Packit a07778
}
Packit a07778
Packit a07778
static void gvir_output_stream_init(GVirOutputStream *stream)
Packit a07778
{
Packit a07778
    stream->priv = G_TYPE_INSTANCE_GET_PRIVATE(stream, GVIR_TYPE_OUTPUT_STREAM, GVirOutputStreamPrivate);
Packit a07778
}
Packit a07778
Packit a07778
GVirOutputStream* _gvir_output_stream_new(GVirStream *stream)
Packit a07778
{
Packit a07778
    return GVIR_OUTPUT_STREAM(g_object_new(GVIR_TYPE_OUTPUT_STREAM, "stream", stream, NULL));
Packit a07778
}