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