1 files changed, 130 insertions, 0 deletions
diff --git a/setup_native/source/win32/customactions/shellextensions/copyeditiondata.cxx b/setup_native/source/win32/customactions/shellextensions/copyeditiondata.cxx
new file mode 100644
index 000000000000..7dc5194a18cb
--- /dev/null
+++ b/setup_native/source/win32/customactions/shellextensions/copyeditiondata.cxx
@@ -0,0 +1,130 @@
+/*************************************************************************
+ *
+ * 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 "sal/config.h"
+
+#include <cstddef>
+#include <new>
+#include <string.h> // <cstring> not supported by old MSC versions
+
+#define WIN32_LEAN_AND_MEAN
+#if defined _MSC_VER
+#pragma warning(push, 1)
+#endif
+#include <windows.h>
+#include <msiquery.h>
+#include <shellapi.h>
+#if defined _MSC_VER
+#pragma warning(pop)
+#endif
+
+#include "boost/scoped_array.hpp"
+
+#define LCL_LENGTH0(s) (sizeof (s) / sizeof *(s))
+#define LCL_STRING0(s) (s), LCL_LENGTH0(s)
+
+namespace {
+
+enum Status { STATUS_NO, STATUS_YES, STATUS_ERROR };
+
+Status fileExists(wchar_t const * path) {
+    return GetFileAttributesW(path) == INVALID_FILE_ATTRIBUTES
+        ? GetLastError() == ERROR_FILE_NOT_FOUND ? STATUS_NO : STATUS_ERROR
+        : STATUS_YES;
+}
+
+wchar_t * getProperty(
+    MSIHANDLE install, wchar_t const * name, wchar_t const * suffix,
+    std::size_t suffixLength, wchar_t ** end = NULL)
+{
+    DWORD n = 0;
+    UINT err = MsiGetPropertyW(install, name, L"", &n);
+    if (err != ERROR_SUCCESS && err != ERROR_MORE_DATA) {
+        return NULL;
+    }
+    DWORD n2 = n + suffixLength; //TODO: overflow
+    wchar_t * data = new(std::nothrow) wchar_t[n2];
+    if (data == NULL) {
+        return NULL;
+    }
+    if (MsiGetPropertyW(install, name, data, &n2) != ERROR_SUCCESS || n2 != n) {
+        delete[] data;
+        return NULL;
+    }
+    memcpy(data + n, suffix, suffixLength * sizeof (wchar_t)); //TODO: overflow
+    if (end != NULL) {
+        *end = data + n + suffixLength;
+    }
+    return data;
+}
+
+}
+
+extern "C" UINT __stdcall copyEditionData(MSIHANDLE install) {
+    boost::scoped_array<wchar_t> from(
+        getProperty(install, L"SourceDir", LCL_STRING0(L"edition\0")));
+    if (!from) {
+        return ERROR_INSTALL_FAILURE;
+    }
+    Status stat = fileExists(from.get());
+    if (stat == STATUS_ERROR) {
+        return ERROR_INSTALL_FAILURE;
+    }
+    if (stat == STATUS_NO) {
+        return ERROR_SUCCESS;
+    }
+    wchar_t * end;
+    boost::scoped_array<wchar_t> to(
+        getProperty(
+            install, L"INSTALLLOCATION",
+            LCL_STRING0(L"program\\edition\0"), &end));
+    if (!to) {
+        return ERROR_INSTALL_FAILURE;
+    }
+    stat = fileExists(to.get());
+    if (stat == STATUS_ERROR) {
+        return ERROR_INSTALL_FAILURE;
+    }
+    if (stat == STATUS_YES) {
+        SHFILEOPSTRUCTW opDelete = {
+            NULL, FO_DELETE, to.get(), NULL, FOF_NOCONFIRMATION | FOF_SILENT,
+            FALSE, NULL, NULL }; //TODO: non-NULL hwnd
+        if (SHFileOperationW(&opDelete) != 0) {
+            return ERROR_INSTALL_FAILURE;
+        }
+    }
+    *(end - LCL_LENGTH0(L"\\edition\0")) = L'\0';
+    *(end - LCL_LENGTH0(L"\\edition\0") + 1) = L'\0';
+    SHFILEOPSTRUCTW opCopy = {
+        NULL, FO_COPY, from.get(), to.get(),
+        FOF_NOCONFIRMATION | FOF_NOCONFIRMMKDIR | FOF_SILENT, FALSE, NULL,
+        NULL }; //TODO: non-NULL hwnd
+    if (SHFileOperationW(&opCopy) != 0) {
+        return ERROR_INSTALL_FAILURE;
+    }
+    return ERROR_SUCCESS;
+}