OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "content/browser/download/mhtml_generation_manager.h" | 5 #include "content/browser/download/mhtml_generation_manager.h" |
6 | 6 |
| 7 #include "base/bind.h" |
7 #include "base/platform_file.h" | 8 #include "base/platform_file.h" |
8 #include "content/browser/renderer_host/render_process_host.h" | 9 #include "content/browser/renderer_host/render_process_host.h" |
9 #include "content/browser/renderer_host/render_view_host.h" | 10 #include "content/browser/renderer_host/render_view_host.h" |
10 #include "content/browser/tab_contents/tab_contents.h" | 11 #include "content/browser/tab_contents/tab_contents.h" |
11 #include "content/public/browser/notification_service.h" | 12 #include "content/public/browser/notification_service.h" |
12 #include "content/common/view_messages.h" | 13 #include "content/common/view_messages.h" |
13 #include "content/public/browser/notification_types.h" | 14 #include "content/public/browser/notification_types.h" |
14 | 15 |
15 using content::BrowserThread; | 16 using content::BrowserThread; |
16 | 17 |
(...skipping 18 matching lines...) Expand all Loading... |
35 int job_id = id_counter++; | 36 int job_id = id_counter++; |
36 Job job; | 37 Job job; |
37 job.file_path = file; | 38 job.file_path = file; |
38 job.process_id = tab_contents->GetRenderProcessHost()->id(); | 39 job.process_id = tab_contents->GetRenderProcessHost()->id(); |
39 job.routing_id = tab_contents->render_view_host()->routing_id(); | 40 job.routing_id = tab_contents->render_view_host()->routing_id(); |
40 id_to_job_[job_id] = job; | 41 id_to_job_[job_id] = job; |
41 | 42 |
42 base::ProcessHandle renderer_process = | 43 base::ProcessHandle renderer_process = |
43 tab_contents->GetRenderProcessHost()->GetHandle(); | 44 tab_contents->GetRenderProcessHost()->GetHandle(); |
44 BrowserThread::PostTask(BrowserThread::FILE, FROM_HERE, | 45 BrowserThread::PostTask(BrowserThread::FILE, FROM_HERE, |
45 NewRunnableMethod(this, &MHTMLGenerationManager::CreateFile, | 46 base::Bind(&MHTMLGenerationManager::CreateFile, this, |
46 job_id, file, renderer_process)); | 47 job_id, file, renderer_process)); |
47 } | 48 } |
48 | 49 |
49 void MHTMLGenerationManager::MHTMLGenerated(int job_id, int64 mhtml_data_size) { | 50 void MHTMLGenerationManager::MHTMLGenerated(int job_id, int64 mhtml_data_size) { |
50 JobFinished(job_id, mhtml_data_size); | 51 JobFinished(job_id, mhtml_data_size); |
51 } | 52 } |
52 | 53 |
53 void MHTMLGenerationManager::CreateFile(int job_id, const FilePath& file_path, | 54 void MHTMLGenerationManager::CreateFile(int job_id, const FilePath& file_path, |
54 base::ProcessHandle renderer_process) { | 55 base::ProcessHandle renderer_process) { |
55 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | 56 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
56 base::PlatformFile browser_file = base::CreatePlatformFile(file_path, | 57 base::PlatformFile browser_file = base::CreatePlatformFile(file_path, |
57 base::PLATFORM_FILE_CREATE_ALWAYS | base::PLATFORM_FILE_WRITE, | 58 base::PLATFORM_FILE_CREATE_ALWAYS | base::PLATFORM_FILE_WRITE, |
58 NULL, NULL); | 59 NULL, NULL); |
59 if (browser_file == base::kInvalidPlatformFileValue) { | 60 if (browser_file == base::kInvalidPlatformFileValue) { |
60 LOG(ERROR) << "Failed to create file to save MHTML at: " << | 61 LOG(ERROR) << "Failed to create file to save MHTML at: " << |
61 file_path.value(); | 62 file_path.value(); |
62 } | 63 } |
63 | 64 |
64 IPC::PlatformFileForTransit renderer_file = | 65 IPC::PlatformFileForTransit renderer_file = |
65 IPC::GetFileHandleForProcess(browser_file, renderer_process, false); | 66 IPC::GetFileHandleForProcess(browser_file, renderer_process, false); |
66 | 67 |
67 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, | 68 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, |
68 NewRunnableMethod(this, &MHTMLGenerationManager::FileCreated, | 69 base::Bind(&MHTMLGenerationManager::FileCreated, this, |
69 job_id, browser_file, renderer_file)); | 70 job_id, browser_file, renderer_file)); |
70 } | 71 } |
71 | 72 |
72 void MHTMLGenerationManager::FileCreated(int job_id, | 73 void MHTMLGenerationManager::FileCreated(int job_id, |
73 base::PlatformFile browser_file, | 74 base::PlatformFile browser_file, |
74 IPC::PlatformFileForTransit renderer_file) { | 75 IPC::PlatformFileForTransit renderer_file) { |
75 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 76 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
76 if (browser_file == base::kInvalidPlatformFileValue) { | 77 if (browser_file == base::kInvalidPlatformFileValue) { |
77 LOG(ERROR) << "Failed to create file"; | 78 LOG(ERROR) << "Failed to create file"; |
78 JobFinished(job_id, -1); | 79 JobFinished(job_id, -1); |
79 return; | 80 return; |
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
115 details.file_path = job.file_path; | 116 details.file_path = job.file_path; |
116 details.file_size = file_size; | 117 details.file_size = file_size; |
117 | 118 |
118 content::NotificationService::current()->Notify( | 119 content::NotificationService::current()->Notify( |
119 content::NOTIFICATION_MHTML_GENERATED, | 120 content::NOTIFICATION_MHTML_GENERATED, |
120 content::Source<RenderViewHost>(rvh), | 121 content::Source<RenderViewHost>(rvh), |
121 content::Details<NotificationDetails>(&details)); | 122 content::Details<NotificationDetails>(&details)); |
122 } | 123 } |
123 | 124 |
124 BrowserThread::PostTask(BrowserThread::FILE, FROM_HERE, | 125 BrowserThread::PostTask(BrowserThread::FILE, FROM_HERE, |
125 NewRunnableMethod(this, &MHTMLGenerationManager::CloseFile, | 126 base::Bind(&MHTMLGenerationManager::CloseFile, this, job.browser_file)); |
126 job.browser_file)); | |
127 | 127 |
128 id_to_job_.erase(job_id); | 128 id_to_job_.erase(job_id); |
129 } | 129 } |
130 | 130 |
131 void MHTMLGenerationManager::CloseFile(base::PlatformFile file) { | 131 void MHTMLGenerationManager::CloseFile(base::PlatformFile file) { |
132 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | 132 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
133 base::ClosePlatformFile(file); | 133 base::ClosePlatformFile(file); |
134 } | 134 } |
OLD | NEW |