a9096718e9
* Fix CID 398788 - memory leak * Fix CID 397077 - remove useless comparison * Fix CID 397076 - remove useless comparison * Fix CID 397074 - remove useless comparison * Fix CID 396207 - memory leak * Fix CID 392014 - bad shift * Fix CID 391965 - NULL dereference * Fix CID 390010 - wrong assert use * Fix CID 378899 - unitialized variable * Fix CID 378859 - integer overflow * Fix CID 366898 - remove useless comparison * Fix CID 364035 - NULL dereference * Fix CID 317001 - closing "-1" file descriptor * Fix CID 316991 - lseek() to negative position * Fix CID 316172 - mmap() with negative size * Fix CID 401510 - double free * Fix CID 395360 - remove useless comparison * Fix CID 352574 - negative loop bound * Fix CID 316187 - memory leak * Fix CID 395425 - memory leak * Fix CID 316975 - buffer overrun * Fix CID 316947 - memory leak * Fix CID 398474 - integer overflow
91 lines
2.2 KiB
C
91 lines
2.2 KiB
C
// SPDX-FileCopyrightText: 2009-2020 ret2libc <sirmy15@gmail.com>
|
|
// SPDX-License-Identifier: LGPL-3.0-only
|
|
|
|
#include <rz_types.h>
|
|
#include <rz_util.h>
|
|
|
|
struct buf_file_user {
|
|
const char *file;
|
|
int perm;
|
|
int mode;
|
|
};
|
|
|
|
struct buf_file_priv {
|
|
int fd;
|
|
ut8 tmp[8];
|
|
};
|
|
|
|
static inline struct buf_file_priv *get_priv_file(RzBuffer *b) {
|
|
struct buf_file_priv *priv = (struct buf_file_priv *)b->priv;
|
|
rz_warn_if_fail(priv);
|
|
return priv;
|
|
}
|
|
|
|
static bool buf_file_init(RzBuffer *b, const void *user) {
|
|
const struct buf_file_user *u = (const struct buf_file_user *)user;
|
|
struct buf_file_priv *priv = RZ_NEW0(struct buf_file_priv);
|
|
if (!priv) {
|
|
return false;
|
|
}
|
|
int fd = rz_sys_open(u->file, u->perm, u->mode);
|
|
if (fd == -1) {
|
|
free(priv);
|
|
return false;
|
|
}
|
|
priv->fd = fd;
|
|
b->priv = priv;
|
|
b->fd = priv->fd;
|
|
return true;
|
|
}
|
|
|
|
static bool buf_file_fini(RzBuffer *b) {
|
|
struct buf_file_priv *priv = get_priv_file(b);
|
|
close(priv->fd);
|
|
RZ_FREE(b->priv);
|
|
return true;
|
|
}
|
|
|
|
static ut64 buf_file_get_size(RzBuffer *b) {
|
|
struct buf_file_priv *priv = get_priv_file(b);
|
|
int pos = lseek(priv->fd, 0, SEEK_CUR);
|
|
int res = lseek(priv->fd, 0, SEEK_END);
|
|
if (pos >= 0) {
|
|
lseek(priv->fd, (off_t)pos, SEEK_SET);
|
|
}
|
|
return (ut64)res;
|
|
}
|
|
|
|
static st64 buf_file_read(RzBuffer *b, ut8 *buf, ut64 len) {
|
|
struct buf_file_priv *priv = get_priv_file(b);
|
|
return read(priv->fd, buf, len);
|
|
}
|
|
|
|
static st64 buf_file_write(RzBuffer *b, const ut8 *buf, ut64 len) {
|
|
struct buf_file_priv *priv = get_priv_file(b);
|
|
return write(priv->fd, buf, len);
|
|
}
|
|
|
|
static st64 buf_file_seek(RzBuffer *b, st64 addr, int whence) {
|
|
struct buf_file_priv *priv = get_priv_file(b);
|
|
switch (whence) {
|
|
case RZ_BUF_CUR: whence = SEEK_CUR; break;
|
|
case RZ_BUF_SET: whence = SEEK_SET; break;
|
|
case RZ_BUF_END: whence = SEEK_END; break;
|
|
}
|
|
return lseek(priv->fd, (off_t)addr, whence);
|
|
}
|
|
|
|
static bool buf_file_resize(RzBuffer *b, ut64 newsize) {
|
|
struct buf_file_priv *priv = get_priv_file(b);
|
|
return rz_sys_truncate_fd(priv->fd, newsize) >= 0;
|
|
}
|
|
|
|
static const RzBufferMethods buffer_file_methods = {
|
|
.init = buf_file_init,
|
|
.fini = buf_file_fini,
|
|
.read = buf_file_read,
|
|
.write = buf_file_write,
|
|
.get_size = buf_file_get_size,
|
|
.resize = buf_file_resize,
|
|
.seek = buf_file_seek,
|
|
};
|