/***************************************************************/

/*  perk: PE Resource Kit                                      */

/*  Copyright (C) 2015--2025  SysDeer Technologies, LLC        */

/*  Released under GPLv2 and GPLv3; see COPYING.PERK.          */

/***************************************************************/

#include <time.h>

#include <locale.h>

#include <inttypes.h>

#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"

#define PERK_PRETTY_FLAGS       (PERK_PRETTY_YAML      \

	                         | PERK_PRETTY_POSIX    \

	                         | PERK_PRETTY_HEXDATA)

#define PPRIU64 "%"PRIu64

const char pe_ar_perm_strs[8][4] = {

	{'-','-','-','\0'},

	{'-','-','x','\0'},

	{'-','w','-','\0'},

	{'-','w','x','\0'},

	{'r','-','-','\0'},

	{'r','-','x','\0'},

	{'r','w','-','\0'},

	{'r','w','x','\0'}

};

static unsigned pe_ar_output_decimal_len_from_val(size_t val, unsigned min)

{

	unsigned ret;

	for (ret=0; val; ret++)

		val /= 10;

	return (ret > min) ? ret : min;

}

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_member_in_members(

	struct ar_meta_member_info *    member,

	const char **                   members)

{

	const char ** pname;

	for (pname=members; *pname; pname++)

		if (!strcmp(*pname,member->ar_file_header.ar_member_name))

			return 1;

	return 0;

}

static int pe_ar_list_one_member_posix(

	int                             fdout,

	struct ar_meta_member_info *    memberp)

{

	return pe_dprintf(

		fdout,"%s\n",

		memberp->ar_file_header.ar_member_name);

}

static int pe_ar_list_one_member_posix_verbose(

	int                             fdout,

	struct ar_meta_member_info *    memberp,

	const char *                    fmtstr,

	locale_t                        arlocale)

{

	unsigned    ownerbits;

	unsigned    groupbits;

	unsigned    worldbits;

	time_t      artimeval;

	struct tm   artimeloc;

	char        artimestr[64] = {0};

	ownerbits = (memberp->ar_file_header.ar_file_mode & 0700) >> 6;

	groupbits = (memberp->ar_file_header.ar_file_mode & 0070) >> 3;

	worldbits = (memberp->ar_file_header.ar_file_mode & 0007);

	artimeval = memberp->ar_file_header.ar_time_date_stamp;

	if (localtime_r(&artimeval,&artimeloc))

		strftime_l(

			artimestr,sizeof(artimestr),

			"%b %e %H:%M %Y",&artimeloc,

			arlocale);

	return pe_dprintf(

		fdout,fmtstr,

		pe_ar_perm_strs[ownerbits],

		pe_ar_perm_strs[groupbits],

		pe_ar_perm_strs[worldbits],

		memberp->ar_file_header.ar_uid,

		memberp->ar_file_header.ar_gid,

		memberp->ar_object_size,

		artimestr,

		memberp->ar_file_header.ar_member_name);

}

static int pe_ar_list_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;

	size_t                          testval;

	size_t                          sizelen;

	size_t                          uidlen;

	size_t                          gidlen;

	locale_t                        arloc;

	char                            fmtstr[64];

	fdout = fdctx->fdout;

	arloc = 0;

	if (dctx->cctx->fmtflags & PERK_PRETTY_VERBOSE) {

		for (sizelen=0,memberp=meta->a_memberv; *memberp; memberp++)

			if ((testval = memberp[0]->ar_object_size) > sizelen)

				if (!members || pe_ar_member_in_members(*memberp,members))

					sizelen = testval;

		for (uidlen=0,memberp=meta->a_memberv; *memberp; memberp++)

			if ((testval = memberp[0]->ar_file_header.ar_uid) > uidlen)

				if (!members || pe_ar_member_in_members(*memberp,members))

					uidlen = testval;

		for (gidlen=0,memberp=meta->a_memberv; *memberp; memberp++)

			if ((testval = memberp[0]->ar_file_header.ar_gid) > gidlen)

				if (!members || pe_ar_member_in_members(*memberp,members))

					gidlen = testval;

		sizelen = pe_ar_output_decimal_len_from_val(sizelen,6);

		uidlen  = pe_ar_output_decimal_len_from_val(uidlen,1);

		gidlen  = pe_ar_output_decimal_len_from_val(gidlen,1);

		arloc   = newlocale(LC_ALL,setlocale(LC_ALL,0),0);

		sprintf(

			fmtstr,

			"%%s%%s%%s "

			"%%"   PPRIU64 "u"

			"/%%-" PPRIU64 "u "

			"%%"   PPRIU64 "u "

			"%%s "

			"%%s\n",

			uidlen,

			gidlen,

			sizelen);

	}

	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 (arloc) {

						if (pe_ar_list_one_member_posix_verbose(

								fdout,minfo,fmtstr,arloc) < 0)

							return PERK_SYSTEM_ERROR(dctx);

					} else {

						if (pe_ar_list_one_member_posix(

								fdout,minfo) < 0)

							return PERK_SYSTEM_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 (arloc) {

					if (pe_ar_list_one_member_posix_verbose(

							fdout,*memberp,fmtstr,arloc) < 0)

						return PERK_SYSTEM_ERROR(dctx);

				} else {

					if (pe_ar_list_one_member_posix(

							fdout,*memberp) < 0)

						return PERK_SYSTEM_ERROR(dctx);

				}

		}

	}

	if (arloc)

		freelocale(arloc);

	return 0;

}

static int pe_ar_list_one_member_yaml(

	int                             fdout,

	struct ar_meta_member_info *    memberp)

{

	return pe_dprintf(

		fdout,

		"    - [ member: %s ]\n",

		memberp->ar_file_header.ar_member_name);

}

static int pe_ar_list_one_member_yaml_verbose(

	int                             fdout,

	struct ar_meta_member_info *    memberp,

	locale_t                        arlocale)

{

	time_t      artimeval;

	struct tm   artimeloc;

	char        artimestr[64] = {0};

	artimeval = memberp->ar_file_header.ar_time_date_stamp;

	if (localtime_r(&artimeval,&artimeloc))

		strftime_l(

			artimestr,sizeof(artimestr),

			"%Y/%m/%d @ %H:%M",&artimeloc,

			arlocale);

	return pe_dprintf(

		fdout,

		"    - Member:\n"

		"      - [ name:       " "%s"    " ]\n"

		"      - [ timestamp:  " "%s"    " ]\n"

		"      - [ filesize:   " PPRIU64 " ]\n"

		"      - [ uid:        " "%d"    " ]\n"

		"      - [ gid:        " "%d"    " ]\n"

		"      - [ mode:       " "%d"    " ]\n\n",

		memberp->ar_file_header.ar_member_name,

		artimestr,

		memberp->ar_object_size,

		memberp->ar_file_header.ar_uid,

		memberp->ar_file_header.ar_gid,

		memberp->ar_file_header.ar_file_mode);

}

static int pe_ar_list_members_yaml(

	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;

	locale_t                        arloc;

	fdout = fdctx->fdout;

	arloc = 0;

	if (dctx->cctx->fmtflags & PERK_PRETTY_VERBOSE) {

		arloc   = newlocale(LC_ALL,setlocale(LC_ALL,0),0);

	}

	if (pe_dprintf(fdctx->fdout,"  - Members:\n") < 0)

		return PERK_SYSTEM_ERROR(dctx);

	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 (arloc) {

						if (pe_ar_list_one_member_yaml_verbose(

								fdout,minfo,arloc) < 0)

							return PERK_SYSTEM_ERROR(dctx);

					} else {

						if (pe_ar_list_one_member_yaml(

								fdout,minfo) < 0)

							return PERK_SYSTEM_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 (arloc) {

					if (pe_ar_list_one_member_yaml_verbose(

							fdout,*memberp,arloc) < 0)

						return PERK_SYSTEM_ERROR(dctx);

				} else {

					if (pe_ar_list_one_member_yaml(

							fdout,*memberp) < 0)

						return PERK_SYSTEM_ERROR(dctx);

				}

		}

	}

	if (arloc)

		freelocale(arloc);

	return 0;

}

int pe_ar_list_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;

	}

	switch (dctx->cctx->fmtflags & PERK_PRETTY_FLAGS) {

		case PERK_PRETTY_YAML:

			return pe_ar_list_members_yaml(

				dctx,meta,&fdctx,members);

		case PERK_PRETTY_POSIX:

			return pe_ar_list_members_posix(

				dctx,meta,&fdctx,members);

		default:

			return pe_ar_list_members_posix(

				dctx,meta,&fdctx,members);

	}

}