/************************************************************************
 *
 * 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;
}
