OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "net/base/platform_mime_util.h" | 5 #include "net/base/platform_mime_util.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/strings/utf_string_conversions.h" | 9 #include "base/strings/utf_string_conversions.h" |
10 #include "base/win/registry.h" | 10 #include "base/win/registry.h" |
11 | 11 |
12 namespace net { | 12 namespace net { |
13 | 13 |
14 bool PlatformMimeUtil::GetPlatformMimeTypeFromExtension( | 14 bool PlatformMimeUtil::GetPlatformMimeTypeFromExtension( |
15 const base::FilePath::StringType& ext, std::string* result) const { | 15 const base::FilePath::StringType& ext, std::string* result) const { |
16 // check windows registry for file extension's mime type (registry key | 16 // check windows registry for file extension's mime type (registry key |
17 // names are not case-sensitive). | 17 // names are not case-sensitive). |
18 std::wstring value, key = L"." + ext; | 18 std::wstring value, key = L"." + ext; |
19 base::win::RegKey(HKEY_CLASSES_ROOT, key.c_str(), KEY_READ).ReadValue( | 19 base::win::RegKey(HKEY_CLASSES_ROOT, key.c_str(), KEY_READ).ReadValue( |
20 L"Content Type", &value); | 20 L"Content Type", &value); |
21 if (!value.empty()) { | 21 if (!value.empty()) { |
22 *result = WideToUTF8(value); | 22 *result = base::WideToUTF8(value); |
23 return true; | 23 return true; |
24 } | 24 } |
25 return false; | 25 return false; |
26 } | 26 } |
27 | 27 |
28 bool PlatformMimeUtil::GetPreferredExtensionForMimeType( | 28 bool PlatformMimeUtil::GetPreferredExtensionForMimeType( |
29 const std::string& mime_type, base::FilePath::StringType* ext) const { | 29 const std::string& mime_type, base::FilePath::StringType* ext) const { |
30 std::wstring key(L"MIME\\Database\\Content Type\\" + UTF8ToWide(mime_type)); | 30 std::wstring key( |
| 31 L"MIME\\Database\\Content Type\\" + base::UTF8ToWide(mime_type)); |
31 if (base::win::RegKey(HKEY_CLASSES_ROOT, key.c_str(), KEY_READ).ReadValue( | 32 if (base::win::RegKey(HKEY_CLASSES_ROOT, key.c_str(), KEY_READ).ReadValue( |
32 L"Extension", ext) != ERROR_SUCCESS) { | 33 L"Extension", ext) != ERROR_SUCCESS) { |
33 return false; | 34 return false; |
34 } | 35 } |
35 // Strip off the leading dot, this should always be the case. | 36 // Strip off the leading dot, this should always be the case. |
36 if (!ext->empty() && ext->at(0) == L'.') | 37 if (!ext->empty() && ext->at(0) == L'.') |
37 ext->erase(ext->begin()); | 38 ext->erase(ext->begin()); |
38 | 39 |
39 return true; | 40 return true; |
40 } | 41 } |
41 | 42 |
42 void PlatformMimeUtil::GetPlatformExtensionsForMimeType( | 43 void PlatformMimeUtil::GetPlatformExtensionsForMimeType( |
43 const std::string& mime_type, | 44 const std::string& mime_type, |
44 base::hash_set<base::FilePath::StringType>* extensions) const { | 45 base::hash_set<base::FilePath::StringType>* extensions) const { |
45 // Multiple extensions could have the given mime type specified as their types | 46 // Multiple extensions could have the given mime type specified as their types |
46 // in their 'HKCR\.<extension>\Content Type' keys. Iterating all the HKCR | 47 // in their 'HKCR\.<extension>\Content Type' keys. Iterating all the HKCR |
47 // entries, though, is wildly impractical. Cheat by returning just the | 48 // entries, though, is wildly impractical. Cheat by returning just the |
48 // preferred extension. | 49 // preferred extension. |
49 base::FilePath::StringType ext; | 50 base::FilePath::StringType ext; |
50 if (GetPreferredExtensionForMimeType(mime_type, &ext)) | 51 if (GetPreferredExtensionForMimeType(mime_type, &ext)) |
51 extensions->insert(ext); | 52 extensions->insert(ext); |
52 } | 53 } |
53 | 54 |
54 } // namespace net | 55 } // namespace net |
OLD | NEW |