Files
libreoffice/pyuno/zipcore/python.cxx

287 lines
9.2 KiB
C++

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/*************************************************************************
*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* Copyright 2000, 2010 Oracle and/or its affiliates.
*
* OpenOffice.org - a multi-platform office productivity suite
*
* This file is part of OpenOffice.org.
*
* OpenOffice.org is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License version 3
* only, as published by the Free Software Foundation.
*
* OpenOffice.org is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License version 3 for more details
* (a copy is included in the LICENSE file that accompanied this code).
*
* You should have received a copy of the GNU Lesser General Public License
* version 3 along with OpenOffice.org. If not, see
* <http://www.openoffice.org/license.html>
* for a copy of the LGPLv3 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);
}
#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);
}
wchar_t urepath[MAX_PATH];
if (tools::buildPath(urepath, path, pathEnd, MY_STRING(L"..\\ure-link"))
== NULL)
{
exit(EXIT_FAILURE);
}
wchar_t * urepathEnd = tools::resolveLink(urepath);
if (urepathEnd == NULL) {
exit(EXIT_FAILURE);
}
urepathEnd = tools::buildPath(
urepath, urepath, urepathEnd, MY_STRING(L"\\bin"));
if (urepathEnd == 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[
(urepathEnd - urepath) + MY_LENGTH(L";") + (pathEnd - path) +
(n == 0 ? 0 : MY_LENGTH(L";") + (n - 1)) + 1]; //TODO: overflow
wsprintfW(value, L"%s;%s%s%s", urepath, 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[
(pathEnd - path) + MY_LENGTH(L";") + (pythonpath2End - pythonpath2) +
MY_LENGTH(L";") + (pythonpath4End - pythonpath4) +
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", path, pythonpath2, pythonpath4, pythonpath3,
n == 0 ? L"" : L";", orig);
#else
value = new wchar_t[
(pathEnd - path) + MY_LENGTH(L";") + (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%s", path, 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);
}
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */