Adding upstream version 1:10.11.6.upstream/1%10.11.6

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 1134 insertions, 0 deletions

diff --git a/win/packaging/ca/CustomAction.cpp b/win/packaging/ca/CustomAction.cpp
new file mode 100644
index 00000000..c397ce23
--- /dev/null
+++ b/win/packaging/ca/CustomAction.cpp
@@ -0,0 +1,1134 @@
+/* Copyright 2010, Oracle and/or its affiliates. All rights reserved.
+
+This program is free software; you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation; version 2 of the License.
+
+This program 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 General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program; if not, write to the Free Software
+Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335 USA */
+
+#ifndef UNICODE
+#define UNICODE
+#endif
+
+#undef NOMINMAX
+
+#include <winsock2.h>
+#include <ws2tcpip.h>
+#include <windows.h>
+#include <winreg.h>
+#include <msi.h>
+#include <msiquery.h>
+#include <wcautil.h>
+#include <strutil.h>
+#include <string.h>
+#include <strsafe.h>
+#include <assert.h>
+#include <shellapi.h>
+#include <stdlib.h>
+#include <winservice.h>
+#include <string>
+#include <iostream>
+#include <fstream>
+#include <sstream>
+#include <vector>
+#include <map>
+
+using namespace std;
+
+
+#define ONE_MB 1048576
+UINT ExecRemoveDataDirectory(wchar_t *dir)
+{
+   /* Strip stray backslash */
+  DWORD len = (DWORD)wcslen(dir);
+  if(len > 0 && dir[len-1]==L'\\')
+    dir[len-1] = 0;
+
+  SHFILEOPSTRUCTW fileop;
+  fileop.hwnd= NULL;    /* no status display */
+  fileop.wFunc= FO_DELETE;  /* delete operation */
+  fileop.pFrom= dir;  /* source file name as double null terminated string */
+  fileop.pTo= NULL;    /* no destination needed */
+  fileop.fFlags= FOF_NOCONFIRMATION|FOF_SILENT;  /* do not prompt the user */
+
+  fileop.fAnyOperationsAborted= FALSE;
+  fileop.lpszProgressTitle= NULL;
+  fileop.hNameMappings= NULL;
+
+  return SHFileOperationW(&fileop);
+}
+
+
+extern "C" UINT __stdcall RemoveDataDirectory(MSIHANDLE hInstall)
+{
+  HRESULT hr = S_OK;
+  UINT er = ERROR_SUCCESS;
+  wchar_t dir[MAX_PATH];
+  DWORD len = MAX_PATH;
+
+  hr = WcaInitialize(hInstall, __FUNCTION__);
+  ExitOnFailure(hr, "Failed to initialize");
+  WcaLog(LOGMSG_STANDARD, "Initialized.");
+
+  MsiGetPropertyW(hInstall, L"CustomActionData", dir, &len);
+
+  er= ExecRemoveDataDirectory(dir);
+  WcaLog(LOGMSG_STANDARD, "SHFileOperation returned %d", er);
+LExit:
+  return WcaFinalize(er);
+}
+
+/*
+  Escape command line parameter fpr pass to CreateProcess().
+
+  We assume out has enough space to include encoded string
+  2*wcslen(in) is enough.
+
+  It is assumed that called will add double quotation marks before and after
+  the string.
+*/
+static void EscapeCommandLine(const wchar_t *in, wchar_t *out, size_t buflen)
+{
+  const wchar_t special_chars[]=L" \t\n\v\"";
+  bool needs_escaping= false;
+  size_t pos;
+
+  for(size_t i=0; i< sizeof(special_chars) -1; i++)
+  {
+    if (wcschr(in, special_chars[i]))
+    {
+      needs_escaping = true;
+      break;
+    }
+  }
+
+  if(!needs_escaping)
+  {
+    wcscpy_s(out, buflen, in);
+    return;
+  }
+
+  pos= 0;
+  for(int i = 0 ; ; i++)
+  {
+    size_t n_backslashes = 0;
+    wchar_t c;
+    while (in[i] == L'\\')
+    {
+      i++;
+      n_backslashes++;
+    }
+
+    c= in[i];
+    if (c == 0)
+    {
+      /*
+        Escape all backslashes, but let the terminating double quotation mark
+        that caller adds be interpreted as a metacharacter.
+      */
+      for(size_t j= 0; j < 2*n_backslashes;j++)
+      {
+        out[pos++]=L'\\';
+      }
+      break;
+    }
+    else if (c == L'"')
+    {
+      /*
+        Escape all backslashes and the following double quotation mark.
+      */
+      for(size_t j= 0; j < 2*n_backslashes + 1; j++)
+      {
+        out[pos++]=L'\\';
+      }
+      out[pos++]= L'"';
+    }
+    else
+    {
+      /* Backslashes aren't special here. */
+      for (size_t j=0; j < n_backslashes; j++)
+        out[pos++] = L'\\';
+
+      out[pos++]= c;
+    }
+  }
+  out[pos++]= 0;
+}
+
+bool IsDirectoryEmptyOrNonExisting(const wchar_t *dir) {
+  wchar_t wildcard[MAX_PATH+3];
+  WIN32_FIND_DATAW data;
+  HANDLE h;
+  wcscpy_s(wildcard, MAX_PATH, dir);
+  wcscat_s(wildcard, MAX_PATH, L"*.*");
+  bool empty= true;
+  h= FindFirstFile(wildcard, &data);
+  if (h != INVALID_HANDLE_VALUE)
+  {
+    for (;;)
+    {
+      if (wcscmp(data.cFileName, L".") && wcscmp(data.cFileName, L".."))
+      {
+        empty= false;
+        break;
+      }
+      if (!FindNextFile(h, &data))
+        break;
+    }
+    FindClose(h);
+  }
+  return empty;
+}
+
+extern "C" UINT __stdcall CheckInstallDirectory(MSIHANDLE hInstall)
+{
+  HRESULT hr= S_OK;
+  UINT er= ERROR_SUCCESS;
+  wchar_t *path= 0;
+
+  hr= WcaInitialize(hInstall, __FUNCTION__);
+  ExitOnFailure(hr, "Failed to initialize");
+  WcaGetFormattedString(L"[INSTALLDIR]", &path);
+  if (!IsDirectoryEmptyOrNonExisting(path))
+  {
+    wchar_t msg[2*MAX_PATH];
+    swprintf(msg,countof(msg), L"Installation directory '%s' exists and is not empty. Choose a "
+                  "different install directory",path);
+    WcaSetProperty(L"INSTALLDIRERROR", msg);
+    goto LExit;
+  }
+
+  WcaSetProperty(L"INSTALLDIRERROR", L"");
+
+LExit:
+  ReleaseStr(path);
+  return WcaFinalize(er);
+}
+
+/*
+  Check for valid data directory is empty during install
+  A valid data directory is non-existing, or empty.
+
+  In addition, it must be different from any directories that
+  are going to be installed. This is required. because the full
+  directory is removed on a feature uninstall, and we do not want
+  it to be lib or bin.
+*/
+extern "C" UINT __stdcall CheckDataDirectory(MSIHANDLE hInstall)
+{
+  HRESULT hr= S_OK;
+  UINT er= ERROR_SUCCESS;
+  wchar_t datadir[MAX_PATH];
+  DWORD len= MAX_PATH;
+  bool empty;
+  wchar_t *path= 0;
+
+  MsiGetPropertyW(hInstall, L"DATADIR", datadir, &len);
+  hr= WcaInitialize(hInstall, __FUNCTION__);
+  ExitOnFailure(hr, "Failed to initialize");
+  WcaLog(LOGMSG_STANDARD, "Initialized.");
+
+  WcaLog(LOGMSG_STANDARD, "Checking files in %S", datadir);
+  empty= IsDirectoryEmptyOrNonExisting(datadir);
+  if (empty)
+    WcaLog(LOGMSG_STANDARD, "DATADIR is empty or non-existent");
+  else
+    WcaLog(LOGMSG_STANDARD, "DATADIR is NOT empty");
+
+  if (!empty)
+  {
+    WcaSetProperty(L"DATADIRERROR", L"data directory exist and not empty");
+    goto LExit;
+  }
+  WcaSetProperty(L"DATADIRERROR", L"");
+
+
+  WcaGetFormattedString(L"[INSTALLDIR]",&path);
+  if (path && !wcsicmp(datadir, path))
+  {
+    WcaSetProperty(L"DATADIRERROR", L"data directory can not be "
+                                    L"installation root directory");
+    ReleaseStr(path);
+    goto LExit;
+  }
+  for (auto dir :
+       {L"[INSTALLDIR]bin\\", L"[INSTALLDIR]include\\",
+        L"[INSTALLDIR]lib\\", L"[INSTALLDIR]share\\"})
+  {
+    WcaGetFormattedString(dir, &path);
+    if (path && !wcsnicmp(datadir, path, wcslen(path)))
+    {
+      const wchar_t *subdir= dir + sizeof("[INSTALLDIR]") - 1;
+      wchar_t msg[MAX_PATH]= L"data directory conflicts with '";
+      wcsncat_s(msg, subdir, wcslen(subdir) - 1);
+      wcscat_s(msg, L"' directory, which is part of this installation");
+      WcaSetProperty(L"DATADIRERROR", msg);
+      ReleaseStr(path);
+      goto LExit;
+    }
+    ReleaseStr(path);
+    path= 0;
+  }
+LExit:
+  return WcaFinalize(er);
+}
+
+
+bool CheckServiceExists(const wchar_t *name)
+{
+   SC_HANDLE manager =0, service=0;
+   manager = OpenSCManager( NULL, NULL, SC_MANAGER_CONNECT);
+   if (!manager)
+   {
+     return false;
+   }
+
+   service = OpenService(manager, name, SC_MANAGER_CONNECT);
+   if(service)
+     CloseServiceHandle(service);
+   CloseServiceHandle(manager);
+
+   return service?true:false;
+}
+
+/* User in rollback of create database custom action */
+bool ExecRemoveService(const wchar_t *name)
+{
+   SC_HANDLE manager =0, service=0;
+   manager = OpenSCManager( NULL, NULL, SC_MANAGER_ALL_ACCESS);
+   bool ret;
+   if (!manager)
+   {
+     return false;
+   }
+   service = OpenService(manager, name, DELETE);
+   if(service)
+   {
+     ret= DeleteService(service);
+   }
+   else
+   {
+     ret= false;
+   }
+   CloseServiceHandle(manager);
+   return ret;
+}
+
+/* Find whether TCP port is in use by trying to bind to the port. */
+static bool IsPortInUse(unsigned short port)
+{
+  struct addrinfo* ai, * a;
+  struct addrinfo hints {};
+
+  char port_buf[NI_MAXSERV];
+  SOCKET ip_sock = INVALID_SOCKET;
+  hints.ai_flags = AI_PASSIVE;
+  hints.ai_socktype = SOCK_STREAM;
+  hints.ai_family = AF_UNSPEC;
+  snprintf(port_buf, NI_MAXSERV, "%u", (unsigned)port);
+
+  if (getaddrinfo(NULL, port_buf, &hints, &ai))
+  {
+    return false;
+  }
+
+  /*
+   Prefer IPv6 socket to IPv4, since we'll use IPv6 dual socket,
+   which coveres both IP versions.
+  */
+  for (a = ai; a; a = a->ai_next)
+  {
+    if (a->ai_family == AF_INET6 &&
+      (ip_sock = socket(a->ai_family, a->ai_socktype, a->ai_protocol)) != INVALID_SOCKET)
+    {
+      break;
+    }
+  }
+
+  if (ip_sock == INVALID_SOCKET)
+  {
+    for (a = ai; a; a = a->ai_next)
+    {
+      if (ai->ai_family == AF_INET &&
+        (ip_sock = socket(a->ai_family, a->ai_socktype, a->ai_protocol)) != INVALID_SOCKET)
+      {
+        break;
+      }
+    }
+  }
+
+  if (ip_sock == INVALID_SOCKET)
+  {
+    return false;
+  }
+
+  /* Use SO_EXCLUSIVEADDRUSE to prevent multiple binding. */
+  int arg = 1;
+  setsockopt(ip_sock, SOL_SOCKET, SO_EXCLUSIVEADDRUSE, (char*)&arg, sizeof(arg));
+
+  /* Allow dual socket, so that IPv4 and IPv6 are both covered.*/
+  if (a->ai_family == AF_INET6)
+  {
+    arg = 0;
+    setsockopt(ip_sock, IPPROTO_IPV6, IPV6_V6ONLY, (char*)&arg, sizeof(arg));
+  }
+
+  bool in_use = false;
+  if (bind(ip_sock, a->ai_addr, (int)a->ai_addrlen) == SOCKET_ERROR)
+  {
+    DWORD last_error = WSAGetLastError();
+    in_use = (last_error ==  WSAEADDRINUSE || last_error == WSAEACCES);
+  }
+
+  freeaddrinfo(ai);
+  closesocket(ip_sock);
+  return in_use;
+}
+
+
+/*
+  Check if TCP port is free
+*/
+bool IsPortFree(unsigned short port)
+{
+  WORD wVersionRequested = MAKEWORD(2, 2);
+  WSADATA wsaData;
+  WSAStartup(wVersionRequested, &wsaData);
+  bool in_use = IsPortInUse(port);
+  WSACleanup();
+  return !in_use;
+}
+
+
+/*
+   Helper function used in filename normalization.
+   Removes leading quote and terminates string at the position of the next one
+   (if applicable, does not change string otherwise). Returns modified string
+*/
+wchar_t *strip_quotes(wchar_t *s)
+{
+   if (s && (*s == L'"'))
+   {
+     s++;
+     wchar_t *p = wcschr(s, L'"');
+     if(p)
+       *p = 0;
+   }
+   return s;
+}
+
+
+/*
+  Checks  for consistency of service configuration.
+
+  It can happen that SERVICENAME or DATADIR
+  MSI properties are in inconsistent state after somebody upgraded database
+  We catch this case during uninstall. In particular, either service is not
+  removed even if SERVICENAME was set (but this name is reused by someone else)
+  or data directory is not removed (if it is used by someone else). To find out
+  whether service name and datadirectory are in use For every service,
+  configuration is read and checked as follows:
+
+  - look if a service has to do something with mysql
+  - If so, check its name against SERVICENAME. if match, check binary path
+    against INSTALLDIR\bin.  If binary path does not match, then service runs
+    under different  installation and won't be removed.
+  - Check options file for datadir and look if this is inside this
+    installation's datadir don't remove datadir if this is the case.
+
+  "Don't remove" in this context means that custom action is removing
+  SERVICENAME property or CLEANUPDATA property, which later on in course of
+  installation mean, that either datadir or service is kept.
+*/
+
+void CheckServiceConfig(
+  wchar_t *my_servicename,    /* SERVICENAME property in this installation*/
+  wchar_t *datadir,           /* DATADIR property in this installation*/
+  wchar_t *bindir,            /* INSTALLDIR\bin */
+  wchar_t *other_servicename, /* Service to check against */
+  QUERY_SERVICE_CONFIGW * config /* Other service's config */
+  )
+{
+
+  bool same_bindir = false;
+  wchar_t * commandline= config->lpBinaryPathName;
+  int numargs;
+  wchar_t **argv= CommandLineToArgvW(commandline, &numargs);
+  wchar_t current_datadir_buf[MAX_PATH]={0};
+  wchar_t normalized_current_datadir[MAX_PATH+1];
+  wchar_t *current_datadir;
+  wchar_t *defaults_file;
+  bool is_my_service;
+
+  WcaLog(LOGMSG_VERBOSE, "CommandLine= %S", commandline);
+  if(!argv  ||  !argv[0]  || ! wcsstr(argv[0], L"mysqld"))
+  {
+    goto end;
+  }
+
+  WcaLog(LOGMSG_STANDARD, "MySQL/MariaDB service %S found: CommandLine= %S",
+    other_servicename, commandline);
+  if (wcsstr(argv[0], bindir))
+  {
+    WcaLog(LOGMSG_STANDARD, "executable under bin directory");
+    same_bindir = true;
+  }
+
+  is_my_service = (_wcsicmp(my_servicename, other_servicename) == 0);
+  if(!is_my_service)
+  {
+    WcaLog(LOGMSG_STANDARD, "service does not match current service");
+    /*
+      TODO probably the best thing possible would be to add temporary
+      row to MSI ServiceConfig table with remove on uninstall
+    */
+  }
+  else if (!same_bindir)
+  {
+    WcaLog(LOGMSG_STANDARD,
+      "Service name matches, but not the executable path directory, mine is %S",
+      bindir);
+    WcaSetProperty(L"SERVICENAME", L"");
+  }
+
+  /* Check if data directory is used */
+  if(!datadir || numargs <= 1 ||  wcsncmp(argv[1],L"--defaults-file=",16) != 0)
+  {
+    goto end;
+  }
+
+  current_datadir= current_datadir_buf;
+  defaults_file= argv[1]+16;
+  defaults_file= strip_quotes(defaults_file);
+
+  WcaLog(LOGMSG_STANDARD, "parsed defaults file is %S", defaults_file);
+
+  if (GetPrivateProfileStringW(L"mysqld", L"datadir", NULL, current_datadir,
+    MAX_PATH, defaults_file) == 0)
+  {
+    WcaLog(LOGMSG_STANDARD,
+      "Cannot find datadir in ini file '%S'", defaults_file);
+    goto end;
+  }
+
+  WcaLog(LOGMSG_STANDARD, "datadir from defaults-file is %S", current_datadir);
+  strip_quotes(current_datadir);
+
+  /* Convert to Windows path */
+  if (GetFullPathNameW(current_datadir, MAX_PATH, normalized_current_datadir,
+    NULL))
+  {
+    /* Add backslash to be compatible with directory formats in MSI */
+    wcsncat(normalized_current_datadir, L"\\", MAX_PATH+1);
+    WcaLog(LOGMSG_STANDARD, "normalized current datadir is '%S'",
+      normalized_current_datadir);
+  }
+
+  if (_wcsicmp(datadir, normalized_current_datadir) == 0 && !same_bindir)
+  {
+    WcaLog(LOGMSG_STANDARD,
+      "database directory from current installation, but different mysqld.exe");
+    WcaSetProperty(L"CLEANUPDATA", L"");
+  }
+
+end:
+  LocalFree((HLOCAL)argv);
+}
+
+/*
+  Checks if database directory or service are modified by user
+  For example, service may point to different mysqld.exe that it was originally
+  installed, or some different service might use this database directory. This
+  would normally mean user has done an upgrade of the database and in this case
+  uninstall should neither delete service nor database directory.
+
+  If this function find that service is modified by user (mysqld.exe used by
+  service does not point to the installation bin directory), MSI public variable
+  SERVICENAME is removed, if DATADIR is used by some other service, variables
+  DATADIR and CLEANUPDATA are removed.
+
+  The effect of variable removal is that service does not get uninstalled and
+  datadir is not touched by uninstallation.
+
+  Note that this function is running without elevation and does not use anything
+  that would require special privileges.
+
+*/
+extern "C" UINT CheckDBInUse(MSIHANDLE hInstall)
+{
+  static BYTE buf[256*1024]; /* largest possible buffer for EnumServices */
+  static char config_buffer[8*1024]; /*largest buffer for QueryServiceConfig */
+  HRESULT hr = S_OK;
+  UINT er = ERROR_SUCCESS;
+  wchar_t *servicename= NULL;
+  wchar_t *datadir= NULL;
+  wchar_t *bindir=NULL;
+
+  SC_HANDLE scm    = NULL;
+  ULONG  bufsize   = sizeof(buf);
+  ULONG  bufneed   = 0x00;
+  ULONG  num_services = 0x00;
+  LPENUM_SERVICE_STATUS_PROCESS info = NULL;
+  BOOL ok;
+
+  hr = WcaInitialize(hInstall, __FUNCTION__);
+  ExitOnFailure(hr, "Failed to initialize");
+  WcaLog(LOGMSG_STANDARD, "Initialized.");
+
+  WcaGetProperty(L"SERVICENAME", &servicename);
+  WcaGetProperty(L"DATADIR", &datadir);
+  WcaGetFormattedString(L"[INSTALLDIR]bin\\", &bindir);
+
+  WcaLog(LOGMSG_STANDARD,"SERVICENAME=%S, DATADIR=%S, bindir=%S",
+    servicename, datadir, bindir);
+
+  scm = OpenSCManager(NULL, NULL,
+    SC_MANAGER_ENUMERATE_SERVICE | SC_MANAGER_CONNECT);
+  if (scm == NULL)
+  {
+    ExitOnFailure(E_FAIL, "OpenSCManager failed");
+  }
+
+  ok = EnumServicesStatusExW(  scm,
+    SC_ENUM_PROCESS_INFO,
+    SERVICE_WIN32,
+    SERVICE_STATE_ALL,
+    buf,
+    bufsize,
+    &bufneed,
+    &num_services,
+    NULL,
+    NULL);
+  if(!ok)
+  {
+    WcaLog(LOGMSG_STANDARD, "last error %d", GetLastError());
+    ExitOnFailure(E_FAIL, "EnumServicesStatusExW failed");
+  }
+  info = (LPENUM_SERVICE_STATUS_PROCESS)buf;
+  for (ULONG i=0; i < num_services; i++)
+  {
+    SC_HANDLE service= OpenServiceW(scm, info[i].lpServiceName,
+      SERVICE_QUERY_CONFIG);
+    if (!service)
+      continue;
+    WcaLog(LOGMSG_VERBOSE, "Checking Service %S", info[i].lpServiceName);
+    QUERY_SERVICE_CONFIGW *config=
+      (QUERY_SERVICE_CONFIGW *)(void *)config_buffer;
+    DWORD needed;
+    BOOL ok= QueryServiceConfigW(service, config,sizeof(config_buffer),
+      &needed);
+    CloseServiceHandle(service);
+    if (ok)
+    {
+       CheckServiceConfig(servicename, datadir, bindir, info[i].lpServiceName,
+         config);
+    }
+  }
+
+LExit:
+  if(scm)
+    CloseServiceHandle(scm);
+
+  ReleaseStr(servicename);
+  ReleaseStr(datadir);
+  ReleaseStr(bindir);
+  return WcaFinalize(er);
+}
+
+/*
+  Get maximum size of the buffer process can allocate.
+  this is calculated as min(RAM,virtualmemorylimit)
+  For 32bit processes, virtual address memory is 2GB (x86 OS)
+  or 4GB(x64 OS).
+
+  Fragmentation due to loaded modules, heap  and stack
+  limit maximum size of continuous memory block further,
+  so that limit for 32 bit process is about 1200 on 32 bit OS
+  or 2000 MB on 64 bit OS(found experimentally).
+*/
+unsigned long long GetMaxBufferSize(unsigned long long totalPhys)
+{
+#ifdef _M_IX86
+  BOOL wow64;
+  if (IsWow64Process(GetCurrentProcess(), &wow64))
+    return min(totalPhys, 2000ULL*ONE_MB);
+  else
+    return min(totalPhys, 1200ULL*ONE_MB);
+#else
+  return totalPhys;
+#endif
+}
+
+
+/*
+  Magic undocumented number for bufferpool minimum,
+  allows innodb to start also for all page sizes.
+*/
+static constexpr unsigned long long minBufferpoolMB= 20;
+
+/*
+  Checks SERVICENAME, PORT and BUFFERSIZE parameters
+*/
+extern "C" UINT  __stdcall CheckDatabaseProperties (MSIHANDLE hInstall)
+{
+  wchar_t ServiceName[MAX_PATH]={0};
+  wchar_t SkipNetworking[MAX_PATH]={0};
+  wchar_t QuickConfig[MAX_PATH]={0};
+  wchar_t Password[MAX_PATH]={0};
+  wchar_t EscapedPassword[2*MAX_PATH+2];
+  wchar_t Port[6];
+  wchar_t BufferPoolSize[16];
+  DWORD PortLen=6;
+  bool haveInvalidPort=false;
+  const wchar_t *ErrorMsg=0;
+  HRESULT hr= S_OK;
+  UINT er= ERROR_SUCCESS;
+  DWORD ServiceNameLen = MAX_PATH;
+  DWORD QuickConfigLen = MAX_PATH;
+  DWORD PasswordLen= MAX_PATH;
+  DWORD SkipNetworkingLen= MAX_PATH;
+
+  hr = WcaInitialize(hInstall, __FUNCTION__);
+  ExitOnFailure(hr, "Failed to initialize");
+  WcaLog(LOGMSG_STANDARD, "Initialized.");
+
+
+  MsiGetPropertyW (hInstall, L"SERVICENAME", ServiceName, &ServiceNameLen);
+  if(ServiceName[0])
+  {
+    if(ServiceNameLen > 256)
+    {
+      ErrorMsg= L"Invalid service name. The maximum length is 256 characters.";
+      goto LExit;
+    }
+    for(DWORD i=0; i< ServiceNameLen;i++)
+    {
+      if(ServiceName[i] == L'\\' || ServiceName[i] == L'/'
+        || ServiceName[i]=='\'' || ServiceName[i] ==L'"')
+      {
+        ErrorMsg =
+          L"Invalid service name. Forward slash and back slash are forbidden."
+          L"Single and double quotes are also not permitted.";
+        goto LExit;
+      }
+    }
+    if(CheckServiceExists(ServiceName))
+    {
+      ErrorMsg=
+        L"A service with the same name already exists. "
+        L"Please use a different name.";
+      goto LExit;
+    }
+  }
+
+  MsiGetPropertyW (hInstall, L"PASSWORD", Password, &PasswordLen);
+  EscapeCommandLine(Password, EscapedPassword,
+    sizeof(EscapedPassword)/sizeof(EscapedPassword[0]));
+  MsiSetPropertyW(hInstall,L"ESCAPEDPASSWORD",EscapedPassword);
+
+  MsiGetPropertyW(hInstall, L"SKIPNETWORKING", SkipNetworking,
+    &SkipNetworkingLen);
+  MsiGetPropertyW(hInstall, L"PORT", Port, &PortLen);
+
+  if(SkipNetworking[0]==0 && Port[0] != 0)
+  {
+    /* Strip spaces */
+    for(DWORD i=PortLen-1; i > 0; i--)
+    {
+      if(Port[i]== ' ')
+        Port[i] = 0;
+    }
+
+    if(PortLen > 5 || PortLen <= 3)
+      haveInvalidPort = true;
+    else
+    {
+      for (DWORD i=0; i< PortLen && Port[i] != 0;i++)
+      {
+        if(Port[i] < '0' || Port[i] >'9')
+        {
+          haveInvalidPort=true;
+          break;
+        }
+      }
+    }
+    if (haveInvalidPort)
+    {
+      ErrorMsg =
+        L"Invalid port number. Please use a number between 1025 and 65535.";
+      goto LExit;
+    }
+
+    unsigned short port = (unsigned short)_wtoi(Port);
+    if (!IsPortFree(port))
+    {
+      ErrorMsg =
+        L"The TCP Port you selected is already in use. "
+        L"Please choose a different port.";
+      goto LExit;
+    }
+  }
+
+  MsiGetPropertyW (hInstall, L"STDCONFIG", QuickConfig, &QuickConfigLen);
+  if(QuickConfig[0] !=0)
+  {
+     MEMORYSTATUSEX memstatus;
+     memstatus.dwLength =sizeof(memstatus);
+     wchar_t invalidValueMsg[256];
+
+     if (!GlobalMemoryStatusEx(&memstatus))
+     {
+        WcaLog(LOGMSG_STANDARD, "Error %u from GlobalMemoryStatusEx",
+          GetLastError());
+        er= ERROR_INSTALL_FAILURE;
+        goto LExit;
+     }
+     DWORD BufferPoolSizeLen= 16;
+     MsiGetPropertyW(hInstall, L"BUFFERPOOLSIZE", BufferPoolSize, &BufferPoolSizeLen);
+     /* Strip spaces */
+     for(DWORD i=BufferPoolSizeLen-1; i > 0; i--)
+     {
+      if(BufferPoolSize[i]== ' ')
+        BufferPoolSize[i] = 0;
+     }
+     unsigned long long availableMemory=
+       GetMaxBufferSize(memstatus.ullTotalPhys)/ONE_MB;
+     swprintf_s(invalidValueMsg,
+        L"Invalid buffer pool size. Please use a number between %llu and %llu",
+         minBufferpoolMB, availableMemory);
+     if (BufferPoolSizeLen == 0 || BufferPoolSizeLen > 15 || !BufferPoolSize[0])
+     {
+       ErrorMsg= invalidValueMsg;
+       goto LExit;
+     }
+
+     BufferPoolSize[BufferPoolSizeLen]=0;
+     MsiSetPropertyW(hInstall, L"BUFFERPOOLSIZE", BufferPoolSize);
+     wchar_t *end;
+     unsigned long long sz = wcstoull(BufferPoolSize, &end, 10);
+     if (sz > availableMemory || sz < minBufferpoolMB || *end)
+     {
+       if (*end == 0)
+       {
+         if(sz > availableMemory)
+         {
+           swprintf_s(invalidValueMsg,
+             L"Value for buffer pool size is too large."
+             L"Only approximately %llu MB is available for allocation."
+             L"Please use a number between %llu and %llu.",
+             availableMemory, minBufferpoolMB, availableMemory);
+         }
+         else if(sz < minBufferpoolMB)
+         {
+           swprintf_s(invalidValueMsg,
+             L"Value for buffer pool size is too small."
+             L"Please use a number between %llu and %llu.",
+             minBufferpoolMB, availableMemory);
+         }
+       }
+       ErrorMsg= invalidValueMsg;
+       goto LExit;
+     }
+  }
+LExit:
+  MsiSetPropertyW (hInstall, L"WarningText", ErrorMsg);
+  return WcaFinalize(er);
+}
+
+/*
+  Sets Innodb buffer pool size (1/8 of RAM by default), if not already specified
+  via command line.
+  Calculates innodb log file size as min(100, innodb buffer pool size/4)
+*/
+extern "C" UINT __stdcall PresetDatabaseProperties(MSIHANDLE hInstall)
+{
+  unsigned long long InnodbBufferPoolSize= 256;
+  unsigned long long InnodbLogFileSize= 100;
+  wchar_t buff[MAX_PATH];
+  UINT er = ERROR_SUCCESS;
+  HRESULT hr= S_OK;
+  MEMORYSTATUSEX memstatus;
+  DWORD BufferPoolsizeParamLen = MAX_PATH;
+  hr = WcaInitialize(hInstall, __FUNCTION__);
+  ExitOnFailure(hr, "Failed to initialize");
+  WcaLog(LOGMSG_STANDARD, "Initialized.");
+
+  /* Check if bufferpoolsize parameter was given on the command line*/
+  MsiGetPropertyW(hInstall, L"BUFFERPOOLSIZE", buff, &BufferPoolsizeParamLen);
+
+  if (BufferPoolsizeParamLen && buff[0])
+  {
+    WcaLog(LOGMSG_STANDARD, "BUFFERPOOLSIZE=%S, len=%u",buff, BufferPoolsizeParamLen);
+    InnodbBufferPoolSize= _wtoi64(buff);
+  }
+  else
+  {
+    memstatus.dwLength = sizeof(memstatus);
+    if (!GlobalMemoryStatusEx(&memstatus))
+    {
+       WcaLog(LOGMSG_STANDARD, "Error %u from GlobalMemoryStatusEx",
+         GetLastError());
+       er= ERROR_INSTALL_FAILURE;
+       goto LExit;
+    }
+    unsigned long long totalPhys= memstatus.ullTotalPhys;
+    /* Give innodb 12.5% of available physical memory. */
+    InnodbBufferPoolSize= totalPhys/ONE_MB/8;
+ #ifdef _M_IX86
+    /*
+      For 32 bit processes, take virtual address space limitation into account.
+      Do not try to use more than 3/4 of virtual address space, even if there
+      is plenty of physical memory.
+    */
+    InnodbBufferPoolSize= min(GetMaxBufferSize(totalPhys)/ONE_MB*3/4,
+      InnodbBufferPoolSize);
+ #endif
+    swprintf_s(buff, L"%llu",InnodbBufferPoolSize);
+    MsiSetPropertyW(hInstall, L"BUFFERPOOLSIZE", buff);
+  }
+  InnodbLogFileSize = min(100, 2 * InnodbBufferPoolSize);
+  swprintf_s(buff, L"%llu",InnodbLogFileSize);
+  MsiSetPropertyW(hInstall, L"LOGFILESIZE", buff);
+
+LExit:
+  return WcaFinalize(er);
+}
+
+static BOOL FindErrorLog(const wchar_t *dir, wchar_t * ErrorLogFile, size_t ErrorLogLen)
+{
+  WIN32_FIND_DATA FindFileData;
+  HANDLE hFind;
+  wchar_t name[MAX_PATH];
+  wcsncpy_s(name,dir, MAX_PATH);
+  wcsncat_s(name,L"\\*.err", MAX_PATH);
+  hFind = FindFirstFileW(name,&FindFileData);
+  if (hFind != INVALID_HANDLE_VALUE)
+  {
+    _snwprintf(ErrorLogFile, ErrorLogLen,
+      L"%s\\%s",dir, FindFileData.cFileName);
+    FindClose(hFind);
+    return TRUE;
+  }
+  return FALSE;
+}
+
+static void DumpErrorLog(const wchar_t *dir)
+{
+  wchar_t filepath[MAX_PATH];
+  if (!FindErrorLog(dir, filepath, MAX_PATH))
+    return;
+  FILE *f= _wfopen(filepath, L"r");
+  if (!f)
+    return;
+  char buf[2048];
+  WcaLog(LOGMSG_STANDARD,"=== dumping error log %S === ",filepath);
+  while (fgets(buf, sizeof(buf), f))
+  {
+     /* Strip off EOL chars. */
+     size_t len = strlen(buf);
+     if (len > 0 && buf[len-1] == '\n')
+       buf[--len]= 0;
+     if (len > 0 && buf[len-1] == '\r')
+       buf[--len]= 0;
+     WcaLog(LOGMSG_STANDARD,"%s",buf);
+  }
+  fclose(f);
+  WcaLog(LOGMSG_STANDARD,"=== end of error log ===");
+}
+
+/* Remove service and data directory created by CreateDatabase operation */
+extern "C" UINT __stdcall CreateDatabaseRollback(MSIHANDLE hInstall)
+{
+  HRESULT hr = S_OK;
+  UINT er = ERROR_SUCCESS;
+  wchar_t* service= 0;
+  wchar_t* dir= 0;
+  wchar_t data[2*MAX_PATH];
+  DWORD len= MAX_PATH;
+
+  hr = WcaInitialize(hInstall, __FUNCTION__);
+  ExitOnFailure(hr, "Failed to initialize");
+  WcaLog(LOGMSG_STANDARD, "Initialized.");
+
+  MsiGetPropertyW(hInstall, L"CustomActionData", data, &len);
+
+  /* Property is encoded as [SERVICENAME]\[DBLOCATION] */
+  if(data[0] == L'\\')
+  {
+    dir= data+1;
+  }
+  else
+  {
+    service= data;
+    dir= wcschr(data, '\\');
+    if (dir)
+    {
+     *dir=0;
+     dir++;
+    }
+  }
+
+  if(service)
+  {
+    ExecRemoveService(service);
+  }
+  if(dir)
+  {
+    DumpErrorLog(dir);
+    ExecRemoveDataDirectory(dir);
+  }
+LExit:
+  return WcaFinalize(er);
+}
+
+
+/*
+  Enables/disables optional "Launch upgrade wizard" checkbox at the end of
+  installation
+*/
+#define MAX_VERSION_PROPERTY_SIZE 64
+
+extern "C" UINT __stdcall CheckServiceUpgrades(MSIHANDLE hInstall)
+{
+  HRESULT hr = S_OK;
+  UINT er = ERROR_SUCCESS;
+  wchar_t installerVersion[MAX_VERSION_PROPERTY_SIZE];
+  char installDir[MAX_PATH];
+  DWORD size =MAX_VERSION_PROPERTY_SIZE;
+  int installerMajorVersion, installerMinorVersion, installerPatchVersion;
+  bool upgradableServiceFound=false;
+  LPENUM_SERVICE_STATUS_PROCESSW info;
+  DWORD bufsize;
+  int index;
+  BOOL ok;
+  SC_HANDLE scm = NULL;
+
+  hr = WcaInitialize(hInstall, __FUNCTION__);
+   WcaLog(LOGMSG_STANDARD, "Initialized.");
+  if (MsiGetPropertyW(hInstall, L"ProductVersion", installerVersion, &size)
+    != ERROR_SUCCESS)
+  {
+    hr = HRESULT_FROM_WIN32(GetLastError());
+    ExitOnFailure(hr, "MsiGetPropertyW failed");
+  }
+   if (swscanf(installerVersion,L"%d.%d.%d",
+    &installerMajorVersion, &installerMinorVersion, &installerPatchVersion) !=3)
+  {
+    assert(FALSE);
+  }
+
+  size= MAX_PATH;
+  if (MsiGetPropertyA(hInstall,"INSTALLDIR", installDir, &size)
+    != ERROR_SUCCESS)
+  {
+    hr = HRESULT_FROM_WIN32(GetLastError());
+    ExitOnFailure(hr, "MsiGetPropertyW failed");
+  }
+
+
+  scm = OpenSCManager(NULL, NULL,
+    SC_MANAGER_ENUMERATE_SERVICE | SC_MANAGER_CONNECT);
+  if (scm == NULL)
+  {
+    hr = HRESULT_FROM_WIN32(GetLastError());
+    ExitOnFailure(hr,"OpenSCManager failed");
+  }
+
+  static BYTE buf[64*1024];
+  static BYTE config_buffer[8*1024];
+
+  bufsize= sizeof(buf);
+  DWORD bufneed;
+  DWORD num_services;
+  ok= EnumServicesStatusExW(scm, SC_ENUM_PROCESS_INFO,  SERVICE_WIN32,
+    SERVICE_STATE_ALL,  buf, bufsize,  &bufneed, &num_services, NULL, NULL);
+  if(!ok)
+  {
+    hr = HRESULT_FROM_WIN32(GetLastError());
+    ExitOnFailure(hr,"EnumServicesStatusEx failed");
+  }
+  info =
+    (LPENUM_SERVICE_STATUS_PROCESSW)buf;
+  index=-1;
+  for (ULONG i=0; i < num_services; i++)
+  {
+    SC_HANDLE service= OpenServiceW(scm, info[i].lpServiceName,
+      SERVICE_QUERY_CONFIG);
+    if (!service)
+      continue;
+    QUERY_SERVICE_CONFIGW *config=
+      (QUERY_SERVICE_CONFIGW*)(void *)config_buffer;
+    DWORD needed;
+    ok= QueryServiceConfigW(service, config,sizeof(config_buffer),
+      &needed) && (config->dwStartType != SERVICE_DISABLED);
+    CloseServiceHandle(service);
+    if (ok)
+    {
+       mysqld_service_properties props;
+       if (get_mysql_service_properties(config->lpBinaryPathName, &props))
+                  continue;
+        /*
+          Only look for services that have mysqld.exe outside of the current
+          installation directory.
+        */
+       if(installDir[0] == 0 || strstr(props.mysqld_exe,installDir) == 0)
+        {
+           WcaLog(LOGMSG_STANDARD, "found service %S, major=%d, minor=%d",
+            info[i].lpServiceName, props.version_major, props.version_minor);
+          if(props.version_major < installerMajorVersion
+            || (props.version_major == installerMajorVersion &&
+                props.version_minor <= installerMinorVersion))
+          {
+            upgradableServiceFound= true;
+            break;
+          }
+       }
+    }
+  }
+
+  if(!upgradableServiceFound)
+  {
+    /* Disable optional checkbox at the end of installation */
+    MsiSetPropertyW(hInstall, L"WIXUI_EXITDIALOGOPTIONALCHECKBOXTEXT", L"");
+    MsiSetPropertyW(hInstall, L"WIXUI_EXITDIALOGOPTIONALCHECKBOX",L"");
+  }
+  else
+  {
+    MsiSetPropertyW(hInstall, L"UpgradableServiceFound", L"1");
+    MsiSetPropertyW(hInstall, L"WIXUI_EXITDIALOGOPTIONALCHECKBOX",L"1");
+  }
+LExit:
+  if(scm)
+    CloseServiceHandle(scm);
+  return WcaFinalize(er);
+}
+
+
+/* DllMain - Initialize and cleanup WiX custom action utils */
+extern "C" BOOL WINAPI DllMain(
+  __in HINSTANCE hInst,
+  __in ULONG ulReason,
+  __in LPVOID
+  )
+{
+  switch(ulReason)
+  {
+  case DLL_PROCESS_ATTACH:
+    WcaGlobalInitialize(hInst);
+    break;
+
+  case DLL_PROCESS_DETACH:
+    WcaGlobalFinalize();
+    break;
+  }
+
+  return TRUE;
+}
+