OLD | NEW |
---|---|
(Empty) | |
1 // Copyright 2016 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_CHILD_WEBSOCKET_MESSAGE_FILTER_H_ | |
6 #define CONTENT_CHILD_WEBSOCKET_MESSAGE_FILTER_H_ | |
7 | |
8 #include "base/macros.h" | |
9 #include "base/memory/ref_counted.h" | |
10 #include "base/memory/weak_ptr.h" | |
11 #include "ipc/message_filter.h" | |
12 | |
13 namespace base { | |
14 class SingleThreadTaskRunner; | |
15 } | |
16 | |
17 namespace content { | |
18 | |
19 class WebSocketDispatcher; | |
20 | |
21 // Delegates IPC messages to a WebSocketDispatcher on the loading task queue | |
22 // instead of the default task queue. | |
23 class WebSocketMessageFilter : public IPC::MessageFilter { | |
24 public: | |
25 explicit WebSocketMessageFilter(WebSocketDispatcher* websocket_dispatcher); | |
26 | |
27 // IPC::MessageFilter implementation. | |
28 bool OnMessageReceived(const IPC::Message& message) override; | |
29 | |
30 void SetLoadingTaskRunner( | |
31 scoped_refptr<base::SingleThreadTaskRunner> loading_task_runner) { | |
32 loading_task_runner_ = loading_task_runner; | |
33 } | |
34 | |
35 private: | |
36 ~WebSocketMessageFilter() override; | |
37 void OnMessageReceivedOnMainThread(const IPC::Message& message); | |
38 | |
39 scoped_refptr<base::SingleThreadTaskRunner> loading_task_runner_; | |
40 | |
41 base::WeakPtr<WebSocketDispatcher> websocket_dispatcher_; | |
kinuko
2016/05/26 10:09:54
nit: add a comment to note this must be de-ref'ed
hajimehoshi
2016/05/26 10:18:42
Done.
| |
42 | |
43 DISALLOW_COPY_AND_ASSIGN(WebSocketMessageFilter); | |
44 }; | |
45 | |
46 } // namespace content | |
47 | |
48 #endif // CONTENT_CHILD_WEBSOCKET_MESSAGE_FILTER_H_ | |
OLD | NEW |