pxt-ev3/libs/core/mmap.cpp
2017-07-10 14:16:31 +01:00

114 lines
2.3 KiB
C++

#include "pxt.h"
#include <unistd.h>
#include <sys/mman.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <sys/ioctl.h>
namespace pxt {
PXT_VTABLE_CTOR(MMap) {
length = 0;
fd = -1;
data = 0;
}
void MMap::print() {
DMESG("MMap %p r=%d len=%d fd=%d data=%p", this, refcnt, length, fd, data);
}
void MMap::destroy() {
munmap(data, length);
close(fd);
}
}
namespace control {
/** Create new file mapping in memory */
//%
MMap *mmap(String filename, int size, int offset) {
DMESG("mmap %s len=%d off=%d", filename->data, size, offset);
int fd = open(filename->data, O_RDWR, 0);
if (fd < 0)
return 0;
void *data = NULL;
if (size > 0) {
data = ::mmap(NULL, size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, offset);
if (data == MAP_FAILED) {
close(fd);
return 0;
}
}
auto r = new MMap();
r->fd = fd;
r->length = size;
r->data = (uint8_t *)data;
return r;
}
}
namespace MMapMethods {
/**
* Write a number in specified format in the buffer.
*/
//%
void setNumber(MMap *buf, NumberFormat format, int offset, TNumber value) {
if (offset < 0)
return;
setNumberCore(buf->data + offset, buf->length - offset, format, value);
}
/**
* Read a number in specified format from the buffer.
*/
//%
TNumber getNumber(MMap *buf, NumberFormat format, int offset) {
if (offset < 0)
return fromInt(0);
return getNumberCore(buf->data + offset, buf->length - offset, format);
}
/**
* Read a range of bytes into a buffer.
*/
//%
Buffer slice(MMap *buf, int offset = 0, int length = -1) {
offset = min((int)buf->length, offset);
if (length < 0)
length = buf->length;
length = min(length, buf->length - offset);
return mkBuffer(buf->data + offset, length);
}
/** Returns the length of a Buffer object. */
//% property
int length(MMap *s) {
return s->length;
}
/** Perform ioctl(2) on the underlaying file */
//%
int ioctl(MMap *mmap, uint32_t id, Buffer data) {
return ::ioctl(mmap->fd, id, data->data);
}
/** Perform write(2) on the underlaying file */
//%
int write(MMap *mmap, Buffer data) {
return ::write(mmap->fd, data->data, data->length);
}
/** Perform read(2) on the underlaying file */
//%
int read(MMap *mmap, Buffer data) {
return ::read(mmap->fd, data->data, data->length);
}
}