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