1 ! Copyright (C) 2008, 2009 Slava Pestov.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: accessors classes.struct kernel destructors bit-arrays
4 sequences assocs specialized-arrays math namespaces
5 libc locals fry unix unix.linux.epoll unix.time io.ports
6 io.backend.unix io.backend.unix.multiplexers ;
7 SPECIALIZED-ARRAY: epoll-event
8 IN: io.backend.unix.multiplexers.epoll
10 TUPLE: epoll-mx < mx events ;
12 ! We read up to 256 events at a time. This is an arbitrary
14 CONSTANT: max-events 256
16 : <epoll-mx> ( -- mx )
18 max-events epoll_create dup io-error >>fd
19 max-events <epoll-event-array> >>events ;
21 M: epoll-mx dispose* fd>> close-file ;
23 : make-event ( fd events -- event )
26 swap over data>> fd<< ;
28 :: do-epoll-ctl ( fd mx what events -- )
29 mx fd>> what fd fd events make-event epoll_ctl io-error ;
31 : do-epoll-add ( fd mx events -- )
32 EPOLL_CTL_ADD swap EPOLLONESHOT bitor do-epoll-ctl ;
34 : do-epoll-del ( fd mx events -- )
35 EPOLL_CTL_DEL swap do-epoll-ctl ;
37 M: epoll-mx add-input-callback ( thread fd mx -- )
38 [ EPOLLIN do-epoll-add ] [ call-next-method ] 2bi ;
40 M: epoll-mx add-output-callback ( thread fd mx -- )
41 [ EPOLLOUT do-epoll-add ] [ call-next-method ] 2bi ;
43 M: epoll-mx remove-input-callbacks ( fd mx -- seq )
45 [ call-next-method ] [ EPOLLIN do-epoll-del ] 2bi
48 M: epoll-mx remove-output-callbacks ( fd mx -- seq )
50 [ EPOLLOUT do-epoll-del ] [ call-next-method ] 2bi
53 : wait-event ( mx nanos -- n )
54 [ [ fd>> ] [ events>> ] bi dup length ] [ 1000000 /i ] bi*
55 epoll_wait multiplexer-error ;
57 : handle-event ( event mx -- )
59 [ EPOLLIN EPOLLOUT bitor do-epoll-del ]
60 [ input-available ] [ output-available ] 2tri ;
62 : handle-events ( mx n -- )
63 [ dup events>> ] dip head-slice swap '[ _ handle-event ] each ;
65 M: epoll-mx wait-for-events ( nanos mx -- )
66 swap 60000000 or dupd wait-event handle-events ;