1 // Copyright (c) 2012 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 #ifndef CONTENT_BROWSER_WORKER_HOST_WORKER_DOCUMENT_SET_H_ 6 #define CONTENT_BROWSER_WORKER_HOST_WORKER_DOCUMENT_SET_H_ 7 8 #include <set> 9 10 #include "base/basictypes.h" 11 #include "base/memory/ref_counted.h" 12 13 namespace content { 14 class WorkerMessageFilter; 15 16 // The WorkerDocumentSet tracks all of the DOM documents associated with a 17 // set of workers. With nested workers, multiple workers can share the same 18 // WorkerDocumentSet (meaning that they all share the same lifetime, and will 19 // all exit when the last document in that set exits, per the WebWorkers spec). 20 class WorkerDocumentSet : public base::RefCounted<WorkerDocumentSet> { 21 public: 22 WorkerDocumentSet(); 23 24 // The information we track for each document 25 class DocumentInfo { 26 public: 27 DocumentInfo(WorkerMessageFilter* filter, unsigned long long document_id, 28 int renderer_process_id, int render_view_id); filter()29 WorkerMessageFilter* filter() const { return filter_; } document_id()30 unsigned long long document_id() const { return document_id_; } render_process_id()31 int render_process_id() const { return render_process_id_; } render_view_id()32 int render_view_id() const { return render_view_id_; } 33 34 // Define operator "<", which is used to determine uniqueness within 35 // the set. 36 bool operator <(const DocumentInfo& other) const { 37 // Items are identical if the sender and document_id are identical, 38 // otherwise create an arbitrary stable ordering based on the document 39 // id/filter. 40 if (filter() == other.filter()) { 41 return document_id() < other.document_id(); 42 } else { 43 return reinterpret_cast<unsigned long long>(filter()) < 44 reinterpret_cast<unsigned long long>(other.filter()); 45 } 46 } 47 48 private: 49 WorkerMessageFilter* filter_; 50 unsigned long long document_id_; 51 int render_process_id_; 52 int render_view_id_; 53 }; 54 55 // Adds a document to a shared worker's document set. Also includes the 56 // associated render_process_id the document is associated with, to enable 57 // communication with the parent tab for things like http auth dialogs. 58 void Add(WorkerMessageFilter* parent, 59 unsigned long long document_id, 60 int render_process_id, 61 int render_view_id); 62 63 // Checks to see if a document is in a shared worker's document set. 64 bool Contains(WorkerMessageFilter* parent, 65 unsigned long long document_id) const; 66 67 // Removes a specific document from a worker's document set when that document 68 // is detached. 69 void Remove(WorkerMessageFilter* parent, unsigned long long document_id); 70 71 // Invoked when a render process exits, to remove all associated documents 72 // from a worker's document set. 73 void RemoveAll(WorkerMessageFilter* parent); 74 IsEmpty()75 bool IsEmpty() const { return document_set_.empty(); } 76 77 // Define a typedef for convenience here when declaring iterators, etc. 78 typedef std::set<DocumentInfo> DocumentInfoSet; 79 80 // Returns the set of documents associated with this worker. documents()81 const DocumentInfoSet& documents() { return document_set_; } 82 83 private: 84 friend class base::RefCounted<WorkerDocumentSet>; 85 virtual ~WorkerDocumentSet(); 86 87 DocumentInfoSet document_set_; 88 }; 89 90 } // namespace content 91 92 #endif // CONTENT_BROWSER_WORKER_HOST_WORKER_DOCUMENT_SET_H_ 93