|
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 |
}
|