desktop/win32/source/rebase/rebase.cxx

   1 /*************************************************************************
   2  *
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * Copyright 2008 by Sun Microsystems, Inc.
   6  *
   7  * OpenOffice.org - a multi-platform office productivity suite
   8  *
   9  * $RCSfile: md5.cxx,v $
  10  * $Revision: 1.8 $
  11  *
  12  * This file is part of OpenOffice.org.
  13  *
  14  * OpenOffice.org is free software: you can redistribute it and/or modify
  15  * it under the terms of the GNU Lesser General Public License version 3
  16  * only, as published by the Free Software Foundation.
  17  *
  18  * OpenOffice.org is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU Lesser General Public License version 3 for more details
  22  * (a copy is included in the LICENSE file that accompanied this code).
  23  *
  24  * You should have received a copy of the GNU Lesser General Public License
  25  * version 3 along with OpenOffice.org.  If not, see
  26  * <http://www.openoffice.org/license.html>
  27  * for a copy of the LGPLv3 License.
  28  *
  29  ************************************************************************/
  30
  31 #include "precompiled_desktop.hxx"
  32 #define UNICODE
  33 #define _UNICODE
  34
  35 #define WIN32_LEAN_AND_MEAN
  36 #if defined _MSC_VER
  37 #pragma warning(push, 1)
  38 #endif
  39 #include <windows.h>
  40 #include <shellapi.h>
  41 #include <imagehlp.h>
  42 #include <wchar.h>
  43 #if defined _MSC_VER
  44 #pragma warning(pop)
  45 #endif
  46
  47 #include <time.h>
  48 #include "sal/config.h"
  49 #include "tools/pathutils.hxx"
  50
  51 #define MY_LENGTH(s) (sizeof (s) / sizeof *(s) - 1)
  52 #define MY_STRING(s) (s), MY_LENGTH(s)
  53
  54 const int   FORMAT_MESSAGE_SIZE = 4096;
  55 const DWORD PE_Signature        = 0x00004550;
  56 const DWORD BASEVIRTUALADDRESS  = 0x10000000;
  57
  58 namespace
  59 {
  60
  61 bool IsValidHandle( HANDLE handle )
  62 {
  63     return ((NULL != handle) && (INVALID_HANDLE_VALUE != handle));
  64 }
  65
  66 void fail()
  67 {
  68     LPWSTR buf = NULL;
  69     FormatMessageW(
  70         FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, NULL,
  71         GetLastError(), 0, reinterpret_cast< LPWSTR >(&buf), 0, NULL);
  72     MessageBoxW(NULL, buf, NULL, MB_OK | MB_ICONERROR);
  73     LocalFree(buf);
  74     TerminateProcess(GetCurrentProcess(), 255);
  75 }
  76
  77 bool rebaseImage( wchar_t* pszFilePath, ULONG nNewImageBase)
  78 {
  79     ULONG ulOldImageSize;
  80     ULONG_PTR lpOldImageBase;
  81     ULONG ulNewImageSize;
  82     ULONG_PTR lpNewImageBase  = nNewImageBase;
  83     ULONG         ulDateTimeStamp = 0;
  84     bool      bResult(false);
  85
  86     char cszFilePath[_MAX_PATH+1] = {0};
  87     int nResult = WideCharToMultiByte(CP_ACP, 0, pszFilePath, -1, cszFilePath, _MAX_PATH, NULL, NULL);
  88
  89     if (nResult != 0)
  90     {
  91         BOOL bResult = ReBaseImage(
  92             cszFilePath,
  93             "",
  94             TRUE,
  95             FALSE,
  96             FALSE,
  97             0,
  98             &ulOldImageSize,
  99             &lpOldImageBase,
 100             &ulNewImageSize,
 101             &lpNewImageBase,
 102             ulDateTimeStamp );
 103     }
 104
 105     return bResult;
 106 }
 107
 108 wchar_t* getBrandPath(wchar_t * path)
 109 {
 110     DWORD n = GetModuleFileNameW(NULL, path, MAX_PATH);
 111     if (n == 0 || n >= MAX_PATH) {
 112         exit(EXIT_FAILURE);
 113     }
 114     return tools::filename(path);
 115 }
 116
 117 void rebaseImagesInFolder( wchar_t* pszFolder, DWORD nNewImageBase )
 118 {
 119     wchar_t szPattern[MAX_PATH];
 120     wchar_t     *lpLastSlash = wcsrchr( pszFolder, '\\' );
 121     if ( lpLastSlash )
 122     {
 123         size_t len = lpLastSlash - pszFolder + 1;
 124         wcsncpy( szPattern, pszFolder, len );
 125         wcsncpy( szPattern + len, TEXT("*.dll"), sizeof(szPattern)/sizeof(szPattern[0]) - len );
 126     }
 127
 128     WIN32_FIND_DATA     aFindFileData;
 129     HANDLE      hFind = FindFirstFile( szPattern, &aFindFileData );
 130
 131     if ( IsValidHandle(hFind) )
 132     {
 133         BOOL fSuccess = false;
 134
 135         do
 136         {
 137             wchar_t szLibFilePath[MAX_PATH];
 138             wchar_t     *lpLastSlash = wcsrchr( pszFolder, '\\' );
 139             if ( lpLastSlash )
 140             {
 141                 size_t len = lpLastSlash - pszFolder + 1;
 142                 wcsncpy( szLibFilePath, pszFolder, len );
 143                 wcsncpy( szLibFilePath + len, aFindFileData.cFileName, sizeof(szLibFilePath)/sizeof(szLibFilePath[0]) - len );
 144             }
 145
 146             rebaseImage( szLibFilePath, nNewImageBase );
 147             fSuccess = FindNextFile( hFind, &aFindFileData );
 148         }
 149         while ( fSuccess );
 150
 151         FindClose( hFind );
 152     }
 153 }
 154
 155 }
 156
 157 extern "C" int APIENTRY WinMain( HINSTANCE, HINSTANCE, LPSTR, int )
 158 {
 159     wchar_t path[MAX_PATH];
 160
 161     wchar_t * pathEnd = getBrandPath(path);
 162
 163     if (tools::buildPath(path, path, pathEnd, MY_STRING(L"")) == NULL)
 164         fail();
 165     rebaseImagesInFolder(path, BASEVIRTUALADDRESS);
 166
 167     if (tools::buildPath(path, path, pathEnd, MY_STRING(L"..\\basis-link")) == NULL)
 168         fail();
 169     pathEnd = tools::resolveLink(path);
 170
 171     if ( pathEnd == NULL )
 172         return 0;
 173
 174     if (tools::buildPath(path, path, pathEnd, MY_STRING(L"\\program\\")) == NULL)
 175         fail();
 176     rebaseImagesInFolder(path, BASEVIRTUALADDRESS);
 177
 178     if (tools::buildPath(path, path, pathEnd, MY_STRING(L"\\ure-link")) == NULL)
 179         fail();
 180     pathEnd = tools::resolveLink(path);
 181
 182     if ( pathEnd == NULL )
 183         return 0;
 184
 185     if (tools::buildPath(path, path, pathEnd, MY_STRING(L"\\bin\\")) == NULL)
 186         fail();
 187     rebaseImagesInFolder(path, BASEVIRTUALADDRESS);
 188
 189     return 0;
 190 }