mirror of
https://github.com/RGBCube/serenity
synced 2025-05-31 10:28:10 +00:00
fe00393941
Everyone used this hook in the same way: immediately accept() on the socket and then do something with the newly accepted fd. This patch simplifies the hook by having LocalServer do the accepting automatically.
47 lines
1.2 KiB
C++
47 lines
1.2 KiB
C++
/*
|
|
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <LibCore/EventLoop.h>
|
|
#include <LibCore/LocalServer.h>
|
|
#include <SQLServer/ClientConnection.h>
|
|
#include <stdio.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
|
|
int main([[maybe_unused]] int argc, [[maybe_unused]] char** argv)
|
|
{
|
|
if (pledge("stdio accept unix rpath wpath cpath", nullptr) < 0) {
|
|
perror("pledge");
|
|
return 1;
|
|
}
|
|
|
|
if (mkdir("/home/anon/sql", 0700) < 0 && errno != EEXIST) {
|
|
perror("mkdir");
|
|
return 1;
|
|
}
|
|
|
|
if (unveil("/home/anon/sql", "rwc") < 0) {
|
|
perror("unveil");
|
|
return 1;
|
|
}
|
|
if (unveil(nullptr, nullptr) < 0) {
|
|
perror("unveil");
|
|
return 1;
|
|
}
|
|
|
|
Core::EventLoop event_loop;
|
|
auto server = Core::LocalServer::construct();
|
|
bool ok = server->take_over_from_system_server();
|
|
VERIFY(ok);
|
|
|
|
server->on_accept = [&](auto client_socket) {
|
|
static int s_next_client_id = 0;
|
|
int client_id = ++s_next_client_id;
|
|
IPC::new_client_connection<SQLServer::ClientConnection>(client_socket, client_id);
|
|
};
|
|
|
|
return event_loop.exec();
|
|
}
|