diff options
Diffstat (limited to 'src/cmds/pe_cmd_ar.c')
| -rw-r--r-- | src/cmds/pe_cmd_ar.c | 53 |
1 files changed, 48 insertions, 5 deletions
diff --git a/src/cmds/pe_cmd_ar.c b/src/cmds/pe_cmd_ar.c index 4b4f284..97e94c7 100644 --- a/src/cmds/pe_cmd_ar.c +++ b/src/cmds/pe_cmd_ar.c @@ -9,12 +9,50 @@ #include "perk_errinfo_impl.h" #include "perk_ar_impl.h" +int pe_ar_list_members(const struct pe_archive_meta *, const char **); + +static int pe_cmd_ar_perform_unit_actions( + const struct pe_driver_ctx * dctx, + const char * arname, + const char ** members, + uint64_t action) +{ + struct pe_unit_ctx * arctx = 0; + int (*pe_ar_fn)(const struct pe_archive_meta *,const char **); + + if (action == PERK_DRIVER_AR_LIST_MEMBERS) { + pe_ar_fn = pe_ar_list_members; + + } else if (action == PERK_DRIVER_AR_PRINT_MEMBERS) { + pe_ar_fn = pe_ar_print_members; + + } else { + return 0; + } + + if (pe_lib_get_unit_ctx(dctx,arname,&arctx) < 0) + return PERK_NESTED_ERROR(dctx); + + if (arctx->armeta == 0) + return PERK_CUSTOM_ERROR(dctx, + PERK_ERR_AR_NON_ARCHIVE_IMAGE); + + if (pe_ar_fn(arctx->armeta,members) < 0) { + pe_lib_free_unit_ctx(arctx); + return PERK_NESTED_ERROR(dctx); + } + + pe_lib_free_unit_ctx(arctx); + + return 0; +} + static int pe_cmd_ar_verify_cmdline( const struct pe_driver_ctx * dctx, uint64_t flags, const char * posname, const char * arname, - const char ** units) + const char ** members) { uint64_t action; uint64_t poscmd; @@ -24,7 +62,7 @@ static int pe_cmd_ar_verify_cmdline( poscmd = (flags & AR_POSNAME_MASK); vercmd = (flags & PERK_DRIVER_VERSION); - if (vercmd && !posname && !arname && !units) + if (vercmd && !posname && !arname && !members) return 0; switch (action) { @@ -36,7 +74,7 @@ static int pe_cmd_ar_verify_cmdline( case PERK_DRIVER_AR_DELETE_MEMBERS: case PERK_DRIVER_AR_APPEND_MEMBERS: case PERK_DRIVER_AR_EXTRACT_MEMBERS: - case PERK_DRIVER_AR_PRINT_ARCHIVE: + case PERK_DRIVER_AR_PRINT_MEMBERS: if (poscmd || posname) return PERK_CUSTOM_ERROR(dctx, PERK_ERR_AR_INVALID_ANCHORS); @@ -82,9 +120,14 @@ int pe_cmd_ar( uint64_t flags, const char * posname, const char * arname, - const char ** units) + const char ** members) { - if (pe_cmd_ar_verify_cmdline(dctx,flags,posname,arname,units) < 0) + uint64_t action = (flags & AR_ACTION_MASK); + + if (pe_cmd_ar_verify_cmdline(dctx,flags,posname,arname,members) < 0) + return PERK_NESTED_ERROR(dctx); + + if (pe_cmd_ar_perform_unit_actions(dctx,arname,members,action) < 0) return PERK_NESTED_ERROR(dctx); return 0; |