Index: util/net/http_transport_test.cc |
diff --git a/util/net/http_transport_test.cc b/util/net/http_transport_test.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..974e9f05411a1152b7c69e9593e94c2043220cd7 |
--- /dev/null |
+++ b/util/net/http_transport_test.cc |
@@ -0,0 +1,242 @@ |
+// Copyright 2014 The Crashpad Authors. All rights reserved. |
+// |
+// Licensed under the Apache License, Version 2.0 (the "License"); |
+// you may not use this file except in compliance with the License. |
+// You may obtain a copy of the License at |
+// |
+// http://www.apache.org/licenses/LICENSE-2.0 |
+// |
+// Unless required by applicable law or agreed to in writing, software |
+// distributed under the License is distributed on an "AS IS" BASIS, |
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
+// See the License for the specific language governing permissions and |
+// limitations under the License. |
+ |
+#include "util/net/http_transport.h" |
+ |
+#include <stdint.h> |
+#include <string.h> |
+ |
+#include <vector> |
+ |
+#include "base/logging.h" |
+#include "base/memory/scoped_ptr.h" |
+#include "base/strings/stringprintf.h" |
+#include "build/build_config.h" |
+#include "gtest/gtest.h" |
+#include "util/file/fd_io.h" |
+#include "util/net/http_body.h" |
+#include "util/net/http_headers.h" |
+#include "util/net/http_multipart_builder.h" |
+#include "util/test/multiprocess_exec.h" |
+ |
+namespace crashpad { |
+namespace test { |
+namespace { |
+ |
+class HTTPTransportTestFixture : public MultiprocessExec { |
+ public: |
+ using RequestValidator = |
+ void(*)(HTTPTransportTestFixture*, const std::string&); |
+ |
+ HTTPTransportTestFixture(const HTTPHeaders& headers, |
+ scoped_ptr<HTTPBodyStream> body_stream, |
+ uint16_t http_response_code, |
+ RequestValidator request_validator) |
+ : MultiprocessExec(), |
+ headers_(headers), |
+ body_stream_(body_stream.Pass()), |
+ response_code_(http_response_code), |
+ request_validator_(request_validator) { |
+ // TODO(rsesek): Use a more robust mechanism to locate testdata |
+ // <https://code.google.com/p/crashpad/issues/detail?id=4>. |
+ SetChildCommand("util/net/http_transport_test_server.py", nullptr); |
+ } |
+ |
+ const HTTPHeaders& headers() { return headers_; } |
+ |
+ private: |
+ void MultiprocessParent() override { |
+ // The child will write the HTTP server port number as a packed unsigned |
+ // short to stdout. |
+ uint16_t port; |
+ CheckedReadFD(ReadPipeFD(), &port, sizeof(port)); |
+ |
+ // Then the parent will tell the web server what response code to send |
+ // for the HTTP request. |
+ CheckedWriteFD(WritePipeFD(), &response_code_, sizeof(response_code_)); |
+ |
+ // Now execute the HTTP request. |
+ scoped_ptr<HTTPTransport> transport(HTTPTransport::Create()); |
+ transport->SetMethod("POST"); |
+ transport->SetURL(base::StringPrintf("http://127.0.0.1:%d/upload", port)); |
+ for (const auto& pair : headers_) { |
+ transport->SetHeader(pair.first, pair.second); |
+ } |
+ transport->SetBodyStream(body_stream_.Pass()); |
+ |
+ EXPECT_EQ(transport->ExecuteSynchronously(), (response_code_ == 200)); |
+ |
+ // Read until the child's stdout closes. |
+ std::string request; |
+ char buf[32]; |
+ ssize_t bytes_read; |
+ while ((bytes_read = ReadFD(ReadPipeFD(), buf, sizeof(buf))) != 0) { |
+ ASSERT_GE(bytes_read, 0); |
+ request.append(buf, bytes_read); |
+ } |
+ |
+ if (request_validator_) |
+ request_validator_(this, request); |
+ } |
+ |
+ HTTPHeaders headers_; |
+ scoped_ptr<HTTPBodyStream> body_stream_; |
+ uint16_t response_code_; |
+ RequestValidator request_validator_; |
+}; |
+ |
+const char kMultipartFormData[] = "multipart/form-data"; |
+ |
+void GetHeaderField(const std::string& request, |
+ const std::string& header, |
+ std::string* value) { |
+ size_t index = request.find(header); |
+ ASSERT_NE(std::string::npos, index); |
+ // Since the header is never the first line of the request, it should always |
+ // be preceded by a CRLF. |
+ EXPECT_EQ('\n', request[index - 1]); |
+ EXPECT_EQ('\r', request[index - 2]); |
+ |
+ index += header.length(); |
+ EXPECT_EQ(':', request[index++]); |
+ // Per RFC 7230 ยง3.2, there can be one or more spaces or horizontal tabs. |
+ // For testing purposes, just assume one space. |
+ EXPECT_EQ(' ', request[index++]); |
+ |
+ size_t header_end = request.find('\r', index); |
+ ASSERT_NE(std::string::npos, header_end); |
+ |
+ *value = request.substr(index, header_end - index); |
+} |
+ |
+void GetMultipartBoundary(const std::string& request, |
+ std::string* multipart_boundary) { |
+ std::string content_type; |
+ GetHeaderField(request, kContentType, &content_type); |
+ |
+ ASSERT_GE(content_type.length(), strlen(kMultipartFormData)); |
+ size_t index = strlen(kMultipartFormData); |
+ EXPECT_EQ(kMultipartFormData, content_type.substr(0, index)); |
+ |
+ EXPECT_EQ(';', content_type[index++]); |
+ |
+ size_t boundary_begin = content_type.find('=', index); |
+ ASSERT_NE(std::string::npos, boundary_begin); |
+ EXPECT_EQ('=', content_type[boundary_begin++]); |
+ if (multipart_boundary) { |
+ *multipart_boundary = content_type.substr(boundary_begin); |
+ } |
+} |
+ |
+const char kBoundaryEq[] = "boundary="; |
+ |
+void ValidFormData(HTTPTransportTestFixture* fixture, |
+ const std::string& request) { |
+ std::string actual_boundary; |
+ GetMultipartBoundary(request, &actual_boundary); |
+ |
+ const auto& content_type = fixture->headers().find(kContentType); |
+ ASSERT_NE(fixture->headers().end(), content_type); |
+ |
+ size_t boundary = content_type->second.find(kBoundaryEq); |
+ ASSERT_NE(std::string::npos, boundary); |
+ std::string expected_boundary = |
+ content_type->second.substr(boundary + strlen(kBoundaryEq)); |
+ EXPECT_EQ(expected_boundary, actual_boundary); |
+ |
+ size_t body_start = request.find("\r\n\r\n"); |
+ ASSERT_NE(std::string::npos, body_start); |
+ body_start += 4; |
+ |
+ std::string expected = "--" + expected_boundary + "\r\n"; |
+ expected += "Content-Disposition: form-data; name=\"key1\"\r\n\r\n"; |
+ expected += "test\r\n"; |
+ ASSERT_LT(body_start + expected.length(), request.length()); |
+ EXPECT_EQ(expected, request.substr(body_start, expected.length())); |
+ |
+ body_start += expected.length(); |
+ |
+ expected = "--" + expected_boundary + "\r\n"; |
+ expected += "Content-Disposition: form-data; name=\"key2\"\r\n\r\n"; |
+ expected += "--abcdefg123\r\n"; |
+ expected += "--" + expected_boundary + "--\r\n"; |
+ ASSERT_EQ(body_start + expected.length(), request.length()); |
+ EXPECT_EQ(expected, request.substr(body_start)); |
+} |
+ |
+TEST(HTTPTransport, ValidFormData) { |
+ HTTPMultipartBuilder builder; |
+ builder.SetFormData("key1", "test"); |
+ builder.SetFormData("key2", "--abcdefg123"); |
+ |
+ HTTPHeaders headers; |
+ headers.insert(builder.GetContentType()); |
+ |
+ HTTPTransportTestFixture test(headers, builder.GetBodyStream(), 200, |
+ &ValidFormData); |
+ test.Run(); |
+} |
+ |
+const char kTextPlain[] = "text/plain"; |
+ |
+void ErrorResponse(HTTPTransportTestFixture* fixture, |
+ const std::string& request) { |
+ std::string content_type; |
+ GetHeaderField(request, kContentType, &content_type); |
+ EXPECT_EQ(kTextPlain, content_type); |
+} |
+ |
+TEST(HTTPTransport, ErrorResponse) { |
+ HTTPMultipartBuilder builder; |
+ HTTPHeaders headers; |
+ headers[kContentType] = kTextPlain; |
+ HTTPTransportTestFixture test(headers, builder.GetBodyStream(), |
+ 404, &ErrorResponse); |
+ test.Run(); |
+} |
+ |
+const char kTextBody[] = "hello world"; |
+ |
+void UnchunkedPlainText(HTTPTransportTestFixture* fixture, |
+ const std::string& request) { |
+ std::string header_value; |
+ GetHeaderField(request, kContentType, &header_value); |
+ EXPECT_EQ(kTextPlain, header_value); |
+ |
+ GetHeaderField(request, kContentLength, &header_value); |
+ const auto& content_length = fixture->headers().find(kContentLength); |
+ ASSERT_NE(fixture->headers().end(), content_length); |
+ EXPECT_EQ(content_length->second, header_value); |
+ |
+ size_t body_start = request.rfind("\r\n"); |
+ ASSERT_NE(std::string::npos, body_start); |
+ |
+ EXPECT_EQ(kTextBody, request.substr(body_start + 2)); |
+} |
+ |
+TEST(HTTPTransport, UnchunkedPlainText) { |
+ scoped_ptr<HTTPBodyStream> body_stream(new StringHTTPBodyStream(kTextBody)); |
+ |
+ HTTPHeaders headers; |
+ headers[kContentType] = kTextPlain; |
+ headers[kContentLength] = base::StringPrintf("%zu", strlen(kTextBody)); |
+ |
+ HTTPTransportTestFixture test(headers, body_stream.Pass(), 200, |
+ &UnchunkedPlainText); |
+ test.Run(); |
+} |
+ |
+} // namespace |
+} // namespace test |
+} // namespace crashpad |