OLD | NEW |
1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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 "chrome/browser/renderer_host/database_dispatcher_host.h" | 5 #include "chrome/browser/renderer_host/database_dispatcher_host.h" |
6 | 6 |
7 #if defined(OS_WIN) | 7 #if defined(OS_WIN) |
8 #include <windows.h> | 8 #include <windows.h> |
9 #endif | 9 #endif |
10 | 10 |
11 #if defined(USE_SYSTEM_SQLITE) | 11 #if defined(USE_SYSTEM_SQLITE) |
12 #include <sqlite3.h> | 12 #include <sqlite3.h> |
13 #else | 13 #else |
14 #include "third_party/sqlite/preprocessed/sqlite3.h" | 14 #include "third_party/sqlite/preprocessed/sqlite3.h" |
15 #endif | 15 #endif |
16 | 16 |
17 #include "base/thread.h" | 17 #include "chrome/browser/chrome_thread.h" |
18 #include "chrome/browser/browser_process.h" | |
19 #include "chrome/browser/renderer_host/resource_message_filter.h" | 18 #include "chrome/browser/renderer_host/resource_message_filter.h" |
20 #include "chrome/common/render_messages.h" | 19 #include "chrome/common/render_messages.h" |
21 #include "webkit/database/vfs_backend.h" | 20 #include "webkit/database/vfs_backend.h" |
22 | 21 |
23 #if defined(OS_POSIX) | 22 #if defined(OS_POSIX) |
24 #include "base/file_descriptor_posix.h" | 23 #include "base/file_descriptor_posix.h" |
25 #endif | 24 #endif |
26 | 25 |
27 using webkit_database::VfsBackend; | 26 using webkit_database::VfsBackend; |
28 | 27 |
(...skipping 25 matching lines...) Expand all Loading... |
54 // message filterto make sure it doesn't get destroyed before we have | 53 // message filterto make sure it doesn't get destroyed before we have |
55 // a chance to send the reply back. So we need to Release() is here | 54 // a chance to send the reply back. So we need to Release() is here |
56 // and allow it to be destroyed if needed. | 55 // and allow it to be destroyed if needed. |
57 sender->Release(); | 56 sender->Release(); |
58 } | 57 } |
59 | 58 |
60 // Scheduled by the IO thread on the file thread. | 59 // Scheduled by the IO thread on the file thread. |
61 // Opens the given database file, then schedules | 60 // Opens the given database file, then schedules |
62 // a task on the IO thread's message loop to send an IPC back to | 61 // a task on the IO thread's message loop to send an IPC back to |
63 // corresponding renderer process with the file handle. | 62 // corresponding renderer process with the file handle. |
64 static void DatabaseOpenFile(MessageLoop* io_thread_message_loop, | 63 static void DatabaseOpenFile(const OpenFileParams& params, |
65 const OpenFileParams& params, | |
66 int32 message_id, | 64 int32 message_id, |
67 ResourceMessageFilter* sender) { | 65 ResourceMessageFilter* sender) { |
68 base::PlatformFile target_handle = base::kInvalidPlatformFileValue; | 66 base::PlatformFile target_handle = base::kInvalidPlatformFileValue; |
69 base::PlatformFile target_dir_handle = base::kInvalidPlatformFileValue; | 67 base::PlatformFile target_dir_handle = base::kInvalidPlatformFileValue; |
70 VfsBackend::OpenFile(params.file_name, params.db_dir, params.desired_flags, | 68 VfsBackend::OpenFile(params.file_name, params.db_dir, params.desired_flags, |
71 params.handle, &target_handle, &target_dir_handle); | 69 params.handle, &target_handle, &target_dir_handle); |
72 | 70 |
73 ViewMsg_DatabaseOpenFileResponse_Params response_params; | 71 ViewMsg_DatabaseOpenFileResponse_Params response_params; |
74 #if defined(OS_WIN) | 72 #if defined(OS_WIN) |
75 response_params.file_handle = target_handle; | 73 response_params.file_handle = target_handle; |
76 #elif defined(OS_POSIX) | 74 #elif defined(OS_POSIX) |
77 response_params.file_handle = base::FileDescriptor(target_handle, true); | 75 response_params.file_handle = base::FileDescriptor(target_handle, true); |
78 response_params.dir_handle = base::FileDescriptor(target_dir_handle, true); | 76 response_params.dir_handle = base::FileDescriptor(target_dir_handle, true); |
79 #endif | 77 #endif |
80 io_thread_message_loop->PostTask(FROM_HERE, | 78 ChromeThread::PostTask( |
| 79 ChromeThread::IO, FROM_HERE, |
81 NewRunnableFunction(SendMessage, sender, | 80 NewRunnableFunction(SendMessage, sender, |
82 new ViewMsg_DatabaseOpenFileResponse(message_id, response_params))); | 81 new ViewMsg_DatabaseOpenFileResponse(message_id, response_params))); |
83 } | 82 } |
84 | 83 |
85 // Scheduled by the IO thread on the file thread. | 84 // Scheduled by the IO thread on the file thread. |
86 // Deletes the given database file, then schedules | 85 // Deletes the given database file, then schedules |
87 // a task on the IO thread's message loop to send an IPC back to | 86 // a task on the IO thread's message loop to send an IPC back to |
88 // corresponding renderer process with the error code. | 87 // corresponding renderer process with the error code. |
89 static void DatabaseDeleteFile(MessageLoop* io_thread_message_loop, | 88 static void DatabaseDeleteFile(const DeleteFileParams& params, |
90 const DeleteFileParams& params, | |
91 int32 message_id, | 89 int32 message_id, |
92 int reschedule_count, | 90 int reschedule_count, |
93 ResourceMessageFilter* sender) { | 91 ResourceMessageFilter* sender) { |
94 // Return an error if the file could not be deleted | 92 // Return an error if the file could not be deleted |
95 // after kNumDeleteRetries times. | 93 // after kNumDeleteRetries times. |
96 if (!reschedule_count) { | 94 if (!reschedule_count) { |
97 io_thread_message_loop->PostTask(FROM_HERE, | 95 ChromeThread::PostTask( |
| 96 ChromeThread::IO, FROM_HERE, |
98 NewRunnableFunction(SendMessage, sender, | 97 NewRunnableFunction(SendMessage, sender, |
99 new ViewMsg_DatabaseDeleteFileResponse(message_id, | 98 new ViewMsg_DatabaseDeleteFileResponse(message_id, |
100 SQLITE_IOERR_DELETE))); | 99 SQLITE_IOERR_DELETE))); |
101 return; | 100 return; |
102 } | 101 } |
103 | 102 |
104 int error_code = | 103 int error_code = |
105 VfsBackend::DeleteFile(params.file_name, params.db_dir, params.sync_dir); | 104 VfsBackend::DeleteFile(params.file_name, params.db_dir, params.sync_dir); |
106 if (error_code == SQLITE_IOERR_DELETE) { | 105 if (error_code == SQLITE_IOERR_DELETE) { |
107 // If the file could not be deleted, try again. | 106 // If the file could not be deleted, try again. |
108 MessageLoop::current()->PostDelayedTask(FROM_HERE, | 107 MessageLoop::current()->PostDelayedTask(FROM_HERE, |
109 NewRunnableFunction(DatabaseDeleteFile, io_thread_message_loop, | 108 NewRunnableFunction(DatabaseDeleteFile, params, message_id, |
110 params, message_id, reschedule_count - 1, sender), | 109 reschedule_count - 1, sender), |
111 kDelayDeleteRetryMs); | 110 kDelayDeleteRetryMs); |
112 return; | 111 return; |
113 } | 112 } |
114 | 113 |
115 io_thread_message_loop->PostTask(FROM_HERE, | 114 ChromeThread::PostTask( |
| 115 ChromeThread::IO, FROM_HERE, |
116 NewRunnableFunction(SendMessage, sender, | 116 NewRunnableFunction(SendMessage, sender, |
117 new ViewMsg_DatabaseDeleteFileResponse(message_id, error_code))); | 117 new ViewMsg_DatabaseDeleteFileResponse(message_id, error_code))); |
118 } | 118 } |
119 | 119 |
120 // Scheduled by the IO thread on the file thread. | 120 // Scheduled by the IO thread on the file thread. |
121 // Gets the attributes of the given database file, then schedules | 121 // Gets the attributes of the given database file, then schedules |
122 // a task on the IO thread's message loop to send an IPC back to | 122 // a task on the IO thread's message loop to send an IPC back to |
123 // corresponding renderer process. | 123 // corresponding renderer process. |
124 static void DatabaseGetFileAttributes(MessageLoop* io_thread_message_loop, | 124 static void DatabaseGetFileAttributes(const FilePath& file_name, |
125 const FilePath& file_name, | |
126 int32 message_id, | 125 int32 message_id, |
127 ResourceMessageFilter* sender) { | 126 ResourceMessageFilter* sender) { |
128 uint32 attributes = VfsBackend::GetFileAttributes(file_name); | 127 uint32 attributes = VfsBackend::GetFileAttributes(file_name); |
129 io_thread_message_loop->PostTask(FROM_HERE, | 128 ChromeThread::PostTask( |
| 129 ChromeThread::IO, FROM_HERE, |
130 NewRunnableFunction(SendMessage, sender, | 130 NewRunnableFunction(SendMessage, sender, |
131 new ViewMsg_DatabaseGetFileAttributesResponse( | 131 new ViewMsg_DatabaseGetFileAttributesResponse( |
132 message_id, attributes))); | 132 message_id, attributes))); |
133 } | 133 } |
134 | 134 |
135 // Scheduled by the IO thread on the file thread. | 135 // Scheduled by the IO thread on the file thread. |
136 // Gets the size of the given file, then schedules a task | 136 // Gets the size of the given file, then schedules a task |
137 // on the IO thread's message loop to send an IPC back to | 137 // on the IO thread's message loop to send an IPC back to |
138 // the corresponding renderer process. | 138 // the corresponding renderer process. |
139 static void DatabaseGetFileSize(MessageLoop* io_thread_message_loop, | 139 static void DatabaseGetFileSize(const FilePath& file_name, |
140 const FilePath& file_name, | |
141 int32 message_id, | 140 int32 message_id, |
142 ResourceMessageFilter* sender) { | 141 ResourceMessageFilter* sender) { |
143 int64 size = VfsBackend::GetFileSize(file_name); | 142 int64 size = VfsBackend::GetFileSize(file_name); |
144 io_thread_message_loop->PostTask(FROM_HERE, | 143 ChromeThread::PostTask( |
| 144 ChromeThread::IO, FROM_HERE, |
145 NewRunnableFunction(SendMessage, sender, | 145 NewRunnableFunction(SendMessage, sender, |
146 new ViewMsg_DatabaseGetFileSizeResponse(message_id, size))); | 146 new ViewMsg_DatabaseGetFileSizeResponse(message_id, size))); |
147 } | 147 } |
148 | 148 |
149 } // namespace | 149 } // namespace |
150 | 150 |
151 DatabaseDispatcherHost::DatabaseDispatcherHost( | 151 DatabaseDispatcherHost::DatabaseDispatcherHost( |
152 const FilePath& profile_path, | 152 const FilePath& profile_path, |
153 ResourceMessageFilter* resource_message_filter) | 153 ResourceMessageFilter* resource_message_filter) |
154 : profile_path_(profile_path), | 154 : profile_path_(profile_path), |
155 resource_message_filter_(resource_message_filter), | 155 resource_message_filter_(resource_message_filter) { |
156 file_thread_message_loop_( | |
157 g_browser_process->file_thread()->message_loop()) { | |
158 } | |
159 | |
160 | |
161 bool DatabaseDispatcherHost::IsDBMessage(const IPC::Message& message) { | |
162 switch (message.type()) { | |
163 case ViewHostMsg_DatabaseOpenFile::ID: | |
164 case ViewHostMsg_DatabaseDeleteFile::ID: | |
165 case ViewHostMsg_DatabaseGetFileAttributes::ID: | |
166 case ViewHostMsg_DatabaseGetFileSize::ID: | |
167 return true; | |
168 default: | |
169 return false; | |
170 } | |
171 } | 156 } |
172 | 157 |
173 bool DatabaseDispatcherHost::OnMessageReceived( | 158 bool DatabaseDispatcherHost::OnMessageReceived( |
174 const IPC::Message& message, bool* message_was_ok) { | 159 const IPC::Message& message, bool* message_was_ok) { |
175 if (!IsDBMessage(message)) | |
176 return false; | |
177 *message_was_ok = true; | 160 *message_was_ok = true; |
178 | 161 |
179 bool handled = true; | 162 bool handled = true; |
180 IPC_BEGIN_MESSAGE_MAP_EX(DatabaseDispatcherHost, message, *message_was_ok) | 163 IPC_BEGIN_MESSAGE_MAP_EX(DatabaseDispatcherHost, message, *message_was_ok) |
181 IPC_MESSAGE_HANDLER(ViewHostMsg_DatabaseOpenFile, OnDatabaseOpenFile); | 164 IPC_MESSAGE_HANDLER(ViewHostMsg_DatabaseOpenFile, OnDatabaseOpenFile); |
182 IPC_MESSAGE_HANDLER(ViewHostMsg_DatabaseDeleteFile, OnDatabaseDeleteFile); | 165 IPC_MESSAGE_HANDLER(ViewHostMsg_DatabaseDeleteFile, OnDatabaseDeleteFile); |
183 IPC_MESSAGE_HANDLER(ViewHostMsg_DatabaseGetFileAttributes, | 166 IPC_MESSAGE_HANDLER(ViewHostMsg_DatabaseGetFileAttributes, |
184 OnDatabaseGetFileAttributes); | 167 OnDatabaseGetFileAttributes); |
185 IPC_MESSAGE_HANDLER(ViewHostMsg_DatabaseGetFileSize, | 168 IPC_MESSAGE_HANDLER(ViewHostMsg_DatabaseGetFileSize, |
186 OnDatabaseGetFileSize); | 169 OnDatabaseGetFileSize); |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
223 message_id, response_params)); | 206 message_id, response_params)); |
224 return; | 207 return; |
225 } | 208 } |
226 | 209 |
227 OpenFileParams params; | 210 OpenFileParams params; |
228 params.db_dir = GetDBDir(); | 211 params.db_dir = GetDBDir(); |
229 params.file_name = db_file_name; | 212 params.file_name = db_file_name; |
230 params.desired_flags = desired_flags; | 213 params.desired_flags = desired_flags; |
231 params.handle = resource_message_filter_->handle(); | 214 params.handle = resource_message_filter_->handle(); |
232 resource_message_filter_->AddRef(); | 215 resource_message_filter_->AddRef(); |
233 file_thread_message_loop_->PostTask(FROM_HERE, | 216 ChromeThread::PostTask( |
234 NewRunnableFunction(DatabaseOpenFile, MessageLoop::current(), | 217 ChromeThread::FILE, FROM_HERE, |
235 params, message_id, resource_message_filter_)); | 218 NewRunnableFunction( |
| 219 DatabaseOpenFile, params, message_id, resource_message_filter_)); |
236 } | 220 } |
237 | 221 |
238 void DatabaseDispatcherHost::OnDatabaseDeleteFile( | 222 void DatabaseDispatcherHost::OnDatabaseDeleteFile( |
239 const FilePath& file_name, const bool& sync_dir, int32 message_id) { | 223 const FilePath& file_name, const bool& sync_dir, int32 message_id) { |
240 FilePath db_file_name = GetDBFileFullPath(file_name); | 224 FilePath db_file_name = GetDBFileFullPath(file_name); |
241 if (db_file_name.empty()) { | 225 if (db_file_name.empty()) { |
242 resource_message_filter_->Send(new ViewMsg_DatabaseDeleteFileResponse( | 226 resource_message_filter_->Send(new ViewMsg_DatabaseDeleteFileResponse( |
243 message_id, SQLITE_IOERR_DELETE)); | 227 message_id, SQLITE_IOERR_DELETE)); |
244 return; | 228 return; |
245 } | 229 } |
246 | 230 |
247 DeleteFileParams params; | 231 DeleteFileParams params; |
248 params.db_dir = GetDBDir(); | 232 params.db_dir = GetDBDir(); |
249 params.file_name = db_file_name; | 233 params.file_name = db_file_name; |
250 params.sync_dir = sync_dir; | 234 params.sync_dir = sync_dir; |
251 resource_message_filter_->AddRef(); | 235 resource_message_filter_->AddRef(); |
252 file_thread_message_loop_->PostTask(FROM_HERE, | 236 ChromeThread::PostTask( |
253 NewRunnableFunction(DatabaseDeleteFile, MessageLoop::current(), | 237 ChromeThread::FILE, FROM_HERE, |
254 params, message_id, kNumDeleteRetries, | 238 NewRunnableFunction( |
255 resource_message_filter_)); | 239 DatabaseDeleteFile, params, message_id, kNumDeleteRetries, |
| 240 resource_message_filter_)); |
256 } | 241 } |
257 | 242 |
258 void DatabaseDispatcherHost::OnDatabaseGetFileAttributes( | 243 void DatabaseDispatcherHost::OnDatabaseGetFileAttributes( |
259 const FilePath& file_name, int32 message_id) { | 244 const FilePath& file_name, int32 message_id) { |
260 FilePath db_file_name = GetDBFileFullPath(file_name); | 245 FilePath db_file_name = GetDBFileFullPath(file_name); |
261 if (db_file_name.empty()) { | 246 if (db_file_name.empty()) { |
262 resource_message_filter_->Send( | 247 resource_message_filter_->Send( |
263 new ViewMsg_DatabaseGetFileAttributesResponse(message_id, -1)); | 248 new ViewMsg_DatabaseGetFileAttributesResponse(message_id, -1)); |
264 return; | 249 return; |
265 } | 250 } |
266 | 251 |
267 resource_message_filter_->AddRef(); | 252 resource_message_filter_->AddRef(); |
268 file_thread_message_loop_->PostTask(FROM_HERE, | 253 ChromeThread::PostTask( |
269 NewRunnableFunction(DatabaseGetFileAttributes, MessageLoop::current(), | 254 ChromeThread::FILE, FROM_HERE, |
270 db_file_name, message_id, resource_message_filter_)); | 255 NewRunnableFunction( |
| 256 DatabaseGetFileAttributes, db_file_name, message_id, |
| 257 resource_message_filter_)); |
271 } | 258 } |
272 | 259 |
273 void DatabaseDispatcherHost::OnDatabaseGetFileSize( | 260 void DatabaseDispatcherHost::OnDatabaseGetFileSize( |
274 const FilePath& file_name, int32 message_id) { | 261 const FilePath& file_name, int32 message_id) { |
275 FilePath db_file_name = GetDBFileFullPath(file_name); | 262 FilePath db_file_name = GetDBFileFullPath(file_name); |
276 if (db_file_name.empty()) { | 263 if (db_file_name.empty()) { |
277 resource_message_filter_->Send(new ViewMsg_DatabaseGetFileSizeResponse( | 264 resource_message_filter_->Send(new ViewMsg_DatabaseGetFileSizeResponse( |
278 message_id, 0)); | 265 message_id, 0)); |
279 return; | 266 return; |
280 } | 267 } |
281 | 268 |
282 resource_message_filter_->AddRef(); | 269 resource_message_filter_->AddRef(); |
283 file_thread_message_loop_->PostTask(FROM_HERE, | 270 ChromeThread::PostTask( |
284 NewRunnableFunction(DatabaseGetFileSize, MessageLoop::current(), | 271 ChromeThread::FILE, FROM_HERE, |
285 db_file_name, message_id, resource_message_filter_)); | 272 NewRunnableFunction( |
| 273 DatabaseGetFileSize, db_file_name, message_id, |
| 274 resource_message_filter_)); |
286 } | 275 } |
OLD | NEW |