windows-updates-20010108
[openafs.git] / src / WINNT / afsd / afsd_service.c
1 /*
2  * Copyright 2000, International Business Machines Corporation and others.
3  * All Rights Reserved.
4  * 
5  * This software has been released under the terms of the IBM Public
6  * License.  For details, see the LICENSE file in the top-level source
7  * directory or online at http://www.openafs.org/dl/license10.html
8  */
9
10 #include <afs/param.h>
11 #include <afs/stds.h>
12
13 #include <windows.h>
14 #include <string.h>
15 #include <setjmp.h>
16 #include "afsd.h"
17 #include "afsd_init.h"
18 #include <stdio.h>
19 #include <stdlib.h>
20 #include <winsock2.h>
21
22 #include <osi.h>
23 \r
24 #ifdef DEBUG\r
25 //#define NOTSERVICE\r
26 #endif\r
27
28 extern void afsi_log(char *pattern, ...);
29
30 extern char AFSConfigKeyName[];
31
32 HANDLE WaitToTerminate;
33
34 int GlobalStatus;
35
36 unsigned int MainThreadId;
37 jmp_buf notifier_jmp;
38
39 extern int traceOnPanic;
40
41 /*
42  * Notifier function for use by osi_panic
43  */
44 static void afsd_notifier(char *msgp, char *filep, long line)
45 {
46         char tbuffer[100];
47         char *ptbuf[1];
48         HANDLE h;
49
50         if (filep)
51                 sprintf(tbuffer, "Error at file %s, line %d: %s",
52                         filep, line, msgp);
53         else
54                 sprintf(tbuffer, "Error at unknown location: %s", msgp);
55
56         h = RegisterEventSource(NULL, AFS_DAEMON_EVENT_NAME);
57         ptbuf[0] = tbuffer;
58         ReportEvent(h, EVENTLOG_ERROR_TYPE, 0, line, NULL, 1, 0, ptbuf, NULL);
59         DeregisterEventSource(h);
60
61         GlobalStatus = line;
62
63         osi_LogEnable(afsd_logp);
64
65         afsd_ForceTrace(TRUE);
66
67         if (traceOnPanic) {
68                 _asm int 3h;
69         }
70
71         SetEvent(WaitToTerminate);
72
73         if (GetCurrentThreadId() == MainThreadId)
74                 longjmp(notifier_jmp, 1);
75         else
76                 ExitThread(1);
77 }
78
79 /*
80  * For use miscellaneously in smb.c; need to do better
81  */
82 static int DummyMessageBox(HWND h, LPCTSTR l1, LPCTSTR l2, UINT ui)
83 {
84         return 0;
85 }
86
87 static SERVICE_STATUS           ServiceStatus;
88 static SERVICE_STATUS_HANDLE    StatusHandle;
89
90 void afsd_ServiceControlHandler(DWORD ctrlCode)
91 {
92         HKEY parmKey;
93         DWORD dummyLen, doTrace;
94         long code;
95
96         switch (ctrlCode) {
97                 case SERVICE_CONTROL_STOP:
98                         /* Shutdown RPC */
99                         RpcMgmtStopServerListening(NULL);
100
101                         /* Force trace if requested */
102                         code = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
103                                             AFSConfigKeyName,
104                                             0, KEY_QUERY_VALUE, &parmKey);
105                         if (code != ERROR_SUCCESS)
106                                 goto doneTrace;
107
108                         dummyLen = sizeof(doTrace);
109                         code = RegQueryValueEx(parmKey, "TraceOnShutdown",
110                                                 NULL, NULL,
111                                                 (BYTE *) &doTrace, &dummyLen);
112                         RegCloseKey (parmKey);
113                         if (code != ERROR_SUCCESS)
114                                 doTrace = 0;
115                         if (doTrace)
116                                 afsd_ForceTrace(FALSE);
117
118 doneTrace:
119                         ServiceStatus.dwCurrentState = SERVICE_STOP_PENDING;
120                         ServiceStatus.dwWin32ExitCode = NO_ERROR;
121                         ServiceStatus.dwCheckPoint = 1;
122                         ServiceStatus.dwWaitHint = 10000;
123                         ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP;
124                         SetServiceStatus(StatusHandle, &ServiceStatus);
125                         SetEvent(WaitToTerminate);
126                         break;
127                 case SERVICE_CONTROL_INTERROGATE:
128                         ServiceStatus.dwCurrentState = SERVICE_RUNNING;
129                         ServiceStatus.dwWin32ExitCode = NO_ERROR;
130                         ServiceStatus.dwCheckPoint = 0;
131                         ServiceStatus.dwWaitHint = 0;
132                         ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP;
133                         SetServiceStatus(StatusHandle, &ServiceStatus);
134                         break;
135                 /* XXX handle system shutdown */
136                 /* XXX handle pause & continue */
137         }
138 }
139
140 #if 0
141 /* This code was moved to Drivemap.cpp*/
142 /* Mount a drive into AFS if the user wants us to */
143 void CheckMountDrive()
144 {
145         char szAfsPath[_MAX_PATH];
146         char szDriveToMapTo[5];
147         DWORD dwResult;
148         char szKeyName[256];
149         HKEY hKey;
150         DWORD dwIndex = 0;
151         DWORD dwDriveSize;
152         DWORD dwSubMountSize;
153         char szSubMount[256];
154         DWORD dwType;
155
156         sprintf(szKeyName, "%s\\GlobalAutoMapper", AFSConfigKeyName);
157
158         dwResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE, szKeyName, 0, KEY_QUERY_VALUE, &hKey);
159         if (dwResult != ERROR_SUCCESS)
160                 return;
161
162         while (1) {
163                 dwDriveSize = sizeof(szDriveToMapTo);
164                 dwSubMountSize = sizeof(szSubMount);
165                 dwResult = RegEnumValue(hKey, dwIndex++, szDriveToMapTo, &dwDriveSize, 0, &dwType, szSubMount, &dwSubMountSize);
166                 if (dwResult != ERROR_MORE_DATA) {
167                         if (dwResult != ERROR_SUCCESS) {
168                                 if (dwResult != ERROR_NO_MORE_ITEMS)
169                                         afsi_log("Failed to read GlobalAutoMapper values: %d\n", dwResult);
170                                 break;
171                         }
172                 }
173                 
174                 sprintf(szAfsPath, "\\Device\\LanmanRedirector\\%s\\%s-AFS\\%s", szDriveToMapTo, cm_HostName, szSubMount);
175         
176                 dwResult = DefineDosDevice(DDD_RAW_TARGET_PATH, szDriveToMapTo, szAfsPath);
177                 afsi_log("GlobalAutoMap of %s to %s %s", szDriveToMapTo, szSubMount, dwResult ? "succeeded" : "failed");
178         }        
179
180         RegCloseKey(hKey);
181 }
182 #endif
183
184 void afsd_Main()
185 {
186         long code;
187         char *reason;
188         int jmpret;
189
190         osi_InitPanic(afsd_notifier);
191         osi_InitTraceOption();
192
193         GlobalStatus = 0;
194
195         WaitToTerminate = CreateEvent(NULL, TRUE, FALSE, NULL);
196
197 #ifndef NOTSERVICE\r
198         StatusHandle = RegisterServiceCtrlHandler(AFS_DAEMON_SERVICE_NAME,
199                         (LPHANDLER_FUNCTION) afsd_ServiceControlHandler);
200
201         ServiceStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
202         ServiceStatus.dwServiceSpecificExitCode = 0;
203         ServiceStatus.dwCurrentState = SERVICE_START_PENDING;
204         ServiceStatus.dwWin32ExitCode = NO_ERROR;
205         ServiceStatus.dwCheckPoint = 1;
206         ServiceStatus.dwWaitHint = 15000;
207         ServiceStatus.dwControlsAccepted = 0;
208         SetServiceStatus(StatusHandle, &ServiceStatus);
209 #endif
210 {       
211         HANDLE h; char *ptbuf[1];
212         h = RegisterEventSource(NULL, AFS_DAEMON_EVENT_NAME);
213         ptbuf[0] = "AFS start pending";
214         ReportEvent(h, EVENTLOG_INFORMATION_TYPE, 0, 0, NULL, 1, 0, ptbuf, NULL);
215         DeregisterEventSource(h);
216 }
217
218         afsi_start();
219
220         MainThreadId = GetCurrentThreadId();
221         jmpret = setjmp(notifier_jmp);
222
223         if (jmpret == 0) {
224                 code = afsd_InitCM(&reason);
225                 if (code != 0)
226                         osi_panic(reason, __FILE__, __LINE__);
227
228                 code = afsd_InitDaemons(&reason);
229                 if (code != 0)
230                         osi_panic(reason, __FILE__, __LINE__);
231
232                 code = afsd_InitSMB(&reason, DummyMessageBox);
233                 if (code != 0)
234                         osi_panic(reason, __FILE__, __LINE__);
235
236 #ifndef NOTSERVICE
237                 ServiceStatus.dwCurrentState = SERVICE_RUNNING;
238                 ServiceStatus.dwWin32ExitCode = NO_ERROR;
239                 ServiceStatus.dwCheckPoint = 0;
240                 ServiceStatus.dwWaitHint = 0;
241                 ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP;
242                 SetServiceStatus(StatusHandle, &ServiceStatus);
243 #endif
244         {
245                 HANDLE h; char *ptbuf[1];
246                 h = RegisterEventSource(NULL, AFS_DAEMON_EVENT_NAME);
247                 ptbuf[0] = "AFS running";
248                 ReportEvent(h, EVENTLOG_INFORMATION_TYPE, 0, 0, NULL, 1, 0, ptbuf, NULL);
249                 DeregisterEventSource(h);
250         }
251         }
252
253         /* Check if we should mount a drive into AFS */
254 /*        CheckMountDrive();*/
255
256         WaitForSingleObject(WaitToTerminate, INFINITE);
257         
258 {   
259         HANDLE h; char *ptbuf[1];
260         h = RegisterEventSource(NULL, AFS_DAEMON_EVENT_NAME);
261         ptbuf[0] = "AFS quitting";
262         ReportEvent(h,
263                 GlobalStatus ? EVENTLOG_ERROR_TYPE : EVENTLOG_INFORMATION_TYPE,
264                 0, 0, NULL, 1, 0, ptbuf, NULL);
265         DeregisterEventSource(h);
266 }
267
268         ServiceStatus.dwCurrentState = SERVICE_STOPPED;
269         ServiceStatus.dwWin32ExitCode = NO_ERROR;
270         ServiceStatus.dwCheckPoint = 0;
271         ServiceStatus.dwWaitHint = 0;
272         ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP;
273         SetServiceStatus(StatusHandle, &ServiceStatus);
274 }
275
276 #ifdef NOTSERVICE
277 void main()
278 {
279         afsd_Main();
280         Sleep(1000);
281         return ;
282 }
283 #else
284 void _CRTAPI1 main()
285 {
286         LONG status = ERROR_SUCCESS;
287         SERVICE_TABLE_ENTRY dispatchTable[] = {
288                 {AFS_DAEMON_SERVICE_NAME, (LPSERVICE_MAIN_FUNCTION) afsd_Main},
289                 {NULL, NULL}
290         };
291
292         if (!StartServiceCtrlDispatcher(dispatchTable))
293                 status = GetLastError();
294 }
295 #endif