Index: remoting/protocol/client_control_dispatcher.h |
diff --git a/remoting/protocol/client_control_dispatcher.h b/remoting/protocol/client_control_dispatcher.h |
index a867e137ad0bd9d337b3a1b60798ed0f026ae031..014662c131570c86834c2c47f2de0d0fdf54e9fc 100644 |
--- a/remoting/protocol/client_control_dispatcher.h |
+++ b/remoting/protocol/client_control_dispatcher.h |
@@ -8,6 +8,7 @@ |
#include "base/memory/ref_counted.h" |
#include "remoting/protocol/channel_dispatcher_base.h" |
#include "remoting/protocol/clipboard_stub.h" |
+#include "remoting/protocol/cursor_shape_stub.h" |
#include "remoting/protocol/host_stub.h" |
#include "remoting/protocol/message_reader.h" |
@@ -20,8 +21,8 @@ class BufferedSocketWriter; |
class Session; |
// ClientControlDispatcher dispatches incoming messages on the control |
-// channel to ClientStub or ClipboardStub, and also implements ClipboardStub |
-// and HostStub for outgoing messages. |
+// channel to ClientStub, ClipboardStub or CursorShapeStub. |
+// It also implements ClipboardStub and HostStub for outgoing messages. |
class ClientControlDispatcher : public ChannelDispatcherBase, |
public ClipboardStub, |
public HostStub { |
@@ -47,6 +48,11 @@ class ClientControlDispatcher : public ChannelDispatcherBase, |
clipboard_stub_ = clipboard_stub; |
} |
+ // Sets stub to call for incoming cursor shape messages. |
+ void set_cursor_shape_stub(CursorShapeStub* cursor_shape_stub) { |
+ cursor_shape_stub_ = cursor_shape_stub; |
+ } |
+ |
protected: |
// ChannelDispatcherBase overrides. |
virtual void OnInitialized() OVERRIDE; |
@@ -57,6 +63,7 @@ class ClientControlDispatcher : public ChannelDispatcherBase, |
ClientStub* client_stub_; |
ClipboardStub* clipboard_stub_; |
+ CursorShapeStub* cursor_shape_stub_; |
ProtobufMessageReader<ControlMessage> reader_; |
scoped_refptr<BufferedSocketWriter> writer_; |