Mac: Proof-of-concept task manager

chrome/common/mach_ipc_mac.mm

 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/common/mach_ipc_mac.h"
 
 #import <Foundation/Foundation.h>
 
 #include <stdio.h>
 #include "base/logging.h"
@@ skipping 167 matching lines @@
 
 //==============================================================================
 // create a new mach port for receiving messages and register a name for it
 ReceivePort::ReceivePort(const char *receive_port_name) {
   mach_port_t current_task = mach_task_self();
 
   init_result_ = mach_port_allocate(current_task,
                                     MACH_PORT_RIGHT_RECEIVE,
                                     &port_);
 
-  if (init_result_ != KERN_SUCCESS)
+  if (init_result_ != KERN_SUCCESS) {
+fprintf(stderr, "failed to allocate send port\n");

[Mark Mentovai, 2010/01/11 20:22:46]

What's this decoration about?

     return;
+  }
 
   init_result_ = mach_port_insert_right(current_task,
                                         port_,
                                         port_,
                                         MACH_MSG_TYPE_MAKE_SEND);
 
-  if (init_result_ != KERN_SUCCESS)
+  if (init_result_ != KERN_SUCCESS) {
+fprintf(stderr, "failed to insert send right\n");

[Mark Mentovai, 2010/01/11 20:22:46]

Same

     return;
+  }
 
   NSPort *ns_port = [NSMachPort portWithMachPort:port_];
   NSString *port_name = [NSString stringWithUTF8String:receive_port_name];
-  [[NSMachBootstrapServer sharedInstance] registerPort:ns_port name:port_name];
+  if (![[NSMachBootstrapServer sharedInstance] registerPort:ns_port name:port_name]) {

[Mark Mentovai, 2010/01/11 20:22:46]

80

+NSLog(@"failed to register ort %@ with name %@", ns_port, port_name);

[Mark Mentovai, 2010/01/11 20:22:46]

NSLog?

And you misspelled port.  :)

+  }
 }
 
 //==============================================================================
 // create a new mach port for receiving messages
 ReceivePort::ReceivePort() {
   mach_port_t current_task = mach_task_self();
 
   init_result_ = mach_port_allocate(current_task,
                                     MACH_PORT_RIGHT_RECEIVE,
                                     &port_);
 
-  if (init_result_ != KERN_SUCCESS)
+  if (init_result_ != KERN_SUCCESS) {
+fprintf(stderr, "failed to allocate send port2\n");

[Mark Mentovai, 2010/01/11 20:22:46]

t’let

     return;
+  }
 
   init_result_ = mach_port_insert_right(current_task,
                                         port_,
                                         port_,
                                         MACH_MSG_TYPE_MAKE_SEND);
 }
 
 //==============================================================================
 // Given an already existing mach port, use it.  We take ownership of the
 // port and deallocate it in our destructor.
 ReceivePort::ReceivePort(mach_port_t receive_port)
   : port_(receive_port),
     init_result_(KERN_SUCCESS) {
 }
 
 //==============================================================================
 ReceivePort::~ReceivePort() {
   if (init_result_ == KERN_SUCCESS)
     mach_port_deallocate(mach_task_self(), port_);
 }
 
 //==============================================================================
 kern_return_t ReceivePort::WaitForMessage(MachReceiveMessage *out_message,
                                           mach_msg_timeout_t timeout) {
   if (!out_message) {
     return KERN_INVALID_ARGUMENT;
   }
 
   // return any error condition encountered in constructor
-  if (init_result_ != KERN_SUCCESS)
+  if (init_result_ != KERN_SUCCESS) {
+fprintf(stderr, "failed to insert send right2\n");

[Mark Mentovai, 2010/01/11 20:22:46]

…

     return init_result_;
+  }
 
   out_message->Head()->msgh_bits = 0;
   out_message->Head()->msgh_local_port = port_;
   out_message->Head()->msgh_remote_port = MACH_PORT_NULL;
   out_message->Head()->msgh_reserved = 0;
   out_message->Head()->msgh_id = 0;
 
   kern_return_t result = mach_msg(out_message->Head(),
                                   MACH_RCV_MSG | MACH_RCV_TIMEOUT,
                                   0,
                                   out_message->MaxSize(),
                                   port_,
                                   timeout,              // timeout in ms
                                   MACH_PORT_NULL);
 
   return result;
 }
 
 #pragma mark -
 
 //==============================================================================
 // get a port with send rights corresponding to a named registered service
 MachPortSender::MachPortSender(const char *receive_port_name) {
   mach_port_t bootstrap_port = 0;
   init_result_ = task_get_bootstrap_port(mach_task_self(), &bootstrap_port);
 
-  if (init_result_ != KERN_SUCCESS)
+  if (init_result_ != KERN_SUCCESS) {
+fprintf(stderr, "receiving bootstrap port failed\n");

[Mark Mentovai, 2010/01/11 20:22:46]

…

     return;
+  }
 
   init_result_ = bootstrap_look_up(bootstrap_port,
                     const_cast<char*>(receive_port_name),
                     &send_port_);
 }
 
 //==============================================================================
 MachPortSender::MachPortSender(mach_port_t send_port)
   : send_port_(send_port),
     init_result_(KERN_SUCCESS) {
 }
 
 //==============================================================================
 kern_return_t MachPortSender::SendMessage(MachSendMessage &message,
                                           mach_msg_timeout_t timeout) {
   if (message.Head()->msgh_size == 0) {
     NOTREACHED();
     return KERN_INVALID_VALUE;    // just for safety -- never should occur
   };
 
-  if (init_result_ != KERN_SUCCESS)
+  if (init_result_ != KERN_SUCCESS) {
+fprintf(stderr, "bootstrap lookup failed\n");

[Mark Mentovai, 2010/01/11 20:22:46]

…

     return init_result_;
+  }
 
   message.Head()->msgh_remote_port = send_port_;
 
   kern_return_t result = mach_msg(message.Head(),
                                   MACH_SEND_MSG | MACH_SEND_TIMEOUT,
                                   message.Head()->msgh_size,
                                   0,
                                   MACH_PORT_NULL,
                                   timeout,              // timeout in ms
                                   MACH_PORT_NULL);
 
   return result;
 }