1ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen// Copyright (c) 2011 The Chromium Authors. All rights reserved. 2c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// Use of this source code is governed by a BSD-style license that can be 3c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// found in the LICENSE file. 4c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 53345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#include "base/environment.h" 6c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 7c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#if defined(OS_POSIX) 8c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include <stdlib.h> 9c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#elif defined(OS_WIN) 10c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include <windows.h> 11c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#endif 12c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 13c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "base/string_util.h" 14c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 15c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#if defined(OS_WIN) 16ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen#include "base/memory/scoped_ptr.h" 17c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "base/utf_string_conversions.h" 18c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#endif 19c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 20c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochnamespace { 21c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 223345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrickclass EnvironmentImpl : public base::Environment { 23c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch public: 243345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick virtual bool GetVar(const char* variable_name, std::string* result) { 253345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick if (GetVarImpl(variable_name, result)) 26c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return true; 27c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 28c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // Some commonly used variable names are uppercase while others 29c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // are lowercase, which is inconsistent. Let's try to be helpful 30c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // and look for a variable name with the reverse case. 31c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // I.e. HTTP_PROXY may be http_proxy for some users/systems. 32c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch char first_char = variable_name[0]; 33c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string alternate_case_var; 34c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (first_char >= 'a' && first_char <= 'z') 35c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch alternate_case_var = StringToUpperASCII(std::string(variable_name)); 36c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch else if (first_char >= 'A' && first_char <= 'Z') 37c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch alternate_case_var = StringToLowerASCII(std::string(variable_name)); 38c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch else 39c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return false; 403345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return GetVarImpl(alternate_case_var.c_str(), result); 41c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 42c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 433345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick virtual bool SetVar(const char* variable_name, const std::string& new_value) { 443345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return SetVarImpl(variable_name, new_value); 453345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick } 463345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 473345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick virtual bool UnSetVar(const char* variable_name) { 483345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return UnSetVarImpl(variable_name); 49c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 50c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 51c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch private: 523345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick bool GetVarImpl(const char* variable_name, std::string* result) { 53c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#if defined(OS_POSIX) 54c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const char* env_value = getenv(variable_name); 55c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (!env_value) 56c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return false; 57c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // Note that the variable may be defined but empty. 58c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (result) 59c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch *result = env_value; 60c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return true; 61c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#elif defined(OS_WIN) 62c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DWORD value_length = ::GetEnvironmentVariable( 63c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch UTF8ToWide(variable_name).c_str(), NULL, 0); 64c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (value_length == 0) 65c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return false; 66c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (result) { 67c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch scoped_array<wchar_t> value(new wchar_t[value_length]); 68c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch ::GetEnvironmentVariable(UTF8ToWide(variable_name).c_str(), value.get(), 69c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch value_length); 70c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch *result = WideToUTF8(value.get()); 71c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 72c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return true; 73c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#else 74c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#error need to port 75c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#endif 76c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 77c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 783345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick bool SetVarImpl(const char* variable_name, const std::string& new_value) { 793345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#if defined(OS_POSIX) 803345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick // On success, zero is returned. 813345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return !setenv(variable_name, new_value.c_str(), 1); 823345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#elif defined(OS_WIN) 833345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick // On success, a nonzero value is returned. 843345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return !!SetEnvironmentVariable(UTF8ToWide(variable_name).c_str(), 853345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick UTF8ToWide(new_value).c_str()); 863345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#endif 873345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick } 883345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 893345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick bool UnSetVarImpl(const char* variable_name) { 90c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#if defined(OS_POSIX) 91c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // On success, zero is returned. 923345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return !unsetenv(variable_name); 93c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#elif defined(OS_WIN) 943345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick // On success, a nonzero value is returned. 953345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return !!SetEnvironmentVariable(UTF8ToWide(variable_name).c_str(), NULL); 96c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#endif 97c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 98c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch}; 99c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 100c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} // namespace 101c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 102c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochnamespace base { 103c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 104c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochnamespace env_vars { 105c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 106c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#if defined(OS_POSIX) 107c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// On Posix systems, this variable contains the location of the user's home 108c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// directory. (e.g, /home/username/). 109c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char kHome[] = "HOME"; 110c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#endif 111c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 112c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} // namespace env_vars 113c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 1143345a6884c488ff3a535c2c9acdd33d74b37e311Iain MerrickEnvironment::~Environment() {} 115c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 1163345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick// static 1173345a6884c488ff3a535c2c9acdd33d74b37e311Iain MerrickEnvironment* Environment::Create() { 1183345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return new EnvironmentImpl(); 119c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 120c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 1213345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrickbool Environment::HasVar(const char* variable_name) { 1223345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return GetVar(variable_name, NULL); 123c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 124c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 125c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} // namespace base 126