Mac: Proof-of-concept task manager

chrome/browser/child_process_launcher.cc

 // Copyright (c) 2009 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/browser/child_process_launcher.h"
 
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "base/scoped_ptr.h"
 #include "base/thread.h"
 #include "chrome/browser/chrome_thread.h"
 #include "chrome/common/chrome_descriptors.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/process_watcher.h"
 #include "chrome/common/result_codes.h"
 
 #if defined(OS_WIN)
 #include "chrome/browser/sandbox_policy.h"
 #elif defined(OS_LINUX)
 #include "base/singleton.h"
 #include "chrome/browser/crash_handler_host_linux.h"
 #include "chrome/browser/zygote_host_linux.h"
 #include "chrome/browser/renderer_host/render_sandbox_host_linux.h"
 #endif
 
 #if defined(OS_POSIX)
 #include "base/global_descriptors_posix.h"
 #endif
 
+#if defined(OS_MACOSX)
+#include "ipc/ipc_switches.h"
+#include "chrome/browser/mach_broker_mac.h"
+#include "chrome/common/mach_ipc_mac.h"
+#endif
+
+#if defined(OS_MACOSX)
+class MachTask : public Task {
+ public:
+  MachTask(std::string channel_name, mach_port_t* task)
+      : task_(task) {
+    // TODO(thakis): Move some place central

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

Wanna do this now?

+    const std::string kMachChannelPrefix = "com.Google.Chrome";
+    std::string channel = kMachChannelPrefix + channel_name;

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

Unless channel_name begins with a dot, this will give us names like
"com.Google.ChromeMyNewAwesomeChannelName".

+
+    MachBroker::instance();  // make sure registrations are registered on the right thread.

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

80

+
+    // This creates our named server port -- needs to happen on the current
+    // thread.
+printf("Creating receive port %s\n", channel.c_str());

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

Debugging turdlet?

(I leave my debugging turdlets flush with the left margin too, to make 'em
easier to spot when I'm done debugging.)

+    port_.reset(new ReceivePort(channel.c_str()));
+  }
+
+  virtual void Run() {
+    // TODO(thakis): Move some place central
+    const int kMachPortMessageID = 57;

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

Magic?

+
+    const int kMachPortMessageReceiveWaitMs = 1000;
+
+    // TODO(thakis): time histogram between creation and port reception?
+    MachReceiveMessage message;
+    kern_return_t result = port_->WaitForMessage(
+        &message, kMachPortMessageReceiveWaitMs);
+    if (result == KERN_SUCCESS) {
+      CHECK(kMachPortMessageID == message.GetMessageID());

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

CHECK_EQ?

+      CHECK(1 == message.GetDescriptorCount());

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

CHECK_EQ?

+      *task_ = message.GetTranslatedPort(0);
+      printf("yay\n");

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

Debugging turdlet?

+    } else {
+      // TODO(thakis): Log somewhere?
+      *task_ = 0;
+      printf("nay\n");

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

LOG?

+    }
+  }
+
+ private:
+  scoped_ptr<ReceivePort> port_;
+  mach_port_t* task_;
+};
+
+class MachTask2 : public Task {

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

Class-level comments would be nice for this and for MachTask, so that a casual
reader can tell the difference between the two classes.

+ public:
+  MachTask2(const mach_port_t& task, base::ProcessHandle pid)  // must be refs, might be 0 at call time and only be set later on the other thread

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

80

+      : task_(task), pid_(pid) {}
+
+  virtual void Run() {
+    if (task_)

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

You need {} here because this thing spills over a bunch of lines.

+      MachBroker::instance()->RegisterPid(
+          pid_,
+          MachBroker::MachInfo().SetTask(task_));
+  }
+ private:
+  const mach_port_t& task_;
+  base::ProcessHandle pid_;
+};
+
+class MachThread : public base::Thread {
+ public:
+  MachThread() : base::Thread("MachThread"), task_(0) {}

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

task_(MACH_PORT_NULL) ?

+
+  void DoIt(const std::string& channel_name) {
+    DCHECK(message_loop());
+    message_loop()->PostTask(FROM_HERE,
+                             new MachTask(channel_name, &task_));
+  }
+
+  void DoIt2(base::ProcessHandle pid) {

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

Oh, so is this just to avoid the ban on overloading?

DoItChannelName and DoItPid would be better names, then.

"Do it.  Do it.  DO IT."

+    DCHECK(message_loop());
+    message_loop()->PostTask(FROM_HERE,
+                             new MachTask2(task_, pid));
+  }
+
+ private:
+  mach_port_t task_;
+};
+#endif
+
 // Having the functionality of ChildProcessLauncher be in an internal
 // ref counted object allows us to automatically terminate the process when the
 // parent class destructs, while still holding on to state that we need.
 class ChildProcessLauncher::Context
     : public base::RefCountedThreadSafe<ChildProcessLauncher::Context> {
  public:
   Context()
       : starting_(true)
 #if defined(OS_LINUX)
         , zygote_(false)
@@ skipping 113 matching lines @@
       }
       if (is_renderer) {
         const int sandbox_fd =
             Singleton<RenderSandboxHostLinux>()->GetRendererSocket();
         fds_to_map.push_back(std::make_pair(
             sandbox_fd,
             kSandboxIPCChannel + base::GlobalDescriptors::kBaseDescriptor));
       }
 #endif  // defined(OS_LINUX)
 
+#if defined(OS_MACOSX)
+      // TODO(thakis): Possibly somewhere else?
+      // (then again, the fds duping stuff is here too, so maybe it's ok)
+      bool is_renderer = cmd_line->GetSwitchValueASCII(switches::kProcessType)
+          == switches::kRendererProcess;
+    bool is_extension = cmd_line->GetSwitchValueASCII(switches::kProcessType) ==

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

Fix indent?

This whole section (253-256) should be broken up somehow to make it a little
more readable.  For example, factor out the GetSwitchValueASCII call?

+        switches::kExtensionProcess;
+
+for (size_t ii = 0; ii < cmd_line->argv().size(); ++ii)

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

Debugging turdlets?

+  fprintf(stderr, "%s ", cmd_line->argv()[ii].c_str());
+fprintf(stderr, "\n");
+fprintf(stderr, "child type: %s\n", cmd_line->GetSwitchValueASCII(switches::kProcessType).c_str());
+      MachThread mach_thread;

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

Fix the indent in here too?

+      if (true || is_renderer || is_extension) {

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

Remove "true" - debugging turdlet?

+        CHECK(mach_thread.Start());
+        mach_thread.DoIt(
+            cmd_line->GetSwitchValueASCII(switches::kProcessChannelID));
+      }
+#endif
+
+
       // Actually launch the app.
       if (!base::LaunchApp(cmd_line->argv(), env, fds_to_map, false, &handle))
         handle = base::kNullProcessHandle;
+
+#if defined(OS_MACOSX)
+      if (true || is_renderer || is_extension) {

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

true

+        // TODO(thakis): Check |handle| first.
+        mach_thread.DoIt2(handle);
+      }
+#endif
     }
 #endif
 
     ChromeThread::PostTask(
         client_thread_id_, FROM_HERE,
         NewRunnableMethod(
             this,
             &ChildProcessLauncher::Context::Notify,
 #if defined(OS_LINUX)
             zygote,
@@ skipping 94 matching lines @@
 
 ChildProcessLauncher::~ChildProcessLauncher() {
   context_->ResetClient();
 }
 
 bool ChildProcessLauncher::IsStarting() {
   return context_->starting_;
 }
 
 base::ProcessHandle ChildProcessLauncher::GetHandle() {
-  DCHECK(!context_->starting_);
+//  DCHECK(!context_->starting_);

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

?

   return context_->process_.handle();
 }
 
 bool ChildProcessLauncher::DidProcessCrash() {
   bool did_crash, child_exited;
   base::ProcessHandle handle = context_->process_.handle();
 #if defined(OS_LINUX)
   if (context_->zygote_) {
     did_crash = Singleton<ZygoteHost>()->DidProcessCrash(handle, &child_exited);
   } else
@@ skipping 12 matching lines @@
   if (child_exited)
     context_->process_.Close();
 
   return did_crash;
 }
 
 void ChildProcessLauncher::SetProcessBackgrounded(bool background) {
   DCHECK(!context_->starting_);
   context_->process_.SetProcessBackgrounded(background);
 }