File Coverage

Shared.xs

Criterion	Covered	Total	%
statement	113	150	75.3
branch	116	250	46.4
condition			n/a
subroutine			n/a
pod			n/a
total	229	400	57.2

#define PERL_NO_GET_CONTEXT
#include "EXTERN.h"
#include "perl.h"
#include "XSUB.h"
#include "ppport.h"
#include "stack.h"

#define EXTRACT_STK(sv) \
    if (!sv_isobject(sv) || !sv_derived_from(sv, "Data::Stack::Shared")) \
        croak("Expected a Data::Stack::Shared object"); \
    StkHandle *h = INT2PTR(StkHandle*, SvIV(SvRV(sv))); \
    if (!h) croak("Attempted to use a destroyed Data::Stack::Shared object")

#define MAKE_OBJ(class, handle) \
    SV *obj = newSViv(PTR2IV(handle)); \
    SV *ref = newRV_noinc(obj); \
    sv_bless(ref, gv_stashpv(class, GV_ADD)); \
    RETVAL = ref



MODULE = Data::Stack::Shared  PACKAGE = Data::Stack::Shared

PROTOTYPES: DISABLE

void
DESTROY(self)
    SV *self
  CODE:
    if (!SvROK(self)) return;
    StkHandle *h = INT2PTR(StkHandle*, SvIV(SvRV(self)));
    if (!h) return;
    sv_setiv(SvRV(self), 0);
    stk_destroy(h);

UV
size(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    RETVAL = stk_size(h);
  OUTPUT:
    RETVAL

UV
capacity(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    RETVAL = (UV)h->hdr->capacity;
  OUTPUT:
    RETVAL

bool
is_empty(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    RETVAL = stk_size(h) == 0;
  OUTPUT:
    RETVAL

bool
is_full(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    RETVAL = stk_size(h) >= (UV)h->hdr->capacity;
  OUTPUT:
    RETVAL

void
clear(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    stk_clear(h);

UV
drain(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    RETVAL = stk_drain(h);
  OUTPUT:
    RETVAL

SV *
path(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    RETVAL = h->path ? newSVpv(h->path, 0) : &PL_sv_undef;
  OUTPUT:
    RETVAL

IV
memfd(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    RETVAL = h->backing_fd;
  OUTPUT:
    RETVAL

IV
eventfd(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    RETVAL = stk_create_eventfd(h);
    if (RETVAL < 0) croak("eventfd: %s", strerror(errno));
  OUTPUT:
    RETVAL

void
eventfd_set(self, fd)
    SV *self
    int fd
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    if (h->notify_fd >= 0 && h->notify_fd != fd) close(h->notify_fd);
    h->notify_fd = fd;

IV
fileno(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    RETVAL = h->notify_fd;
  OUTPUT:
    RETVAL

bool
notify(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    RETVAL = stk_notify(h);
  OUTPUT:
    RETVAL

SV *
eventfd_consume(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    int64_t v = stk_eventfd_consume(h);
    RETVAL = (v >= 0) ? newSViv((IV)v) : &PL_sv_undef;
  OUTPUT:
    RETVAL

void
sync(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    if (stk_msync(h) != 0) croak("msync: %s", strerror(errno));

void
unlink(self_or_class, ...)
    SV *self_or_class
  CODE:
    const char *p;
    if (sv_isobject(self_or_class)) {
        StkHandle *h = INT2PTR(StkHandle*, SvIV(SvRV(self_or_class)));
        if (!h) croak("Attempted to use a destroyed object");
        p = h->path;
    } else {
        if (items < 2) croak("Usage: ...->unlink($path)");
        p = SvPV_nolen(ST(1));
    }
    if (!p) croak("cannot unlink anonymous or memfd object");
    if (unlink(p) != 0) croak("unlink(%s): %s", p, strerror(errno));

SV *
stats(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    HV *hv = newHV();
    StkHeader *hdr = h->hdr;
    hv_store(hv, "size", 4, newSVuv(stk_size(h)), 0);
    hv_store(hv, "capacity", 8, newSVuv((UV)hdr->capacity), 0);
    hv_store(hv, "pushes", 6, newSVuv((UV)__atomic_load_n(&hdr->stat_pushes, __ATOMIC_RELAXED)), 0);
    hv_store(hv, "pops", 4, newSVuv((UV)__atomic_load_n(&hdr->stat_pops, __ATOMIC_RELAXED)), 0);
    hv_store(hv, "waits", 5, newSVuv((UV)__atomic_load_n(&hdr->stat_waits, __ATOMIC_RELAXED)), 0);
    hv_store(hv, "timeouts", 8, newSVuv((UV)__atomic_load_n(&hdr->stat_timeouts, __ATOMIC_RELAXED)), 0);
    hv_store(hv, "mmap_size", 9, newSVuv((UV)h->mmap_size), 0);
    RETVAL = newRV_noinc((SV *)hv);
  OUTPUT:
    RETVAL



MODULE = Data::Stack::Shared  PACKAGE = Data::Stack::Shared::Int

PROTOTYPES: DISABLE

SV *
new(class, path, capacity)
    const char *class
    SV *path
    UV capacity
  PREINIT:
    char errbuf[STK_ERR_BUFLEN];
  CODE:
    const char *p = SvOK(path) ? SvPV_nolen(path) : NULL;
    StkHandle *h = stk_create(p, capacity, sizeof(int64_t), STK_VAR_INT, errbuf);
    if (!h) croak("Data::Stack::Shared::Int->new: %s", errbuf);
    MAKE_OBJ(class, h);
  OUTPUT:
    RETVAL

SV *
new_memfd(class, name, capacity)
    const char *class
    const char *name
    UV capacity
  PREINIT:
    char errbuf[STK_ERR_BUFLEN];
  CODE:
    StkHandle *h = stk_create_memfd(name, capacity, sizeof(int64_t), STK_VAR_INT, errbuf);
    if (!h) croak("Data::Stack::Shared::Int->new_memfd: %s", errbuf);
    MAKE_OBJ(class, h);
  OUTPUT:
    RETVAL

SV *
new_from_fd(class, fd)
    const char *class
    int fd
  PREINIT:
    char errbuf[STK_ERR_BUFLEN];
  CODE:
    StkHandle *h = stk_open_fd(fd, STK_VAR_INT, errbuf);
    if (!h) croak("Data::Stack::Shared::Int->new_from_fd: %s", errbuf);
    MAKE_OBJ(class, h);
  OUTPUT:
    RETVAL

bool
push(self, val)
    SV *self
    IV val
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    int64_t v = (int64_t)val;
    RETVAL = stk_try_push(h, &v, sizeof(v));
  OUTPUT:
    RETVAL

bool
push_wait(self, val, ...)
    SV *self
    IV val
  PREINIT:
    EXTRACT_STK(self);
    double timeout = -1;
  CODE:
    if (items > 2) timeout = SvNV(ST(2));
    int64_t v = (int64_t)val;
    RETVAL = stk_push(h, &v, sizeof(v), timeout);
  OUTPUT:
    RETVAL

SV *
pop(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    int64_t v;
    RETVAL = stk_try_pop(h, &v) ? newSViv((IV)v) : &PL_sv_undef;
  OUTPUT:
    RETVAL

SV *
pop_wait(self, ...)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
    double timeout = -1;
  CODE:
    if (items > 1) timeout = SvNV(ST(1));
    int64_t v;
    RETVAL = stk_pop(h, &v, timeout) ? newSViv((IV)v) : &PL_sv_undef;
  OUTPUT:
    RETVAL

SV *
peek(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    int64_t v;
    RETVAL = stk_peek(h, &v) ? newSViv((IV)v) : &PL_sv_undef;
  OUTPUT:
    RETVAL



MODULE = Data::Stack::Shared  PACKAGE = Data::Stack::Shared::Str

PROTOTYPES: DISABLE

SV *
new(class, path, capacity, max_len)
    const char *class
    SV *path
    UV capacity
    UV max_len
  PREINIT:
    char errbuf[STK_ERR_BUFLEN];
  CODE:
    if (max_len == 0) croak("max_len must be > 0");
    if (max_len > (UV)(UINT32_MAX - sizeof(uint32_t)))
        croak("max_len too large");
    uint32_t elem_size = (uint32_t)(sizeof(uint32_t) + max_len);
    const char *p = SvOK(path) ? SvPV_nolen(path) : NULL;
    StkHandle *h = stk_create(p, capacity, elem_size, STK_VAR_STR, errbuf);
    if (!h) croak("Data::Stack::Shared::Str->new: %s", errbuf);
    MAKE_OBJ(class, h);
  OUTPUT:
    RETVAL

SV *
new_memfd(class, name, capacity, max_len)
    const char *class
    const char *name
    UV capacity
    UV max_len
  PREINIT:
    char errbuf[STK_ERR_BUFLEN];
  CODE:
    if (max_len == 0) croak("max_len must be > 0");
    if (max_len > (UV)(UINT32_MAX - sizeof(uint32_t)))
        croak("max_len too large");
    uint32_t elem_size = (uint32_t)(sizeof(uint32_t) + max_len);
    StkHandle *h = stk_create_memfd(name, capacity, elem_size, STK_VAR_STR, errbuf);
    if (!h) croak("Data::Stack::Shared::Str->new_memfd: %s", errbuf);
    MAKE_OBJ(class, h);
  OUTPUT:
    RETVAL

SV *
new_from_fd(class, fd)
    const char *class
    int fd
  PREINIT:
    char errbuf[STK_ERR_BUFLEN];
  CODE:
    StkHandle *h = stk_open_fd(fd, STK_VAR_STR, errbuf);
    if (!h) croak("Data::Stack::Shared::Str->new_from_fd: %s", errbuf);
    MAKE_OBJ(class, h);
  OUTPUT:
    RETVAL

bool
push(self, val)
    SV *self
    SV *val
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    STRLEN slen;
    const char *s = SvPV(val, slen);
    uint32_t max_len = h->elem_size - sizeof(uint32_t);
    if (slen > max_len) slen = max_len;
    uint8_t *buf;
    Newxz(buf, h->elem_size, uint8_t);
    uint32_t l32 = (uint32_t)slen;
    memcpy(buf, &l32, sizeof(uint32_t));
    memcpy(buf + sizeof(uint32_t), s, slen);
    RETVAL = stk_try_push(h, buf, h->elem_size);
    Safefree(buf);
  OUTPUT:
    RETVAL

bool
push_wait(self, val, ...)
    SV *self
    SV *val
  PREINIT:
    EXTRACT_STK(self);
    double timeout = -1;
  CODE:
    if (items > 2) timeout = SvNV(ST(2));
    STRLEN slen;
    const char *s = SvPV(val, slen);
    uint32_t max_len = h->elem_size - sizeof(uint32_t);
    if (slen > max_len) slen = max_len;
    uint8_t *buf;
    Newxz(buf, h->elem_size, uint8_t);
    uint32_t l32 = (uint32_t)slen;
    memcpy(buf, &l32, sizeof(uint32_t));
    memcpy(buf + sizeof(uint32_t), s, slen);
    RETVAL = stk_push(h, buf, h->elem_size, timeout);
    Safefree(buf);
  OUTPUT:
    RETVAL

SV *
pop(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    uint8_t *buf;
    Newx(buf, h->elem_size, uint8_t);
    if (stk_try_pop(h, buf)) {
        uint32_t len;
        memcpy(&len, buf, sizeof(uint32_t));
        uint32_t max_len = h->elem_size - sizeof(uint32_t);
        if (len > max_len) len = max_len;
        RETVAL = newSVpvn((char *)buf + sizeof(uint32_t), len);
    } else {
        RETVAL = &PL_sv_undef;
    }
    Safefree(buf);
  OUTPUT:
    RETVAL

SV *
pop_wait(self, ...)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
    double timeout = -1;
  CODE:
    if (items > 1) timeout = SvNV(ST(1));
    uint8_t *buf;
    Newx(buf, h->elem_size, uint8_t);
    if (stk_pop(h, buf, timeout)) {
        uint32_t len;
        memcpy(&len, buf, sizeof(uint32_t));
        uint32_t max_len = h->elem_size - sizeof(uint32_t);
        if (len > max_len) len = max_len;
        RETVAL = newSVpvn((char *)buf + sizeof(uint32_t), len);
    } else {
        RETVAL = &PL_sv_undef;
    }
    Safefree(buf);
  OUTPUT:
    RETVAL

SV *
peek(self)
    SV *self
  PREINIT:
    EXTRACT_STK(self);
  CODE:
    uint8_t *buf;
    Newx(buf, h->elem_size, uint8_t);
    if (stk_peek(h, buf)) {
        uint32_t len;
        memcpy(&len, buf, sizeof(uint32_t));
        uint32_t max_len = h->elem_size - sizeof(uint32_t);
        if (len > max_len) len = max_len;
        RETVAL = newSVpvn((char *)buf + sizeof(uint32_t), len);
    } else {
        RETVAL = &PL_sv_undef;
    }
    Safefree(buf);
  OUTPUT:
    RETVAL