pki tool supports single letter short options

This commit is contained in:
Martin Willi
2009-09-15 10:20:22 +02:00
parent 012f62a820
commit 4fdb9f6f74
8 changed files with 42 additions and 13 deletions
+30 -6
View File
@@ -46,19 +46,26 @@ static int help_idx;
struct option command_opts[MAX_COMMANDS > MAX_OPTIONS ?: MAX_OPTIONS]; struct option command_opts[MAX_COMMANDS > MAX_OPTIONS ?: MAX_OPTIONS];
/** /**
* Build long_opts for a specific command * Global optstring used by all subcommands
*/
char command_optstring[(MAX_COMMANDS > MAX_OPTIONS ?: MAX_OPTIONS) * 3];
/**
* Build command_opts/command_optstr for the active command
*/ */
static void build_opts() static void build_opts()
{ {
int i; int i, pos = 0;
memset(command_opts, 0, sizeof(command_opts)); memset(command_opts, 0, sizeof(command_opts));
memset(command_optstring, 0, sizeof(command_optstring));
if (active == help_idx) if (active == help_idx)
{ {
for (i = 0; cmds[i].cmd; i++) for (i = 0; cmds[i].cmd; i++)
{ {
command_opts[i].name = cmds[i].cmd; command_opts[i].name = cmds[i].cmd;
command_opts[i].val = cmds[i].op; command_opts[i].val = cmds[i].op;
command_optstring[i] = cmds[i].op;
} }
} }
else else
@@ -68,6 +75,19 @@ static void build_opts()
command_opts[i].name = cmds[active].options[i].name; command_opts[i].name = cmds[active].options[i].name;
command_opts[i].has_arg = cmds[active].options[i].arg; command_opts[i].has_arg = cmds[active].options[i].arg;
command_opts[i].val = cmds[active].options[i].op; command_opts[i].val = cmds[active].options[i].op;
command_optstring[pos++] = cmds[active].options[i].op;
switch (cmds[active].options[i].arg)
{
case optional_argument:
command_optstring[pos++] = ':';
/* FALL */
case required_argument:
command_optstring[pos++] = ':';
/* FALL */
case no_argument:
default:
break;
}
} }
} }
} }
@@ -86,6 +106,7 @@ void command_register(command_t command)
int command_usage(char *error) int command_usage(char *error)
{ {
FILE *out = stdout; FILE *out = stdout;
char buf[64];
int i; int i;
if (error) if (error)
@@ -99,7 +120,8 @@ int command_usage(char *error)
{ {
for (i = 0; cmds[i].cmd; i++) for (i = 0; cmds[i].cmd; i++)
{ {
fprintf(out, " pki --%-6s %s\n", cmds[i].cmd, cmds[i].description); snprintf(buf, sizeof(buf), "--%s (-%c)", cmds[i].cmd, cmds[i].op);
fprintf(out, " pki %-14s %s\n", buf, cmds[i].description);
} }
} }
else else
@@ -118,8 +140,10 @@ int command_usage(char *error)
} }
for (i = 0; cmds[active].options[i].name; i++) for (i = 0; cmds[active].options[i].name; i++)
{ {
fprintf(out, " --%-8s %s\n", snprintf(buf, sizeof(buf), "--%s (-%c)",
cmds[active].options[i].name, cmds[active].options[i].desc); cmds[active].options[i].name, cmds[active].options[i].op);
fprintf(out, " %-15s %s\n",
buf, cmds[active].options[i].desc);
} }
} }
return error != NULL; return error != NULL;
@@ -145,7 +169,7 @@ int command_dispatch(int argc, char *argv[])
command_register((command_t){help, 'h', "help", "show usage information"}); command_register((command_t){help, 'h', "help", "show usage information"});
build_opts(); build_opts();
op = getopt_long(argc, argv, "", command_opts, NULL); op = getopt_long(argc, argv, command_optstring, command_opts, NULL);
for (i = 0; cmds[i].cmd; i++) for (i = 0; cmds[i].cmd; i++)
{ {
if (cmds[i].op == op) if (cmds[i].op == op)
+5
View File
@@ -74,6 +74,11 @@ struct command_t {
*/ */
extern struct option command_opts[]; extern struct option command_opts[];
/**
* Short option string of the active command.
*/
extern char command_optstring[];
/** /**
* Register a command. * Register a command.
*/ */
+1 -1
View File
@@ -28,7 +28,7 @@ static int gen(int argc, char *argv[])
while (TRUE) while (TRUE)
{ {
switch (getopt_long(argc, argv, "", command_opts, NULL)) switch (getopt_long(argc, argv, command_optstring, command_opts, NULL))
{ {
case 'h': case 'h':
return command_usage(NULL); return command_usage(NULL);
+1 -1
View File
@@ -53,7 +53,7 @@ static int issue(int argc, char *argv[])
while (TRUE) while (TRUE)
{ {
switch (getopt_long(argc, argv, "", command_opts, NULL)) switch (getopt_long(argc, argv, command_optstring, command_opts, NULL))
{ {
case 'h': case 'h':
goto usage; goto usage;
+1 -1
View File
@@ -34,7 +34,7 @@ static int keyid(int argc, char *argv[])
while (TRUE) while (TRUE)
{ {
switch (getopt_long(argc, argv, "", command_opts, NULL)) switch (getopt_long(argc, argv, command_optstring, command_opts, NULL))
{ {
case 'h': case 'h':
return command_usage(NULL); return command_usage(NULL);
+1 -1
View File
@@ -35,7 +35,7 @@ static int pub(int argc, char *argv[])
while (TRUE) while (TRUE)
{ {
switch (getopt_long(argc, argv, "", command_opts, NULL)) switch (getopt_long(argc, argv, command_optstring, command_opts, NULL))
{ {
case 'h': case 'h':
return command_usage(NULL); return command_usage(NULL);
+1 -1
View File
@@ -48,7 +48,7 @@ static int self(int argc, char *argv[])
while (TRUE) while (TRUE)
{ {
switch (getopt_long(argc, argv, "", command_opts, NULL)) switch (getopt_long(argc, argv, command_optstring, command_opts, NULL))
{ {
case 'h': case 'h':
goto usage; goto usage;
+1 -1
View File
@@ -29,7 +29,7 @@ static int verify(int argc, char *argv[])
while (TRUE) while (TRUE)
{ {
switch (getopt_long(argc, argv, "", command_opts, NULL)) switch (getopt_long(argc, argv, command_optstring, command_opts, NULL))
{ {
case 'h': case 'h':
return command_usage(NULL); return command_usage(NULL);