device/hid/hid_service_linux.cc

   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 #include "device/hid/hid_service_linux.h"
   6
   7 #include <fcntl.h>
   8 #include <limits>
   9 #include <string>
  10
  11 #include "base/bind.h"
  12 #include "base/files/file.h"
  13 #include "base/files/file_path.h"
  14 #include "base/files/file_util.h"
  15 #include "base/location.h"
  16 #include "base/scoped_observer.h"
  17 #include "base/strings/string_number_conversions.h"
  18 #include "base/strings/string_split.h"
  19 #include "base/thread_task_runner_handle.h"
  20 #include "base/threading/thread_restrictions.h"
  21 #include "components/device_event_log/device_event_log.h"
  22 #include "device/hid/device_monitor_linux.h"
  23 #include "device/hid/hid_connection_linux.h"
  24 #include "device/hid/hid_device_info_linux.h"
  25 #include "device/udev_linux/scoped_udev.h"
  26 #include "net/base/net_util.h"
  27
  28 #if defined(OS_CHROMEOS)
  29 #include "base/sys_info.h"
  30 #include "chromeos/dbus/dbus_thread_manager.h"
  31 #include "chromeos/dbus/permission_broker_client.h"
  32 #endif  // defined(OS_CHROMEOS)
  33
  34 namespace device {
  35
  36 namespace {
  37
  38 const char kHidrawSubsystem[] = "hidraw";
  39 const char kHIDID[] = "HID_ID";
  40 const char kHIDName[] = "HID_NAME";
  41 const char kHIDUnique[] = "HID_UNIQ";
  42 const char kSysfsReportDescriptorKey[] = "report_descriptor";
  43
  44 }  // namespace
  45
  46 struct HidServiceLinux::ConnectParams {
  47   ConnectParams(scoped_refptr<HidDeviceInfoLinux> device_info,
  48                 const ConnectCallback& callback,
  49                 scoped_refptr<base::SingleThreadTaskRunner> task_runner,
  50                 scoped_refptr<base::SingleThreadTaskRunner> file_task_runner)
  51       : device_info(device_info),
  52         callback(callback),
  53         task_runner(task_runner),
  54         file_task_runner(file_task_runner) {}
  55   ~ConnectParams() {}
  56
  57   scoped_refptr<HidDeviceInfoLinux> device_info;
  58   ConnectCallback callback;
  59   scoped_refptr<base::SingleThreadTaskRunner> task_runner;
  60   scoped_refptr<base::SingleThreadTaskRunner> file_task_runner;
  61   base::File device_file;
  62 };
  63
  64 class HidServiceLinux::FileThreadHelper
  65     : public DeviceMonitorLinux::Observer,
  66       public base::MessageLoop::DestructionObserver {
  67  public:
  68   FileThreadHelper(base::WeakPtr<HidServiceLinux> service,
  69                    scoped_refptr<base::SingleThreadTaskRunner> task_runner)
  70       : observer_(this), service_(service), task_runner_(task_runner) {}
  71
  72   ~FileThreadHelper() override {
  73     DCHECK(thread_checker_.CalledOnValidThread());
  74     base::MessageLoop::current()->RemoveDestructionObserver(this);
  75   }
  76
  77   static void Start(scoped_ptr<FileThreadHelper> self) {
  78     base::ThreadRestrictions::AssertIOAllowed();
  79     self->thread_checker_.DetachFromThread();
  80
  81     DeviceMonitorLinux* monitor = DeviceMonitorLinux::GetInstance();
  82     self->observer_.Add(monitor);
  83     monitor->Enumerate(base::Bind(&FileThreadHelper::OnDeviceAdded,
  84                                   base::Unretained(self.get())));
  85     self->task_runner_->PostTask(
  86         FROM_HERE,
  87         base::Bind(&HidServiceLinux::FirstEnumerationComplete, self->service_));
  88
  89     // |self| is now owned by the current message loop.
  90     base::MessageLoop::current()->AddDestructionObserver(self.release());
  91   }
  92
  93  private:
  94   // DeviceMonitorLinux::Observer:
  95   void OnDeviceAdded(udev_device* device) override {
  96     DCHECK(thread_checker_.CalledOnValidThread());
  97     const char* device_path = udev_device_get_syspath(device);
  98     if (!device_path) {
  99       return;
 100     }
 101     HidDeviceId device_id = device_path;
 102
 103     const char* subsystem = udev_device_get_subsystem(device);
 104     if (!subsystem || strcmp(subsystem, kHidrawSubsystem) != 0) {
 105       return;
 106     }
 107
 108     const char* str_property = udev_device_get_devnode(device);
 109     if (!str_property) {
 110       return;
 111     }
 112     std::string device_node = str_property;
 113
 114     udev_device* parent = udev_device_get_parent(device);
 115     if (!parent) {
 116       return;
 117     }
 118
 119     const char* hid_id = udev_device_get_property_value(parent, kHIDID);
 120     if (!hid_id) {
 121       return;
 122     }
 123
 124     std::vector<std::string> parts = base::SplitString(
 125         hid_id, ":", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL);
 126     if (parts.size() != 3) {
 127       return;
 128     }
 129
 130     uint32_t int_property = 0;
 131     if (!HexStringToUInt(base::StringPiece(parts[1]), &int_property) ||
 132         int_property > std::numeric_limits<uint16_t>::max()) {
 133       return;
 134     }
 135     uint16_t vendor_id = int_property;
 136
 137     if (!HexStringToUInt(base::StringPiece(parts[2]), &int_property) ||
 138         int_property > std::numeric_limits<uint16_t>::max()) {
 139       return;
 140     }
 141     uint16_t product_id = int_property;
 142
 143     std::string serial_number;
 144     str_property = udev_device_get_property_value(parent, kHIDUnique);
 145     if (str_property != NULL) {
 146       serial_number = str_property;
 147     }
 148
 149     std::string product_name;
 150     str_property = udev_device_get_property_value(parent, kHIDName);
 151     if (str_property != NULL) {
 152       product_name = str_property;
 153     }
 154
 155     const char* parent_sysfs_path = udev_device_get_syspath(parent);
 156     if (!parent_sysfs_path) {
 157       return;
 158     }
 159     base::FilePath report_descriptor_path =
 160         base::FilePath(parent_sysfs_path).Append(kSysfsReportDescriptorKey);
 161     std::string report_descriptor_str;
 162     if (!base::ReadFileToString(report_descriptor_path,
 163                                 &report_descriptor_str)) {
 164       return;
 165     }
 166
 167     scoped_refptr<HidDeviceInfo> device_info(new HidDeviceInfoLinux(
 168         device_id, device_node, vendor_id, product_id, product_name,
 169         serial_number,
 170         kHIDBusTypeUSB,  // TODO(reillyg): Detect Bluetooth. crbug.com/443335
 171         std::vector<uint8>(report_descriptor_str.begin(),
 172                            report_descriptor_str.end())));
 173
 174     task_runner_->PostTask(FROM_HERE, base::Bind(&HidServiceLinux::AddDevice,
 175                                                  service_, device_info));
 176   }
 177
 178   void OnDeviceRemoved(udev_device* device) override {
 179     DCHECK(thread_checker_.CalledOnValidThread());
 180     const char* device_path = udev_device_get_syspath(device);
 181     if (device_path) {
 182       task_runner_->PostTask(
 183           FROM_HERE, base::Bind(&HidServiceLinux::RemoveDevice, service_,
 184                                 std::string(device_path)));
 185     }
 186   }
 187
 188   // base::MessageLoop::DestructionObserver:
 189   void WillDestroyCurrentMessageLoop() override {
 190     DCHECK(thread_checker_.CalledOnValidThread());
 191     delete this;
 192   }
 193
 194   base::ThreadChecker thread_checker_;
 195   ScopedObserver<DeviceMonitorLinux, DeviceMonitorLinux::Observer> observer_;
 196
 197   // This weak pointer is only valid when checked on this task runner.
 198   base::WeakPtr<HidServiceLinux> service_;
 199   scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
 200
 201   DISALLOW_COPY_AND_ASSIGN(FileThreadHelper);
 202 };
 203
 204 HidServiceLinux::HidServiceLinux(
 205     scoped_refptr<base::SingleThreadTaskRunner> file_task_runner)
 206     : file_task_runner_(file_task_runner), weak_factory_(this) {
 207   task_runner_ = base::ThreadTaskRunnerHandle::Get();
 208   scoped_ptr<FileThreadHelper> helper(
 209       new FileThreadHelper(weak_factory_.GetWeakPtr(), task_runner_));
 210   helper_ = helper.get();
 211   file_task_runner_->PostTask(
 212       FROM_HERE, base::Bind(&FileThreadHelper::Start, base::Passed(&helper)));
 213 }
 214
 215 void HidServiceLinux::Connect(const HidDeviceId& device_id,
 216                               const ConnectCallback& callback) {
 217   DCHECK(thread_checker_.CalledOnValidThread());
 218
 219   const auto& map_entry = devices().find(device_id);
 220   if (map_entry == devices().end()) {
 221     task_runner_->PostTask(FROM_HERE, base::Bind(callback, nullptr));
 222     return;
 223   }
 224   scoped_refptr<HidDeviceInfoLinux> device_info =
 225       static_cast<HidDeviceInfoLinux*>(map_entry->second.get());
 226
 227   scoped_ptr<ConnectParams> params(new ConnectParams(
 228       device_info, callback, task_runner_, file_task_runner_));
 229
 230 #if defined(OS_CHROMEOS)
 231   chromeos::PermissionBrokerClient* client =
 232       chromeos::DBusThreadManager::Get()->GetPermissionBrokerClient();
 233   DCHECK(client) << "Could not get permission broker client.";
 234   client->OpenPath(
 235       device_info->device_node(),
 236       base::Bind(&HidServiceLinux::OnPathOpened, base::Passed(&params)));
 237 #else
 238   file_task_runner_->PostTask(FROM_HERE,
 239                               base::Bind(&HidServiceLinux::OpenOnBlockingThread,
 240                                          base::Passed(&params)));
 241 #endif  // defined(OS_CHROMEOS)
 242 }
 243
 244 HidServiceLinux::~HidServiceLinux() {
 245   file_task_runner_->DeleteSoon(FROM_HERE, helper_);
 246 }
 247
 248 #if defined(OS_CHROMEOS)
 249
 250 // static
 251 void HidServiceLinux::OnPathOpened(scoped_ptr<ConnectParams> params,
 252                                    dbus::FileDescriptor fd) {
 253   scoped_refptr<base::SingleThreadTaskRunner> file_task_runner =
 254       params->file_task_runner;
 255   file_task_runner->PostTask(
 256       FROM_HERE, base::Bind(&HidServiceLinux::ValidateFdOnBlockingThread,
 257                             base::Passed(&params), base::Passed(&fd)));
 258 }
 259
 260 // static
 261 void HidServiceLinux::ValidateFdOnBlockingThread(
 262     scoped_ptr<ConnectParams> params,
 263     dbus::FileDescriptor fd) {
 264   base::ThreadRestrictions::AssertIOAllowed();
 265
 266   fd.CheckValidity();
 267   if (fd.is_valid()) {
 268     params->device_file = base::File(fd.TakeValue());
 269     FinishOpen(params.Pass());
 270   } else {
 271     HID_LOG(EVENT) << "Permission broker denied access to '"
 272                    << params->device_info->device_node() << "'.";
 273     params->task_runner->PostTask(FROM_HERE,
 274                                   base::Bind(params->callback, nullptr));
 275   }
 276 }
 277
 278 #else
 279
 280 // static
 281 void HidServiceLinux::OpenOnBlockingThread(scoped_ptr<ConnectParams> params) {
 282   base::ThreadRestrictions::AssertIOAllowed();
 283   scoped_refptr<base::SingleThreadTaskRunner> task_runner = params->task_runner;
 284
 285   base::FilePath device_path(params->device_info->device_node());
 286   base::File& device_file = params->device_file;
 287   int flags =
 288       base::File::FLAG_OPEN | base::File::FLAG_READ | base::File::FLAG_WRITE;
 289   device_file.Initialize(device_path, flags);
 290   if (!device_file.IsValid()) {
 291     base::File::Error file_error = device_file.error_details();
 292
 293     if (file_error == base::File::FILE_ERROR_ACCESS_DENIED) {
 294       HID_LOG(EVENT)
 295           << "Access denied opening device read-write, trying read-only.";
 296       flags = base::File::FLAG_OPEN | base::File::FLAG_READ;
 297       device_file.Initialize(device_path, flags);
 298     }
 299   }
 300   if (!device_file.IsValid()) {
 301     HID_LOG(EVENT) << "Failed to open '" << params->device_info->device_node()
 302                    << "': "
 303                    << base::File::ErrorToString(device_file.error_details());
 304     task_runner->PostTask(FROM_HERE, base::Bind(params->callback, nullptr));
 305     return;
 306   }
 307
 308   FinishOpen(params.Pass());
 309 }
 310
 311 #endif  // defined(OS_CHROMEOS)
 312
 313 // static
 314 void HidServiceLinux::FinishOpen(scoped_ptr<ConnectParams> params) {
 315   base::ThreadRestrictions::AssertIOAllowed();
 316   scoped_refptr<base::SingleThreadTaskRunner> task_runner = params->task_runner;
 317
 318   int result = net::SetNonBlocking(params->device_file.GetPlatformFile());
 319   if (result == -1) {
 320     HID_PLOG(ERROR) << "Failed to set the non-blocking flag on the device fd";
 321     task_runner->PostTask(FROM_HERE, base::Bind(params->callback, nullptr));
 322     return;
 323   }
 324
 325   task_runner->PostTask(
 326       FROM_HERE,
 327       base::Bind(&HidServiceLinux::CreateConnection, base::Passed(&params)));
 328 }
 329
 330 // static
 331 void HidServiceLinux::CreateConnection(scoped_ptr<ConnectParams> params) {
 332   DCHECK(params->device_file.IsValid());
 333   params->callback.Run(make_scoped_refptr(
 334       new HidConnectionLinux(params->device_info, params->device_file.Pass(),
 335                              params->file_task_runner)));
 336 }
 337
 338 }  // namespace device