base/file_version_info_win.cc

   1 // Copyright 2011 The Chromium Authors
   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 "base/file_version_info_win.h"
   6
   7 #include <windows.h>
   8 #include <stddef.h>
   9
  10 #include <utility>
  11
  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"
  18
  19 namespace {
  20
  21 struct LanguageAndCodePage {
  22   WORD language;
  23   WORD code_page;
  24 };
  25
  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;
  31   UINT dummy_size;
  32   if (::VerQueryValue(data, kTranslation, &translate, &dummy_size))
  33     return static_cast<LanguageAndCodePage*>(translate);
  34   return nullptr;
  35 }
  36
  37 const VS_FIXEDFILEINFO& GetVsFixedFileInfo(const void* data) {
  38   static constexpr wchar_t kRoot[] = L"\\";
  39   LPVOID fixed_file_info = nullptr;
  40   UINT dummy_size;
  41   CHECK(::VerQueryValue(data, kRoot, &fixed_file_info, &dummy_size));
  42   return *static_cast<VS_FIXEDFILEINFO*>(fixed_file_info);
  43 }
  44
  45 }  // namespace
  46
  47 FileVersionInfoWin::~FileVersionInfoWin() = default;
  48
  49 // static
  50 std::unique_ptr<FileVersionInfo>
  51 FileVersionInfo::CreateFileVersionInfoForModule(HMODULE module) {
  52   void* data;
  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)
  57     return nullptr;
  58
  59   const LanguageAndCodePage* translate = GetTranslate(data);
  60   if (!translate)
  61     return nullptr;
  62
  63   return base::WrapUnique(
  64       new FileVersionInfoWin(data, translate->language, translate->code_page));
  65 }
  66
  67 // static
  68 std::unique_ptr<FileVersionInfo> FileVersionInfo::CreateFileVersionInfo(
  69     const base::FilePath& file_path) {
  70   return FileVersionInfoWin::CreateFileVersionInfoWin(file_path);
  71 }
  72
  73 // static
  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);
  78
  79   DWORD dummy;
  80   const wchar_t* path = file_path.value().c_str();
  81   const DWORD length = ::GetFileVersionInfoSize(path, &dummy);
  82   if (length == 0)
  83     return nullptr;
  84
  85   std::vector<uint8_t> data(length, 0);
  86
  87   if (!::GetFileVersionInfo(path, dummy, length, data.data()))
  88     return nullptr;
  89
  90   const LanguageAndCodePage* translate = GetTranslate(data.data());
  91   if (!translate)
  92     return nullptr;
  93
  94   return base::WrapUnique(new FileVersionInfoWin(
  95       std::move(data), translate->language, translate->code_page));
  96 }
  97
  98 std::u16string FileVersionInfoWin::company_name() {
  99   return GetStringValue(u"CompanyName");
 100 }
 101
 102 std::u16string FileVersionInfoWin::company_short_name() {
 103   return GetStringValue(u"CompanyShortName");
 104 }
 105
 106 std::u16string FileVersionInfoWin::internal_name() {
 107   return GetStringValue(u"InternalName");
 108 }
 109
 110 std::u16string FileVersionInfoWin::product_name() {
 111   return GetStringValue(u"ProductName");
 112 }
 113
 114 std::u16string FileVersionInfoWin::product_short_name() {
 115   return GetStringValue(u"ProductShortName");
 116 }
 117
 118 std::u16string FileVersionInfoWin::product_version() {
 119   return GetStringValue(u"ProductVersion");
 120 }
 121
 122 std::u16string FileVersionInfoWin::file_description() {
 123   return GetStringValue(u"FileDescription");
 124 }
 125
 126 std::u16string FileVersionInfoWin::file_version() {
 127   return GetStringValue(u"FileVersion");
 128 }
 129
 130 std::u16string FileVersionInfoWin::original_filename() {
 131   return GetStringValue(u"OriginalFilename");
 132 }
 133
 134 std::u16string FileVersionInfoWin::special_build() {
 135   return GetStringValue(u"SpecialBuild");
 136 }
 137
 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).
 146       {language_, 1252},
 147       // Use the default language and Latin codepage (most common).
 148       {::GetUserDefaultLangID(), 1252},
 149   };
 150
 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;
 157     uint32_t size;
 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);
 161       return true;
 162     }
 163   }
 164   return false;
 165 }
 166
 167 std::u16string FileVersionInfoWin::GetStringValue(const char16_t* name) const {
 168   std::u16string str;
 169   GetValue(name, &str);
 170   return str;
 171 }
 172
 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)});
 178 }
 179
 180 FileVersionInfoWin::FileVersionInfoWin(std::vector<uint8_t>&& data,
 181                                        WORD language,
 182                                        WORD code_page)
 183     : owned_data_(std::move(data)),
 184       data_(owned_data_.data()),
 185       language_(language),
 186       code_page_(code_page),
 187       fixed_file_info_(GetVsFixedFileInfo(data_)) {
 188   DCHECK(!owned_data_.empty());
 189 }
 190
 191 FileVersionInfoWin::FileVersionInfoWin(void* data,
 192                                        WORD language,
 193                                        WORD code_page)
 194     : data_(data),
 195       language_(language),
 196       code_page_(code_page),
 197       fixed_file_info_(GetVsFixedFileInfo(data)) {
 198   DCHECK(data_);
 199 }