windows-updates-including-dont-lose-tokens-20011110
[openafs.git] / src / WINNT / client_creds / advtab.cpp
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 extern "C" {
11 #include <afs/param.h>
12 #include <afs/stds.h>
13 }
14
15 #include "afscreds.h"
16
17
18 /*
19  * PROTOTYPES _________________________________________________________________
20  *
21  */
22
23 void Advanced_OnInitDialog (HWND hDlg);
24 void Advanced_StartTimer (HWND hDlg);
25 void Advanced_OnServiceTimer (HWND hDlg);
26 void Advanced_OnOpenCPL (HWND hDlg);
27 void Advanced_OnChangeService (HWND hDlg, WORD wCmd);
28 void Advanced_OnStartup (HWND hDlg);
29
30
31 /*
32  * ROUTINES ___________________________________________________________________
33  *
34  */
35
36 BOOL CALLBACK Advanced_DlgProc (HWND hDlg, UINT msg, WPARAM wp, LPARAM lp)
37 {
38    switch (msg)
39       {
40       case WM_INITDIALOG:
41          RECT rTab;
42          GetClientRect (GetParent(hDlg), &rTab);
43          TabCtrl_AdjustRect (GetParent (hDlg), FALSE, &rTab); 
44          SetWindowPos (hDlg, NULL, rTab.left, rTab.top, 0, 0, SWP_NOSIZE | SWP_NOACTIVATE | SWP_NOZORDER);
45
46          Advanced_OnInitDialog (hDlg);
47          break;
48
49       case WM_TIMER:
50          Advanced_OnServiceTimer (hDlg);
51          break;
52
53       case WM_COMMAND:
54          switch (LOWORD(wp))
55             {
56             case IDC_SERVICE_STOP:
57             case IDC_SERVICE_START:
58             case IDC_SERVICE_AUTO:
59                Advanced_OnChangeService (hDlg, LOWORD(wp));
60                break;
61
62             case IDC_OPEN_CPL:
63                Advanced_OnOpenCPL (hDlg);
64                break;
65
66             case IDC_STARTUP:
67                Advanced_OnStartup (hDlg);
68                break;
69
70             case IDHELP:
71                Advanced_DlgProc (hDlg, WM_HELP, 0, 0);
72                break;
73             }
74          break;
75
76       case WM_HELP:
77          WinHelp (hDlg, g.szHelpFile, HELP_CONTEXT, IDH_AFSCREDS_TAB_ADVANCED);
78          break;
79       }
80    return FALSE;
81 }
82
83
84 void Advanced_OnInitDialog (HWND hDlg)
85 {
86    CheckDlgButton (hDlg, IDC_STARTUP, g.fStartup);
87    Advanced_OnServiceTimer (hDlg);
88    Advanced_StartTimer (hDlg);
89 }
90
91
92 void Advanced_StartTimer (HWND hDlg)
93 {
94    KillTimer (hDlg, ID_SERVICE_TIMER);
95    SetTimer (hDlg, ID_SERVICE_TIMER, (ULONG)cmsecSERVICE, NULL);
96 }
97
98
99 void Advanced_OnServiceTimer (HWND hDlg)
100 {
101    BOOL fFinal = TRUE;
102    BOOL fFound = FALSE;
103
104    struct {
105       QUERY_SERVICE_CONFIG Config;
106       BYTE buf[1024];
107    } Config;
108    memset (&Config, 0x00, sizeof(Config));
109    Config.Config.dwStartType = SERVICE_DEMAND_START;
110
111    SERVICE_STATUS Status;
112    memset (&Status, 0x00, sizeof(Status));
113    Status.dwCurrentState = SERVICE_STOPPED;
114
115    SC_HANDLE hManager;
116    if ((hManager = OpenSCManager (NULL, NULL, GENERIC_READ)) != NULL)
117       {
118       SC_HANDLE hService;
119       if ((hService = OpenService (hManager, TEXT("TransarcAFSDaemon"), GENERIC_READ)) != NULL)
120          {
121          fFound = TRUE;
122          DWORD dwSize = sizeof(Config);
123          QueryServiceConfig (hService, (QUERY_SERVICE_CONFIG*)&Config, sizeof(Config), &dwSize);
124          QueryServiceStatus (hService, &Status);
125                  TestAndDoMapShare(Status.dwCurrentState);
126
127          CloseServiceHandle (hService);
128          }
129
130       CloseServiceHandle (hManager);
131       }
132
133    if (fFound)
134       {
135       if ((Status.dwCurrentState == SERVICE_STOP_PENDING) ||
136           (Status.dwCurrentState == SERVICE_START_PENDING))
137          {
138          fFinal = FALSE;
139          }
140       }
141
142    EnableWindow (GetDlgItem (hDlg, IDC_SERVICE_START), fFound && (Status.dwCurrentState == SERVICE_STOPPED));
143    EnableWindow (GetDlgItem (hDlg, IDC_SERVICE_STOP), fFound && (Status.dwCurrentState == SERVICE_RUNNING));
144    EnableWindow (GetDlgItem (hDlg, IDC_SERVICE_AUTO), fFound);
145    CheckDlgButton (hDlg, IDC_SERVICE_AUTO, fFound && (Config.Config.dwStartType == SERVICE_AUTO_START));
146
147    TCHAR szStatus[cchRESOURCE];
148    if (!fFound)
149       GetString (szStatus, IDS_SERVICE_BROKEN);
150    else if (Status.dwCurrentState == SERVICE_RUNNING)
151       GetString (szStatus, IDS_SERVICE_RUNNING);
152    else if (Status.dwCurrentState == SERVICE_STOPPED)
153       GetString (szStatus, IDS_SERVICE_STOPPED);
154    else if (Status.dwCurrentState == SERVICE_STOP_PENDING)
155       GetString (szStatus, IDS_SERVICE_STOPPING);
156    else if (Status.dwCurrentState == SERVICE_START_PENDING)
157       GetString (szStatus, IDS_SERVICE_STARTING);
158    else
159       GetString (szStatus, IDS_SERVICE_UNKNOWN);
160    TestAndDoMapShare(Status.dwCurrentState);
161    SetDlgItemText (hDlg, IDC_SERVICE_STATUS, szStatus);
162
163    if (fFinal && GetWindowLong (hDlg, DWL_USER))
164       {
165       SetWindowLong (hDlg, DWL_USER, 0);
166       Main_RepopulateTabs (FALSE);
167       }
168
169    if (fFinal)
170       {
171       KillTimer (hDlg, ID_SERVICE_TIMER);
172       }
173 }
174
175
176 void Advanced_OnChangeService (HWND hDlg, WORD wCmd)
177 {
178    BOOL fSuccess = FALSE;
179    ULONG error = 0;
180
181    SC_HANDLE hManager;
182    if ((hManager = OpenSCManager (NULL, NULL, SC_MANAGER_ALL_ACCESS)) != NULL)
183       {
184       SC_HANDLE hService;
185       if ((hService = OpenService (hManager, TEXT("TransarcAFSDaemon"), SERVICE_ALL_ACCESS)) != NULL)
186          {
187          switch (wCmd)
188             {
189             case IDC_SERVICE_AUTO:
190                DWORD StartType;
191                StartType = (IsDlgButtonChecked (hDlg, wCmd)) ? SERVICE_AUTO_START : SERVICE_DEMAND_START;
192
193                if (ChangeServiceConfig (hService, SERVICE_NO_CHANGE, StartType, SERVICE_NO_CHANGE, 0, 0, 0, 0, 0, 0, 0))
194                   fSuccess = TRUE;
195                break;
196
197             case IDC_SERVICE_START:
198                if (StartService (hService, 0, 0))
199                            {
200                                   TestAndDoMapShare(SERVICE_START_PENDING);
201                   fSuccess = TRUE;
202                            }
203                break;
204
205             case IDC_SERVICE_STOP:
206                SERVICE_STATUS Status;
207                if (ControlService (hService, SERVICE_CONTROL_STOP, &Status))
208                                    TestAndDoUnMapShare();
209                   fSuccess = TRUE;
210                break;
211             }
212
213          CloseServiceHandle (hService);
214          }
215
216       CloseServiceHandle (hManager);
217       }
218
219    if (fSuccess)
220       {
221       if (wCmd == IDC_SERVICE_STOP)
222          SetWindowLong (hDlg, DWL_USER, TRUE);
223       Advanced_OnServiceTimer (hDlg);
224       Advanced_StartTimer (hDlg);
225       }
226    else
227       {
228       if (!error)
229          error = GetLastError();
230
231       int ids;
232       switch (wCmd)
233          {
234          case IDC_SERVICE_START:
235             ids = IDS_SERVICE_FAIL_START;
236             break;
237          case IDC_SERVICE_STOP:
238             ids = IDS_SERVICE_FAIL_STOP;
239             break;
240          default:
241             ids = IDS_SERVICE_FAIL_CONFIG;
242             break;
243          }
244
245       Message (MB_OK | MB_ICONHAND, IDS_SERVICE_ERROR, ids, TEXT("%08lX"), error);
246       }
247 }
248
249
250 void Advanced_OnOpenCPL (HWND hDlg)
251 {
252    WinExec ("afs_config.exe", SW_SHOW);
253 }
254
255
256 void Advanced_OnStartup (HWND hDlg)
257 {
258    g.fStartup = IsDlgButtonChecked (hDlg, IDC_STARTUP);
259
260    HKEY hk;
261    if (RegCreateKey (HKEY_LOCAL_MACHINE, TEXT("System\\CurrentControlSet\\Services\\TransarcAFSDaemon\\Parameters"), &hk) == 0)
262       {
263       DWORD dwSize = sizeof(g.fStartup);
264       DWORD dwType = REG_DWORD;
265       RegSetValueEx (hk, TEXT("ShowTrayIcon"), NULL, dwType, (PBYTE)&g.fStartup, dwSize);
266       RegCloseKey (hk);
267       }
268
269    Shortcut_FixStartup (cszSHORTCUT_NAME, g.fStartup);
270 }
271