spice/server/red-pipe-item.cpp
Frediano Ziglio 45e964dc5a red-pipe-item: Better encapsulate marshaller_unref_pipe_item
To avoid memory errors marshaller_unref_pipe_item should be
matched with a previous call to red_pipe_item_ref.
This is correctly done but to reduce the possibility of breaking
this rule move both referencing and unreferencing into a new
RedPipeItem::add_to_marshaller method.

Signed-off-by: Frediano Ziglio <freddy77@gmail.com>
Acked-by: Julien Ropé <jrope@gmail.com>
2020-07-13 19:47:45 +01:00

62 lines
1.8 KiB
C++

/* -*- Mode: C; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/*
Copyright (C) 2016 Red Hat, Inc.
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with this library; if not, see <http://www.gnu.org/licenses/>.
*/
#include <config.h>
#include "red-channel.h"
#include "red-pipe-item.h"
RedPipeItem *red_pipe_item_ref(RedPipeItem *item)
{
g_return_val_if_fail(item->refcount > 0, NULL);
g_atomic_int_inc(&item->refcount);
return item;
}
void red_pipe_item_unref(RedPipeItem *item)
{
g_return_if_fail(item->refcount > 0);
if (g_atomic_int_dec_and_test(&item->refcount)) {
item->free_func(item);
}
}
void red_pipe_item_init_full(RedPipeItem *item,
gint type,
red_pipe_item_free_t *free_func)
{
item->type = type;
item->refcount = 1;
item->free_func = free_func ? free_func : (red_pipe_item_free_t *)g_free;
}
static void marshaller_unref_pipe_item(uint8_t *, void *opaque)
{
RedPipeItem *item = (RedPipeItem*) opaque;
red_pipe_item_unref(item);
}
void RedPipeItem::add_to_marshaller(SpiceMarshaller *m, uint8_t *data, size_t size)
{
red_pipe_item_ref(this);
spice_marshaller_add_by_ref_full(m, data, size,
marshaller_unref_pipe_item, this);
}