setup_native/source/win32/customactions/quickstarter/quickstarter.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * This file is part of the LibreOffice project.
   4  *
   5  * This Source Code Form is subject to the terms of the Mozilla Public
   6  * License, v. 2.0. If a copy of the MPL was not distributed with this
   7  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8  *
   9  * This file incorporates work covered by the following license notice:
  10  *
  11  *   Licensed to the Apache Software Foundation (ASF) under one or more
  12  *   contributor license agreements. See the NOTICE file distributed
  13  *   with this work for additional information regarding copyright
  14  *   ownership. The ASF licenses this file to you under the Apache
  15  *   License, Version 2.0 (the "License"); you may not use this file
  16  *   except in compliance with the License. You may obtain a copy of
  17  *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
  18  */
  19
  20 #include "quickstarter.hxx"
  21
  22 #include <psapi.h>
  23
  24 #include <malloc.h>
  25
  26 std::wstring GetOfficeInstallationPathW(MSIHANDLE handle)
  27 {
  28     std::wstring progpath;
  29     DWORD sz = 0;
  30     PWSTR dummy = const_cast<PWSTR>(L"");
  31
  32     if (MsiGetPropertyW(handle, L"INSTALLLOCATION", dummy, &sz) == ERROR_MORE_DATA)
  33     {
  34         sz++; // space for the final '\0'
  35         DWORD nbytes = sz * sizeof(WCHAR);
  36         PWSTR buff = static_cast<PWSTR>(_alloca(nbytes));
  37         ZeroMemory(buff, nbytes);
  38         MsiGetPropertyW(handle, L"INSTALLLOCATION", buff, &sz);
  39         progpath = buff;
  40     }
  41     return progpath;
  42 }
  43
  44 std::wstring GetOfficeProductNameW(MSIHANDLE handle)
  45 {
  46     std::wstring productname;
  47     DWORD sz = 0;
  48     PWSTR dummy = const_cast<PWSTR>(L"");
  49
  50     if (MsiGetPropertyW(handle, L"ProductName", dummy, &sz) == ERROR_MORE_DATA)
  51     {
  52         sz++; // space for the final '\0'
  53         DWORD nbytes = sz * sizeof(WCHAR);
  54         PWSTR buff = static_cast<PWSTR>(_alloca(nbytes));
  55         ZeroMemory(buff, nbytes);
  56         MsiGetPropertyW(handle, L"ProductName", buff, &sz);
  57         productname = buff;
  58     }
  59     return productname;
  60 }
  61
  62 std::wstring GetQuickstarterLinkNameW(MSIHANDLE handle)
  63 {
  64     std::wstring quickstarterlinkname;
  65     DWORD sz = 0;
  66     PWSTR dummy = const_cast<PWSTR>(L"");
  67
  68     if (MsiGetPropertyW(handle, L"Quickstarterlinkname", dummy, &sz) == ERROR_MORE_DATA)
  69     {
  70         sz++; // space for the final '\0'
  71         DWORD nbytes = sz * sizeof(WCHAR);
  72         PWSTR buff = static_cast<PWSTR>(_alloca(nbytes));
  73         ZeroMemory(buff, nbytes);
  74         MsiGetPropertyW(handle, L"Quickstarterlinkname", buff, &sz);
  75         quickstarterlinkname = buff;
  76     }
  77     else if (MsiGetPropertyW(handle, L"ProductName", dummy, &sz) == ERROR_MORE_DATA)
  78     {
  79         sz++; // space for the final '\0'
  80         DWORD nbytes = sz * sizeof(WCHAR);
  81         PWSTR buff = static_cast<PWSTR>(_alloca(nbytes));
  82         ZeroMemory(buff, nbytes);
  83         MsiGetPropertyW(handle, L"ProductName", buff, &sz);
  84         quickstarterlinkname = buff;
  85     }
  86     return quickstarterlinkname;
  87 }
  88
  89 static bool IsValidHandle( HANDLE handle )
  90 {
  91     return nullptr != handle && INVALID_HANDLE_VALUE != handle;
  92 }
  93
  94 static DWORD WINAPI GetModuleFileNameExW_( HANDLE hProcess, HMODULE hModule, PWSTR lpFileName, DWORD nSize )
  95 {
  96     static auto lpProc = []() {
  97         HMODULE hLibrary = LoadLibraryW(L"PSAPI.DLL");
  98         decltype(GetModuleFileNameExW)* pRet = nullptr;
  99         if (hLibrary)
 100             pRet = reinterpret_cast<decltype(GetModuleFileNameExW)*>(
 101                 GetProcAddress(hLibrary, "GetModuleFileNameExW"));
 102         return pRet;
 103     }();
 104
 105     if ( lpProc )
 106         return lpProc( hProcess, hModule, lpFileName, nSize );
 107
 108     return 0;
 109
 110 }
 111
 112 std::wstring GetProcessImagePathW( DWORD dwProcessId )
 113 {
 114     std::wstring sImagePath;
 115
 116     HANDLE  hProcess = OpenProcess( PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, dwProcessId );
 117
 118     if ( IsValidHandle( hProcess ) )
 119     {
 120         WCHAR szPathBuffer[MAX_PATH] = L"";
 121
 122         if ( GetModuleFileNameExW_( hProcess, nullptr, szPathBuffer, sizeof(szPathBuffer)/sizeof(szPathBuffer[0]) ) )
 123             sImagePath = szPathBuffer;
 124
 125         CloseHandle( hProcess );
 126     }
 127
 128     return sImagePath;
 129 }
 130
 131 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */