OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 SANDBOX_MAC_XPC_MESSAGE_SERVER_H_ | 5 #ifndef SANDBOX_MAC_XPC_MESSAGE_SERVER_H_ |
6 #define SANDBOX_MAC_XPC_MESSAGE_SERVER_H_ | 6 #define SANDBOX_MAC_XPC_MESSAGE_SERVER_H_ |
7 | 7 |
8 #include "base/mac/scoped_mach_port.h" | 8 #include "base/mac/scoped_mach_port.h" |
9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
10 #include "sandbox/mac/message_server.h" | 10 #include "sandbox/mac/message_server.h" |
11 #include "sandbox/mac/xpc.h" | 11 #include "sandbox/mac/xpc.h" |
12 #include "sandbox/sandbox_export.h" | 12 #include "sandbox/sandbox_export.h" |
13 | 13 |
14 namespace sandbox { | 14 namespace sandbox { |
15 | 15 |
16 class DispatchSourceMach; | 16 class DispatchSourceMach; |
17 | 17 |
18 // An implementation of MessageServer that uses XPC pipes to read and write XPC | 18 // An implementation of MessageServer that uses XPC pipes to read and write XPC |
19 // messages from a Mach port. | 19 // messages from a Mach port. |
20 class SANDBOX_EXPORT XPCMessageServer : public MessageServer { | 20 class SANDBOX_EXPORT XPCMessageServer : public MessageServer { |
21 public: | 21 public: |
22 // Creates a new XPC message server that will send messages to |demuxer| | 22 // Creates a new XPC message server that will send messages to |demuxer| |
23 // for handling. If the |server_receive_right| is non-NULL, this class will | 23 // for handling. If the |server_receive_right| is non-NULL, this class will |
24 // take ownership of the port and it will be used to receive messages. | 24 // take ownership of the port and it will be used to receive messages. |
25 // Otherwise the server will create a new receive right on which to listen. | 25 // Otherwise the server will create a new receive right on which to listen. |
26 XPCMessageServer(MessageDemuxer* demuxer, | 26 XPCMessageServer(MessageDemuxer* demuxer, |
27 mach_port_t server_receive_right); | 27 mach_port_t server_receive_right); |
28 virtual ~XPCMessageServer(); | 28 ~XPCMessageServer() override; |
29 | 29 |
30 // MessageServer: | 30 // MessageServer: |
31 virtual bool Initialize() override; | 31 bool Initialize() override; |
32 virtual pid_t GetMessageSenderPID(IPCMessage request) override; | 32 pid_t GetMessageSenderPID(IPCMessage request) override; |
33 virtual IPCMessage CreateReply(IPCMessage request) override; | 33 IPCMessage CreateReply(IPCMessage request) override; |
34 virtual bool SendReply(IPCMessage reply) override; | 34 bool SendReply(IPCMessage reply) override; |
35 virtual void ForwardMessage(IPCMessage request, | 35 void ForwardMessage(IPCMessage request, mach_port_t destination) override; |
36 mach_port_t destination) override; | |
37 // Creates an error reply message with a field "error" set to |error_code|. | 36 // Creates an error reply message with a field "error" set to |error_code|. |
38 virtual void RejectMessage(IPCMessage request, int error_code) override; | 37 void RejectMessage(IPCMessage request, int error_code) override; |
39 virtual mach_port_t GetServerPort() const override; | 38 mach_port_t GetServerPort() const override; |
40 | 39 |
41 private: | 40 private: |
42 // Reads a message from the XPC pipe. | 41 // Reads a message from the XPC pipe. |
43 void ReceiveMessage(); | 42 void ReceiveMessage(); |
44 | 43 |
45 // The demuxer delegate. Weak. | 44 // The demuxer delegate. Weak. |
46 MessageDemuxer* demuxer_; | 45 MessageDemuxer* demuxer_; |
47 | 46 |
48 // The Mach port on which the server is receiving requests. | 47 // The Mach port on which the server is receiving requests. |
49 base::mac::ScopedMachReceiveRight server_port_; | 48 base::mac::ScopedMachReceiveRight server_port_; |
50 | 49 |
51 // MACH_RECV dispatch source that handles the |server_port_|. | 50 // MACH_RECV dispatch source that handles the |server_port_|. |
52 scoped_ptr<DispatchSourceMach> dispatch_source_; | 51 scoped_ptr<DispatchSourceMach> dispatch_source_; |
53 | 52 |
54 // The reply message, if one has been created. | 53 // The reply message, if one has been created. |
55 xpc_object_t reply_message_; | 54 xpc_object_t reply_message_; |
56 | 55 |
57 DISALLOW_COPY_AND_ASSIGN(XPCMessageServer); | 56 DISALLOW_COPY_AND_ASSIGN(XPCMessageServer); |
58 }; | 57 }; |
59 | 58 |
60 } // namespace sandbox | 59 } // namespace sandbox |
61 | 60 |
62 #endif // SANDBOX_MAC_XPC_MESSAGE_SERVER_H_ | 61 #endif // SANDBOX_MAC_XPC_MESSAGE_SERVER_H_ |
OLD | NEW |