Source code
Revision control
Copy as Markdown
Other Tools
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
/*
** File: env.c
** Description: Testing environment variable operations
**
*/
#include "prenv.h"
#include "prmem.h"
#include "plgetopt.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
PRIntn debug = 0;
PRIntn verbose = 0;
PRIntn secure = 0;
PRBool failedAlready = PR_FALSE;
#define ENVNAME "NSPR_ENVIRONMENT_TEST_VARIABLE"
#define ENVVALUE "The expected result"
#define ENVBUFSIZE 256
char* envBuf; /* buffer pointer. We leak memory here on purpose! */
static char* NewBuffer(size_t size) {
char* buf = malloc(size);
if (NULL == buf) {
printf("env: NewBuffer() failed\n");
exit(1);
}
return (buf);
} /* end NewBuffer() */
int main(int argc, char** argv) {
char* value;
PRStatus rc;
{ /* Get command line options */
PLOptStatus os;
PLOptState* opt = PL_CreateOptState(argc, argv, "vds");
while (PL_OPT_EOL != (os = PL_GetNextOpt(opt))) {
if (PL_OPT_BAD == os) {
continue;
}
switch (opt->option) {
case 'd': /* debug */
debug = 1;
break;
case 'v': /* verbose */
verbose = 1;
break;
case 's': /* secure / set[ug]id */
/*
** To test PR_GetEnvSecure, make this executable (or a
** copy of it) setuid / setgid / otherwise inherently
** privileged (e.g., file capabilities) and run it
** with this flag.
*/
secure = 1;
break;
default:
break;
}
}
PL_DestroyOptState(opt);
} /* end block "Get command line options" */
#if 0
{
/*
** This uses Windows native environment manipulation
** as an experiment. Note the separation of namespace!
*/
BOOL rv;
DWORD size;
rv = SetEnvironmentVariable( ENVNAME, ENVVALUE );
if ( rv == 0 ) {
if (debug) {
printf("env: Shit! SetEnvironmentVariable() failed\n");
}
failedAlready = PR_TRUE;
}
if (verbose) {
printf("env: SetEnvironmentVariable() worked\n");
}
size = GetEnvironmentVariable( ENVNAME, envBuf, ENVBUFSIZE );
if ( size == 0 ) {
if (debug) {
printf("env: Shit! GetEnvironmentVariable() failed. Found: %s\n", envBuf );
}
failedAlready = PR_TRUE;
}
if (verbose) {
printf("env: GetEnvironmentVariable() worked. Found: %s\n", envBuf);
}
value = PR_GetEnv( ENVNAME );
if ( (NULL == value ) || (strcmp( value, ENVVALUE))) {
if (debug) {
printf( "env: PR_GetEnv() failed retrieving WinNative. Found: %s\n", value);
}
failedAlready = PR_TRUE;
}
if (verbose) {
printf("env: PR_GetEnv() worked. Found: %s\n", value);
}
}
#endif
/* set an environment variable, read it back */
envBuf = NewBuffer(ENVBUFSIZE);
sprintf(envBuf, ENVNAME "=" ENVVALUE);
rc = PR_SetEnv(envBuf);
if (PR_FAILURE == rc) {
if (debug) {
printf("env: PR_SetEnv() failed setting\n");
}
failedAlready = PR_TRUE;
} else {
if (verbose) {
printf("env: PR_SetEnv() worked.\n");
}
}
value = PR_GetEnv(ENVNAME);
if ((NULL == value) || (strcmp(value, ENVVALUE))) {
if (debug) {
printf("env: PR_GetEnv() Failed after setting\n");
}
failedAlready = PR_TRUE;
} else {
if (verbose) {
printf("env: PR_GetEnv() worked after setting it. Found: %s\n", value);
}
}
if (secure) {
/*
** In this case we've been run with elevated privileges, so
** test that PR_GetEnvSecure *doesn't* find that env var.
*/
value = PR_GetEnvSecure(ENVNAME);
if (NULL != value) {
if (debug) {
printf("env: PR_GetEnvSecure() failed; expected NULL, found \"%s\"\n",
value);
}
failedAlready = PR_TRUE;
} else {
if (verbose) {
printf("env: PR_GetEnvSecure() worked\n");
}
}
} else {
/*
** In this case the program is being run normally, so do the
** same check for PR_GetEnvSecure as for PR_GetEnv.
*/
value = PR_GetEnvSecure(ENVNAME);
if ((NULL == value) || (strcmp(value, ENVVALUE))) {
if (debug) {
printf("env: PR_GetEnvSecure() Failed after setting\n");
}
failedAlready = PR_TRUE;
} else {
if (verbose) {
printf("env: PR_GetEnvSecure() worked after setting it. Found: %s\n",
value);
}
}
}
/* ---------------------------------------------------------------------- */
/* check that PR_DuplicateEnvironment() agrees with PR_GetEnv() */
{
#if defined(XP_UNIX) && (!defined(DARWIN) || defined(HAVE_CRT_EXTERNS_H))
static const PRBool expect_failure = PR_FALSE;
#else
static const PRBool expect_failure = PR_TRUE;
#endif
char **i, **dupenv = PR_DuplicateEnvironment();
if (NULL == dupenv) {
if (expect_failure) {
if (verbose)
printf(
"env: PR_DuplicateEnvironment failed, "
"as expected on this platform.\n");
} else {
if (debug) {
printf("env: PR_DuplicateEnvironment() failed.\n");
}
failedAlready = PR_TRUE;
}
} else {
unsigned found = 0;
if (expect_failure) {
if (debug)
printf(
"env: PR_DuplicateEnvironment() succeeded, "
"but failure is expected on this platform.\n");
failedAlready = PR_TRUE;
} else {
if (verbose) {
printf("env: PR_DuplicateEnvironment() succeeded.\n");
}
}
for (i = dupenv; *i; i++) {
char* equals = strchr(*i, '=');
if (equals == NULL) {
if (debug)
printf(
"env: PR_DuplicateEnvironment() returned a string"
" with no '=': %s\n",
*i);
failedAlready = PR_TRUE;
} else {
/* We own this string, so we can temporarily alter it */
/* *i is the null-terminated name; equals + 1 is the value */
*equals = '\0';
if (strcmp(*i, ENVNAME) == 0) {
found++;
if (verbose)
printf("env: PR_DuplicateEnvironment() found " ENVNAME
" (%u so far).\n",
found);
}
/* Multiple values for the same name can't happen, according to POSIX.
*/
value = PR_GetEnv(*i);
if (value == NULL) {
if (debug)
printf(
"env: PR_DuplicateEnvironment() returned a name"
" which PR_GetEnv() failed to find: %s\n",
*i);
failedAlready = PR_TRUE;
} else if (strcmp(equals + 1, value) != 0) {
if (debug)
printf(
"env: PR_DuplicateEnvironment() returned the wrong"
" value for %s: expected %s; found %s\n",
*i, value, equals + 1);
failedAlready = PR_TRUE;
} else {
if (verbose)
printf(
"env: PR_DuplicateEnvironment() agreed with"
" PR_GetEnv() about %s\n",
*i);
}
}
PR_Free(*i);
}
PR_Free(dupenv);
if (found != 1) {
if (debug)
printf("env: PR_DuplicateEnvironment() found %u entries for " ENVNAME
" (expected 1)\n",
found);
failedAlready = PR_TRUE;
} else {
if (verbose) {
printf("env: PR_DuplicateEnvironment() found 1 entry for " ENVNAME
"\n");
}
}
}
}
/* ---------------------------------------------------------------------- */
/* un-set the variable, using RAW name... should not work */
envBuf = NewBuffer(ENVBUFSIZE);
sprintf(envBuf, ENVNAME);
rc = PR_SetEnv(envBuf);
if (PR_FAILURE == rc) {
if (verbose) {
printf("env: PR_SetEnv() not un-set using RAW name. Good!\n");
}
} else {
if (debug) {
printf("env: PR_SetEnv() un-set using RAW name. Bad!\n");
}
failedAlready = PR_TRUE;
}
value = PR_GetEnv(ENVNAME);
if (NULL == value) {
if (debug) {
printf("env: PR_GetEnv() after un-set using RAW name. Bad!\n");
}
failedAlready = PR_TRUE;
} else {
if (verbose) {
printf("env: PR_GetEnv() after RAW un-set found: %s\n", value);
}
}
/* ---------------------------------------------------------------------- */
/* set it again ... */
envBuf = NewBuffer(ENVBUFSIZE);
sprintf(envBuf, ENVNAME "=" ENVVALUE);
rc = PR_SetEnv(envBuf);
if (PR_FAILURE == rc) {
if (debug) {
printf("env: PR_SetEnv() failed setting the second time.\n");
}
failedAlready = PR_TRUE;
} else {
if (verbose) {
printf("env: PR_SetEnv() worked.\n");
}
}
/* un-set the variable using the form name= */
envBuf = NewBuffer(ENVBUFSIZE);
sprintf(envBuf, ENVNAME "=");
rc = PR_SetEnv(envBuf);
if (PR_FAILURE == rc) {
if (debug) {
printf("env: PR_SetEnv() failed un-setting using name=\n");
}
failedAlready = PR_TRUE;
} else {
if (verbose) {
printf("env: PR_SetEnv() un-set using name= worked\n");
}
}
value = PR_GetEnv(ENVNAME);
if ((NULL == value) || (0x00 == *value)) {
if (verbose) {
printf("env: PR_GetEnv() after un-set using name= worked\n");
}
} else {
if (debug) {
printf("env: PR_GetEnv() after un-set using name=. Found: %s\n", value);
}
failedAlready = PR_TRUE;
}
/* ---------------------------------------------------------------------- */
/* un-set the variable using the form name= */
envBuf = NewBuffer(ENVBUFSIZE);
sprintf(envBuf, ENVNAME "999=");
rc = PR_SetEnv(envBuf);
if (PR_FAILURE == rc) {
if (debug) {
printf("env: PR_SetEnv() failed un-setting using name=\n");
}
failedAlready = PR_TRUE;
} else {
if (verbose) {
printf("env: PR_SetEnv() un-set using name= worked\n");
}
}
value = PR_GetEnv(ENVNAME "999");
if ((NULL == value) || (0x00 == *value)) {
if (verbose) {
printf("env: PR_GetEnv() after un-set using name= worked\n");
}
} else {
if (debug) {
printf("env: PR_GetEnv() after un-set using name=. Found: %s\n", value);
}
failedAlready = PR_TRUE;
}
/* ---------------------------------------------------------------------- */
if (debug || verbose) {
printf("\n%s\n", (failedAlready) ? "FAILED" : "PASSED");
}
return ((failedAlready) ? 1 : 0);
} /* main() */
/* env.c */