/* -*- 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/. */

#include <string.h>

#include <stdlib.h>

#include "primpl.h"

#include "prmem.h"

#if defined(XP_UNIX)

#  include <unistd.h>

#  if defined(DARWIN)

#    if defined(HAVE_CRT_EXTERNS_H)

#      include <crt_externs.h>

#    endif /* HAVE_CRT_EXTERNS_H */

#  else    /* DARWIN */

PR_IMPORT_DATA(char**) environ;

#  endif   /* DARWIN */

#endif     /* XP_UNIX */

#if !defined(HAVE_SECURE_GETENV) && defined(HAVE___SECURE_GETENV)

#  define secure_getenv __secure_getenv

#  define HAVE_SECURE_GETENV 1

#endif

/* Lock used to lock the environment */

#if defined(_PR_NO_PREEMPT)

#  define _PR_NEW_LOCK_ENV()

#  define _PR_DELETE_LOCK_ENV()

#  define _PR_LOCK_ENV()

#  define _PR_UNLOCK_ENV()

#elif defined(_PR_LOCAL_THREADS_ONLY)

extern _PRCPU* _pr_primordialCPU;

static PRIntn _is;

#  define _PR_NEW_LOCK_ENV()

#  define _PR_DELETE_LOCK_ENV()

#  define _PR_LOCK_ENV() \

    if (_pr_primordialCPU) _PR_INTSOFF(_is);

#  define _PR_UNLOCK_ENV() \

    if (_pr_primordialCPU) _PR_INTSON(_is);

#else

static PRLock* _pr_envLock = NULL;

#  define _PR_NEW_LOCK_ENV() \

    { _pr_envLock = PR_NewLock(); }

#  define _PR_DELETE_LOCK_ENV()      \

    {                                \

      if (_pr_envLock) {             \

        PR_DestroyLock(_pr_envLock); \

        _pr_envLock = NULL;          \

      }                              \

    }

#  define _PR_LOCK_ENV()                     \

    {                                        \

      if (_pr_envLock) PR_Lock(_pr_envLock); \

    }

#  define _PR_UNLOCK_ENV()                     \

    {                                          \

      if (_pr_envLock) PR_Unlock(_pr_envLock); \

    }

#endif

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

void _PR_InitEnv(void) { _PR_NEW_LOCK_ENV(); }

void _PR_CleanupEnv(void) { _PR_DELETE_LOCK_ENV(); }

PR_IMPLEMENT(char*) PR_GetEnv(const char* var) {

  char* ev;

  if (!_pr_initialized) {

    _PR_ImplicitInitialization();

  }

  _PR_LOCK_ENV();

  ev = _PR_MD_GET_ENV(var);

  _PR_UNLOCK_ENV();

  return ev;

}

PR_IMPLEMENT(char*) PR_GetEnvSecure(const char* var) {

#ifdef HAVE_SECURE_GETENV

  char* ev;

  if (!_pr_initialized) {

    _PR_ImplicitInitialization();

  }

  _PR_LOCK_ENV();

  ev = secure_getenv(var);

  _PR_UNLOCK_ENV();

  return ev;

#else

#  ifdef XP_UNIX

  /*

  ** Fall back to checking uids and gids.  This won't detect any other

  ** privilege-granting mechanisms the platform may have.  This also

  ** can't detect the case where the process already called

  ** setuid(geteuid()) and/or setgid(getegid()).

  */

  if (getuid() != geteuid() || getgid() != getegid()) {

    return NULL;

  }

#  endif /* XP_UNIX */

  return PR_GetEnv(var);

#endif   /* HAVE_SECURE_GETENV */

}

PR_IMPLEMENT(PRStatus) PR_SetEnv(const char* string) {

  PRIntn result;

  if (!_pr_initialized) {

    _PR_ImplicitInitialization();

  }

  if (!strchr(string, '=')) {

    return (PR_FAILURE);

  }

  _PR_LOCK_ENV();

  result = _PR_MD_PUT_ENV((char*)string);

  _PR_UNLOCK_ENV();

  return result ? PR_FAILURE : PR_SUCCESS;

}

#if defined(XP_UNIX) && (!defined(DARWIN) || defined(HAVE_CRT_EXTERNS_H))

PR_IMPLEMENT(char**) PR_DuplicateEnvironment(void) {

  char **the_environ, **result, **end, **src, **dst;

  _PR_LOCK_ENV();

#  ifdef DARWIN

  the_environ = *(_NSGetEnviron());

#  else

  the_environ = environ;

#  endif

  for (end = the_environ; *end != NULL; end++) /* empty loop body */;

  result = (char**)PR_Malloc(sizeof(char*) * (end - the_environ + 1));

  if (result != NULL) {

    for (src = the_environ, dst = result; src != end; src++, dst++) {

      size_t len;

      len = strlen(*src) + 1;

      *dst = PR_Malloc(len);

      if (*dst == NULL) {

        /* Allocation failed.  Must clean up the half-copied env. */

        char** to_delete;

        for (to_delete = result; to_delete != dst; to_delete++) {

          PR_Free(*to_delete);

        }

        PR_Free(result);

        result = NULL;

        goto out;

      }

      memcpy(*dst, *src, len);

    }

    *dst = NULL;

  }

out:

  _PR_UNLOCK_ENV();

  return result;

}

#else

/* This platform doesn't support raw access to the environ block. */

PR_IMPLEMENT(char**) PR_DuplicateEnvironment(void) { return NULL; }

#endif