diff options
Diffstat (limited to 'src/arbits/output/pe_ar_print_members.c')
| -rw-r--r-- | src/arbits/output/pe_ar_print_members.c | 181 |
1 files changed, 181 insertions, 0 deletions
diff --git a/src/arbits/output/pe_ar_print_members.c b/src/arbits/output/pe_ar_print_members.c new file mode 100644 index 0000000..3bdf539 --- /dev/null +++ b/src/arbits/output/pe_ar_print_members.c @@ -0,0 +1,181 @@ +/***************************************************************/ +/* perk: PE Resource Kit */ +/* Copyright (C) 2015--2025 SysDeer Technologies, LLC */ +/* Released under GPLv2 and GPLv3; see COPYING.PERK. */ +/***************************************************************/ + +#include <perk/perk.h> +#include <perk/perk_output.h> +#include "perk_driver_impl.h" +#include "perk_dprintf_impl.h" +#include "perk_errinfo_impl.h" +#include "perk_ar_impl.h" + +static void pe_ar_member_not_found( + const struct pe_archive_meta * meta, + const char * name) +{ + const struct pe_driver_ctx * dctx; + int fderr; + + dctx = (pe_archive_meta_ictx(meta))->dctx; + + if ((fderr = pe_driver_fderr(dctx)) >= 0) + pe_dprintf( + fderr, + "%s: error: member '%s' was not found in the current archive\n", + dctx->program,name); +} + +static struct ar_meta_member_info * pe_ar_get_member_info( + const struct pe_archive_meta * meta, + const char * name) +{ + struct ar_meta_member_info ** memberp; + const char * slash; + + if ((slash = strrchr(name,'/'))) + name = ++slash; + + for (memberp=meta->a_memberv; *memberp; memberp++) + if (!strcmp(memberp[0]->ar_file_header.ar_member_name,name)) + return *memberp; + + pe_ar_member_not_found(meta,name); + + return 0; +} + +static int pe_ar_print_one_member_posix( + int fdout, + const struct pe_driver_ctx * dctx, + struct ar_meta_member_info * minfo) +{ + ssize_t ret; + size_t cnt; + const char * ch; + + ret = 0; + cnt = minfo->ar_file_header.ar_file_size; + ch = minfo->ar_object_data; + + for (; cnt && ret>=0; ) { + ch += ret; + cnt -= ret; + + ret = write(fdout,ch,cnt); + + while ((ret < 0) && (errno == EINTR)) + ret = write(fdout,ch,cnt); + } + + if (ret < 0) + return PERK_SYSTEM_ERROR(dctx); + + return 0; +} + +static int pe_ar_print_one_member_posix_verbose( + int fdout, + const struct pe_driver_ctx * dctx, + struct ar_meta_member_info * minfo, + const char * aname) +{ + const char * name = aname ? aname : minfo->ar_file_header.ar_member_name; + + if (pe_dprintf(fdout,"\n<%s>\n\n",name) < 0) + return PERK_SYSTEM_ERROR(dctx); + + if (pe_ar_print_one_member_posix(fdout,dctx,minfo) < 0) + return PERK_NESTED_ERROR(dctx); + + return 0; +} + +static int pe_ar_print_members_posix( + const struct pe_driver_ctx * dctx, + const struct pe_archive_meta * meta, + const struct pe_fd_ctx * fdctx, + const char ** members) +{ + struct ar_meta_member_info ** memberp; + struct ar_meta_member_info * minfo; + const char ** pname; + int fdout; + bool fname; + + fdout = fdctx->fdout; + fname = dctx->cctx->fmtflags & PERK_PRETTY_VERBOSE; + + for (pname=members; pname && *pname; pname++) { + if ((minfo = pe_ar_get_member_info(meta,*pname))) { + switch (minfo->ar_member_attr) { + case AR_MEMBER_ATTR_ARMAP: + case AR_MEMBER_ATTR_LINKINFO: + case AR_MEMBER_ATTR_NAMESTRS: + break; + + default: + if (fname) { + if (pe_ar_print_one_member_posix_verbose( + fdout,dctx,minfo,*pname) < 0) + return PERK_NESTED_ERROR(dctx); + } else { + if (pe_ar_print_one_member_posix( + fdout,dctx,minfo) < 0) + return PERK_NESTED_ERROR(dctx); + } + } + } + } + + for (memberp=meta->a_memberv; !members && *memberp; memberp++) { + switch ((*memberp)->ar_member_attr) { + case AR_MEMBER_ATTR_ARMAP: + case AR_MEMBER_ATTR_LINKINFO: + case AR_MEMBER_ATTR_NAMESTRS: + break; + + default: + if (fname) { + if (pe_ar_print_one_member_posix_verbose( + fdout,dctx,*memberp,0) < 0) + return PERK_NESTED_ERROR(dctx); + } else { + if (pe_ar_print_one_member_posix( + fdout,dctx,*memberp) < 0) + return PERK_NESTED_ERROR(dctx); + } + } + } + + return 0; +} + +int pe_ar_print_members(const struct pe_archive_meta * meta, const char ** members) +{ + const struct pe_driver_ctx * dctx; + struct pe_fd_ctx fdctx; + + const char ** pmember; + + dctx = (pe_archive_meta_ictx(meta))->dctx; + + if (pe_lib_get_driver_fdctx(dctx,&fdctx) < 0) + return PERK_NESTED_ERROR(dctx); + + if (members && !members[0]) + members = 0; + + if (!meta->a_memberv) { + for (pmember=members; pmember && *pmember; pmember++) + pe_ar_member_not_found(meta,*pmember); + + return members ? 1 : 0; + } + + if (pe_ar_print_members_posix(dctx,meta,&fdctx,members) < 0) + return PERK_NESTED_ERROR(dctx); + + return 0; +} |