зеркало из https://github.com/mozilla/gecko-dev.git
371 строка
12 KiB
C
371 строка
12 KiB
C
/* -*- 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, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
/*
|
|
** 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 */
|