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 3 // found in the LICENSE file. 4 5 #ifndef SANDBOX_MAC_MACH_MESSAGE_SERVER_H_ 6 #define SANDBOX_MAC_MACH_MESSAGE_SERVER_H_ 7 8 #include <mach/mach.h> 9 10 #include "base/mac/scoped_mach_port.h" 11 #include "base/mac/scoped_mach_vm.h" 12 #include "base/memory/scoped_ptr.h" 13 #include "sandbox/mac/message_server.h" 14 15 namespace sandbox { 16 17 class DispatchSourceMach; 18 19 // A Mach message server that operates a receive port. Messages are received 20 // and then passed to the MessageDemuxer for handling. The Demuxer 21 // can use the server class to send a reply, forward the message to a 22 // different port, or reply to the message with a MIG error. 23 class MachMessageServer : public MessageServer { 24 public: 25 // Creates a new Mach message server that will send messages to |demuxer| 26 // for handling. If the |server_receive_right| is non-NULL, this class will 27 // take ownership of the port and it will be used to receive messages. 28 // Otherwise the server will create a new receive right. 29 // The maximum size of messages is specified by |buffer_size|. 30 MachMessageServer(MessageDemuxer* demuxer, 31 mach_port_t server_receive_right, 32 mach_msg_size_t buffer_size); 33 virtual ~MachMessageServer(); 34 35 // MessageServer: 36 virtual bool Initialize() OVERRIDE; 37 virtual pid_t GetMessageSenderPID(IPCMessage request) OVERRIDE; 38 virtual IPCMessage CreateReply(IPCMessage request) OVERRIDE; 39 virtual bool SendReply(IPCMessage reply) OVERRIDE; 40 virtual void ForwardMessage(IPCMessage request, 41 mach_port_t destination) OVERRIDE; 42 // Replies to the message with the specified |error_code| as a MIG 43 // error_reply RetCode. 44 virtual void RejectMessage(IPCMessage request, int error_code) OVERRIDE; 45 virtual mach_port_t GetServerPort() const OVERRIDE; 46 47 private: 48 // Event handler for the |server_source_| that reads a message from the queue 49 // and processes it. 50 void ReceiveMessage(); 51 52 // The demuxer delegate. Weak. 53 MessageDemuxer* demuxer_; 54 55 // The Mach port on which the server is receiving requests. 56 base::mac::ScopedMachReceiveRight server_port_; 57 58 // The size of the two message buffers below. 59 const mach_msg_size_t buffer_size_; 60 61 // Request and reply buffers used in ReceiveMessage. 62 base::mac::ScopedMachVM request_buffer_; 63 base::mac::ScopedMachVM reply_buffer_; 64 65 // MACH_RECV dispatch source that handles the |server_port_|. 66 scoped_ptr<DispatchSourceMach> dispatch_source_; 67 68 // Whether or not ForwardMessage() was called during ReceiveMessage(). 69 bool did_forward_message_; 70 }; 71 72 } // namespace sandbox 73 74 #endif // SANDBOX_MAC_MACH_MESSAGE_SERVER_H_ 75