123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139 |
- /*
- * Copyright (c) 2018-2021, Andreas Kling <kling@serenityos.org>
- *
- * SPDX-License-Identifier: BSD-2-Clause
- */
- #include <LibCore/LocalServer.h>
- #include <LibCore/Notifier.h>
- #include <LibCore/SessionManagement.h>
- #include <LibCore/Socket.h>
- #include <LibCore/System.h>
- #include <LibCore/SystemServerTakeover.h>
- #include <fcntl.h>
- #include <stdio.h>
- #include <sys/socket.h>
- #include <sys/stat.h>
- #include <unistd.h>
- #ifndef SOCK_NONBLOCK
- # include <sys/ioctl.h>
- #endif
- namespace Core {
- LocalServer::LocalServer(EventReceiver* parent)
- : EventReceiver(parent)
- {
- }
- LocalServer::~LocalServer()
- {
- if (m_fd >= 0)
- ::close(m_fd);
- }
- ErrorOr<void> LocalServer::take_over_from_system_server(DeprecatedString const& socket_path)
- {
- if (m_listening)
- return Error::from_string_literal("Core::LocalServer: Can't perform socket takeover when already listening");
- auto const parsed_path = TRY(Core::SessionManagement::parse_path_with_sid(socket_path));
- auto socket = TRY(take_over_socket_from_system_server(parsed_path));
- m_fd = TRY(socket->release_fd());
- m_listening = true;
- setup_notifier();
- return {};
- }
- void LocalServer::setup_notifier()
- {
- m_notifier = Notifier::construct(m_fd, Notifier::Type::Read, this);
- m_notifier->on_activation = [this] {
- if (on_accept) {
- auto maybe_client_socket = accept();
- if (maybe_client_socket.is_error()) {
- dbgln("LocalServer::on_ready_to_read: Error accepting a connection: {}", maybe_client_socket.error());
- if (on_accept_error)
- on_accept_error(maybe_client_socket.release_error());
- return;
- }
- on_accept(maybe_client_socket.release_value());
- }
- };
- }
- bool LocalServer::listen(DeprecatedString const& address)
- {
- if (m_listening)
- return false;
- int rc;
- #ifdef SOCK_NONBLOCK
- m_fd = socket(AF_LOCAL, SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, 0);
- #else
- m_fd = socket(AF_LOCAL, SOCK_STREAM, 0);
- int option = 1;
- ioctl(m_fd, FIONBIO, &option);
- fcntl(m_fd, F_SETFD, FD_CLOEXEC);
- #endif
- VERIFY(m_fd >= 0);
- #ifndef AK_OS_MACOS
- rc = fchmod(m_fd, 0600);
- if (rc < 0) {
- perror("fchmod");
- VERIFY_NOT_REACHED();
- }
- #endif
- auto socket_address = SocketAddress::local(address);
- auto un_optional = socket_address.to_sockaddr_un();
- if (!un_optional.has_value()) {
- perror("bind");
- return false;
- }
- auto un = un_optional.value();
- rc = ::bind(m_fd, (sockaddr const*)&un, sizeof(un));
- if (rc < 0) {
- perror("bind");
- return false;
- }
- rc = ::listen(m_fd, 5);
- if (rc < 0) {
- perror("listen");
- return false;
- }
- m_listening = true;
- setup_notifier();
- return true;
- }
- ErrorOr<NonnullOwnPtr<LocalSocket>> LocalServer::accept()
- {
- VERIFY(m_listening);
- sockaddr_un un;
- socklen_t un_size = sizeof(un);
- #ifndef AK_OS_MACOS
- int accepted_fd = ::accept4(m_fd, (sockaddr*)&un, &un_size, SOCK_NONBLOCK | SOCK_CLOEXEC);
- #else
- int accepted_fd = ::accept(m_fd, (sockaddr*)&un, &un_size);
- #endif
- if (accepted_fd < 0) {
- return Error::from_syscall("accept"sv, -errno);
- }
- #ifdef AK_OS_MACOS
- int option = 1;
- ioctl(m_fd, FIONBIO, &option);
- (void)fcntl(accepted_fd, F_SETFD, FD_CLOEXEC);
- #endif
- return LocalSocket::adopt_fd(accepted_fd, Socket::PreventSIGPIPE::Yes);
- }
- }
|