diff options
| -rw-r--r-- | include/perk/perk.h | 2 | ||||
| -rw-r--r-- | src/driver/pe_unit_ctx.c | 2 | ||||
| -rw-r--r-- | src/logic/pe_map_raw_image.c | 18 |
3 files changed, 13 insertions, 9 deletions
diff --git a/include/perk/perk.h b/include/perk/perk.h index be5c079..bb2a05b 100644 --- a/include/perk/perk.h +++ b/include/perk/perk.h @@ -188,7 +188,7 @@ perk_api int pe_output_export_symbols (const struct pe_image_meta *, const stru perk_api int pe_output_import_libraries(const struct pe_image_meta *, const struct pe_common_ctx *, FILE *); /* high-level api */ -perk_api int pe_map_raw_image (int fd, const char * path, int prot, struct pe_raw_image *); +perk_api int pe_map_raw_image (const struct pe_driver_ctx *, int fd, const char * path, int prot, struct pe_raw_image *); perk_api int pe_unmap_raw_image (struct pe_raw_image *); perk_api int pe_get_image_meta (const struct pe_driver_ctx *, const struct pe_raw_image *, struct pe_image_meta **); diff --git a/src/driver/pe_unit_ctx.c b/src/driver/pe_unit_ctx.c index 2ea690e..5330659 100644 --- a/src/driver/pe_unit_ctx.c +++ b/src/driver/pe_unit_ctx.c @@ -43,7 +43,7 @@ int pe_get_unit_ctx( ? PROT_READ | PROT_WRITE : PROT_READ; - if (pe_map_raw_image(dctx->cctx->ioctx->fdin,path,prot,&ctx->map)) + if (pe_map_raw_image(dctx,dctx->cctx->ioctx->fdin,path,prot,&ctx->map)) return pe_free_unit_ctx_impl(ctx, PERK_SYSTEM_ERROR(dctx)); diff --git a/src/logic/pe_map_raw_image.c b/src/logic/pe_map_raw_image.c index 0598d85..4c6ec92 100644 --- a/src/logic/pe_map_raw_image.c +++ b/src/logic/pe_map_raw_image.c @@ -13,12 +13,14 @@ #include <sys/stat.h> #include <perk/perk.h> +#include "perk_errinfo_impl.h" int pe_map_raw_image( - int fd, - const char * path, - int prot, - struct pe_raw_image * map) + const struct pe_driver_ctx * dctx, + int fd, + const char * path, + int prot, + struct pe_raw_image * map) { struct stat st; bool fnew; @@ -28,13 +30,13 @@ int pe_map_raw_image( fd = open(path,O_RDONLY | O_CLOEXEC); if (fd < 0) - return -1; + return PERK_SYSTEM_ERROR(dctx); if ((ret = fstat(fd,&st) < 0) && fnew) close(fd); if (ret < 0) - return -1; + return PERK_SYSTEM_ERROR(dctx); map->size = st.st_size; map->addr = mmap(0,map->size,prot,MAP_PRIVATE,fd,0); @@ -42,7 +44,9 @@ int pe_map_raw_image( if (fnew) close(fd); - return (map->addr == MAP_FAILED) ? -1 : 0; + return (map->addr == MAP_FAILED) + ? PERK_SYSTEM_ERROR(dctx) + : 0; } int pe_unmap_raw_image(struct pe_raw_image * map) |