blob: 70f4460505470bb2650e5461c0bf3a46fbdd23d6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
// 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 "chrome/common/extensions/image_writer/image_writer_util_mac.h"
#include <CoreFoundation/CoreFoundation.h>
#include "base/mac/scoped_cftyperef.h"
#include "base/mac/scoped_ioobject.h"
namespace extensions {
bool IsUsbDevice(io_object_t disk_obj) {
io_object_t current_obj = disk_obj;
io_object_t parent_obj = 0;
// Keep scoped object outside the loop so the object lives to the next
// GetParentEntry.
base::mac::ScopedIOObject<io_object_t> parent_obj_ref(parent_obj);
while ((IORegistryEntryGetParentEntry(
current_obj, kIOServicePlane, &parent_obj)) == KERN_SUCCESS) {
current_obj = parent_obj;
parent_obj_ref.reset(parent_obj);
base::ScopedCFTypeRef<CFStringRef> class_name(
IOObjectCopyClass(current_obj));
if (!class_name) {
LOG(ERROR) << "Could not get object class of IO Registry Entry.";
continue;
}
if (CFStringCompare(class_name.get(), CFSTR("IOUSBMassStorageClass"), 0) ==
kCFCompareEqualTo) {
return true;
}
}
return false;
}
} // namespace extensions
|
