file_version_info_win.cc 6.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199
  1. // Copyright (c) 2011 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. #include "base/file_version_info_win.h"
  5. #include <windows.h>
  6. #include <stddef.h>
  7. #include <utility>
  8. #include "base/check.h"
  9. #include "base/files/file_path.h"
  10. #include "base/memory/ptr_util.h"
  11. #include "base/strings/string_util.h"
  12. #include "base/threading/scoped_blocking_call.h"
  13. #include "base/win/resource_util.h"
  14. namespace {
  15. struct LanguageAndCodePage {
  16. WORD language;
  17. WORD code_page;
  18. };
  19. // Returns the \VarFileInfo\Translation value extracted from the
  20. // VS_VERSION_INFO resource in |data|.
  21. LanguageAndCodePage* GetTranslate(const void* data) {
  22. static constexpr wchar_t kTranslation[] = L"\\VarFileInfo\\Translation";
  23. LPVOID translate = nullptr;
  24. UINT dummy_size;
  25. if (::VerQueryValue(data, kTranslation, &translate, &dummy_size))
  26. return static_cast<LanguageAndCodePage*>(translate);
  27. return nullptr;
  28. }
  29. const VS_FIXEDFILEINFO& GetVsFixedFileInfo(const void* data) {
  30. static constexpr wchar_t kRoot[] = L"\\";
  31. LPVOID fixed_file_info = nullptr;
  32. UINT dummy_size;
  33. CHECK(::VerQueryValue(data, kRoot, &fixed_file_info, &dummy_size));
  34. return *static_cast<VS_FIXEDFILEINFO*>(fixed_file_info);
  35. }
  36. } // namespace
  37. FileVersionInfoWin::~FileVersionInfoWin() = default;
  38. // static
  39. std::unique_ptr<FileVersionInfo>
  40. FileVersionInfo::CreateFileVersionInfoForModule(HMODULE module) {
  41. void* data;
  42. size_t version_info_length;
  43. const bool has_version_resource = base::win::GetResourceFromModule(
  44. module, VS_VERSION_INFO, RT_VERSION, &data, &version_info_length);
  45. if (!has_version_resource)
  46. return nullptr;
  47. const LanguageAndCodePage* translate = GetTranslate(data);
  48. if (!translate)
  49. return nullptr;
  50. return base::WrapUnique(
  51. new FileVersionInfoWin(data, translate->language, translate->code_page));
  52. }
  53. // static
  54. std::unique_ptr<FileVersionInfo> FileVersionInfo::CreateFileVersionInfo(
  55. const base::FilePath& file_path) {
  56. return FileVersionInfoWin::CreateFileVersionInfoWin(file_path);
  57. }
  58. // static
  59. std::unique_ptr<FileVersionInfoWin>
  60. FileVersionInfoWin::CreateFileVersionInfoWin(const base::FilePath& file_path) {
  61. base::ScopedBlockingCall scoped_blocking_call(FROM_HERE,
  62. base::BlockingType::MAY_BLOCK);
  63. DWORD dummy;
  64. const wchar_t* path = file_path.value().c_str();
  65. const DWORD length = ::GetFileVersionInfoSize(path, &dummy);
  66. if (length == 0)
  67. return nullptr;
  68. std::vector<uint8_t> data(length, 0);
  69. if (!::GetFileVersionInfo(path, dummy, length, data.data()))
  70. return nullptr;
  71. const LanguageAndCodePage* translate = GetTranslate(data.data());
  72. if (!translate)
  73. return nullptr;
  74. return base::WrapUnique(new FileVersionInfoWin(
  75. std::move(data), translate->language, translate->code_page));
  76. }
  77. std::u16string FileVersionInfoWin::company_name() {
  78. return GetStringValue(u"CompanyName");
  79. }
  80. std::u16string FileVersionInfoWin::company_short_name() {
  81. return GetStringValue(u"CompanyShortName");
  82. }
  83. std::u16string FileVersionInfoWin::internal_name() {
  84. return GetStringValue(u"InternalName");
  85. }
  86. std::u16string FileVersionInfoWin::product_name() {
  87. return GetStringValue(u"ProductName");
  88. }
  89. std::u16string FileVersionInfoWin::product_short_name() {
  90. return GetStringValue(u"ProductShortName");
  91. }
  92. std::u16string FileVersionInfoWin::product_version() {
  93. return GetStringValue(u"ProductVersion");
  94. }
  95. std::u16string FileVersionInfoWin::file_description() {
  96. return GetStringValue(u"FileDescription");
  97. }
  98. std::u16string FileVersionInfoWin::file_version() {
  99. return GetStringValue(u"FileVersion");
  100. }
  101. std::u16string FileVersionInfoWin::original_filename() {
  102. return GetStringValue(u"OriginalFilename");
  103. }
  104. std::u16string FileVersionInfoWin::special_build() {
  105. return GetStringValue(u"SpecialBuild");
  106. }
  107. bool FileVersionInfoWin::GetValue(const char16_t* name,
  108. std::u16string* value) const {
  109. const struct LanguageAndCodePage lang_codepages[] = {
  110. // Use the language and codepage from the DLL.
  111. {language_, code_page_},
  112. // Use the default language and codepage from the DLL.
  113. {::GetUserDefaultLangID(), code_page_},
  114. // Use the language from the DLL and Latin codepage (most common).
  115. {language_, 1252},
  116. // Use the default language and Latin codepage (most common).
  117. {::GetUserDefaultLangID(), 1252},
  118. };
  119. for (const auto& lang_codepage : lang_codepages) {
  120. wchar_t sub_block[MAX_PATH];
  121. _snwprintf_s(sub_block, MAX_PATH, MAX_PATH,
  122. L"\\StringFileInfo\\%04x%04x\\%ls", lang_codepage.language,
  123. lang_codepage.code_page, base::as_wcstr(name));
  124. LPVOID value_ptr = nullptr;
  125. uint32_t size;
  126. BOOL r = ::VerQueryValue(data_, sub_block, &value_ptr, &size);
  127. if (r && value_ptr && size) {
  128. value->assign(static_cast<char16_t*>(value_ptr), size - 1);
  129. return true;
  130. }
  131. }
  132. return false;
  133. }
  134. std::u16string FileVersionInfoWin::GetStringValue(const char16_t* name) const {
  135. std::u16string str;
  136. GetValue(name, &str);
  137. return str;
  138. }
  139. base::Version FileVersionInfoWin::GetFileVersion() const {
  140. return base::Version({HIWORD(fixed_file_info_.dwFileVersionMS),
  141. LOWORD(fixed_file_info_.dwFileVersionMS),
  142. HIWORD(fixed_file_info_.dwFileVersionLS),
  143. LOWORD(fixed_file_info_.dwFileVersionLS)});
  144. }
  145. FileVersionInfoWin::FileVersionInfoWin(std::vector<uint8_t>&& data,
  146. WORD language,
  147. WORD code_page)
  148. : owned_data_(std::move(data)),
  149. data_(owned_data_.data()),
  150. language_(language),
  151. code_page_(code_page),
  152. fixed_file_info_(GetVsFixedFileInfo(data_)) {
  153. DCHECK(!owned_data_.empty());
  154. }
  155. FileVersionInfoWin::FileVersionInfoWin(void* data,
  156. WORD language,
  157. WORD code_page)
  158. : data_(data),
  159. language_(language),
  160. code_page_(code_page),
  161. fixed_file_info_(GetVsFixedFileInfo(data)) {
  162. DCHECK(data_);
  163. }