# # Asyncdispatch2 Handles # (c) Copyright 2018 # Status Research & Development GmbH # # Licensed under either of # Apache License, version 2.0, (LICENSE-APACHEv2) # MIT license (LICENSE-MIT) import net, nativesockets, asyncloop when defined(windows): import winlean const asyncInvalidSocket* = AsyncFD(SocketHandle(-1)) else: import posix const asyncInvalidSocket* = AsyncFD(posix.INVALID_SOCKET) proc setSocketBlocking*(s: SocketHandle, blocking: bool): bool = ## Sets blocking mode on socket. when defined(windows): result = true var mode = clong(ord(not blocking)) if ioctlsocket(s, FIONBIO, addr(mode)) == -1: result = false else: result = true var x: int = fcntl(s, F_GETFL, 0) if x == -1: result = false else: var mode = if blocking: x and not O_NONBLOCK else: x or O_NONBLOCK if fcntl(s, F_SETFL, mode) == -1: result = false proc setSockOpt*(socket: SocketHandle | AsyncFD, level, optname, optval: int): bool = ## `setsockopt()` for integer options. ## Returns ``true`` on success, ``false`` on error. result = true var value = cint(optval) if setsockopt(SocketHandle(socket), cint(level), cint(optname), addr(value), sizeof(value).SockLen) < 0'i32: result = false proc getSockOpt*(socket: SocketHandle | AsyncFD, level, optname: int, value: var int): bool = ## `getsockopt()` for integer options. var res: cint var size = sizeof(res).SockLen result = true if getsockopt(SocketHandle(socket), cint(level), cint(optname), addr(res), addr(size)) < 0'i32: return false value = int(res) proc getSocketError*(socket: SocketHandle | AsyncFD, err: var int): bool = if not getSockOpt(socket, cint(SOL_SOCKET), cint(SO_ERROR), err): result = false else: result = true proc createAsyncSocket*(domain: Domain, sockType: SockType, protocol: Protocol): AsyncFD = ## Creates new asynchronous socket. ## Returns ``asyncInvalidSocket`` on error. let handle = createNativeSocket(domain, sockType, protocol) if handle == osInvalidSocket: return asyncInvalidSocket if not setSocketBlocking(handle, false): close(handle) return asyncInvalidSocket when defined(macosx) and not defined(nimdoc): if not handle.setSockOpt(SOL_SOCKET, SO_NOSIGPIPE, 1): close(handle) return asyncInvalidSocket result = AsyncFD(handle) register(result) proc wrapAsyncSocket*(sock: SocketHandle): AsyncFD = ## Wraps normal socket to asynchronous socket. ## Return ``asyncInvalidSocket`` on error. if not setSocketBlocking(sock, false): close(sock) return asyncInvalidSocket when defined(macosx) and not defined(nimdoc): if not sock.setSockOpt(SOL_SOCKET, SO_NOSIGPIPE, 1): close(sock) return asyncInvalidSocket result = AsyncFD(sock) register(result) proc closeAsyncSocket*(s: AsyncFD) {.inline.} = unregister(s) close(SocketHandle(s))