Function parse_address

crates/stdlib/src/overlapped.rs:300–384

Parse a Python address tuple to SOCKADDR

(addr_obj: &PyTupleRef, vm: &VirtualMachine)

    /// Parse a Python address tuple to SOCKADDR
    fn parse_address(addr_obj: &PyTupleRef, vm: &VirtualMachine) -> PyResult<(Vec<u8>, i32)> {
        use windows_sys::Win32::Networking::WinSock::{WSAGetLastError, WSAStringToAddressW};

        match addr_obj.len() {
            2 => {
                // IPv4: (host, port)
                let host: PyStrRef = addr_obj[0].clone().try_into_value(vm)?;
                let port: u16 = addr_obj[1].clone().try_to_value(vm)?;

                let mut addr: SOCKADDR_IN = unsafe { core::mem::zeroed() };
                addr.sin_family = AF_INET;

                let host_wide: Vec<u16> = host.as_wtf8().encode_wide().chain([0]).collect();
                let mut addr_len = core::mem::size_of::<SOCKADDR_IN>() as i32;

                let ret = unsafe {
                    WSAStringToAddressW(
                        host_wide.as_ptr(),
                        AF_INET as i32,
                        core::ptr::null(),
                        &mut addr as *mut _ as *mut SOCKADDR,
                        &mut addr_len,
                    )
                };

                if ret < 0 {
                    let err = unsafe { WSAGetLastError() } as u32;
                    return Err(set_from_windows_err(err, vm));
                }

                // Restore port (WSAStringToAddressW overwrites it)
                addr.sin_port = port.to_be();

                let bytes = unsafe {
                    core::slice::from_raw_parts(
                        &addr as *const _ as *const u8,
                        core::mem::size_of::<SOCKADDR_IN>(),
                    )
                };
                Ok((bytes.to_vec(), addr_len))
            }
            4 => {
                // IPv6: (host, port, flowinfo, scope_id)
                let host: PyStrRef = addr_obj[0].clone().try_into_value(vm)?;
                let port: u16 = addr_obj[1].clone().try_to_value(vm)?;
                let flowinfo: u32 = addr_obj[2].clone().try_to_value(vm)?;
                let scope_id: u32 = addr_obj[3].clone().try_to_value(vm)?;

                let mut addr: SOCKADDR_IN6 = unsafe { core::mem::zeroed() };
                addr.sin6_family = AF_INET6;

                let host_wide: Vec<u16> = host.as_wtf8().encode_wide().chain([0]).collect();
                let mut addr_len = core::mem::size_of::<SOCKADDR_IN6>() as i32;

                let ret = unsafe {
                    WSAStringToAddressW(
                        host_wide.as_ptr(),
                        AF_INET6 as i32,