base/file_version_info_win.cc

   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
   5 #include "base/file_version_info_win.h"
   6
   7 #include <windows.h>
   8
   9 #include "base/file_version_info.h"
  10 #include "base/files/file_path.h"
  11 #include "base/logging.h"
  12 #include "base/threading/thread_restrictions.h"
  13
  14 using base::FilePath;
  15
  16 FileVersionInfoWin::FileVersionInfoWin(void* data,
  17                                        WORD language,
  18                                        WORD code_page)
  19     : language_(language), code_page_(code_page) {
  20   base::ThreadRestrictions::AssertIOAllowed();
  21   data_.reset((char*) data);
  22   fixed_file_info_ = NULL;
  23   UINT size;
  24   ::VerQueryValue(data_.get(), L"\\", (LPVOID*)&fixed_file_info_, &size);
  25 }
  26
  27 FileVersionInfoWin::~FileVersionInfoWin() {
  28   DCHECK(data_.get());
  29 }
  30
  31 typedef struct {
  32   WORD language;
  33   WORD code_page;
  34 } LanguageAndCodePage;
  35
  36 // static
  37 FileVersionInfo* FileVersionInfo::CreateFileVersionInfoForModule(
  38     HMODULE module) {
  39   // Note that the use of MAX_PATH is basically in line with what we do for
  40   // all registered paths (PathProviderWin).
  41   wchar_t system_buffer[MAX_PATH];
  42   system_buffer[0] = 0;
  43   if (!GetModuleFileName(module, system_buffer, MAX_PATH))
  44     return NULL;
  45
  46   FilePath app_path(system_buffer);
  47   return CreateFileVersionInfo(app_path);
  48 }
  49
  50 // static
  51 FileVersionInfo* FileVersionInfo::CreateFileVersionInfo(
  52     const FilePath& file_path) {
  53   base::ThreadRestrictions::AssertIOAllowed();
  54
  55   DWORD dummy;
  56   const wchar_t* path = file_path.value().c_str();
  57   DWORD length = ::GetFileVersionInfoSize(path, &dummy);
  58   if (length == 0)
  59     return NULL;
  60
  61   void* data = calloc(length, 1);
  62   if (!data)
  63     return NULL;
  64
  65   if (!::GetFileVersionInfo(path, dummy, length, data)) {
  66     free(data);
  67     return NULL;
  68   }
  69
  70   LanguageAndCodePage* translate = NULL;
  71   uint32 page_count;
  72   BOOL query_result = VerQueryValue(data, L"\\VarFileInfo\\Translation",
  73                                    (void**) &translate, &page_count);
  74
  75   if (query_result && translate) {
  76     return new FileVersionInfoWin(data, translate->language,
  77                                   translate->code_page);
  78
  79   } else {
  80     free(data);
  81     return NULL;
  82   }
  83 }
  84
  85 base::string16 FileVersionInfoWin::company_name() {
  86   return GetStringValue(L"CompanyName");
  87 }
  88
  89 base::string16 FileVersionInfoWin::company_short_name() {
  90   return GetStringValue(L"CompanyShortName");
  91 }
  92
  93 base::string16 FileVersionInfoWin::internal_name() {
  94   return GetStringValue(L"InternalName");
  95 }
  96
  97 base::string16 FileVersionInfoWin::product_name() {
  98   return GetStringValue(L"ProductName");
  99 }
 100
 101 base::string16 FileVersionInfoWin::product_short_name() {
 102   return GetStringValue(L"ProductShortName");
 103 }
 104
 105 base::string16 FileVersionInfoWin::comments() {
 106   return GetStringValue(L"Comments");
 107 }
 108
 109 base::string16 FileVersionInfoWin::legal_copyright() {
 110   return GetStringValue(L"LegalCopyright");
 111 }
 112
 113 base::string16 FileVersionInfoWin::product_version() {
 114   return GetStringValue(L"ProductVersion");
 115 }
 116
 117 base::string16 FileVersionInfoWin::file_description() {
 118   return GetStringValue(L"FileDescription");
 119 }
 120
 121 base::string16 FileVersionInfoWin::legal_trademarks() {
 122   return GetStringValue(L"LegalTrademarks");
 123 }
 124
 125 base::string16 FileVersionInfoWin::private_build() {
 126   return GetStringValue(L"PrivateBuild");
 127 }
 128
 129 base::string16 FileVersionInfoWin::file_version() {
 130   return GetStringValue(L"FileVersion");
 131 }
 132
 133 base::string16 FileVersionInfoWin::original_filename() {
 134   return GetStringValue(L"OriginalFilename");
 135 }
 136
 137 base::string16 FileVersionInfoWin::special_build() {
 138   return GetStringValue(L"SpecialBuild");
 139 }
 140
 141 base::string16 FileVersionInfoWin::last_change() {
 142   return GetStringValue(L"LastChange");
 143 }
 144
 145 bool FileVersionInfoWin::is_official_build() {
 146   return (GetStringValue(L"Official Build").compare(L"1") == 0);
 147 }
 148
 149 bool FileVersionInfoWin::GetValue(const wchar_t* name,
 150                                   std::wstring* value_str) {
 151   WORD lang_codepage[8];
 152   int i = 0;
 153   // Use the language and codepage from the DLL.
 154   lang_codepage[i++] = language_;
 155   lang_codepage[i++] = code_page_;
 156   // Use the default language and codepage from the DLL.
 157   lang_codepage[i++] = ::GetUserDefaultLangID();
 158   lang_codepage[i++] = code_page_;
 159   // Use the language from the DLL and Latin codepage (most common).
 160   lang_codepage[i++] = language_;
 161   lang_codepage[i++] = 1252;
 162   // Use the default language and Latin codepage (most common).
 163   lang_codepage[i++] = ::GetUserDefaultLangID();
 164   lang_codepage[i++] = 1252;
 165
 166   i = 0;
 167   while (i < arraysize(lang_codepage)) {
 168     wchar_t sub_block[MAX_PATH];
 169     WORD language = lang_codepage[i++];
 170     WORD code_page = lang_codepage[i++];
 171     _snwprintf_s(sub_block, MAX_PATH, MAX_PATH,
 172                  L"\\StringFileInfo\\%04x%04x\\%ls", language, code_page, name);
 173     LPVOID value = NULL;
 174     uint32 size;
 175     BOOL r = ::VerQueryValue(data_.get(), sub_block, &value, &size);
 176     if (r && value) {
 177       value_str->assign(static_cast<wchar_t*>(value));
 178       return true;
 179     }
 180   }
 181   return false;
 182 }
 183
 184 std::wstring FileVersionInfoWin::GetStringValue(const wchar_t* name) {
 185   std::wstring str;
 186   if (GetValue(name, &str))
 187     return str;
 188   else
 189     return L"";
 190 }