276da39af9
Approved by: roberto, delphij Security: VuXML: 0d0f3050-1f69-11e5-9ba9-d050996490d0 Security: http://bugs.ntp.org/show_bug.cgi?id=2853 Security: https://www.kb.cert.org/vuls/id/668167 Security: http://support.ntp.org/bin/view/Main/SecurityNotice#June_2015_NTP_Security_Vulnerabi
268 lines
7.2 KiB
C
268 lines
7.2 KiB
C
|
|
/**
|
|
* \file environment.c
|
|
*
|
|
* This file contains all of the routines that must be linked into
|
|
* an executable to use the generated option processing. The optional
|
|
* routines are in separately compiled modules so that they will not
|
|
* necessarily be linked in.
|
|
*
|
|
* @addtogroup autoopts
|
|
* @{
|
|
*/
|
|
/*
|
|
* This file is part of AutoOpts, a companion to AutoGen.
|
|
* AutoOpts is free software.
|
|
* AutoOpts is Copyright (C) 1992-2015 by Bruce Korb - all rights reserved
|
|
*
|
|
* AutoOpts is available under any one of two licenses. The license
|
|
* in use must be one of these two and the choice is under the control
|
|
* of the user of the license.
|
|
*
|
|
* The GNU Lesser General Public License, version 3 or later
|
|
* See the files "COPYING.lgplv3" and "COPYING.gplv3"
|
|
*
|
|
* The Modified Berkeley Software Distribution License
|
|
* See the file "COPYING.mbsd"
|
|
*
|
|
* These files have the following sha256 sums:
|
|
*
|
|
* 8584710e9b04216a394078dc156b781d0b47e1729104d666658aecef8ee32e95 COPYING.gplv3
|
|
* 4379e7444a0e2ce2b12dd6f5a52a27a4d02d39d247901d3285c88cf0d37f477b COPYING.lgplv3
|
|
* 13aa749a5b0a454917a944ed8fffc530b784f5ead522b1aacaf4ec8aa55a6239 COPYING.mbsd
|
|
*/
|
|
|
|
/* = = = START-STATIC-FORWARD = = = */
|
|
static void
|
|
do_env_opt(tOptState * os, char * env_name,
|
|
tOptions * pOpts, teEnvPresetType type);
|
|
/* = = = END-STATIC-FORWARD = = = */
|
|
|
|
/*
|
|
* doPrognameEnv - check for preset values from the ${PROGNAME}
|
|
* environment variable. This is accomplished by parsing the text into
|
|
* tokens, temporarily replacing the arg vector and calling
|
|
* immediate_opts and/or regular_opts.
|
|
*/
|
|
LOCAL void
|
|
doPrognameEnv(tOptions * pOpts, teEnvPresetType type)
|
|
{
|
|
char const * env_opts = getenv(pOpts->pzPROGNAME);
|
|
token_list_t * pTL;
|
|
int sv_argc;
|
|
proc_state_mask_t sv_flag;
|
|
char ** sv_argv;
|
|
|
|
/*
|
|
* No such beast? Then bail now.
|
|
*/
|
|
if (env_opts == NULL)
|
|
return;
|
|
|
|
/*
|
|
* Tokenize the string. If there's nothing of interest, we'll bail
|
|
* here immediately.
|
|
*/
|
|
pTL = ao_string_tokenize(env_opts);
|
|
if (pTL == NULL)
|
|
return;
|
|
|
|
/*
|
|
* Substitute our $PROGNAME argument list for the real one
|
|
*/
|
|
sv_argc = (int)pOpts->origArgCt;
|
|
sv_argv = pOpts->origArgVect;
|
|
sv_flag = pOpts->fOptSet;
|
|
|
|
/*
|
|
* We add a bogus pointer to the start of the list. The program name
|
|
* has already been pulled from "argv", so it won't get dereferenced.
|
|
* The option scanning code will skip the "program name" at the start
|
|
* of this list of tokens, so we accommodate this way ....
|
|
*/
|
|
{
|
|
uintptr_t v = (uintptr_t)(pTL->tkn_list);
|
|
pOpts->origArgVect = VOIDP(v - sizeof(char *));
|
|
}
|
|
pOpts->origArgCt = (unsigned int)pTL->tkn_ct + 1;
|
|
pOpts->fOptSet &= ~OPTPROC_ERRSTOP;
|
|
|
|
pOpts->curOptIdx = 1;
|
|
pOpts->pzCurOpt = NULL;
|
|
|
|
switch (type) {
|
|
case ENV_IMM:
|
|
(void)immediate_opts(pOpts);
|
|
break;
|
|
|
|
case ENV_ALL:
|
|
(void)immediate_opts(pOpts);
|
|
pOpts->curOptIdx = 1;
|
|
pOpts->pzCurOpt = NULL;
|
|
/* FALLTHROUGH */
|
|
|
|
case ENV_NON_IMM:
|
|
(void)regular_opts(pOpts);
|
|
}
|
|
|
|
/*
|
|
* Free up the temporary arg vector and restore the original program args.
|
|
*/
|
|
free(pTL);
|
|
pOpts->origArgVect = sv_argv;
|
|
pOpts->origArgCt = (unsigned int)sv_argc;
|
|
pOpts->fOptSet = sv_flag;
|
|
}
|
|
|
|
static void
|
|
do_env_opt(tOptState * os, char * env_name,
|
|
tOptions * pOpts, teEnvPresetType type)
|
|
{
|
|
os->pzOptArg = getenv(env_name);
|
|
if (os->pzOptArg == NULL)
|
|
return;
|
|
|
|
os->flags = OPTST_PRESET | OPTST_ALLOC_ARG | os->pOD->fOptState;
|
|
os->optType = TOPT_UNDEFINED;
|
|
|
|
if ( (os->pOD->pz_DisablePfx != NULL)
|
|
&& (streqvcmp(os->pzOptArg, os->pOD->pz_DisablePfx) == 0)) {
|
|
os->flags |= OPTST_DISABLED;
|
|
os->pzOptArg = NULL;
|
|
handle_opt(pOpts, os);
|
|
return;
|
|
}
|
|
|
|
switch (type) {
|
|
case ENV_IMM:
|
|
/*
|
|
* Process only immediate actions
|
|
*/
|
|
if (DO_IMMEDIATELY(os->flags))
|
|
break;
|
|
return;
|
|
|
|
case ENV_NON_IMM:
|
|
/*
|
|
* Process only NON immediate actions
|
|
*/
|
|
if (DO_NORMALLY(os->flags) || DO_SECOND_TIME(os->flags))
|
|
break;
|
|
return;
|
|
|
|
default: /* process everything */
|
|
break;
|
|
}
|
|
|
|
/*
|
|
* Make sure the option value string is persistent and consistent.
|
|
*
|
|
* The interpretation of the option value depends
|
|
* on the type of value argument the option takes
|
|
*/
|
|
if (OPTST_GET_ARGTYPE(os->pOD->fOptState) == OPARG_TYPE_NONE) {
|
|
/*
|
|
* Ignore any value.
|
|
*/
|
|
os->pzOptArg = NULL;
|
|
|
|
} else if (os->pzOptArg[0] == NUL) {
|
|
/*
|
|
* If the argument is the empty string and the argument is
|
|
* optional, then treat it as if the option was not specified.
|
|
*/
|
|
if ((os->pOD->fOptState & OPTST_ARG_OPTIONAL) == 0)
|
|
return;
|
|
os->pzOptArg = NULL;
|
|
|
|
} else {
|
|
AGDUPSTR(os->pzOptArg, os->pzOptArg, "option argument");
|
|
os->flags |= OPTST_ALLOC_ARG;
|
|
}
|
|
|
|
handle_opt(pOpts, os);
|
|
}
|
|
|
|
/*
|
|
* env_presets - check for preset values from the envrionment
|
|
* This routine should process in all, immediate or normal modes....
|
|
*/
|
|
LOCAL void
|
|
env_presets(tOptions * pOpts, teEnvPresetType type)
|
|
{
|
|
int ct;
|
|
tOptState st;
|
|
char * pzFlagName;
|
|
size_t spaceLeft;
|
|
char zEnvName[ AO_NAME_SIZE ];
|
|
|
|
/*
|
|
* Finally, see if we are to look at the environment
|
|
* variables for initial values.
|
|
*/
|
|
if ((pOpts->fOptSet & OPTPROC_ENVIRON) == 0)
|
|
return;
|
|
|
|
doPrognameEnv(pOpts, type);
|
|
|
|
ct = pOpts->presetOptCt;
|
|
st.pOD = pOpts->pOptDesc;
|
|
|
|
pzFlagName = zEnvName
|
|
+ snprintf(zEnvName, sizeof(zEnvName), "%s_", pOpts->pzPROGNAME);
|
|
spaceLeft = AO_NAME_SIZE - (unsigned long)(pzFlagName - zEnvName) - 1;
|
|
|
|
for (;ct-- > 0; st.pOD++) {
|
|
size_t nln;
|
|
|
|
/*
|
|
* If presetting is disallowed, then skip this entry
|
|
*/
|
|
if ( ((st.pOD->fOptState & OPTST_NO_INIT) != 0)
|
|
|| (st.pOD->optEquivIndex != NO_EQUIVALENT) )
|
|
continue;
|
|
|
|
/*
|
|
* IF there is no such environment variable,
|
|
* THEN skip this entry, too.
|
|
*/
|
|
nln = strlen(st.pOD->pz_NAME) + 1;
|
|
if (nln <= spaceLeft) {
|
|
/*
|
|
* Set up the option state
|
|
*/
|
|
memcpy(pzFlagName, st.pOD->pz_NAME, nln);
|
|
do_env_opt(&st, zEnvName, pOpts, type);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Special handling for ${PROGNAME_LOAD_OPTS}
|
|
*/
|
|
if ( (pOpts->specOptIdx.save_opts != NO_EQUIVALENT)
|
|
&& (pOpts->specOptIdx.save_opts != 0)) {
|
|
size_t nln;
|
|
st.pOD = pOpts->pOptDesc + pOpts->specOptIdx.save_opts + 1;
|
|
|
|
if (st.pOD->pz_NAME == NULL)
|
|
return;
|
|
|
|
nln = strlen(st.pOD->pz_NAME) + 1;
|
|
|
|
if (nln > spaceLeft)
|
|
return;
|
|
|
|
memcpy(pzFlagName, st.pOD->pz_NAME, nln);
|
|
do_env_opt(&st, zEnvName, pOpts, type);
|
|
}
|
|
}
|
|
|
|
/** @}
|
|
*
|
|
* Local Variables:
|
|
* mode: C
|
|
* c-file-style: "stroustrup"
|
|
* indent-tabs-mode: nil
|
|
* End:
|
|
* end of autoopts/environment.c */
|