forked from MarcoBuster/Magisk
131 lines
2.7 KiB
C++
131 lines
2.7 KiB
C++
#include <utils.hpp>
|
|
#include <logging.hpp>
|
|
#include <stream.hpp>
|
|
|
|
static int strm_read(void *v, char *buf, int len) {
|
|
auto strm = reinterpret_cast<stream *>(v);
|
|
return strm->read(buf, len);
|
|
}
|
|
|
|
static int strm_write(void *v, const char *buf, int len) {
|
|
auto strm = reinterpret_cast<stream *>(v);
|
|
return strm->write(buf, len);
|
|
}
|
|
|
|
static fpos_t strm_seek(void *v, fpos_t off, int whence) {
|
|
auto strm = reinterpret_cast<stream *>(v);
|
|
return strm->seek(off, whence);
|
|
}
|
|
|
|
static int strm_close(void *v) {
|
|
auto strm = reinterpret_cast<stream *>(v);
|
|
delete strm;
|
|
return 0;
|
|
}
|
|
|
|
sFILE make_stream_fp(stream_ptr &&strm) {
|
|
sFILE fp(funopen(strm.release(), strm_read, strm_write, strm_seek, strm_close), fclose);
|
|
setbuf(fp.get(), nullptr);
|
|
return fp;
|
|
}
|
|
|
|
int stream::read(void *buf, size_t len) {
|
|
LOGE("This stream does not support read");
|
|
return -1;
|
|
}
|
|
|
|
int stream::write(const void *buf, size_t len) {
|
|
LOGE("This stream does not support write");
|
|
return -1;
|
|
}
|
|
|
|
off_t stream::seek(off_t off, int whence) {
|
|
LOGE("This stream does not support seek");
|
|
return -1;
|
|
}
|
|
|
|
int fp_stream::read(void *buf, size_t len) {
|
|
return fread(buf, 1, len, fp.get());
|
|
}
|
|
|
|
int fp_stream::write(const void *buf, size_t len) {
|
|
return fwrite(buf, 1, len, fp.get());
|
|
}
|
|
|
|
off_t fp_stream::seek(off_t off, int whence) {
|
|
return fseek(fp.get(), off, whence);
|
|
}
|
|
|
|
int filter_stream::read(void *buf, size_t len) {
|
|
return base->read(buf, len);
|
|
}
|
|
|
|
int filter_stream::write(const void *buf, size_t len) {
|
|
return base->write(buf, len);
|
|
}
|
|
|
|
byte_stream::byte_stream(uint8_t *&buf, size_t &len) : _buf(buf), _len(len) {
|
|
buf = nullptr;
|
|
len = 0;
|
|
}
|
|
|
|
int byte_stream::read(void *buf, size_t len) {
|
|
len = std::min(len, _len - _pos);
|
|
memcpy(buf, _buf + _pos, len);
|
|
return len;
|
|
}
|
|
|
|
int byte_stream::write(const void *buf, size_t len) {
|
|
resize(_pos + len);
|
|
memcpy(_buf + _pos, buf, len);
|
|
_pos += len;
|
|
_len = std::max(_len, _pos);
|
|
return len;
|
|
}
|
|
|
|
off_t byte_stream::seek(off_t off, int whence) {
|
|
off_t np;
|
|
switch (whence) {
|
|
case SEEK_CUR:
|
|
np = _pos + off;
|
|
break;
|
|
case SEEK_END:
|
|
np = _len + off;
|
|
break;
|
|
case SEEK_SET:
|
|
np = off;
|
|
break;
|
|
default:
|
|
return -1;
|
|
}
|
|
resize(np, true);
|
|
_pos = np;
|
|
return np;
|
|
}
|
|
|
|
void byte_stream::resize(size_t new_pos, bool zero) {
|
|
bool resize = false;
|
|
size_t old_cap = _cap;
|
|
while (new_pos > _cap) {
|
|
_cap = _cap ? (_cap << 1) - (_cap >> 1) : 1 << 12;
|
|
resize = true;
|
|
}
|
|
if (resize) {
|
|
_buf = (uint8_t *) xrealloc(_buf, _cap);
|
|
if (zero)
|
|
memset(_buf + old_cap, 0, _cap - old_cap);
|
|
}
|
|
}
|
|
|
|
int fd_stream::read(void *buf, size_t len) {
|
|
return ::read(fd, buf, len);
|
|
}
|
|
|
|
int fd_stream::write(const void *buf, size_t len) {
|
|
return ::write(fd, buf, len);
|
|
}
|
|
|
|
off_t fd_stream::seek(off_t off, int whence) {
|
|
return lseek(fd, off, whence);
|
|
}
|