196b6346ba
setenv(3) by tracking the size of the memory allocated instead of using strlen() on the current value. Convert all calls to POSIX from historic BSD API: - unsetenv returns an int. - putenv takes a char * instead of const char *. - putenv no longer makes a copy of the input string. - errno is set appropriately for POSIX. Exceptions involve bad environ variable and internal initialization code. These both set errno to EFAULT. Several patches to base utilities to handle the POSIX changes from Andrey Chernov's previous commit. A few I re-wrote to use setenv() instead of putenv(). New regression module for tools/regression/environ to test these functions. It also can be used to test the performance. Bump __FreeBSD_version to 700050 due to API change. PR: kern/99826 Approved by: wes Approved by: re (kensmith)
330 lines
7.9 KiB
C
330 lines
7.9 KiB
C
/*
|
|
* The new sysinstall program.
|
|
*
|
|
* This is probably the last program in the `sysinstall' line - the next
|
|
* generation being essentially a complete rewrite.
|
|
*
|
|
* $FreeBSD$
|
|
*
|
|
* Copyright (c) 1995
|
|
* Jordan Hubbard. All rights reserved.
|
|
* Copyright (c) 2001
|
|
* Murray Stokely. 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,
|
|
* verbatim and that no modifications are made prior to this
|
|
* point in the file.
|
|
* 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.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY JORDAN HUBBARD ``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 JORDAN HUBBARD OR HIS PETS 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, LIFE 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 "sysinstall.h"
|
|
|
|
/* Routines for dealing with variable lists */
|
|
|
|
static void
|
|
make_variable(char *var, char *value, int dirty)
|
|
{
|
|
Variable *vp;
|
|
|
|
/* Trim leading and trailing whitespace */
|
|
var = string_skipwhite(string_prune(var));
|
|
|
|
if (!var || !*var)
|
|
return;
|
|
|
|
|
|
/* Now search to see if it's already in the list */
|
|
for (vp = VarHead; vp; vp = vp->next) {
|
|
if (!strcmp(vp->name, var)) {
|
|
if (vp->dirty && !dirty)
|
|
return;
|
|
setenv(var, value, 1);
|
|
free(vp->value);
|
|
vp->value = strdup(value);
|
|
if (dirty != -1)
|
|
vp->dirty = dirty;
|
|
return;
|
|
}
|
|
}
|
|
|
|
setenv(var, value, 1);
|
|
/* No? Create a new one */
|
|
vp = (Variable *)safe_malloc(sizeof(Variable));
|
|
vp->name = strdup(var);
|
|
vp->value = strdup(value);
|
|
if (dirty == -1)
|
|
dirty = 0;
|
|
vp->dirty = dirty;
|
|
vp->next = VarHead;
|
|
VarHead = vp;
|
|
}
|
|
|
|
void
|
|
variable_set(char *var, int dirty)
|
|
{
|
|
char tmp[1024], *cp;
|
|
|
|
if (!var)
|
|
msgFatal("NULL variable name & value passed.");
|
|
else if (!*var)
|
|
msgDebug("Warning: Zero length name & value passed to variable_set()\n");
|
|
SAFE_STRCPY(tmp, var);
|
|
if ((cp = index(tmp, '=')) == NULL)
|
|
msgFatal("Invalid variable format: %s", var);
|
|
*(cp++) = '\0';
|
|
make_variable(tmp, string_skipwhite(cp), dirty);
|
|
}
|
|
|
|
void
|
|
variable_set2(char *var, char *value, int dirty)
|
|
{
|
|
if (!var || !value)
|
|
msgFatal("Null name or value passed to set_variable2(%s) = %s!",
|
|
var ? var : "", value ? value : "");
|
|
else if (!*var || !*value)
|
|
msgDebug("Warning: Zero length name or value passed to variable_set2(%s) = %s\n",
|
|
var, value);
|
|
make_variable(var, value, dirty);
|
|
}
|
|
|
|
char *
|
|
variable_get(char *var)
|
|
{
|
|
return getenv(var);
|
|
}
|
|
|
|
int
|
|
variable_cmp(char *var, char *value)
|
|
{
|
|
char *val;
|
|
|
|
if ((val = variable_get(var)))
|
|
return strcmp(val, value);
|
|
return -1;
|
|
}
|
|
|
|
void
|
|
variable_unset(char *var)
|
|
{
|
|
Variable *vp;
|
|
char name[512], *cp;
|
|
|
|
if ((cp = index(var, '=')) != NULL)
|
|
sstrncpy(name, var, cp - var);
|
|
else
|
|
SAFE_STRCPY(name, var);
|
|
unsetenv(name);
|
|
/* Now search to see if it's in our list, if we have one.. */
|
|
if (!VarHead)
|
|
return;
|
|
else if (!VarHead->next && !strcmp(VarHead->name, name)) {
|
|
safe_free(VarHead->name);
|
|
safe_free(VarHead->value);
|
|
free(VarHead);
|
|
VarHead = NULL;
|
|
}
|
|
else {
|
|
for (vp = VarHead; vp; vp = vp->next) {
|
|
if (!strcmp(vp->name, name)) {
|
|
Variable *save = vp->next;
|
|
|
|
safe_free(vp->name);
|
|
safe_free(vp->value);
|
|
*vp = *save;
|
|
safe_free(save);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Prompt user for the name of a variable */
|
|
char *
|
|
variable_get_value(char *var, char *prompt, int dirty)
|
|
{
|
|
char *cp;
|
|
|
|
cp = variable_get(var);
|
|
if (cp && variable_get(VAR_NONINTERACTIVE))
|
|
return cp;
|
|
else if ((cp = msgGetInput(cp, "%s", prompt)) != NULL)
|
|
variable_set2(var, cp, dirty);
|
|
else
|
|
cp = NULL;
|
|
return cp;
|
|
}
|
|
|
|
/* Check if value passed in data (in the form "variable=value") is
|
|
* valid, and it's status compared to the value of variable stored in
|
|
* env
|
|
*
|
|
* Possible return values :
|
|
* -3: Invalid line, the data string is NOT set as an env variable
|
|
* -2: Invalid line, the data string is set as an env variable
|
|
* -1: Invalid line
|
|
* 0: Valid line, is NOT equal to env version
|
|
* 1: Valid line, is equal to env version
|
|
* 2: Valid line, value empty - e.g. foo=""
|
|
* 3: Valid line, does not exist in env
|
|
*/
|
|
int
|
|
variable_check2(char *data)
|
|
{
|
|
char *cp, *cp2, *cp3, tmp[256];
|
|
|
|
if (data == NULL)
|
|
return -1;
|
|
SAFE_STRCPY(tmp, data);
|
|
if ((cp = index(tmp, '=')) != NULL) {
|
|
*(cp++) = '\0';
|
|
if (*cp == '"') { /* smash quotes if present */
|
|
++cp;
|
|
if ((cp3 = index(cp, '"')) != NULL)
|
|
*cp3 = '\0';
|
|
}
|
|
else if ((cp3 = index(cp, ',')) != NULL)
|
|
*cp3 = '\0';
|
|
cp2 = variable_get(tmp);
|
|
if (cp2 != NULL) {
|
|
if (*cp == '\0')
|
|
return 2;
|
|
else
|
|
return strcmp(cp, cp2) == 0 ? 1 : 0;
|
|
}
|
|
else
|
|
return 3;
|
|
}
|
|
else
|
|
return variable_get(tmp) != NULL ? -2 : -3;
|
|
}
|
|
|
|
/* Check if the value passed in data (in the form "variable=value") is
|
|
equal to the value of variable stored in env */
|
|
int
|
|
variable_check(char *data)
|
|
{
|
|
int ret;
|
|
ret = variable_check2(data);
|
|
|
|
switch(ret) {
|
|
case -2:
|
|
case 1:
|
|
case 2:
|
|
return TRUE;
|
|
/* NOT REACHED */
|
|
default:
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
int
|
|
dump_variables(dialogMenuItem *unused)
|
|
{
|
|
FILE *fp;
|
|
Variable *vp;
|
|
|
|
if (isDebug())
|
|
msgDebug("Writing sysinstall variables to file..\n");
|
|
|
|
fp = fopen("/etc/sysinstall.vars", "w");
|
|
if (!fp) {
|
|
msgConfirm("Unable to write to /etc/sysinstall.vars: %s",
|
|
strerror(errno));
|
|
return DITEM_FAILURE;
|
|
}
|
|
|
|
for (vp = VarHead; vp; vp = vp->next)
|
|
fprintf(fp, "%s=\"%s\" (%d)\n", vp->name, vp->value, vp->dirty);
|
|
|
|
fclose(fp);
|
|
|
|
return DITEM_SUCCESS;
|
|
}
|
|
|
|
/* Free all of the variables, useful to really start over as when the
|
|
user selects "restart" from the interrupt menu. */
|
|
void
|
|
free_variables(void)
|
|
{
|
|
Variable *vp;
|
|
|
|
/* Free the variables from our list, if we have one.. */
|
|
if (!VarHead)
|
|
return;
|
|
else if (!VarHead->next) {
|
|
unsetenv(VarHead->name);
|
|
safe_free(VarHead->name);
|
|
safe_free(VarHead->value);
|
|
free(VarHead);
|
|
VarHead = NULL;
|
|
}
|
|
else {
|
|
for (vp = VarHead; vp; ) {
|
|
Variable *save = vp;
|
|
unsetenv(vp->name);
|
|
safe_free(vp->name);
|
|
safe_free(vp->value);
|
|
vp = vp->next;
|
|
safe_free(save);
|
|
}
|
|
VarHead = NULL;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Persistent variables. The variables modified by these functions
|
|
* are not cleared between invocations of sysinstall. This is useful
|
|
* to allow the user to completely restart sysinstall, without having
|
|
* it load all of the modules again from the installation media which
|
|
* are still in memory.
|
|
*/
|
|
|
|
void
|
|
pvariable_set(char *var)
|
|
{
|
|
char *p;
|
|
char tmp[1024];
|
|
|
|
if (!var)
|
|
msgFatal("NULL variable name & value passed.");
|
|
else if (!*var)
|
|
msgDebug("Warning: Zero length name & value passed to variable_set()\n");
|
|
/* Add a trivial namespace to whatever name the caller chooses. */
|
|
SAFE_STRCPY(tmp, "SYSINSTALL_PVAR");
|
|
if (index(var, '=') == NULL)
|
|
msgFatal("Invalid variable format: %s", var);
|
|
strlcat(tmp, var, 1024);
|
|
p = strchr(tmp, '=');
|
|
*p = '\0';
|
|
setenv(tmp, p + 1, 1);
|
|
}
|
|
|
|
char *
|
|
pvariable_get(char *var)
|
|
{
|
|
char tmp[1024];
|
|
|
|
SAFE_STRCPY(tmp, "SYSINSTALL_PVAR");
|
|
strlcat(tmp, var, 1024);
|
|
return getenv(tmp);
|
|
}
|