118 lines
2.3 KiB
C
118 lines
2.3 KiB
C
#include "common-static.h"
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "option.h"
|
|
|
|
int option_flag (void *data, _unused_ int argc, _unused_ char **argv)
|
|
{
|
|
const int one = 1;
|
|
byte_cpy(data, &one, sizeof(one));
|
|
|
|
return 0;
|
|
}
|
|
|
|
int option_str (void *data, int argc, char **argv)
|
|
{
|
|
if (argc<2 || !argv[1]) {
|
|
printf("option `%s' need a string argument\n", argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
byte_cpy(data, &argv[1], sizeof(argv[1]));
|
|
|
|
return 1;
|
|
}
|
|
|
|
int option_long (void *data, int argc, char **argv)
|
|
{
|
|
if (argc<2 || !argv[1]) {
|
|
printf("option `%s' need an integer argument\n", argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
errno = 0;
|
|
char *end;
|
|
long val = strtol(argv[1], &end, 0);
|
|
|
|
if (errno || argv[1]==end) {
|
|
printf("argument `%s' is not a valid integer\n", argv[1]);
|
|
return -1;
|
|
}
|
|
|
|
byte_cpy(data, &val, sizeof(val));
|
|
|
|
return 1;
|
|
}
|
|
|
|
int option_option (void *data, int argc, char **argv)
|
|
{
|
|
struct option *opts = (struct option *)data;
|
|
|
|
for (int i=1; i<argc; i++) {
|
|
int found = 0;
|
|
|
|
for (int k=0; opts[k].name; k++) {
|
|
if (str_cmp(opts[k].name, argv[i]))
|
|
continue;
|
|
|
|
int ret = opts[k].call(opts[k].data, argc-i, &argv[i]);
|
|
|
|
if (ret<0)
|
|
return -1;
|
|
|
|
i += ret;
|
|
found = 1;
|
|
break;
|
|
}
|
|
|
|
if (!found)
|
|
return i-1;
|
|
}
|
|
|
|
return argc;
|
|
}
|
|
|
|
static void option_usage (struct option *opts, char *name)
|
|
{
|
|
char *usage = "usage: ";
|
|
size_t slen = str_len(usage)+str_len(name);
|
|
size_t len = slen;
|
|
|
|
printf("%s%s", usage, name);
|
|
|
|
if (slen>40)
|
|
slen = 12;
|
|
|
|
for (int k=0; opts[k].name; k++) {
|
|
int isflag = opts[k].call==option_flag;
|
|
size_t inc = str_len(opts[k].name)+(isflag?0:4)+4;
|
|
|
|
if (len+inc>60) {
|
|
printf("\n%*s", (int)slen, "");
|
|
len = 0;
|
|
}
|
|
printf(" [%s%s]", opts[k].name, isflag?"":" ARG");
|
|
len += inc;
|
|
}
|
|
|
|
printf("\n");
|
|
}
|
|
|
|
int option (struct option *opts, int argc, char **argv)
|
|
{
|
|
int ret = option_option(opts, argc, argv);
|
|
|
|
if (ret==argc)
|
|
return 0;
|
|
|
|
if (ret<0 || ret+1>=argc)
|
|
return 1;
|
|
|
|
printf("option `%s' is unknown\n", argv[ret+1]);
|
|
option_usage(opts, argv[0]);
|
|
|
|
return 1;
|
|
}
|