1 /* 2 Copyright (c) 2005-2022 Intel Corporation 3 4 Licensed under the Apache License, Version 2.0 (the "License"); 5 you may not use this file except in compliance with the License. 6 You may obtain a copy of the License at 7 8 http://www.apache.org/licenses/LICENSE-2.0 9 10 Unless required by applicable law or agreed to in writing, software 11 distributed under the License is distributed on an "AS IS" BASIS, 12 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 See the License for the specific language governing permissions and 14 limitations under the License. 15 */ 16 17 #ifndef __TBB_test_common_utils_env_H 18 #define __TBB_test_common_utils_env_H 19 20 #include "test.h" 21 22 #include <cstdlib> // getenv/setenv 23 24 namespace utils { 25 26 int SetEnv( const char *envname, const char *envval ) { 27 CHECK_MESSAGE( (envname && envval), "Harness::SetEnv() requires two valid C strings" ); 28 #if __TBB_WIN8UI_SUPPORT 29 CHECK_MESSAGE( false, "Harness::SetEnv() should not be called in code built for win8ui" ); 30 return -1; 31 #elif !(_MSC_VER || __MINGW32__ || __MINGW64__) 32 // On POSIX systems use setenv 33 return setenv(envname, envval, /*overwrite=*/1); 34 #elif __STDC_SECURE_LIB__>=200411 35 // this macro is set in VC & MinGW if secure API functions are present 36 return _putenv_s(envname, envval); 37 #else 38 // If no secure API on Windows, use _putenv 39 size_t namelen = strlen(envname), valuelen = strlen(envval); 40 char* buf = new char[namelen+valuelen+2]; 41 strncpy(buf, envname, namelen); 42 buf[namelen] = '='; 43 strncpy(buf+namelen+1, envval, valuelen); 44 buf[namelen+1+valuelen] = char(0); 45 int status = _putenv(buf); 46 delete[] buf; 47 return status; 48 #endif 49 } 50 51 char* GetEnv(const char *envname) { 52 CHECK_MESSAGE(envname, "Harness::GetEnv() requires a valid C string"); 53 #if __TBB_WIN8UI_SUPPORT 54 return nullptr; 55 #else 56 return std::getenv(envname); 57 #endif 58 } 59 60 } // namespace utils 61 62 #endif // __TBB_test_common_utils_env_H 63 64