[4/4][Ozone-Drm] Keep track of DRM devices in browser process

ui/ozone/platform/drm/host/drm_native_display_delegate.cc

 // Copyright 2014 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 "ui/ozone/platform/drm/host/drm_native_display_delegate.h"
 
 #include <stdio.h>
-#include <xf86drm.h>
 
 #include "base/logging.h"
 #include "base/thread_task_runner_handle.h"
 #include "base/threading/thread_restrictions.h"
 #include "ui/display/types/display_snapshot.h"
 #include "ui/display/types/native_display_observer.h"
 #include "ui/events/ozone/device/device_event.h"
 #include "ui/events/ozone/device/device_manager.h"
 #include "ui/ozone/common/display_snapshot_proxy.h"
 #include "ui/ozone/common/display_util.h"
 #include "ui/ozone/common/gpu/ozone_gpu_messages.h"
 #include "ui/ozone/platform/drm/host/display_manager.h"
+#include "ui/ozone/platform/drm/host/drm_device_handle.h"
 #include "ui/ozone/platform/drm/host/drm_gpu_platform_support_host.h"
 
 namespace ui {
 
 namespace {
 
-typedef base::Callback<void(const base::FilePath&, base::File)>
-    OnOpenDeviceReplyCallback;
+typedef base::Callback<void(const base::FilePath&)> OnOpenDeviceReplyCallback;
 
 const char* kDisplayActionString[] = {
     "ADD",
     "REMOVE",
     "CHANGE",
 };
 
-bool Authenticate(int fd) {
-  drm_magic_t magic = 0;
-  // We need to make sure the DRM device has enough privilege. Use the DRM
-  // authentication logic to figure out if the device has enough permissions.
-  return !drmGetMagic(fd, &magic) && !drmAuthMagic(fd, magic);
-}
-
-base::File OpenDrmDevice(const base::FilePath& path) {
-  base::File file;
-  bool print_warning = true;
-  while (true) {
-    file = base::File(path, base::File::FLAG_OPEN | base::File::FLAG_READ |
-                                base::File::FLAG_WRITE);
-
-    base::File::Info info;
-    file.GetInfo(&info);
-
-    CHECK(!info.is_directory);
-    CHECK(path.DirName() == base::FilePath("/dev/dri"));
-
-    if (!file.IsValid()) {
-      LOG(ERROR) << "Failed to open " << path.value() << ": "
-                 << base::File::ErrorToString(file.error_details());
-      return file.Pass();
-    }
-
-    if (Authenticate(file.GetPlatformFile()))
-      break;
-
-    LOG_IF(WARNING, print_warning) << "Failed to authenticate " << path.value();
-
-    print_warning = false;
-    usleep(100000);
-  }
-
-  VLOG(1) << "Succeeded authenticating " << path.value();
-  return file.Pass();
-}
-
 void OpenDeviceOnWorkerThread(
     const base::FilePath& path,
+    DrmDeviceHandle* handle,
     const scoped_refptr<base::TaskRunner>& reply_runner,
     const OnOpenDeviceReplyCallback& callback) {
-  base::File file = OpenDrmDevice(path);
-  reply_runner->PostTask(FROM_HERE,
-                         base::Bind(callback, path, base::Passed(file.Pass())));
+  if (handle->Initialize(path))
+    reply_runner->PostTask(FROM_HERE, base::Bind(callback, path));
 }
 
 void UpdateDeviceOnWorkerThread(
     const scoped_refptr<base::TaskRunner>& reply_runner,
     const base::Closure& callback) {
   // Just reply since there isn't a need for extra operations.
   reply_runner->PostTask(FROM_HERE, callback);
 }
 
+void CloseDeviceOnWorkerThread(
+    scoped_ptr<DrmDeviceHandle> handle,
+    const scoped_refptr<base::TaskRunner>& reply_runner,
+    const base::Closure& callback) {
+  handle->Shutdown();
+  reply_runner->PostTask(FROM_HERE, callback);
+}
+
 class DrmDisplaySnapshotProxy : public DisplaySnapshotProxy {
  public:
   DrmDisplaySnapshotProxy(const DisplaySnapshot_Params& params,
                           DisplayManager* display_manager)
       : DisplaySnapshotProxy(params), display_manager_(display_manager) {
     display_manager_->RegisterDisplay(this);
   }
 
   ~DrmDisplaySnapshotProxy() override {
     display_manager_->UnregisterDisplay(this);
@@ skipping 12 matching lines @@
     DeviceManager* device_manager,
     DisplayManager* display_manager,
     const base::FilePath& primary_graphics_card_path)
     : proxy_(proxy),
       device_manager_(device_manager),
       display_manager_(display_manager),
       primary_graphics_card_path_(primary_graphics_card_path),
       has_dummy_display_(false),
       weak_ptr_factory_(this) {
   proxy_->RegisterHandler(this);
-  drm_devices_.insert(primary_graphics_card_path);
 }
 
 DrmNativeDisplayDelegate::~DrmNativeDisplayDelegate() {
   device_manager_->RemoveObserver(this);
   proxy_->UnregisterHandler(this);
 }
 
 void DrmNativeDisplayDelegate::Initialize() {
   device_manager_->AddObserver(this);
   device_manager_->ScanDevices(this);
@@ skipping 128 matching lines @@
 
 void DrmNativeDisplayDelegate::OnDeviceEvent(const DeviceEvent& event) {
   if (event.device_type() != DeviceEvent::DISPLAY)
     return;
 
   VLOG(1) << "Got display event " << kDisplayActionString[event.action_type()]
           << " for " << event.path().value();
   switch (event.action_type()) {
     case DeviceEvent::ADD:
       if (drm_devices_.find(event.path()) == drm_devices_.end()) {
-        drm_devices_.insert(event.path());
+        scoped_ptr<DrmDeviceHandle> handle(new DrmDeviceHandle());
         scheduler_.PostNamedSequencedWorkerTask(
             event.path().value(), FROM_HERE,
             base::Bind(
-                &OpenDeviceOnWorkerThread, event.path(),
+                &OpenDeviceOnWorkerThread, event.path(), handle.get(),
                 base::ThreadTaskRunnerHandle::Get(),
                 base::Bind(&DrmNativeDisplayDelegate::OnAddGraphicsDevice,
                            weak_ptr_factory_.GetWeakPtr())));
+        drm_devices_.add(event.path(), handle.Pass());

[dnicoara, 2015/04/27 16:46:55]

Hmm, this will be problematic since the handle may be accessed by
OnChannelEstablished() while the worker thread is opening it.

       }
       return;
     case DeviceEvent::CHANGE:
       scheduler_.PostNamedSequencedWorkerTask(
           event.path().value(), FROM_HERE,
           base::Bind(
               &UpdateDeviceOnWorkerThread, base::ThreadTaskRunnerHandle::Get(),
               base::Bind(&DrmNativeDisplayDelegate::OnUpdateGraphicsDevice,
                          weak_ptr_factory_.GetWeakPtr())));
       return;
     case DeviceEvent::REMOVE:
       DCHECK(event.path() != primary_graphics_card_path_)
           << "Removing primary graphics card";
       auto it = drm_devices_.find(event.path());
       if (it != drm_devices_.end()) {
-        drm_devices_.erase(it);
         scheduler_.PostNamedSequencedWorkerTask(
             event.path().value(), FROM_HERE,
             base::Bind(
-                &UpdateDeviceOnWorkerThread,
+                &CloseDeviceOnWorkerThread,
+                base::Passed(drm_devices_.take_and_erase(it)),
                 base::ThreadTaskRunnerHandle::Get(),
                 base::Bind(&DrmNativeDisplayDelegate::OnRemoveGraphicsDevice,
                            weak_ptr_factory_.GetWeakPtr(), event.path())));
       }
       return;
   }
 }
 
-void DrmNativeDisplayDelegate::OnAddGraphicsDevice(const base::FilePath& path,
-                                                   base::File file) {
-  if (!file.IsValid())
+void DrmNativeDisplayDelegate::OnAddGraphicsDevice(const base::FilePath& path) {
+  auto it = drm_devices_.find(path);
+  if (it == drm_devices_.end())
     return;
 
   proxy_->Send(new OzoneGpuMsg_AddGraphicsDevice(
-      path, base::FileDescriptor(file.Pass())));
+      path, base::FileDescriptor(it->second->DuplicateFile())));
   FOR_EACH_OBSERVER(NativeDisplayObserver, observers_,
                     OnConfigurationChanged());
 }
 
 void DrmNativeDisplayDelegate::OnUpdateGraphicsDevice() {
   FOR_EACH_OBSERVER(NativeDisplayObserver, observers_,
                     OnConfigurationChanged());
 }
 
 void DrmNativeDisplayDelegate::OnRemoveGraphicsDevice(
     const base::FilePath& path) {
   proxy_->Send(new OzoneGpuMsg_RemoveGraphicsDevice(path));
   FOR_EACH_OBSERVER(NativeDisplayObserver, observers_,
                     OnConfigurationChanged());
 }
 
 void DrmNativeDisplayDelegate::OnChannelEstablished(
     int host_id,
     scoped_refptr<base::SingleThreadTaskRunner> send_runner,
     const base::Callback<void(IPC::Message*)>& send_callback) {
-  drm_devices_.clear();
-  drm_devices_.insert(primary_graphics_card_path_);
-  {
+  if (drm_devices_.find(primary_graphics_card_path_) == drm_devices_.end()) {
     // First device needs to be treated specially. We need to open this
     // synchronously since the GPU process will need it to initialize the
     // graphics state.
     base::ThreadRestrictions::ScopedAllowIO allow_io;
-    base::File file = OpenDrmDevice(primary_graphics_card_path_);
-    if (!file.IsValid()) {
+    scoped_ptr<DrmDeviceHandle> handle(new DrmDeviceHandle());
+    if (!handle->Initialize(primary_graphics_card_path_)) {
       LOG(FATAL) << "Failed to open primary graphics card";
       return;
     }
-    OnAddGraphicsDevice(primary_graphics_card_path_, file.Pass());
+    drm_devices_.add(primary_graphics_card_path_, handle.Pass());
+    OnAddGraphicsDevice(primary_graphics_card_path_);
   }
 
+  for (auto pair : drm_devices_)
+    OnAddGraphicsDevice(pair.first);
+
   device_manager_->ScanDevices(this);
   FOR_EACH_OBSERVER(NativeDisplayObserver, observers_,
                     OnConfigurationChanged());
 }
 
 void DrmNativeDisplayDelegate::OnChannelDestroyed(int host_id) {
   // If the channel got destroyed in the middle of a configuration then just
   // respond with failure.
   if (!get_displays_callback_.is_null()) {
     base::ThreadTaskRunnerHandle::Get()->PostTask(
@@ skipping 48 matching lines @@
     configure_callback_map_.erase(it);
   }
 }
 
 void DrmNativeDisplayDelegate::RunUpdateDisplaysCallback(
     const GetDisplaysCallback& callback) const {
   callback.Run(displays_.get());
 }
 
 }  // namespace ui