HttpsJob.cpp 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. /*
  2. * Copyright (c) 2020, the SerenityOS developers.
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <AK/Debug.h>
  7. #include <LibCore/EventLoop.h>
  8. #include <LibHTTP/HttpResponse.h>
  9. #include <LibHTTP/HttpsJob.h>
  10. #include <LibTLS/TLSv12.h>
  11. #include <stdio.h>
  12. #include <unistd.h>
  13. namespace HTTP {
  14. void HttpsJob::start()
  15. {
  16. VERIFY(!m_socket);
  17. m_socket = TLS::TLSv12::construct(this);
  18. m_socket->set_root_certificates(m_override_ca_certificates ? *m_override_ca_certificates : DefaultRootCACertificates::the().certificates());
  19. m_socket->on_tls_connected = [this] {
  20. dbgln_if(HTTPSJOB_DEBUG, "HttpsJob: on_connected callback");
  21. on_socket_connected();
  22. };
  23. m_socket->on_tls_error = [&](TLS::AlertDescription error) {
  24. if (error == TLS::AlertDescription::HandshakeFailure) {
  25. deferred_invoke([this](auto&) {
  26. return did_fail(Core::NetworkJob::Error::ProtocolFailed);
  27. });
  28. } else if (error == TLS::AlertDescription::DecryptError) {
  29. deferred_invoke([this](auto&) {
  30. return did_fail(Core::NetworkJob::Error::ConnectionFailed);
  31. });
  32. } else {
  33. deferred_invoke([this](auto&) {
  34. return did_fail(Core::NetworkJob::Error::TransmissionFailed);
  35. });
  36. }
  37. };
  38. m_socket->on_tls_finished = [&] {
  39. finish_up();
  40. };
  41. m_socket->on_tls_certificate_request = [this](auto&) {
  42. if (on_certificate_requested)
  43. on_certificate_requested(*this);
  44. };
  45. bool success = ((TLS::TLSv12&)*m_socket).connect(m_request.url().host(), m_request.url().port());
  46. if (!success) {
  47. deferred_invoke([this](auto&) {
  48. return did_fail(Core::NetworkJob::Error::ConnectionFailed);
  49. });
  50. }
  51. }
  52. void HttpsJob::shutdown()
  53. {
  54. if (!m_socket)
  55. return;
  56. m_socket->on_tls_ready_to_read = nullptr;
  57. m_socket->on_tls_connected = nullptr;
  58. remove_child(*m_socket);
  59. m_socket = nullptr;
  60. }
  61. void HttpsJob::set_certificate(String certificate, String private_key)
  62. {
  63. if (!m_socket->add_client_key(certificate.bytes(), private_key.bytes())) {
  64. dbgln("LibHTTP: Failed to set a client certificate");
  65. // FIXME: Do something about this failure
  66. VERIFY_NOT_REACHED();
  67. }
  68. }
  69. void HttpsJob::read_while_data_available(Function<IterationDecision()> read)
  70. {
  71. while (m_socket->can_read()) {
  72. if (read() == IterationDecision::Break)
  73. break;
  74. }
  75. }
  76. void HttpsJob::register_on_ready_to_read(Function<void()> callback)
  77. {
  78. m_socket->on_tls_ready_to_read = [callback = move(callback)](auto&) {
  79. callback();
  80. };
  81. }
  82. void HttpsJob::register_on_ready_to_write(Function<void()> callback)
  83. {
  84. m_socket->on_tls_ready_to_write = [callback = move(callback)](auto&) {
  85. callback();
  86. };
  87. }
  88. bool HttpsJob::can_read_line() const
  89. {
  90. return m_socket->can_read_line();
  91. }
  92. String HttpsJob::read_line(size_t size)
  93. {
  94. return m_socket->read_line(size);
  95. }
  96. ByteBuffer HttpsJob::receive(size_t size)
  97. {
  98. return m_socket->read(size);
  99. }
  100. bool HttpsJob::can_read() const
  101. {
  102. return m_socket->can_read();
  103. }
  104. bool HttpsJob::eof() const
  105. {
  106. return m_socket->eof();
  107. }
  108. bool HttpsJob::write(ReadonlyBytes data)
  109. {
  110. return m_socket->write(data);
  111. }
  112. }