Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 1 | /* |
| 2 | * kmp_environment.h -- Handle environment varoiables OS-independently. |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 3 | */ |
| 4 | |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 5 | //===----------------------------------------------------------------------===// |
| 6 | // |
| 7 | // The LLVM Compiler Infrastructure |
| 8 | // |
| 9 | // This file is dual licensed under the MIT and the University of Illinois Open |
| 10 | // Source Licenses. See LICENSE.txt for details. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 14 | #ifndef KMP_ENVIRONMENT_H |
| 15 | #define KMP_ENVIRONMENT_H |
| 16 | |
| 17 | #ifdef __cplusplus |
| 18 | extern "C" { |
| 19 | #endif |
| 20 | |
Jonathan Peyton | 3041982 | 2017-05-12 18:01:32 +0000 | [diff] [blame] | 21 | // Return a copy of the value of environment variable or NULL if the variable |
| 22 | // does not exist. |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 23 | // *Note*: Returned pointed *must* be freed after use with __kmp_env_free(). |
Jonathan Peyton | 3041982 | 2017-05-12 18:01:32 +0000 | [diff] [blame] | 24 | char *__kmp_env_get(char const *name); |
| 25 | void __kmp_env_free(char const **value); |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 26 | |
| 27 | // Return 1 if the environment variable exists or 0 if does not exist. |
Jonathan Peyton | 3041982 | 2017-05-12 18:01:32 +0000 | [diff] [blame] | 28 | int __kmp_env_exists(char const *name); |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 29 | |
| 30 | // Set the environment variable. |
Jonathan Peyton | 3041982 | 2017-05-12 18:01:32 +0000 | [diff] [blame] | 31 | void __kmp_env_set(char const *name, char const *value, int overwrite); |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 32 | |
| 33 | // Unset (remove) environment variable. |
Jonathan Peyton | 3041982 | 2017-05-12 18:01:32 +0000 | [diff] [blame] | 34 | void __kmp_env_unset(char const *name); |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 35 | |
Jonathan Peyton | 3041982 | 2017-05-12 18:01:32 +0000 | [diff] [blame] | 36 | // ----------------------------------------------------------------------------- |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 37 | // Working with environment blocks. |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 38 | |
Jonathan Peyton | 3041982 | 2017-05-12 18:01:32 +0000 | [diff] [blame] | 39 | /* kmp_env_blk_t is read-only collection of environment variables (or |
| 40 | environment-like). Usage: |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 41 | |
Jonathan Peyton | 3041982 | 2017-05-12 18:01:32 +0000 | [diff] [blame] | 42 | kmp_env_blk_t block; |
| 43 | __kmp_env_blk_init( & block, NULL ); // Initialize block from process |
| 44 | // environment. |
| 45 | // or |
| 46 | __kmp_env_blk_init( & block, "KMP_WARNING=1|KMP_AFFINITY=none" ); // from string |
| 47 | __kmp_env_blk_sort( & block ); // Optionally, sort list. |
| 48 | for ( i = 0; i < block.count; ++ i ) { |
| 49 | // Process block.vars[ i ].name and block.vars[ i ].value... |
Jonathan Peyton | bd3a763 | 2017-09-27 20:36:27 +0000 | [diff] [blame] | 50 | } |
Jonathan Peyton | 3041982 | 2017-05-12 18:01:32 +0000 | [diff] [blame] | 51 | __kmp_env_block_free( & block ); |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 52 | */ |
| 53 | |
| 54 | struct __kmp_env_var { |
Jonathan Peyton | 3041982 | 2017-05-12 18:01:32 +0000 | [diff] [blame] | 55 | char const *name; |
| 56 | char const *value; |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 57 | }; |
| 58 | typedef struct __kmp_env_var kmp_env_var_t; |
| 59 | |
| 60 | struct __kmp_env_blk { |
Jonathan Peyton | 3041982 | 2017-05-12 18:01:32 +0000 | [diff] [blame] | 61 | char const *bulk; |
| 62 | kmp_env_var_t const *vars; |
| 63 | int count; |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 64 | }; |
| 65 | typedef struct __kmp_env_blk kmp_env_blk_t; |
| 66 | |
Jonathan Peyton | 3041982 | 2017-05-12 18:01:32 +0000 | [diff] [blame] | 67 | void __kmp_env_blk_init(kmp_env_blk_t *block, char const *bulk); |
| 68 | void __kmp_env_blk_free(kmp_env_blk_t *block); |
| 69 | void __kmp_env_blk_sort(kmp_env_blk_t *block); |
| 70 | char const *__kmp_env_blk_var(kmp_env_blk_t *block, char const *name); |
Jim Cownie | 5e8470a | 2013-09-27 10:38:44 +0000 | [diff] [blame] | 71 | |
| 72 | #ifdef __cplusplus |
| 73 | } |
| 74 | #endif |
| 75 | |
| 76 | #endif // KMP_ENVIRONMENT_H |
| 77 | |
| 78 | // end of file // |