Function initialize_winsock_extensions

crates/stdlib/src/overlapped.rs:58–139

(vm: &VirtualMachine)

    static TRANSMIT_FILE: std::sync::OnceLock<usize> = std::sync::OnceLock::new();

    fn initialize_winsock_extensions(vm: &VirtualMachine) -> PyResult<()> {
        use windows_sys::Win32::Networking::WinSock::{
            INVALID_SOCKET, IPPROTO_TCP, SIO_GET_EXTENSION_FUNCTION_POINTER, SOCK_STREAM,
            SOCKET_ERROR, WSAGetLastError, WSAIoctl, closesocket, socket,
        };

        // GUIDs for extension functions
        const WSAID_ACCEPTEX: windows_sys::core::GUID = windows_sys::core::GUID {
            data1: 0xb5367df1,
            data2: 0xcbac,
            data3: 0x11cf,
            data4: [0x95, 0xca, 0x00, 0x80, 0x5f, 0x48, 0xa1, 0x92],
        };
        const WSAID_CONNECTEX: windows_sys::core::GUID = windows_sys::core::GUID {
            data1: 0x25a207b9,
            data2: 0xddf3,
            data3: 0x4660,
            data4: [0x8e, 0xe9, 0x76, 0xe5, 0x8c, 0x74, 0x06, 0x3e],
        };
        const WSAID_DISCONNECTEX: windows_sys::core::GUID = windows_sys::core::GUID {
            data1: 0x7fda2e11,
            data2: 0x8630,
            data3: 0x436f,
            data4: [0xa0, 0x31, 0xf5, 0x36, 0xa6, 0xee, 0xc1, 0x57],
        };
        const WSAID_TRANSMITFILE: windows_sys::core::GUID = windows_sys::core::GUID {
            data1: 0xb5367df0,
            data2: 0xcbac,
            data3: 0x11cf,
            data4: [0x95, 0xca, 0x00, 0x80, 0x5f, 0x48, 0xa1, 0x92],
        };

        // Check all four locks to prevent partial initialization
        if ACCEPT_EX.get().is_some()
            && CONNECT_EX.get().is_some()
            && DISCONNECT_EX.get().is_some()
            && TRANSMIT_FILE.get().is_some()
        {
            return Ok(());
        }

        let s = unsafe { socket(AF_INET as i32, SOCK_STREAM, IPPROTO_TCP) };
        if s == INVALID_SOCKET {
            let err = unsafe { WSAGetLastError() } as u32;
            return Err(set_from_windows_err(err, vm));
        }

        let mut dw_bytes: u32 = 0;

        macro_rules! get_extension {
            ($guid:expr, $lock:expr) => {{
                let mut func_ptr: usize = 0;
                let ret = unsafe {
                    WSAIoctl(
                        s,
                        SIO_GET_EXTENSION_FUNCTION_POINTER,
                        &$guid as *const _ as *const _,
                        core::mem::size_of_val(&$guid) as u32,