1 2bc5ec6b 2023-12-19 op // This is free and unencumbered software released into the public domain.
3 2bc5ec6b 2023-12-19 op // Anyone is free to copy, modify, publish, use, compile, sell, or
4 2bc5ec6b 2023-12-19 op // distribute this software, either in source code form or as a compiled
5 2bc5ec6b 2023-12-19 op // binary, for any purpose, commercial or non-commercial, and by any
8 2bc5ec6b 2023-12-19 op // In jurisdictions that recognize copyright laws, the author or authors
9 2bc5ec6b 2023-12-19 op // of this software dedicate any and all copyright interest in the
10 2bc5ec6b 2023-12-19 op // software to the public domain. We make this dedication for the benefit
11 2bc5ec6b 2023-12-19 op // of the public at large and to the detriment of our heirs and
12 2bc5ec6b 2023-12-19 op // successors. We intend this dedication to be an overt act of
13 2bc5ec6b 2023-12-19 op // relinquishment in perpetuity of all present and future rights to this
14 2bc5ec6b 2023-12-19 op // software under copyright law.
16 2bc5ec6b 2023-12-19 op // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17 2bc5ec6b 2023-12-19 op // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18 2bc5ec6b 2023-12-19 op // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
19 2bc5ec6b 2023-12-19 op // IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
20 2bc5ec6b 2023-12-19 op // OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
21 2bc5ec6b 2023-12-19 op // ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
22 2bc5ec6b 2023-12-19 op // OTHER DEALINGS IN THE SOFTWARE.
24 2bc5ec6b 2023-12-19 op // XXX would be nice to eventually have multiple backends for poll,
25 2bc5ec6b 2023-12-19 op // kqueue, epoll.
30 2bc5ec6b 2023-12-19 op use unix::poll;
31 2bc5ec6b 2023-12-19 op use unix::signal;
33 2bc5ec6b 2023-12-19 op export type event = enum i16 {
38 f2f593e9 2023-12-19 op export def READ = event::READ;
39 f2f593e9 2023-12-19 op export def WRITE = event::WRITE;
41 2bc5ec6b 2023-12-19 op type cb = struct {
42 01312690 2023-12-19 op cb: *fn(io::file, event, nullable *opaque) void,
43 01312690 2023-12-19 op data: nullable *opaque,
46 e9feffee 2023-12-19 op type queue = struct {
47 2bc5ec6b 2023-12-19 op pfds: []poll::pollfd,
51 094f9ac0 2023-12-19 op // XXX we should hide the poll::error and return 'just' an error.
53 094f9ac0 2023-12-19 op export type evloop = struct {
54 094f9ac0 2023-12-19 op add: *fn(*evloop, io::file, event, *fn(io::file, event, nullable *opaque) void, nullable *opaque) void,
55 094f9ac0 2023-12-19 op del: *fn(*evloop, io::file) void,
56 094f9ac0 2023-12-19 op loop: *fn(*evloop) (void | poll::error),
57 7a04e249 2023-12-19 op loopbreak: *fn(*evloop) void,
60 e9feffee 2023-12-19 op type base = struct {
62 e9feffee 2023-12-19 op working: queue,
64 ae2955a0 2023-12-19 op sigpipe: (void | (io::file, io::file)),
65 ae2955a0 2023-12-19 op sigcb: (void | *fn(signal::sig, nullable *opaque) void),
66 ae2955a0 2023-12-19 op sigdata: nullable *opaque,
67 2bc5ec6b 2023-12-19 op stop: bool, // signal to stop
68 2bc5ec6b 2023-12-19 op // TODO: timeout
71 094f9ac0 2023-12-19 op export fn new() *evloop = alloc(base {
75 7a04e249 2023-12-19 op loopbreak = &evloopbreak,
76 094f9ac0 2023-12-19 op sigpipe = void,
81 2bc5ec6b 2023-12-19 op fn ev2poll(ev: event) poll::event = switch (ev) {
82 2bc5ec6b 2023-12-19 op case event::READ => yield poll::event::POLLIN;
83 2bc5ec6b 2023-12-19 op case event::WRITE => yield poll::event::POLLOUT;
86 2bc5ec6b 2023-12-19 op fn poll2ev(ev: i16) event = {
87 2bc5ec6b 2023-12-19 op if ((ev & poll::event::POLLOUT) != 0) {
88 2bc5ec6b 2023-12-19 op return event::WRITE;
90 2bc5ec6b 2023-12-19 op return event::READ;
93 1b9c9f8a 2023-12-19 op export fn evadd(
94 1b9c9f8a 2023-12-19 op evloop: *evloop,
97 a3fc3b94 2023-12-19 op f: *fn(io::file, event, nullable *opaque) void,
98 1b9c9f8a 2023-12-19 op data: nullable *opaque
100 094f9ac0 2023-12-19 op const b = evloop: *base;
101 e9feffee 2023-12-19 op append(b.wip.pfds, poll::pollfd {
103 2bc5ec6b 2023-12-19 op events = ev2poll(ev),
106 e9feffee 2023-12-19 op append(b.wip.cbs, cb {
112 094f9ac0 2023-12-19 op export fn evdel(evloop: *evloop, fd: io::file) void = {
113 094f9ac0 2023-12-19 op const b = evloop: *base;
114 e9feffee 2023-12-19 op for (let i = 0z; i < len(b.wip.pfds); i += 1) {
115 e9feffee 2023-12-19 op if (b.wip.pfds[i].fd != fd) {
118 e9feffee 2023-12-19 op delete(b.wip.pfds[i]);
119 e9feffee 2023-12-19 op delete(b.wip.cbs[i]);
124 094f9ac0 2023-12-19 op fn prepare_queue(b: *base) void = {
125 e9feffee 2023-12-19 op let wip = b.wip;
127 6b358d51 2023-12-19 op delete(b.working.pfds[..]);
128 6b358d51 2023-12-19 op delete(b.working.cbs[..]);
130 e9feffee 2023-12-19 op for (let i = 0z; i < len(wip.pfds); i += 1) {
131 e9feffee 2023-12-19 op append(b.working.pfds, wip.pfds[i]);
132 e9feffee 2023-12-19 op append(b.working.cbs, wip.cbs[i]);
136 094f9ac0 2023-12-19 op export fn loop(evloop: *evloop) (void | poll::error) = {
137 094f9ac0 2023-12-19 op const b = evloop: *base;
139 2bc5ec6b 2023-12-19 op const mask = poll::event::POLLIN | poll::event::POLLOUT |
140 2bc5ec6b 2023-12-19 op poll::event::POLLHUP;
142 2bc5ec6b 2023-12-19 op for (!b.stop) {
143 094f9ac0 2023-12-19 op prepare_queue(b);
145 e9feffee 2023-12-19 op let q = b.working;
147 ae2955a0 2023-12-19 op let n = match (poll::poll(q.pfds, poll::INDEF)) {
148 ae2955a0 2023-12-19 op case let err: errors::interrupted =>
150 ae2955a0 2023-12-19 op case let err: errors::error =>
152 ae2955a0 2023-12-19 op case let n: uint => yield n;
155 e9feffee 2023-12-19 op for (let i = 0z; i < len(q.pfds); i += 1) {
156 e9feffee 2023-12-19 op if ((q.pfds[i].revents & mask) == 0) {
159 e9feffee 2023-12-19 op q.cbs[i].cb(q.pfds[i].fd, poll2ev(q.pfds[i].revents),
160 e9feffee 2023-12-19 op q.cbs[i].data);
165 7a04e249 2023-12-19 op export fn evloopbreak(evloop: *evloop) void = {
166 7a04e249 2023-12-19 op const b = evloop: *base;
167 7a04e249 2023-12-19 op b.stop = true;