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