72 lines
1.6 KiB
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);
|
||
|
|
}
|