diff options
Diffstat (limited to 'win/packaging/ca')
| -rw-r--r-- | win/packaging/ca/CMakeLists.txt | 24 | ||||
| -rw-r--r-- | win/packaging/ca/CustomAction.cpp | 1134 | ||||
| -rw-r--r-- | win/packaging/ca/CustomAction.def | 11 | ||||
| -rw-r--r-- | win/packaging/ca/CustomAction.rc | 18 |
4 files changed, 1187 insertions, 0 deletions
diff --git a/win/packaging/ca/CMakeLists.txt b/win/packaging/ca/CMakeLists.txt new file mode 100644 index 00000000..368a844f --- /dev/null +++ b/win/packaging/ca/CMakeLists.txt @@ -0,0 +1,24 @@ +# 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 St, Fifth Floor, Boston, MA 02110-1335 USA + +SET(WIXCA_SOURCES CustomAction.cpp CustomAction.def) + +INCLUDE_DIRECTORIES(${CMAKE_SOURCE_DIR}/sql ${CMAKE_CURRENT_SOURCE_DIR} ${WIX_INCLUDE_DIR}) + +# Custom action should not depend on C runtime, since we do not know if CRT is installed. +FORCE_STATIC_CRT() +ADD_VERSION_INFO(wixca SHARED WIXCA_SOURCES) +ADD_LIBRARY(wixca SHARED EXCLUDE_FROM_ALL ${WIXCA_SOURCES} ${CMAKE_SOURCE_DIR}/sql/winservice.c) +TARGET_LINK_LIBRARIES(wixca ${WIX_WCAUTIL_LIBRARY} ${WIX_DUTIL_LIBRARY} msi version) 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; +} + diff --git a/win/packaging/ca/CustomAction.def b/win/packaging/ca/CustomAction.def new file mode 100644 index 00000000..c18a0d92 --- /dev/null +++ b/win/packaging/ca/CustomAction.def @@ -0,0 +1,11 @@ +LIBRARY "wixca" +VERSION 1.0 +EXPORTS +PresetDatabaseProperties +RemoveDataDirectory +CreateDatabaseRollback +CheckDatabaseProperties +CheckDataDirectory +CheckDBInUse +CheckServiceUpgrades +CheckInstallDirectory diff --git a/win/packaging/ca/CustomAction.rc b/win/packaging/ca/CustomAction.rc new file mode 100644 index 00000000..3f37126e --- /dev/null +++ b/win/packaging/ca/CustomAction.rc @@ -0,0 +1,18 @@ +#include "afxres.h" +#undef APSTUDIO_READONLY_SYMBOLS + +VS_VERSION_INFO VERSIONINFO + FILEVERSION 1,0,0,1 + PRODUCTVERSION 1,0,0,1 + FILEFLAGSMASK 0x17L +#ifdef _DEBUG + FILEFLAGS 0x1L +#else + FILEFLAGS 0x0L +#endif + FILEOS 0x4L + FILETYPE 0x0L + FILESUBTYPE 0x0L +BEGIN +END + |