Function with_cstr

codelldb/adapter/crates/lldb/src/strings.rs:6–33

Call `f` with a NUL-terminated copy of `s`.

(s: S, f: F)

/// Call `f` with a NUL-terminated copy of `s`.
pub(crate) fn with_cstr<S, F, R>(s: S, f: F) -> R
where
    S: AsRef<OsStr>,
    F: FnOnce(*const c_char) -> R,
{
    #[cfg(unix)]
    fn os_str_bytes(s: &OsStr) -> &[u8] {
        use std::os::unix::prelude::OsStrExt;
        s.as_bytes()
    }
    #[cfg(windows)]
    fn os_str_bytes(s: &OsStr) -> &[u8] {
        s.to_str().unwrap().as_bytes()
    }

    let bytes = os_str_bytes(s.as_ref());
    let allocated;
    let mut buffer = [0u8; 256];
    let ptr: *const c_char = if bytes.len() < buffer.len() {
        buffer[0..bytes.len()].clone_from_slice(bytes);
        buffer[bytes.len()] = 0;
        buffer.as_ptr() as *const c_char
    } else {
        allocated = Some(CString::new(bytes).unwrap());
        allocated.as_ref().unwrap().as_ptr()
    };
    f(ptr)
}

/// Call `f` with a NUL-terminated copy of `s`, or a null pointer if `s` is None.
pub(crate) fn with_opt_cstr<S, F, R>(s: Option<S>, f: F) -> R