Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(775)

Side by Side Diff: chrome/browser/metrics/drive_metrics_provider_win.cc

Issue 1254983003: Componentize DriveMetricsProvider* (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Response to review Created 5 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « chrome/browser/metrics/drive_metrics_provider_unittest.cc ('k') | chrome/chrome_browser.gypi » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 // Copyright 2015 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 "chrome/browser/metrics/drive_metrics_provider.h"
6
7 #include <windows.h>
8 #include <ntddscsi.h>
9 #include <winioctl.h>
10 #include <vector>
11
12 #include "base/files/file.h"
13 #include "base/files/file_path.h"
14 #include "base/macros.h"
15 #include "base/strings/stringprintf.h"
16 #include "base/win/windows_version.h"
17
18 namespace {
19
20 // Semi-copy of similarly named struct from ata.h in WinDDK.
21 struct IDENTIFY_DEVICE_DATA {
22 USHORT UnusedWords[217];
23 USHORT NominalMediaRotationRate;
24 USHORT MoreUnusedWords[38];
25 };
26 COMPILE_ASSERT(sizeof(IDENTIFY_DEVICE_DATA) == 512, IdentifyDeviceDataSize);
27
28 struct AtaRequest {
29 ATA_PASS_THROUGH_EX query;
30 IDENTIFY_DEVICE_DATA result;
31 };
32
33 } // namespace
34
35 // static
36 bool DriveMetricsProvider::HasSeekPenalty(const base::FilePath& path,
37 bool* has_seek_penalty) {
38 std::vector<base::FilePath::StringType> components;
39 path.GetComponents(&components);
40
41 int flags = base::File::FLAG_OPEN;
42 bool win7_or_higher = base::win::GetVersion() >= base::win::VERSION_WIN7;
43 if (!win7_or_higher)
44 flags |= base::File::FLAG_READ | base::File::FLAG_WRITE;
45
46 base::File volume(base::FilePath(L"\\\\.\\" + components[0]), flags);
47 if (!volume.IsValid())
48 return false;
49
50 if (win7_or_higher) {
51 STORAGE_PROPERTY_QUERY query = {};
52 query.QueryType = PropertyStandardQuery;
53 query.PropertyId = StorageDeviceSeekPenaltyProperty;
54
55 DEVICE_SEEK_PENALTY_DESCRIPTOR result;
56 DWORD bytes_returned;
57 BOOL success = DeviceIoControl(volume.GetPlatformFile(),
58 IOCTL_STORAGE_QUERY_PROPERTY,
59 &query, sizeof(query),
60 &result, sizeof(result),
61 &bytes_returned, NULL);
62 if (success == FALSE || bytes_returned < sizeof(result))
63 return false;
64
65 *has_seek_penalty = result.IncursSeekPenalty != FALSE;
66 } else {
67 AtaRequest request = {};
68 request.query.AtaFlags = ATA_FLAGS_DATA_IN;
69 request.query.CurrentTaskFile[6] = ID_CMD;
70 request.query.DataBufferOffset = sizeof(request.query);
71 request.query.DataTransferLength = sizeof(request.result);
72 request.query.Length = sizeof(request.query);
73 request.query.TimeOutValue = 10;
74
75 DWORD bytes_returned;
76 BOOL success = DeviceIoControl(volume.GetPlatformFile(),
77 IOCTL_ATA_PASS_THROUGH,
78 &request, sizeof(request),
79 &request, sizeof(request),
80 &bytes_returned, NULL);
81 if (success == FALSE || bytes_returned < sizeof(request) ||
82 request.query.CurrentTaskFile[0]) {
83 return false;
84 }
85
86 *has_seek_penalty = request.result.NominalMediaRotationRate != 1;
87 }
88
89 return true;
90 }
OLDNEW
« no previous file with comments | « chrome/browser/metrics/drive_metrics_provider_unittest.cc ('k') | chrome/chrome_browser.gypi » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698