diff options
-rw-r--r-- | lib/pure/ioselects/ioselectors_epoll.nim | 52 | ||||
-rw-r--r-- | lib/pure/ioselects/ioselectors_kqueue.nim | 21 | ||||
-rw-r--r-- | lib/pure/ioselects/ioselectors_poll.nim | 10 | ||||
-rw-r--r-- | lib/pure/ioselects/ioselectors_select.nim | 28 |
4 files changed, 58 insertions, 53 deletions
diff --git a/lib/pure/ioselects/ioselectors_epoll.nim b/lib/pure/ioselects/ioselectors_epoll.nim index cb555f001..f8feb7361 100644 --- a/lib/pure/ioselects/ioselectors_epoll.nim +++ b/lib/pure/ioselects/ioselectors_epoll.nim @@ -131,11 +131,12 @@ proc newSelector*[T](): Selector[T] = result.fds = newSeq[SelectorKey[T]](maxFD) proc close*[T](s: Selector[T]) = - if posix.close(s.epollFD) != 0: - raiseIOSelectorsError(osLastError()) + let res = posix.close(s.epollFD) when hasThreadSupport: deallocSharedArray(s.fds) deallocShared(cast[pointer](s)) + if res != 0: + raiseIOSelectorsError(osLastError()) template clearKey[T](key: ptr SelectorKey[T]) = var empty: T @@ -157,9 +158,10 @@ proc setEvent*(ev: SelectEvent) = raiseIOSelectorsError(osLastError()) proc close*(ev: SelectEvent) = - if posix.close(ev.efd) == -1: - raiseIOSelectorsError(osLastError()) + let res = posix.close(ev.efd) deallocShared(cast[pointer](ev)) + if res != 0: + raiseIOSelectorsError(osLastError()) template checkFd(s, f) = if f >= s.maxFD: @@ -176,7 +178,7 @@ proc registerHandle*[T](s: Selector[T], fd: SocketHandle, epv.data.u64 = fdi.uint if Event.Read in events: epv.events = epv.events or EPOLLIN if Event.Write in events: epv.events = epv.events or EPOLLOUT - if epoll_ctl(s.epollFD, EPOLL_CTL_ADD, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_ADD, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) inc(s.count) @@ -196,15 +198,15 @@ proc updateHandle*[T](s: Selector[T], fd: SocketHandle, events: set[Event]) = if Event.Write in events: epv.events = epv.events or EPOLLOUT if pkey.events == {}: - if epoll_ctl(s.epollFD, EPOLL_CTL_ADD, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_ADD, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) inc(s.count) else: if events != {}: - if epoll_ctl(s.epollFD, EPOLL_CTL_MOD, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_MOD, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) else: - if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) dec(s.count) pkey.events = events @@ -219,20 +221,20 @@ proc unregister*[T](s: Selector[T], fd: int|SocketHandle) = when not defined(android): if pkey.events * {Event.Read, Event.Write} != {}: var epv = epoll_event() - if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) dec(s.count) elif Event.Timer in pkey.events: if Event.Finished notin pkey.events: var epv = epoll_event() - if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) dec(s.count) - if posix.close(cint(fdi)) == -1: + if posix.close(cint(fdi)) != 0: raiseIOSelectorsError(osLastError()) elif Event.Signal in pkey.events: var epv = epoll_event() - if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) var nmask, omask: Sigset discard sigemptyset(nmask) @@ -240,12 +242,12 @@ proc unregister*[T](s: Selector[T], fd: int|SocketHandle) = discard sigaddset(nmask, cint(s.fds[fdi].param)) unblockSignals(nmask, omask) dec(s.count) - if posix.close(cint(fdi)) == -1: + if posix.close(cint(fdi)) != 0: raiseIOSelectorsError(osLastError()) elif Event.Process in pkey.events: if Event.Finished notin pkey.events: var epv = epoll_event() - if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) var nmask, omask: Sigset discard sigemptyset(nmask) @@ -253,21 +255,21 @@ proc unregister*[T](s: Selector[T], fd: int|SocketHandle) = discard sigaddset(nmask, SIGCHLD) unblockSignals(nmask, omask) dec(s.count) - if posix.close(cint(fdi)) == -1: + if posix.close(cint(fdi)) != 0: raiseIOSelectorsError(osLastError()) else: if pkey.events * {Event.Read, Event.Write} != {}: var epv = epoll_event() - if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) dec(s.count) elif Event.Timer in pkey.events: if Event.Finished notin pkey.events: var epv = epoll_event() - if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) dec(s.count) - if posix.close(cint(fdi)) == -1: + if posix.close(cint(fdi)) != 0: raiseIOSelectorsError(osLastError()) clearKey(pkey) @@ -278,7 +280,7 @@ proc unregister*[T](s: Selector[T], ev: SelectEvent) = doAssert(pkey.ident != 0) doAssert(Event.User in pkey.events) var epv = epoll_event() - if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) dec(s.count) clearKey(pkey) @@ -312,9 +314,9 @@ proc registerTimer*[T](s: Selector[T], timeout: int, oneshot: bool, new_ts.it_value.tv_sec = new_ts.it_interval.tv_sec new_ts.it_value.tv_nsec = new_ts.it_interval.tv_nsec - if timerfd_settime(fdi.cint, cint(0), new_ts, old_ts) == -1: + if timerfd_settime(fdi.cint, cint(0), new_ts, old_ts) != 0: raiseIOSelectorsError(osLastError()) - if epoll_ctl(s.epollFD, EPOLL_CTL_ADD, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_ADD, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) s.setKey(fdi, events, 0, data) inc(s.count) @@ -342,7 +344,7 @@ when not defined(android): var epv = epoll_event(events: EPOLLIN or EPOLLRDHUP) epv.data.u64 = fdi.uint - if epoll_ctl(s.epollFD, EPOLL_CTL_ADD, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_ADD, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) s.setKey(fdi, {Event.Signal}, signal, data) inc(s.count) @@ -370,7 +372,7 @@ when not defined(android): var epv = epoll_event(events: EPOLLIN or EPOLLRDHUP) epv.data.u64 = fdi.uint epv.events = EPOLLIN or EPOLLRDHUP - if epoll_ctl(s.epollFD, EPOLL_CTL_ADD, fdi.cint, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_ADD, fdi.cint, addr epv) != 0: raiseIOSelectorsError(osLastError()) s.setKey(fdi, {Event.Process, Event.Oneshot}, pid, data) inc(s.count) @@ -382,7 +384,7 @@ proc registerEvent*[T](s: Selector[T], ev: SelectEvent, data: T) = s.setKey(fdi, {Event.User}, 0, data) var epv = epoll_event(events: EPOLLIN or EPOLLRDHUP) epv.data.u64 = ev.efd.uint - if epoll_ctl(s.epollFD, EPOLL_CTL_ADD, ev.efd, addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_ADD, ev.efd, addr epv) != 0: raiseIOSelectorsError(osLastError()) inc(s.count) @@ -480,7 +482,7 @@ proc selectInto*[T](s: Selector[T], timeout: int, if Event.Oneshot in pkey.events: var epv = epoll_event() - if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, cint(fdi), addr epv) == -1: + if epoll_ctl(s.epollFD, EPOLL_CTL_DEL, cint(fdi), addr epv) != 0: raiseIOSelectorsError(osLastError()) # we will not clear key until it will be unregistered, so # application can obtain data, but we will decrease counter, diff --git a/lib/pure/ioselects/ioselectors_kqueue.nim b/lib/pure/ioselects/ioselectors_kqueue.nim index f4111263a..578bce063 100644 --- a/lib/pure/ioselects/ioselectors_kqueue.nim +++ b/lib/pure/ioselects/ioselectors_kqueue.nim @@ -116,12 +116,13 @@ proc newSelector*[T](): Selector[T] = raiseIOSelectorsError(osLastError()) proc close*[T](s: Selector[T]) = - if posix.close(s.kqFD) != 0: - raiseIOSelectorsError(osLastError()) + let res = posix.close(s.kqFD) when hasThreadSupport: deinitLock(s.changesLock) deallocSharedArray(s.fds) deallocShared(cast[pointer](s)) + if res != 0: + raiseIOSelectorsError(osLastError()) template clearKey[T](key: ptr SelectorKey[T]) = var empty: T @@ -131,7 +132,7 @@ template clearKey[T](key: ptr SelectorKey[T]) = proc newSelectEvent*(): SelectEvent = var fds: array[2, cint] - if posix.pipe(fds) == -1: + if posix.pipe(fds) != 0: raiseIOSelectorsError(osLastError()) setNonBlocking(fds[0]) setNonBlocking(fds[1]) @@ -145,11 +146,11 @@ proc setEvent*(ev: SelectEvent) = raiseIOSelectorsError(osLastError()) proc close*(ev: SelectEvent) = - if posix.close(cint(ev.rfd)) == -1: - raiseIOSelectorsError(osLastError()) - if posix.close(cint(ev.wfd)) == -1: - raiseIOSelectorsError(osLastError()) + let res1 = posix.close(rfd) + let res2 = posix.close(wfd) deallocShared(cast[pointer](ev)) + if res1 != 0 or res2 != 0: + raiseIOSelectorsError(osLastError()) template checkFd(s, f) = if f >= s.maxFD: @@ -386,7 +387,7 @@ proc unregister*[T](s: Selector[T], fd: int|SocketHandle) = when not declared(CACHE_EVENTS): flushKQueue(s) dec(s.count) - if posix.close(cint(pkey.ident)) == -1: + if posix.close(cint(pkey.ident)) != 0: raiseIOSelectorsError(osLastError()) elif Event.Signal in pkey.events: var nmask, omask: Sigset @@ -400,7 +401,7 @@ proc unregister*[T](s: Selector[T], fd: int|SocketHandle) = when not declared(CACHE_EVENTS): flushKQueue(s) dec(s.count) - if posix.close(cint(pkey.ident)) == -1: + if posix.close(cint(pkey.ident)) != 0: raiseIOSelectorsError(osLastError()) elif Event.Process in pkey.events: if Event.Finished notin pkey.events: @@ -408,7 +409,7 @@ proc unregister*[T](s: Selector[T], fd: int|SocketHandle) = when not declared(CACHE_EVENTS): flushKQueue(s) dec(s.count) - if posix.close(cint(pkey.ident)) == -1: + if posix.close(cint(pkey.ident)) != 0: raiseIOSelectorsError(osLastError()) elif Event.Vnode in pkey.events: modifyKQueue(s, uint(fdi), EVFILT_VNODE, EV_DELETE, 0, 0, nil) diff --git a/lib/pure/ioselects/ioselectors_poll.nim b/lib/pure/ioselects/ioselectors_poll.nim index 26fe60b54..9c6f9796f 100644 --- a/lib/pure/ioselects/ioselectors_poll.nim +++ b/lib/pure/ioselects/ioselectors_poll.nim @@ -199,7 +199,7 @@ proc unregister*[T](s: Selector[T], ev: SelectEvent) = proc newSelectEvent*(): SelectEvent = var fds: array[2, cint] - if posix.pipe(fds) == -1: + if posix.pipe(fds) != 0: raiseIOSelectorsError(osLastError()) setNonBlocking(fds[0]) setNonBlocking(fds[1]) @@ -213,11 +213,11 @@ proc setEvent*(ev: SelectEvent) = raiseIOSelectorsError(osLastError()) proc close*(ev: SelectEvent) = - if posix.close(cint(ev.rfd)) == -1: - raiseIOSelectorsError(osLastError()) - if posix.close(cint(ev.wfd)) == -1: - raiseIOSelectorsError(osLastError()) + let res1 = posix.close(ev.rfd) + let res2 = posix.close(ev.wfd) deallocShared(cast[pointer](ev)) + if res1 != 0 or res2 != 0: + raiseIOSelectorsError(osLastError()) proc selectInto*[T](s: Selector[T], timeout: int, results: var openarray[ReadyKey]): int = diff --git a/lib/pure/ioselects/ioselectors_select.nim b/lib/pure/ioselects/ioselectors_select.nim index 19d68f0fe..7a7d23982 100644 --- a/lib/pure/ioselects/ioselectors_select.nim +++ b/lib/pure/ioselects/ioselectors_select.nim @@ -122,17 +122,17 @@ when defined(windows): sizeof(saddr).SockLen) < 0'i32: raiseIOSelectorsError(osLastError()) - if winlean.listen(ssock, 1) == -1: + if winlean.listen(ssock, 1) != 0: raiseIOSelectorsError(osLastError()) var namelen = sizeof(saddr).SockLen if getsockname(ssock, cast[ptr SockAddr](addr(saddr)), - addr(namelen)) == -1'i32: + addr(namelen)) != 0'i32: raiseIOSelectorsError(osLastError()) saddr.sin_addr.s_addr = 0x0100007F if winlean.connect(wsock, cast[ptr SockAddr](addr(saddr)), - sizeof(saddr).SockLen) == -1: + sizeof(saddr).SockLen) != 0: raiseIOSelectorsError(osLastError()) namelen = sizeof(saddr).SockLen rsock = winlean.accept(ssock, cast[ptr SockAddr](addr(saddr)), @@ -140,14 +140,14 @@ when defined(windows): if rsock == SocketHandle(-1): raiseIOSelectorsError(osLastError()) - if winlean.closesocket(ssock) == -1: + if winlean.closesocket(ssock) != 0: raiseIOSelectorsError(osLastError()) var mode = clong(1) - if ioctlsocket(rsock, FIONBIO, addr(mode)) == -1: + if ioctlsocket(rsock, FIONBIO, addr(mode)) != 0: raiseIOSelectorsError(osLastError()) mode = clong(1) - if ioctlsocket(wsock, FIONBIO, addr(mode)) == -1: + if ioctlsocket(wsock, FIONBIO, addr(mode)) != 0: raiseIOSelectorsError(osLastError()) result = cast[SelectEvent](allocShared0(sizeof(SelectEventImpl))) @@ -161,14 +161,16 @@ when defined(windows): raiseIOSelectorsError(osLastError()) proc close*(ev: SelectEvent) = - discard winlean.closesocket(ev.rsock) - discard winlean.closesocket(ev.wsock) + let res1 = winlean.closesocket(ev.rsock) + let res2 = winlean.closesocket(ev.wsock) deallocShared(cast[pointer](ev)) + if res1 != 0 or res2 != 0: + raiseIOSelectorsError(osLastError()) else: proc newSelectEvent*(): SelectEvent = var fds: array[2, cint] - if posix.pipe(fds) == -1: + if posix.pipe(fds) != 0: raiseIOSelectorsError(osLastError()) setNonBlocking(fds[0]) setNonBlocking(fds[1]) @@ -182,11 +184,11 @@ else: raiseIOSelectorsError(osLastError()) proc close*(ev: SelectEvent) = - if posix.close(cint(ev.rsock)) == -1: - raiseIOSelectorsError(osLastError()) - if posix.close(cint(ev.wsock)) == -1: - raiseIOSelectorsError(osLastError()) + let res1 = posix.close(cint(ev.rsock)) + let res2 = posix.close(cint(ev.wsock)) deallocShared(cast[pointer](ev)) + if res1 != 0 or res2 != 0: + raiseIOSelectorsError(osLastError()) proc setSelectKey[T](s: Selector[T], fd: SocketHandle, events: set[Event], data: T) = |