xref: /aosp_15_r20/external/crosvm/base/src/sys/windows/shm.rs (revision bb4ee6a4ae7042d18b07a98463b9c8b875e44b39)

// Copyright 2022 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use std::ffi::CStr;

use win_util::create_file_mapping;
use winapi::um::winnt::PAGE_EXECUTE_READWRITE;

use crate::descriptor::FromRawDescriptor;
use crate::descriptor::SafeDescriptor;
use crate::shm::PlatformSharedMemory;
use crate::Result;
use crate::SharedMemory;

impl PlatformSharedMemory for SharedMemory {
    fn new(_debug_name: &CStr, size: u64) -> Result<SharedMemory> {
        let mapping_handle =
            // SAFETY:
            // Safe because we do not provide a handle.
            unsafe { create_file_mapping(None, size, PAGE_EXECUTE_READWRITE, None) }
                .map_err(super::Error::from)?;

        Self::from_safe_descriptor(
            // SAFETY:
            // Safe because we have exclusive ownership of mapping_handle & it is valid.
            unsafe { SafeDescriptor::from_raw_descriptor(mapping_handle) },
            size,
        )
    }

    fn from_safe_descriptor(mapping_handle: SafeDescriptor, size: u64) -> Result<SharedMemory> {
        Ok(SharedMemory {
            descriptor: mapping_handle,
            size,
        })
    }
}

#[cfg(test)]
mod test {
    use winapi::shared::winerror::ERROR_NOT_ENOUGH_MEMORY;

    use super::*;

    #[cfg_attr(all(target_os = "windows", target_env = "gnu"), ignore)]
    #[test]
    fn new_too_huge() {
        let result = SharedMemory::new("test", 0x8000_0000_0000_0000);
        assert_eq!(
            result.err().unwrap().errno(),
            ERROR_NOT_ENOUGH_MEMORY as i32
        );
    }
}