1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
/***************************************************************/
/* perk: PE Resource Kit */
/* Copyright (C) 2015--2025 SysDeer Technologies, LLC */
/* Released under GPLv2 and GPLv3; see COPYING.PERK. */
/***************************************************************/
#include <stdio.h>
#include <unistd.h>
#include <perk/perk.h>
#include <perk/perk_output.h>
#include "perk_driver_impl.h"
#include "perk_dprintf_impl.h"
#include "perk_ar_impl.h"
#ifndef PERK_DRIVER_FLAGS
#define PERK_DRIVER_FLAGS PERK_DRIVER_VERBOSITY_ERRORS \
| PERK_DRIVER_VERBOSITY_UTILITY \
| PERK_DRIVER_AR_OBJECT_VECTOR
#endif
static const char vermsg[] = "%s%s%s (https://git.foss21.org/perk): "
"version %s%d.%d.%d%s.\n"
"[commit reference: %s%s%s]\n";
static const char * const pe_ver_color[6] = {
"\x1b[1m\x1b[35m","\x1b[0m",
"\x1b[1m\x1b[32m","\x1b[0m",
"\x1b[1m\x1b[34m","\x1b[0m"
};
static const char * const pe_ver_plain[6] = {
"","",
"","",
"",""
};
static ssize_t pe_version(struct pe_driver_ctx * dctx, int fdout)
{
const struct pe_source_version * verinfo;
const char * const * verclr;
verinfo = pe_source_version();
verclr = isatty(fdout) ? pe_ver_color : pe_ver_plain;
return pe_dprintf(
fdout,vermsg,
verclr[0],dctx->program,verclr[1],
verclr[2],verinfo->major,verinfo->minor,
verinfo->revision,verclr[3],
verclr[4],verinfo->commit,verclr[5]);
}
static int pe_exit(struct pe_driver_ctx * dctx, int ret)
{
pe_output_error_vector(dctx);
pe_lib_free_driver_ctx(dctx);
return ret;
}
typedef int (*pe_cmd)(const struct pe_driver_ctx *, const char *);
static void pe_cmd_common(
const struct pe_driver_ctx * dctx,
pe_cmd cmdfn)
{
const char ** unit;
for (unit=dctx->units; *unit; unit++)
cmdfn(dctx,*unit);
}
int pe_main(char ** argv, char ** envp, const struct pe_fd_ctx * fdctx)
{
int ret;
int fdout;
uint64_t flags;
struct pe_driver_ctx * dctx;
const char * posname;
const char * arname;
const char ** arfiles;
uint64_t arflags;
flags = PERK_DRIVER_FLAGS;
fdout = fdctx ? fdctx->fdout : STDOUT_FILENO;
if ((ret = pe_lib_get_driver_ctx(argv,envp,flags,fdctx,&dctx)))
return (ret == PERK_USAGE)
? !argv || !argv[0] || !argv[1]
: PERK_ERROR;
if (dctx->cctx->drvflags & PERK_DRIVER_VERSION)
if ((pe_version(dctx,fdout)) < 0)
return pe_exit(dctx,PERK_ERROR);
switch (dctx->cctx->cmd) {
case PERK_CMD_PERK:
pe_cmd_common(dctx,pe_cmd_perk);
break;
case PERK_CMD_AR:
arflags = dctx->cctx->drvflags;
if (arflags & AR_POSNAME_MASK) {
posname = dctx->units[0];
arname = posname ? dctx->units[1] : 0;
arfiles = arname ? &dctx->units[2] : 0;
} else {
posname = 0;
arname = dctx->units[0];
arfiles = arname ? &dctx->units[1] : 0;
}
pe_cmd_ar(dctx,arflags,posname,arname,arfiles);
break;
case PERK_CMD_NM:
pe_cmd_common(dctx,pe_cmd_nm);
break;
case PERK_CMD_SIZE:
pe_cmd_common(dctx,pe_cmd_size);
break;
default:
break;
}
return pe_exit(dctx,dctx->errv[0] ? PERK_ERROR : PERK_OK);
}
|