CHttpJob.cpp 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. #include <LibCore/CHttpJob.h>
  2. #include <LibCore/CHttpResponse.h>
  3. #include <LibCore/CTCPSocket.h>
  4. #include <stdio.h>
  5. #include <unistd.h>
  6. CHttpJob::CHttpJob(const CHttpRequest& request)
  7. : m_request(request)
  8. {
  9. }
  10. CHttpJob::~CHttpJob()
  11. {
  12. }
  13. void CHttpJob::on_socket_connected()
  14. {
  15. auto raw_request = m_request.to_raw_request();
  16. #if 0
  17. dbg() << "CHttpJob: raw_request:";
  18. dbg() << String::copy(raw_request).characters();
  19. #endif
  20. bool success = m_socket->send(raw_request);
  21. if (!success)
  22. return deferred_invoke([this](auto&) { did_fail(CNetworkJob::Error::TransmissionFailed); });
  23. m_socket->on_ready_to_read = [&] {
  24. if (m_state == State::InStatus) {
  25. if (!m_socket->can_read_line())
  26. return;
  27. auto line = m_socket->read_line(PAGE_SIZE);
  28. if (line.is_null()) {
  29. fprintf(stderr, "CHttpJob: Expected HTTP status\n");
  30. return deferred_invoke([this](auto&) { did_fail(CNetworkJob::Error::TransmissionFailed); });
  31. }
  32. auto parts = String::copy(line, Chomp).split(' ');
  33. if (parts.size() < 3) {
  34. fprintf(stderr, "CHttpJob: Expected 3-part HTTP status, got '%s'\n", line.pointer());
  35. return deferred_invoke([this](auto&) { did_fail(CNetworkJob::Error::ProtocolFailed); });
  36. }
  37. bool ok;
  38. m_code = parts[1].to_uint(ok);
  39. if (!ok) {
  40. fprintf(stderr, "CHttpJob: Expected numeric HTTP status\n");
  41. return deferred_invoke([this](auto&) { did_fail(CNetworkJob::Error::ProtocolFailed); });
  42. }
  43. m_state = State::InHeaders;
  44. return;
  45. }
  46. if (m_state == State::InHeaders) {
  47. if (!m_socket->can_read_line())
  48. return;
  49. auto line = m_socket->read_line(PAGE_SIZE);
  50. if (line.is_null()) {
  51. fprintf(stderr, "CHttpJob: Expected HTTP header\n");
  52. return did_fail(CNetworkJob::Error::ProtocolFailed);
  53. }
  54. auto chomped_line = String::copy(line, Chomp);
  55. if (chomped_line.is_empty()) {
  56. m_state = State::InBody;
  57. return;
  58. }
  59. auto parts = chomped_line.split(':');
  60. if (parts.is_empty()) {
  61. fprintf(stderr, "CHttpJob: Expected HTTP header with key/value\n");
  62. return deferred_invoke([this](auto&) { did_fail(CNetworkJob::Error::ProtocolFailed); });
  63. }
  64. auto name = parts[0];
  65. if (chomped_line.length() < name.length() + 2) {
  66. fprintf(stderr, "CHttpJob: Malformed HTTP header: '%s' (%d)\n", chomped_line.characters(), chomped_line.length());
  67. return deferred_invoke([this](auto&) { did_fail(CNetworkJob::Error::ProtocolFailed); });
  68. }
  69. auto value = chomped_line.substring(name.length() + 2, chomped_line.length() - name.length() - 2);
  70. m_headers.set(name, value);
  71. dbg() << "CHttpJob: [" << name << "] = '" << value << "'";
  72. return;
  73. }
  74. ASSERT(m_state == State::InBody);
  75. ASSERT(m_socket->can_read());
  76. auto payload = m_socket->receive(PAGE_SIZE);
  77. if (!payload) {
  78. if (m_socket->eof())
  79. return finish_up();
  80. return deferred_invoke([this](auto&) { did_fail(CNetworkJob::Error::ProtocolFailed); });
  81. }
  82. m_received_buffers.append(payload);
  83. m_received_size += payload.size();
  84. auto content_length_header = m_headers.get("Content-Length");
  85. if (content_length_header.has_value()) {
  86. bool ok;
  87. if (m_received_size >= content_length_header.value().to_uint(ok) && ok)
  88. return finish_up();
  89. }
  90. };
  91. }
  92. void CHttpJob::finish_up()
  93. {
  94. m_state = State::Finished;
  95. auto flattened_buffer = ByteBuffer::create_uninitialized(m_received_size);
  96. u8* flat_ptr = flattened_buffer.data();
  97. for (auto& received_buffer : m_received_buffers) {
  98. memcpy(flat_ptr, received_buffer.data(), received_buffer.size());
  99. flat_ptr += received_buffer.size();
  100. }
  101. m_received_buffers.clear();
  102. auto response = CHttpResponse::create(m_code, move(m_headers), move(flattened_buffer));
  103. deferred_invoke([this, response](auto&) {
  104. did_finish(move(response));
  105. });
  106. }
  107. void CHttpJob::start()
  108. {
  109. ASSERT(!m_socket);
  110. m_socket = new CTCPSocket(this);
  111. m_socket->on_connected = [this] {
  112. dbg() << "CHttpJob: on_connected callback";
  113. on_socket_connected();
  114. };
  115. bool success = m_socket->connect(m_request.hostname(), m_request.port());
  116. if (!success)
  117. return did_fail(CNetworkJob::Error::ConnectionFailed);
  118. }