1 files changed, 280 insertions, 0 deletions
diff --git a/content/browser/in_process_webkit/indexed_db_dispatcher_host.h b/content/browser/in_process_webkit/indexed_db_dispatcher_host.h
new file mode 100644
index 0000000..1823954
--- /dev/null
+++ b/content/browser/in_process_webkit/indexed_db_dispatcher_host.h
@@ -0,0 +1,280 @@
+// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef CONTENT_BROWSER_IN_PROCESS_WEBKIT_INDEXED_DB_DISPATCHER_HOST_H_
+#define CONTENT_BROWSER_IN_PROCESS_WEBKIT_INDEXED_DB_DISPATCHER_HOST_H_
+#pragma once
+
+#include "base/basictypes.h"
+#include "base/id_map.h"
+#include "content/browser/browser_message_filter.h"
+#include "content/browser/in_process_webkit/webkit_context.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebExceptionCode.h"
+
+class HostContentSettingsMap;
+class IndexedDBKey;
+class NullableString16;
+class Profile;
+class SerializedScriptValue;
+struct IndexedDBHostMsg_DatabaseCreateObjectStore_Params;
+struct IndexedDBHostMsg_FactoryDeleteDatabase_Params;
+struct IndexedDBHostMsg_FactoryOpen_Params;
+struct IndexedDBHostMsg_IndexOpenCursor_Params;
+struct IndexedDBHostMsg_ObjectStoreCreateIndex_Params;
+struct IndexedDBHostMsg_ObjectStoreOpenCursor_Params;
+struct IndexedDBHostMsg_ObjectStorePut_Params;
+
+namespace WebKit {
+class WebIDBCursor;
+class WebIDBDatabase;
+class WebIDBIndex;
+class WebIDBObjectStore;
+class WebIDBTransaction;
+}
+
+// Handles all IndexedDB related messages from a particular renderer process.
+class IndexedDBDispatcherHost : public BrowserMessageFilter {
+ public:
+  // Only call the constructor from the UI thread.
+  IndexedDBDispatcherHost(int process_id, Profile* profile);
+
+  // BrowserMessageFilter implementation.
+  virtual void OnChannelClosing();
+  virtual void OverrideThreadForMessage(const IPC::Message& message,
+                                        BrowserThread::ID* thread);
+  virtual bool OnMessageReceived(const IPC::Message& message,
+                                 bool* message_was_ok);
+
+  // A shortcut for accessing our context.
+  IndexedDBContext* Context() {
+    return webkit_context_->indexed_db_context();
+  }
+
+  // The various IndexedDBCallbacks children call these methods to add the
+  // results into the applicable map.  See below for more details.
+  int32 Add(WebKit::WebIDBCursor* idb_cursor);
+  int32 Add(WebKit::WebIDBDatabase* idb_database);
+  int32 Add(WebKit::WebIDBIndex* idb_index);
+  int32 Add(WebKit::WebIDBObjectStore* idb_object_store);
+  int32 Add(WebKit::WebIDBTransaction* idb_transaction);
+
+ private:
+  ~IndexedDBDispatcherHost();
+
+  // True if the given |origin| can use databases according to the content
+  // settings.
+  bool CheckContentSetting(const string16& origin,
+                           const string16& description,
+                           int routing_id,
+                           int response_id);
+
+  // Message processing. Most of the work is delegated to the dispatcher hosts
+  // below.
+  void OnIDBFactoryOpen(const IndexedDBHostMsg_FactoryOpen_Params& p);
+
+  void OnIDBFactoryDeleteDatabase(
+      const IndexedDBHostMsg_FactoryDeleteDatabase_Params& p);
+
+  // Helper templates.
+  template <class ReturnType>
+  ReturnType* GetOrTerminateProcess(
+    IDMap<ReturnType, IDMapOwnPointer>* map, int32 return_object_id);
+
+  template <typename ReplyType, typename WebObjectType, typename Method>
+  void SyncGetter(IDMap<WebObjectType, IDMapOwnPointer>* map, int32 object_id,
+                  ReplyType* reply, Method method);
+
+  template <typename ObjectType>
+  void DestroyObject(IDMap<ObjectType, IDMapOwnPointer>* map, int32 object_id);
+
+  class DatabaseDispatcherHost {
+   public:
+    explicit DatabaseDispatcherHost(IndexedDBDispatcherHost* parent);
+    ~DatabaseDispatcherHost();
+
+    bool OnMessageReceived(const IPC::Message& message, bool *msg_is_ok);
+    void Send(IPC::Message* message);
+
+    void OnName(int32 idb_database_id, string16* name);
+    void OnVersion(int32 idb_database_id, string16* version);
+    void OnObjectStoreNames(int32 idb_database_id,
+                            std::vector<string16>* object_stores);
+    void OnCreateObjectStore(
+        const IndexedDBHostMsg_DatabaseCreateObjectStore_Params& params,
+        int32* object_store_id, WebKit::WebExceptionCode* ec);
+    void OnDeleteObjectStore(int32 idb_database_id,
+                             const string16& name,
+                             int32 transaction_id,
+                             WebKit::WebExceptionCode* ec);
+    void OnSetVersion(int32 idb_database_id,
+                      int32 response_id,
+                      const string16& version,
+                      WebKit::WebExceptionCode* ec);
+    void OnTransaction(int32 idb_database_id,
+                       const std::vector<string16>& names,
+                       int32 mode, int32 timeout,
+                       int32* idb_transaction_id,
+                       WebKit::WebExceptionCode* ec);
+    void OnClose(int32 idb_database_id);
+    void OnDestroyed(int32 idb_database_id);
+
+    IndexedDBDispatcherHost* parent_;
+    IDMap<WebKit::WebIDBDatabase, IDMapOwnPointer> map_;
+  };
+
+  class IndexDispatcherHost {
+   public:
+    explicit IndexDispatcherHost(IndexedDBDispatcherHost* parent);
+    ~IndexDispatcherHost();
+
+    bool OnMessageReceived(const IPC::Message& message, bool *msg_is_ok);
+    void Send(IPC::Message* message);
+
+    void OnName(int32 idb_index_id, string16* name);
+    void OnStoreName(int32 idb_index_id, string16* store_name);
+    void OnKeyPath(int32 idb_index_id, NullableString16* key_path);
+    void OnUnique(int32 idb_index_id, bool* unique);
+    void OnOpenObjectCursor(
+        const IndexedDBHostMsg_IndexOpenCursor_Params& params,
+        WebKit::WebExceptionCode* ec);
+    void OnOpenKeyCursor(const IndexedDBHostMsg_IndexOpenCursor_Params& params,
+                         WebKit::WebExceptionCode* ec);
+    void OnGetObject(int idb_index_id,
+                     int32 response_id,
+                     const IndexedDBKey& key,
+                     int32 transaction_id,
+                     WebKit::WebExceptionCode* ec);
+    void OnGetKey(int idb_index_id,
+                  int32 response_id,
+                  const IndexedDBKey& key,
+                  int32 transaction_id,
+                  WebKit::WebExceptionCode* ec);
+    void OnDestroyed(int32 idb_index_id);
+
+    IndexedDBDispatcherHost* parent_;
+    IDMap<WebKit::WebIDBIndex, IDMapOwnPointer> map_;
+  };
+
+  class ObjectStoreDispatcherHost {
+   public:
+    explicit ObjectStoreDispatcherHost(IndexedDBDispatcherHost* parent);
+    ~ObjectStoreDispatcherHost();
+
+    bool OnMessageReceived(const IPC::Message& message, bool *msg_is_ok);
+    void Send(IPC::Message* message);
+
+    void OnName(int32 idb_object_store_id, string16* name);
+    void OnKeyPath(int32 idb_object_store_id, NullableString16* keyPath);
+    void OnIndexNames(int32 idb_object_store_id,
+                      std::vector<string16>* index_names);
+    void OnGet(int idb_object_store_id,
+               int32 response_id,
+               const IndexedDBKey& key,
+               int32 transaction_id,
+               WebKit::WebExceptionCode* ec);
+    void OnPut(const IndexedDBHostMsg_ObjectStorePut_Params& params,
+               WebKit::WebExceptionCode* ec);
+    void OnDelete(int idb_object_store_id,
+                  int32 response_id,
+                  const IndexedDBKey& key,
+                  int32 transaction_id,
+                  WebKit::WebExceptionCode* ec);
+    void OnClear(int idb_object_store_id,
+                 int32 response_id,
+                 int32 transaction_id,
+                 WebKit::WebExceptionCode* ec);
+    void OnCreateIndex(
+        const IndexedDBHostMsg_ObjectStoreCreateIndex_Params& params,
+        int32* index_id,
+        WebKit::WebExceptionCode* ec);
+    void OnIndex(int32 idb_object_store_id,
+                 const string16& name,
+                 int32* idb_index_id,
+                 WebKit::WebExceptionCode* ec);
+    void OnDeleteIndex(int32 idb_object_store_id,
+                       const string16& name,
+                       int32 transaction_id,
+                       WebKit::WebExceptionCode* ec);
+    void OnOpenCursor(
+        const IndexedDBHostMsg_ObjectStoreOpenCursor_Params& params,
+        WebKit::WebExceptionCode* ec);
+    void OnDestroyed(int32 idb_object_store_id);
+
+    IndexedDBDispatcherHost* parent_;
+    IDMap<WebKit::WebIDBObjectStore, IDMapOwnPointer> map_;
+  };
+
+  class CursorDispatcherHost {
+   public:
+    explicit CursorDispatcherHost(IndexedDBDispatcherHost* parent);
+    ~CursorDispatcherHost();
+
+    bool OnMessageReceived(const IPC::Message& message, bool *msg_is_ok);
+    void Send(IPC::Message* message);
+
+    void OnDirection(int32 idb_object_store_id, int32* direction);
+    void OnKey(int32 idb_object_store_id, IndexedDBKey* key);
+    void OnValue(int32 idb_object_store_id,
+                 SerializedScriptValue* script_value,
+                 IndexedDBKey* key);
+    void OnUpdate(int32 idb_object_store_id,
+                  int32 response_id,
+                  const SerializedScriptValue& value,
+                  WebKit::WebExceptionCode* ec);
+    void OnContinue(int32 idb_object_store_id,
+                    int32 response_id,
+                    const IndexedDBKey& key,
+                    WebKit::WebExceptionCode* ec);
+    void OnDelete(int32 idb_object_store_id,
+                  int32 response_id,
+                  WebKit::WebExceptionCode* ec);
+    void OnDestroyed(int32 idb_cursor_id);
+
+    IndexedDBDispatcherHost* parent_;
+    IDMap<WebKit::WebIDBCursor, IDMapOwnPointer> map_;
+  };
+
+  class TransactionDispatcherHost {
+   public:
+    explicit TransactionDispatcherHost(IndexedDBDispatcherHost* parent);
+    ~TransactionDispatcherHost();
+
+    bool OnMessageReceived(const IPC::Message& message, bool *msg_is_ok);
+    void Send(IPC::Message* message);
+
+    // TODO: add the rest of the transaction methods.
+    void OnAbort(int32 transaction_id);
+    void OnMode(int32 transaction_id, int* mode);
+    void OnObjectStore(int32 transaction_id,
+                       const string16& name,
+                       int32* object_store_id,
+                       WebKit::WebExceptionCode* ec);
+    void OnDidCompleteTaskEvents(int transaction_id);
+    void OnDestroyed(int32 idb_transaction_id);
+
+    IndexedDBDispatcherHost* parent_;
+    typedef IDMap<WebKit::WebIDBTransaction, IDMapOwnPointer> MapType;
+    MapType map_;
+  };
+
+  // Data shared between renderer processes with the same profile.
+  scoped_refptr<WebKitContext> webkit_context_;
+
+  // Tells us whether the user wants to allow databases to be opened.
+  scoped_refptr<HostContentSettingsMap> host_content_settings_map_;
+
+  // Only access on WebKit thread.
+  scoped_ptr<DatabaseDispatcherHost> database_dispatcher_host_;
+  scoped_ptr<IndexDispatcherHost> index_dispatcher_host_;
+  scoped_ptr<ObjectStoreDispatcherHost> object_store_dispatcher_host_;
+  scoped_ptr<CursorDispatcherHost> cursor_dispatcher_host_;
+  scoped_ptr<TransactionDispatcherHost> transaction_dispatcher_host_;
+
+  // Used to dispatch messages to the correct view host.
+  int process_id_;
+
+  DISALLOW_IMPLICIT_CONSTRUCTORS(IndexedDBDispatcherHost);
+};
+
+#endif  // CONTENT_BROWSER_IN_PROCESS_WEBKIT_INDEXED_DB_DISPATCHER_HOST_H_