diff options
| -rw-r--r-- | include/perk/perk.h | 2 | ||||
| -rw-r--r-- | src/output/pe_output_export_symbols.c | 22 |
2 files changed, 21 insertions, 3 deletions
diff --git a/include/perk/perk.h b/include/perk/perk.h index ca51be4..22f7bd5 100644 --- a/include/perk/perk.h +++ b/include/perk/perk.h @@ -157,7 +157,7 @@ perk_api int pe_get_unit_ctx (struct pe_driver_ctx *, const char * name, struct perk_api int pe_free_unit_ctx (struct pe_unit_ctx *); /* utility api */ -perk_api int pe_output_export_symbols (const struct pe_image_meta *, uint32_t flags, FILE *); +perk_api int pe_output_export_symbols (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 * name, int prot, struct pe_raw_image *); diff --git a/src/output/pe_output_export_symbols.c b/src/output/pe_output_export_symbols.c index 099b540..fde10c5 100644 --- a/src/output/pe_output_export_symbols.c +++ b/src/output/pe_output_export_symbols.c @@ -6,13 +6,28 @@ #include <perk/perk.h> -int pe_output_export_symbols (const struct pe_image_meta * m, uint32_t flags, FILE * f) +int pe_output_export_symbols (const struct pe_image_meta * m, const struct pe_common_ctx * cctx, FILE * f) { uint32_t offset; uint32_t * symrva; + int fdout; int i; - if (!m->hedata) return 0; + if (!m->hedata) + return 0; + else if (f) + fdout = -1; + else if (!cctx) + return -1; + else if (cctx->fdout < 0) { + f = stdout; + fdout = -1; + } else if ((fdout = dup(cctx->fdout)) < 0) + return -1; + else if (!(f = fdopen(fdout,"a"))) { + close(fdout); + return -1; + } offset = m->hedata->virtual_addr - m->hedata->ptr_to_raw_data; symrva = (uint32_t *)((uintptr_t)m->image.addr + (m->edata.name_ptr_rva - offset)); @@ -20,5 +35,8 @@ int pe_output_export_symbols (const struct pe_image_meta * m, uint32_t flags, FI for (i=0; i<m->edata.num_of_name_ptrs; i++) fprintf(f,"%s\n",(char *)((uintptr_t)m->image.addr + symrva[i] - offset)); + if (fdout >= 0) + fclose(f); + return 0; } |