mhtml_generator.cc revision f2477e01787aa58f445919b809d89e252beef54f 
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
13namespace content {
14
15MHTMLGenerator::MHTMLGenerator(RenderViewImpl* render_view)
16    : RenderViewObserver(render_view),
17      file_(base::kInvalidPlatformFileValue) {
18}
19
20MHTMLGenerator::~MHTMLGenerator() {
21}
22
23// RenderViewObserver implementation:
24bool 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
33void 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
43void 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.
50int64 MHTMLGenerator::GenerateMHTML() {
51  blink::WebCString mhtml =
52      blink::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
70