| 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 |