/*
   Copyright (c) 2011, 2012, Monty Program Ab

   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 */

/*
  Extract properties of a windows service binary path
*/
#pragma once

#ifdef __cplusplus
extern "C" {
#endif

#include <windows.h>
#ifdef _MSC_VER
#pragma warning(push)
#pragma warning(disable : 4995)
#endif
#include <winsvc.h>
#ifdef _MSC_VER
#pragma warning(pop)
#endif

typedef struct mysqld_service_properties_st
{
  char mysqld_exe[MAX_PATH];
  char inifile[MAX_PATH];
  char datadir[MAX_PATH];
  int  version_major;
  int  version_minor;
  int  version_patch;
} mysqld_service_properties;

extern int get_mysql_service_properties(const wchar_t *bin_path,
  mysqld_service_properties *props);


#if !defined(UNICODE)
/*
  The following wrappers workaround Windows bugs
  with CreateService/OpenService with ANSI codepage UTF8.

  Apparently, these function in ANSI mode, for this codepage only
  do *not* behave as expected (as-if string parameters were
  converted to UTF16 and "wide" function were called)
*/
#include <malloc.h>
static inline wchar_t* awstrdup(const char *str)
{
  if (!str)
    return NULL;
  size_t len= strlen(str) + 1;
  wchar_t *wstr= (wchar_t *) malloc(sizeof(wchar_t)*len);
  if (MultiByteToWideChar(GetACP(), 0, str, (int)len, wstr, (int)len) == 0)
  {
    free(wstr);
    return NULL;
  }
  return wstr;
}

#define AWSTRDUP(dest, src)                                                   \
  dest= awstrdup(src);                                                        \
  if (src && !dest)                                                           \
  {                                                                           \
    ok= FALSE;                                                                \
    last_error = ERROR_OUTOFMEMORY;                                           \
    goto end;                                                                 \
  }

static inline SC_HANDLE my_OpenService(SC_HANDLE hSCManager, LPCSTR lpServiceName, DWORD dwDesiredAccess)
{
  wchar_t *w_ServiceName= NULL;
  BOOL ok=TRUE;
  DWORD last_error=0;
  SC_HANDLE sch=NULL;

  AWSTRDUP(w_ServiceName, lpServiceName);
  sch= OpenServiceW(hSCManager, w_ServiceName, dwDesiredAccess);
  if (!sch)
  {
    ok= FALSE;
    last_error= GetLastError();
  }

end:
  free(w_ServiceName);
  if (!ok)
    SetLastError(last_error);
  return sch;
}

static inline SC_HANDLE my_CreateService(SC_HANDLE hSCManager,
  LPCSTR lpServiceName, LPCSTR lpDisplayName,
  DWORD dwDesiredAccess,  DWORD dwServiceType,
  DWORD dwStartType, DWORD dwErrorControl,
  LPCSTR lpBinaryPathName, LPCSTR lpLoadOrderGroup,
  LPDWORD lpdwTagId, LPCSTR lpDependencies,
  LPCSTR lpServiceStartName, LPCSTR lpPassword)
{
  wchar_t *w_ServiceName= NULL;
  wchar_t *w_DisplayName= NULL;
  wchar_t *w_BinaryPathName= NULL;
  wchar_t *w_LoadOrderGroup= NULL;
  wchar_t *w_Dependencies= NULL;
  wchar_t *w_ServiceStartName= NULL;
  wchar_t *w_Password= NULL;
  SC_HANDLE sch = NULL;
  DWORD last_error=0;
  BOOL ok= TRUE;

  AWSTRDUP(w_ServiceName,lpServiceName);
  AWSTRDUP(w_DisplayName,lpDisplayName);
  AWSTRDUP(w_BinaryPathName, lpBinaryPathName);
  AWSTRDUP(w_LoadOrderGroup, lpLoadOrderGroup);
  AWSTRDUP(w_Dependencies, lpDependencies);
  AWSTRDUP(w_ServiceStartName, lpServiceStartName);
  AWSTRDUP(w_Password, lpPassword);

  sch= CreateServiceW(
      hSCManager, w_ServiceName, w_DisplayName, dwDesiredAccess, dwServiceType,
      dwStartType, dwErrorControl, w_BinaryPathName, w_LoadOrderGroup,
      lpdwTagId, w_Dependencies, w_ServiceStartName, w_Password);
  if(!sch)
  {
    ok= FALSE;
    last_error= GetLastError();
  }

end:
  free(w_ServiceName);
  free(w_DisplayName);
  free(w_BinaryPathName);
  free(w_LoadOrderGroup);
  free(w_Dependencies);
  free(w_ServiceStartName);
  free(w_Password);

  if (!ok)
    SetLastError(last_error);
  return sch;
}

static inline BOOL my_ChangeServiceConfig(SC_HANDLE hService, DWORD dwServiceType,
   DWORD dwStartType, DWORD dwErrorControl,
   LPCSTR lpBinaryPathName, LPCSTR lpLoadOrderGroup,
   LPDWORD lpdwTagId, LPCSTR lpDependencies,
   LPCSTR lpServiceStartName, LPCSTR lpPassword,
   LPCSTR lpDisplayName)
{
  wchar_t *w_DisplayName= NULL;
  wchar_t *w_BinaryPathName= NULL;
  wchar_t *w_LoadOrderGroup= NULL;
  wchar_t *w_Dependencies= NULL;
  wchar_t *w_ServiceStartName= NULL;
  wchar_t *w_Password= NULL;
  DWORD last_error=0;
  BOOL ok= TRUE;

  AWSTRDUP(w_DisplayName, lpDisplayName);
  AWSTRDUP(w_BinaryPathName, lpBinaryPathName);
  AWSTRDUP(w_LoadOrderGroup, lpLoadOrderGroup);
  AWSTRDUP(w_Dependencies, lpDependencies);
  AWSTRDUP(w_ServiceStartName, lpServiceStartName);
  AWSTRDUP(w_Password, lpPassword);

  ok= ChangeServiceConfigW(
      hService, dwServiceType, dwStartType, dwErrorControl, w_BinaryPathName,
      w_LoadOrderGroup, lpdwTagId, w_Dependencies, w_ServiceStartName,
      w_Password, w_DisplayName);
  if (!ok)
  {
    last_error= GetLastError();
  }

end:
  free(w_DisplayName);
  free(w_BinaryPathName);
  free(w_LoadOrderGroup);
  free(w_Dependencies);
  free(w_ServiceStartName);
  free(w_Password);

  if (last_error)
    SetLastError(last_error);
  return ok;
}
#undef AWSTRDUP

#undef OpenService
#define OpenService my_OpenService
#undef ChangeServiceConfig
#define ChangeServiceConfig my_ChangeServiceConfig
#undef CreateService
#define CreateService my_CreateService
#endif

#ifdef __cplusplus
}
#endif