content: Pass IOSurface references using Mach IPC.

content/browser/mach_broker_mac_unittest.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 "content/browser/mach_broker_mac.h"
 
 #include "base/synchronization/lock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace content {
 
 class MachBrokerTest : public testing::Test {
  public:
   // Helper function to acquire/release locks and call |PlaceholderForPid()|.
-  void AddPlaceholderForPid(base::ProcessHandle pid, int child_process_id) {
+  void AddPlaceholderForPid(base::ProcessHandle pid,
+                            int child_process_id,
+                            bool is_gpu_process) {
     base::AutoLock lock(broker_.GetLock());
-    broker_.AddPlaceholderForPid(pid, child_process_id);
+    broker_.AddPlaceholderForPid(pid, child_process_id, is_gpu_process);
   }
 
   void InvalidateChildProcessId(int child_process_id) {
     broker_.InvalidateChildProcessId(child_process_id);
   }
 
   int GetChildProcessCount(int child_process_id) {
     return broker_.child_process_id_map_.count(child_process_id);
   }
 
   // Helper function to acquire/release locks and call |FinalizePid()|.
   void FinalizePid(base::ProcessHandle pid,
                    mach_port_t task_port) {
     base::AutoLock lock(broker_.GetLock());
     broker_.FinalizePid(pid, task_port);
   }
 
+  // Helper function to acquire/release locks and call |RegisterIOSurface()|.
+  bool RegisterIOSurface(base::ProcessHandle pid,
+                         int io_surface_id,
+                         int client_id,
+                         mach_port_t io_surface_port) {
+    base::AutoLock lock(broker_.GetLock());
+    return broker_.RegisterIOSurface(pid, io_surface_id, client_id,
+                                     io_surface_port);
+  }
+
+  // Helper function to acquire/release locks and call |UnregisterIOSurface()|.
+  void UnregisterIOSurface(base::ProcessHandle pid,
+                           int io_surface_id,
+                           int client_id) {
+    base::AutoLock lock(broker_.GetLock());
+    broker_.UnregisterIOSurface(pid, io_surface_id, client_id);
+  }
+
+  // Helper function to acquire/release locks and call |AcquireIOSurface()|.
+  mach_port_t AcquireIOSurface(base::ProcessHandle pid, int io_surface_id) {
+    base::AutoLock lock(broker_.GetLock());
+    return broker_.AcquireIOSurface(pid, io_surface_id);
+  }
+
  protected:
   MachBroker broker_;
 };
 
 TEST_F(MachBrokerTest, Locks) {
   // Acquire and release the locks.  Nothing bad should happen.
   base::AutoLock lock(broker_.GetLock());
 }
 
 TEST_F(MachBrokerTest, AddPlaceholderAndFinalize) {
   // Add a placeholder for PID 1.
-  AddPlaceholderForPid(1, 1);
+  AddPlaceholderForPid(1, 1, false);
   EXPECT_EQ(0u, broker_.TaskForPid(1));
 
   // Finalize PID 1.
   FinalizePid(1, 100u);
   EXPECT_EQ(100u, broker_.TaskForPid(1));
 
   // Should be no entry for PID 2.
   EXPECT_EQ(0u, broker_.TaskForPid(2));
 }
 
 TEST_F(MachBrokerTest, InvalidateChildProcessId) {
   // Add a placeholder for PID 1 and child process id 50.
-  AddPlaceholderForPid(1, 50);
+  AddPlaceholderForPid(1, 50, false);
   FinalizePid(1, 100u);
 
   EXPECT_EQ(100u, broker_.TaskForPid(1));
   InvalidateChildProcessId(50);
   EXPECT_EQ(0u, broker_.TaskForPid(1));
 }
 
 TEST_F(MachBrokerTest, ValidateChildProcessIdMap) {
   // Add a placeholder for PID 1 and child process id 50.
-  AddPlaceholderForPid(1, 50);
+  AddPlaceholderForPid(1, 50, false);
   FinalizePid(1, 100u);
 
   EXPECT_EQ(1, GetChildProcessCount(50));
   InvalidateChildProcessId(50);
   EXPECT_EQ(0, GetChildProcessCount(50));
 }
 
 TEST_F(MachBrokerTest, FinalizeUnknownPid) {
   // Finalizing an entry for an unknown pid should not add it to the map.
   FinalizePid(1u, 100u);
   EXPECT_EQ(0u, broker_.TaskForPid(1u));
 }
 
+TEST_F(MachBrokerTest, IOSurfaces) {
+  const int kIOSurfaceId = 0;
+  // Add a placeholder and finalize PID 1 (GPU process).
+  AddPlaceholderForPid(1, 1, true);
+  FinalizePid(1, 100u);
+
+  // Add a placeholder and finalize PID 2.
+  AddPlaceholderForPid(2, 2, false);
+  FinalizePid(2, 101u);
+
+  // Add a placeholder and finalize PID 3.
+  AddPlaceholderForPid(3, 3, false);
+  FinalizePid(3, 103u);
+
+  // Attempt to register an IOSurface using PID 2. Should fail as only GPU
+  // process is allowed to register IOSurfaces.
+  bool rv = RegisterIOSurface(2, kIOSurfaceId, 2, 200u);
+  EXPECT_FALSE(rv);
+  EXPECT_EQ(0u, AcquireIOSurface(2, kIOSurfaceId));
+
+  // Register an IOSurface using PID 1 for usage by child process id 2.
+  rv = RegisterIOSurface(1, kIOSurfaceId, 2, 200u);
+  EXPECT_TRUE(rv);
+
+  // PID 3 should not be able to acquire an IOSurface reference.
+  EXPECT_EQ(0u, AcquireIOSurface(3, kIOSurfaceId));
+
+  // PID 2 should be able to acquire an IOSurface reference.
+  EXPECT_EQ(200u, AcquireIOSurface(2, kIOSurfaceId));
+
+  // PID 2/3 should not be able to unregister IOSurface.
+  UnregisterIOSurface(2, kIOSurfaceId, 2);
+  UnregisterIOSurface(3, kIOSurfaceId, 2);
+  EXPECT_EQ(200u, AcquireIOSurface(2, kIOSurfaceId));
+
+  // GPU process can unregister IOSurface.
+  UnregisterIOSurface(1, kIOSurfaceId, 2);
+  EXPECT_EQ(0u, AcquireIOSurface(2, kIOSurfaceId));
+}
+
 }  // namespace content