Get signals working in the EXTRA_DATA section of MHTML

content/browser/download/mhtml_extra_data_impl.cc

Index: content/browser/download/mhtml_extra_data_impl.cc
diff --git a/content/browser/download/mhtml_extra_data_impl.cc b/content/browser/download/mhtml_extra_data_impl.cc
new file mode 100644
index 0000000000000000000000000000000000000000..8aef7b69cc561e2b90e847bb6f39990f38a86fd2
--- /dev/null
+++ b/content/browser/download/mhtml_extra_data_impl.cc
@@ -0,0 +1,45 @@
+// Copyright 2017 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 "content/browser/download/mhtml_extra_data_impl.h"
+
+namespace {
+const int kMHTMLExtraDataKey = 0;

[Dmitry Titov, 2017/03/29 18:53:06]

no need to initialize this value. But adding a short comment that only an
address of this variable is used would be nice.

[Pete Williamson, 2017/03/29 22:05:42]

Comment added.  The compiler really wanted me to initialize it though, so I left
the '= 0' in. 

+}
+
+namespace content {
+
+MHTMLExtraDataImpl::MHTMLExtraDataImpl() {}
+MHTMLExtraDataImpl::~MHTMLExtraDataImpl() {}
+
+MHTMLExtraData* MHTMLExtraData::FromWebContents(WebContents* contents) {
+  // Get the MHTMLExtraDataImpl from the web contents.
+  MHTMLExtraDataImpl* extra_data_impl = static_cast<MHTMLExtraDataImpl*>(
+      contents->GetUserData(&kMHTMLExtraDataKey));
+
+  // If we did not have one on the web contents already, make one and put it on
+  // the web contents.
+  if (extra_data_impl == nullptr) {
+    extra_data_impl = new MHTMLExtraDataImpl();
+    contents->SetUserData(&kMHTMLExtraDataKey,
+                          std::unique_ptr<MHTMLExtraData>(
+                              static_cast<MHTMLExtraData*>(extra_data_impl)));
+  }
+  return static_cast<MHTMLExtraData*>(extra_data_impl);
+}
+
+void MHTMLExtraDataImpl::AddExtraMHTMLPart(const std::string& content_type,
+                                           const std::string& content_location,
+                                           const std::string& body) {
+  MHTMLExtraDataPart part;
+  part.content_type = content_type;
+  part.content_location = content_location;
+  part.body = body;
+
+  // Add this part to the list of parts to be saved out when the file is
+  // written.
+  parts_.push_back(part);
+}
+
+}  // namespace content