]> git.sur5r.net Git - openldap/blobdiff - libraries/liblutil/ntservice.c
Really fix the fix to the past commit
[openldap] / libraries / liblutil / ntservice.c
index 53c6841106ab0fdfa5b9fdac5f9228ce80e62658..bc4ae422dc3642b232c7736ffcbadfe210056d30 100644 (file)
@@ -1,24 +1,44 @@
-// ntservice.c
+/* $OpenLDAP$ */
+/*
+ * Copyright 1998-2000 The OpenLDAP Foundation, All Rights Reserved.
+ * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
+ */
+
+/*
+ * NT Service manager utilities for OpenLDAP services
+ *     these should NOT be slapd specific, but are
+ */
+
 #include "portable.h"
+
+#ifdef HAVE_NT_SERVICE_MANAGER
+
+#include <ac/stdlib.h>
+#include <ac/string.h>
+
 #include <stdio.h>
+
 #include <windows.h>
-#include "ldap.h"
+#include <winsvc.h>
+
+#include <ldap.h>
+
+#define ldap_debug slap_debug
+extern int slap_debug;
+
 #include "ldap_log.h"
 #include "ldap_pvt_thread.h"
-#include <winsvc.h>
-#include <sys/types.h>
-#include <ac/string.h>
+
 
 #include "ldap_defaults.h"
-//#include "slap.h"
 
 #include "slapdmsg.h"
 
 #define SCM_NOTIFICATION_INTERVAL      5000
 #define THIRTY_SECONDS                         (30 * 1000)
 
-int      is_NT_Service = 1;    // assume this is an NT service until determined that 
-                                                                       // startup was from the command line
+int      is_NT_Service = 1;    /* assume this is an NT service until determined that */
+                                                       /* startup was from the command line */
 
 SERVICE_STATUS                 SLAPDServiceStatus;
 SERVICE_STATUS_HANDLE  hSLAPDServiceStatus;
@@ -28,17 +48,10 @@ ldap_pvt_thread_t           start_status_tid,       stop_status_tid;
 
 void (*stopfunc)(int);
 
-// in main.c
-void WINAPI ServiceMain( DWORD argc, LPTSTR *argv );
-
+static char *GetLastErrorString( void );
 
-// in wsa_err.c
-char *WSAGetLastErrorString( void );
-
-// in nt_err.c
-char *GetLastErrorString( void );
-
-int srv_install(LPCTSTR lpszServiceName, LPCTSTR lpszBinaryPathName)
+int srv_install(LPCTSTR lpszServiceName, LPCTSTR lpszDisplayName,
+               LPCTSTR lpszBinaryPathName, BOOL auto_start)
 {
        HKEY            hKey;
        DWORD           dwValue, dwDisposition;
@@ -50,10 +63,10 @@ int srv_install(LPCTSTR lpszServiceName, LPCTSTR lpszBinaryPathName)
                if ((schService = CreateService( 
                                                        schSCManager, 
                                                        lpszServiceName, 
-                                                       TEXT("OpenLDAP Directory Service")
-                                                       SC_MANAGER_CREATE_SERVICE
+                                                       lpszDisplayName
+                                                       SERVICE_ALL_ACCESS
                                                        SERVICE_WIN32_OWN_PROCESS, 
-                                                       SERVICE_DEMAND_START, 
+                                                       auto_start ? SERVICE_AUTO_START : SERVICE_DEMAND_START, 
                                                        SERVICE_ERROR_NORMAL, 
                                                        lpszBinaryPathName, 
                                                        NULL, NULL, NULL, NULL, NULL)) != NULL)
@@ -64,19 +77,19 @@ int srv_install(LPCTSTR lpszServiceName, LPCTSTR lpszBinaryPathName)
 
                        sprintf( regpath, "SYSTEM\\CurrentControlSet\\Services\\EventLog\\Application\\%s",
                                lpszServiceName );
-                       // Create the registry key for event logging to the Windows NT event log.
+                       /* Create the registry key for event logging to the Windows NT event log. */
                        if ( RegCreateKeyEx(HKEY_LOCAL_MACHINE, 
                                regpath, 0, 
                                "REG_SZ", REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hKey, 
                                &dwDisposition) != ERROR_SUCCESS)
                        {
-                               fprintf( stderr, "RegCreateKeyEx() failed. GetLastError=%d (%s)\n", GetLastError(), GetLastErrorString() );
+                               fprintf( stderr, "RegCreateKeyEx() failed. GetLastError=%lu (%s)\n", GetLastError(), GetLastErrorString() );
                                RegCloseKey(hKey);
                                return(0);
                        }
                        if ( RegSetValueEx(hKey, "EventMessageFile", 0, REG_EXPAND_SZ, lpszBinaryPathName, strlen(lpszBinaryPathName) + 1) != ERROR_SUCCESS)
                        {
-                               fprintf( stderr, "RegSetValueEx(EventMessageFile) failed. GetLastError=%d (%s)\n", GetLastError(), GetLastErrorString() );
+                               fprintf( stderr, "RegSetValueEx(EventMessageFile) failed. GetLastError=%lu (%s)\n", GetLastError(), GetLastErrorString() );
                                RegCloseKey(hKey);
                                return(0);
                        }
@@ -84,7 +97,7 @@ int srv_install(LPCTSTR lpszServiceName, LPCTSTR lpszBinaryPathName)
                        dwValue = EVENTLOG_ERROR_TYPE | EVENTLOG_WARNING_TYPE | EVENTLOG_INFORMATION_TYPE;
                        if ( RegSetValueEx(hKey, "TypesSupported", 0, REG_DWORD, (LPBYTE) &dwValue, sizeof(DWORD)) != ERROR_SUCCESS) 
                        {
-                               fprintf( stderr, "RegCreateKeyEx(TypesSupported) failed. GetLastError=%d (%s)\n", GetLastError(), GetLastErrorString() );
+                               fprintf( stderr, "RegCreateKeyEx(TypesSupported) failed. GetLastError=%lu (%s)\n", GetLastError(), GetLastErrorString() );
                                RegCloseKey(hKey);
                                return(0);
                        }
@@ -93,13 +106,13 @@ int srv_install(LPCTSTR lpszServiceName, LPCTSTR lpszBinaryPathName)
                }
                else
                {
-                       fprintf( stderr, "CreateService() failed. GetLastError=%d (%s)\n", GetLastError(), GetLastErrorString() );
+                       fprintf( stderr, "CreateService() failed. GetLastError=%lu (%s)\n", GetLastError(), GetLastErrorString() );
                        CloseServiceHandle(schSCManager);
                        return(0);
                }
        }
        else
-               fprintf( stderr, "OpenSCManager() failed. GetLastError=%d (%s)\n", GetLastError(), GetLastErrorString() );
+               fprintf( stderr, "OpenSCManager() failed. GetLastError=%lu (%s)\n", GetLastError(), GetLastErrorString() );
        return(0);
 }
 
@@ -119,24 +132,75 @@ int srv_remove(LPCTSTR lpszServiceName, LPCTSTR lpszBinaryPathName)
                                CloseServiceHandle(schSCManager);
                                return(1);
                        } else {
-                               fprintf( stderr, "DeleteService() failed. GetLastError=%d (%s)\n", GetLastError(), GetLastErrorString() );
+                               fprintf( stderr, "DeleteService() failed. GetLastError=%lu (%s)\n", GetLastError(), GetLastErrorString() );
                                fprintf( stderr, "The %s service has not been removed.\n", lpszBinaryPathName);
                                CloseServiceHandle(schService);
                                CloseServiceHandle(schSCManager);
                                return(0);
                        }
                } else {
-                       fprintf( stderr, "OpenService() failed. GetLastError=%d (%s)\n", GetLastError(), GetLastErrorString() );
+                       fprintf( stderr, "OpenService() failed. GetLastError=%lu (%s)\n", GetLastError(), GetLastErrorString() );
                        CloseServiceHandle(schSCManager);
                        return(0);
                }
        }
        else
-               fprintf( stderr, "OpenSCManager() failed. GetLastError=%d (%s)\n", GetLastError(), GetLastErrorString() );
+               fprintf( stderr, "OpenSCManager() failed. GetLastError=%lu (%s)\n", GetLastError(), GetLastErrorString() );
        return(0);
 }
 
 
+DWORD
+svc_installed (LPTSTR lpszServiceName, LPTSTR lpszBinaryPathName)
+{
+       char buf[256];
+       HKEY key;
+       DWORD rc;
+       DWORD type;
+       long len;
+
+       strcpy(buf, TEXT("SYSTEM\\CurrentControlSet\\Services\\"));
+       strcat(buf, lpszServiceName);
+       if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, buf, 0, KEY_QUERY_VALUE, &key) != ERROR_SUCCESS)
+               return(-1);
+
+       rc = 0;
+       if (lpszBinaryPathName) {
+               len = sizeof(buf);
+               if (RegQueryValueEx(key, "ImagePath", NULL, &type, buf, &len) == ERROR_SUCCESS) {
+                       if (strcmp(lpszBinaryPathName, buf))
+                               rc = -1;
+               }
+       }
+       RegCloseKey(key);
+       return(rc);
+}
+
+
+DWORD
+svc_running (LPTSTR lpszServiceName)
+{
+       SC_HANDLE service;
+       SC_HANDLE scm;
+       DWORD rc;
+       SERVICE_STATUS ss;
+
+       if (!(scm = OpenSCManager(NULL, NULL, GENERIC_READ)))
+               return(GetLastError());
+
+       rc = 1;
+       service = OpenService(scm, lpszServiceName, SERVICE_QUERY_STATUS);
+       if (service) {
+               if (!QueryServiceStatus(service, &ss))
+                       rc = GetLastError();
+               else if (ss.dwCurrentState != SERVICE_STOPPED)
+                       rc = 0;
+               CloseServiceHandle(service);
+       }
+       CloseServiceHandle(scm);
+       return(rc);
+}
+
 
 static void *start_status_routine( void *ptr )
 {
@@ -150,9 +214,9 @@ static void *start_status_routine( void *ptr )
                {
                        case WAIT_ABANDONED:
                        case WAIT_OBJECT_0:
-                               // the object that we were waiting for has been destroyed (ABANDONED) or
-                               // signalled (TIMEOUT_0). We can assume that the startup process is
-                               // complete and tell the Service Control Manager that we are now runnng
+                               /* the object that we were waiting for has been destroyed (ABANDONED) or
+                                * signalled (TIMEOUT_0). We can assume that the startup process is
+                                * complete and tell the Service Control Manager that we are now runnng */
                                SLAPDServiceStatus.dwCurrentState       = SERVICE_RUNNING;
                                SLAPDServiceStatus.dwWin32ExitCode      = NO_ERROR;
                                SLAPDServiceStatus.dwCheckPoint++;
@@ -161,16 +225,16 @@ static void *start_status_routine( void *ptr )
                                done = 1;
                                break;
                        case WAIT_TIMEOUT:
-                               // We've waited for the required time, so send an update to the Service Control 
-                               // Manager saying to wait again.
+                               /* We've waited for the required time, so send an update to the Service Control 
+                                * Manager saying to wait again. */
                                SLAPDServiceStatus.dwCheckPoint++;
                                SLAPDServiceStatus.dwWaitHint = SCM_NOTIFICATION_INTERVAL * 2;
                                SetServiceStatus(hSLAPDServiceStatus, &SLAPDServiceStatus);
                                break;
                        case WAIT_FAILED:
-                               // theres been some proble with WaitForSingleObject so tell the Service
-                               // Control Manager to wait 30 seconds before deploying its assasin and 
-                               // then leave the thread.
+                               /* theres been some problem with WaitForSingleObject so tell the Service
+                                * Control Manager to wait 30 seconds before deploying its assasin and 
+                                * then leave the thread. */
                                SLAPDServiceStatus.dwCheckPoint++;
                                SLAPDServiceStatus.dwWaitHint = THIRTY_SECONDS;
                                SetServiceStatus(hSLAPDServiceStatus, &SLAPDServiceStatus);
@@ -196,23 +260,23 @@ static void *stop_status_routine( void *ptr )
                {
                        case WAIT_ABANDONED:
                        case WAIT_OBJECT_0:
-                               // the object that we were waiting for has been destroyed (ABANDONED) or
-                               // signalled (TIMEOUT_0). The shutting down process is therefore complete 
-                               // and the final SERVICE_STOPPED message will be sent to the service control
-                               // manager prior to the process terminating.
+                               /* the object that we were waiting for has been destroyed (ABANDONED) or
+                                * signalled (TIMEOUT_0). The shutting down process is therefore complete 
+                                * and the final SERVICE_STOPPED message will be sent to the service control
+                                * manager prior to the process terminating. */
                                done = 1;
                                break;
                        case WAIT_TIMEOUT:
-                               // We've waited for the required time, so send an update to the Service Control 
-                               // Manager saying to wait again.
+                               /* We've waited for the required time, so send an update to the Service Control 
+                                * Manager saying to wait again. */
                                SLAPDServiceStatus.dwCheckPoint++;
                                SLAPDServiceStatus.dwWaitHint = SCM_NOTIFICATION_INTERVAL * 2;
                                SetServiceStatus(hSLAPDServiceStatus, &SLAPDServiceStatus);
                                break;
                        case WAIT_FAILED:
-                               // theres been some proble with WaitForSingleObject so tell the Service
-                               // Control Manager to wait 30 seconds before deploying its assasin and 
-                               // then leave the thread.
+                               /* theres been some problem with WaitForSingleObject so tell the Service
+                                * Control Manager to wait 30 seconds before deploying its assasin and 
+                                * then leave the thread. */
                                SLAPDServiceStatus.dwCheckPoint++;
                                SLAPDServiceStatus.dwWaitHint = THIRTY_SECONDS;
                                SetServiceStatus(hSLAPDServiceStatus, &SLAPDServiceStatus);
@@ -233,7 +297,7 @@ void WINAPI SLAPDServiceCtrlHandler( IN DWORD Opcode)
        case SERVICE_CONTROL_STOP:
        case SERVICE_CONTROL_SHUTDOWN:
 
-               Debug( LDAP_DEBUG_TRACE, "Service Shutdown ordered\n", 0, 0 );
+               Debug( LDAP_DEBUG_TRACE, "Service Shutdown ordered\n", 0, 0, 0 );
                SLAPDServiceStatus.dwCurrentState       = SERVICE_STOP_PENDING;
                SLAPDServiceStatus.dwCheckPoint++;
                SLAPDServiceStatus.dwWaitHint           = SCM_NOTIFICATION_INTERVAL * 2;
@@ -242,25 +306,25 @@ void WINAPI SLAPDServiceCtrlHandler( IN DWORD Opcode)
                ldap_pvt_thread_cond_init( &stopped_event );
                if ( stopped_event == NULL )
                {
-                       // the event was not created. We will ask the service control manager for 30
-                       // seconds to shutdown
+                       /* the event was not created. We will ask the service control manager for 30
+                        * seconds to shutdown */
                        SLAPDServiceStatus.dwCheckPoint++;
                        SLAPDServiceStatus.dwWaitHint           = THIRTY_SECONDS;
                        SetServiceStatus(hSLAPDServiceStatus, &SLAPDServiceStatus);
                }
                else
                {
-                       // start a thread to report the progress to the service control manager 
-                       // until the stopped_event is fired.
+                       /* start a thread to report the progress to the service control manager 
+                        * until the stopped_event is fired. */
                        if ( ldap_pvt_thread_create( &stop_status_tid, 0, stop_status_routine, NULL ) == 0 )
                        {
                                
                        }
                        else {
-                               // failed to create the thread that tells the Service Control Manager that the
-                               // service stopping is proceeding. 
-                               // tell the Service Control Manager to wait another 30 seconds before deploying its
-                               // assasin.
+                               /* failed to create the thread that tells the Service Control Manager that the
+                                * service stopping is proceeding. 
+                                * tell the Service Control Manager to wait another 30 seconds before deploying its
+                                * assasin.  */
                                SLAPDServiceStatus.dwCheckPoint++;
                                SLAPDServiceStatus.dwWaitHint = THIRTY_SECONDS;
                                SetServiceStatus(hSLAPDServiceStatus, &SLAPDServiceStatus);
@@ -276,52 +340,45 @@ void WINAPI SLAPDServiceCtrlHandler( IN DWORD Opcode)
        return;
 }
 
-
 void *getRegParam( char *svc, char *value )
 {
-       HANDLE hkey;
+       HKEY hkey;
        char path[255];
-       int i = 0, rc;
-       char vName[256];
-       DWORD vNameLen = 255;
        DWORD vType;
        static char vValue[1024];
-       DWORD valLen = 1024;
+       DWORD valLen = sizeof( vValue );
 
        if ( svc != NULL )
-               sprintf ( path, "SOFTWARE\\OpenLDAP\\%s\\Parameters", svc );
+               sprintf ( path, "SOFTWARE\\%s", svc );
        else
                strcpy (path, "SOFTWARE\\OpenLDAP\\Parameters" );
-
-       if( (rc = RegOpenKeyEx( HKEY_LOCAL_MACHINE,
-               path, 0, KEY_READ, &hkey )) != ERROR_SUCCESS )
+       
+       if ( RegOpenKeyEx( HKEY_LOCAL_MACHINE, path, 0, KEY_READ, &hkey ) != ERROR_SUCCESS )
        {
-               //Debug( LDAP_DEBUG_ANY, "%s\n", GetLastErrorString(), 0, 0 );
+               /*Debug( LDAP_DEBUG_ANY, "RegOpenKeyEx() %s\n", GetLastErrorString(), 0, 0); */
                return NULL;
        }
 
-       while ( !RegEnumValue( hkey, i, vName, &vNameLen, NULL,
-               &vType, vValue, &valLen ) )
+       if ( RegQueryValueEx( hkey, value, NULL, &vType, vValue, &valLen ) != ERROR_SUCCESS )
        {
-               if ( !strcasecmp( value, vName ) )
-               {
-                       switch ( vType )
-                       {
-                       case REG_BINARY:
-                       case REG_DWORD:
-                               return (void*)&vValue;
-                       case REG_SZ:
-                               return (void*)strdup( vValue );
-                       }
-               }
-               i++;
-               vNameLen = 255;
-               valLen = 1024;
+               /*Debug( LDAP_DEBUG_ANY, "RegQueryValueEx() %s\n", GetLastErrorString(), 0, 0 );*/
+               RegCloseKey( hkey );
+               return NULL;
+       }
+       RegCloseKey( hkey );
+       
+       switch ( vType )
+       {
+       case REG_BINARY:
+       case REG_DWORD:
+               return (void*)&vValue;
+       case REG_SZ:
+               return (void*)&vValue;
        }
        return (void*)NULL;
 }
 
-void LogSlapdStartedEvent( char *svc, int slap_debug, char *configfile, short port, int udp )
+void LogSlapdStartedEvent( char *svc, int slap_debug, char *configfile, char *urls )
 {
        char *Inserts[5];
        WORD i = 0, j;
@@ -331,13 +388,12 @@ void LogSlapdStartedEvent( char *svc, int slap_debug, char *configfile, short po
 
        Inserts[i] = (char *)malloc( 20 );
        itoa( slap_debug, Inserts[i++], 10 );
-       Inserts[i++] = ldap_pvt_strdup( configfile );
-       Inserts[i] = (char *)malloc( 20 );
-       itoa( port, Inserts[i++], 10 );
-       Inserts[i++] = ldap_pvt_strdup( udp ? "udp" : "tcp" );
-       Inserts[i++] = ldap_pvt_strdup( is_NT_Service ? "svc" : "cmd" );
+       Inserts[i++] = strdup( configfile );
+       Inserts[i++] = strdup( urls ? urls : "ldap:///" );
+       Inserts[i++] = strdup( is_NT_Service ? "svc" : "cmd" );
 
-       ReportEvent( hEventLog, EVENTLOG_INFORMATION_TYPE, 0, MSG_SLAPD_STARTED, NULL, i, 0, Inserts, NULL );
+       ReportEvent( hEventLog, EVENTLOG_INFORMATION_TYPE, 0,
+               MSG_SLAPD_STARTED, NULL, i, 0, (LPCSTR *) Inserts, NULL );
 
        for ( j = 0; j < i; j++ )
                ldap_memfree( Inserts[j] );
@@ -351,7 +407,8 @@ void LogSlapdStoppedEvent( char *svc )
        HANDLE hEventLog;
        
        hEventLog = RegisterEventSource( NULL, svc );
-       ReportEvent( hEventLog, EVENTLOG_INFORMATION_TYPE, 0, MSG_SLAPD_STOPPED, NULL, 0, 0, NULL, NULL );
+       ReportEvent( hEventLog, EVENTLOG_INFORMATION_TYPE, 0,
+               MSG_SLAPD_STOPPED, NULL, 0, 0, NULL, NULL );
        DeregisterEventSource( hEventLog );
 }
 
@@ -374,32 +431,32 @@ void CommenceStartupProcessing( LPCTSTR lpszServiceName,
 
        SetServiceStatus(hSLAPDServiceStatus, &SLAPDServiceStatus);
 
-       // start up a thread to keep sending SERVICE_START_PENDING to the Service Control Manager
-       // until the slapd listener is completed and listening. Only then should we send 
-       // SERVICE_RUNNING to the Service Control Manager.
+       /* start up a thread to keep sending SERVICE_START_PENDING to the Service Control Manager
+        * until the slapd listener is completed and listening. Only then should we send 
+        * SERVICE_RUNNING to the Service Control Manager. */
        ldap_pvt_thread_cond_init( &started_event );
        if ( started_event == NULL)
        {
-               // failed to create the event to determine when the startup process is complete so
-               // tell the Service Control Manager to wait another 30 seconds before deploying its
-               // assasin
+               /* failed to create the event to determine when the startup process is complete so
+                * tell the Service Control Manager to wait another 30 seconds before deploying its
+                * assasin  */
                SLAPDServiceStatus.dwCheckPoint++;
                SLAPDServiceStatus.dwWaitHint = THIRTY_SECONDS;
                SetServiceStatus(hSLAPDServiceStatus, &SLAPDServiceStatus);
        }
        else
        {
-               // start a thread to report the progress to the service control manager 
-               // until the started_event is fired.
+               /* start a thread to report the progress to the service control manager 
+                * until the started_event is fired.  */
                if ( ldap_pvt_thread_create( &start_status_tid, 0, start_status_routine, NULL ) == 0 )
                {
                        
                }
                else {
-                       // failed to create the thread that tells the Service Control Manager that the
-                       // service startup is proceeding. 
-                       // tell the Service Control Manager to wait another 30 seconds before deploying its
-                       // assasin.
+                       /* failed to create the thread that tells the Service Control Manager that the
+                        * service startup is proceeding. 
+                        * tell the Service Control Manager to wait another 30 seconds before deploying its
+                        * assasin.  */
                        SLAPDServiceStatus.dwCheckPoint++;
                        SLAPDServiceStatus.dwWaitHint = THIRTY_SECONDS;
                        SetServiceStatus(hSLAPDServiceStatus, &SLAPDServiceStatus);
@@ -411,12 +468,12 @@ void ReportSlapdShutdownComplete(  )
 {
        if ( is_NT_Service )
        {
-               // stop sending SERVICE_STOP_PENDING messages to the Service Control Manager
+               /* stop sending SERVICE_STOP_PENDING messages to the Service Control Manager */
                ldap_pvt_thread_cond_signal( &stopped_event );
                ldap_pvt_thread_cond_destroy( &stopped_event );
 
-               // wait for the thread sending the SERVICE_STOP_PENDING messages to the Service Control Manager to die.
-               // if the wait fails then put ourselves to sleep for half the Service Control Manager update interval
+               /* wait for the thread sending the SERVICE_STOP_PENDING messages to the Service Control Manager to die.
+                * if the wait fails then put ourselves to sleep for half the Service Control Manager update interval */
                if (ldap_pvt_thread_join( stop_status_tid, (void *) NULL ) == -1)
                        ldap_pvt_thread_sleep( SCM_NOTIFICATION_INTERVAL / 2 );
 
@@ -426,3 +483,22 @@ void ReportSlapdShutdownComplete(  )
                SetServiceStatus(hSLAPDServiceStatus, &SLAPDServiceStatus);
        }
 }
+
+static char *GetErrorString( int err )
+{
+       static char msgBuf[1024];
+
+       FormatMessage(
+               FORMAT_MESSAGE_FROM_SYSTEM,
+               NULL,
+               err, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
+               msgBuf, 1024, NULL );
+
+       return msgBuf;
+}
+
+static char *GetLastErrorString( void )
+{
+       return GetErrorString( GetLastError() );
+}
+#endif