open-coroutine-core 0.7.0

The open-coroutine is a simple, efficient and generic coroutine library.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

use std::ffi::c_int;
use windows_sys::core::PSTR;
use crate::syscall::windows::{RECV_TIME_LIMIT, SEND_TIME_LIMIT};
use windows_sys::Win32::Networking::WinSock::{SO_RCVTIMEO, SO_SNDTIMEO, SOCKET, SOL_SOCKET};

trait SetsockoptSyscall {
    extern "system" fn setsockopt(
        &self,
        fn_ptr: Option<&extern "system" fn(SOCKET, c_int, c_int, PSTR, c_int) -> c_int>,
        socket: SOCKET,
        level: c_int,
        name: c_int,
        value: PSTR,
        option_len: c_int
    ) -> c_int;
}

impl_syscall!(SetsockoptSyscallFacade, NioSetsockoptSyscall, RawSetsockoptSyscall,
    setsockopt(socket: SOCKET, level: c_int, name: c_int, value: PSTR, option_len: c_int) -> c_int
);

impl_facade!(SetsockoptSyscallFacade, SetsockoptSyscall,
    setsockopt(socket: SOCKET, level: c_int, name: c_int, value: PSTR, option_len: c_int) -> c_int
);

#[repr(C)]
#[derive(Debug, Default)]
struct NioSetsockoptSyscall<I: SetsockoptSyscall> {
    inner: I,
}

#[allow(clippy::cast_ptr_alignment)]
impl<I: SetsockoptSyscall> SetsockoptSyscall for NioSetsockoptSyscall<I> {
    extern "system" fn setsockopt(
        &self,
        fn_ptr: Option<&extern "system" fn(SOCKET, c_int, c_int, PSTR, c_int) -> c_int>,
        socket: SOCKET,
        level: c_int,
        name: c_int,
        value: PSTR,
        option_len: c_int
    ) -> c_int {
        let r= self.inner.setsockopt(fn_ptr, socket, level, name, value, option_len);
        if 0 == r && SOL_SOCKET == level {
            if SO_SNDTIMEO == name {
                let ms = unsafe { *value.cast::<c_int>() };
                let mut time_limit = u64::try_from(ms)
                    .expect("overflow")
                    .saturating_mul(1_000_000);
                if 0 == time_limit {
                    // 取消超时
                    time_limit = u64::MAX;
                }
                assert!(SEND_TIME_LIMIT.insert(socket, time_limit).is_none());
            } else if SO_RCVTIMEO == name {
                let ms = unsafe { *value.cast::<c_int>() };
                let mut time_limit = u64::try_from(ms)
                    .expect("overflow")
                    .saturating_mul(1_000_000);
                if 0 == time_limit {
                    // 取消超时
                    time_limit = u64::MAX;
                }
                assert!(RECV_TIME_LIMIT.insert(socket, time_limit).is_none());
            }
        }
        r
    }
}

impl_raw!(RawSetsockoptSyscall, SetsockoptSyscall, windows_sys::Win32::Networking::WinSock,
    setsockopt(socket: SOCKET, level: c_int, name: c_int, value: PSTR, option_len: c_int) -> c_int
);