Adding a command-line flag to set MHTML generation options.

content/public/common/mhtml_generation_params.cc

 // Copyright 2016 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/public/common/mhtml_generation_params.h"
 
+#include "base/command_line.h"
 #include "base/files/file_path.h"
+#include "content/public/common/content_switches.h"
 
 namespace content {
 
 MHTMLGenerationParams::MHTMLGenerationParams(const base::FilePath& file_path)
-    : file_path(file_path) {}
+    : file_path(file_path) {
+  // Check which variant of MHTML generation is required.
+  std::string mhtmlGeneratorOptionFlag =
+      base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
+      switches::kMHTMLGeneratorOption);
+  if (mhtmlGeneratorOptionFlag == switches::kMHTMLSkipNostoreMain) {
+    cache_control_policy =
+        blink::WebFrameSerializerCacheControlPolicy::
+        FailForNoStoreMainFrame;
+  } else if (mhtmlGeneratorOptionFlag == switches::kMHTMLSkipNostoreAll) {
+    cache_control_policy =
+        blink::WebFrameSerializerCacheControlPolicy::
+        SkipAnyFrameOrResourceMarkedNoStore;
+  }
+}
 
 }  // namespace content