Skip to content

Commit ffabd02

Browse files
committed
fix: unsafe env
Signed-off-by: if0ne <pavel.agafonov.al@gmail.com>
1 parent 9481d07 commit ffabd02

2 files changed

Lines changed: 15 additions & 5 deletions

File tree

rust/driver_manager/src/profile.rs

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -848,7 +848,9 @@ key = "value"
848848

849849
for TestCase(name, env_vars, input, expected) in test_cases {
850850
for (k, v) in &env_vars {
851-
std::env::set_var(k, v);
851+
unsafe {
852+
std::env::set_var(k, v);
853+
}
852854
}
853855

854856
let result = process_profile_value(input);
@@ -889,7 +891,9 @@ key = "value"
889891
}
890892

891893
for (k, _) in &env_vars {
892-
std::env::remove_var(k);
894+
unsafe {
895+
std::env::remove_var(k);
896+
}
893897
}
894898
}
895899
}

rust/driver_manager/tests/common/mod.rs

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -356,7 +356,9 @@ pub struct SetEnv {
356356
impl SetEnv {
357357
pub fn new(env_var: &'static str, new_value: impl AsRef<OsStr>) -> Self {
358358
let original_value = std::env::var_os(env_var);
359-
std::env::set_var(env_var, new_value);
359+
unsafe {
360+
std::env::set_var(env_var, new_value);
361+
}
360362
Self {
361363
env_var,
362364
original_value,
@@ -367,9 +369,13 @@ impl SetEnv {
367369
impl Drop for SetEnv {
368370
fn drop(&mut self) {
369371
if let Some(original_value) = &self.original_value {
370-
std::env::set_var(self.env_var, original_value);
372+
unsafe {
373+
std::env::set_var(self.env_var, original_value);
374+
}
371375
} else {
372-
std::env::remove_var(self.env_var);
376+
unsafe {
377+
std::env::remove_var(self.env_var);
378+
}
373379
}
374380
}
375381
}

0 commit comments

Comments
 (0)