legacy-libs/grpc-cloned/deps/grpc/src/core/lib/gpr/env_linux.cc

   1 /*
   2  *
   3  * Copyright 2015 gRPC authors.
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *     http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  */
  18
  19 /* for secure_getenv. */
  20 #ifndef _GNU_SOURCE
  21 #define _GNU_SOURCE
  22 #endif
  23
  24 #include <grpc/support/port_platform.h>
  25
  26 #ifdef GPR_LINUX_ENV
  27
  28 #include "src/core/lib/gpr/env.h"
  29
  30 #include <dlfcn.h>
  31 #include <features.h>
  32 #include <stdlib.h>
  33 #include <string.h>
  34
  35 #include <grpc/support/log.h>
  36 #include <grpc/support/string_util.h>
  37
  38 #include "src/core/lib/gpr/string.h"
  39 #include "src/core/lib/gpr/useful.h"
  40
  41 char* gpr_getenv(const char* name) {
  42   char* result = nullptr;
  43 #if defined(GPR_BACKWARDS_COMPATIBILITY_MODE)
  44   typedef char* (*getenv_type)(const char*);
  45   static getenv_type getenv_func = nullptr;
  46   /* Check to see which getenv variant is supported (go from most
  47    * to least secure) */
  48   if (getenv_func == nullptr) {
  49     const char* names[] = {"secure_getenv", "__secure_getenv", "getenv"};
  50     for (size_t i = 0; i < GPR_ARRAY_SIZE(names); i++) {
  51       getenv_func = (getenv_type)dlsym(RTLD_DEFAULT, names[i]);
  52       if (getenv_func != nullptr) {
  53         break;
  54       }
  55     }
  56   }
  57   result = getenv_func(name);
  58 #elif __GLIBC__ > 2 || (__GLIBC__ == 2 && __GLIBC_MINOR__ >= 17)
  59   result = secure_getenv(name);
  60 #else
  61   result = getenv(name);
  62 #endif
  63   return result == nullptr ? result : gpr_strdup(result);
  64 }
  65
  66 void gpr_setenv(const char* name, const char* value) {
  67   int res = setenv(name, value, 1);
  68   GPR_ASSERT(res == 0);
  69 }
  70
  71 void gpr_unsetenv(const char* name) {
  72   int res = unsetenv(name);
  73   GPR_ASSERT(res == 0);
  74 }
  75
  76 #endif /* GPR_LINUX_ENV */