Idris2/support/refc/buffer.c
Hattori, Hiroki 5c9f8e36a1
[ RefC ] Add 16 and 32 bit access to base/Data.Buffer . (#2609)
* Fix symbom mangling

* Revert "Fix symbom mangling"

This reverts commit 6481e80155.

* Fix typo

* [RefC] Add missed prims of setBuffer* .

* [ fix ] formatting

* [ re #2609 ] Use 'UInt' instead of 'Word'

More descriptive/to the point / Less assumed knowledge.

There are no *LE suffixes for UInt8, since endianness is to do with
multiple bytes and UInt8 is a single one.

Co-authored-by: Guillaume Allais <guillaume.allais@ens-lyon.org>
Co-authored-by: Thomas E. Hansen <teh6@st-andrews.ac.uk>
2022-11-15 12:15:06 +01:00

93 lines
2.2 KiB
C

#include "buffer.h"
#include "refc_util.h"
#include <string.h>
#include <sys/stat.h>
void *newBuffer(int bytes) {
size_t size = sizeof(Buffer) + bytes * sizeof(uint8_t);
Buffer *buf = malloc(size);
if (buf == NULL) {
return NULL;
}
buf->size = bytes;
memset(buf->data, 0, bytes);
return (void *)buf;
}
static void assert_valid_range(Buffer *buf, int64_t offset, int64_t len) {
IDRIS2_REFC_VERIFY(offset >= 0, "offset (%lld) < 0", (long long)offset);
IDRIS2_REFC_VERIFY(len >= 0, "len (%lld) < 0", (long long)offset);
IDRIS2_REFC_VERIFY(offset + len <= buf->size,
"offset (%lld) + len (%lld) > buf.size (%lld)",
(long long)offset, (long long)len, (long long)buf->size);
}
void copyBuffer(void *from, int from_offset, int len, void *to, int to_offset) {
Buffer *bfrom = from;
Buffer *bto = to;
assert_valid_range(bfrom, from_offset, len);
assert_valid_range(bto, to_offset, len);
memcpy(bto->data + to_offset, bfrom->data + from_offset, len);
}
int getBufferSize(void *buffer) { return ((Buffer *)buffer)->size; }
void setBufferUIntLE(void *b, int loc, uint64_t val, size_t len) {
assert_valid_range((Buffer *)b, loc, len);
while (len--) {
((Buffer *)b)->data[loc++] = (uint8_t)val;
val >>= 8;
}
}
uint64_t getBufferUIntLE(void *b, int loc, size_t len) {
assert_valid_range((Buffer *)b, loc, len);
uint64_t r = 0;
loc += len;
while (len--) {
r <<= 8;
r += (uint8_t)(((Buffer *)b)->data[--loc]);
}
return r;
}
void setBufferDouble(void *buffer, int loc, double val) {
union {
double d;
uint64_t i;
} tmp;
tmp.d = val;
setBufferUIntLE(buffer, loc, tmp.i, 8);
}
void setBufferString(void *buffer, int loc, char *str) {
Buffer *b = buffer;
size_t len = strlen(str);
assert_valid_range(b, loc, len);
memcpy((b->data) + loc, str, len);
}
double getBufferDouble(void *buffer, int loc) {
union {
double d;
uint64_t i;
} tmp;
tmp.i = getBufferUIntLE(buffer, loc, 8);
return tmp.d;
}
char *getBufferString(void *buffer, int loc, int len) {
Buffer *b = buffer;
assert_valid_range(b, loc, len);
char *s = (char *)(b->data + loc);
char *rs = malloc(len + 1);
strncpy(rs, s, len);
rs[len] = '\0';
return rs;
}