/* * Copyright (c) 2020, the SerenityOS developers. * * SPDX-License-Identifier: BSD-2-Clause */ #include #include #include #include #include #include #include namespace Gemini { void GeminiJob::start(NonnullRefPtr socket) { VERIFY(!m_socket); VERIFY(is(*socket)); m_socket = static_ptr_cast(socket); m_socket->on_tls_error = [this](TLS::AlertDescription error) { if (error == TLS::AlertDescription::HandshakeFailure) { deferred_invoke([this] { return did_fail(Core::NetworkJob::Error::ProtocolFailed); }); } else if (error == TLS::AlertDescription::DecryptError) { deferred_invoke([this] { return did_fail(Core::NetworkJob::Error::ConnectionFailed); }); } else { deferred_invoke([this] { return did_fail(Core::NetworkJob::Error::TransmissionFailed); }); } }; m_socket->on_tls_finished = [this] { finish_up(); }; m_socket->on_tls_certificate_request = [this](auto&) { if (on_certificate_requested) on_certificate_requested(*this); }; m_socket->set_idle(false); if (m_socket->is_established()) { deferred_invoke([this] { on_socket_connected(); }); } else { m_socket->set_root_certificates(m_override_ca_certificates ? *m_override_ca_certificates : DefaultRootCACertificates::the().certificates()); m_socket->on_tls_connected = [this] { on_socket_connected(); }; bool success = ((TLS::TLSv12&)*m_socket).connect(m_request.url().host(), m_request.url().port_or_default()); if (!success) { deferred_invoke([this] { return did_fail(Core::NetworkJob::Error::ConnectionFailed); }); } } } void GeminiJob::shutdown(ShutdownMode mode) { if (!m_socket) return; if (mode == ShutdownMode::CloseSocket) { m_socket->close(); } else { m_socket->on_tls_ready_to_read = nullptr; m_socket->on_tls_connected = nullptr; m_socket->set_idle(true); m_socket = nullptr; } } void GeminiJob::read_while_data_available(Function read) { while (m_socket->can_read()) { if (read() == IterationDecision::Break) break; } } void GeminiJob::set_certificate(String certificate, String private_key) { if (!m_socket->add_client_key(certificate.bytes(), private_key.bytes())) { dbgln("LibGemini: Failed to set a client certificate"); // FIXME: Do something about this failure VERIFY_NOT_REACHED(); } } void GeminiJob::register_on_ready_to_read(Function callback) { m_socket->on_tls_ready_to_read = [callback = move(callback)](auto&) { callback(); }; } void GeminiJob::register_on_ready_to_write(Function callback) { m_socket->set_on_tls_ready_to_write([callback = move(callback)](auto& tls) { Core::deferred_invoke([&tls] { tls.set_on_tls_ready_to_write(nullptr); }); callback(); }); } bool GeminiJob::can_read_line() const { return m_socket->can_read_line(); } String GeminiJob::read_line(size_t size) { return m_socket->read_line(size); } ByteBuffer GeminiJob::receive(size_t size) { return m_socket->read(size); } bool GeminiJob::can_read() const { return m_socket->can_read(); } bool GeminiJob::eof() const { return m_socket->eof(); } bool GeminiJob::write(ReadonlyBytes bytes) { return m_socket->write(bytes); } }