/************************************************************************

  *

  * environ.cpp - definitions of testsuite helpers

  *

  * $Id: environ.cpp 290013 2005-09-18 23:47:55Z sebor $

  *

  ************************************************************************

  *

  * Copyright (c) 1994-2005 Quovadx,  Inc., acting through its  Rogue Wave

  * Software division. Licensed under the Apache License, Version 2.0 (the

  * "License");  you may  not use this file except  in compliance with the

  * License.    You    may   obtain   a   copy   of    the   License    at

  * http://www.apache.org/licenses/LICENSE-2.0.    Unless   required    by

  * applicable law  or agreed to  in writing,  software  distributed under

  * the License is distributed on an "AS IS" BASIS,  WITHOUT WARRANTIES OR

  * CONDITIONS OF  ANY KIND, either  express or implied.  See  the License

  * for the specific language governing permissions  and limitations under

  * the License.

  * 

  **************************************************************************/

 // expand _TEST_EXPORT macros

 #define _RWSTD_TEST_SRC

 #include <environ.h>

 #include <assert.h>   // for assert

 #include <stdlib.h>   // for getenv, malloc, putenv

 #include <string.h>   // for strchr, strlen, ...

 extern "C" {

 #ifndef _RWSTD_NO_PUTENV_CONST_CHAR

 IMPORT_C int putenv (const char*) _LIBC_THROWS ();

 #else   // if defined (_RWSTD_NO_PUTENV_CONST_CHAR)

 _RWSTD_DLLIMPORT int putenv (char*) _LIBC_THROWS ();

 #endif   // _RWSTD_NO_PUTENV_CONST_CHAR

 }   // extern "C"

 // sets one or more sep-separated environment variables

 _TEST_EXPORT int

 rw_putenv (const char* str, int sep /* = -1 */)

 {

     int nset = 0;

     int ret  = 0;

     if (!str) {

         str = getenv ("RW_PUTENV");

         if (str)

             sep = *str++;

         else

             return 0;

     }

     for (const char *pvar = str; pvar && *pvar; ++nset) {

         const char *pend = strchr (pvar, sep);

         if (!pend)

             pend = pvar + strlen (pvar);

         const size_t varlen = pend - pvar;

         char* const envvar = (char*)malloc (pend - pvar + 1);

         memcpy (envvar, pvar, varlen);

         envvar [varlen] = '\0';

         // Note: calling Solaris 7 putenv() during program startup

         // (i.e., from ctors of namespace-scope objects) prevents

         // getenv() from finding that variable at program runtime

         ret = putenv (envvar);

         // determine wheteher putenv() made copy of the variable

         // or if it simply used the pointer passed to it; if the

         // former, deallocate the buffer dynamically allocated

         // above

         char namebuf [256];

         char* const equals = strchr (envvar, '=');

         if (equals) {

             assert (size_t (equals - envvar) < sizeof namebuf);

             memcpy (namebuf, envvar, equals - envvar);

             namebuf [equals - envvar] = '\0';

             const char* const var = getenv (namebuf);

             if (equals + 1 != var)

                 free (envvar);

         }

         else

             free (envvar);

         pvar = pend + !!*pend;

     }

     if (1 == nset)

         return ret;

     return nset;

 }