| /************************************************************** |
| * |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| * |
| *************************************************************/ |
| |
| |
| |
| #include "precompiled_desktop.hxx" |
| #include "sal/config.h" |
| |
| #include <cstddef> |
| |
| #define WIN32_LEAN_AND_MEAN |
| #if defined _MSC_VER |
| #pragma warning(push, 1) |
| #endif |
| #include <windows.h> |
| #include <shlwapi.h> |
| #if defined _MSC_VER |
| #pragma warning(pop) |
| #endif |
| |
| #include "tools/pathutils.hxx" |
| |
| #include "extendloaderenvironment.hxx" |
| |
| namespace { |
| |
| void fail() { |
| LPWSTR buf = NULL; |
| FormatMessageW( |
| FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, NULL, |
| GetLastError(), 0, reinterpret_cast< LPWSTR >(&buf), 0, NULL); |
| MessageBoxW(NULL, buf, NULL, MB_OK | MB_ICONERROR); |
| LocalFree(buf); |
| TerminateProcess(GetCurrentProcess(), 255); |
| } |
| |
| bool contains(WCHAR const * paths, WCHAR const * path, WCHAR const * pathEnd) { |
| WCHAR const * q = path; |
| for (WCHAR const * p = paths;; ++p) { |
| WCHAR c = *p; |
| switch (c) { |
| case L'\0': |
| return q == pathEnd; |
| case L';': |
| if (q == pathEnd) { |
| return true; |
| } |
| q = path; |
| break; |
| default: |
| if (q != NULL) { |
| if (q != pathEnd && *q == c) { |
| ++q; |
| } else { |
| q = NULL; |
| } |
| } |
| break; |
| } |
| } |
| } |
| |
| } |
| |
| namespace desktop_win32 { |
| |
| void extendLoaderEnvironment(WCHAR * binPath, WCHAR * iniDirectory) { |
| if (!GetModuleFileNameW(NULL, iniDirectory, MAX_PATH)) { |
| fail(); |
| } |
| WCHAR * iniDirEnd = tools::filename(iniDirectory); |
| WCHAR name[MAX_PATH + MY_LENGTH(L".bin")]; |
| // hopefully std::size_t is large enough to not overflow |
| WCHAR * nameEnd = name; |
| for (WCHAR * p = iniDirEnd; *p != L'\0'; ++p) { |
| *nameEnd++ = *p; |
| } |
| if (!(nameEnd - name >= 4 && nameEnd[-4] == L'.' && |
| (nameEnd[-3] == L'E' || nameEnd[-3] == L'e') && |
| (nameEnd[-2] == L'X' || nameEnd[-2] == L'x') && |
| (nameEnd[-1] == L'E' || nameEnd[-1] == L'e'))) |
| { |
| *nameEnd = L'.'; |
| nameEnd += 4; |
| } |
| nameEnd[-3] = 'b'; |
| nameEnd[-2] = 'i'; |
| nameEnd[-1] = 'n'; |
| tools::buildPath(binPath, iniDirectory, iniDirEnd, name, nameEnd - name); |
| *iniDirEnd = L'\0'; |
| WCHAR path[MAX_PATH]; |
| WCHAR * pathEnd = tools::buildPath( |
| path, iniDirectory, iniDirEnd, MY_STRING(L"..\\basis-link")); |
| if (pathEnd == NULL) { |
| fail(); |
| } |
| std::size_t const maxEnv = 32767; |
| WCHAR pad[2 * MAX_PATH + maxEnv]; |
| // hopefully std::size_t is large enough to not overflow |
| WCHAR * padEnd = NULL; |
| WCHAR env[maxEnv]; |
| DWORD n = GetEnvironmentVariableW(L"PATH", env, maxEnv); |
| if (n >= maxEnv || n == 0 && GetLastError() != ERROR_ENVVAR_NOT_FOUND) { |
| fail(); |
| } |
| env[n] = L'\0'; |
| bool exclude1; |
| pathEnd = tools::resolveLink(path); |
| if (pathEnd == NULL) { |
| if (GetLastError() != ERROR_FILE_NOT_FOUND) { |
| fail(); |
| } |
| // This path is only taken by testtool.exe in basis program directory; |
| // its PATH needs to include the brand program directory: |
| pathEnd = tools::buildPath( |
| path, iniDirectory, iniDirEnd, MY_STRING(L"..")); |
| if (pathEnd == NULL) { |
| fail(); |
| } |
| padEnd = tools::buildPath( |
| pad, path, pathEnd, MY_STRING(L"\\..\\program")); |
| if (padEnd == NULL) { |
| fail(); |
| } |
| exclude1 = contains(env, pad, padEnd); |
| } else { |
| exclude1 = true; |
| } |
| WCHAR * pad2 = exclude1 ? pad : padEnd + 1; |
| // pathEnd = tools::buildPath(path, path, pathEnd, MY_STRING(L"\\ure-link")); |
| pathEnd = tools::buildPath(path, path, pathEnd, MY_STRING(L"")); |
| if (pathEnd == NULL) { |
| fail(); |
| } |
| // pathEnd = tools::resolveLink(path); |
| // if (pathEnd == NULL) { |
| // fail(); |
| // } |
| // padEnd = tools::buildPath(pad2, path, pathEnd, MY_STRING(L"\\bin")); |
| // if (padEnd == NULL) { |
| // fail(); |
| // } |
| bool exclude2 = contains(env, pad2, padEnd); |
| if (!(exclude1 && exclude2)) { |
| if (!(exclude1 || exclude2)) { |
| pad2[-1] = L';'; |
| } |
| WCHAR * p = exclude2 ? pad2 - 1 : padEnd; |
| if (n != 0) { |
| *p++ = L';'; |
| } |
| for (DWORD i = 0; i <= n; ++i) { |
| *p++ = env[i]; |
| } |
| if (!SetEnvironmentVariableW(L"PATH", pad)) { |
| fail(); |
| } |
| } |
| } |
| |
| } |