summaryrefslogtreecommitdiffhomepage
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/internal/perk_output.h31
-rw-r--r--src/output/pe_output_export_symbols.c26
2 files changed, 38 insertions, 19 deletions
diff --git a/src/internal/perk_output.h b/src/internal/perk_output.h
new file mode 100644
index 0000000..d60d78d
--- /dev/null
+++ b/src/internal/perk_output.h
@@ -0,0 +1,31 @@
+#include <perk/perk.h>
+
+static inline FILE * pe_output_prolog(const struct pe_common_ctx * cctx, FILE * fout, FILE ** ftmp)
+{
+ int fdout;
+
+ *ftmp = (FILE *)0;
+
+ if (fout)
+ return fout;
+ else if (!cctx)
+ return 0;
+ else if (cctx->fdout < 0)
+ return stdout;
+
+ if ((fdout = dup(cctx->fdout)) < 0)
+ return 0;
+
+ if ((*ftmp = fdopen(fdout,"a")))
+ return *ftmp;
+
+ close(fdout);
+ return 0;
+}
+
+static inline int pe_output_epilog(int ret, FILE * f)
+{
+ if (f)
+ fclose(f);
+ return ret;
+}
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 <errno.h>
#include <perk/perk.h>
+#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; 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);
+ fprintf(fout,"%s\n",(char *)((uintptr_t)m->image.addr + symrva[i] - offset));
- return 0;
+ return pe_output_epilog(0,ftmp);
}