nsprpub/pr/tests/env.c

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6 /*
   7 ** File:        env.c
   8 ** Description: Testing environment variable operations
   9 **
  10 */
  11 #include "prenv.h"
  12 #include "plgetopt.h"
  13
  14 #include <stdio.h>
  15 #include <stdlib.h>
  16 #include <string.h>
  17
  18 PRIntn  debug = 0;
  19 PRIntn  verbose = 0;
  20 PRBool  failedAlready = PR_FALSE;
  21
  22 #define  ENVNAME    "NSPR_ENVIRONMENT_TEST_VARIABLE"
  23 #define  ENVVALUE   "The expected result"
  24 #define  ENVBUFSIZE 256
  25
  26 char    *envBuf; /* buffer pointer. We leak memory here on purpose! */
  27
  28 static char * NewBuffer( size_t size )
  29 {
  30     char *buf = malloc( size );
  31     if ( NULL == buf ) {
  32         printf("env: NewBuffer() failed\n");
  33         exit(1);
  34     }
  35     return(buf);
  36 } /* end NewBuffer() */
  37
  38 int main(int argc, char **argv)
  39 {
  40     char    *value;
  41     PRStatus    rc;
  42
  43     {   /* Get command line options */
  44         PLOptStatus os;
  45         PLOptState *opt = PL_CreateOptState(argc, argv, "vd");
  46
  47             while (PL_OPT_EOL != (os = PL_GetNextOpt(opt)))
  48         {
  49                     if (PL_OPT_BAD == os) continue;
  50             switch (opt->option)
  51             {
  52             case 'd':  /* debug */
  53                 debug = 1;
  54                 break;
  55             case 'v':  /* verbose */
  56                 verbose = 1;
  57                 break;
  58              default:
  59                 break;
  60             }
  61         }
  62             PL_DestroyOptState(opt);
  63     } /* end block "Get command line options" */
  64
  65 #if 0
  66     {
  67         /*
  68         ** This uses Windows native environment manipulation
  69         ** as an experiment. Note the separation of namespace!
  70         */
  71         BOOL rv;
  72         DWORD   size;
  73         rv = SetEnvironmentVariable( ENVNAME, ENVVALUE );
  74         if ( rv == 0 )  {
  75             if (debug) printf("env: Shit! SetEnvironmentVariable() failed\n");
  76             failedAlready = PR_TRUE;
  77         }
  78         if (verbose) printf("env: SetEnvironmentVariable() worked\n");
  79
  80         size = GetEnvironmentVariable( ENVNAME, envBuf, ENVBUFSIZE );
  81         if ( size == 0 )  {
  82             if (debug) printf("env: Shit! GetEnvironmentVariable() failed. Found: %s\n", envBuf );
  83             failedAlready = PR_TRUE;
  84         }
  85         if (verbose) printf("env: GetEnvironmentVariable() worked. Found: %s\n", envBuf);
  86
  87         value = PR_GetEnv( ENVNAME );
  88         if ( (NULL == value ) || (strcmp( value, ENVVALUE)))  {
  89             if (debug) printf( "env: PR_GetEnv() failed retrieving WinNative. Found: %s\n", value);
  90             failedAlready = PR_TRUE;
  91         }
  92         if (verbose) printf("env: PR_GetEnv() worked. Found: %s\n", value);
  93     }
  94 #endif
  95
  96     /* set an environment variable, read it back */
  97     envBuf = NewBuffer( ENVBUFSIZE );
  98     sprintf( envBuf, ENVNAME "=" ENVVALUE );
  99     rc = PR_SetEnv( envBuf );
 100     if ( PR_FAILURE == rc )  {
 101         if (debug) printf( "env: PR_SetEnv() failed setting\n");
 102         failedAlready = PR_TRUE;
 103     } else {
 104         if (verbose) printf("env: PR_SetEnv() worked.\n");
 105     }
 106
 107     value = PR_GetEnv( ENVNAME );
 108     if ( (NULL == value ) || (strcmp( value, ENVVALUE)))  {
 109         if (debug) printf( "env: PR_GetEnv() Failed after setting\n" );
 110         failedAlready = PR_TRUE;
 111     } else {
 112         if (verbose) printf("env: PR_GetEnv() worked after setting it. Found: %s\n", value );
 113     }
 114
 115 /* ---------------------------------------------------------------------- */
 116     /* un-set the variable, using RAW name... should not work */
 117     envBuf = NewBuffer( ENVBUFSIZE );
 118     sprintf( envBuf, ENVNAME );
 119     rc = PR_SetEnv( envBuf );
 120     if ( PR_FAILURE == rc )  {
 121         if (verbose) printf( "env: PR_SetEnv() not un-set using RAW name. Good!\n");
 122     } else {
 123         if (debug) printf("env: PR_SetEnv() un-set using RAW name. Bad!\n" );
 124         failedAlready = PR_TRUE;
 125     }
 126
 127     value = PR_GetEnv( ENVNAME );
 128     if ( NULL == value ) {
 129         if (debug) printf("env: PR_GetEnv() after un-set using RAW name. Bad!\n" );
 130         failedAlready = PR_TRUE;
 131     } else {
 132         if (verbose) printf( "env: PR_GetEnv() after RAW un-set found: %s\n", value );
 133     }
 134
 135 /* ---------------------------------------------------------------------- */
 136     /* set it again ... */
 137     envBuf = NewBuffer( ENVBUFSIZE );
 138     sprintf( envBuf, ENVNAME "=" ENVVALUE );
 139     rc = PR_SetEnv( envBuf );
 140     if ( PR_FAILURE == rc )  {
 141         if (debug) printf( "env: PR_SetEnv() failed setting the second time.\n");
 142         failedAlready = PR_TRUE;
 143     } else {
 144         if (verbose) printf("env: PR_SetEnv() worked.\n");
 145     }
 146
 147     /* un-set the variable using the form name= */
 148     envBuf = NewBuffer( ENVBUFSIZE );
 149     sprintf( envBuf, ENVNAME "=" );
 150     rc = PR_SetEnv( envBuf );
 151     if ( PR_FAILURE == rc )  {
 152         if (debug) printf( "env: PR_SetEnv() failed un-setting using name=\n");
 153         failedAlready = PR_TRUE;
 154     } else {
 155         if (verbose) printf("env: PR_SetEnv() un-set using name= worked\n" );
 156     }
 157
 158     value = PR_GetEnv( ENVNAME );
 159     if (( NULL == value ) || ( 0x00 == *value )) {
 160         if (verbose) printf("env: PR_GetEnv() after un-set using name= worked\n" );
 161     } else {
 162         if (debug) printf( "env: PR_GetEnv() after un-set using name=. Found: %s\n", value );
 163         failedAlready = PR_TRUE;
 164     }
 165 /* ---------------------------------------------------------------------- */
 166     /* un-set the variable using the form name= */
 167     envBuf = NewBuffer( ENVBUFSIZE );
 168     sprintf( envBuf, ENVNAME "999=" );
 169     rc = PR_SetEnv( envBuf );
 170     if ( PR_FAILURE == rc )  {
 171         if (debug) printf( "env: PR_SetEnv() failed un-setting using name=\n");
 172         failedAlready = PR_TRUE;
 173     } else {
 174         if (verbose) printf("env: PR_SetEnv() un-set using name= worked\n" );
 175     }
 176
 177     value = PR_GetEnv( ENVNAME "999" );
 178     if (( NULL == value ) || ( 0x00 == *value )) {
 179         if (verbose) printf("env: PR_GetEnv() after un-set using name= worked\n" );
 180     } else {
 181         if (debug) printf( "env: PR_GetEnv() after un-set using name=. Found: %s\n", value );
 182         failedAlready = PR_TRUE;
 183     }
 184
 185 /* ---------------------------------------------------------------------- */
 186     if (debug || verbose) printf("\n%s\n", (failedAlready)? "FAILED" : "PASSED" );
 187     return( (failedAlready)? 1 : 0 );
 188 }  /* main() */
 189
 190 /* env.c */