diff options
Diffstat (limited to 'src/driver/pe_driver_ctx.c')
| -rw-r--r-- | src/driver/pe_driver_ctx.c | 40 |
1 files changed, 32 insertions, 8 deletions
diff --git a/src/driver/pe_driver_ctx.c b/src/driver/pe_driver_ctx.c index 6a260bd..b99ee7f 100644 --- a/src/driver/pe_driver_ctx.c +++ b/src/driver/pe_driver_ctx.c @@ -29,6 +29,8 @@ static const struct pe_source_version pe_src_version = { static const char * const perk_cmd_name[PERK_CMD_CAP] = { [PERK_CMD_PERK] = "perk", [PERK_CMD_AR] = "ar", + [PERK_CMD_NM] = "nm", + [PERK_CMD_SIZE] = "size", }; /* perk command options */ @@ -36,6 +38,8 @@ static const struct argv_option * perk_cmd_options[PERK_CMD_CAP] = { [PERK_CMD_DEFAULT] = pe_default_options, [PERK_CMD_PERK] = pe_perk_options, [PERK_CMD_AR] = pe_ar_options, + [PERK_CMD_NM] = pe_nm_options, + [PERK_CMD_SIZE] = pe_size_options, }; /* default fd context */ @@ -98,6 +102,18 @@ static int pe_driver_usage( cmdname,cmdname,cmdname); break; + case PERK_CMD_NM: + snprintf(header,sizeof(header), + PERK_NM_CMD_SYNOPSIS, + cmdname,cmdname); + break; + + case PERK_CMD_SIZE: + snprintf(header,sizeof(header), + PERK_SIZE_CMD_SYNOPSIS, + cmdname,cmdname); + break; + case PERK_CMD_PERK: snprintf(header,sizeof(header), PERK_PERK_CMD_SYNOPSIS, @@ -159,6 +175,7 @@ static int pe_cmd_from_program(const char * program) const char * dot; const char * hyphen; const char * mark; + enum pe_cmd cmd; dot = strrchr(program,'.'); hyphen = strrchr(program,'-'); @@ -170,15 +187,12 @@ static int pe_cmd_from_program(const char * program) else mark = program; - if (!strcmp(mark,"perk")) { - return PERK_CMD_PERK; - - } else if (!strcmp(mark,"ar")) { - return PERK_CMD_AR; + for (cmd=0; cmd<PERK_CMD_CAP; cmd++) + if (perk_cmd_name[cmd]) + if (!strcmp(mark,perk_cmd_name[cmd])) + return cmd; - } else { - return PERK_CMD_DEFAULT; - } + return PERK_CMD_DEFAULT; } static int pe_cctx_update( @@ -251,6 +265,10 @@ static int pe_cctx_update( cctx->fmtflags |= PERK_OUTPUT_IMAGE_STRINGS; break; + case TAG_RELOCS: + cctx->fmtflags |= PERK_OUTPUT_BASE_RELOCS; + break; + case TAG_EXPSYMS: cctx->fmtflags |= PERK_OUTPUT_EXPORT_SYMS; break; @@ -483,6 +501,12 @@ int pe_lib_get_driver_ctx( } else if (cctx.cmd == PERK_CMD_AR) { argv_optv_init(pe_ar_options,optv); + + } else if (cctx.cmd == PERK_CMD_NM) { + argv_optv_init(pe_nm_options,optv); + + } else if (cctx.cmd == PERK_CMD_SIZE) { + argv_optv_init(pe_size_options,optv); } /* process the selected tool's command-line arguments */ |