| 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 #ifndef CONTENT_CHILD_QUOTA_DISPATCHER_H_ | 5 #ifndef CONTENT_CHILD_QUOTA_DISPATCHER_H_ |
| 6 #define CONTENT_CHILD_QUOTA_DISPATCHER_H_ | 6 #define CONTENT_CHILD_QUOTA_DISPATCHER_H_ |
| 7 | 7 |
| 8 #include <map> | 8 #include <map> |
| 9 #include <set> | 9 #include <set> |
| 10 | 10 |
| 11 #include "base/basictypes.h" | 11 #include "base/basictypes.h" |
| 12 #include "base/id_map.h" | 12 #include "base/id_map.h" |
| 13 #include "base/memory/ref_counted.h" | 13 #include "ipc/ipc_listener.h" |
| 14 #include "webkit/child/worker_task_runner.h" | |
| 15 #include "webkit/common/quota/quota_types.h" | 14 #include "webkit/common/quota/quota_types.h" |
| 16 | 15 |
| 17 class GURL; | 16 class GURL; |
| 18 | 17 |
| 19 namespace IPC { | 18 namespace IPC { |
| 20 class Message; | 19 class Message; |
| 21 } | 20 } |
| 22 | 21 |
| 23 namespace WebKit { | 22 namespace WebKit { |
| 24 class WebStorageQuotaCallbacks; | 23 class WebStorageQuotaCallbacks; |
| 25 } | 24 } |
| 26 | 25 |
| 27 namespace content { | 26 namespace content { |
| 28 | 27 |
| 29 class ThreadSafeSender; | |
| 30 class QuotaMessageFilter; | |
| 31 | |
| 32 // Dispatches and sends quota related messages sent to/from a child | 28 // Dispatches and sends quota related messages sent to/from a child |
| 33 // process from/to the main browser process. There is one instance | 29 // process from/to the main browser process. There is one instance |
| 34 // per each thread. Thread-specific instance can be obtained by | 30 // per child process. Messages are dispatched on the main child thread. |
| 35 // ThreadSpecificInstance(). | 31 class QuotaDispatcher : public IPC::Listener { |
| 36 class QuotaDispatcher : public webkit_glue::WorkerTaskRunner::Observer { | |
| 37 public: | 32 public: |
| 38 class Callback { | 33 class Callback { |
| 39 public: | 34 public: |
| 40 virtual ~Callback() {} | 35 virtual ~Callback() {} |
| 41 virtual void DidQueryStorageUsageAndQuota(int64 usage, int64 quota) = 0; | 36 virtual void DidQueryStorageUsageAndQuota(int64 usage, int64 quota) = 0; |
| 42 virtual void DidGrantStorageQuota(int64 granted_quota) = 0; | 37 virtual void DidGrantStorageQuota(int64 granted_quota) = 0; |
| 43 virtual void DidFail(quota::QuotaStatusCode status) = 0; | 38 virtual void DidFail(quota::QuotaStatusCode status) = 0; |
| 44 }; | 39 }; |
| 45 | 40 |
| 46 QuotaDispatcher(ThreadSafeSender* thread_safe_sender, | 41 QuotaDispatcher(); |
| 47 QuotaMessageFilter* quota_message_filter); | |
| 48 virtual ~QuotaDispatcher(); | 42 virtual ~QuotaDispatcher(); |
| 49 | 43 |
| 50 // |thread_safe_sender| and |quota_message_filter| are used if | 44 // IPC::Listener implementation. |
| 51 // calling this leads to construction. | 45 virtual bool OnMessageReceived(const IPC::Message& msg) OVERRIDE; |
| 52 static QuotaDispatcher* ThreadSpecificInstance( | |
| 53 ThreadSafeSender* thread_safe_sender, | |
| 54 QuotaMessageFilter* quota_message_filter); | |
| 55 | |
| 56 // webkit_glue::WorkerTaskRunner::Observer implementation. | |
| 57 virtual void OnWorkerRunLoopStopped() OVERRIDE; | |
| 58 | |
| 59 void OnMessageReceived(const IPC::Message& msg); | |
| 60 | 46 |
| 61 void QueryStorageUsageAndQuota(const GURL& gurl, | 47 void QueryStorageUsageAndQuota(const GURL& gurl, |
| 62 quota::StorageType type, | 48 quota::StorageType type, |
| 63 Callback* callback); | 49 Callback* callback); |
| 64 void RequestStorageQuota(int render_view_id, | 50 void RequestStorageQuota(int render_view_id, |
| 65 const GURL& gurl, | 51 const GURL& gurl, |
| 66 quota::StorageType type, | 52 quota::StorageType type, |
| 67 int64 requested_size, | 53 int64 requested_size, |
| 68 Callback* callback); | 54 Callback* callback); |
| 69 | 55 |
| 70 // Creates a new Callback instance for WebStorageQuotaCallbacks. | 56 // Creates a new Callback instance for WebStorageQuotaCallbacks. |
| 71 static Callback* CreateWebStorageQuotaCallbacksWrapper( | 57 static Callback* CreateWebStorageQuotaCallbacksWrapper( |
| 72 WebKit::WebStorageQuotaCallbacks* callbacks); | 58 WebKit::WebStorageQuotaCallbacks* callbacks); |
| 73 | 59 |
| 74 private: | 60 private: |
| 75 // Message handlers. | 61 // Message handlers. |
| 76 void DidQueryStorageUsageAndQuota(int request_id, | 62 void DidQueryStorageUsageAndQuota(int request_id, |
| 77 int64 current_usage, | 63 int64 current_usage, |
| 78 int64 current_quota); | 64 int64 current_quota); |
| 79 void DidGrantStorageQuota(int request_id, | 65 void DidGrantStorageQuota(int request_id, |
| 80 int64 granted_quota); | 66 int64 granted_quota); |
| 81 void DidFail(int request_id, | 67 void DidFail(int request_id, quota::QuotaStatusCode error); |
| 82 quota::QuotaStatusCode error); | |
| 83 | 68 |
| 84 IDMap<Callback, IDMapOwnPointer> pending_quota_callbacks_; | 69 IDMap<Callback, IDMapOwnPointer> pending_quota_callbacks_; |
| 85 | 70 |
| 86 scoped_refptr<ThreadSafeSender> thread_safe_sender_; | |
| 87 scoped_refptr<QuotaMessageFilter> quota_message_filter_; | |
| 88 | |
| 89 DISALLOW_COPY_AND_ASSIGN(QuotaDispatcher); | 71 DISALLOW_COPY_AND_ASSIGN(QuotaDispatcher); |
| 90 }; | 72 }; |
| 91 | 73 |
| 92 } // namespace content | 74 } // namespace content |
| 93 | 75 |
| 94 #endif // CONTENT_CHILD_QUOTA_DISPATCHER_H_ | 76 #endif // CONTENT_CHILD_QUOTA_DISPATCHER_H_ |
| OLD | NEW |