| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "mojo/services/network/http_connection_impl.h" | 5 #include "mojo/services/network/http_connection_impl.h" |
| 6 | 6 |
| 7 #include <limits> | 7 #include <limits> |
| 8 | 8 |
| 9 #include "base/bind_helpers.h" | 9 #include "base/bind_helpers.h" |
| 10 #include "base/callback.h" | 10 #include "base/callback.h" |
| (...skipping 65 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 76 | 76 |
| 77 ScopedDataPipeConsumerHandle consumer_; | 77 ScopedDataPipeConsumerHandle consumer_; |
| 78 common::HandleWatcher watcher_; | 78 common::HandleWatcher watcher_; |
| 79 CompletionCallback completion_callback_; | 79 CompletionCallback completion_callback_; |
| 80 scoped_ptr<std::string> buffer_; | 80 scoped_ptr<std::string> buffer_; |
| 81 | 81 |
| 82 DISALLOW_COPY_AND_ASSIGN(SimpleDataPipeReader); | 82 DISALLOW_COPY_AND_ASSIGN(SimpleDataPipeReader); |
| 83 }; | 83 }; |
| 84 | 84 |
| 85 template <> | 85 template <> |
| 86 struct TypeConverter<URLRequestPtr, net::HttpServerRequestInfo> { | 86 struct TypeConverter<HttpRequestPtr, net::HttpServerRequestInfo> { |
| 87 static URLRequestPtr Convert(const net::HttpServerRequestInfo& obj) { | 87 static HttpRequestPtr Convert(const net::HttpServerRequestInfo& obj) { |
| 88 URLRequestPtr request(URLRequest::New()); | 88 HttpRequestPtr request(HttpRequest::New()); |
| 89 request->method = obj.method; |
| 89 request->url = obj.path; | 90 request->url = obj.path; |
| 90 request->method = obj.method; | |
| 91 request->headers.resize(obj.headers.size()); | 91 request->headers.resize(obj.headers.size()); |
| 92 size_t index = 0; | 92 size_t index = 0; |
| 93 for (const auto& item : obj.headers) { | 93 for (const auto& item : obj.headers) { |
| 94 HTTPHeaderPtr header(HTTPHeader::New()); | 94 HttpHeaderPtr header(HttpHeader::New()); |
| 95 header->name = item.first; | 95 header->name = item.first; |
| 96 header->value = item.second; | 96 header->value = item.second; |
| 97 request->headers[index++] = header.Pass(); | 97 request->headers[index++] = header.Pass(); |
| 98 } | 98 } |
| 99 if (!obj.data.empty()) { | 99 if (!obj.data.empty()) { |
| 100 uint32_t num_bytes = static_cast<uint32_t>(obj.data.size()); | 100 uint32_t num_bytes = static_cast<uint32_t>(obj.data.size()); |
| 101 MojoCreateDataPipeOptions options; | 101 MojoCreateDataPipeOptions options; |
| 102 options.struct_size = sizeof(MojoCreateDataPipeOptions); | 102 options.struct_size = sizeof(MojoCreateDataPipeOptions); |
| 103 options.flags = MOJO_CREATE_DATA_PIPE_OPTIONS_FLAG_NONE; | 103 options.flags = MOJO_CREATE_DATA_PIPE_OPTIONS_FLAG_NONE; |
| 104 options.element_num_bytes = 1; | 104 options.element_num_bytes = 1; |
| 105 options.capacity_num_bytes = num_bytes; | 105 options.capacity_num_bytes = num_bytes; |
| 106 DataPipe data_pipe(options); | 106 DataPipe data_pipe(options); |
| 107 request->body.push_back(data_pipe.consumer_handle.Pass()); | 107 request->body = data_pipe.consumer_handle.Pass(); |
| 108 MojoResult result = | 108 MojoResult result = |
| 109 WriteDataRaw(data_pipe.producer_handle.get(), obj.data.data(), | 109 WriteDataRaw(data_pipe.producer_handle.get(), obj.data.data(), |
| 110 &num_bytes, MOJO_WRITE_DATA_FLAG_ALL_OR_NONE); | 110 &num_bytes, MOJO_WRITE_DATA_FLAG_ALL_OR_NONE); |
| 111 CHECK_EQ(MOJO_RESULT_OK, result); | 111 CHECK_EQ(MOJO_RESULT_OK, result); |
| 112 } | 112 } |
| 113 return request.Pass(); | 113 return request.Pass(); |
| 114 } | 114 } |
| 115 }; | 115 }; |
| 116 | 116 |
| 117 HttpConnectionImpl::HttpConnectionImpl(int connection_id, | 117 HttpConnectionImpl::HttpConnectionImpl(int connection_id, |
| (...skipping 12 matching lines...) Expand all Loading... |
| 130 HttpConnectionImpl::~HttpConnectionImpl() { | 130 HttpConnectionImpl::~HttpConnectionImpl() { |
| 131 STLDeleteElements(&response_body_readers_); | 131 STLDeleteElements(&response_body_readers_); |
| 132 } | 132 } |
| 133 | 133 |
| 134 void HttpConnectionImpl::OnReceivedHttpRequest( | 134 void HttpConnectionImpl::OnReceivedHttpRequest( |
| 135 const net::HttpServerRequestInfo& info) { | 135 const net::HttpServerRequestInfo& info) { |
| 136 if (EncounteredConnectionError()) | 136 if (EncounteredConnectionError()) |
| 137 return; | 137 return; |
| 138 | 138 |
| 139 delegate_->OnReceivedRequest( | 139 delegate_->OnReceivedRequest( |
| 140 URLRequest::From(info), [this](URLResponsePtr response) { | 140 HttpRequest::From(info), [this](HttpResponsePtr response) { |
| 141 if (response->body.is_valid()) { | 141 if (response->body.is_valid()) { |
| 142 SimpleDataPipeReader* reader = new SimpleDataPipeReader; | 142 SimpleDataPipeReader* reader = new SimpleDataPipeReader; |
| 143 response_body_readers_.insert(reader); | 143 response_body_readers_.insert(reader); |
| 144 reader->Start( | 144 reader->Start( |
| 145 response->body.Pass(), | 145 response->body.Pass(), |
| 146 base::Bind(&HttpConnectionImpl::OnFinishedReadingResponseBody, | 146 base::Bind(&HttpConnectionImpl::OnFinishedReadingResponseBody, |
| 147 base::Unretained(this), base::Passed(&response))); | 147 base::Unretained(this), base::Passed(&response))); |
| 148 } else { | 148 } else { |
| 149 OnFinishedReadingResponseBody(response.Pass(), nullptr, nullptr); | 149 OnFinishedReadingResponseBody(response.Pass(), nullptr, nullptr); |
| 150 } | 150 } |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 191 | 191 |
| 192 binding_.Close(); | 192 binding_.Close(); |
| 193 delegate_.reset(); | 193 delegate_.reset(); |
| 194 | 194 |
| 195 // Don't close the connection until all pending responses are sent. | 195 // Don't close the connection until all pending responses are sent. |
| 196 if (response_body_readers_.empty()) | 196 if (response_body_readers_.empty()) |
| 197 owner_->server()->Close(connection_id_); | 197 owner_->server()->Close(connection_id_); |
| 198 } | 198 } |
| 199 | 199 |
| 200 void HttpConnectionImpl::OnFinishedReadingResponseBody( | 200 void HttpConnectionImpl::OnFinishedReadingResponseBody( |
| 201 URLResponsePtr response, | 201 HttpResponsePtr response, |
| 202 SimpleDataPipeReader* reader, | 202 SimpleDataPipeReader* reader, |
| 203 scoped_ptr<std::string> body) { | 203 scoped_ptr<std::string> body) { |
| 204 if (reader) { | 204 if (reader) { |
| 205 delete reader; | 205 delete reader; |
| 206 response_body_readers_.erase(reader); | 206 response_body_readers_.erase(reader); |
| 207 } | 207 } |
| 208 | 208 |
| 209 net::HttpServerResponseInfo info( | 209 net::HttpServerResponseInfo info( |
| 210 static_cast<net::HttpStatusCode>(response->status_code)); | 210 static_cast<net::HttpStatusCode>(response->status_code)); |
| 211 | 211 |
| 212 std::string content_type; | 212 std::string content_type; |
| 213 for (size_t i = 0; i < response->headers.size(); ++i) { | 213 for (size_t i = 0; i < response->headers.size(); ++i) { |
| 214 const HTTPHeader& header = *(response->headers[i]); | 214 const HttpHeader& header = *(response->headers[i]); |
| 215 | 215 |
| 216 if (body) { | 216 if (body) { |
| 217 // net::HttpServerResponseInfo::SetBody() automatically sets | 217 // net::HttpServerResponseInfo::SetBody() automatically sets |
| 218 // Content-Length and Content-Types, so skip the two here. | 218 // Content-Length and Content-Types, so skip the two here. |
| 219 // | 219 // |
| 220 // TODO(yzshen): Consider adding to net::HttpServerResponseInfo a simple | 220 // TODO(yzshen): Consider adding to net::HttpServerResponseInfo a simple |
| 221 // setter for body which doesn't fiddle with headers. | 221 // setter for body which doesn't fiddle with headers. |
| 222 if (base::strcasecmp(header.name.data(), | 222 if (base::strcasecmp(header.name.data(), |
| 223 net::HttpRequestHeaders::kContentLength) == 0) { | 223 net::HttpRequestHeaders::kContentLength) == 0) { |
| 224 continue; | 224 continue; |
| 225 } else if (base::strcasecmp(header.name.data(), | 225 } else if (base::strcasecmp(header.name.data(), |
| 226 net::HttpRequestHeaders::kContentType) == 0) { | 226 net::HttpRequestHeaders::kContentType) == 0) { |
| 227 content_type = header.value; | 227 content_type = header.value; |
| 228 continue; | 228 continue; |
| 229 } | 229 } |
| 230 } | 230 } |
| 231 info.AddHeader(header.name, header.value); | 231 info.AddHeader(header.name, header.value); |
| 232 } | 232 } |
| 233 | 233 |
| 234 if (body) | 234 if (body) |
| 235 info.SetBody(*body, content_type); | 235 info.SetBody(*body, content_type); |
| 236 | 236 |
| 237 owner_->server()->SendResponse(connection_id_, info); | 237 owner_->server()->SendResponse(connection_id_, info); |
| 238 | 238 |
| 239 if (response_body_readers_.empty() && EncounteredConnectionError()) | 239 if (response_body_readers_.empty() && EncounteredConnectionError()) |
| 240 owner_->server()->Close(connection_id_); | 240 owner_->server()->Close(connection_id_); |
| 241 } | 241 } |
| 242 | 242 |
| 243 } // namespace mojo | 243 } // namespace mojo |
| OLD | NEW |