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.
5 #include "base/file_version_info_win.h"
12 #include "base/check.h"
13 #include "base/files/file_path.h"
14 #include "base/memory/ptr_util.h"
15 #include "base/strings/string_util.h"
16 #include "base/threading/scoped_blocking_call.h"
17 #include "base/win/resource_util.h"
21 struct LanguageAndCodePage {
26 // Returns the \VarFileInfo\Translation value extracted from the
27 // VS_VERSION_INFO resource in |data|.
28 LanguageAndCodePage* GetTranslate(const void* data) {
29 static constexpr wchar_t kTranslation[] = L"\\VarFileInfo\\Translation";
30 LPVOID translate = nullptr;
32 if (::VerQueryValue(data, kTranslation, &translate, &dummy_size))
33 return static_cast<LanguageAndCodePage*>(translate);
37 const VS_FIXEDFILEINFO& GetVsFixedFileInfo(const void* data) {
38 static constexpr wchar_t kRoot[] = L"\\";
39 LPVOID fixed_file_info = nullptr;
41 CHECK(::VerQueryValue(data, kRoot, &fixed_file_info, &dummy_size));
42 return *static_cast<VS_FIXEDFILEINFO*>(fixed_file_info);
47 FileVersionInfoWin::~FileVersionInfoWin() = default;
50 std::unique_ptr<FileVersionInfo>
51 FileVersionInfo::CreateFileVersionInfoForModule(HMODULE module) {
53 size_t version_info_length;
54 const bool has_version_resource = base::win::GetResourceFromModule(
55 module, VS_VERSION_INFO, RT_VERSION, &data, &version_info_length);
56 if (!has_version_resource)
59 const LanguageAndCodePage* translate = GetTranslate(data);
63 return base::WrapUnique(
64 new FileVersionInfoWin(data, translate->language, translate->code_page));
68 std::unique_ptr<FileVersionInfo> FileVersionInfo::CreateFileVersionInfo(
69 const base::FilePath& file_path) {
70 return FileVersionInfoWin::CreateFileVersionInfoWin(file_path);
74 std::unique_ptr<FileVersionInfoWin>
75 FileVersionInfoWin::CreateFileVersionInfoWin(const base::FilePath& file_path) {
76 base::ScopedBlockingCall scoped_blocking_call(FROM_HERE,
77 base::BlockingType::MAY_BLOCK);
80 const wchar_t* path = file_path.value().c_str();
81 const DWORD length = ::GetFileVersionInfoSize(path, &dummy);
85 std::vector<uint8_t> data(length, 0);
87 if (!::GetFileVersionInfo(path, dummy, length, data.data()))
90 const LanguageAndCodePage* translate = GetTranslate(data.data());
94 return base::WrapUnique(new FileVersionInfoWin(
95 std::move(data), translate->language, translate->code_page));
98 std::u16string FileVersionInfoWin::company_name() {
99 return GetStringValue(u"CompanyName");
102 std::u16string FileVersionInfoWin::company_short_name() {
103 return GetStringValue(u"CompanyShortName");
106 std::u16string FileVersionInfoWin::internal_name() {
107 return GetStringValue(u"InternalName");
110 std::u16string FileVersionInfoWin::product_name() {
111 return GetStringValue(u"ProductName");
114 std::u16string FileVersionInfoWin::product_short_name() {
115 return GetStringValue(u"ProductShortName");
118 std::u16string FileVersionInfoWin::product_version() {
119 return GetStringValue(u"ProductVersion");
122 std::u16string FileVersionInfoWin::file_description() {
123 return GetStringValue(u"FileDescription");
126 std::u16string FileVersionInfoWin::file_version() {
127 return GetStringValue(u"FileVersion");
130 std::u16string FileVersionInfoWin::original_filename() {
131 return GetStringValue(u"OriginalFilename");
134 std::u16string FileVersionInfoWin::special_build() {
135 return GetStringValue(u"SpecialBuild");
138 bool FileVersionInfoWin::GetValue(const char16_t* name,
139 std::u16string* value) const {
140 const struct LanguageAndCodePage lang_codepages[] = {
141 // Use the language and codepage from the DLL.
142 {language_, code_page_},
143 // Use the default language and codepage from the DLL.
144 {::GetUserDefaultLangID(), code_page_},
145 // Use the language from the DLL and Latin codepage (most common).
147 // Use the default language and Latin codepage (most common).
148 {::GetUserDefaultLangID(), 1252},
151 for (const auto& lang_codepage : lang_codepages) {
152 wchar_t sub_block[MAX_PATH];
153 _snwprintf_s(sub_block, MAX_PATH, MAX_PATH,
154 L"\\StringFileInfo\\%04x%04x\\%ls", lang_codepage.language,
155 lang_codepage.code_page, base::as_wcstr(name));
156 LPVOID value_ptr = nullptr;
158 BOOL r = ::VerQueryValue(data_, sub_block, &value_ptr, &size);
159 if (r && value_ptr && size) {
160 value->assign(static_cast<char16_t*>(value_ptr), size - 1);
167 std::u16string FileVersionInfoWin::GetStringValue(const char16_t* name) const {
169 GetValue(name, &str);
173 base::Version FileVersionInfoWin::GetFileVersion() const {
174 return base::Version({HIWORD(fixed_file_info_.dwFileVersionMS),
175 LOWORD(fixed_file_info_.dwFileVersionMS),
176 HIWORD(fixed_file_info_.dwFileVersionLS),
177 LOWORD(fixed_file_info_.dwFileVersionLS)});
180 FileVersionInfoWin::FileVersionInfoWin(std::vector<uint8_t>&& data,
183 : owned_data_(std::move(data)),
184 data_(owned_data_.data()),
186 code_page_(code_page),
187 fixed_file_info_(GetVsFixedFileInfo(data_)) {
188 DCHECK(!owned_data_.empty());
191 FileVersionInfoWin::FileVersionInfoWin(void* data,
196 code_page_(code_page),
197 fixed_file_info_(GetVsFixedFileInfo(data)) {