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;
102 562f76f0 2023-12-21 op for (let i = 0z; i < len(b.wip.pfds); i += 1) {
103 562f76f0 2023-12-21 op if (b.wip.pfds[i].fd != fd)
105 562f76f0 2023-12-21 op b.wip.pfds[i].events = ev2poll(ev);
106 562f76f0 2023-12-21 op b.wip.cbs[i] = cb { cb = f, data = data };
110 e9feffee 2023-12-19 op append(b.wip.pfds, poll::pollfd {
112 2bc5ec6b 2023-12-19 op events = ev2poll(ev),
115 e9feffee 2023-12-19 op append(b.wip.cbs, cb {
121 094f9ac0 2023-12-19 op export fn evdel(evloop: *evloop, fd: io::file) void = {
122 094f9ac0 2023-12-19 op const b = evloop: *base;
123 e9feffee 2023-12-19 op for (let i = 0z; i < len(b.wip.pfds); i += 1) {
124 e9feffee 2023-12-19 op if (b.wip.pfds[i].fd != fd) {
127 e9feffee 2023-12-19 op delete(b.wip.pfds[i]);
128 e9feffee 2023-12-19 op delete(b.wip.cbs[i]);
133 094f9ac0 2023-12-19 op fn prepare_queue(b: *base) void = {
134 6b358d51 2023-12-19 op delete(b.working.pfds[..]);
135 6b358d51 2023-12-19 op delete(b.working.cbs[..]);
137 710a6fd8 2023-12-21 op for (let i = 0z; i < len(b.wip.pfds); i += 1) {
138 710a6fd8 2023-12-21 op append(b.working.pfds, b.wip.pfds[i]);
139 710a6fd8 2023-12-21 op append(b.working.cbs, b.wip.cbs[i]);
143 094f9ac0 2023-12-19 op export fn loop(evloop: *evloop) (void | poll::error) = {
144 094f9ac0 2023-12-19 op const b = evloop: *base;
146 2bc5ec6b 2023-12-19 op const mask = poll::event::POLLIN | poll::event::POLLOUT |
147 2bc5ec6b 2023-12-19 op poll::event::POLLHUP;
149 2bc5ec6b 2023-12-19 op for (!b.stop) {
150 094f9ac0 2023-12-19 op prepare_queue(b);
152 e9feffee 2023-12-19 op let q = b.working;
154 ae2955a0 2023-12-19 op let n = match (poll::poll(q.pfds, poll::INDEF)) {
155 ae2955a0 2023-12-19 op case let err: errors::interrupted =>
157 ae2955a0 2023-12-19 op case let err: errors::error =>
159 ae2955a0 2023-12-19 op case let n: uint => yield n;
162 e9feffee 2023-12-19 op for (let i = 0z; i < len(q.pfds); i += 1) {
163 0e544c93 2023-12-21 op // XXX check for POLLNVAL ?
165 e9feffee 2023-12-19 op if ((q.pfds[i].revents & mask) == 0) {
168 e9feffee 2023-12-19 op q.cbs[i].cb(q.pfds[i].fd, poll2ev(q.pfds[i].revents),
169 e9feffee 2023-12-19 op q.cbs[i].data);
174 7a04e249 2023-12-19 op export fn evloopbreak(evloop: *evloop) void = {
175 7a04e249 2023-12-19 op const b = evloop: *base;
176 7a04e249 2023-12-19 op b.stop = true;
