win8/metro_driver/winrt_utils.cc

   1 // Copyright (c) 2012 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 "stdafx.h"
   6 #include "winrt_utils.h"
   7
   8 #include <shlobj.h>
   9
  10 #include "base/files/file_path.h"
  11 #include "base/logging.h"
  12 #include "base/win/scoped_com_initializer.h"
  13 #include "base/win/scoped_comptr.h"
  14 #include "chrome/installer/util/browser_distribution.h"
  15 #include "chrome/installer/util/install_util.h"
  16
  17 void CheckHR(HRESULT hr, const char* message) {
  18   if (FAILED(hr)) {
  19     if (message)
  20       PLOG(DFATAL) << message << ", hr = " << std::hex << hr;
  21     else
  22       PLOG(DFATAL) << "COM ERROR" << ", hr = " << std::hex << hr;
  23   }
  24 }
  25
  26 HSTRING MakeHString(const base::string16& str) {
  27   HSTRING hstr;
  28   if (FAILED(::WindowsCreateString(str.c_str(), static_cast<UINT32>(str.size()),
  29                                    &hstr))) {
  30     PLOG(DFATAL) << "Hstring creation failed";
  31   }
  32   return hstr;
  33 }
  34
  35 base::string16 MakeStdWString(HSTRING hstring) {
  36   const wchar_t* str;
  37   UINT32 size = 0;
  38   str = ::WindowsGetStringRawBuffer(hstring, &size);
  39   if (!size)
  40     return base::string16();
  41   return base::string16(str, size);
  42 }
  43
  44 namespace {
  45
  46 #define IMPLEMENT_CREATE_PROPERTY(Name, Type) \
  47 HRESULT Create ## Name ## Property(Type value, \
  48                                    winfoundtn::IPropertyValue** prop) { \
  49   mswr::ComPtr<winfoundtn::IPropertyValueStatics> property_value_statics; \
  50   HRESULT hr = winrt_utils::CreateActivationFactory( \
  51       RuntimeClass_Windows_Foundation_PropertyValue, \
  52       property_value_statics.GetAddressOf()); \
  53   CheckHR(hr, "Can't create IPropertyValueStatics"); \
  54   hr = property_value_statics->Create ## Name ## ( \
  55       value, \
  56       reinterpret_cast<IInspectable**>(prop)); \
  57   CheckHR(hr, "Failed to create Property"); \
  58   return hr; \
  59 }
  60
  61 #define COMPARE_ATOMIC_PROPERTY_VALUES(Name, Type) \
  62   Type lhs_value; \
  63   hr = lhs->Get ## Name ##(&lhs_value); \
  64   CheckHR(hr, "Can't get value for lhs"); \
  65   Type rhs_value; \
  66   hr = rhs->Get ## Name ##(&rhs_value); \
  67   CheckHR(hr, "Can't get value for rhs"); \
  68   if (lhs_value < rhs_value) \
  69     *result = -1; \
  70   else if (lhs_value > rhs_value) \
  71     *result = 1; \
  72   else \
  73     *result = 0; \
  74   hr = S_OK
  75
  76 }  // namespace
  77
  78 namespace winrt_utils {
  79
  80 IMPLEMENT_CREATE_PROPERTY(String, HSTRING);
  81 IMPLEMENT_CREATE_PROPERTY(Int16, INT16);
  82 IMPLEMENT_CREATE_PROPERTY(Int32, INT32);
  83 IMPLEMENT_CREATE_PROPERTY(Int64, INT64);
  84 IMPLEMENT_CREATE_PROPERTY(UInt8, UINT8);
  85 IMPLEMENT_CREATE_PROPERTY(UInt16, UINT16);
  86 IMPLEMENT_CREATE_PROPERTY(UInt32, UINT32);
  87 IMPLEMENT_CREATE_PROPERTY(UInt64, UINT64);
  88
  89 HRESULT CompareProperties(winfoundtn::IPropertyValue* lhs,
  90                           winfoundtn::IPropertyValue* rhs,
  91                           INT32* result) {
  92   if (result == nullptr) {
  93     PLOG(DFATAL) << "Invalid argument to CompareProperties.";
  94     return E_INVALIDARG;
  95   }
  96
  97   if (lhs == rhs) {
  98     *result = 0;
  99     return S_OK;
 100   }
 101
 102   winfoundtn::PropertyType lhs_property_type;
 103   HRESULT hr = lhs->get_Type(&lhs_property_type);
 104   if (FAILED(hr)) {
 105     PLOG(DFATAL) << "Can't get property type for lhs, hr=" << std::hex << hr;
 106   }
 107
 108   winfoundtn::PropertyType rhs_property_type;
 109   hr = rhs->get_Type(&rhs_property_type);
 110   CheckHR(hr, "Can't get property type for rhs");
 111
 112   if (lhs_property_type != rhs_property_type)
 113     return E_INVALIDARG;
 114
 115   switch (lhs_property_type) {
 116     case winfoundtn::PropertyType::PropertyType_String: {
 117       mswrw::HString lhs_string;
 118       hr = lhs->GetString(lhs_string.GetAddressOf());
 119       CheckHR(hr, "Can't get string for lhs");
 120
 121       mswrw::HString rhs_string;
 122       hr = rhs->GetString(rhs_string.GetAddressOf());
 123       CheckHR(hr, "Can't get string for rhs");
 124
 125       hr = WindowsCompareStringOrdinal(
 126           lhs_string.Get(), rhs_string.Get(), result);
 127       break;
 128     }
 129     case winfoundtn::PropertyType::PropertyType_Char16: {
 130       COMPARE_ATOMIC_PROPERTY_VALUES(Char16, wchar_t);
 131       break;
 132     }
 133     case winfoundtn::PropertyType::PropertyType_Double: {
 134       COMPARE_ATOMIC_PROPERTY_VALUES(Double, double);
 135       break;
 136     }
 137     case winfoundtn::PropertyType::PropertyType_Int16: {
 138       COMPARE_ATOMIC_PROPERTY_VALUES(Int16, INT16);
 139       break;
 140     }
 141     case winfoundtn::PropertyType::PropertyType_Int32: {
 142       COMPARE_ATOMIC_PROPERTY_VALUES(Int32, INT32);
 143       break;
 144     }
 145     case winfoundtn::PropertyType::PropertyType_Int64: {
 146       COMPARE_ATOMIC_PROPERTY_VALUES(Int64, INT64);
 147       break;
 148     }
 149     case winfoundtn::PropertyType::PropertyType_UInt8: {
 150       COMPARE_ATOMIC_PROPERTY_VALUES(UInt8, UINT8);
 151       break;
 152     }
 153     case winfoundtn::PropertyType::PropertyType_UInt16: {
 154       COMPARE_ATOMIC_PROPERTY_VALUES(UInt16, UINT16);
 155       break;
 156     }
 157     case winfoundtn::PropertyType::PropertyType_UInt32: {
 158       COMPARE_ATOMIC_PROPERTY_VALUES(UInt32, UINT32);
 159       break;
 160     }
 161     case winfoundtn::PropertyType::PropertyType_UInt64: {
 162       COMPARE_ATOMIC_PROPERTY_VALUES(UInt64, UINT64);
 163       break;
 164     }
 165     default: {
 166       hr = E_NOTIMPL;
 167     }
 168   }
 169   return hr;
 170 }
 171
 172 bool GetArgumentsFromShortcut(const base::FilePath& shortcut,
 173                               base::string16* arguments) {
 174   HRESULT result;
 175   base::win::ScopedComPtr<IShellLink> i_shell_link;
 176   bool is_resolved = false;
 177
 178
 179   base::win::ScopedCOMInitializer sta_com_initializer;
 180
 181   // Get pointer to the IShellLink interface
 182   result = i_shell_link.CreateInstance(CLSID_ShellLink, NULL,
 183                                        CLSCTX_INPROC_SERVER);
 184   if (SUCCEEDED(result)) {
 185     base::win::ScopedComPtr<IPersistFile> persist;
 186     // Query IShellLink for the IPersistFile interface
 187     result = persist.QueryFrom(i_shell_link);
 188     if (SUCCEEDED(result)) {
 189       WCHAR temp_arguments[MAX_PATH];
 190       // Load the shell link
 191       result = persist->Load(shortcut.value().c_str(), STGM_READ);
 192       if (SUCCEEDED(result)) {
 193         result = i_shell_link->GetArguments(temp_arguments, MAX_PATH);
 194         *arguments = temp_arguments;
 195         is_resolved = true;
 196       }
 197     }
 198   }
 199
 200   return is_resolved;
 201 }
 202
 203 base::string16 ReadArgumentsFromPinnedTaskbarShortcut() {
 204   wchar_t path_buffer[MAX_PATH] = {};
 205
 206   if (SUCCEEDED(SHGetFolderPath(NULL, CSIDL_APPDATA, NULL,
 207                                 SHGFP_TYPE_CURRENT, path_buffer))) {
 208     base::FilePath shortcut(path_buffer);
 209     shortcut = shortcut.Append(
 210         L"Microsoft\\Internet Explorer\\Quick Launch\\User Pinned\\TaskBar");
 211
 212     BrowserDistribution* dist = BrowserDistribution::GetDistribution();
 213     base::string16 link_name = dist->GetShortcutName(
 214         BrowserDistribution::SHORTCUT_CHROME) + installer::kLnkExt;
 215     shortcut = shortcut.Append(link_name);
 216
 217     base::string16 arguments;
 218     if (GetArgumentsFromShortcut(shortcut, &arguments)) {
 219       return arguments;
 220     }
 221   }
 222
 223   return L"";
 224 }
 225
 226 }  // namespace winrt_utils