summaryrefslogtreecommitdiffhomepage
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/perk.c41
1 files changed, 38 insertions, 3 deletions
diff --git a/src/perk.c b/src/perk.c
index 983c130..b952a1b 100644
--- a/src/perk.c
+++ b/src/perk.c
@@ -1,11 +1,46 @@
+#include <stdio.h>
#include <perk/perk.h>
-static int perk_main(int argc, char ** argv, char ** envp)
+#ifndef PERK_DRIVER_FLAGS
+#define PERK_DRIVER_FLAGS PERK_DRIVER_VERBOSITY_ERRORS|PERK_DRIVER_VERBOSITY_USAGE
+#endif
+
+static void perk_perform_unit_actions(struct pe_unit_ctx * uctx)
+{
+ uint64_t flags = uctx->cctx.fmtflags;
+
+ if (flags & PERK_OUTPUT_FLAG_EXPORT_SYMS)
+ uctx->cctx.status = pe_output_export_symbols(uctx->meta,&uctx->cctx,0);
+}
+
+static int perk_exit(struct pe_driver_ctx * dctx, int status)
{
- return -1;
+ pe_free_driver_ctx(dctx);
+ return (status < 0) ? 2 : status;
+}
+
+static int perk_main(int argc, const char ** argv, const char ** envp)
+{
+ int ret;
+ struct pe_driver_ctx * dctx;
+ struct pe_unit_ctx * uctx;
+ const char ** unit;
+
+ if ((ret = pe_get_driver_ctx(argv,envp,PERK_DRIVER_FLAGS,&dctx)))
+ return (ret == PERK_USAGE) ? 0 : 2;
+
+ for (unit=dctx->units; *unit; unit++) {
+ if (!(pe_get_unit_ctx(dctx,*unit,&uctx))) {
+ perk_perform_unit_actions(uctx);
+ ret = uctx->cctx.status ? uctx->cctx.status : ret;
+ pe_free_unit_ctx(uctx);
+ }
+ }
+
+ return perk_exit(dctx,ret);
}
-int main(int argc, char ** argv, char ** envp)
+int main(int argc, const char ** argv, const char ** envp)
{
return perk_main(argc,argv,envp);
}