src/content/renderer/mhtml_generator.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/renderer/mhtml_generator.h"
   6
   7 #include "base/platform_file.h"
   8 #include "content/common/view_messages.h"
   9 #include "content/renderer/render_view_impl.h"
  10 #include "third_party/WebKit/public/platform/WebCString.h"
  11 #include "third_party/WebKit/public/web/WebPageSerializer.h"
  12
  13 namespace content {
  14
  15 MHTMLGenerator::MHTMLGenerator(RenderViewImpl* render_view)
  16     : RenderViewObserver(render_view),
  17       file_(base::kInvalidPlatformFileValue) {
  18 }
  19
  20 MHTMLGenerator::~MHTMLGenerator() {
  21 }
  22
  23 // RenderViewObserver implementation:
  24 bool MHTMLGenerator::OnMessageReceived(const IPC::Message& message) {
  25   bool handled = true;
  26   IPC_BEGIN_MESSAGE_MAP(MHTMLGenerator, message)
  27       IPC_MESSAGE_HANDLER(ViewMsg_SavePageAsMHTML, OnSavePageAsMHTML)
  28       IPC_MESSAGE_UNHANDLED(handled = false)
  29   IPC_END_MESSAGE_MAP()
  30   return handled;
  31 }
  32
  33 void MHTMLGenerator::OnSavePageAsMHTML(
  34     int job_id, IPC::PlatformFileForTransit file_for_transit) {
  35   base::PlatformFile file =
  36       IPC::PlatformFileForTransitToPlatformFile(file_for_transit);
  37   file_ = file;
  38   int64 size = GenerateMHTML();
  39   base::ClosePlatformFile(file);
  40   NotifyBrowser(job_id, size);
  41 }
  42
  43 void MHTMLGenerator::NotifyBrowser(int job_id, int64 data_size) {
  44   render_view()->Send(new ViewHostMsg_SavedPageAsMHTML(job_id, data_size));
  45   file_ = base::kInvalidPlatformFileValue;
  46 }
  47
  48 // TODO(jcivelli): write the chunks in deferred tasks to give a chance to the
  49 //                 message loop to process other events.
  50 int64 MHTMLGenerator::GenerateMHTML() {
  51   WebKit::WebCString mhtml =
  52       WebKit::WebPageSerializer::serializeToMHTML(render_view()->GetWebView());
  53   const size_t chunk_size = 1024;
  54   const char* data = mhtml.data();
  55   size_t total_bytes_written = 0;
  56   while (total_bytes_written < mhtml.length()) {
  57     size_t copy_size =
  58         std::min(mhtml.length() - total_bytes_written, chunk_size);
  59     int bytes_written = base::WritePlatformFile(file_, total_bytes_written,
  60                                                 data + total_bytes_written,
  61                                                 copy_size);
  62     if (bytes_written == -1)
  63       return -1;
  64     total_bytes_written += bytes_written;
  65   }
  66   return total_bytes_written;
  67 }
  68
  69 }  // namespace content