1/*	$NetBSD$	*/
2
3/* OpenLDAP: pkg/ldap/servers/slapd/nt_svc.c,v 1.27.2.5 2010/04/13 20:23:17 kurt Exp */
4/* This work is part of OpenLDAP Software <http://www.openldap.org/>.
5 *
6 * Copyright 1998-2010 The OpenLDAP Foundation.
7 * All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted only as authorized by the OpenLDAP
11 * Public License.
12 *
13 * A copy of this license is available in the file LICENSE in the
14 * top-level directory of the distribution or, alternatively, at
15 * <http://www.OpenLDAP.org/license.html>.
16 */
17
18#include "portable.h"
19#include <stdio.h>
20#include <ac/string.h>
21#include "slap.h"
22#include "lutil.h"
23
24#ifdef HAVE_NT_SERVICE_MANAGER
25
26/* in main.c */
27void WINAPI ServiceMain( DWORD argc, LPTSTR *argv );
28
29/* in ntservice.c */
30int main( int argc, LPTSTR *argv )
31{
32	int		length;
33	char	filename[MAX_PATH], *fname_start;
34
35	/*
36	 * Because the service was registered as SERVICE_WIN32_OWN_PROCESS,
37	 * the lpServiceName element of the SERVICE_TABLE_ENTRY will be
38	 * ignored.
39	 */
40
41	SERVICE_TABLE_ENTRY		DispatchTable[] = {
42		{	"",	(LPSERVICE_MAIN_FUNCTION) ServiceMain	},
43		{	NULL,			NULL	}
44	};
45
46	/*
47	 * set the service's current directory to the installation directory
48	 * for the service. this way we don't have to write absolute paths
49	 * in the configuration files
50	 */
51	GetModuleFileName( NULL, filename, sizeof( filename ) );
52	fname_start = strrchr( filename, *LDAP_DIRSEP );
53
54	if ( argc > 1 ) {
55		if ( _stricmp( "install", argv[1] ) == 0 )
56		{
57			char *svcName = SERVICE_NAME;
58			char *displayName = "OpenLDAP Directory Service";
59			BOOL auto_start = FALSE;
60
61			if ( (argc > 2) && (argv[2] != NULL) )
62				svcName = argv[2];
63
64			if ( argc > 3 && argv[3])
65				displayName = argv[3];
66
67			if ( argc > 4 && stricmp(argv[4], "auto") == 0)
68				auto_start = TRUE;
69
70			strcat(filename, " service");
71			if ( !lutil_srv_install(svcName, displayName, filename, auto_start) )
72			{
73				fputs( "service failed installation ...\n", stderr  );
74				return EXIT_FAILURE;
75			}
76			fputs( "service has been installed ...\n", stderr  );
77			return EXIT_SUCCESS;
78		}
79
80		if ( _stricmp( "remove", argv[1] ) == 0 )
81		{
82			char *svcName = SERVICE_NAME;
83			if ( (argc > 2) && (argv[2] != NULL) )
84				svcName = argv[2];
85			if ( !lutil_srv_remove(svcName, filename) )
86			{
87				fputs( "failed to remove the service ...\n", stderr  );
88				return EXIT_FAILURE;
89			}
90			fputs( "service has been removed ...\n", stderr );
91			return EXIT_SUCCESS;
92		}
93		if ( _stricmp( "service", argv[1] ) == 0 )
94		{
95			is_NT_Service = 1;
96			*fname_start = '\0';
97			SetCurrentDirectory( filename );
98		}
99	}
100
101	if (is_NT_Service)
102	{
103		StartServiceCtrlDispatcher(DispatchTable);
104	} else
105	{
106		ServiceMain( argc, argv );
107	}
108
109	return EXIT_SUCCESS;
110}
111
112#endif
113