| /************************************************************** |
| * |
| * 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 <cstddef> |
| #include <stdlib.h> |
| #include <wchar.h> |
| |
| #define WIN32_LEAN_AND_MEAN |
| #if defined _MSC_VER |
| #pragma warning(push, 1) |
| #endif |
| #include <windows.h> |
| #if defined _MSC_VER |
| #pragma warning(pop) |
| #endif |
| |
| #include "tools/pathutils.hxx" |
| |
| #include "pyversion.hxx" |
| |
| #define MY_LENGTH(s) (sizeof (s) / sizeof *(s) - 1) |
| #define MY_STRING(s) (s), MY_LENGTH(s) |
| |
| wchar_t * encode(wchar_t * buffer, wchar_t const * text) { |
| *buffer++ = L'"'; |
| std::size_t n = 0; |
| for (;;) { |
| wchar_t c = *text++; |
| if (c == L'\0') { |
| break; |
| } else if (c == L'"') { |
| // Double any preceding backslashes as required by Windows: |
| for (std::size_t i = 0; i < n; ++i) { |
| *buffer++ = L'\\'; |
| } |
| *buffer++ = L'\\'; |
| *buffer++ = L'"'; |
| n = 0; |
| } else if (c == L'\\') { |
| *buffer++ = L'\\'; |
| ++n; |
| } else { |
| *buffer++ = c; |
| n = 0; |
| } |
| } |
| // The command line will continue with a double quote, so double any |
| // preceding backslashes as required by Windows: |
| for (std::size_t i = 0; i < n; ++i) { |
| *buffer++ = L'\\'; |
| } |
| *buffer++ = L'"'; |
| return buffer; |
| } |
| |
| #ifdef __MINGW32__ |
| int main(int argc, char ** argv, char **) { |
| #else |
| int wmain(int argc, wchar_t ** argv, wchar_t **) { |
| #endif |
| wchar_t path[MAX_PATH]; |
| DWORD n = GetModuleFileNameW(NULL, path, MAX_PATH); |
| if (n == 0 || n >= MAX_PATH) { |
| exit(EXIT_FAILURE); |
| } |
| wchar_t * pathEnd = tools::filename(path); |
| *pathEnd = L'\0'; |
| n = GetEnvironmentVariableW(L"UNO_PATH", NULL, 0); |
| if (n == 0) { |
| if (GetLastError() != ERROR_ENVVAR_NOT_FOUND || |
| !SetEnvironmentVariableW(L"UNO_PATH", path)) |
| { |
| exit(EXIT_FAILURE); |
| } |
| } |
| |
| wchar_t bootstrap[MY_LENGTH(L"vnd.sun.star.pathname:") + MAX_PATH] = |
| L"vnd.sun.star.pathname:"; //TODO: overflow |
| wchar_t * bootstrapEnd = tools::buildPath( |
| bootstrap + MY_LENGTH(L"vnd.sun.star.pathname:"), path, pathEnd, |
| MY_STRING(L"fundamental.ini")); |
| if (bootstrapEnd == NULL) |
| { |
| exit(EXIT_FAILURE); |
| } |
| |
| wchar_t pythonpath2[MAX_PATH]; |
| wchar_t * pythonpath2End = tools::buildPath( |
| pythonpath2, path, pathEnd, |
| MY_STRING(L"python-core-" MY_PYVERSION L"\\lib")); |
| if (pythonpath2End == NULL) { |
| exit(EXIT_FAILURE); |
| } |
| |
| wchar_t pythonpath3[MAX_PATH]; |
| wchar_t * pythonpath3End = tools::buildPath( |
| pythonpath3, path, pathEnd, |
| MY_STRING( |
| L"python-core-" MY_PYVERSION L"\\lib\\site-packages")); |
| if (pythonpath3End == NULL) { |
| exit(EXIT_FAILURE); |
| } |
| |
| #ifdef __MINGW32__ |
| wchar_t pythonpath4[MAX_PATH]; |
| wchar_t * pythonpath4End = tools::buildPath( |
| pythonpath4, path, pathEnd, |
| MY_STRING(L"python-core-" MY_PYVERSION L"\\lib\\lib-dynload")); |
| if (pythonpath4End == NULL) { |
| exit(EXIT_FAILURE); |
| } |
| wchar_t pythonpath5[MAX_PATH]; |
| wchar_t * pythonpath5End = tools::buildPath( |
| pythonpath5, path, pathEnd, |
| MY_STRING(L"python-core-" MY_PYVERSION L"\\lib\\lib-dynload")); |
| if (pythonpath5End == NULL) { |
| exit(EXIT_FAILURE); |
| } |
| #endif |
| wchar_t pythonhome[MAX_PATH]; |
| wchar_t * pythonhomeEnd = tools::buildPath( |
| pythonhome, path, pathEnd, |
| MY_STRING(L"python-core-" MY_PYVERSION)); |
| if (pythonhomeEnd == NULL) { |
| exit(EXIT_FAILURE); |
| } |
| wchar_t pythonexe[MAX_PATH]; |
| wchar_t * pythonexeEnd = tools::buildPath( |
| pythonexe, path, pathEnd, |
| #ifdef __MINGW32__ |
| MY_STRING( |
| L"python-core-" MY_PYVERSION L"\\bin\\python.bin")); |
| #else |
| MY_STRING( |
| L"python-core-" MY_PYVERSION L"\\bin\\python.exe")); |
| #endif |
| if (pythonexeEnd == NULL) { |
| exit(EXIT_FAILURE); |
| } |
| |
| std::size_t clSize = MY_LENGTH(L"\"") + 4 * (pythonexeEnd - pythonexe) + |
| MY_LENGTH(L"\"\0"); //TODO: overflow |
| // 4 * len: each char preceded by backslash, each trailing backslash |
| // doubled |
| for (int i = 1; i < argc; ++i) { |
| #ifdef __MINGW32__ |
| clSize += MY_LENGTH(L" \"") + 4 * strlen(argv[i]) + |
| #else |
| clSize += MY_LENGTH(L" \"") + 4 * wcslen(argv[i]) + |
| #endif |
| MY_LENGTH(L"\""); //TODO: overflow |
| } |
| wchar_t * cl = new wchar_t[clSize]; |
| if (cl == NULL) { |
| exit(EXIT_FAILURE); |
| } |
| wchar_t * cp = encode(cl, pythonhome); |
| for (int i = 1; i < argc; ++i) { |
| *cp++ = L' '; |
| #ifdef __MINGW32__ |
| int nNeededWStrBuffSize = MultiByteToWideChar(CP_ACP, 0, argv[i], -1, NULL, 0); |
| WCHAR *buff = new WCHAR[nNeededWStrBuffSize+1]; |
| MultiByteToWideChar(CP_ACP, 0, argv[i], -1, buff, nNeededWStrBuffSize); |
| buff[nNeededWStrBuffSize] = 0; |
| cp = encode(cp, buff); |
| delete [] buff; |
| #else |
| cp = encode(cp, argv[i]); |
| #endif |
| } |
| *cp = L'\0'; |
| n = GetEnvironmentVariableW(L"PATH", NULL, 0); |
| wchar_t * orig; |
| if (n == 0) { |
| if (GetLastError() != ERROR_ENVVAR_NOT_FOUND) { |
| exit(EXIT_FAILURE); |
| } |
| orig = L""; |
| } else { |
| orig = new wchar_t[n]; |
| if (orig == NULL || |
| GetEnvironmentVariableW(L"PATH", orig, n) != n - 1) |
| { |
| exit(EXIT_FAILURE); |
| } |
| } |
| wchar_t * value = new wchar_t[ |
| (pathEnd - path) + MY_LENGTH(L";") + |
| (n == 0 ? 0 : MY_LENGTH(L";") + (n - 1)) + 1]; //TODO: overflow |
| |
| wsprintfW(value, L"%s%s%s", path, n == 0 ? L"" : L";", orig); |
| |
| if (!SetEnvironmentVariableW(L"PATH", value)) { |
| exit(EXIT_FAILURE); |
| } |
| if (n != 0) { |
| delete [] orig; |
| } |
| delete [] value; |
| n = GetEnvironmentVariableW(L"PYTHONPATH", NULL, 0); |
| if (n == 0) { |
| if (GetLastError() != ERROR_ENVVAR_NOT_FOUND) { |
| exit(EXIT_FAILURE); |
| } |
| orig = L""; |
| } else { |
| orig = new wchar_t[n]; |
| if (orig == NULL || |
| GetEnvironmentVariableW(L"PYTHONPATH", orig, n) != n - 1) |
| { |
| exit(EXIT_FAILURE); |
| } |
| } |
| #ifdef __MINGW32__ |
| value = new wchar_t[ |
| (path1End - path1) + MY_LENGTH(L";") + (pythonpath2End - pythonpath2) + |
| MY_LENGTH(L";") + (pythonpath4End - pythonpath4) + |
| MY_LENGTH(L";") + (pythonpath5End - pythonpath5) + |
| MY_LENGTH(L";") + (pythonpath3End - pythonpath3) + |
| (n == 0 ? 0 : MY_LENGTH(L";") + (n - 1)) + 1]; //TODO: overflow |
| wsprintfW( |
| value, L"%s;%s;%s;%s;%s%s%s", path1, pythonpath2, pythonpath4, |
| pythonpath5, pythonpath3, |
| n == 0 ? L"" : L";", orig); |
| #else |
| value = new wchar_t[ |
| (pythonpath2End - pythonpath2) + |
| MY_LENGTH(L";") + (pythonpath3End - pythonpath3) + |
| (n == 0 ? 0 : MY_LENGTH(L";") + (n - 1)) + 1]; //TODO: overflow |
| wsprintfW( |
| value, L"%s;%s%s%s", pythonpath2, pythonpath3, |
| n == 0 ? L"" : L";", orig); |
| #endif |
| if (!SetEnvironmentVariableW(L"PYTHONPATH", value)) { |
| exit(EXIT_FAILURE); |
| } |
| if (n != 0) { |
| delete [] orig; |
| } |
| delete [] value; |
| if (!SetEnvironmentVariableW(L"PYTHONHOME", pythonhome)) { |
| exit(EXIT_FAILURE); |
| } |
| n = GetEnvironmentVariableW(L"URE_BOOTSTRAP", NULL, 0); |
| if (n == 0) { |
| if (GetLastError() != ERROR_ENVVAR_NOT_FOUND || |
| !SetEnvironmentVariableW(L"URE_BOOTSTRAP", bootstrap)) |
| { |
| exit(EXIT_FAILURE); |
| } |
| } |
| STARTUPINFOW startinfo; |
| ZeroMemory(&startinfo, sizeof (STARTUPINFOW)); |
| startinfo.cb = sizeof (STARTUPINFOW); |
| PROCESS_INFORMATION procinfo; |
| if (!CreateProcessW( |
| pythonexe, cl, NULL, NULL, FALSE, CREATE_UNICODE_ENVIRONMENT, NULL, |
| NULL, &startinfo, &procinfo)) { |
| exit(EXIT_FAILURE); |
| } |
| WaitForSingleObject(procinfo.hProcess,INFINITE); |
| DWORD exitStatus; |
| GetExitCodeProcess(procinfo.hProcess,&exitStatus); |
| exit(exitStatus); |
| } |
