desktop/win32/source/extendloaderenvironment.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: extendloaderenvironment.cxx,v $
  10  * $Revision: 1.7 $
  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 #include "sal/config.h"
  33
  34 #include <cstddef>
  35
  36 #define WIN32_LEAN_AND_MEAN
  37 #if defined _MSC_VER
  38 #pragma warning(push, 1)
  39 #endif
  40 #include <windows.h>
  41 #include <shlwapi.h>
  42 #if defined _MSC_VER
  43 #pragma warning(pop)
  44 #endif
  45
  46 #include "tools/pathutils.hxx"
  47
  48 #include "extendloaderenvironment.hxx"
  49
  50 namespace {
  51
  52 void fail() {
  53     LPWSTR buf = NULL;
  54     FormatMessageW(
  55         FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, NULL,
  56         GetLastError(), 0, reinterpret_cast< LPWSTR >(&buf), 0, NULL);
  57     MessageBoxW(NULL, buf, NULL, MB_OK | MB_ICONERROR);
  58     LocalFree(buf);
  59     TerminateProcess(GetCurrentProcess(), 255);
  60 }
  61
  62 bool contains(WCHAR const * paths, WCHAR const * path, WCHAR const * pathEnd) {
  63     WCHAR const * q = path;
  64     for (WCHAR const * p = paths;; ++p) {
  65         WCHAR c = *p;
  66         switch (c) {
  67         case L'\0':
  68             return q == pathEnd;
  69         case L';':
  70             if (q == pathEnd) {
  71                 return true;
  72             }
  73             q = path;
  74             break;
  75         default:
  76             if (q != NULL) {
  77                 if (q != pathEnd && *q == c) {
  78                     ++q;
  79                 } else {
  80                     q = NULL;
  81                 }
  82             }
  83             break;
  84         }
  85     }
  86 }
  87
  88 }
  89
  90 namespace desktop_win32 {
  91
  92 void extendLoaderEnvironment(WCHAR * binPath, WCHAR * iniDirectory) {
  93     if (!GetModuleFileNameW(NULL, iniDirectory, MAX_PATH)) {
  94         fail();
  95     }
  96     WCHAR * iniDirEnd = tools::filename(iniDirectory);
  97     WCHAR name[MAX_PATH + MY_LENGTH(L".bin")];
  98         // hopefully std::size_t is large enough to not overflow
  99     WCHAR * nameEnd = name;
 100     for (WCHAR * p = iniDirEnd; *p != L'\0'; ++p) {
 101         *nameEnd++ = *p;
 102     }
 103     if (!(nameEnd - name >= 4 && nameEnd[-4] == L'.' &&
 104           (nameEnd[-3] == L'E' || nameEnd[-3] == L'e') &&
 105           (nameEnd[-2] == L'X' || nameEnd[-2] == L'x') &&
 106           (nameEnd[-1] == L'E' || nameEnd[-1] == L'e')))
 107     {
 108         *nameEnd = L'.';
 109         nameEnd += 4;
 110     }
 111     nameEnd[-3] = 'b';
 112     nameEnd[-2] = 'i';
 113     nameEnd[-1] = 'n';
 114     tools::buildPath(binPath, iniDirectory, iniDirEnd, name, nameEnd - name);
 115     *iniDirEnd = L'\0';
 116     WCHAR path[MAX_PATH];
 117     WCHAR * pathEnd = tools::buildPath(
 118         path, iniDirectory, iniDirEnd, MY_STRING(L"..\\basis-link"));
 119     if (pathEnd == NULL) {
 120         fail();
 121     }
 122     std::size_t const maxEnv = 32767;
 123     WCHAR pad[2 * MAX_PATH + maxEnv];
 124         // hopefully std::size_t is large enough to not overflow
 125     WCHAR * padEnd = NULL;
 126     WCHAR env[maxEnv];
 127     DWORD n = GetEnvironmentVariableW(L"PATH", env, maxEnv);
 128     if (n >= maxEnv || n == 0 && GetLastError() != ERROR_ENVVAR_NOT_FOUND) {
 129         fail();
 130     }
 131     env[n] = L'\0';
 132     bool exclude1;
 133     pathEnd = tools::resolveLink(path);
 134     if (pathEnd == NULL) {
 135         if (GetLastError() != ERROR_FILE_NOT_FOUND) {
 136             fail();
 137         }
 138         // This path is only taken by testtool.exe in basis program directory;
 139         // its PATH needs to include the brand program directory:
 140         pathEnd = tools::buildPath(
 141             path, iniDirectory, iniDirEnd, MY_STRING(L".."));
 142         if (pathEnd == NULL) {
 143             fail();
 144         }
 145         padEnd = tools::buildPath(
 146             pad, path, pathEnd, MY_STRING(L"\\..\\program"));
 147         if (padEnd == NULL) {
 148             fail();
 149         }
 150         exclude1 = contains(env, pad, padEnd);
 151     } else {
 152         exclude1 = true;
 153     }
 154     WCHAR * pad2 = exclude1 ? pad : padEnd + 1;
 155     pathEnd = tools::buildPath(path, path, pathEnd, MY_STRING(L"\\ure-link"));
 156     if (pathEnd == NULL) {
 157         fail();
 158     }
 159     pathEnd = tools::resolveLink(path);
 160     if (pathEnd == NULL) {
 161         fail();
 162     }
 163     padEnd = tools::buildPath(pad2, path, pathEnd, MY_STRING(L"\\bin"));
 164     if (padEnd == NULL) {
 165         fail();
 166     }
 167     bool exclude2 = contains(env, pad2, padEnd);
 168     if (!(exclude1 && exclude2)) {
 169         if (!(exclude1 || exclude2)) {
 170             pad2[-1] = L';';
 171         }
 172         WCHAR * p = exclude2 ? pad2 - 1 : padEnd;
 173         if (n != 0) {
 174             *p++ = L';';
 175         }
 176         for (DWORD i = 0; i <= n; ++i) {
 177             *p++ = env[i];
 178         }
 179         if (!SetEnvironmentVariableW(L"PATH", pad)) {
 180             fail();
 181         }
 182     }
 183 }
 184
 185 }