apio/Luna
1
1
Fork 0
Code Issues Pull Requests Actions Projects Releases 8 Wiki Activity

wind: Monitor data on client connections

Browse Source
This commit is contained in:
apio 2023-08-14 13:24:26 +02:00
parent 753a56d14c
commit 2c2c6fbc2d
Signed by: apio
GPG Key ID: B8A7D06E42258954
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
wind/main.cpp
View File

@ -87,6 +87,10 @@ Result<int> luna_main(int argc, char** argv)
TRY(make<Window>(ui::Rect { 600, 130, 350, 250 }, ui::CYAN, "File Manager"_sv));
Vector<os::LocalServer::Client> clients;
Vector<struct pollfd> fds;
TRY(fds.try_append({ .fd = mouse->fd(), .events = POLLIN, .revents = 0 }));
TRY(fds.try_append({ .fd = keyboard->fd(), .events = POLLIN, .revents = 0 }));
TRY(fds.try_append({ .fd = server->fd(), .events = POLLIN, .revents = 0 }));
TRY(os::Security::pledge("stdio rpath unix", NULL));
@ -97,13 +101,9 @@ Result<int> luna_main(int argc, char** argv)
mouse_pointer.draw(screen.canvas());
screen.sync();
struct pollfd fds[] = {
{ .fd = mouse->fd(), .events = POLLIN, .revents = 0 },
{ .fd = keyboard->fd(), .events = POLLIN, .revents = 0 },
{ .fd = server->fd(), .events = POLLIN, .revents = 0 },
};
for (auto& pfd : fds) { pfd.revents = 0; }
int rc = poll(fds, 3, 1000);
int rc = poll(fds.data(), fds.size(), 1000);
if (!rc) continue;
if (rc < 0 && errno != EINTR) { os::println("poll: error: %s", strerror(errno)); }
@ -126,10 +126,22 @@ Result<int> luna_main(int argc, char** argv)
strerror(packet.key)));
}
}
for (usize i = 0; i < clients.size(); i++)
{
if (fds[i + 3].revents & POLLIN) os::println("wind: Client %d sent data!", i);
if (fds[i + 3].revents & POLLHUP)
{
os::println("wind: Client %d disconnected", i);
fds.remove_at(i + 3);
auto client = clients.remove_at(i);
client.disconnect();
}
}
if (fds[2].revents & POLLIN)
{
auto client = TRY(server->accept());
os::println("wind: New client connected!");
TRY(fds.try_append({ .fd = client.fd(), .events = POLLIN, .revents = 0 }));
TRY(clients.try_append(move(client)));
}
}