From 7cd8bca5bbc93ed1352e02089d03c34a23609444 Mon Sep 17 00:00:00 2001 From: midipix Date: Mon, 23 Nov 2015 15:35:04 -0500 Subject: pretty printer: use common prolog/epilog for output stream selection. --- src/output/pe_output_export_symbols.c | 26 +++++++------------------- 1 file changed, 7 insertions(+), 19 deletions(-) (limited to 'src/output') diff --git a/src/output/pe_output_export_symbols.c b/src/output/pe_output_export_symbols.c index fde10c5..529cb9e 100644 --- a/src/output/pe_output_export_symbols.c +++ b/src/output/pe_output_export_symbols.c @@ -5,38 +5,26 @@ #include #include +#include "perk_output.h" -int pe_output_export_symbols (const struct pe_image_meta * m, const struct pe_common_ctx * cctx, FILE * f) +int pe_output_export_symbols (const struct pe_image_meta * m, const struct pe_common_ctx * cctx, FILE * fout) { + FILE * ftmp; uint32_t offset; uint32_t * symrva; - int fdout; int i; 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); + + if (!(fout = pe_output_prolog(cctx,fout,&ftmp))) 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)); for (i=0; iedata.num_of_name_ptrs; i++) - fprintf(f,"%s\n",(char *)((uintptr_t)m->image.addr + symrva[i] - offset)); - - if (fdout >= 0) - fclose(f); + fprintf(fout,"%s\n",(char *)((uintptr_t)m->image.addr + symrva[i] - offset)); - return 0; + return pe_output_epilog(0,ftmp); } -- cgit v1.2.3