src/tools/miri/tests/pass-dep/libc/libc-env.rs

//@ignore-target: windows # No libc
//@compile-flags: -Zmiri-disable-isolation

use std::ffi::CStr;

fn test_getenv() {
    let s = unsafe { libc::getenv(c"MIRI_ENV_VAR_TEST".as_ptr()) };
    assert!(!s.is_null());
    let value = unsafe { CStr::from_ptr(s).to_str().unwrap() };
    assert_eq!(value, "0");

    // Get a non-existing environment variable
    let s = unsafe { libc::getenv(c"MIRI_TEST_NONEXISTENT_VAR".as_ptr()) };
    assert!(s.is_null());

    // Empty string should not crash
    let s = unsafe { libc::getenv(c"".as_ptr()) };
    assert!(s.is_null());
}

fn test_setenv() {
    // Set a new environment variable
    let result = unsafe { libc::setenv(c"MIRI_TEST_VAR".as_ptr(), c"test_value".as_ptr(), 1) };
    assert_eq!(result, 0);

    // Verify it was set
    let s = unsafe { libc::getenv(c"MIRI_TEST_VAR".as_ptr()) };
    assert!(!s.is_null());
    let value = unsafe { CStr::from_ptr(s).to_str().unwrap() };
    assert_eq!(value, "test_value");

    // Test overwriting an existing variable
    let result = unsafe { libc::setenv(c"MIRI_TEST_VAR".as_ptr(), c"new_value".as_ptr(), 1) };
    assert_eq!(result, 0);

    // Verify it was updated
    let s = unsafe { libc::getenv(c"MIRI_TEST_VAR".as_ptr()) };
    assert!(!s.is_null());
    let value = unsafe { CStr::from_ptr(s).to_str().unwrap() };
    assert_eq!(value, "new_value");

    // Test invalid parameters
    let result = unsafe { libc::setenv(std::ptr::null(), c"value".as_ptr(), 1) };
    assert_eq!(result, -1);

    let result = unsafe { libc::setenv(c"".as_ptr(), c"value".as_ptr(), 1) };
    assert_eq!(result, -1);

    let result = unsafe { libc::setenv(c"INVALID=NAME".as_ptr(), c"value".as_ptr(), 1) };
    assert_eq!(result, -1);
}

fn test_unsetenv() {
    // Set a variable
    let result =
        unsafe { libc::setenv(c"MIRI_TEST_UNSET_VAR".as_ptr(), c"to_be_unset".as_ptr(), 1) };
    assert_eq!(result, 0);

    // Verify it exists
    let s = unsafe { libc::getenv(c"MIRI_TEST_UNSET_VAR".as_ptr()) };
    assert!(!s.is_null());

    // Unset it
    let result = unsafe { libc::unsetenv(c"MIRI_TEST_UNSET_VAR".as_ptr()) };
    assert_eq!(result, 0);

    // Verify it was unset
    let s = unsafe { libc::getenv(c"MIRI_TEST_UNSET_VAR".as_ptr()) };
    assert!(s.is_null());

    // Test unsetting a non-existing variable (should succeed)
    let result = unsafe { libc::unsetenv(c"MIRI_TEST_NONEXISTENT_VAR".as_ptr()) };
    assert_eq!(result, 0);

    // Test invalid parameters
    let result = unsafe { libc::unsetenv(std::ptr::null()) };
    assert_eq!(result, -1);

    let result = unsafe { libc::unsetenv(c"".as_ptr()) };
    assert_eq!(result, -1);

    let result = unsafe { libc::unsetenv(c"INVALID=NAME".as_ptr()) };
    assert_eq!(result, -1);
}

fn main() {
    test_getenv();
    test_setenv();
    test_unsetenv();
}