SSL in EmbeddedTestServer

net/test/embedded_test_server/http_response.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/test/embedded_test_server/http_response.h"
 
 #include "base/format_macros.h"
 #include "base/logging.h"
 #include "base/strings/stringprintf.h"
 #include "net/http/http_status_code.h"
 
 namespace net {
 namespace test_server {
 
 HttpResponse::~HttpResponse() {
 }
 
+RawHttpResponse::RawHttpResponse(const std::string& headers,
+                                 const std::string& contents)
+    : headers_(headers), contents_(contents) {}
+
+RawHttpResponse::~RawHttpResponse() {}
+
+void RawHttpResponse::SendResponse(const SendBytesCallback& send,
+                                   const SendCompleteCallback& done) {
+  std::string response;
+  if (!headers_.empty())
+    response = headers_ + "\r\n" + contents_;
+  else
+    response = contents_;
+  send.Run(response, done);
+}
+
+void RawHttpResponse::AddHeader(const std::string& key_value_pair) {
+  headers_.append(base::StringPrintf("%s\r\n", key_value_pair.c_str()));
+}
+
 BasicHttpResponse::BasicHttpResponse() : code_(HTTP_OK) {
 }
 
 BasicHttpResponse::~BasicHttpResponse() {
 }
 
 std::string BasicHttpResponse::ToResponseString() const {
   // Response line with headers.
   std::string response_builder;
 
   std::string http_reason_phrase(GetHttpReasonPhrase(code_));
 
   // TODO(mtomasz): For http/1.0 requests, send http/1.0.
   base::StringAppendF(&response_builder,
                       "HTTP/1.1 %d %s\r\n",
                       code_,
                       http_reason_phrase.c_str());
   base::StringAppendF(&response_builder, "Connection: close\r\n");
-  base::StringAppendF(&response_builder,
-                      "Content-Length: %" PRIuS "\r\n",
+
+  base::StringAppendF(&response_builder, "Content-Length: %" PRIuS "\r\n",
                       content_.size());
-  base::StringAppendF(&response_builder,
-                      "Content-Type: %s\r\n",
+  base::StringAppendF(&response_builder, "Content-Type: %s\r\n",
                       content_type_.c_str());
   for (size_t i = 0; i < custom_headers_.size(); ++i) {
     const std::string& header_name = custom_headers_[i].first;
     const std::string& header_value = custom_headers_[i].second;
     DCHECK(header_value.find_first_of("\n\r") == std::string::npos) <<
         "Malformed header value.";
     base::StringAppendF(&response_builder,
                         "%s: %s\r\n",
                         header_name.c_str(),
                         header_value.c_str());
   }
   base::StringAppendF(&response_builder, "\r\n");
 
   return response_builder + content_;
 }
 
+void BasicHttpResponse::SendResponse(const SendBytesCallback& send,
+                                     const SendCompleteCallback& done) {
+  send.Run(ToResponseString(), done);
+}
+
 }  // namespace test_server
 }  // namespace net