b48bed5f42
Tests that start daemons have to "wait" for them to start. This commit makes Heimdal daemons prep to detach (when requested) by forking early, then having the child signal readiness to the parent when the child really is ready. The parent exits only which the child is ready. This means that tests will no longer need to wait for daemons. However, tests will still need a pidfile or such so they can stop the daemons. Note that the --detach options should not be used on OS X from launchd, only from tests.
608 lines
15 KiB
C
608 lines
15 KiB
C
/*
|
|
* Copyright (c) 1997 - 2002 Kungliga Tekniska Högskolan
|
|
* (Royal Institute of Technology, Stockholm, Sweden).
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* 3. Neither the name of the Institute nor the names of its contributors
|
|
* may be used to endorse or promote products derived from this software
|
|
* without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
* SUCH DAMAGE.
|
|
*/
|
|
|
|
#include <config.h>
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include "roken.h"
|
|
#include "getarg.h"
|
|
|
|
#define ISFLAG(X) ((X).type == arg_flag || (X).type == arg_negative_flag)
|
|
|
|
static size_t
|
|
print_arg (char *string,
|
|
size_t len,
|
|
int mdoc,
|
|
int longp,
|
|
struct getargs *arg,
|
|
char *(i18n)(const char *))
|
|
{
|
|
const char *s;
|
|
|
|
*string = '\0';
|
|
|
|
if (ISFLAG(*arg) || (!longp && arg->type == arg_counter))
|
|
return 0;
|
|
|
|
if(mdoc){
|
|
if(longp)
|
|
strlcat(string, "= Ns", len);
|
|
strlcat(string, " Ar ", len);
|
|
} else {
|
|
if (longp)
|
|
strlcat (string, "=", len);
|
|
else
|
|
strlcat (string, " ", len);
|
|
}
|
|
|
|
if (arg->arg_help)
|
|
s = (*i18n)(arg->arg_help);
|
|
else if (arg->type == arg_integer || arg->type == arg_counter)
|
|
s = "integer";
|
|
else if (arg->type == arg_string)
|
|
s = "string";
|
|
else if (arg->type == arg_strings)
|
|
s = "strings";
|
|
else if (arg->type == arg_double)
|
|
s = "float";
|
|
else
|
|
s = "<undefined>";
|
|
|
|
strlcat(string, s, len);
|
|
return 1 + strlen(s);
|
|
}
|
|
|
|
static void
|
|
mandoc_template(struct getargs *args,
|
|
size_t num_args,
|
|
const char *progname,
|
|
const char *extra_string,
|
|
char *(i18n)(const char *))
|
|
{
|
|
size_t i;
|
|
char timestr[64], cmd[64];
|
|
char buf[128];
|
|
const char *p;
|
|
time_t t;
|
|
|
|
printf(".\\\" Things to fix:\n");
|
|
printf(".\\\" * correct section, and operating system\n");
|
|
printf(".\\\" * remove Op from mandatory flags\n");
|
|
printf(".\\\" * use better macros for arguments (like .Pa for files)\n");
|
|
printf(".\\\"\n");
|
|
t = time(NULL);
|
|
strftime(timestr, sizeof(timestr), "%B %e, %Y", localtime(&t));
|
|
printf(".Dd %s\n", timestr);
|
|
p = strrchr(progname, '/');
|
|
if(p) p++; else p = progname;
|
|
strlcpy(cmd, p, sizeof(cmd));
|
|
strupr(cmd);
|
|
|
|
printf(".Dt %s SECTION\n", cmd);
|
|
printf(".Os OPERATING_SYSTEM\n");
|
|
printf(".Sh NAME\n");
|
|
printf(".Nm %s\n", p);
|
|
printf(".Nd in search of a description\n");
|
|
printf(".Sh SYNOPSIS\n");
|
|
printf(".Nm\n");
|
|
for(i = 0; i < num_args; i++){
|
|
/* we seem to hit a limit on number of arguments if doing
|
|
short and long flags with arguments -- split on two lines */
|
|
if(ISFLAG(args[i]) ||
|
|
args[i].short_name == 0 || args[i].long_name == NULL) {
|
|
printf(".Op ");
|
|
|
|
if(args[i].short_name) {
|
|
print_arg(buf, sizeof(buf), 1, 0, args + i, i18n);
|
|
printf("Fl %c%s", args[i].short_name, buf);
|
|
if(args[i].long_name)
|
|
printf(" | ");
|
|
}
|
|
if(args[i].long_name) {
|
|
print_arg(buf, sizeof(buf), 1, 1, args + i, i18n);
|
|
printf("Fl Fl %s%s%s",
|
|
args[i].type == arg_negative_flag ? "no-" : "",
|
|
args[i].long_name, buf);
|
|
}
|
|
printf("\n");
|
|
} else {
|
|
print_arg(buf, sizeof(buf), 1, 0, args + i, i18n);
|
|
printf(".Oo Fl %c%s \\*(Ba Xo\n", args[i].short_name, buf);
|
|
print_arg(buf, sizeof(buf), 1, 1, args + i, i18n);
|
|
printf(".Fl Fl %s%s\n.Xc\n.Oc\n", args[i].long_name, buf);
|
|
}
|
|
/*
|
|
if(args[i].type == arg_strings)
|
|
fprintf (stderr, "...");
|
|
*/
|
|
}
|
|
if (extra_string && *extra_string)
|
|
printf (".Ar %s\n", extra_string);
|
|
printf(".Sh DESCRIPTION\n");
|
|
printf("Supported options:\n");
|
|
printf(".Bl -tag -width Ds\n");
|
|
for(i = 0; i < num_args; i++){
|
|
printf(".It Xo\n");
|
|
if(args[i].short_name){
|
|
printf(".Fl %c", args[i].short_name);
|
|
print_arg(buf, sizeof(buf), 1, 0, args + i, i18n);
|
|
printf("%s", buf);
|
|
if(args[i].long_name)
|
|
printf(" ,");
|
|
printf("\n");
|
|
}
|
|
if(args[i].long_name){
|
|
printf(".Fl Fl %s%s",
|
|
args[i].type == arg_negative_flag ? "no-" : "",
|
|
args[i].long_name);
|
|
print_arg(buf, sizeof(buf), 1, 1, args + i, i18n);
|
|
printf("%s\n", buf);
|
|
}
|
|
printf(".Xc\n");
|
|
if(args[i].help)
|
|
printf("%s\n", args[i].help);
|
|
/*
|
|
if(args[i].type == arg_strings)
|
|
fprintf (stderr, "...");
|
|
*/
|
|
}
|
|
printf(".El\n");
|
|
printf(".\\\".Sh ENVIRONMENT\n");
|
|
printf(".\\\".Sh FILES\n");
|
|
printf(".\\\".Sh EXAMPLES\n");
|
|
printf(".\\\".Sh DIAGNOSTICS\n");
|
|
printf(".\\\".Sh SEE ALSO\n");
|
|
printf(".\\\".Sh STANDARDS\n");
|
|
printf(".\\\".Sh HISTORY\n");
|
|
printf(".\\\".Sh AUTHORS\n");
|
|
printf(".\\\".Sh BUGS\n");
|
|
}
|
|
|
|
static int
|
|
check_column(FILE *f, int col, int len, int columns)
|
|
{
|
|
if(col + len > columns) {
|
|
fprintf(f, "\n");
|
|
col = fprintf(f, " ");
|
|
}
|
|
return col;
|
|
}
|
|
|
|
static char *
|
|
builtin_i18n(const char *str)
|
|
{
|
|
return rk_UNCONST(str);
|
|
}
|
|
|
|
ROKEN_LIB_FUNCTION void ROKEN_LIB_CALL
|
|
arg_printusage (struct getargs *args,
|
|
size_t num_args,
|
|
const char *progname,
|
|
const char *extra_string)
|
|
{
|
|
arg_printusage_i18n(args, num_args, "Usage",
|
|
progname, extra_string, builtin_i18n);
|
|
}
|
|
|
|
ROKEN_LIB_FUNCTION void ROKEN_LIB_CALL
|
|
arg_printusage_i18n (struct getargs *args,
|
|
size_t num_args,
|
|
const char *usage,
|
|
const char *progname,
|
|
const char *extra_string,
|
|
char *(*i18n)(const char *))
|
|
{
|
|
size_t i, max_len = 0;
|
|
char buf[128];
|
|
int col = 0, columns;
|
|
|
|
if (progname == NULL)
|
|
progname = getprogname();
|
|
|
|
if (i18n == NULL)
|
|
i18n = builtin_i18n;
|
|
|
|
if(getenv("GETARGMANDOC")){
|
|
mandoc_template(args, num_args, progname, extra_string, i18n);
|
|
return;
|
|
}
|
|
if(get_window_size(2, NULL, &columns) == -1)
|
|
columns = 80;
|
|
col = 0;
|
|
col += fprintf (stderr, "%s: %s", usage, progname);
|
|
buf[0] = '\0';
|
|
for (i = 0; i < num_args; ++i) {
|
|
if(args[i].short_name && ISFLAG(args[i])) {
|
|
char s[2];
|
|
if(buf[0] == '\0')
|
|
strlcpy(buf, "[-", sizeof(buf));
|
|
s[0] = args[i].short_name;
|
|
s[1] = '\0';
|
|
strlcat(buf, s, sizeof(buf));
|
|
}
|
|
}
|
|
if(buf[0] != '\0') {
|
|
strlcat(buf, "]", sizeof(buf));
|
|
col = check_column(stderr, col, strlen(buf) + 1, columns);
|
|
col += fprintf(stderr, " %s", buf);
|
|
}
|
|
|
|
for (i = 0; i < num_args; ++i) {
|
|
size_t len = 0;
|
|
|
|
if (args[i].long_name) {
|
|
buf[0] = '\0';
|
|
strlcat(buf, "[--", sizeof(buf));
|
|
len += 2;
|
|
if(args[i].type == arg_negative_flag) {
|
|
strlcat(buf, "no-", sizeof(buf));
|
|
len += 3;
|
|
}
|
|
strlcat(buf, args[i].long_name, sizeof(buf));
|
|
len += strlen(args[i].long_name);
|
|
len += print_arg(buf + strlen(buf), sizeof(buf) - strlen(buf),
|
|
0, 1, &args[i], i18n);
|
|
strlcat(buf, "]", sizeof(buf));
|
|
if(args[i].type == arg_strings)
|
|
strlcat(buf, "...", sizeof(buf));
|
|
col = check_column(stderr, col, strlen(buf) + 1, columns);
|
|
col += fprintf(stderr, " %s", buf);
|
|
}
|
|
if (args[i].short_name && !ISFLAG(args[i])) {
|
|
snprintf(buf, sizeof(buf), "[-%c", args[i].short_name);
|
|
len += 2;
|
|
len += print_arg(buf + strlen(buf), sizeof(buf) - strlen(buf),
|
|
0, 0, &args[i], i18n);
|
|
strlcat(buf, "]", sizeof(buf));
|
|
if(args[i].type == arg_strings)
|
|
strlcat(buf, "...", sizeof(buf));
|
|
col = check_column(stderr, col, strlen(buf) + 1, columns);
|
|
col += fprintf(stderr, " %s", buf);
|
|
}
|
|
if (args[i].long_name && args[i].short_name)
|
|
len += 2; /* ", " */
|
|
max_len = max(max_len, len);
|
|
}
|
|
if (extra_string) {
|
|
check_column(stderr, col, strlen(extra_string) + 1, columns);
|
|
fprintf (stderr, " %s\n", extra_string);
|
|
} else
|
|
fprintf (stderr, "\n");
|
|
for (i = 0; i < num_args; ++i) {
|
|
if (args[i].help) {
|
|
size_t count = 0;
|
|
|
|
if (args[i].short_name) {
|
|
count += fprintf (stderr, "-%c", args[i].short_name);
|
|
print_arg (buf, sizeof(buf), 0, 0, &args[i], i18n);
|
|
count += fprintf(stderr, "%s", buf);
|
|
}
|
|
if (args[i].short_name && args[i].long_name)
|
|
count += fprintf (stderr, ", ");
|
|
if (args[i].long_name) {
|
|
count += fprintf (stderr, "--");
|
|
if (args[i].type == arg_negative_flag)
|
|
count += fprintf (stderr, "no-");
|
|
count += fprintf (stderr, "%s", args[i].long_name);
|
|
print_arg (buf, sizeof(buf), 0, 1, &args[i], i18n);
|
|
count += fprintf(stderr, "%s", buf);
|
|
}
|
|
while(count++ <= max_len)
|
|
putc (' ', stderr);
|
|
fprintf (stderr, "%s\n", (*i18n)(args[i].help));
|
|
}
|
|
}
|
|
}
|
|
|
|
static int
|
|
add_string(getarg_strings *s, char *value)
|
|
{
|
|
char **strings;
|
|
|
|
strings = realloc(s->strings, (s->num_strings + 1) * sizeof(*s->strings));
|
|
if (strings == NULL) {
|
|
free(s->strings);
|
|
s->strings = NULL;
|
|
s->num_strings = 0;
|
|
return ENOMEM;
|
|
}
|
|
s->strings = strings;
|
|
s->strings[s->num_strings] = value;
|
|
s->num_strings++;
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
arg_match_long(struct getargs *args, size_t num_args,
|
|
char *argv, int argc, char **rargv, int *goptind)
|
|
{
|
|
size_t i;
|
|
char *goptarg = NULL;
|
|
int negate = 0;
|
|
int partial_match = 0;
|
|
struct getargs *partial = NULL;
|
|
struct getargs *current = NULL;
|
|
int argv_len;
|
|
char *p;
|
|
int p_len;
|
|
|
|
argv_len = strlen(argv);
|
|
p = strchr (argv, '=');
|
|
if (p != NULL)
|
|
argv_len = p - argv;
|
|
|
|
for (i = 0; i < num_args; ++i) {
|
|
if(args[i].long_name) {
|
|
int len = strlen(args[i].long_name);
|
|
p = argv;
|
|
p_len = argv_len;
|
|
negate = 0;
|
|
|
|
for (;;) {
|
|
if (strncmp (args[i].long_name, p, p_len) == 0) {
|
|
if(p_len == len)
|
|
current = &args[i];
|
|
else {
|
|
++partial_match;
|
|
partial = &args[i];
|
|
}
|
|
goptarg = p + p_len;
|
|
} else if (ISFLAG(args[i]) && strncmp (p, "no-", 3) == 0) {
|
|
negate = !negate;
|
|
p += 3;
|
|
p_len -= 3;
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
if (current)
|
|
break;
|
|
}
|
|
}
|
|
if (current == NULL) {
|
|
if (partial_match == 1)
|
|
current = partial;
|
|
else
|
|
return ARG_ERR_NO_MATCH;
|
|
}
|
|
|
|
if(*goptarg == '\0'
|
|
&& !ISFLAG(*current)
|
|
&& current->type != arg_collect
|
|
&& current->type != arg_counter)
|
|
return ARG_ERR_NO_MATCH;
|
|
switch(current->type){
|
|
case arg_integer:
|
|
{
|
|
int tmp;
|
|
if(sscanf(goptarg + 1, "%d", &tmp) != 1)
|
|
return ARG_ERR_BAD_ARG;
|
|
*(int*)current->value = tmp;
|
|
return 0;
|
|
}
|
|
case arg_string:
|
|
{
|
|
*(char**)current->value = goptarg + 1;
|
|
return 0;
|
|
}
|
|
case arg_strings:
|
|
{
|
|
return add_string((getarg_strings*)current->value, goptarg + 1);
|
|
}
|
|
case arg_flag:
|
|
case arg_negative_flag:
|
|
{
|
|
int *flag = current->value;
|
|
if(*goptarg == '\0' ||
|
|
strcmp(goptarg + 1, "yes") == 0 ||
|
|
strcmp(goptarg + 1, "true") == 0){
|
|
*flag = !negate;
|
|
return 0;
|
|
} else if (*goptarg && strcmp(goptarg + 1, "maybe") == 0) {
|
|
*flag = rk_random() & 1;
|
|
} else {
|
|
*flag = negate;
|
|
return 0;
|
|
}
|
|
return ARG_ERR_BAD_ARG;
|
|
}
|
|
case arg_counter :
|
|
{
|
|
int val;
|
|
|
|
if (*goptarg == '\0')
|
|
val = 1;
|
|
else if(sscanf(goptarg + 1, "%d", &val) != 1)
|
|
return ARG_ERR_BAD_ARG;
|
|
*(int *)current->value += val;
|
|
return 0;
|
|
}
|
|
case arg_double:
|
|
{
|
|
double tmp;
|
|
if(sscanf(goptarg + 1, "%lf", &tmp) != 1)
|
|
return ARG_ERR_BAD_ARG;
|
|
*(double*)current->value = tmp;
|
|
return 0;
|
|
}
|
|
case arg_collect:{
|
|
struct getarg_collect_info *c = current->value;
|
|
int o = argv - rargv[*goptind];
|
|
return (*c->func)(FALSE, argc, rargv, goptind, &o, c->data);
|
|
}
|
|
|
|
default:
|
|
abort ();
|
|
UNREACHABLE(return 0);
|
|
}
|
|
}
|
|
|
|
static int
|
|
arg_match_short (struct getargs *args, size_t num_args,
|
|
char *argv, int argc, char **rargv, int *goptind)
|
|
{
|
|
size_t j, k;
|
|
|
|
for(j = 1; j > 0 && j < strlen(rargv[*goptind]); j++) {
|
|
for(k = 0; k < num_args; k++) {
|
|
char *goptarg;
|
|
|
|
if(args[k].short_name == 0)
|
|
continue;
|
|
if(argv[j] == args[k].short_name) {
|
|
if(args[k].type == arg_flag) {
|
|
*(int*)args[k].value = 1;
|
|
break;
|
|
}
|
|
if(args[k].type == arg_negative_flag) {
|
|
*(int*)args[k].value = 0;
|
|
break;
|
|
}
|
|
if(args[k].type == arg_counter) {
|
|
++*(int *)args[k].value;
|
|
break;
|
|
}
|
|
if(args[k].type == arg_collect) {
|
|
struct getarg_collect_info *c = args[k].value;
|
|
int a = (int)j;
|
|
|
|
if((*c->func)(TRUE, argc, rargv, goptind, &a, c->data))
|
|
return ARG_ERR_BAD_ARG;
|
|
j = a;
|
|
break;
|
|
}
|
|
|
|
if(argv[j + 1])
|
|
goptarg = &argv[j + 1];
|
|
else {
|
|
++*goptind;
|
|
goptarg = rargv[*goptind];
|
|
}
|
|
if(goptarg == NULL) {
|
|
--*goptind;
|
|
return ARG_ERR_NO_ARG;
|
|
}
|
|
if(args[k].type == arg_integer) {
|
|
int tmp;
|
|
if(sscanf(goptarg, "%d", &tmp) != 1)
|
|
return ARG_ERR_BAD_ARG;
|
|
*(int*)args[k].value = tmp;
|
|
return 0;
|
|
} else if(args[k].type == arg_string) {
|
|
*(char**)args[k].value = goptarg;
|
|
return 0;
|
|
} else if(args[k].type == arg_strings) {
|
|
return add_string((getarg_strings*)args[k].value, goptarg);
|
|
} else if(args[k].type == arg_double) {
|
|
double tmp;
|
|
if(sscanf(goptarg, "%lf", &tmp) != 1)
|
|
return ARG_ERR_BAD_ARG;
|
|
*(double*)args[k].value = tmp;
|
|
return 0;
|
|
}
|
|
return ARG_ERR_BAD_ARG;
|
|
}
|
|
}
|
|
if (k == num_args)
|
|
return ARG_ERR_NO_MATCH;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
|
|
getarg(struct getargs *args, size_t num_args,
|
|
int argc, char **argv, int *goptind)
|
|
{
|
|
int i;
|
|
int ret = 0;
|
|
|
|
rk_random_init();
|
|
(*goptind)++;
|
|
for(i = *goptind; i < argc; i++) {
|
|
if(argv[i][0] != '-')
|
|
break;
|
|
if(argv[i][1] == '-'){
|
|
if(argv[i][2] == 0){
|
|
i++;
|
|
break;
|
|
}
|
|
ret = arg_match_long (args, num_args, argv[i] + 2,
|
|
argc, argv, &i);
|
|
} else {
|
|
ret = arg_match_short (args, num_args, argv[i],
|
|
argc, argv, &i);
|
|
}
|
|
if(ret)
|
|
break;
|
|
}
|
|
*goptind = i;
|
|
return ret;
|
|
}
|
|
|
|
ROKEN_LIB_FUNCTION void ROKEN_LIB_CALL
|
|
free_getarg_strings (getarg_strings *s)
|
|
{
|
|
free (s->strings);
|
|
}
|
|
|
|
#if TEST
|
|
int foo_flag = 2;
|
|
int flag1 = 0;
|
|
int flag2 = 0;
|
|
int bar_int;
|
|
char *baz_string;
|
|
|
|
struct getargs args[] = {
|
|
{ NULL, '1', arg_flag, &flag1, "one", NULL },
|
|
{ NULL, '2', arg_flag, &flag2, "two", NULL },
|
|
{ "foo", 'f', arg_negative_flag, &foo_flag, "foo", NULL },
|
|
{ "bar", 'b', arg_integer, &bar_int, "bar", "seconds"},
|
|
{ "baz", 'x', arg_string, &baz_string, "baz", "name" },
|
|
};
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int goptind = 0;
|
|
while (getarg(args, 5, argc, argv, &goptind))
|
|
printf("Bad arg: %s\n", argv[goptind]);
|
|
printf("flag1 = %d\n", flag1);
|
|
printf("flag2 = %d\n", flag2);
|
|
printf("foo_flag = %d\n", foo_flag);
|
|
printf("bar_int = %d\n", bar_int);
|
|
printf("baz_flag = %s\n", baz_string);
|
|
arg_printusage (args, 5, argv[0], "nothing here");
|
|
}
|
|
#endif
|