neatvnc/src/rcbuf.c

72 lines
1.6 KiB
C

/*
* Copyright (c) 2020 Andri Yngvason
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
* REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
* AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
* INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
* LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
* OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
* PERFORMANCE OF THIS SOFTWARE.
*/
#include "rcbuf.h"
#include <stdlib.h>
#include <string.h>
#include <assert.h>
struct rcbuf* rcbuf_new(void* payload, size_t size)
{
struct rcbuf* self = calloc(1, sizeof(*self));
if (!self)
return NULL;
self->ref = 1;
self->payload = payload;
self->size = size;
return self;
}
struct rcbuf* rcbuf_from_string(const char* str)
{
char* value = strdup(str);
return value ? rcbuf_new(value, strlen(str)) : NULL;
}
struct rcbuf* rcbuf_from_mem(const void* addr, size_t size)
{
void* mem = malloc(size);
if (!mem)
return NULL;
memcpy(mem, addr, size);
struct rcbuf* rcbuf = rcbuf_new(mem, size);
if (!rcbuf)
free(mem);
return rcbuf;
}
void rcbuf_ref(struct rcbuf* self)
{
self->ref++;
}
void rcbuf_unref(struct rcbuf* self)
{
assert(self->ref > 0);
if (--self->ref > 0)
return;
free(self->payload);
free(self);
}