Get signals working in the EXTRA_DATA section of MHTML

content/browser/download/mhtml_generation_manager.h

 // 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.
 #ifndef CONTENT_BROWSER_DOWNLOAD_MHTML_GENERATION_MANAGER_H_
 #define CONTENT_BROWSER_DOWNLOAD_MHTML_GENERATION_MANAGER_H_
 
 #include <stdint.h>
 
 #include <map>
 #include <memory>
 #include <set>
 #include <string>
 
 #include "base/files/file.h"
 #include "base/macros.h"
 #include "base/memory/singleton.h"
 #include "base/process/process.h"
+#include "base/supports_user_data.h"
 #include "content/common/download/mhtml_save_status.h"
+#include "content/public/browser/web_contents.h"
 #include "content/public/common/mhtml_generation_params.h"
 #include "ipc/ipc_platform_file.h"
 
 namespace base {
 class FilePath;
 }
 
 namespace content {
 
 class RenderFrameHostImpl;
 class WebContents;
 
 // The class and all of its members live on the UI thread.  Only static methods
 // are executed on other threads.
 //
 // MHTMLGenerationManager is a singleton.  Each call to SaveMHTML method creates
 // a new instance of MHTMLGenerationManager::Job that tracks generation of a
 // single MHTML file.
-class MHTMLGenerationManager {
+class CONTENT_EXPORT MHTMLGenerationManager {
  public:
   static MHTMLGenerationManager* GetInstance();
 
   // GenerateMHTMLCallback is called to report completion and status of MHTML
   // generation.  On success |file_size| indicates the size of the
   // generated file.  On failure |file_size| is -1.
   typedef base::Callback<void(int64_t file_size)> GenerateMHTMLCallback;
 
   // Instructs the RenderFrames in |web_contents| to generate a MHTML
   // representation of the current page.
   void SaveMHTML(WebContents* web_contents,
                  const MHTMLGenerationParams& params,
                  const GenerateMHTMLCallback& callback);
 
   // Handler for FrameHostMsg_SerializeAsMHTMLResponse (a notification from the
   // renderer that the MHTML generation finished for a single frame).
   void OnSerializeAsMHTMLResponse(
       RenderFrameHostImpl* sender,
       int job_id,
       MhtmlSaveStatus save_status,
       const std::set<std::string>& digests_of_uris_of_serialized_resources,
       base::TimeDelta renderer_main_thread_time);
 
+  // Contained class that represents an extra data part.  With the addition of
+  // the MHTMLExtraData class, we can add enough information to create an extra
+  // MHTML part to be added into the complete file.  For instance, this might
+  // be used for gathering load time signals in debug code for analysis, but
+  // the mechanism is fairly general, and could be used to add arbitrary parts.
+  // Specify the content type, content location, and body, and the methods in
+  // MHTMLExtraData will do the rest.
+  struct MHTMLExtraPart {
+    std::string content_type;
+    std::string content_location;
+    std::string body;
+  };
+
+  // Helper class to put extra parts into the WebContents user data, and take
+  // them back out again.
+  class MHTMLExtraData : public base::SupportsUserData::Data {
+   public:
+    MHTMLExtraData();
+    ~MHTMLExtraData() override;
+
+    // Get the data string out of the web contents. The web contents retains
+    // ownership of the vector.  Returns nullptr if none found.
+    static std::vector<MHTMLExtraPart>* FromWebContents(
+        content::WebContents* contents);
+
+    // Add the extra part to the data structure stored by the WebContents.
+    static void AddPartToWebContents(content::WebContents* contents,
+                                     MHTMLExtraPart& part);
+
+   private:
+    std::vector<MHTMLExtraPart> parts_;
+  };
+
  private:
   friend struct base::DefaultSingletonTraits<MHTMLGenerationManager>;
   class Job;
 
   MHTMLGenerationManager();
   virtual ~MHTMLGenerationManager();
 
   // Called on the file thread to create |file|.
   static base::File CreateFile(const base::FilePath& file_path);
 
@@ skipping 23 matching lines @@
   std::map<int, std::unique_ptr<Job>> id_to_job_;
 
   int next_job_id_;
 
   DISALLOW_COPY_AND_ASSIGN(MHTMLGenerationManager);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_DOWNLOAD_MHTML_GENERATION_MANAGER_H_