[jbossnative-commits] JBoss Native SVN: r2690 - in tags/jbosssvc: 2.0.8/procrun/apps/prunmgr and 2 other directories.
jbossnative-commits at lists.jboss.org
jbossnative-commits at lists.jboss.org
Wed Jul 21 16:14:08 EDT 2010
Author: mladen.turk at jboss.com
Date: 2010-07-21 16:14:05 -0400 (Wed, 21 Jul 2010)
New Revision: 2690
Added:
tags/jbosssvc/2.0.8/
tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.c
tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.h
tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.rc
tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.c
tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.h
tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.rc
tags/jbosssvc/2.0.8/procrun/src/cmdline.c
tags/jbosssvc/2.0.8/procrun/src/handles.c
tags/jbosssvc/2.0.8/procrun/src/javajni.c
tags/jbosssvc/2.0.8/procrun/src/log.c
tags/jbosssvc/2.0.8/procrun/src/mclib.c
tags/jbosssvc/2.0.8/procrun/src/private.h
tags/jbosssvc/2.0.8/procrun/src/registry.c
tags/jbosssvc/2.0.8/procrun/src/rprocess.c
tags/jbosssvc/2.0.8/procrun/src/service.c
tags/jbosssvc/2.0.8/procrun/src/utils.c
Removed:
tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.c
tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.h
tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.rc
tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.c
tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.h
tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.rc
tags/jbosssvc/2.0.8/procrun/src/cmdline.c
tags/jbosssvc/2.0.8/procrun/src/handles.c
tags/jbosssvc/2.0.8/procrun/src/javajni.c
tags/jbosssvc/2.0.8/procrun/src/log.c
tags/jbosssvc/2.0.8/procrun/src/mclib.c
tags/jbosssvc/2.0.8/procrun/src/private.h
tags/jbosssvc/2.0.8/procrun/src/registry.c
tags/jbosssvc/2.0.8/procrun/src/rprocess.c
tags/jbosssvc/2.0.8/procrun/src/service.c
tags/jbosssvc/2.0.8/procrun/src/utils.c
Log:
Tag 2.0.8
Copied: tags/jbosssvc/2.0.8 (from rev 2688, trunk/utils/windows/native/service)
Deleted: tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.c
===================================================================
--- trunk/utils/windows/native/service/procrun/apps/prunmgr/prunmgr.c 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,1740 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-/* ====================================================================
- * prunsrv -- Service Runner.
- * Contributed by Mladen Turk <mturk at apache.org>
- * 05 Aug 2003
- * ====================================================================
- */
-
-/* Force the JNI vprintf functions */
-#define _DEBUG_JNI 1
-#include "apxwin.h"
-#include "prunmgr.h"
-
-LPAPXGUISTORE _gui_store = NULL;
-#define PRUNMGR_CLASS TEXT("PRUNMGR")
-#define TMNU_CONF TEXT("Configure...")
-#define TMNU_START TEXT("Start service")
-#define TMNU_STOP TEXT("Stop service")
-#define TMNU_EXIT TEXT("Exit")
-#define TMNU_ABOUT TEXT("About")
-#define TMNU_DUMP TEXT("Thread Dump")
-
-/* Display only Started/Paused status */
-#define STAT_STARTED TEXT("Started")
-#define STAT_PAUSED TEXT("Paused")
-#define STAT_STOPPED TEXT("Stopped")
-#define STAT_DISABLED TEXT("Disabled")
-#define STAT_NONE TEXT("")
-#define STAT_SYSTEM L"LocalSystem"
-
-#define LOGL_ERROR L"Error"
-#define LOGL_DEBUG L"Debug"
-#define LOGL_INFO L"Info"
-#define LOGL_WARN L"Warning"
-
-
-#define START_AUTO L"Automatic"
-#define START_MANUAL L"Manual"
-#define START_DISABLED L"Disabled"
-#define START_BOOT L"Boot"
-#define START_SYSTEM L"SystemInit"
-#define EMPTY_PASSWORD L" "
-
-#ifdef WIN64
-#define KREG_WOW6432 KEY_WOW64_32KEY
-#else
-#define KREG_WOW6432 0
-#endif
-
-/* Main application pool */
-APXHANDLE hPool = NULL;
-APXHANDLE hService = NULL;
-APXHANDLE hRegistry = NULL;
-APXHANDLE hRegserv = NULL;
-HICON hIcoRun = NULL;
-HICON hIcoStop = NULL;
-
-LPAPXSERVENTRY _currentEntry = NULL;
-
-BOOL bEnableTry = FALSE;
-DWORD startPage = 0;
-
-static LPCWSTR _s_log = L"Log";
-static LPCWSTR _s_java = L"Java";
-static LPCWSTR _s_start = L"Start";
-static LPCWSTR _s_stop = L"Stop";
-
-/* Allowed procrun commands */
-static LPCWSTR _commands[] = {
- L"ES", /* 1 Manage Service (default)*/
- L"MS", /* 2 Monitor Service */
- L"MR", /* 3 Monitor Service and start if not runing */
- L"MQ", /* 4 Quit all running Monitor applications */
- NULL
-};
-
-/* Allowed procrun parameters */
-static APXCMDLINEOPT _options[] = {
-/* 0 */ { L"AllowMultiInstances", NULL, NULL, APXCMDOPT_INT, NULL, 0},
- /* NULL terminate the array */
- { NULL }
-};
-
-/* Create RBUTTON try menu
- * Configure... (default, or lbutton dblclick)
- * Start <service name>
- * Stop <service name>
- * Exit
- * Logo
- */
-static void createRbuttonTryMenu(HWND hWnd)
-{
- HMENU hMnu;
- POINT pt;
- BOOL canStop = FALSE;
- BOOL canStart = FALSE;
- hMnu = CreatePopupMenu();
-
- if (_currentEntry) {
- if (_currentEntry->stServiceStatus.dwCurrentState == SERVICE_RUNNING) {
- if (_currentEntry->stServiceStatus.dwControlsAccepted & SERVICE_ACCEPT_STOP)
- canStop = TRUE;
- }
- else if (_currentEntry->stServiceStatus.dwCurrentState == SERVICE_STOPPED) {
- if (_currentEntry->lpConfig->dwStartType != SERVICE_DISABLED)
- canStart = TRUE;
- }
- }
- apxAppendMenuItem(hMnu, IDM_TM_CONFIG, TMNU_CONF, TRUE, TRUE);
- apxAppendMenuItem(hMnu, IDM_TM_START, TMNU_START, FALSE, canStart);
- apxAppendMenuItem(hMnu, IDM_TM_STOP, TMNU_STOP, FALSE, canStop);
- apxAppendMenuItem(hMnu, IDM_TM_DUMP, TMNU_DUMP, FALSE, canStop);
- apxAppendMenuItem(hMnu, IDM_TM_EXIT, TMNU_EXIT, FALSE, TRUE);
- apxAppendMenuItem(hMnu, -1, NULL, FALSE, FALSE);
- apxAppendMenuItem(hMnu, IDM_TM_ABOUT, TMNU_ABOUT, FALSE, TRUE);
-
- /* Ensure we have a focus */
- if (!SetForegroundWindow(hWnd))
- SetForegroundWindow(NULL);
- GetCursorPos(&pt);
- /* Display the try menu */
- TrackPopupMenu(hMnu, TPM_LEFTALIGN | TPM_RIGHTBUTTON,
- pt.x, pt.y, 0, hWnd, NULL);
- DestroyMenu(hMnu);
-}
-
-/* wParam progress dialog handle
- */
-static BOOL __startServiceCallback(APXHANDLE hObject, UINT uMsg,
- WPARAM wParam, LPARAM lParam)
-{
- HWND hDlg = (HWND)hObject;
-
- switch (uMsg) {
- case WM_USER+1:
- hDlg = (HWND)lParam;
- if (IS_INVALID_HANDLE(hService)) {
- EndDialog(hDlg, IDOK);
- PostMessage(_gui_store->hMainWnd, WM_COMMAND,
- MAKEWPARAM(IDMS_REFRESH, 0), 0);
- return FALSE;
- }
- if (apxServiceControl(hService, SERVICE_CONTROL_CONTINUE, WM_USER+2,
- __startServiceCallback, hDlg)) {
- _currentEntry->stServiceStatus.dwCurrentState = SERVICE_RUNNING;
- _currentEntry->stStatusProcess.dwCurrentState = SERVICE_RUNNING;
-
- }
- EndDialog(hDlg, IDOK);
- PostMessage(_gui_store->hMainWnd, WM_COMMAND,
- MAKEWPARAM(IDMS_REFRESH, 0), 0);
- break;
- case WM_USER+2:
- SendMessage(hDlg, WM_USER+1, 0, 0);
- Sleep(500);
- break;
- }
- return TRUE;
-}
-
-static BOOL __stopServiceCallback(APXHANDLE hObject, UINT uMsg,
- WPARAM wParam, LPARAM lParam)
-{
- HWND hDlg = (HWND)hObject;
-
- switch (uMsg) {
- case WM_USER+1:
- hDlg = (HWND)lParam;
- if (IS_INVALID_HANDLE(hService))
- return FALSE;
- if (apxServiceControl(hService, SERVICE_CONTROL_STOP, WM_USER+2,
- __stopServiceCallback, hDlg)) {
- }
- EndDialog(hDlg, IDOK);
- PostMessage(_gui_store->hMainWnd, WM_COMMAND,
- MAKEWPARAM(IDMS_REFRESH, 0), 0);
- break;
- case WM_USER+2:
- if (wParam == 4)
- AplCopyMemory(&_currentEntry->stServiceStatus,
- (LPVOID)lParam, sizeof(SERVICE_STATUS));
- SendMessage(hDlg, WM_USER+1, 0, 0);
- Sleep(100);
- break;
- }
- return TRUE;
-}
-
-static BOOL __restartServiceCallback(APXHANDLE hObject, UINT uMsg,
- WPARAM wParam, LPARAM lParam)
-{
- HWND hDlg = (HWND)hObject;
- switch (uMsg) {
- case WM_USER+1:
- hDlg = (HWND)lParam;
- if (IS_INVALID_HANDLE(hService))
- return FALSE;
- /* TODO: use 128 as controll code */
- if (apxServiceControl(hService, 128, WM_USER+2,
- __restartServiceCallback, hDlg)) {
-
- }
- EndDialog(hDlg, IDOK);
- PostMessage(_gui_store->hMainWnd, WM_COMMAND,
- MAKEWPARAM(IDMS_REFRESH, 0), 0);
- break;
- case WM_USER+2:
- if (wParam == 4)
- AplCopyMemory(&_currentEntry->stServiceStatus,
- (LPVOID)lParam, sizeof(SERVICE_STATUS));
-
- SendMessage(hDlg, WM_USER+1, 0, 0);
- Sleep(100);
- break;
- }
- return TRUE;
-}
-
-static BOOL __pauseServiceCallback(APXHANDLE hObject, UINT uMsg,
- WPARAM wParam, LPARAM lParam)
-{
- HWND hDlg = (HWND)hObject;
- switch (uMsg) {
- case WM_USER+1:
- hDlg = (HWND)lParam;
- if (IS_INVALID_HANDLE(hService))
- return FALSE;
- if (apxServiceControl(hService, SERVICE_CONTROL_PAUSE, WM_USER+2,
- __pauseServiceCallback, hDlg)) {
- }
- EndDialog(hDlg, IDOK);
- PostMessage(_gui_store->hMainWnd, WM_COMMAND,
- MAKEWPARAM(IDMS_REFRESH, 0), 0);
- break;
- case WM_USER+2:
- if (wParam == 4)
- AplCopyMemory(&_currentEntry->stServiceStatus,
- (LPVOID)lParam, sizeof(SERVICE_STATUS));
- SendMessage(hDlg, WM_USER+1, 0, 0);
- Sleep(100);
- break;
- }
- return TRUE;
-}
-
-static DWORD _propertyChanged;
-static BOOL _propertyOpened = FALSE;
-static HWND _propertyHwnd = NULL;
-/* Service property pages */
-int CALLBACK __propertyCallback(HWND hwndPropSheet, UINT uMsg, LPARAM lParam)
-{
- switch(uMsg) {
- case PSCB_PRECREATE:
- {
- LPDLGTEMPLATE lpTemplate = (LPDLGTEMPLATE)lParam;
- if (!(lpTemplate->style & WS_SYSMENU))
- lpTemplate->style |= WS_SYSMENU;
- _propertyHwnd = hwndPropSheet;
-
- _propertyChanged = 0;
- _propertyOpened = TRUE;
- return TRUE;
- }
- break;
- case PSCB_INITIALIZED:
- break;
- }
- return TRUE;
-}
-
-BOOL __generalPropertySave(HWND hDlg)
-{
- WCHAR szN[256];
- WCHAR szD[256];
- DWORD dwStartType = SERVICE_NO_CHANGE;
- int i;
-
- if (!(TST_BIT_FLAG(_propertyChanged, 1)))
- return TRUE;
- CLR_BIT_FLAG(_propertyChanged, 1);
-
- if (IS_INVALID_HANDLE(hService))
- return FALSE;
- GetDlgItemTextW(hDlg, IDC_PPSGDISP, szN, 255);
- GetDlgItemTextW(hDlg, IDC_PPSGDESC, szD, 1023);
- i = ComboBox_GetCurSel(GetDlgItem(hDlg, IDC_PPSGCMBST));
- if (i == 0)
- dwStartType = SERVICE_AUTO_START;
- else if (i == 1)
- dwStartType = SERVICE_DEMAND_START;
- else if (i == 2)
- dwStartType = SERVICE_DISABLED;
- apxServiceSetNames(hService, NULL, szN, szD, NULL, NULL);
- apxServiceSetOptions(hService, SERVICE_NO_CHANGE, dwStartType, SERVICE_NO_CHANGE);
-
- if (!(TST_BIT_FLAG(_propertyChanged, 2)))
- PostMessage(_gui_store->hMainWnd, WM_COMMAND, MAKEWPARAM(IDMS_REFRESH, 0), 0);
-
- return TRUE;
-}
-
-BOOL __generalLogonSave(HWND hDlg)
-{
- WCHAR szU[64];
- WCHAR szP[64];
- WCHAR szC[64];
- DWORD dwStartType = SERVICE_NO_CHANGE;
-
- if (!(TST_BIT_FLAG(_propertyChanged, 2)))
- return TRUE;
- CLR_BIT_FLAG(_propertyChanged, 2);
-
- if (IS_INVALID_HANDLE(hService))
- return FALSE;
- GetDlgItemTextW(hDlg, IDC_PPSLUSER, szU, 63);
- GetDlgItemTextW(hDlg, IDC_PPSLPASS, szP, 63);
- GetDlgItemTextW(hDlg, IDC_PPSLCPASS, szC, 63);
-
- if (lstrlenW(szU) && lstrcmpiW(szU, STAT_SYSTEM)) {
- if (szP[0] != L' ' && szC[0] != L' ' && !lstrcmpW(szP, szC))
- apxServiceSetNames(hService, NULL, NULL, NULL, szU, szP);
- else {
- MessageBoxW(hDlg, apxLoadResourceW(IDS_VALIDPASS, 0),
- apxLoadResourceW(IDS_APPLICATION, 1),
- MB_OK | MB_ICONSTOP);
- return FALSE;
- }
- }
- else {
- apxServiceSetNames(hService, NULL, NULL, NULL, STAT_SYSTEM, L"");
- if (IsDlgButtonChecked(hDlg, IDC_PPSLID) == BST_CHECKED)
- apxServiceSetOptions(hService,
- _currentEntry->stServiceStatus.dwServiceType | SERVICE_INTERACTIVE_PROCESS,
- SERVICE_NO_CHANGE, SERVICE_NO_CHANGE);
- else
- apxServiceSetOptions(hService,
- _currentEntry->stServiceStatus.dwServiceType & ~SERVICE_INTERACTIVE_PROCESS,
- SERVICE_NO_CHANGE, SERVICE_NO_CHANGE);
- }
- if (!(TST_BIT_FLAG(_propertyChanged, 1)))
- PostMessage(_gui_store->hMainWnd, WM_COMMAND, MAKEWPARAM(IDMS_REFRESH, 0), 0);
- return TRUE;
-}
-
-BOOL __generalLoggingSave(HWND hDlg)
-{
- WCHAR szB[SIZ_DESLEN];
-
- if (!(TST_BIT_FLAG(_propertyChanged, 3)))
- return TRUE;
- CLR_BIT_FLAG(_propertyChanged, 3);
-
- if (IS_INVALID_HANDLE(hService))
- return FALSE;
-
- GetDlgItemTextW(hDlg, IDC_PPLGLEVEL, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_log, L"Level", szB);
- GetDlgItemTextW(hDlg, IDC_PPLGPATH, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_log, L"Path", szB);
- GetDlgItemTextW(hDlg, IDC_PPLGPREFIX, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_log, L"Prefix", szB);
- GetDlgItemTextW(hDlg, IDC_PPLGSTDOUT, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_log, L"StdOutput", szB);
- GetDlgItemTextW(hDlg, IDC_PPLGSTDERR, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_log, L"StdError", szB);
-
- if (!(TST_BIT_FLAG(_propertyChanged, 1)))
- PostMessage(_gui_store->hMainWnd, WM_COMMAND, MAKEWPARAM(IDMS_REFRESH, 0), 0);
- return TRUE;
-}
-
-BOOL __generalJvmSave(HWND hDlg)
-{
- WCHAR szB[SIZ_DESLEN];
- LPWSTR p, s;
- DWORD l;
- if (!(TST_BIT_FLAG(_propertyChanged, 4)))
- return TRUE;
- CLR_BIT_FLAG(_propertyChanged, 4);
-
- if (IS_INVALID_HANDLE(hService))
- return FALSE;
- if (!IsDlgButtonChecked(hDlg, IDC_PPJAUTO)) {
- GetDlgItemTextW(hDlg, IDC_PPJJVM, szB, SIZ_DESMAX);
- }
- else
- lstrcpyW(szB, L"auto");
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_java, L"Jvm", szB);
- GetDlgItemTextW(hDlg, IDC_PPJCLASSPATH, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_java, L"Classpath", szB);
-
- l = GetWindowTextLength(GetDlgItem(hDlg, IDC_PPJOPTIONS));
- p = apxPoolAlloc(hPool, (l + 2) * sizeof(WCHAR));
- GetDlgItemTextW(hDlg, IDC_PPJOPTIONS, p, l + 1);
- s = apxCRLFToMszW(hPool, p, &l);
- apxFree(p);
- apxRegistrySetMzStrW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_java, L"Options", s, l);
- if (!GetDlgItemTextW(hDlg, IDC_PPJMS, szB, SIZ_DESMAX))
- szB[0] = L'\0';
-
- apxRegistrySetNumW(hRegserv, APXREG_PARAMSOFTWARE, _s_java, L"JvmMs",
- apxAtoulW(szB));
- if (!GetDlgItemTextW(hDlg, IDC_PPJMX, szB, SIZ_DESMAX))
- szB[0] = L'\0';
- apxRegistrySetNumW(hRegserv, APXREG_PARAMSOFTWARE, _s_java, L"JvmMx",
- apxAtoulW(szB));
- if (!GetDlgItemTextW(hDlg, IDC_PPJSS, szB, SIZ_DESMAX))
- szB[0] = L'\0';
- apxRegistrySetNumW(hRegserv, APXREG_PARAMSOFTWARE, _s_java, L"JvmSs",
- apxAtoulW(szB));
- apxFree(s);
- if (!(TST_BIT_FLAG(_propertyChanged, 1)))
- PostMessage(_gui_store->hMainWnd, WM_COMMAND, MAKEWPARAM(IDMS_REFRESH, 0), 0);
- return TRUE;
-}
-
-BOOL __generalStartSave(HWND hDlg)
-{
- WCHAR szB[SIZ_DESLEN];
- LPWSTR p, s;
- DWORD l;
-
- if (!(TST_BIT_FLAG(_propertyChanged, 5)))
- return TRUE;
- CLR_BIT_FLAG(_propertyChanged, 5);
-
- if (IS_INVALID_HANDLE(hService))
- return FALSE;
-
- GetDlgItemTextW(hDlg, IDC_PPRCLASS, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_start, L"Class", szB);
- GetDlgItemTextW(hDlg, IDC_PPRIMAGE, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_start, L"Image", szB);
- GetDlgItemTextW(hDlg, IDC_PPRWPATH, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_start, L"WorkingPath", szB);
- GetDlgItemTextW(hDlg, IDC_PPRMETHOD, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_start, L"Method", szB);
- GetDlgItemTextW(hDlg, IDC_PPRMODE, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_start, L"Mode", szB);
-
- l = GetWindowTextLength(GetDlgItem(hDlg, IDC_PPRARGS));
- p = apxPoolAlloc(hPool, (l + 2) * sizeof(WCHAR));
- GetDlgItemTextW(hDlg, IDC_PPRARGS, p, l + 1);
- s = apxCRLFToMszW(hPool, p, &l);
- apxFree(p);
- apxRegistrySetMzStrW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_start, L"Params", s, l);
- apxFree(s);
-
- if (!(TST_BIT_FLAG(_propertyChanged, 1)))
- PostMessage(_gui_store->hMainWnd, WM_COMMAND, MAKEWPARAM(IDMS_REFRESH, 0), 0);
- return TRUE;
-}
-
-BOOL __generalStopSave(HWND hDlg)
-{
- WCHAR szB[SIZ_DESLEN];
- LPWSTR p, s;
- DWORD l;
-
- if (!(TST_BIT_FLAG(_propertyChanged, 6)))
- return TRUE;
- CLR_BIT_FLAG(_propertyChanged, 6);
-
- if (IS_INVALID_HANDLE(hService))
- return FALSE;
-
- GetDlgItemTextW(hDlg, IDC_PPSCLASS, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_stop, L"Class", szB);
- GetDlgItemTextW(hDlg, IDC_PPSIMAGE, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_stop, L"Image", szB);
- GetDlgItemTextW(hDlg, IDC_PPSWPATH, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_stop, L"WorkingPath", szB);
- GetDlgItemTextW(hDlg, IDC_PPSMETHOD, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_stop, L"Method", szB);
- GetDlgItemTextW(hDlg, IDC_PPSTIMEOUT, szB, SIZ_DESMAX);
- apxRegistrySetNumW(hRegserv, APXREG_PARAMSOFTWARE, _s_stop, L"Timeout", apxAtoulW(szB));
- GetDlgItemTextW(hDlg, IDC_PPSMODE, szB, SIZ_DESMAX);
- apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_stop, L"Mode", szB);
-
- l = GetWindowTextLength(GetDlgItem(hDlg, IDC_PPSARGS));
- p = apxPoolAlloc(hPool, (l + 2) * sizeof(WCHAR));
- GetDlgItemTextW(hDlg, IDC_PPSARGS, p, l + 1);
- s = apxCRLFToMszW(hPool, p, &l);
- apxFree(p);
- apxRegistrySetMzStrW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_stop, L"Params", s, l);
- apxFree(s);
-
- if (!(TST_BIT_FLAG(_propertyChanged, 1)))
- PostMessage(_gui_store->hMainWnd, WM_COMMAND, MAKEWPARAM(IDMS_REFRESH, 0), 0);
- return TRUE;
-}
-
-void __generalPropertyRefresh(HWND hDlg)
-{
- Button_Enable(GetDlgItem(hDlg, IDC_PPSGSTART), FALSE);
- Button_Enable(GetDlgItem(hDlg, IDC_PPSGSTOP), FALSE);
- Button_Enable(GetDlgItem(hDlg, IDC_PPSGPAUSE), FALSE);
- Button_Enable(GetDlgItem(hDlg, IDC_PPSGRESTART), FALSE);
- switch (_currentEntry->stServiceStatus.dwCurrentState) {
- case SERVICE_RUNNING:
- if (_currentEntry->stServiceStatus.dwControlsAccepted & SERVICE_ACCEPT_STOP ||
- _currentEntry->lpConfig->dwStartType != SERVICE_DISABLED) {
- Button_Enable(GetDlgItem(hDlg, IDC_PPSGSTOP), TRUE);
- SetDlgItemText(hDlg, IDC_PPSGSTATUS, STAT_STARTED);
- }
- else
- SetDlgItemText(hDlg, IDC_PPSGSTATUS, STAT_DISABLED);
- if (_currentEntry->stServiceStatus.dwControlsAccepted & SERVICE_ACCEPT_PAUSE_CONTINUE) {
- Button_Enable(GetDlgItem(hDlg, IDC_PPSGPAUSE), TRUE);
- Button_Enable(GetDlgItem(hDlg, IDC_PPSGRESTART), TRUE);
- }
- break;
- case SERVICE_PAUSED:
- Button_Enable(GetDlgItem(hDlg, IDC_PPSGSTART), TRUE);
- Button_Enable(GetDlgItem(hDlg, IDC_PPSGRESTART), TRUE);
- SetDlgItemText(hDlg, IDC_PPSGSTATUS, STAT_PAUSED);
- break;
- case SERVICE_STOPPED:
- if (_currentEntry->lpConfig->dwStartType != SERVICE_DISABLED) {
- Button_Enable(GetDlgItem(hDlg, IDC_PPSGSTART), TRUE);
- SetDlgItemText(hDlg, IDC_PPSGSTATUS, STAT_STOPPED);
- }
- else
- SetDlgItemText(hDlg, IDC_PPSGSTATUS, STAT_DISABLED);
- break;
- default:
- break;
- }
-}
-
-static BOOL bpropCentered = FALSE;
-LRESULT CALLBACK __generalProperty(HWND hDlg,
- UINT uMessage,
- WPARAM wParam,
- LPARAM lParam)
-{
- LPPSHNOTIFY lpShn;
- WCHAR szBuf[1024];
-
- switch (uMessage) {
- case WM_INITDIALOG:
- {
- if (!bEnableTry)
- apxCenterWindow(GetParent(hDlg), NULL);
- else if (!bpropCentered)
- apxCenterWindow(GetParent(hDlg), NULL);
- bpropCentered = TRUE;
- startPage = 0;
- if (!bEnableTry)
- apxCenterWindow(GetParent(hDlg), NULL);
- SendMessage(GetDlgItem(hDlg, IDC_PPSGDISP), EM_LIMITTEXT, 255, 0);
- SendMessage(GetDlgItem(hDlg, IDC_PPSGDESC), EM_LIMITTEXT, 1023, 0);
-
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPSGCMBST), START_AUTO);
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPSGCMBST), START_MANUAL);
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPSGCMBST), START_DISABLED);
- if (_currentEntry->lpConfig->dwStartType == SERVICE_AUTO_START)
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPSGCMBST), 0);
- else if (_currentEntry->lpConfig->dwStartType == SERVICE_DEMAND_START)
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPSGCMBST), 1);
- else if (_currentEntry->lpConfig->dwStartType == SERVICE_DISABLED)
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPSGCMBST), 2);
-
- SetDlgItemTextW(hDlg, IDC_PPSGNAME, _currentEntry->szServiceName);
- SetDlgItemTextW(hDlg, IDC_PPSGDISP, _currentEntry->lpConfig->lpDisplayName);
- SetDlgItemTextW(hDlg, IDC_PPSGDESC, _currentEntry->szServiceDescription);
- SetDlgItemTextW(hDlg, IDC_PPSGDEXE, _currentEntry->lpConfig->lpBinaryPathName);
- __generalPropertyRefresh(hDlg);
- }
- break;
- case WM_COMMAND:
- switch (LOWORD(wParam)) {
- case IDC_PPSGCMBST:
- if (HIWORD(wParam) == CBN_SELCHANGE) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 1);
- }
- break;
- case IDC_PPSGDISP:
- if (HIWORD(wParam) == EN_CHANGE) {
- GetDlgItemTextW(hDlg, IDC_PPSGDISP, szBuf, 255);
- if (!lstrcmpW(szBuf, _currentEntry->lpConfig->lpDisplayName)) {
- PropSheet_UnChanged(GetParent(hDlg), hDlg);
- CLR_BIT_FLAG(_propertyChanged, 1);
- }
- else {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 1);
- }
- }
- break;
- case IDC_PPSGDESC:
- if (HIWORD(wParam) == EN_CHANGE) {
- GetDlgItemTextW(hDlg, IDC_PPSGDESC, szBuf, 1023);
- if (!lstrcmpW(szBuf, _currentEntry->szServiceDescription)) {
- PropSheet_UnChanged(GetParent(hDlg), hDlg);
- CLR_BIT_FLAG(_propertyChanged, 1);
- }
- else {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 1);
- }
- }
- break;
- case IDC_PPSGSTART:
- apxProgressBox(hDlg, apxLoadResource(IDS_HSSTART, 0),
- _currentEntry->lpConfig->lpDisplayName,
- __startServiceCallback, NULL);
- __generalPropertyRefresh(hDlg);
- break;
- case IDC_PPSGSTOP:
- apxProgressBox(hDlg, apxLoadResource(IDS_HSSTOP, 0),
- _currentEntry->lpConfig->lpDisplayName,
- __stopServiceCallback, NULL);
- __generalPropertyRefresh(hDlg);
- break;
- case IDC_PPSGPAUSE:
- apxProgressBox(hDlg, apxLoadResource(IDS_HSPAUSE, 0),
- _currentEntry->lpConfig->lpDisplayName,
- __pauseServiceCallback, NULL);
- __generalPropertyRefresh(hDlg);
- break;
- case IDC_PPSGRESTART:
- apxProgressBox(hDlg, apxLoadResource(IDS_HSRESTART, 0),
- _currentEntry->lpConfig->lpDisplayName,
- __restartServiceCallback, NULL);
- __generalPropertyRefresh(hDlg);
- break;
- }
- break;
- case WM_NOTIFY:
- lpShn = (LPPSHNOTIFY )lParam;
- switch (lpShn->hdr.code) {
- case PSN_APPLY: /* sent when OK or Apply button pressed */
- if (__generalPropertySave(hDlg)) {
- PropSheet_UnChanged(GetParent(hDlg), hDlg);
- }
- else {
- SET_BIT_FLAG(_propertyChanged, 1);
- SetWindowLong(hDlg, DWLP_MSGRESULT,
- PSNRET_INVALID_NOCHANGEPAGE);
- return TRUE;
- }
-
- break;
- default:
- break;
- }
- break;
- default:
- break;
- }
-
- return FALSE;
-}
-
-LRESULT CALLBACK __logonProperty(HWND hDlg,
- UINT uMessage,
- WPARAM wParam,
- LPARAM lParam)
-{
- LPPSHNOTIFY lpShn;
- WCHAR szBuf[1024];
- switch (uMessage) {
- case WM_INITDIALOG:
- {
- BOOL bAccount = FALSE;
- startPage = 1;
- if (!bpropCentered)
- apxCenterWindow(GetParent(hDlg), NULL);
- bpropCentered = TRUE;
-
- SendMessage(GetDlgItem(hDlg, IDC_PPSLUSER), EM_LIMITTEXT, 63, 0);
- SendMessage(GetDlgItem(hDlg, IDC_PPSLPASS), EM_LIMITTEXT, 63, 0);
- SendMessage(GetDlgItem(hDlg, IDC_PPSLCPASS), EM_LIMITTEXT, 63, 0);
- /* Check if we use LocalSystem or user defined account */
- if (lstrcmpiW(_currentEntry->szObjectName, STAT_SYSTEM)) {
- bAccount = TRUE;
- CheckRadioButton(hDlg, IDC_PPSLLS, IDC_PPSLUA, IDC_PPSLUA);
- SetDlgItemTextW(hDlg, IDC_PPSLUSER, _currentEntry->szObjectName);
- SetDlgItemTextW(hDlg, IDC_PPSLPASS, EMPTY_PASSWORD);
- SetDlgItemTextW(hDlg, IDC_PPSLCPASS, EMPTY_PASSWORD);
- }
- else {
- CheckRadioButton(hDlg, IDC_PPSLLS, IDC_PPSLUA, IDC_PPSLLS);
- if (_currentEntry->lpConfig->dwServiceType &
- SERVICE_INTERACTIVE_PROCESS)
- CheckDlgButton(hDlg, IDC_PPSLID, BST_CHECKED);
- }
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLID), !bAccount);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLUSER), bAccount);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLBROWSE), bAccount);
- EnableWindow(GetDlgItem(hDlg, IDL_PPSLPASS), bAccount);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLPASS), bAccount);
- EnableWindow(GetDlgItem(hDlg, IDL_PPSLCPASS), bAccount);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLCPASS), bAccount);
- }
- break;
- case WM_COMMAND:
- switch (LOWORD(wParam)) {
- case IDC_PPSLLS:
- SetDlgItemTextW(hDlg, IDC_PPSLUSER, L"");
- SetDlgItemTextW(hDlg, IDC_PPSLPASS, L"");
- SetDlgItemTextW(hDlg, IDC_PPSLCPASS, L"");
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLID), TRUE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLUSER), FALSE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLBROWSE), FALSE);
- EnableWindow(GetDlgItem(hDlg, IDL_PPSLPASS), FALSE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLPASS), FALSE);
- EnableWindow(GetDlgItem(hDlg, IDL_PPSLCPASS), FALSE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLCPASS), FALSE);
- CheckRadioButton(hDlg, IDC_PPSLLS, IDC_PPSLUA, (INT)wParam);
- if (lstrcmpiW(_currentEntry->szObjectName, STAT_SYSTEM)) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 2);
- }
- else {
- PropSheet_UnChanged(GetParent(hDlg), hDlg);
- CLR_BIT_FLAG(_propertyChanged, 2);
- }
- break;
- case IDC_PPSLUA:
- SetDlgItemTextW(hDlg, IDC_PPSLUSER, _currentEntry->szObjectName);
- SetDlgItemTextW(hDlg, IDC_PPSLPASS, EMPTY_PASSWORD);
- SetDlgItemTextW(hDlg, IDC_PPSLCPASS, EMPTY_PASSWORD);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLID), FALSE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLUSER), TRUE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLBROWSE), TRUE);
- EnableWindow(GetDlgItem(hDlg, IDL_PPSLPASS), TRUE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLPASS), TRUE);
- EnableWindow(GetDlgItem(hDlg, IDL_PPSLCPASS), TRUE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSLCPASS), TRUE);
- CheckRadioButton(hDlg, IDC_PPSLLS, IDC_PPSLUA, (INT)wParam);
- if (lstrcmpW(_currentEntry->szObjectName, STAT_SYSTEM)) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 2);
- }
- else {
- PropSheet_UnChanged(GetParent(hDlg), hDlg);
- CLR_BIT_FLAG(_propertyChanged, 2);
- }
- break;
- case IDC_PPSLID:
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 2);
- break;
- case IDC_PPSLUSER:
- if (HIWORD(wParam) == EN_CHANGE) {
- GetDlgItemTextW(hDlg, IDC_PPSLUSER, szBuf, 63);
- if (!lstrcmpiW(szBuf, _currentEntry->szObjectName)) {
- PropSheet_UnChanged(GetParent(hDlg), hDlg);
- CLR_BIT_FLAG(_propertyChanged, 2);
- }
- else {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 2);
- }
- }
- break;
- case IDC_PPSLPASS:
- case IDC_PPSLCPASS:
- if (HIWORD(wParam) == EN_CHANGE) {
- WCHAR szP[64];
- WCHAR szC[64];
- GetDlgItemTextW(hDlg, IDC_PPSLPASS, szP, 63);
- GetDlgItemTextW(hDlg, IDC_PPSLCPASS, szC, 63);
- /* check for valid password match */
- if (szP[0] == L' ' && szC[0] == L' ') {
- PropSheet_UnChanged(GetParent(hDlg), hDlg);
- CLR_BIT_FLAG(_propertyChanged, 2);
- }
- else if (!lstrcmpW(szP, szC)) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 2);
- }
- }
- break;
- case IDC_PPSLBROWSE:
- {
- WCHAR szUser[SIZ_RESLEN];
- if (apxDlgSelectUser(hDlg, szUser))
- SetDlgItemTextW(hDlg, IDC_PPSLUSER, szUser);
- }
- break;
- }
- break;
- case WM_NOTIFY:
- lpShn = (LPPSHNOTIFY )lParam;
- switch (lpShn->hdr.code) {
- case PSN_APPLY: /* sent when OK or Apply button pressed */
- if (__generalLogonSave(hDlg))
- PropSheet_UnChanged(GetParent(hDlg), hDlg);
- else {
- SET_BIT_FLAG(_propertyChanged, 2);
- SetWindowLong(hDlg, DWLP_MSGRESULT,
- PSNRET_INVALID_NOCHANGEPAGE);
- return TRUE;
- }
-
- break;
- }
- break;
-
- default:
- break;
- }
- return FALSE;
-}
-
-LRESULT CALLBACK __loggingProperty(HWND hDlg,
- UINT uMessage,
- WPARAM wParam,
- LPARAM lParam)
-{
- LPPSHNOTIFY lpShn;
- LPWSTR lpBuf;
-
- switch (uMessage) {
- case WM_INITDIALOG:
- {
- LPWSTR b;
- startPage = 2;
- if (!bpropCentered)
- apxCenterWindow(GetParent(hDlg), NULL);
- bpropCentered = TRUE;
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPLGLEVEL), LOGL_ERROR);
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPLGLEVEL), LOGL_INFO);
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPLGLEVEL), LOGL_WARN);
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPLGLEVEL), LOGL_DEBUG);
- if ((b = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_log, L"Level")) != NULL) {
- if (!lstrcmpiW(b, LOGL_ERROR))
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPLGLEVEL), 0);
- else if (!lstrcmpiW(b, LOGL_INFO))
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPLGLEVEL), 1);
- else if (!lstrcmpiW(b, LOGL_WARN))
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPLGLEVEL), 2);
- else
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPLGLEVEL), 3);
- apxFree(b);
- }
- else
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPLGLEVEL), 0);
- if ((b = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_log, L"Path")) != NULL) {
- SetDlgItemTextW(hDlg, IDC_PPLGPATH, b);
- apxFree(b);
- }
- if ((b = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_log, L"StdOutput")) != NULL) {
- SetDlgItemTextW(hDlg, IDC_PPLGSTDOUT, b);
- apxFree(b);
- }
- if ((b = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_log, L"StdError")) != NULL) {
- SetDlgItemTextW(hDlg, IDC_PPLGSTDERR, b);
- apxFree(b);
- }
- }
- break;
- case WM_COMMAND:
- switch (LOWORD(wParam)) {
- case IDC_PPLGLEVEL:
- if (HIWORD(wParam) == CBN_SELCHANGE) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 3);
- }
- break;
- case IDC_PPLGPATH:
- if (HIWORD(wParam) == EN_CHANGE) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 3);
- }
- break;
- case IDC_PPLGPREFIX:
- if (HIWORD(wParam) == EN_CHANGE) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 3);
- }
- break;
- case IDC_PPLGSTDERR:
- if (HIWORD(wParam) == EN_CHANGE) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 3);
- }
- break;
- case IDC_PPLGSTDOUT:
- if (HIWORD(wParam) == EN_CHANGE) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 3);
- }
- break;
- case IDC_PPLGBPATH:
- lpBuf = apxBrowseForFolderW(hDlg, apxLoadResourceW(IDS_LGPATHTITLE, 0),
- NULL);
- if (lpBuf) {
- SetDlgItemTextW(hDlg, IDC_PPLGPATH, lpBuf);
- apxFree(lpBuf);
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 3);
- }
- break;
- case IDC_PPLGBSTDOUT:
- lpBuf = apxGetFileNameW(hDlg, apxLoadResourceW(IDS_LGSTDOUT, 0),
- apxLoadResourceW(IDS_ALLFILES, 1), NULL,
- NULL, FALSE, NULL);
- if (lpBuf) {
- SetDlgItemTextW(hDlg, IDC_PPLGSTDOUT, lpBuf);
- apxFree(lpBuf);
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 3);
- }
- break;
- case IDC_PPLGBSTDERR:
- lpBuf = apxGetFileNameW(hDlg, apxLoadResourceW(IDS_LGSTDERR, 0),
- apxLoadResourceW(IDS_ALLFILES, 1), NULL,
- NULL, FALSE, NULL);
- if (lpBuf) {
- SetDlgItemTextW(hDlg, IDC_PPLGSTDERR, lpBuf);
- apxFree(lpBuf);
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 3);
- }
- break;
- }
- break;
- case WM_NOTIFY:
- lpShn = (LPPSHNOTIFY )lParam;
- switch (lpShn->hdr.code) {
- case PSN_APPLY: /* sent when OK or Apply button pressed */
- if (__generalLoggingSave(hDlg))
- PropSheet_UnChanged(GetParent(hDlg), hDlg);
- else {
- SET_BIT_FLAG(_propertyChanged, 3);
- SetWindowLong(hDlg, DWLP_MSGRESULT,
- PSNRET_INVALID_NOCHANGEPAGE);
- return TRUE;
- }
-
- break;
- }
- break;
-
- default:
- break;
- }
- return FALSE;
-}
-
-LRESULT CALLBACK __jvmProperty(HWND hDlg,
- UINT uMessage,
- WPARAM wParam,
- LPARAM lParam)
-{
- LPPSHNOTIFY lpShn;
- LPWSTR lpBuf, b;
- DWORD v;
- CHAR bn[32];
-
- switch (uMessage) {
- case WM_INITDIALOG:
- {
- startPage = 3;
- if (!bpropCentered)
- apxCenterWindow(GetParent(hDlg), NULL);
- bpropCentered = TRUE;
- if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_java, L"Jvm")) != NULL) {
- if (!lstrcmpiW(lpBuf, L"auto")) {
- CheckDlgButton(hDlg, IDC_PPJAUTO, BST_CHECKED);
- apxFree(lpBuf);
- lpBuf = apxGetJavaSoftRuntimeLib(hPool);
- EnableWindow(GetDlgItem(hDlg, IDC_PPJJVM), FALSE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPJBJVM), FALSE);
- }
- if (lpBuf) {
- SetDlgItemTextW(hDlg, IDC_PPJJVM, lpBuf);
- apxFree(lpBuf);
- }
- }
- if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_java, L"Classpath")) != NULL) {
- SetDlgItemTextW(hDlg, IDC_PPJCLASSPATH, lpBuf);
- apxFree(lpBuf);
- }
- if ((lpBuf = apxRegistryGetMzStrW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_java, L"Options", NULL, NULL)) != NULL) {
- LPWSTR p = apxMszToCRLFW(hPool, lpBuf);
- SetDlgItemTextW(hDlg, IDC_PPJOPTIONS, p);
- apxFree(lpBuf);
- apxFree(p);
- }
- v = apxRegistryGetNumberW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_java, L"JvmMs");
- if (v && v != 0xFFFFFFFF) {
- wsprintfA(bn, "%d", v);
- SetDlgItemTextA(hDlg, IDC_PPJMS, bn);
- }
- v = apxRegistryGetNumberW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_java, L"JvmMx");
- if (v && v != 0xFFFFFFFF) {
- wsprintfA(bn, "%d", v);
- SetDlgItemTextA(hDlg, IDC_PPJMX, bn);
- }
- v = apxRegistryGetNumberW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_java, L"JvmSs");
- if (v && v != 0xFFFFFFFF) {
- wsprintfA(bn, "%d", v);
- SetDlgItemTextA(hDlg, IDC_PPJSS, bn);
- }
-
- }
- break;
- case WM_COMMAND:
- switch (LOWORD(wParam)) {
- case IDC_PPJBJVM:
- b = apxGetJavaSoftHome(hPool, TRUE);
- lpBuf = apxGetFileNameW(hDlg, apxLoadResourceW(IDS_PPJBJVM, 0),
- apxLoadResourceW(IDS_DLLFILES, 1), NULL,
- b,
- TRUE, NULL);
- apxFree(b);
- if (lpBuf) {
- SetDlgItemTextW(hDlg, IDC_PPJJVM, lpBuf);
- apxFree(lpBuf);
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 4);
- }
- break;
- case IDC_PPJAUTO:
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 4);
- if (IsDlgButtonChecked(hDlg, IDC_PPJAUTO)) {
- EnableWindow(GetDlgItem(hDlg, IDC_PPJJVM), FALSE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPJBJVM), FALSE);
- lpBuf = apxGetJavaSoftRuntimeLib(hPool);
- if (lpBuf) {
- SetDlgItemTextW(hDlg, IDC_PPJJVM, lpBuf);
- apxFree(lpBuf);
- }
- }
- else {
- EnableWindow(GetDlgItem(hDlg, IDC_PPJJVM), TRUE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPJBJVM), TRUE);
- }
- break;
- case IDC_PPJJVM:
- case IDC_PPJCLASSPATH:
- case IDC_PPJOPTIONS:
- case IDC_PPJMX:
- case IDC_PPJMS:
- case IDC_PPJSS:
- if (HIWORD(wParam) == EN_CHANGE) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 4);
- }
- break;
- }
- break;
- case WM_NOTIFY:
- lpShn = (LPPSHNOTIFY )lParam;
- switch (lpShn->hdr.code) {
- case PSN_APPLY: /* sent when OK or Apply button pressed */
- if (__generalJvmSave(hDlg))
- PropSheet_UnChanged(GetParent(hDlg), hDlg);
- else {
- SET_BIT_FLAG(_propertyChanged, 4);
- SetWindowLong(hDlg, DWLP_MSGRESULT,
- PSNRET_INVALID_NOCHANGEPAGE);
- return TRUE;
- }
-
- break;
- }
- break;
-
- default:
- break;
- }
- return FALSE;
-}
-
-LRESULT CALLBACK __startProperty(HWND hDlg,
- UINT uMessage,
- WPARAM wParam,
- LPARAM lParam)
-{
- LPPSHNOTIFY lpShn;
- LPWSTR lpBuf, b;
-
- switch (uMessage) {
- case WM_INITDIALOG:
- {
- startPage = 4;
- if (!bpropCentered)
- apxCenterWindow(GetParent(hDlg), NULL);
- bpropCentered = TRUE;
-
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPRMODE), L"");
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPRMODE), L"jvm");
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPRMODE), _s_java);
-
- if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_start, L"Class")) != NULL) {
- SetDlgItemTextW(hDlg, IDC_PPRCLASS, lpBuf);
- apxFree(lpBuf);
- }
- if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_start, L"Image")) != NULL) {
- SetDlgItemTextW(hDlg, IDC_PPRIMAGE, lpBuf);
- apxFree(lpBuf);
- }
- if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_start, L"WorkingPath")) != NULL) {
- SetDlgItemTextW(hDlg, IDC_PPRWPATH, lpBuf);
- apxFree(lpBuf);
- }
- if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_start, L"Method")) != NULL) {
- SetDlgItemTextW(hDlg, IDC_PPRMETHOD, lpBuf);
- apxFree(lpBuf);
- }
- if ((lpBuf = apxRegistryGetMzStrW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_start, L"Params", NULL, NULL)) != NULL) {
- b = apxMszToCRLFW(hPool, lpBuf);
- SetDlgItemTextW(hDlg, IDC_PPRARGS, b);
- apxFree(lpBuf);
- apxFree(b);
- }
- if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_start, L"Mode")) != NULL) {
- if (!lstrcmpiW(lpBuf, L"jvm")) {
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPRMODE), 1);
-
- }
- else if (!lstrcmpiW(lpBuf, _s_java)) {
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPRMODE), 2);
- }
- else {
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPRMODE), 0);
- EnableWindow(GetDlgItem(hDlg, IDC_PPRIMAGE), TRUE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPRBIMAGE), TRUE);
- }
- apxFree(lpBuf);
- }
- }
- break;
- case WM_COMMAND:
- switch (LOWORD(wParam)) {
- case IDC_PPRBWPATH:
- lpBuf = apxBrowseForFolderW(hDlg, apxLoadResourceW(IDS_PPWPATH, 0),
- NULL);
- if (lpBuf) {
- SetDlgItemTextW(hDlg, IDC_PPRWPATH, lpBuf);
- apxFree(lpBuf);
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 5);
- }
- break;
- case IDC_PPRBIMAGE:
- lpBuf = apxGetFileNameW(hDlg, apxLoadResourceW(IDS_PPIMAGE, 0),
- apxLoadResourceW(IDS_EXEFILES, 1), NULL,
- NULL, TRUE, NULL);
- if (lpBuf) {
- SetDlgItemTextW(hDlg, IDC_PPRIMAGE, lpBuf);
- apxFree(lpBuf);
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 5);
- }
- break;
- case IDC_PPRCLASS:
- case IDC_PPRMETHOD:
- case IDC_PPRARGS:
- case IDC_PPRIMAGE:
- case IDC_PPRWPATH:
- if (HIWORD(wParam) == EN_CHANGE) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 5);
- }
- break;
- case IDC_PPRMODE:
- if (HIWORD(wParam) == CBN_SELCHANGE) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 5);
- if (ComboBox_GetCurSel(GetDlgItem(hDlg, IDC_PPRMODE))) {
- EnableWindow(GetDlgItem(hDlg, IDC_PPRIMAGE), FALSE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPRBIMAGE), FALSE);
- }
- else {
- EnableWindow(GetDlgItem(hDlg, IDC_PPRIMAGE), TRUE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPRBIMAGE), TRUE);
- }
- }
- break;
-
- }
- break;
- case WM_NOTIFY:
- lpShn = (LPPSHNOTIFY )lParam;
- switch (lpShn->hdr.code) {
- case PSN_APPLY: /* sent when OK or Apply button pressed */
- if (__generalStartSave(hDlg))
- PropSheet_UnChanged(GetParent(hDlg), hDlg);
- else {
- SET_BIT_FLAG(_propertyChanged, 5);
- SetWindowLong(hDlg, DWLP_MSGRESULT,
- PSNRET_INVALID_NOCHANGEPAGE);
- return TRUE;
- }
-
- break;
- }
- break;
-
- default:
- break;
- }
- return FALSE;
-}
-
-LRESULT CALLBACK __stopProperty(HWND hDlg,
- UINT uMessage,
- WPARAM wParam,
- LPARAM lParam)
-{
- LPPSHNOTIFY lpShn;
- LPWSTR lpBuf, b;
- DWORD v;
-
- switch (uMessage) {
- case WM_INITDIALOG:
- {
- startPage = 5;
- if (!bpropCentered)
- apxCenterWindow(GetParent(hDlg), NULL);
- bpropCentered = TRUE;
-
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPSMODE), L"");
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPSMODE), L"jvm");
- ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPSMODE), _s_java);
-
- if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_stop, L"Class")) != NULL) {
- SetDlgItemTextW(hDlg, IDC_PPSCLASS, lpBuf);
- apxFree(lpBuf);
- }
- if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_stop, L"Image")) != NULL) {
- SetDlgItemTextW(hDlg, IDC_PPSIMAGE, lpBuf);
- apxFree(lpBuf);
- }
- if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_stop, L"WorkingPath")) != NULL) {
- SetDlgItemTextW(hDlg, IDC_PPSWPATH, lpBuf);
- apxFree(lpBuf);
- }
- if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_stop, L"Method")) != NULL) {
- SetDlgItemTextW(hDlg, IDC_PPSMETHOD, lpBuf);
- apxFree(lpBuf);
- }
- if ((lpBuf = apxRegistryGetMzStrW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_stop, L"Params", NULL, NULL)) != NULL) {
- b = apxMszToCRLFW(hPool, lpBuf);
- SetDlgItemTextW(hDlg, IDC_PPSARGS, b);
- apxFree(lpBuf);
- apxFree(b);
- }
- v = apxRegistryGetNumberW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_stop, L"Timeout");
- {
- CHAR bn[32];
- wsprintfA(bn, "%d", v);
- SetDlgItemTextA(hDlg, IDC_PPSTIMEOUT, bn);
- }
- if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
- _s_stop, L"Mode")) != NULL) {
- if (!lstrcmpiW(lpBuf, L"jvm")) {
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPSMODE), 1);
-
- }
- else if (!lstrcmpiW(lpBuf, _s_java)) {
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPSMODE), 2);
- }
- else {
- ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPSMODE), 0);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSIMAGE), TRUE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSBIMAGE), TRUE);
- }
- apxFree(lpBuf);
- }
- }
- break;
- case WM_COMMAND:
- switch (LOWORD(wParam)) {
- case IDC_PPSBWPATH:
- lpBuf = apxBrowseForFolderW(hDlg, apxLoadResourceW(IDS_PPWPATH, 0),
- NULL);
- if (lpBuf) {
- SetDlgItemTextW(hDlg, IDC_PPSWPATH, lpBuf);
- apxFree(lpBuf);
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 6);
- }
- break;
- case IDC_PPSBIMAGE:
- lpBuf = apxGetFileNameW(hDlg, apxLoadResourceW(IDS_PPIMAGE, 0),
- apxLoadResourceW(IDS_EXEFILES, 1), NULL,
- NULL, TRUE, NULL);
- if (lpBuf) {
- SetDlgItemTextW(hDlg, IDC_PPSIMAGE, lpBuf);
- apxFree(lpBuf);
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 6);
- }
- break;
- case IDC_PPSCLASS:
- case IDC_PPSMETHOD:
- case IDC_PPSTIMEOUT:
- case IDC_PPSARGS:
- case IDC_PPSIMAGE:
- case IDC_PPSWPATH:
- if (HIWORD(wParam) == EN_CHANGE) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 6);
- }
- break;
- case IDC_PPSMODE:
- if (HIWORD(wParam) == CBN_SELCHANGE) {
- PropSheet_Changed(GetParent(hDlg), hDlg);
- SET_BIT_FLAG(_propertyChanged, 6);
- if (ComboBox_GetCurSel(GetDlgItem(hDlg, IDC_PPSMODE))) {
- EnableWindow(GetDlgItem(hDlg, IDC_PPSIMAGE), FALSE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSBIMAGE), FALSE);
- }
- else {
- EnableWindow(GetDlgItem(hDlg, IDC_PPSIMAGE), TRUE);
- EnableWindow(GetDlgItem(hDlg, IDC_PPSBIMAGE), TRUE);
- }
- }
- break;
-
- }
- break;
- case WM_NOTIFY:
- lpShn = (LPPSHNOTIFY )lParam;
- switch (lpShn->hdr.code) {
- case PSN_APPLY: /* sent when OK or Apply button pressed */
- if (__generalStopSave(hDlg))
- PropSheet_UnChanged(GetParent(hDlg), hDlg);
- else {
- SET_BIT_FLAG(_propertyChanged, 6);
- SetWindowLong(hDlg, DWLP_MSGRESULT,
- PSNRET_INVALID_NOCHANGEPAGE);
- return TRUE;
- }
-
- break;
- }
- break;
-
- default:
- break;
- }
- return FALSE;
-}
-
-void __initPpage(PROPSHEETPAGEW *lpPage, INT iDlg, INT iTitle, DLGPROC pfnDlgProc)
-{
- lpPage->dwSize = sizeof(PROPSHEETPAGE);
- lpPage->dwFlags = PSP_USETITLE;
- lpPage->hInstance = _gui_store->hInstance;
- lpPage->pszTemplate = MAKEINTRESOURCEW(iDlg);
- lpPage->pszIcon = NULL;
- lpPage->pfnDlgProc = pfnDlgProc;
- lpPage->pszTitle = MAKEINTRESOURCEW(iTitle);
- lpPage->lParam = 0;
-}
-
-void ShowServiceProperties(HWND hWnd)
-{
- PROPSHEETPAGEW psP[6];
- PROPSHEETHEADERW psH;
- WCHAR szT[1024] = {0};
-
- if (_propertyOpened) {
- SetForegroundWindow(_gui_store->hMainWnd);
- return;
- }
- __initPpage(&psP[0], IDD_PROPPAGE_SGENERAL, IDS_PPGENERAL,
- __generalProperty);
- __initPpage(&psP[1], IDD_PROPPAGE_LOGON, IDS_PPLOGON,
- __logonProperty);
- __initPpage(&psP[2], IDD_PROPPAGE_LOGGING, IDS_PPLOGGING,
- __loggingProperty);
- __initPpage(&psP[3], IDD_PROPPAGE_JVM, IDS_PPJAVAVM,
- __jvmProperty);
- __initPpage(&psP[4], IDD_PROPPAGE_START, IDS_PPSTART,
- __startProperty);
- __initPpage(&psP[5], IDD_PROPPAGE_STOP, IDS_PPSTOP,
- __stopProperty);
-
- if (_currentEntry && _currentEntry->lpConfig)
- lstrcpyW(szT, _currentEntry->lpConfig->lpDisplayName);
- else
- return;
- lstrcatW(szT, L" Properties");
-
- psH.dwSize = sizeof(PROPSHEETHEADER);
- psH.dwFlags = PSH_PROPSHEETPAGE | PSH_USEICONID | PSH_USECALLBACK | PSH_NOCONTEXTHELP;
- psH.hwndParent = bEnableTry ? hWnd : NULL;
- psH.hInstance = _gui_store->hInstance;
- psH.pszIcon = MAKEINTRESOURCEW(IDI_MAINICON);
- psH.pszCaption = szT;
- psH.nPages = 6;
- psH.ppsp = (LPCPROPSHEETPAGEW) &psP;
- psH.pfnCallback = (PFNPROPSHEETCALLBACK)__propertyCallback;
- psH.nStartPage = startPage;
-
- PropertySheetW(&psH);
- _propertyOpened = FALSE;
- if (!bEnableTry)
- PostQuitMessage(0);
- bpropCentered = FALSE;
-
-}
-
-static void signalService(LPCWSTR szServiceName)
-{
- HANDLE event;
- WCHAR en[SIZ_DESLEN];
- int i;
-
- lstrcpyW(en, L"Global\\");
- lstrcatW(en, szServiceName);
- lstrcatW(en, L"SIGNAL");
- for (i = 7; i < lstrlenW(en); i++) {
- if (en[i] >= L'a' && en[i] <= L'z')
- en[i] = en[i] - 32;
- }
-
-
- event = OpenEventW(EVENT_MODIFY_STATE, FALSE, en);
- if (event) {
- SetEvent(event);
- CloseHandle(event);
- }
- else
- apxDisplayError(TRUE, NULL, 0, "Unable to open the Event Mutex");
-
-}
-
-LRESULT CALLBACK MainWndProc(HWND hWnd, UINT uMsg,
- WPARAM wParam, LPARAM lParam)
-{
- switch (uMsg) {
- case WM_CREATE:
- if (bEnableTry) {
- if (_currentEntry && _currentEntry->lpConfig) {
- BOOL isRunning = _currentEntry->stServiceStatus.dwCurrentState == SERVICE_RUNNING;
- apxManageTryIconW(hWnd, NIM_ADD, NULL,
- _currentEntry->lpConfig->lpDisplayName,
- isRunning ? hIcoRun : hIcoStop);
- }
- else {
- apxManageTryIconA(hWnd, NIM_ADD, NULL,
- apxLoadResourceA(IDS_APPLICATION, 0),
- NULL);
- }
- }
- else
- ShowServiceProperties(hWnd);
-
- break;
- case WM_COMMAND:
- switch(LOWORD(wParam)) {
- case IDM_TM_CONFIG:
- ShowServiceProperties(hWnd);
- break;
- case IDM_TM_ABOUT:
- apxAboutBox(hWnd);
- break;
- case IDM_TM_EXIT:
- SendMessage(hWnd, WM_CLOSE, 0, 0);
- break;
- case IDM_TM_START:
- if (!_propertyOpened)
- apxProgressBox(hWnd, apxLoadResource(IDS_HSSTART, 0),
- _currentEntry->lpConfig->lpDisplayName,
- __startServiceCallback, NULL);
- break;
- case IDM_TM_STOP:
- if (!_propertyOpened)
- apxProgressBox(hWnd, apxLoadResource(IDS_HSSTOP, 0),
- _currentEntry->lpConfig->lpDisplayName,
- __stopServiceCallback, NULL);
- break;
- case IDM_TM_PAUSE:
- if (!_propertyOpened)
- apxProgressBox(hWnd, apxLoadResource(IDS_HSPAUSE, 0),
- _currentEntry->lpConfig->lpDisplayName,
- __pauseServiceCallback, NULL);
- break;
- case IDM_TM_RESTART:
- if (!_propertyOpened)
- apxProgressBox(hWnd, apxLoadResource(IDS_HSRESTART, 0),
- _currentEntry->lpConfig->lpDisplayName,
- __restartServiceCallback, NULL);
- break;
- case IDM_TM_DUMP:
- signalService(_currentEntry->szServiceName);
- break;
- case IDMS_REFRESH:
- if (bEnableTry &&
- (_currentEntry = apxServiceEntry(hService, TRUE)) != NULL) {
- BOOL isRunning = _currentEntry->stServiceStatus.dwCurrentState == SERVICE_RUNNING;
- apxManageTryIconW(hWnd, NIM_MODIFY, NULL,
- _currentEntry->lpConfig->lpDisplayName,
- isRunning ? hIcoRun : hIcoStop);
- }
- break;
-
- }
- break;
- case WM_TRAYMESSAGE:
- switch(lParam) {
- case WM_LBUTTONDBLCLK:
- ShowServiceProperties(hWnd);
- break;
- case WM_RBUTTONUP:
- _currentEntry = apxServiceEntry(hService, TRUE);
- createRbuttonTryMenu(hWnd);
- break;
- }
- break;
- case WM_QUIT:
- return DefWindowProc(hWnd, uMsg, wParam, lParam);
- break;
- case WM_DESTROY:
- if (bEnableTry)
- apxManageTryIconA(hWnd, NIM_DELETE, NULL, NULL, NULL);
- PostQuitMessage(0);
- break;
- default:
- return DefWindowProc(hWnd, uMsg, wParam, lParam);
- break;
- }
-
- return FALSE;
-}
-
-static BOOL loadConfiguration()
-{
- return TRUE;
-}
-
-static BOOL saveConfiguration()
-{
- return TRUE;
-}
-
-/* Main program entry
- * Since we are inependant from CRT
- * the arguments are not used
- */
-#ifdef _NO_CRTLIBRARY
-int xMain(void)
-#else
-int WINAPI WinMain(HINSTANCE hInstance,
- HINSTANCE hPrevInstance,
- LPSTR lpCmdLine,
- int nCmdShow)
-#endif
-{
- MSG msg;
- LPAPXCMDLINE lpCmdline;
- HANDLE mutex = NULL;
- BOOL quiet = FALSE;
-
- apxHandleManagerInitialize();
- hPool = apxPoolCreate(NULL, 0);
-
- /* Parse the command line */
- if ((lpCmdline = apxCmdlineParse(hPool, _options, _commands)) == NULL) {
- /* TODO: dispalay error message */
- apxDisplayError(TRUE, NULL, 0, "Error parsing command line");
- goto cleanup;
- }
-
- if (!lpCmdline->dwCmdIndex) {
- /* Skip sytem error message */
- SetLastError(ERROR_SUCCESS);
- apxDisplayError(TRUE, NULL, 0,
- apxLoadResourceA(IDS_ERRORCMD, 0),
- lpCmdLine);
- goto cleanup;
- }
- else if (lpCmdline->dwCmdIndex == 4)
- quiet = TRUE;
- else if (lpCmdline->dwCmdIndex >= 2)
- bEnableTry = TRUE;
- hService = apxCreateService(hPool, SC_MANAGER_CONNECT, FALSE);
- if (IS_INVALID_HANDLE(hService)) {
- if (!quiet)
- apxDisplayError(TRUE, NULL, 0, "Unable to open the Service Manager");
- goto cleanup;
- }
- /* Open the main service handle */
- if (!apxServiceOpen(hService, lpCmdline->szApplication,
- SERVICE_ALL_ACCESS)) {
- LPWSTR w = lpCmdline->szApplication + lstrlenW(lpCmdline->szApplication) - 1;
- if (*w == L'w')
- *w = L'\0';
- if (!apxServiceOpen(hService, lpCmdline->szApplication,
- SERVICE_ALL_ACCESS)) {
- if (!apxServiceOpen(hService, lpCmdline->szApplication,
- GENERIC_READ | GENERIC_EXECUTE)) {
-
- if (!quiet)
- apxDisplayError(TRUE, NULL, 0, "Unable to open the service '%S'",
- lpCmdline->szApplication);
- goto cleanup;
- }
- }
- }
- /* Obtain service parameters and status */
- if (!(_currentEntry = apxServiceEntry(hService, TRUE))) {
- if (!quiet)
- apxDisplayError(TRUE, NULL, 0, "Unable to query the service '%S' status",
- lpCmdline->szApplication);
- goto cleanup;
- }
-#ifdef _UNICODE
- _gui_store = apxGuiInitialize(MainWndProc, lpCmdline->szApplication);
-#else
- {
- CHAR szApp[MAX_PATH];
- _gui_store = apxGuiInitialize(MainWndProc,
- WideToAscii(lpCmdline->szApplication, szApp));
- }
-#endif
- if (!_gui_store) {
- if (!quiet)
- apxDisplayError(TRUE, NULL, 0, "Unable to initialize GUI manager");
- goto cleanup;
- }
- hIcoRun = LoadImage(_gui_store->hInstance, MAKEINTRESOURCE(IDI_ICONRUN),
- IMAGE_ICON, 16, 16, LR_DEFAULTCOLOR);
- hIcoStop = LoadImage(_gui_store->hInstance, MAKEINTRESOURCE(IDI_ICONSTOP),
- IMAGE_ICON, 16, 16, LR_DEFAULTCOLOR);
-
- /* Handle //MQ// option */
- if (lpCmdline->dwCmdIndex == 4) {
- HANDLE hOther = FindWindow(_gui_store->szWndClass, NULL);
- if (hOther)
- SendMessage(hOther, WM_CLOSE, 0, 0);
- goto cleanup;
- }
-
- if (!_options[0].dwValue) {
- mutex = CreateMutex(NULL, FALSE, _gui_store->szWndMutex);
- if ((mutex == NULL) || (GetLastError() == ERROR_ALREADY_EXISTS)) {
- /* Skip sytem error message */
- SetLastError(ERROR_SUCCESS);
- if (!quiet)
- apxDisplayError(TRUE, NULL, 0, apxLoadResourceA(IDS_ALREAY_RUNING, 0),
- lpCmdline->szApplication);
- goto cleanup;
- }
- }
- hRegistry = apxCreateRegistry(hPool, KEY_ALL_ACCESS, NULL,
- apxLoadResource(IDS_APPLICATION, 0),
- APXREG_USER);
- loadConfiguration();
- hRegserv = apxCreateRegistryW(hPool, KEY_READ | KEY_WRITE | KREG_WOW6432,
- PRG_REGROOT,
- lpCmdline->szApplication,
- APXREG_SOFTWARE | APXREG_SERVICE);
-
- if (IS_INVALID_HANDLE(hRegserv)) {
- if (!quiet)
- apxDisplayError(TRUE, NULL, 0, apxLoadResourceA(IDS_ERRSREG, 0));
- return FALSE;
- }
- /* Create main invisible window */
- _gui_store->hMainWnd = CreateWindow(_gui_store->szWndClass,
- apxLoadResource(IDS_APPLICATION, 0),
- 0, 0, 0, 0, 0,
- NULL, NULL,
- _gui_store->hInstance,
- NULL);
-
- if (!_gui_store->hMainWnd) {
- goto cleanup;
- }
- if (lpCmdline->dwCmdIndex == 3)
- PostMessage(_gui_store->hMainWnd, WM_COMMAND, IDM_TM_START, 0);
-
- while (GetMessage(&msg, NULL, 0, 0)) {
- if(!TranslateAccelerator(_gui_store->hMainWnd,
- _gui_store->hAccel, &msg)) {
- TranslateMessage(&msg);
- DispatchMessage(&msg);
- }
- }
- saveConfiguration();
-
-cleanup:
- if (hIcoStop)
- DestroyIcon(hIcoStop);
- if (hIcoRun)
- DestroyIcon(hIcoRun);
- if (mutex)
- CloseHandle(mutex);
- if (lpCmdline)
- apxCmdlineFree(lpCmdline);
- apxCloseHandle(hService);
- apxHandleManagerDestroy();
- ExitProcess(0);
- return 0;
-}
-
Copied: tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.c (from rev 2689, trunk/utils/windows/native/service/procrun/apps/prunmgr/prunmgr.c)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.c (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,1763 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/* ====================================================================
+ * prunmgr -- Service Manager Application.
+ * Contributed by Mladen Turk <mturk at apache.org>
+ * 05 Aug 2003
+ * ====================================================================
+ */
+
+/* Force the JNI vprintf functions */
+#define _DEBUG_JNI 1
+#include "apxwin.h"
+#include "prunmgr.h"
+
+LPAPXGUISTORE _gui_store = NULL;
+#define PRUNMGR_CLASS TEXT("PRUNMGR")
+#define TMNU_CONF TEXT("Configure...")
+#define TMNU_START TEXT("Start service")
+#define TMNU_STOP TEXT("Stop service")
+#define TMNU_EXIT TEXT("Exit")
+#define TMNU_ABOUT TEXT("About")
+#define TMNU_DUMP TEXT("Thread Dump")
+
+/* Display only Started/Paused status */
+#define STAT_STARTED TEXT("Started")
+#define STAT_PAUSED TEXT("Paused")
+#define STAT_STOPPED TEXT("Stopped")
+#define STAT_DISABLED TEXT("Disabled")
+#define STAT_NONE TEXT("")
+#define STAT_SYSTEM L"LocalSystem"
+
+#define LOGL_ERROR L"Error"
+#define LOGL_DEBUG L"Debug"
+#define LOGL_INFO L"Info"
+#define LOGL_WARN L"Warning"
+
+
+#define START_AUTO L"Automatic"
+#define START_MANUAL L"Manual"
+#define START_DISABLED L"Disabled"
+#define START_BOOT L"Boot"
+#define START_SYSTEM L"SystemInit"
+#define EMPTY_PASSWORD L" "
+
+#ifdef WIN64
+#define KREG_WOW6432 KEY_WOW64_32KEY
+#else
+#define KREG_WOW6432 0
+#endif
+
+/* Main application pool */
+APXHANDLE hPool = NULL;
+APXHANDLE hService = NULL;
+APXHANDLE hRegistry = NULL;
+APXHANDLE hRegserv = NULL;
+HICON hIcoRun = NULL;
+HICON hIcoStop = NULL;
+
+LPAPXSERVENTRY _currentEntry = NULL;
+
+BOOL bEnableTry = FALSE;
+DWORD startPage = 0;
+
+static LPCWSTR _s_log = L"Log";
+static LPCWSTR _s_java = L"Java";
+static LPCWSTR _s_start = L"Start";
+static LPCWSTR _s_stop = L"Stop";
+
+/* Allowed prunmgr commands */
+static LPCWSTR _commands[] = {
+ L"ES", /* 1 Manage Service (default)*/
+ L"MS", /* 2 Monitor Service */
+ L"MR", /* 3 Monitor Service and start if not runing */
+ L"MQ", /* 4 Quit all running Monitor applications */
+ NULL
+};
+
+/* Allowed procrun parameters */
+static APXCMDLINEOPT _options[] = {
+/* 0 */ { L"AllowMultiInstances", NULL, NULL, APXCMDOPT_INT, NULL, 0},
+ /* NULL terminate the array */
+ { NULL }
+};
+
+/* Create RBUTTON try menu
+ * Configure... (default, or lbutton dblclick)
+ * Start <service name>
+ * Stop <service name>
+ * Exit
+ * Logo
+ */
+static void createRbuttonTryMenu(HWND hWnd)
+{
+ HMENU hMnu;
+ POINT pt;
+ BOOL canStop = FALSE;
+ BOOL canStart = FALSE;
+ hMnu = CreatePopupMenu();
+
+ if (_currentEntry) {
+ if (_currentEntry->stServiceStatus.dwCurrentState == SERVICE_RUNNING) {
+ if (_currentEntry->stServiceStatus.dwControlsAccepted & SERVICE_ACCEPT_STOP)
+ canStop = TRUE;
+ }
+ else if (_currentEntry->stServiceStatus.dwCurrentState == SERVICE_STOPPED) {
+ if (_currentEntry->lpConfig->dwStartType != SERVICE_DISABLED)
+ canStart = TRUE;
+ }
+ }
+ apxAppendMenuItem(hMnu, IDM_TM_CONFIG, TMNU_CONF, TRUE, TRUE);
+ apxAppendMenuItem(hMnu, IDM_TM_START, TMNU_START, FALSE, canStart);
+ apxAppendMenuItem(hMnu, IDM_TM_STOP, TMNU_STOP, FALSE, canStop);
+ apxAppendMenuItem(hMnu, IDM_TM_DUMP, TMNU_DUMP, FALSE, canStop);
+ apxAppendMenuItem(hMnu, IDM_TM_EXIT, TMNU_EXIT, FALSE, TRUE);
+ apxAppendMenuItem(hMnu, -1, NULL, FALSE, FALSE);
+ apxAppendMenuItem(hMnu, IDM_TM_ABOUT, TMNU_ABOUT, FALSE, TRUE);
+
+ /* Ensure we have a focus */
+ if (!SetForegroundWindow(hWnd))
+ SetForegroundWindow(NULL);
+ GetCursorPos(&pt);
+ /* Display the try menu */
+ TrackPopupMenu(hMnu, TPM_LEFTALIGN | TPM_RIGHTBUTTON,
+ pt.x, pt.y, 0, hWnd, NULL);
+ DestroyMenu(hMnu);
+}
+
+/* wParam progress dialog handle
+ */
+static BOOL __startServiceCallback(APXHANDLE hObject, UINT uMsg,
+ WPARAM wParam, LPARAM lParam)
+{
+ HWND hDlg = (HWND)hObject;
+
+ switch (uMsg) {
+ case WM_USER+1:
+ hDlg = (HWND)lParam;
+ if (IS_INVALID_HANDLE(hService)) {
+ EndDialog(hDlg, IDOK);
+ PostMessage(_gui_store->hMainWnd, WM_COMMAND,
+ MAKEWPARAM(IDMS_REFRESH, 0), 0);
+ return FALSE;
+ }
+ if (apxServiceControl(hService, SERVICE_CONTROL_CONTINUE, WM_USER+2,
+ __startServiceCallback, hDlg)) {
+ _currentEntry->stServiceStatus.dwCurrentState = SERVICE_RUNNING;
+ _currentEntry->stStatusProcess.dwCurrentState = SERVICE_RUNNING;
+
+ }
+ EndDialog(hDlg, IDOK);
+ PostMessage(_gui_store->hMainWnd, WM_COMMAND,
+ MAKEWPARAM(IDMS_REFRESH, 0), 0);
+ break;
+ case WM_USER+2:
+ SendMessage(hDlg, WM_USER+1, 0, 0);
+ Sleep(500);
+ break;
+ }
+ return TRUE;
+}
+
+static BOOL __stopServiceCallback(APXHANDLE hObject, UINT uMsg,
+ WPARAM wParam, LPARAM lParam)
+{
+ HWND hDlg = (HWND)hObject;
+
+ switch (uMsg) {
+ case WM_USER+1:
+ hDlg = (HWND)lParam;
+ if (IS_INVALID_HANDLE(hService))
+ return FALSE;
+ if (apxServiceControl(hService, SERVICE_CONTROL_STOP, WM_USER+2,
+ __stopServiceCallback, hDlg)) {
+ }
+ EndDialog(hDlg, IDOK);
+ PostMessage(_gui_store->hMainWnd, WM_COMMAND,
+ MAKEWPARAM(IDMS_REFRESH, 0), 0);
+ break;
+ case WM_USER+2:
+ if (wParam == 4)
+ AplCopyMemory(&_currentEntry->stServiceStatus,
+ (LPVOID)lParam, sizeof(SERVICE_STATUS));
+ SendMessage(hDlg, WM_USER+1, 0, 0);
+ Sleep(100);
+ break;
+ }
+ return TRUE;
+}
+
+static BOOL __restartServiceCallback(APXHANDLE hObject, UINT uMsg,
+ WPARAM wParam, LPARAM lParam)
+{
+ HWND hDlg = (HWND)hObject;
+ switch (uMsg) {
+ case WM_USER+1:
+ hDlg = (HWND)lParam;
+ if (IS_INVALID_HANDLE(hService))
+ return FALSE;
+ /* TODO: use 128 as controll code */
+ if (apxServiceControl(hService, 128, WM_USER+2,
+ __restartServiceCallback, hDlg)) {
+
+ }
+ EndDialog(hDlg, IDOK);
+ PostMessage(_gui_store->hMainWnd, WM_COMMAND,
+ MAKEWPARAM(IDMS_REFRESH, 0), 0);
+ break;
+ case WM_USER+2:
+ if (wParam == 4)
+ AplCopyMemory(&_currentEntry->stServiceStatus,
+ (LPVOID)lParam, sizeof(SERVICE_STATUS));
+
+ SendMessage(hDlg, WM_USER+1, 0, 0);
+ Sleep(100);
+ break;
+ }
+ return TRUE;
+}
+
+static BOOL __pauseServiceCallback(APXHANDLE hObject, UINT uMsg,
+ WPARAM wParam, LPARAM lParam)
+{
+ HWND hDlg = (HWND)hObject;
+ switch (uMsg) {
+ case WM_USER+1:
+ hDlg = (HWND)lParam;
+ if (IS_INVALID_HANDLE(hService))
+ return FALSE;
+ if (apxServiceControl(hService, SERVICE_CONTROL_PAUSE, WM_USER+2,
+ __pauseServiceCallback, hDlg)) {
+ }
+ EndDialog(hDlg, IDOK);
+ PostMessage(_gui_store->hMainWnd, WM_COMMAND,
+ MAKEWPARAM(IDMS_REFRESH, 0), 0);
+ break;
+ case WM_USER+2:
+ if (wParam == 4)
+ AplCopyMemory(&_currentEntry->stServiceStatus,
+ (LPVOID)lParam, sizeof(SERVICE_STATUS));
+ SendMessage(hDlg, WM_USER+1, 0, 0);
+ Sleep(100);
+ break;
+ }
+ return TRUE;
+}
+
+static DWORD _propertyChanged;
+static BOOL _propertyOpened = FALSE;
+static HWND _propertyHwnd = NULL;
+/* Service property pages */
+int CALLBACK __propertyCallback(HWND hwndPropSheet, UINT uMsg, LPARAM lParam)
+{
+ switch(uMsg) {
+ case PSCB_PRECREATE:
+ {
+ LPDLGTEMPLATE lpTemplate = (LPDLGTEMPLATE)lParam;
+ if (!(lpTemplate->style & WS_SYSMENU))
+ lpTemplate->style |= WS_SYSMENU;
+ _propertyHwnd = hwndPropSheet;
+
+ _propertyChanged = 0;
+ _propertyOpened = TRUE;
+ return TRUE;
+ }
+ break;
+ case PSCB_INITIALIZED:
+ break;
+ }
+ return TRUE;
+}
+
+BOOL __generalPropertySave(HWND hDlg)
+{
+ WCHAR szN[SIZ_RESLEN];
+ WCHAR szD[SIZ_DESLEN];
+ DWORD dwStartType = SERVICE_NO_CHANGE;
+ int i;
+
+ if (!(TST_BIT_FLAG(_propertyChanged, 1)))
+ return TRUE;
+ CLR_BIT_FLAG(_propertyChanged, 1);
+
+ if (IS_INVALID_HANDLE(hService))
+ return FALSE;
+ GetDlgItemTextW(hDlg, IDC_PPSGDISP, szN, SIZ_RESMAX);
+ GetDlgItemTextW(hDlg, IDC_PPSGDESC, szD, SIZ_DESMAX);
+ i = ComboBox_GetCurSel(GetDlgItem(hDlg, IDC_PPSGCMBST));
+ if (i == 0)
+ dwStartType = SERVICE_AUTO_START;
+ else if (i == 1)
+ dwStartType = SERVICE_DEMAND_START;
+ else if (i == 2)
+ dwStartType = SERVICE_DISABLED;
+ apxServiceSetNames(hService, NULL, szN, szD, NULL, NULL);
+ apxServiceSetOptions(hService, SERVICE_NO_CHANGE, dwStartType, SERVICE_NO_CHANGE);
+
+ if (!(TST_BIT_FLAG(_propertyChanged, 2)))
+ PostMessage(_gui_store->hMainWnd, WM_COMMAND, MAKEWPARAM(IDMS_REFRESH, 0), 0);
+
+ return TRUE;
+}
+
+BOOL __generalLogonSave(HWND hDlg)
+{
+ WCHAR szU[SIZ_RESLEN];
+ WCHAR szP[SIZ_RESLEN];
+ WCHAR szC[SIZ_RESLEN];
+ DWORD dwStartType = SERVICE_NO_CHANGE;
+
+ if (!(TST_BIT_FLAG(_propertyChanged, 2)))
+ return TRUE;
+ CLR_BIT_FLAG(_propertyChanged, 2);
+
+ if (IS_INVALID_HANDLE(hService))
+ return FALSE;
+ GetDlgItemTextW(hDlg, IDC_PPSLUSER, szU, SIZ_RESMAX);
+ GetDlgItemTextW(hDlg, IDC_PPSLPASS, szP, SIZ_RESMAX);
+ GetDlgItemTextW(hDlg, IDC_PPSLCPASS, szC, SIZ_RESMAX);
+
+ if (lstrlenW(szU) && lstrcmpiW(szU, STAT_SYSTEM)) {
+ if (szP[0] != L' ' && szC[0] != L' ' && !lstrcmpW(szP, szC)) {
+ apxServiceSetNames(hService, NULL, NULL, NULL, szU, szP);
+ lstrlcpyW(_currentEntry->szObjectName, SIZ_RESLEN, szU);
+ }
+ else {
+ MessageBoxW(hDlg, apxLoadResourceW(IDS_VALIDPASS, 0),
+ apxLoadResourceW(IDS_APPLICATION, 1),
+ MB_OK | MB_ICONSTOP);
+ return FALSE;
+ }
+ }
+ else {
+ apxServiceSetNames(hService, NULL, NULL, NULL, STAT_SYSTEM, L"");
+ lstrlcpyW(_currentEntry->szObjectName, SIZ_RESLEN, STAT_SYSTEM);
+ if (IsDlgButtonChecked(hDlg, IDC_PPSLID) == BST_CHECKED)
+ apxServiceSetOptions(hService,
+ _currentEntry->stServiceStatus.dwServiceType | SERVICE_INTERACTIVE_PROCESS,
+ SERVICE_NO_CHANGE, SERVICE_NO_CHANGE);
+ else
+ apxServiceSetOptions(hService,
+ _currentEntry->stServiceStatus.dwServiceType & ~SERVICE_INTERACTIVE_PROCESS,
+ SERVICE_NO_CHANGE, SERVICE_NO_CHANGE);
+ }
+ if (!(TST_BIT_FLAG(_propertyChanged, 1)))
+ PostMessage(_gui_store->hMainWnd, WM_COMMAND, MAKEWPARAM(IDMS_REFRESH, 0), 0);
+ return TRUE;
+}
+
+BOOL __generalLoggingSave(HWND hDlg)
+{
+ WCHAR szB[SIZ_DESLEN];
+
+ if (!(TST_BIT_FLAG(_propertyChanged, 3)))
+ return TRUE;
+ CLR_BIT_FLAG(_propertyChanged, 3);
+
+ if (IS_INVALID_HANDLE(hService))
+ return FALSE;
+
+ GetDlgItemTextW(hDlg, IDC_PPLGLEVEL, szB, SIZ_DESMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_log, L"Level", szB);
+ GetDlgItemTextW(hDlg, IDC_PPLGPATH, szB, SIZ_DESMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_log, L"Path", szB);
+ GetDlgItemTextW(hDlg, IDC_PPLGPREFIX, szB, SIZ_DESMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_log, L"Prefix", szB);
+ GetDlgItemTextW(hDlg, IDC_PPLGPIDFILE, szB, SIZ_DESMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_log, L"PidFile", szB);
+ GetDlgItemTextW(hDlg, IDC_PPLGSTDOUT, szB, SIZ_DESMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_log, L"StdOutput", szB);
+ GetDlgItemTextW(hDlg, IDC_PPLGSTDERR, szB, SIZ_DESMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_log, L"StdError", szB);
+
+ if (!(TST_BIT_FLAG(_propertyChanged, 1)))
+ PostMessage(_gui_store->hMainWnd, WM_COMMAND, MAKEWPARAM(IDMS_REFRESH, 0), 0);
+ return TRUE;
+}
+
+BOOL __generalJvmSave(HWND hDlg)
+{
+ WCHAR szB[SIZ_HUGLEN];
+ LPWSTR p, s;
+ DWORD l;
+ if (!(TST_BIT_FLAG(_propertyChanged, 4)))
+ return TRUE;
+ CLR_BIT_FLAG(_propertyChanged, 4);
+
+ if (IS_INVALID_HANDLE(hService))
+ return FALSE;
+ if (!IsDlgButtonChecked(hDlg, IDC_PPJAUTO)) {
+ GetDlgItemTextW(hDlg, IDC_PPJJVM, szB, SIZ_HUGMAX);
+ }
+ else
+ lstrcpyW(szB, L"auto");
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_java, L"Jvm", szB);
+ GetDlgItemTextW(hDlg, IDC_PPJCLASSPATH, szB, SIZ_HUGMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_java, L"Classpath", szB);
+
+ l = GetWindowTextLength(GetDlgItem(hDlg, IDC_PPJOPTIONS));
+ p = apxPoolAlloc(hPool, (l + 2) * sizeof(WCHAR));
+ GetDlgItemTextW(hDlg, IDC_PPJOPTIONS, p, l + 1);
+ s = apxCRLFToMszW(hPool, p, &l);
+ apxFree(p);
+ apxRegistrySetMzStrW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_java, L"Options", s, l);
+ if (!GetDlgItemTextW(hDlg, IDC_PPJMS, szB, SIZ_HUGMAX))
+ szB[0] = L'\0';
+
+ apxRegistrySetNumW(hRegserv, APXREG_PARAMSOFTWARE, _s_java, L"JvmMs",
+ apxAtoulW(szB));
+ if (!GetDlgItemTextW(hDlg, IDC_PPJMX, szB, SIZ_DESMAX))
+ szB[0] = L'\0';
+ apxRegistrySetNumW(hRegserv, APXREG_PARAMSOFTWARE, _s_java, L"JvmMx",
+ apxAtoulW(szB));
+ if (!GetDlgItemTextW(hDlg, IDC_PPJSS, szB, SIZ_DESMAX))
+ szB[0] = L'\0';
+ apxRegistrySetNumW(hRegserv, APXREG_PARAMSOFTWARE, _s_java, L"JvmSs",
+ apxAtoulW(szB));
+ apxFree(s);
+ if (!(TST_BIT_FLAG(_propertyChanged, 1)))
+ PostMessage(_gui_store->hMainWnd, WM_COMMAND, MAKEWPARAM(IDMS_REFRESH, 0), 0);
+ return TRUE;
+}
+
+BOOL __generalStartSave(HWND hDlg)
+{
+ WCHAR szB[SIZ_HUGLEN];
+ LPWSTR p, s;
+ DWORD l;
+
+ if (!(TST_BIT_FLAG(_propertyChanged, 5)))
+ return TRUE;
+ CLR_BIT_FLAG(_propertyChanged, 5);
+
+ if (IS_INVALID_HANDLE(hService))
+ return FALSE;
+
+ GetDlgItemTextW(hDlg, IDC_PPRCLASS, szB, SIZ_HUGMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_start, L"Class", szB);
+ GetDlgItemTextW(hDlg, IDC_PPRIMAGE, szB, SIZ_HUGMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_start, L"Image", szB);
+ GetDlgItemTextW(hDlg, IDC_PPRWPATH, szB, SIZ_HUGMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_start, L"WorkingPath", szB);
+ GetDlgItemTextW(hDlg, IDC_PPRMETHOD, szB, SIZ_HUGMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_start, L"Method", szB);
+ GetDlgItemTextW(hDlg, IDC_PPRMODE, szB, SIZ_HUGMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_start, L"Mode", szB);
+
+ l = GetWindowTextLength(GetDlgItem(hDlg, IDC_PPRARGS));
+ p = apxPoolAlloc(hPool, (l + 2) * sizeof(WCHAR));
+ GetDlgItemTextW(hDlg, IDC_PPRARGS, p, l + 1);
+ s = apxCRLFToMszW(hPool, p, &l);
+ apxFree(p);
+ apxRegistrySetMzStrW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_start, L"Params", s, l);
+ apxFree(s);
+
+ if (!(TST_BIT_FLAG(_propertyChanged, 1)))
+ PostMessage(_gui_store->hMainWnd, WM_COMMAND, MAKEWPARAM(IDMS_REFRESH, 0), 0);
+ return TRUE;
+}
+
+BOOL __generalStopSave(HWND hDlg)
+{
+ WCHAR szB[SIZ_HUGLEN];
+ LPWSTR p, s;
+ DWORD l;
+
+ if (!(TST_BIT_FLAG(_propertyChanged, 6)))
+ return TRUE;
+ CLR_BIT_FLAG(_propertyChanged, 6);
+
+ if (IS_INVALID_HANDLE(hService))
+ return FALSE;
+
+ GetDlgItemTextW(hDlg, IDC_PPSCLASS, szB, SIZ_HUGMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_stop, L"Class", szB);
+ GetDlgItemTextW(hDlg, IDC_PPSIMAGE, szB, SIZ_HUGMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_stop, L"Image", szB);
+ GetDlgItemTextW(hDlg, IDC_PPSWPATH, szB, SIZ_HUGMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_stop, L"WorkingPath", szB);
+ GetDlgItemTextW(hDlg, IDC_PPSMETHOD, szB, SIZ_HUGMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_stop, L"Method", szB);
+ GetDlgItemTextW(hDlg, IDC_PPSTIMEOUT, szB, SIZ_HUGMAX);
+ apxRegistrySetNumW(hRegserv, APXREG_PARAMSOFTWARE, _s_stop, L"Timeout", apxAtoulW(szB));
+ GetDlgItemTextW(hDlg, IDC_PPSMODE, szB, SIZ_HUGMAX);
+ apxRegistrySetStrW(hRegserv, APXREG_PARAMSOFTWARE, _s_stop, L"Mode", szB);
+
+ l = GetWindowTextLength(GetDlgItem(hDlg, IDC_PPSARGS));
+ p = apxPoolAlloc(hPool, (l + 2) * sizeof(WCHAR));
+ GetDlgItemTextW(hDlg, IDC_PPSARGS, p, l + 1);
+ s = apxCRLFToMszW(hPool, p, &l);
+ apxFree(p);
+ apxRegistrySetMzStrW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_stop, L"Params", s, l);
+ apxFree(s);
+
+ if (!(TST_BIT_FLAG(_propertyChanged, 1)))
+ PostMessage(_gui_store->hMainWnd, WM_COMMAND, MAKEWPARAM(IDMS_REFRESH, 0), 0);
+ return TRUE;
+}
+
+void __generalPropertyRefresh(HWND hDlg)
+{
+ Button_Enable(GetDlgItem(hDlg, IDC_PPSGSTART), FALSE);
+ Button_Enable(GetDlgItem(hDlg, IDC_PPSGSTOP), FALSE);
+ Button_Enable(GetDlgItem(hDlg, IDC_PPSGPAUSE), FALSE);
+ Button_Enable(GetDlgItem(hDlg, IDC_PPSGRESTART), FALSE);
+ switch (_currentEntry->stServiceStatus.dwCurrentState) {
+ case SERVICE_RUNNING:
+ if (_currentEntry->stServiceStatus.dwControlsAccepted & SERVICE_ACCEPT_STOP ||
+ _currentEntry->lpConfig->dwStartType != SERVICE_DISABLED) {
+ Button_Enable(GetDlgItem(hDlg, IDC_PPSGSTOP), TRUE);
+ SetDlgItemText(hDlg, IDC_PPSGSTATUS, STAT_STARTED);
+ }
+ else
+ SetDlgItemText(hDlg, IDC_PPSGSTATUS, STAT_DISABLED);
+ if (_currentEntry->stServiceStatus.dwControlsAccepted & SERVICE_ACCEPT_PAUSE_CONTINUE) {
+ Button_Enable(GetDlgItem(hDlg, IDC_PPSGPAUSE), TRUE);
+ Button_Enable(GetDlgItem(hDlg, IDC_PPSGRESTART), TRUE);
+ }
+ break;
+ case SERVICE_PAUSED:
+ Button_Enable(GetDlgItem(hDlg, IDC_PPSGSTART), TRUE);
+ Button_Enable(GetDlgItem(hDlg, IDC_PPSGRESTART), TRUE);
+ SetDlgItemText(hDlg, IDC_PPSGSTATUS, STAT_PAUSED);
+ break;
+ case SERVICE_STOPPED:
+ if (_currentEntry->lpConfig->dwStartType != SERVICE_DISABLED) {
+ Button_Enable(GetDlgItem(hDlg, IDC_PPSGSTART), TRUE);
+ SetDlgItemText(hDlg, IDC_PPSGSTATUS, STAT_STOPPED);
+ }
+ else
+ SetDlgItemText(hDlg, IDC_PPSGSTATUS, STAT_DISABLED);
+ break;
+ default:
+ break;
+ }
+}
+
+static BOOL bpropCentered = FALSE;
+LRESULT CALLBACK __generalProperty(HWND hDlg,
+ UINT uMessage,
+ WPARAM wParam,
+ LPARAM lParam)
+{
+ LPPSHNOTIFY lpShn;
+ WCHAR szBuf[SIZ_DESLEN];
+
+ switch (uMessage) {
+ case WM_INITDIALOG:
+ {
+ if (!bEnableTry)
+ apxCenterWindow(GetParent(hDlg), NULL);
+ else if (!bpropCentered)
+ apxCenterWindow(GetParent(hDlg), NULL);
+ bpropCentered = TRUE;
+ startPage = 0;
+ if (!bEnableTry)
+ apxCenterWindow(GetParent(hDlg), NULL);
+ SendMessage(GetDlgItem(hDlg, IDC_PPSGDISP), EM_LIMITTEXT, SIZ_RESMAX, 0);
+ SendMessage(GetDlgItem(hDlg, IDC_PPSGDESC), EM_LIMITTEXT, SIZ_DESMAX, 0);
+
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPSGCMBST), START_AUTO);
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPSGCMBST), START_MANUAL);
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPSGCMBST), START_DISABLED);
+ if (_currentEntry->lpConfig->dwStartType == SERVICE_AUTO_START)
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPSGCMBST), 0);
+ else if (_currentEntry->lpConfig->dwStartType == SERVICE_DEMAND_START)
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPSGCMBST), 1);
+ else if (_currentEntry->lpConfig->dwStartType == SERVICE_DISABLED)
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPSGCMBST), 2);
+
+ SetDlgItemTextW(hDlg, IDC_PPSGNAME, _currentEntry->szServiceName);
+ SetDlgItemTextW(hDlg, IDC_PPSGDISP, _currentEntry->lpConfig->lpDisplayName);
+ SetDlgItemTextW(hDlg, IDC_PPSGDESC, _currentEntry->szServiceDescription);
+ SetDlgItemTextW(hDlg, IDC_PPSGDEXE, _currentEntry->lpConfig->lpBinaryPathName);
+ __generalPropertyRefresh(hDlg);
+ }
+ break;
+ case WM_COMMAND:
+ switch (LOWORD(wParam)) {
+ case IDC_PPSGCMBST:
+ if (HIWORD(wParam) == CBN_SELCHANGE) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 1);
+ }
+ break;
+ case IDC_PPSGDISP:
+ if (HIWORD(wParam) == EN_CHANGE) {
+ GetDlgItemTextW(hDlg, IDC_PPSGDISP, szBuf, SIZ_RESMAX);
+ if (!lstrcmpW(szBuf, _currentEntry->lpConfig->lpDisplayName)) {
+ PropSheet_UnChanged(GetParent(hDlg), hDlg);
+ CLR_BIT_FLAG(_propertyChanged, 1);
+ }
+ else {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 1);
+ }
+ }
+ break;
+ case IDC_PPSGDESC:
+ if (HIWORD(wParam) == EN_CHANGE) {
+ GetDlgItemTextW(hDlg, IDC_PPSGDESC, szBuf, SIZ_DESMAX);
+ if (!lstrcmpW(szBuf, _currentEntry->szServiceDescription)) {
+ PropSheet_UnChanged(GetParent(hDlg), hDlg);
+ CLR_BIT_FLAG(_propertyChanged, 1);
+ }
+ else {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 1);
+ }
+ }
+ break;
+ case IDC_PPSGSTART:
+ apxProgressBox(hDlg, apxLoadResource(IDS_HSSTART, 0),
+ _currentEntry->lpConfig->lpDisplayName,
+ __startServiceCallback, NULL);
+ __generalPropertyRefresh(hDlg);
+ break;
+ case IDC_PPSGSTOP:
+ apxProgressBox(hDlg, apxLoadResource(IDS_HSSTOP, 0),
+ _currentEntry->lpConfig->lpDisplayName,
+ __stopServiceCallback, NULL);
+ __generalPropertyRefresh(hDlg);
+ break;
+ case IDC_PPSGPAUSE:
+ apxProgressBox(hDlg, apxLoadResource(IDS_HSPAUSE, 0),
+ _currentEntry->lpConfig->lpDisplayName,
+ __pauseServiceCallback, NULL);
+ __generalPropertyRefresh(hDlg);
+ break;
+ case IDC_PPSGRESTART:
+ apxProgressBox(hDlg, apxLoadResource(IDS_HSRESTART, 0),
+ _currentEntry->lpConfig->lpDisplayName,
+ __restartServiceCallback, NULL);
+ __generalPropertyRefresh(hDlg);
+ break;
+ }
+ break;
+ case WM_NOTIFY:
+ lpShn = (LPPSHNOTIFY )lParam;
+ switch (lpShn->hdr.code) {
+ case PSN_APPLY: /* sent when OK or Apply button pressed */
+ if (__generalPropertySave(hDlg)) {
+ PropSheet_UnChanged(GetParent(hDlg), hDlg);
+ }
+ else {
+ SET_BIT_FLAG(_propertyChanged, 1);
+ SetWindowLong(hDlg, DWLP_MSGRESULT,
+ PSNRET_INVALID_NOCHANGEPAGE);
+ return TRUE;
+ }
+
+ break;
+ default:
+ break;
+ }
+ break;
+ default:
+ break;
+ }
+
+ return FALSE;
+}
+
+LRESULT CALLBACK __logonProperty(HWND hDlg,
+ UINT uMessage,
+ WPARAM wParam,
+ LPARAM lParam)
+{
+ LPPSHNOTIFY lpShn;
+ WCHAR szBuf[SIZ_DESLEN];
+ switch (uMessage) {
+ case WM_INITDIALOG:
+ {
+ BOOL bAccount = FALSE;
+ startPage = 1;
+ if (!bpropCentered)
+ apxCenterWindow(GetParent(hDlg), NULL);
+ bpropCentered = TRUE;
+
+ SendMessage(GetDlgItem(hDlg, IDC_PPSLUSER), EM_LIMITTEXT, 63, 0);
+ SendMessage(GetDlgItem(hDlg, IDC_PPSLPASS), EM_LIMITTEXT, 63, 0);
+ SendMessage(GetDlgItem(hDlg, IDC_PPSLCPASS), EM_LIMITTEXT, 63, 0);
+ /* Check if we use LocalSystem or user defined account */
+ if (lstrcmpiW(_currentEntry->szObjectName, STAT_SYSTEM)) {
+ bAccount = TRUE;
+ CheckRadioButton(hDlg, IDC_PPSLLS, IDC_PPSLUA, IDC_PPSLUA);
+ SetDlgItemTextW(hDlg, IDC_PPSLUSER, _currentEntry->szObjectName);
+ SetDlgItemTextW(hDlg, IDC_PPSLPASS, EMPTY_PASSWORD);
+ SetDlgItemTextW(hDlg, IDC_PPSLCPASS, EMPTY_PASSWORD);
+ }
+ else {
+ CheckRadioButton(hDlg, IDC_PPSLLS, IDC_PPSLUA, IDC_PPSLLS);
+ if (_currentEntry->lpConfig->dwServiceType &
+ SERVICE_INTERACTIVE_PROCESS)
+ CheckDlgButton(hDlg, IDC_PPSLID, BST_CHECKED);
+ }
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLID), !bAccount);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLUSER), bAccount);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLBROWSE), bAccount);
+ EnableWindow(GetDlgItem(hDlg, IDL_PPSLPASS), bAccount);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLPASS), bAccount);
+ EnableWindow(GetDlgItem(hDlg, IDL_PPSLCPASS), bAccount);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLCPASS), bAccount);
+ }
+ break;
+ case WM_COMMAND:
+ switch (LOWORD(wParam)) {
+ case IDC_PPSLLS:
+ SetDlgItemTextW(hDlg, IDC_PPSLUSER, L"");
+ SetDlgItemTextW(hDlg, IDC_PPSLPASS, L"");
+ SetDlgItemTextW(hDlg, IDC_PPSLCPASS, L"");
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLID), TRUE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLUSER), FALSE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLBROWSE), FALSE);
+ EnableWindow(GetDlgItem(hDlg, IDL_PPSLPASS), FALSE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLPASS), FALSE);
+ EnableWindow(GetDlgItem(hDlg, IDL_PPSLCPASS), FALSE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLCPASS), FALSE);
+ CheckRadioButton(hDlg, IDC_PPSLLS, IDC_PPSLUA, (INT)wParam);
+ if (lstrcmpiW(_currentEntry->szObjectName, STAT_SYSTEM)) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 2);
+ }
+ else {
+ PropSheet_UnChanged(GetParent(hDlg), hDlg);
+ CLR_BIT_FLAG(_propertyChanged, 2);
+ }
+ break;
+ case IDC_PPSLUA:
+ SetDlgItemTextW(hDlg, IDC_PPSLUSER, _currentEntry->szObjectName);
+ SetDlgItemTextW(hDlg, IDC_PPSLPASS, EMPTY_PASSWORD);
+ SetDlgItemTextW(hDlg, IDC_PPSLCPASS, EMPTY_PASSWORD);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLID), FALSE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLUSER), TRUE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLBROWSE), TRUE);
+ EnableWindow(GetDlgItem(hDlg, IDL_PPSLPASS), TRUE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLPASS), TRUE);
+ EnableWindow(GetDlgItem(hDlg, IDL_PPSLCPASS), TRUE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSLCPASS), TRUE);
+ CheckRadioButton(hDlg, IDC_PPSLLS, IDC_PPSLUA, (INT)wParam);
+ if (lstrcmpW(_currentEntry->szObjectName, STAT_SYSTEM)) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 2);
+ }
+ else {
+ PropSheet_UnChanged(GetParent(hDlg), hDlg);
+ CLR_BIT_FLAG(_propertyChanged, 2);
+ }
+ break;
+ case IDC_PPSLID:
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 2);
+ break;
+ case IDC_PPSLUSER:
+ if (HIWORD(wParam) == EN_CHANGE) {
+ GetDlgItemTextW(hDlg, IDC_PPSLUSER, szBuf, SIZ_RESMAX);
+ if (!lstrcmpiW(szBuf, _currentEntry->szObjectName)) {
+ PropSheet_UnChanged(GetParent(hDlg), hDlg);
+ CLR_BIT_FLAG(_propertyChanged, 2);
+ }
+ else {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 2);
+ }
+ }
+ break;
+ case IDC_PPSLPASS:
+ case IDC_PPSLCPASS:
+ if (HIWORD(wParam) == EN_CHANGE) {
+ WCHAR szP[SIZ_RESLEN];
+ WCHAR szC[SIZ_RESLEN];
+ GetDlgItemTextW(hDlg, IDC_PPSLPASS, szP, SIZ_RESMAX);
+ GetDlgItemTextW(hDlg, IDC_PPSLCPASS, szC, SIZ_RESMAX);
+ /* check for valid password match */
+ if (szP[0] == L' ' && szC[0] == L' ') {
+ PropSheet_UnChanged(GetParent(hDlg), hDlg);
+ CLR_BIT_FLAG(_propertyChanged, 2);
+ }
+ else if (!lstrcmpW(szP, szC)) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 2);
+ }
+ }
+ break;
+ case IDC_PPSLBROWSE:
+ {
+ WCHAR szUser[SIZ_RESLEN];
+ if (apxDlgSelectUser(hDlg, szUser))
+ SetDlgItemTextW(hDlg, IDC_PPSLUSER, szUser);
+ }
+ break;
+ }
+ break;
+ case WM_NOTIFY:
+ lpShn = (LPPSHNOTIFY )lParam;
+ switch (lpShn->hdr.code) {
+ case PSN_APPLY: /* sent when OK or Apply button pressed */
+ if (__generalLogonSave(hDlg))
+ PropSheet_UnChanged(GetParent(hDlg), hDlg);
+ else {
+ SET_BIT_FLAG(_propertyChanged, 2);
+ SetWindowLong(hDlg, DWLP_MSGRESULT,
+ PSNRET_INVALID_NOCHANGEPAGE);
+ return TRUE;
+ }
+
+ break;
+ }
+ break;
+
+ default:
+ break;
+ }
+ return FALSE;
+}
+
+LRESULT CALLBACK __loggingProperty(HWND hDlg,
+ UINT uMessage,
+ WPARAM wParam,
+ LPARAM lParam)
+{
+ LPPSHNOTIFY lpShn;
+ LPWSTR lpBuf;
+
+ switch (uMessage) {
+ case WM_INITDIALOG:
+ {
+ LPWSTR b;
+ startPage = 2;
+ if (!bpropCentered)
+ apxCenterWindow(GetParent(hDlg), NULL);
+ bpropCentered = TRUE;
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPLGLEVEL), LOGL_ERROR);
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPLGLEVEL), LOGL_INFO);
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPLGLEVEL), LOGL_WARN);
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPLGLEVEL), LOGL_DEBUG);
+ if ((b = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_log, L"Level")) != NULL) {
+ if (!lstrcmpiW(b, LOGL_ERROR))
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPLGLEVEL), 0);
+ else if (!lstrcmpiW(b, LOGL_INFO))
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPLGLEVEL), 1);
+ else if (!lstrcmpiW(b, LOGL_WARN))
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPLGLEVEL), 2);
+ else
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPLGLEVEL), 3);
+ apxFree(b);
+ }
+ else
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPLGLEVEL), 1);
+ if ((b = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_log, L"Path")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPLGPATH, b);
+ apxFree(b);
+ }
+ if ((b = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_log, L"Prefix")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPLGPREFIX, b);
+ apxFree(b);
+ }
+ else
+ SetDlgItemTextW(hDlg, IDC_PPLGPREFIX, L"commons-daemon");
+ if ((b = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_log, L"PidFile")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPLGPIDFILE, b);
+ apxFree(b);
+ }
+ if ((b = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_log, L"StdOutput")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPLGSTDOUT, b);
+ apxFree(b);
+ }
+ if ((b = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_log, L"StdError")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPLGSTDERR, b);
+ apxFree(b);
+ }
+ }
+ break;
+ case WM_COMMAND:
+ switch (LOWORD(wParam)) {
+ case IDC_PPLGLEVEL:
+ if (HIWORD(wParam) == CBN_SELCHANGE) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 3);
+ }
+ break;
+ case IDC_PPLGPATH:
+ if (HIWORD(wParam) == EN_CHANGE) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 3);
+ }
+ break;
+ case IDC_PPLGPREFIX:
+ if (HIWORD(wParam) == EN_CHANGE) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 3);
+ }
+ break;
+ case IDC_PPLGPIDFILE:
+ if (HIWORD(wParam) == EN_CHANGE) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 3);
+ }
+ break;
+ case IDC_PPLGSTDERR:
+ if (HIWORD(wParam) == EN_CHANGE) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 3);
+ }
+ break;
+ case IDC_PPLGSTDOUT:
+ if (HIWORD(wParam) == EN_CHANGE) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 3);
+ }
+ break;
+ case IDC_PPLGBPATH:
+ lpBuf = apxBrowseForFolderW(hDlg, apxLoadResourceW(IDS_LGPATHTITLE, 0),
+ NULL);
+ if (lpBuf) {
+ SetDlgItemTextW(hDlg, IDC_PPLGPATH, lpBuf);
+ apxFree(lpBuf);
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 3);
+ }
+ break;
+ case IDC_PPLGBSTDOUT:
+ lpBuf = apxGetFileNameW(hDlg, apxLoadResourceW(IDS_LGSTDOUT, 0),
+ apxLoadResourceW(IDS_ALLFILES, 1), NULL,
+ NULL, FALSE, NULL);
+ if (lpBuf) {
+ SetDlgItemTextW(hDlg, IDC_PPLGSTDOUT, lpBuf);
+ apxFree(lpBuf);
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 3);
+ }
+ break;
+ case IDC_PPLGBSTDERR:
+ lpBuf = apxGetFileNameW(hDlg, apxLoadResourceW(IDS_LGSTDERR, 0),
+ apxLoadResourceW(IDS_ALLFILES, 1), NULL,
+ NULL, FALSE, NULL);
+ if (lpBuf) {
+ SetDlgItemTextW(hDlg, IDC_PPLGSTDERR, lpBuf);
+ apxFree(lpBuf);
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 3);
+ }
+ break;
+ }
+ break;
+ case WM_NOTIFY:
+ lpShn = (LPPSHNOTIFY )lParam;
+ switch (lpShn->hdr.code) {
+ case PSN_APPLY: /* sent when OK or Apply button pressed */
+ if (__generalLoggingSave(hDlg))
+ PropSheet_UnChanged(GetParent(hDlg), hDlg);
+ else {
+ SET_BIT_FLAG(_propertyChanged, 3);
+ SetWindowLong(hDlg, DWLP_MSGRESULT,
+ PSNRET_INVALID_NOCHANGEPAGE);
+ return TRUE;
+ }
+
+ break;
+ }
+ break;
+
+ default:
+ break;
+ }
+ return FALSE;
+}
+
+LRESULT CALLBACK __jvmProperty(HWND hDlg,
+ UINT uMessage,
+ WPARAM wParam,
+ LPARAM lParam)
+{
+ LPPSHNOTIFY lpShn;
+ LPWSTR lpBuf, b;
+ DWORD v;
+ CHAR bn[32];
+
+ switch (uMessage) {
+ case WM_INITDIALOG:
+ {
+ startPage = 3;
+ if (!bpropCentered)
+ apxCenterWindow(GetParent(hDlg), NULL);
+ bpropCentered = TRUE;
+ if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_java, L"Jvm")) != NULL) {
+ if (!lstrcmpiW(lpBuf, L"auto")) {
+ CheckDlgButton(hDlg, IDC_PPJAUTO, BST_CHECKED);
+ apxFree(lpBuf);
+ lpBuf = apxGetJavaSoftRuntimeLib(hPool);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPJJVM), FALSE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPJBJVM), FALSE);
+ }
+ if (lpBuf) {
+ SetDlgItemTextW(hDlg, IDC_PPJJVM, lpBuf);
+ apxFree(lpBuf);
+ }
+ }
+ if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_java, L"Classpath")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPJCLASSPATH, lpBuf);
+ apxFree(lpBuf);
+ }
+ if ((lpBuf = apxRegistryGetMzStrW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_java, L"Options", NULL, NULL)) != NULL) {
+ LPWSTR p = apxMszToCRLFW(hPool, lpBuf);
+ SetDlgItemTextW(hDlg, IDC_PPJOPTIONS, p);
+ apxFree(lpBuf);
+ apxFree(p);
+ }
+ v = apxRegistryGetNumberW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_java, L"JvmMs");
+ if (v && v != 0xFFFFFFFF) {
+ wsprintfA(bn, "%d", v);
+ SetDlgItemTextA(hDlg, IDC_PPJMS, bn);
+ }
+ v = apxRegistryGetNumberW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_java, L"JvmMx");
+ if (v && v != 0xFFFFFFFF) {
+ wsprintfA(bn, "%d", v);
+ SetDlgItemTextA(hDlg, IDC_PPJMX, bn);
+ }
+ v = apxRegistryGetNumberW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_java, L"JvmSs");
+ if (v && v != 0xFFFFFFFF) {
+ wsprintfA(bn, "%d", v);
+ SetDlgItemTextA(hDlg, IDC_PPJSS, bn);
+ }
+
+ }
+ break;
+ case WM_COMMAND:
+ switch (LOWORD(wParam)) {
+ case IDC_PPJBJVM:
+ b = apxGetJavaSoftHome(hPool, TRUE);
+ lpBuf = apxGetFileNameW(hDlg, apxLoadResourceW(IDS_PPJBJVM, 0),
+ apxLoadResourceW(IDS_DLLFILES, 1), NULL,
+ b,
+ TRUE, NULL);
+ apxFree(b);
+ if (lpBuf) {
+ SetDlgItemTextW(hDlg, IDC_PPJJVM, lpBuf);
+ apxFree(lpBuf);
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 4);
+ }
+ break;
+ case IDC_PPJAUTO:
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 4);
+ if (IsDlgButtonChecked(hDlg, IDC_PPJAUTO)) {
+ EnableWindow(GetDlgItem(hDlg, IDC_PPJJVM), FALSE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPJBJVM), FALSE);
+ lpBuf = apxGetJavaSoftRuntimeLib(hPool);
+ if (lpBuf) {
+ SetDlgItemTextW(hDlg, IDC_PPJJVM, lpBuf);
+ apxFree(lpBuf);
+ }
+ }
+ else {
+ EnableWindow(GetDlgItem(hDlg, IDC_PPJJVM), TRUE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPJBJVM), TRUE);
+ }
+ break;
+ case IDC_PPJJVM:
+ case IDC_PPJCLASSPATH:
+ case IDC_PPJOPTIONS:
+ case IDC_PPJMX:
+ case IDC_PPJMS:
+ case IDC_PPJSS:
+ if (HIWORD(wParam) == EN_CHANGE) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 4);
+ }
+ break;
+ }
+ break;
+ case WM_NOTIFY:
+ lpShn = (LPPSHNOTIFY )lParam;
+ switch (lpShn->hdr.code) {
+ case PSN_APPLY: /* sent when OK or Apply button pressed */
+ if (__generalJvmSave(hDlg))
+ PropSheet_UnChanged(GetParent(hDlg), hDlg);
+ else {
+ SET_BIT_FLAG(_propertyChanged, 4);
+ SetWindowLong(hDlg, DWLP_MSGRESULT,
+ PSNRET_INVALID_NOCHANGEPAGE);
+ return TRUE;
+ }
+
+ break;
+ }
+ break;
+
+ default:
+ break;
+ }
+ return FALSE;
+}
+
+LRESULT CALLBACK __startProperty(HWND hDlg,
+ UINT uMessage,
+ WPARAM wParam,
+ LPARAM lParam)
+{
+ LPPSHNOTIFY lpShn;
+ LPWSTR lpBuf, b;
+
+ switch (uMessage) {
+ case WM_INITDIALOG:
+ {
+ startPage = 4;
+ if (!bpropCentered)
+ apxCenterWindow(GetParent(hDlg), NULL);
+ bpropCentered = TRUE;
+
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPRMODE), L"exe");
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPRMODE), L"jvm");
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPRMODE), _s_java);
+
+ if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_start, L"Class")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPRCLASS, lpBuf);
+ apxFree(lpBuf);
+ }
+ if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_start, L"Image")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPRIMAGE, lpBuf);
+ apxFree(lpBuf);
+ }
+ if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_start, L"WorkingPath")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPRWPATH, lpBuf);
+ apxFree(lpBuf);
+ }
+ if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_start, L"Method")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPRMETHOD, lpBuf);
+ apxFree(lpBuf);
+ }
+ if ((lpBuf = apxRegistryGetMzStrW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_start, L"Params", NULL, NULL)) != NULL) {
+ b = apxMszToCRLFW(hPool, lpBuf);
+ SetDlgItemTextW(hDlg, IDC_PPRARGS, b);
+ apxFree(lpBuf);
+ apxFree(b);
+ }
+ if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_start, L"Mode")) != NULL) {
+ if (!lstrcmpiW(lpBuf, L"jvm")) {
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPRMODE), 1);
+
+ }
+ else if (!lstrcmpiW(lpBuf, _s_java)) {
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPRMODE), 2);
+ }
+ else {
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPRMODE), 0);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPRIMAGE), TRUE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPRBIMAGE), TRUE);
+ }
+ apxFree(lpBuf);
+ }
+ }
+ break;
+ case WM_COMMAND:
+ switch (LOWORD(wParam)) {
+ case IDC_PPRBWPATH:
+ lpBuf = apxBrowseForFolderW(hDlg, apxLoadResourceW(IDS_PPWPATH, 0),
+ NULL);
+ if (lpBuf) {
+ SetDlgItemTextW(hDlg, IDC_PPRWPATH, lpBuf);
+ apxFree(lpBuf);
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 5);
+ }
+ break;
+ case IDC_PPRBIMAGE:
+ lpBuf = apxGetFileNameW(hDlg, apxLoadResourceW(IDS_PPIMAGE, 0),
+ apxLoadResourceW(IDS_EXEFILES, 1), NULL,
+ NULL, TRUE, NULL);
+ if (lpBuf) {
+ SetDlgItemTextW(hDlg, IDC_PPRIMAGE, lpBuf);
+ apxFree(lpBuf);
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 5);
+ }
+ break;
+ case IDC_PPRCLASS:
+ case IDC_PPRMETHOD:
+ case IDC_PPRARGS:
+ case IDC_PPRIMAGE:
+ case IDC_PPRWPATH:
+ if (HIWORD(wParam) == EN_CHANGE) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 5);
+ }
+ break;
+ case IDC_PPRMODE:
+ if (HIWORD(wParam) == CBN_SELCHANGE) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 5);
+ if (ComboBox_GetCurSel(GetDlgItem(hDlg, IDC_PPRMODE))) {
+ EnableWindow(GetDlgItem(hDlg, IDC_PPRIMAGE), FALSE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPRBIMAGE), FALSE);
+ }
+ else {
+ EnableWindow(GetDlgItem(hDlg, IDC_PPRIMAGE), TRUE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPRBIMAGE), TRUE);
+ }
+ }
+ break;
+
+ }
+ break;
+ case WM_NOTIFY:
+ lpShn = (LPPSHNOTIFY )lParam;
+ switch (lpShn->hdr.code) {
+ case PSN_APPLY: /* sent when OK or Apply button pressed */
+ if (__generalStartSave(hDlg))
+ PropSheet_UnChanged(GetParent(hDlg), hDlg);
+ else {
+ SET_BIT_FLAG(_propertyChanged, 5);
+ SetWindowLong(hDlg, DWLP_MSGRESULT,
+ PSNRET_INVALID_NOCHANGEPAGE);
+ return TRUE;
+ }
+
+ break;
+ }
+ break;
+
+ default:
+ break;
+ }
+ return FALSE;
+}
+
+LRESULT CALLBACK __stopProperty(HWND hDlg,
+ UINT uMessage,
+ WPARAM wParam,
+ LPARAM lParam)
+{
+ LPPSHNOTIFY lpShn;
+ LPWSTR lpBuf, b;
+ DWORD v;
+
+ switch (uMessage) {
+ case WM_INITDIALOG:
+ {
+ startPage = 5;
+ if (!bpropCentered)
+ apxCenterWindow(GetParent(hDlg), NULL);
+ bpropCentered = TRUE;
+
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPSMODE), L"exe");
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPSMODE), L"jvm");
+ ComboBox_AddStringW(GetDlgItem(hDlg, IDC_PPSMODE), _s_java);
+
+ if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_stop, L"Class")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPSCLASS, lpBuf);
+ apxFree(lpBuf);
+ }
+ if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_stop, L"Image")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPSIMAGE, lpBuf);
+ apxFree(lpBuf);
+ }
+ if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_stop, L"WorkingPath")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPSWPATH, lpBuf);
+ apxFree(lpBuf);
+ }
+ if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_stop, L"Method")) != NULL) {
+ SetDlgItemTextW(hDlg, IDC_PPSMETHOD, lpBuf);
+ apxFree(lpBuf);
+ }
+ if ((lpBuf = apxRegistryGetMzStrW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_stop, L"Params", NULL, NULL)) != NULL) {
+ b = apxMszToCRLFW(hPool, lpBuf);
+ SetDlgItemTextW(hDlg, IDC_PPSARGS, b);
+ apxFree(lpBuf);
+ apxFree(b);
+ }
+ v = apxRegistryGetNumberW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_stop, L"Timeout");
+ {
+ CHAR bn[32];
+ wsprintfA(bn, "%d", v);
+ SetDlgItemTextA(hDlg, IDC_PPSTIMEOUT, bn);
+ }
+ if ((lpBuf = apxRegistryGetStringW(hRegserv, APXREG_PARAMSOFTWARE,
+ _s_stop, L"Mode")) != NULL) {
+ if (!lstrcmpiW(lpBuf, L"jvm")) {
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPSMODE), 1);
+
+ }
+ else if (!lstrcmpiW(lpBuf, _s_java)) {
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPSMODE), 2);
+ }
+ else {
+ ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_PPSMODE), 0);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSIMAGE), TRUE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSBIMAGE), TRUE);
+ }
+ apxFree(lpBuf);
+ }
+ }
+ break;
+ case WM_COMMAND:
+ switch (LOWORD(wParam)) {
+ case IDC_PPSBWPATH:
+ lpBuf = apxBrowseForFolderW(hDlg, apxLoadResourceW(IDS_PPWPATH, 0),
+ NULL);
+ if (lpBuf) {
+ SetDlgItemTextW(hDlg, IDC_PPSWPATH, lpBuf);
+ apxFree(lpBuf);
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 6);
+ }
+ break;
+ case IDC_PPSBIMAGE:
+ lpBuf = apxGetFileNameW(hDlg, apxLoadResourceW(IDS_PPIMAGE, 0),
+ apxLoadResourceW(IDS_EXEFILES, 1), NULL,
+ NULL, TRUE, NULL);
+ if (lpBuf) {
+ SetDlgItemTextW(hDlg, IDC_PPSIMAGE, lpBuf);
+ apxFree(lpBuf);
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 6);
+ }
+ break;
+ case IDC_PPSCLASS:
+ case IDC_PPSMETHOD:
+ case IDC_PPSTIMEOUT:
+ case IDC_PPSARGS:
+ case IDC_PPSIMAGE:
+ case IDC_PPSWPATH:
+ if (HIWORD(wParam) == EN_CHANGE) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 6);
+ }
+ break;
+ case IDC_PPSMODE:
+ if (HIWORD(wParam) == CBN_SELCHANGE) {
+ PropSheet_Changed(GetParent(hDlg), hDlg);
+ SET_BIT_FLAG(_propertyChanged, 6);
+ if (ComboBox_GetCurSel(GetDlgItem(hDlg, IDC_PPSMODE))) {
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSIMAGE), FALSE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSBIMAGE), FALSE);
+ }
+ else {
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSIMAGE), TRUE);
+ EnableWindow(GetDlgItem(hDlg, IDC_PPSBIMAGE), TRUE);
+ }
+ }
+ break;
+
+ }
+ break;
+ case WM_NOTIFY:
+ lpShn = (LPPSHNOTIFY )lParam;
+ switch (lpShn->hdr.code) {
+ case PSN_APPLY: /* sent when OK or Apply button pressed */
+ if (__generalStopSave(hDlg))
+ PropSheet_UnChanged(GetParent(hDlg), hDlg);
+ else {
+ SET_BIT_FLAG(_propertyChanged, 6);
+ SetWindowLong(hDlg, DWLP_MSGRESULT,
+ PSNRET_INVALID_NOCHANGEPAGE);
+ return TRUE;
+ }
+
+ break;
+ }
+ break;
+
+ default:
+ break;
+ }
+ return FALSE;
+}
+
+void __initPpage(PROPSHEETPAGEW *lpPage, INT iDlg, INT iTitle, DLGPROC pfnDlgProc)
+{
+ lpPage->dwSize = sizeof(PROPSHEETPAGE);
+ lpPage->dwFlags = PSP_USETITLE;
+ lpPage->hInstance = _gui_store->hInstance;
+ lpPage->pszTemplate = MAKEINTRESOURCEW(iDlg);
+ lpPage->pszIcon = NULL;
+ lpPage->pfnDlgProc = pfnDlgProc;
+ lpPage->pszTitle = MAKEINTRESOURCEW(iTitle);
+ lpPage->lParam = 0;
+}
+
+void ShowServiceProperties(HWND hWnd)
+{
+ PROPSHEETPAGEW psP[6];
+ PROPSHEETHEADERW psH;
+ WCHAR szT[SIZ_DESLEN] = {0};
+
+ if (_propertyOpened) {
+ SetForegroundWindow(_gui_store->hMainWnd);
+ return;
+ }
+ __initPpage(&psP[0], IDD_PROPPAGE_SGENERAL, IDS_PPGENERAL,
+ __generalProperty);
+ __initPpage(&psP[1], IDD_PROPPAGE_LOGON, IDS_PPLOGON,
+ __logonProperty);
+ __initPpage(&psP[2], IDD_PROPPAGE_LOGGING, IDS_PPLOGGING,
+ __loggingProperty);
+ __initPpage(&psP[3], IDD_PROPPAGE_JVM, IDS_PPJAVAVM,
+ __jvmProperty);
+ __initPpage(&psP[4], IDD_PROPPAGE_START, IDS_PPSTART,
+ __startProperty);
+ __initPpage(&psP[5], IDD_PROPPAGE_STOP, IDS_PPSTOP,
+ __stopProperty);
+
+ if (_currentEntry && _currentEntry->lpConfig)
+ lstrlcpyW(szT, SIZ_DESMAX, _currentEntry->lpConfig->lpDisplayName);
+ else
+ return;
+ lstrlcatW(szT, SIZ_DESMAX, L" Properties");
+
+ psH.dwSize = sizeof(PROPSHEETHEADER);
+ psH.dwFlags = PSH_PROPSHEETPAGE | PSH_USEICONID | PSH_USECALLBACK | PSH_NOCONTEXTHELP;
+ psH.hwndParent = bEnableTry ? hWnd : NULL;
+ psH.hInstance = _gui_store->hInstance;
+ psH.pszIcon = MAKEINTRESOURCEW(IDI_MAINICON);
+ psH.pszCaption = szT;
+ psH.nPages = 6;
+ psH.ppsp = (LPCPROPSHEETPAGEW) &psP;
+ psH.pfnCallback = (PFNPROPSHEETCALLBACK)__propertyCallback;
+ psH.nStartPage = startPage;
+
+ PropertySheetW(&psH);
+ _propertyOpened = FALSE;
+ if (!bEnableTry)
+ PostQuitMessage(0);
+ bpropCentered = FALSE;
+
+}
+
+static void signalService(LPCWSTR szServiceName)
+{
+ HANDLE event;
+ WCHAR en[SIZ_DESLEN];
+ int i;
+
+ lstrlcpyW(en, SIZ_DESLEN, L"Global\\");
+ lstrlcatW(en, SIZ_DESLEN, szServiceName);
+ lstrlcatW(en, SIZ_DESLEN, L"SIGNAL");
+ for (i = 7; i < lstrlenW(en); i++) {
+ if (en[i] >= L'a' && en[i] <= L'z')
+ en[i] = en[i] - 32;
+ }
+
+
+ event = OpenEventW(EVENT_MODIFY_STATE, FALSE, en);
+ if (event) {
+ SetEvent(event);
+ CloseHandle(event);
+ }
+ else
+ apxDisplayError(TRUE, NULL, 0, "Unable to open the Event Mutex");
+
+}
+
+LRESULT CALLBACK MainWndProc(HWND hWnd, UINT uMsg,
+ WPARAM wParam, LPARAM lParam)
+{
+ switch (uMsg) {
+ case WM_CREATE:
+ if (bEnableTry) {
+ if (_currentEntry && _currentEntry->lpConfig) {
+ BOOL isRunning = _currentEntry->stServiceStatus.dwCurrentState == SERVICE_RUNNING;
+ apxManageTryIconW(hWnd, NIM_ADD, NULL,
+ _currentEntry->lpConfig->lpDisplayName,
+ isRunning ? hIcoRun : hIcoStop);
+ }
+ else {
+ apxManageTryIconA(hWnd, NIM_ADD, NULL,
+ apxLoadResourceA(IDS_APPLICATION, 0),
+ NULL);
+ }
+ }
+ else
+ ShowServiceProperties(hWnd);
+
+ break;
+ case WM_COMMAND:
+ switch(LOWORD(wParam)) {
+ case IDM_TM_CONFIG:
+ ShowServiceProperties(hWnd);
+ break;
+ case IDM_TM_ABOUT:
+ apxAboutBox(hWnd);
+ break;
+ case IDM_TM_EXIT:
+ SendMessage(hWnd, WM_CLOSE, 0, 0);
+ break;
+ case IDM_TM_START:
+ if (!_propertyOpened)
+ apxProgressBox(hWnd, apxLoadResource(IDS_HSSTART, 0),
+ _currentEntry->lpConfig->lpDisplayName,
+ __startServiceCallback, NULL);
+ break;
+ case IDM_TM_STOP:
+ if (!_propertyOpened)
+ apxProgressBox(hWnd, apxLoadResource(IDS_HSSTOP, 0),
+ _currentEntry->lpConfig->lpDisplayName,
+ __stopServiceCallback, NULL);
+ break;
+ case IDM_TM_PAUSE:
+ if (!_propertyOpened)
+ apxProgressBox(hWnd, apxLoadResource(IDS_HSPAUSE, 0),
+ _currentEntry->lpConfig->lpDisplayName,
+ __pauseServiceCallback, NULL);
+ break;
+ case IDM_TM_RESTART:
+ if (!_propertyOpened)
+ apxProgressBox(hWnd, apxLoadResource(IDS_HSRESTART, 0),
+ _currentEntry->lpConfig->lpDisplayName,
+ __restartServiceCallback, NULL);
+ break;
+ case IDM_TM_DUMP:
+ signalService(_currentEntry->szServiceName);
+ break;
+ case IDMS_REFRESH:
+ if (bEnableTry &&
+ (_currentEntry = apxServiceEntry(hService, TRUE)) != NULL) {
+ BOOL isRunning = _currentEntry->stServiceStatus.dwCurrentState == SERVICE_RUNNING;
+ apxManageTryIconW(hWnd, NIM_MODIFY, NULL,
+ _currentEntry->lpConfig->lpDisplayName,
+ isRunning ? hIcoRun : hIcoStop);
+ }
+ break;
+
+ }
+ break;
+ case WM_TRAYMESSAGE:
+ switch(lParam) {
+ case WM_LBUTTONDBLCLK:
+ ShowServiceProperties(hWnd);
+ break;
+ case WM_RBUTTONUP:
+ _currentEntry = apxServiceEntry(hService, TRUE);
+ createRbuttonTryMenu(hWnd);
+ break;
+ }
+ break;
+ case WM_QUIT:
+ return DefWindowProc(hWnd, uMsg, wParam, lParam);
+ break;
+ case WM_DESTROY:
+ if (bEnableTry)
+ apxManageTryIconA(hWnd, NIM_DELETE, NULL, NULL, NULL);
+ PostQuitMessage(0);
+ break;
+ default:
+ return DefWindowProc(hWnd, uMsg, wParam, lParam);
+ break;
+ }
+
+ return FALSE;
+}
+
+static BOOL loadConfiguration()
+{
+ return TRUE;
+}
+
+static BOOL saveConfiguration()
+{
+ return TRUE;
+}
+
+/* Main program entry
+ * Since we are inependant from CRT
+ * the arguments are not used
+ */
+#ifdef _NO_CRTLIBRARY
+int xMain(void)
+#else
+int WINAPI WinMain(HINSTANCE hInstance,
+ HINSTANCE hPrevInstance,
+ LPSTR lpCmdLine,
+ int nCmdShow)
+#endif
+{
+ MSG msg;
+ LPAPXCMDLINE lpCmdline;
+ HANDLE mutex = NULL;
+ BOOL quiet = FALSE;
+
+ apxHandleManagerInitialize();
+ hPool = apxPoolCreate(NULL, 0);
+
+ /* Parse the command line */
+ if ((lpCmdline = apxCmdlineParse(hPool, _options, _commands)) == NULL) {
+ /* TODO: dispalay error message */
+ apxDisplayError(TRUE, NULL, 0, "Error parsing command line");
+ goto cleanup;
+ }
+
+ if (!lpCmdline->dwCmdIndex) {
+ /* Skip sytem error message */
+ SetLastError(ERROR_SUCCESS);
+ apxDisplayError(TRUE, NULL, 0,
+ apxLoadResourceA(IDS_ERRORCMD, 0),
+ lpCmdLine);
+ goto cleanup;
+ }
+ else if (lpCmdline->dwCmdIndex == 4)
+ quiet = TRUE;
+ else if (lpCmdline->dwCmdIndex >= 2)
+ bEnableTry = TRUE;
+ hService = apxCreateService(hPool, SC_MANAGER_CONNECT, FALSE);
+ if (IS_INVALID_HANDLE(hService)) {
+ if (!quiet)
+ apxDisplayError(TRUE, NULL, 0, "Unable to open the Service Manager");
+ goto cleanup;
+ }
+ /* Open the main service handle */
+ if (!apxServiceOpen(hService, lpCmdline->szApplication,
+ SERVICE_ALL_ACCESS)) {
+ LPWSTR w = lpCmdline->szApplication + lstrlenW(lpCmdline->szApplication) - 1;
+ if (*w == L'w')
+ *w = L'\0';
+ if (!apxServiceOpen(hService, lpCmdline->szApplication,
+ SERVICE_ALL_ACCESS)) {
+ if (!apxServiceOpen(hService, lpCmdline->szApplication,
+ GENERIC_READ | GENERIC_EXECUTE)) {
+
+ if (!quiet)
+ apxDisplayError(TRUE, NULL, 0, "Unable to open the service '%S'",
+ lpCmdline->szApplication);
+ goto cleanup;
+ }
+ }
+ }
+ /* Obtain service parameters and status */
+ if (!(_currentEntry = apxServiceEntry(hService, TRUE))) {
+ if (!quiet)
+ apxDisplayError(TRUE, NULL, 0, "Unable to query the service '%S' status",
+ lpCmdline->szApplication);
+ goto cleanup;
+ }
+#ifdef _UNICODE
+ _gui_store = apxGuiInitialize(MainWndProc, lpCmdline->szApplication);
+#else
+ {
+ CHAR szApp[MAX_PATH];
+ _gui_store = apxGuiInitialize(MainWndProc,
+ WideToAscii(lpCmdline->szApplication, szApp));
+ }
+#endif
+ if (!_gui_store) {
+ if (!quiet)
+ apxDisplayError(TRUE, NULL, 0, "Unable to initialize GUI manager");
+ goto cleanup;
+ }
+ hIcoRun = LoadImage(_gui_store->hInstance, MAKEINTRESOURCE(IDI_ICONRUN),
+ IMAGE_ICON, 16, 16, LR_DEFAULTCOLOR);
+ hIcoStop = LoadImage(_gui_store->hInstance, MAKEINTRESOURCE(IDI_ICONSTOP),
+ IMAGE_ICON, 16, 16, LR_DEFAULTCOLOR);
+
+ /* Handle //MQ// option */
+ if (lpCmdline->dwCmdIndex == 4) {
+ HANDLE hOther = FindWindow(_gui_store->szWndClass, NULL);
+ if (hOther)
+ SendMessage(hOther, WM_CLOSE, 0, 0);
+ goto cleanup;
+ }
+
+ if (!_options[0].dwValue) {
+ mutex = CreateMutex(NULL, FALSE, _gui_store->szWndMutex);
+ if ((mutex == NULL) || (GetLastError() == ERROR_ALREADY_EXISTS)) {
+ /* Skip sytem error message */
+ SetLastError(ERROR_SUCCESS);
+ if (!quiet)
+ apxDisplayError(TRUE, NULL, 0, apxLoadResourceA(IDS_ALREAY_RUNING, 0),
+ lpCmdline->szApplication);
+ goto cleanup;
+ }
+ }
+ hRegistry = apxCreateRegistry(hPool, KEY_ALL_ACCESS, NULL,
+ apxLoadResource(IDS_APPLICATION, 0),
+ APXREG_USER);
+ loadConfiguration();
+ hRegserv = apxCreateRegistryW(hPool, KEY_READ | KEY_WRITE | KREG_WOW6432,
+ PRG_REGROOT,
+ lpCmdline->szApplication,
+ APXREG_SOFTWARE | APXREG_SERVICE);
+
+ if (IS_INVALID_HANDLE(hRegserv)) {
+ if (!quiet)
+ apxDisplayError(TRUE, NULL, 0, apxLoadResourceA(IDS_ERRSREG, 0));
+ return FALSE;
+ }
+ /* Create main invisible window */
+ _gui_store->hMainWnd = CreateWindow(_gui_store->szWndClass,
+ apxLoadResource(IDS_APPLICATION, 0),
+ 0, 0, 0, 0, 0,
+ NULL, NULL,
+ _gui_store->hInstance,
+ NULL);
+
+ if (!_gui_store->hMainWnd) {
+ goto cleanup;
+ }
+ if (lpCmdline->dwCmdIndex == 3)
+ PostMessage(_gui_store->hMainWnd, WM_COMMAND, IDM_TM_START, 0);
+
+ while (GetMessage(&msg, NULL, 0, 0)) {
+ if(!TranslateAccelerator(_gui_store->hMainWnd,
+ _gui_store->hAccel, &msg)) {
+ TranslateMessage(&msg);
+ DispatchMessage(&msg);
+ }
+ }
+ saveConfiguration();
+
+cleanup:
+ if (hIcoStop)
+ DestroyIcon(hIcoStop);
+ if (hIcoRun)
+ DestroyIcon(hIcoRun);
+ if (mutex)
+ CloseHandle(mutex);
+ if (lpCmdline)
+ apxCmdlineFree(lpCmdline);
+ apxCloseHandle(hService);
+ apxHandleManagerDestroy();
+ ExitProcess(0);
+ return 0;
+}
+
Deleted: tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.h
===================================================================
--- trunk/utils/windows/native/service/procrun/apps/prunmgr/prunmgr.h 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.h 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,141 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-/* ====================================================================
- *
- * Contributed by Mladen Turk <mturk at apache.org>
- * 05 Aug 2003
- * ====================================================================
- */
-
-#ifndef _PRUNMGR_H
-#define _PRUNMGR_H
-
-#undef PRG_VERSION
-#define PRG_VERSION "2.0.4.1"
-#define CSV_VERSION 2,0,4,1
-#define PRG_REGROOT L"JBoss\\JBoss Web 2"
-
-#define IDM_TM_EXIT 2000
-#define IDM_TM_START 2001
-#define IDM_TM_STOP 2002
-#define IDM_TM_PAUSE 2003
-#define IDM_TM_RESTART 2004
-#define IDM_TM_CONFIG 2005
-#define IDM_TM_ABOUT 2006
-#define IDM_TM_DUMP 2007
-
-#define IDMS_REFRESH 2020
-
-#define IDI_ICONSTOP 2030
-#define IDI_ICONRUN 2031
-
-
-
-/* Property pages */
-
-#define IDD_PROPPAGE_SGENERAL 2600
-#define IDC_PPSGNAME 2601
-#define IDC_PPSGDISP 2602
-#define IDC_PPSGDESC 2603
-#define IDC_PPSGDEXE 2604
-#define IDC_PPSGCMBST 2605
-#define IDC_PPSGSTATUS 2606
-#define IDC_PPSGSTART 2607
-#define IDC_PPSGSTOP 2608
-#define IDC_PPSGPAUSE 2609
-#define IDC_PPSGRESTART 2610
-
-#define IDD_PROPPAGE_LOGON 2620
-#define IDC_PPSLLS 2621
-#define IDC_PPSLID 2622
-#define IDC_PPSLUA 2623
-#define IDC_PPSLUSER 2624
-#define IDC_PPSLBROWSE 2625
-#define IDC_PPSLPASS 2626
-#define IDC_PPSLCPASS 2627
-#define IDL_PPSLPASS 2628
-#define IDL_PPSLCPASS 2629
-
-#define IDD_PROPPAGE_LOGGING 2640
-#define IDC_PPLGLEVEL 2641
-#define IDC_PPLGPATH 2642
-#define IDC_PPLGBPATH 2643
-#define IDC_PPLGPREFIX 2644
-#define IDC_PPLGSTDOUT 2645
-#define IDC_PPLGBSTDOUT 2646
-#define IDC_PPLGSTDERR 2647
-#define IDC_PPLGBSTDERR 2648
-
-#define IDD_PROPPAGE_JVM 2660
-#define IDC_PPJAUTO 2661
-#define IDC_PPJJVM 2662
-#define IDC_PPJBJVM 2663
-#define IDC_PPJCLASSPATH 2664
-#define IDC_PPJOPTIONS 2665
-#define IDC_PPJMS 2666
-#define IDC_PPJMX 2667
-#define IDC_PPJSS 2668
-
-#define IDD_PROPPAGE_START 2680
-#define IDC_PPRCLASS 2681
-#define IDC_PPRIMAGE 2682
-#define IDC_PPRBIMAGE 2683
-#define IDC_PPRWPATH 2684
-#define IDC_PPRBWPATH 2685
-#define IDC_PPRMETHOD 2686
-#define IDC_PPRARGS 2687
-#define IDC_PPRTIMEOUT 2688
-#define IDC_PPRMODE 2689
-
-#define IDD_PROPPAGE_STOP 2700
-#define IDC_PPSCLASS 2701
-#define IDC_PPSIMAGE 2702
-#define IDC_PPSBIMAGE 2703
-#define IDC_PPSWPATH 2704
-#define IDC_PPSBWPATH 2705
-#define IDC_PPSMETHOD 2706
-#define IDC_PPSARGS 2707
-#define IDC_PPSTIMEOUT 2708
-#define IDC_PPSMODE 2709
-
-#define IDS_ALREAY_RUNING 3100
-#define IDS_ERRORCMD 3101
-#define IDS_HSSTART 3102
-#define IDS_HSSTOP 3103
-#define IDS_HSPAUSE 3104
-#define IDS_HSRESTART 3105
-#define IDS_VALIDPASS 3106
-#define IDS_PPGENERAL 3107
-#define IDS_PPLOGON 3108
-#define IDS_PPLOGGING 3109
-#define IDS_PPJAVAVM 3110
-#define IDS_PPSTART 3111
-#define IDS_PPSTOP 3112
-#define IDS_LGPATHTITLE 3113
-#define IDS_ALLFILES 3114
-#define IDS_DLLFILES 3115
-#define IDS_EXEFILES 3116
-#define IDS_LGSTDERR 3117
-#define IDS_LGSTDOUT 3118
-#define IDS_PPJBJVM 3119
-#define IDS_PPWPATH 3120
-#define IDS_PPIMAGE 3121
-#define IDS_ERRSREG 3122
-
-#define IDS_NOTIMPLEMENTED 3199
-
-#endif /* _PRUNMGR_H */
Copied: tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.h (from rev 2689, trunk/utils/windows/native/service/procrun/apps/prunmgr/prunmgr.h)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.h (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.h 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,141 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/* ====================================================================
+ *
+ * Contributed by Mladen Turk <mturk at apache.org>
+ * 05 Aug 2003
+ * ====================================================================
+ */
+
+#ifndef _PRUNMGR_H
+#define _PRUNMGR_H
+
+#undef PRG_VERSION
+#define PRG_VERSION "2.0.5.1"
+#define CSV_VERSION 2,0,5,1
+#define PRG_REGROOT L"JBoss\\JBoss Web 2"
+
+#define IDM_TM_EXIT 2000
+#define IDM_TM_START 2001
+#define IDM_TM_STOP 2002
+#define IDM_TM_PAUSE 2003
+#define IDM_TM_RESTART 2004
+#define IDM_TM_CONFIG 2005
+#define IDM_TM_ABOUT 2006
+#define IDM_TM_DUMP 2007
+
+#define IDMS_REFRESH 2020
+
+#define IDI_ICONSTOP 2030
+#define IDI_ICONRUN 2031
+
+
+
+/* Property pages */
+
+#define IDD_PROPPAGE_SGENERAL 2600
+#define IDC_PPSGNAME 2601
+#define IDC_PPSGDISP 2602
+#define IDC_PPSGDESC 2603
+#define IDC_PPSGDEXE 2604
+#define IDC_PPSGCMBST 2605
+#define IDC_PPSGSTATUS 2606
+#define IDC_PPSGSTART 2607
+#define IDC_PPSGSTOP 2608
+#define IDC_PPSGPAUSE 2609
+#define IDC_PPSGRESTART 2610
+
+#define IDD_PROPPAGE_LOGON 2620
+#define IDC_PPSLLS 2621
+#define IDC_PPSLID 2622
+#define IDC_PPSLUA 2623
+#define IDC_PPSLUSER 2624
+#define IDC_PPSLBROWSE 2625
+#define IDC_PPSLPASS 2626
+#define IDC_PPSLCPASS 2627
+#define IDL_PPSLPASS 2628
+#define IDL_PPSLCPASS 2629
+
+#define IDD_PROPPAGE_LOGGING 2640
+#define IDC_PPLGLEVEL 2641
+#define IDC_PPLGPATH 2642
+#define IDC_PPLGBPATH 2643
+#define IDC_PPLGPREFIX 2644
+#define IDC_PPLGSTDOUT 2645
+#define IDC_PPLGBSTDOUT 2646
+#define IDC_PPLGSTDERR 2647
+#define IDC_PPLGBSTDERR 2648
+
+#define IDD_PROPPAGE_JVM 2660
+#define IDC_PPJAUTO 2661
+#define IDC_PPJJVM 2662
+#define IDC_PPJBJVM 2663
+#define IDC_PPJCLASSPATH 2664
+#define IDC_PPJOPTIONS 2665
+#define IDC_PPJMS 2666
+#define IDC_PPJMX 2667
+#define IDC_PPJSS 2668
+
+#define IDD_PROPPAGE_START 2680
+#define IDC_PPRCLASS 2681
+#define IDC_PPRIMAGE 2682
+#define IDC_PPRBIMAGE 2683
+#define IDC_PPRWPATH 2684
+#define IDC_PPRBWPATH 2685
+#define IDC_PPRMETHOD 2686
+#define IDC_PPRARGS 2687
+#define IDC_PPRTIMEOUT 2688
+#define IDC_PPRMODE 2689
+
+#define IDD_PROPPAGE_STOP 2700
+#define IDC_PPSCLASS 2701
+#define IDC_PPSIMAGE 2702
+#define IDC_PPSBIMAGE 2703
+#define IDC_PPSWPATH 2704
+#define IDC_PPSBWPATH 2705
+#define IDC_PPSMETHOD 2706
+#define IDC_PPSARGS 2707
+#define IDC_PPSTIMEOUT 2708
+#define IDC_PPSMODE 2709
+
+#define IDS_ALREAY_RUNING 3100
+#define IDS_ERRORCMD 3101
+#define IDS_HSSTART 3102
+#define IDS_HSSTOP 3103
+#define IDS_HSPAUSE 3104
+#define IDS_HSRESTART 3105
+#define IDS_VALIDPASS 3106
+#define IDS_PPGENERAL 3107
+#define IDS_PPLOGON 3108
+#define IDS_PPLOGGING 3109
+#define IDS_PPJAVAVM 3110
+#define IDS_PPSTART 3111
+#define IDS_PPSTOP 3112
+#define IDS_LGPATHTITLE 3113
+#define IDS_ALLFILES 3114
+#define IDS_DLLFILES 3115
+#define IDS_EXEFILES 3116
+#define IDS_LGSTDERR 3117
+#define IDS_LGSTDOUT 3118
+#define IDS_PPJBJVM 3119
+#define IDS_PPWPATH 3120
+#define IDS_PPIMAGE 3121
+#define IDS_ERRSREG 3122
+
+#define IDS_NOTIMPLEMENTED 3199
+
+#endif /* _PRUNMGR_H */
Deleted: tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.rc
===================================================================
--- trunk/utils/windows/native/service/procrun/apps/prunmgr/prunmgr.rc 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.rc 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,296 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#include "apxwin.h"
-#include "prunmgr.h"
-
-#define RSTR_SCMATS "Service Manager is attempting to "
-
-#define STR_COPYRIGHT "Copyright © 2008 Red Hat Middleware, LLC. " \
- "or its licensors, as applicable."
-
-#define STR_LICENSE "Distributable under LGPL license. " \
- "See terms of license at gnu.org."
-
-#define STR_COMPANY "Red Hat®, Inc."
-#define STR_TRADEMARK "™ Red Hat Inc."
-#define STR_PRODUCT "JBoss Web Service manager"
-
-IDI_MAINICON ICON "../../resources/jboss.ico"
-IDI_REDHATICON ICON "../../resources/redhat.ico"
-IDI_ICONSTOP ICON "../../resources/procruns.ico"
-IDI_ICONRUN ICON "../../resources/procrunr.ico"
-IDR_LICENSE RTF "../../resources/license.rtf"
-BMP_JBOSSLOGO BITMAP "../../resources/jbosslogo.bmp"
-
-CREATEPROCESS_MANIFEST_RESOURCE_ID RT_MANIFEST "prunmgr.manifest"
-
-IDD_ABOUTBOX DIALOGEX 0, 0, 337, 187
-STYLE DS_SETFONT | DS_MODALFRAME | WS_CAPTION | WS_SYSMENU
-CAPTION "JBoss Web Service Manager"
-FONT 8, "Microsoft Sans Serif", 400, 0, 0x0
-BEGIN
- DEFPUSHBUTTON "&OK",IDOK,285,150,50,14
- CONTROL "",IDC_LICENSE,"RichEdit20A",ES_MULTILINE |
- ES_READONLY | WS_BORDER | WS_VSCROLL,0,31,335,115
- CONTROL "BMP_JBOSSLOGO",IDC_STATIC,"Static",SS_BITMAP|0x00000040L,0,0,337,30
- LTEXT " ",IDC_ABOUTAPP,2,150,270,12
- LTEXT "Copyright © 2008 Red Hat Middleware, LLC.",IDC_STATIC,2,160,270,12
- LTEXT "http://www.jboss.org",IDC_STATIC,2,170,270,12
- PUSHBUTTON "&System Info",IAB_SYSINF,285,170,50,14
-END
-
-IDD_PROGRESS DIALOGEX 0, 0, 322, 92
-STYLE DS_SYSMODAL | DS_SETFONT | DS_MODALFRAME | DS_FIXEDSYS | WS_POPUP |
- WS_CAPTION | WS_SYSMENU
-EXSTYLE WS_EX_TOPMOST
-CAPTION "JBoss Web Service Manager"
-FONT 8, "MS Shell Dlg", 400, 0, 0x1
-BEGIN
- DEFPUSHBUTTON "Cl&ose",IDOK,261,73,50,14
- PUSHBUTTON "&Cancel",IDCANCEL,202,73,50,14
- LTEXT " ",
- IDDP_HEAD,40,4,250,8
- LTEXT " ",IDDP_TEXT,40,15,250,44
- CONTROL "",IDDP_PROGRESS,"msctls_progress32",WS_BORDER,10,50,302,
- 14
- ICON IDI_MAINICON,IDC_STATIC,10,4,20,20,0,WS_EX_TRANSPARENT
-END
-
-IDD_SELUSER DIALOGEX 0, 0, 410, 201
-STYLE DS_SETFONT | DS_FIXEDSYS | WS_POPUP | WS_CAPTION | WS_SYSMENU |
- WS_THICKFRAME
-EXSTYLE WS_EX_CONTEXTHELP
-CAPTION "Select User"
-FONT 8, "MS Shell Dlg", 400, 0, 0x1
-BEGIN
- LTEXT "Name:",IDSU_SELNAME,10,183,22,8
- EDITTEXT IDSU_SELECTED,48,200,213,14,ES_AUTOHSCROLL
- DEFPUSHBUTTON "OK",IDOK,292,180,50,14,WS_DISABLED
- PUSHBUTTON "Cancel",IDCANCEL,348,180,50,14
- LTEXT "Look In:",IDC_STATIC,10,9,27,8
- CONTROL "",IDSU_LIST,"SysListView32",LVS_REPORT | LVS_SINGLESEL |
- LVS_SORTASCENDING | LVS_SHAREIMAGELISTS | LVS_ALIGNLEFT | WS_BORDER |
- WS_TABSTOP,0,26,410,149,WS_EX_CLIENTEDGE
- CONTROL "",IDSU_COMBO,"ComboBoxEx32",CBS_DROPDOWNLIST |
- WS_VSCROLL | WS_TABSTOP,47,6,260,80
-END
-
-IDD_PROPPAGE_SGENERAL DIALOGEX 0, 0, 260, 186
-STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD
-FONT 8, "MS Shell Dlg", 0, 0, 0x0
-BEGIN
- LTEXT "Service Name: ",IDC_STATIC,10,8,52,8
- LTEXT " ",IDC_PPSGNAME,70,8,240,8
- LTEXT "Display &name: ",IDC_STATIC,10,23,50,8
- EDITTEXT IDC_PPSGDISP,70,22,180,12,ES_AUTOHSCROLL
- LTEXT "&Description: ",IDC_STATIC,10,41,43,8
- EDITTEXT IDC_PPSGDESC,70,40,180,12,ES_AUTOHSCROLL
- LTEXT "Pat&h to executable:",IDC_STATIC,10,63,66,8
- EDITTEXT IDC_PPSGDEXE,10,75,240,12,ES_AUTOHSCROLL | WS_DISABLED
- LTEXT "Startup typ&e:",IDC_STATIC,10,94,46,8
- COMBOBOX IDC_PPSGCMBST,70,93,180,80,CBS_DROPDOWNLIST | WS_VSCROLL |
- WS_TABSTOP
- CONTROL "",IDC_STATIC,"Static",SS_ETCHEDHORZ,10,128,240,1
- LTEXT "Service Status:",IDC_STATIC,10,138,52,8
- LTEXT " ",IDC_PPSGSTATUS,70,138,240,8
- PUSHBUTTON "&Start",IDC_PPSGSTART,10,160,55,14,WS_DISABLED
- PUSHBUTTON "S&top",IDC_PPSGSTOP,71,160,55,14,WS_DISABLED
- PUSHBUTTON "&Pause",IDC_PPSGPAUSE,132,160,55,14,WS_DISABLED
- PUSHBUTTON "&Restart",IDC_PPSGRESTART,193,160,55,14,WS_DISABLED
-END
-
-IDD_PROPPAGE_LOGON DIALOGEX 0, 0, 260, 186
-STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD
-FONT 8, "MS Shell Dlg", 0, 0, 0x0
-BEGIN
- LTEXT "Log on as:",IDC_STATIC,10,8,51,8
- CONTROL "&Local System account",IDC_PPSLLS,"Button",
- BS_AUTORADIOBUTTON,10,23,85,10
- CONTROL "Allo&w service to interact with desktop",IDC_PPSLID,
- "Button",BS_AUTOCHECKBOX | WS_TABSTOP,21,37,135,10
- CONTROL "&This account:",IDC_PPSLUA,"Button",BS_AUTORADIOBUTTON,
- 10,59,61,10
- EDITTEXT IDC_PPSLUSER,86,58,114,12,ES_AUTOHSCROLL
- PUSHBUTTON "&Browse...",IDC_PPSLBROWSE,205,57,50,14
- LTEXT "&Password:",IDL_PPSLPASS,21,76,36,8
- EDITTEXT IDC_PPSLPASS,86,75,114,12,ES_PASSWORD | ES_AUTOHSCROLL
- LTEXT "&Confirm Password:",IDL_PPSLCPASS,20,93,63,8
- EDITTEXT IDC_PPSLCPASS,86,92,114,12,ES_PASSWORD | ES_AUTOHSCROLL
-END
-
-IDD_PROPPAGE_LOGGING DIALOGEX 0, 0, 260, 186
-STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD
-FONT 8, "MS Shell Dlg", 0, 0, 0x0
-BEGIN
- LTEXT "&Level:",IDC_STATIC,10,18,46,8
- COMBOBOX IDC_PPLGLEVEL,70,17,180,80,CBS_DROPDOWNLIST | WS_VSCROLL |
- WS_TABSTOP
- LTEXT "Log &path: ",IDC_STATIC,10,33,50,8
- EDITTEXT IDC_PPLGPATH,10,45,218,12,ES_AUTOHSCROLL
- PUSHBUTTON "...",IDC_PPLGBPATH,232,44,18,14
- LTEXT "Log prefix: ",IDC_STATIC,10,63,50,8
- EDITTEXT IDC_PPLGPREFIX,70,62,158,12,ES_AUTOHSCROLL
-
- LTEXT "Redirect Stdout: ",IDC_STATIC,10,85,80,8
- EDITTEXT IDC_PPLGSTDOUT,10,97,218,12,ES_AUTOHSCROLL
- PUSHBUTTON "...",IDC_PPLGBSTDOUT,232,96,18,14
- LTEXT "Redirect Stderror: ",IDC_STATIC,10,115,80,8
- EDITTEXT IDC_PPLGSTDERR,10,127,218,12,ES_AUTOHSCROLL
- PUSHBUTTON "...",IDC_PPLGBSTDERR,232,126,18,14
-END
-
-IDD_PROPPAGE_JVM DIALOGEX 0, 0, 260, 186
-STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD
-FONT 8, "MS Shell Dlg", 0, 0, 0x0
-BEGIN
- CONTROL " Use default",IDC_PPJAUTO,"Button",BS_AUTOCHECKBOX |
- WS_TABSTOP,10,8,72,12
- LTEXT "Java Virtual Machine: ",IDC_STATIC,10,23,80,8
- EDITTEXT IDC_PPJJVM,10,35,218,12,ES_AUTOHSCROLL
- PUSHBUTTON "...",IDC_PPJBJVM,232,34,18,14
- LTEXT "Java Classpath:",IDC_STATIC,10,53,66,8
- EDITTEXT IDC_PPJCLASSPATH,10,65,240,12,ES_AUTOHSCROLL
- LTEXT "Java Options:",IDC_STATIC,10,83,66,8
- EDITTEXT IDC_PPJOPTIONS,10,95,240,40,ES_MULTILINE | ES_AUTOHSCROLL |
- ES_WANTRETURN | WS_VSCROLL
- LTEXT "Initial memory pool:",IDC_STATIC,10,140,80,8
- EDITTEXT IDC_PPJMS,90,139,100,12,ES_AUTOHSCROLL | ES_NUMBER
- LTEXT "MB",IDC_STATIC,195,140,25,8
- LTEXT "Maximum memory pool:",IDC_STATIC,10,156,80,8
- EDITTEXT IDC_PPJMX,90,155,100,12,ES_AUTOHSCROLL | ES_NUMBER
- LTEXT "MB",IDC_STATIC,195,156,25,8
- LTEXT "Thread stack size:",IDC_STATIC,10,172,80,8
- EDITTEXT IDC_PPJSS,90,171,100,12,ES_AUTOHSCROLL | ES_NUMBER
- LTEXT "KB",IDC_STATIC,195,172,25,8
-END
-
-IDD_PROPPAGE_START DIALOGEX 0, 0, 260, 186
-STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD
-FONT 8, "MS Shell Dlg", 0, 0, 0x0
-BEGIN
- LTEXT "&Class: ",IDC_STATIC,10,8,50,8
- EDITTEXT IDC_PPRCLASS,10,20,240,12,ES_AUTOHSCROLL
- LTEXT "Image: ",IDC_STATIC,10,38,80,8
- EDITTEXT IDC_PPRIMAGE,10,50,218,12,ES_AUTOHSCROLL | WS_DISABLED
- PUSHBUTTON "...",IDC_PPRBIMAGE,232,49,18,14,WS_DISABLED
- LTEXT "Working Path: ",IDC_STATIC,10,68,80,8
- EDITTEXT IDC_PPRWPATH,10,80,218,12,ES_AUTOHSCROLL
- PUSHBUTTON "...",IDC_PPRBWPATH,232,79,18,14
- LTEXT "&Method: ",IDC_STATIC,10,98,50,8
- EDITTEXT IDC_PPRMETHOD,70,97,180,12,ES_AUTOHSCROLL
- LTEXT "&Arguments: ",IDC_STATIC,10,115,50,8
- EDITTEXT IDC_PPRARGS,70,114,180,30,ES_MULTILINE | ES_AUTOHSCROLL |
- ES_WANTRETURN | WS_VSCROLL
- LTEXT "&Timeout: ",IDC_STATIC,10,150,50,8
- EDITTEXT IDC_PPRTIMEOUT,70,149,100,12,ES_AUTOHSCROLL | WS_DISABLED
- LTEXT "sec.",IDC_STATIC,175,150,25,8
- LTEXT "&Mode: ",IDC_STATIC,10,168,50,8
- COMBOBOX IDC_PPRMODE,70,167,180,80,CBS_DROPDOWNLIST | WS_VSCROLL |
- WS_TABSTOP
-END
-
-IDD_PROPPAGE_STOP DIALOGEX 0, 0, 260, 186
-STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD
-FONT 8, "MS Shell Dlg", 0, 0, 0x0
-BEGIN
- LTEXT "&Class: ",IDC_STATIC,10,8,50,8
- EDITTEXT IDC_PPSCLASS,10,20,240,12,ES_AUTOHSCROLL
- LTEXT "Image: ",IDC_STATIC,10,38,80,8
- EDITTEXT IDC_PPSIMAGE,10,50,218,12,ES_AUTOHSCROLL | WS_DISABLED
- PUSHBUTTON "...",IDC_PPSBIMAGE,232,49,18,14,WS_DISABLED
- LTEXT "Working Path: ",IDC_STATIC,10,68,80,8
- EDITTEXT IDC_PPSWPATH,10,80,218,12,ES_AUTOHSCROLL
- PUSHBUTTON "...",IDC_PPSBWPATH,232,79,18,14
- LTEXT "&Method: ",IDC_STATIC,10,98,50,8
- EDITTEXT IDC_PPSMETHOD,70,97,180,12,ES_AUTOHSCROLL
- LTEXT "&Arguments: ",IDC_STATIC,10,115,50,8
- EDITTEXT IDC_PPSARGS,70,114,180,30,ES_MULTILINE | ES_AUTOHSCROLL |
- ES_WANTRETURN | WS_VSCROLL
- LTEXT "&Timeout: ",IDC_STATIC,10,150,50,8
- EDITTEXT IDC_PPSTIMEOUT,70,149,100,12,ES_AUTOHSCROLL | ES_NUMBER
- LTEXT "sec.",IDC_STATIC,175,150,25,8
- LTEXT "&Mode: ",IDC_STATIC,10,168,50,8
- COMBOBOX IDC_PPSMODE,70,167,180,80,CBS_DROPDOWNLIST | WS_VSCROLL |
- WS_TABSTOP
-END
-
-STRINGTABLE
-BEGIN
- IDS_APPLICATION STR_PRODUCT
- IDS_APPVERSION "Version 2.0.4"
- IDS_APPFULLNAME STR_PRODUCT " Version " PRG_VERSION
- IDS_APPCOPYRIGHT "Copyright © 2008 Red Hat Middleware, LLC."
- IDS_APPDESCRIPTION "JBoss Web Service Management Tool"
- IDS_ALREAY_RUNING "An instance of '%S' application is already running"
- IDS_ERRORCMD "Unknown command line option '%s'\nSee the manual for command line usage."
- IDS_HSSTART RSTR_SCMATS "start the following service ..."
- IDS_HSSTOP RSTR_SCMATS "stop the following service ..."
- IDS_HSRESTART RSTR_SCMATS "to restart the following service ..."
- IDS_HSPAUSE RSTR_SCMATS "to pause the following service ..."
- IDS_VALIDPASS "Please enter a valid password"
- IDS_PPGENERAL "General"
- IDS_PPLOGON "Log On"
- IDS_PPLOGGING "Logging"
- IDS_PPJAVAVM "Java"
- IDS_PPSTART "Startup"
- IDS_PPSTOP "Shutdown"
- IDS_NOTIMPLEMENTED "Sorry, but this feature is not implemented yet"
- IDS_LGPATHTITLE "Select Log root folder"
- IDS_ALLFILES "All Files (*.*)\0*.*\0"
- IDS_DLLFILES "Dynamic Link Libraries (*.dll)\0*.dll\0"
- IDS_EXEFILES "Executables (*.exe)\0*.exe\0"
- IDS_LGSTDOUT "Select Stdoutput filename"
- IDS_LGSTDERR "Select Stderror filename"
- IDS_PPJBJVM "Select Java Virtual Machine DLL"
- IDS_PPWPATH "Select Working path"
- IDS_PPIMAGE "Select Executable Image"
- IDS_ERRSREG "Unable to open the service registry key"
-END
-
-1 VERSIONINFO
- FILEVERSION CSV_VERSION
- PRODUCTVERSION CSV_VERSION
- FILEFLAGSMASK 0x3fL
-#if defined(_DEBUG)
- FILEFLAGS 0x03L
-#else
- FILEFLAGS 0x02L
-#endif
- FILEOS 0x40004L
- FILETYPE 0x1L
- FILESUBTYPE 0x0L
-BEGIN
- BLOCK "StringFileInfo"
- BEGIN
- BLOCK "040904b0"
- BEGIN
- VALUE "Comments", STR_COPYRIGHT "\0"
- VALUE "CompanyName", STR_COMPANY "\0"
- VALUE "FileDescription", STR_PRODUCT "\0"
- VALUE "FileVersion", PRG_VERSION
- VALUE "InternalName", STR_PRODUCT "\0"
- VALUE "LegalCopyright", STR_COPYRIGHT "\0"
- VALUE "OriginalFilename", "jbosswebw.exe\0"
- VALUE "ProductName", STR_PRODUCT "\0"
- VALUE "ProductVersion", PRG_VERSION
- END
- END
- BLOCK "VarFileInfo"
- BEGIN
- VALUE "Translation", 0x409, 1200
- END
-END
Copied: tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.rc (from rev 2689, trunk/utils/windows/native/service/procrun/apps/prunmgr/prunmgr.rc)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.rc (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/apps/prunmgr/prunmgr.rc 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,299 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "apxwin.h"
+#include "prunmgr.h"
+
+#define RSTR_SCMATS "Service Manager is attempting to "
+
+#define STR_COPYRIGHT "Copyright � 2010 Red Hat, Inc. " \
+ "or its licensors, as applicable."
+
+#define STR_LICENSE "Distributable under LGPL license. " \
+ "See terms of license at gnu.org."
+
+#define STR_COMPANY "Red Hat�, Inc."
+#define STR_TRADEMARK "� Red Hat Inc."
+#define STR_PRODUCT "JBoss Web Service manager"
+
+IDI_MAINICON ICON "../../resources/jboss.ico"
+IDI_REDHATICON ICON "../../resources/redhat.ico"
+IDI_ICONSTOP ICON "../../resources/procruns.ico"
+IDI_ICONRUN ICON "../../resources/procrunr.ico"
+IDR_LICENSE RTF "../../resources/license.rtf"
+BMP_JBOSSLOGO BITMAP "../../resources/jbosslogo.bmp"
+
+CREATEPROCESS_MANIFEST_RESOURCE_ID RT_MANIFEST "prunmgr.manifest"
+
+IDD_ABOUTBOX DIALOGEX 0, 0, 337, 187
+STYLE DS_SETFONT | DS_MODALFRAME | WS_CAPTION | WS_SYSMENU
+CAPTION "JBoss Web Service Manager"
+FONT 8, "Microsoft Sans Serif", 400, 0, 0x0
+BEGIN
+ DEFPUSHBUTTON "&OK",IDOK,285,150,50,14
+ CONTROL "",IDC_LICENSE,"RichEdit20A",ES_MULTILINE |
+ ES_READONLY | WS_BORDER | WS_VSCROLL,0,31,335,115
+ CONTROL "BMP_JBOSSLOGO",IDC_STATIC,"Static",SS_BITMAP|0x00000040L,0,0,337,30
+ LTEXT " ",IDC_ABOUTAPP,2,150,270,12
+ LTEXT "Copyright � 2008 Red Hat Middleware, LLC.",IDC_STATIC,2,160,270,12
+ LTEXT "http://www.jboss.org",IDC_STATIC,2,170,270,12
+ PUSHBUTTON "&System Info",IAB_SYSINF,285,170,50,14
+END
+
+IDD_PROGRESS DIALOGEX 0, 0, 322, 92
+STYLE DS_SYSMODAL | DS_SETFONT | DS_MODALFRAME | DS_FIXEDSYS | WS_POPUP |
+ WS_CAPTION | WS_SYSMENU
+EXSTYLE WS_EX_TOPMOST
+CAPTION "JBoss Web Service Manager"
+FONT 8, "MS Shell Dlg", 400, 0, 0x1
+BEGIN
+ DEFPUSHBUTTON "Cl&ose",IDOK,261,73,50,14
+ PUSHBUTTON "&Cancel",IDCANCEL,202,73,50,14
+ LTEXT " ",
+ IDDP_HEAD,40,4,250,8
+ LTEXT " ",IDDP_TEXT,40,15,250,44
+ CONTROL "",IDDP_PROGRESS,"msctls_progress32",WS_BORDER,10,50,302,
+ 14
+ ICON IDI_MAINICON,IDC_STATIC,10,4,20,20,0,WS_EX_TRANSPARENT
+END
+
+IDD_SELUSER DIALOGEX 0, 0, 410, 201
+STYLE DS_SETFONT | DS_FIXEDSYS | WS_POPUP | WS_CAPTION | WS_SYSMENU |
+ WS_THICKFRAME
+EXSTYLE WS_EX_CONTEXTHELP
+CAPTION "Select User"
+FONT 8, "MS Shell Dlg", 400, 0, 0x1
+BEGIN
+ LTEXT "Name:",IDSU_SELNAME,10,183,22,8
+ EDITTEXT IDSU_SELECTED,48,200,213,14,ES_AUTOHSCROLL
+ DEFPUSHBUTTON "OK",IDOK,292,180,50,14,WS_DISABLED
+ PUSHBUTTON "Cancel",IDCANCEL,348,180,50,14
+ LTEXT "Look In:",IDC_STATIC,10,9,27,8
+ CONTROL "",IDSU_LIST,"SysListView32",LVS_REPORT | LVS_SINGLESEL |
+ LVS_SORTASCENDING | LVS_SHAREIMAGELISTS | LVS_ALIGNLEFT | WS_BORDER |
+ WS_TABSTOP,0,26,410,149,WS_EX_CLIENTEDGE
+ CONTROL "",IDSU_COMBO,"ComboBoxEx32",CBS_DROPDOWNLIST |
+ WS_VSCROLL | WS_TABSTOP,47,6,260,80
+END
+
+IDD_PROPPAGE_SGENERAL DIALOGEX 0, 0, 260, 186
+STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD
+FONT 8, "MS Shell Dlg", 0, 0, 0x0
+BEGIN
+ LTEXT "Service Name: ",IDC_STATIC,10,8,52,8
+ LTEXT " ",IDC_PPSGNAME,70,8,240,8
+ LTEXT "Display &name: ",IDC_STATIC,10,23,50,8
+ EDITTEXT IDC_PPSGDISP,70,22,180,12,ES_AUTOHSCROLL
+ LTEXT "&Description: ",IDC_STATIC,10,41,43,8
+ EDITTEXT IDC_PPSGDESC,70,40,180,12,ES_AUTOHSCROLL
+ LTEXT "Pat&h to executable:",IDC_STATIC,10,63,66,8
+ EDITTEXT IDC_PPSGDEXE,10,75,240,12,ES_AUTOHSCROLL | WS_DISABLED
+ LTEXT "Startup typ&e:",IDC_STATIC,10,94,46,8
+ COMBOBOX IDC_PPSGCMBST,70,93,180,80,CBS_DROPDOWNLIST | WS_VSCROLL |
+ WS_TABSTOP
+ CONTROL "",IDC_STATIC,"Static",SS_ETCHEDHORZ,10,128,240,1
+ LTEXT "Service Status:",IDC_STATIC,10,138,52,8
+ LTEXT " ",IDC_PPSGSTATUS,70,138,240,8
+ PUSHBUTTON "&Start",IDC_PPSGSTART,10,160,55,14,WS_DISABLED
+ PUSHBUTTON "S&top",IDC_PPSGSTOP,71,160,55,14,WS_DISABLED
+ PUSHBUTTON "&Pause",IDC_PPSGPAUSE,132,160,55,14,WS_DISABLED
+ PUSHBUTTON "&Restart",IDC_PPSGRESTART,193,160,55,14,WS_DISABLED
+END
+
+IDD_PROPPAGE_LOGON DIALOGEX 0, 0, 260, 186
+STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD
+FONT 8, "MS Shell Dlg", 0, 0, 0x0
+BEGIN
+ LTEXT "Log on as:",IDC_STATIC,10,8,51,8
+ CONTROL "&Local System account",IDC_PPSLLS,"Button",
+ BS_AUTORADIOBUTTON,10,23,85,10
+ CONTROL "Allo&w service to interact with desktop",IDC_PPSLID,
+ "Button",BS_AUTOCHECKBOX | WS_TABSTOP,21,37,135,10
+ CONTROL "&This account:",IDC_PPSLUA,"Button",BS_AUTORADIOBUTTON,
+ 10,59,61,10
+ EDITTEXT IDC_PPSLUSER,86,58,114,12,ES_AUTOHSCROLL
+ PUSHBUTTON "&Browse...",IDC_PPSLBROWSE,205,57,50,14
+ LTEXT "&Password:",IDL_PPSLPASS,21,76,36,8
+ EDITTEXT IDC_PPSLPASS,86,75,114,12,ES_PASSWORD | ES_AUTOHSCROLL
+ LTEXT "&Confirm Password:",IDL_PPSLCPASS,20,93,63,8
+ EDITTEXT IDC_PPSLCPASS,86,92,114,12,ES_PASSWORD | ES_AUTOHSCROLL
+END
+
+IDD_PROPPAGE_LOGGING DIALOGEX 0, 0, 260, 186
+STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD
+FONT 8, "MS Shell Dlg", 0, 0, 0x0
+BEGIN
+ LTEXT "&Level:",IDC_STATIC,10,18,46,8
+ COMBOBOX IDC_PPLGLEVEL,70,17,180,80,CBS_DROPDOWNLIST | WS_VSCROLL |
+ WS_TABSTOP
+ LTEXT "Log &path: ",IDC_STATIC,10,33,50,8
+ EDITTEXT IDC_PPLGPATH,10,45,218,12,ES_AUTOHSCROLL
+ PUSHBUTTON "...",IDC_PPLGBPATH,232,44,18,14
+ LTEXT "Log prefix: ",IDC_STATIC,10,63,50,8
+ EDITTEXT IDC_PPLGPREFIX,70,62,158,12,ES_AUTOHSCROLL
+
+ LTEXT "Pid file: ",IDC_STATIC,10,81,50,8
+ EDITTEXT IDC_PPLGPIDFILE,70,82,158,12,ES_AUTOHSCROLL
+
+ LTEXT "Redirect Stdout: ",IDC_STATIC,10,105,80,8
+ EDITTEXT IDC_PPLGSTDOUT,10,117,218,12,ES_AUTOHSCROLL
+ PUSHBUTTON "...",IDC_PPLGBSTDOUT,232,116,18,14
+ LTEXT "Redirect Stderror: ",IDC_STATIC,10,135,80,8
+ EDITTEXT IDC_PPLGSTDERR,10,147,218,12,ES_AUTOHSCROLL
+ PUSHBUTTON "...",IDC_PPLGBSTDERR,232,146,18,14
+END
+
+IDD_PROPPAGE_JVM DIALOGEX 0, 0, 260, 186
+STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD
+FONT 8, "MS Shell Dlg", 0, 0, 0x0
+BEGIN
+ CONTROL " Use default",IDC_PPJAUTO,"Button",BS_AUTOCHECKBOX |
+ WS_TABSTOP,10,8,72,12
+ LTEXT "Java Virtual Machine: ",IDC_STATIC,10,23,80,8
+ EDITTEXT IDC_PPJJVM,10,35,218,12,ES_AUTOHSCROLL
+ PUSHBUTTON "...",IDC_PPJBJVM,232,34,18,14
+ LTEXT "Java Classpath:",IDC_STATIC,10,53,66,8
+ EDITTEXT IDC_PPJCLASSPATH,10,65,240,12,ES_AUTOHSCROLL
+ LTEXT "Java Options:",IDC_STATIC,10,83,66,8
+ EDITTEXT IDC_PPJOPTIONS,10,95,240,40,ES_MULTILINE | ES_AUTOHSCROLL |
+ ES_WANTRETURN | WS_VSCROLL
+ LTEXT "Initial memory pool:",IDC_STATIC,10,140,80,8
+ EDITTEXT IDC_PPJMS,90,139,100,12,ES_AUTOHSCROLL | ES_NUMBER
+ LTEXT "MB",IDC_STATIC,195,140,25,8
+ LTEXT "Maximum memory pool:",IDC_STATIC,10,156,80,8
+ EDITTEXT IDC_PPJMX,90,155,100,12,ES_AUTOHSCROLL | ES_NUMBER
+ LTEXT "MB",IDC_STATIC,195,156,25,8
+ LTEXT "Thread stack size:",IDC_STATIC,10,172,80,8
+ EDITTEXT IDC_PPJSS,90,171,100,12,ES_AUTOHSCROLL | ES_NUMBER
+ LTEXT "KB",IDC_STATIC,195,172,25,8
+END
+
+IDD_PROPPAGE_START DIALOGEX 0, 0, 260, 186
+STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD
+FONT 8, "MS Shell Dlg", 0, 0, 0x0
+BEGIN
+ LTEXT "&Class: ",IDC_STATIC,10,8,50,8
+ EDITTEXT IDC_PPRCLASS,10,20,240,12,ES_AUTOHSCROLL
+ LTEXT "Image: ",IDC_STATIC,10,38,80,8
+ EDITTEXT IDC_PPRIMAGE,10,50,218,12,ES_AUTOHSCROLL | WS_DISABLED
+ PUSHBUTTON "...",IDC_PPRBIMAGE,232,49,18,14,WS_DISABLED
+ LTEXT "Working Path: ",IDC_STATIC,10,68,80,8
+ EDITTEXT IDC_PPRWPATH,10,80,218,12,ES_AUTOHSCROLL
+ PUSHBUTTON "...",IDC_PPRBWPATH,232,79,18,14
+ LTEXT "&Method: ",IDC_STATIC,10,98,50,8
+ EDITTEXT IDC_PPRMETHOD,70,97,180,12,ES_AUTOHSCROLL
+ LTEXT "&Arguments: ",IDC_STATIC,10,115,50,8
+ EDITTEXT IDC_PPRARGS,70,114,180,30,ES_MULTILINE | ES_AUTOHSCROLL |
+ ES_WANTRETURN | WS_VSCROLL
+ LTEXT "&Timeout: ",IDC_STATIC,10,150,50,8
+ EDITTEXT IDC_PPRTIMEOUT,70,149,100,12,ES_AUTOHSCROLL | WS_DISABLED
+ LTEXT "sec.",IDC_STATIC,175,150,25,8
+ LTEXT "&Mode: ",IDC_STATIC,10,168,50,8
+ COMBOBOX IDC_PPRMODE,70,167,180,80,CBS_DROPDOWNLIST | WS_VSCROLL |
+ WS_TABSTOP
+END
+
+IDD_PROPPAGE_STOP DIALOGEX 0, 0, 260, 186
+STYLE DS_SETFONT | DS_FIXEDSYS | WS_CHILD
+FONT 8, "MS Shell Dlg", 0, 0, 0x0
+BEGIN
+ LTEXT "&Class: ",IDC_STATIC,10,8,50,8
+ EDITTEXT IDC_PPSCLASS,10,20,240,12,ES_AUTOHSCROLL
+ LTEXT "Image: ",IDC_STATIC,10,38,80,8
+ EDITTEXT IDC_PPSIMAGE,10,50,218,12,ES_AUTOHSCROLL | WS_DISABLED
+ PUSHBUTTON "...",IDC_PPSBIMAGE,232,49,18,14,WS_DISABLED
+ LTEXT "Working Path: ",IDC_STATIC,10,68,80,8
+ EDITTEXT IDC_PPSWPATH,10,80,218,12,ES_AUTOHSCROLL
+ PUSHBUTTON "...",IDC_PPSBWPATH,232,79,18,14
+ LTEXT "&Method: ",IDC_STATIC,10,98,50,8
+ EDITTEXT IDC_PPSMETHOD,70,97,180,12,ES_AUTOHSCROLL
+ LTEXT "&Arguments: ",IDC_STATIC,10,115,50,8
+ EDITTEXT IDC_PPSARGS,70,114,180,30,ES_MULTILINE | ES_AUTOHSCROLL |
+ ES_WANTRETURN | WS_VSCROLL
+ LTEXT "&Timeout: ",IDC_STATIC,10,150,50,8
+ EDITTEXT IDC_PPSTIMEOUT,70,149,100,12,ES_AUTOHSCROLL | ES_NUMBER
+ LTEXT "sec.",IDC_STATIC,175,150,25,8
+ LTEXT "&Mode: ",IDC_STATIC,10,168,50,8
+ COMBOBOX IDC_PPSMODE,70,167,180,80,CBS_DROPDOWNLIST | WS_VSCROLL |
+ WS_TABSTOP
+END
+
+STRINGTABLE
+BEGIN
+ IDS_APPLICATION STR_PRODUCT
+ IDS_APPVERSION "Version 2.0.5"
+ IDS_APPFULLNAME STR_PRODUCT " Version " PRG_VERSION
+ IDS_APPCOPYRIGHT "Copyright � 2010 Red Hat, Inc."
+ IDS_APPDESCRIPTION "JBoss Web Service Management Tool"
+ IDS_ALREAY_RUNING "An instance of '%S' application is already running"
+ IDS_ERRORCMD "Unknown command line option '%s'\nSee the manual for command line usage."
+ IDS_HSSTART RSTR_SCMATS "start the following service ..."
+ IDS_HSSTOP RSTR_SCMATS "stop the following service ..."
+ IDS_HSRESTART RSTR_SCMATS "to restart the following service ..."
+ IDS_HSPAUSE RSTR_SCMATS "to pause the following service ..."
+ IDS_VALIDPASS "Please enter a valid password"
+ IDS_PPGENERAL "General"
+ IDS_PPLOGON "Log On"
+ IDS_PPLOGGING "Logging"
+ IDS_PPJAVAVM "Java"
+ IDS_PPSTART "Startup"
+ IDS_PPSTOP "Shutdown"
+ IDS_NOTIMPLEMENTED "Sorry, but this feature is not implemented yet"
+ IDS_LGPATHTITLE "Select Log root folder"
+ IDS_ALLFILES "All Files (*.*)\0*.*\0"
+ IDS_DLLFILES "Dynamic Link Libraries (*.dll)\0*.dll\0"
+ IDS_EXEFILES "Executables (*.exe)\0*.exe\0"
+ IDS_LGSTDOUT "Select Stdoutput filename"
+ IDS_LGSTDERR "Select Stderror filename"
+ IDS_PPJBJVM "Select Java Virtual Machine DLL"
+ IDS_PPWPATH "Select Working path"
+ IDS_PPIMAGE "Select Executable Image"
+ IDS_ERRSREG "Unable to open the service registry key"
+END
+
+1 VERSIONINFO
+ FILEVERSION CSV_VERSION
+ PRODUCTVERSION CSV_VERSION
+ FILEFLAGSMASK 0x3fL
+#if defined(_DEBUG)
+ FILEFLAGS 0x03L
+#else
+ FILEFLAGS 0x02L
+#endif
+ FILEOS 0x40004L
+ FILETYPE 0x1L
+ FILESUBTYPE 0x0L
+BEGIN
+ BLOCK "StringFileInfo"
+ BEGIN
+ BLOCK "040904b0"
+ BEGIN
+ VALUE "Comments", STR_COPYRIGHT "\0"
+ VALUE "CompanyName", STR_COMPANY "\0"
+ VALUE "FileDescription", STR_PRODUCT "\0"
+ VALUE "FileVersion", PRG_VERSION
+ VALUE "InternalName", STR_PRODUCT "\0"
+ VALUE "LegalCopyright", STR_COPYRIGHT "\0"
+ VALUE "OriginalFilename", "jbosswebw.exe\0"
+ VALUE "ProductName", STR_PRODUCT "\0"
+ VALUE "ProductVersion", PRG_VERSION
+ END
+ END
+ BLOCK "VarFileInfo"
+ BEGIN
+ VALUE "Translation", 0x409, 1200
+ END
+END
Deleted: tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.c
===================================================================
--- trunk/utils/windows/native/service/procrun/apps/prunsrv/prunsrv.c 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,1399 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-/* ====================================================================
- * prunsrv -- Service Runner.
- * Contributed by Mladen Turk <mturk at apache.org>
- * 05 Aug 2003
- * ====================================================================
- */
-
-/* Force the JNI vprintf functions */
-#define _DEBUG_JNI 1
-#include "apxwin.h"
-#include "prunsrv.h"
-
-#include <stdio.h>
-#include <stdlib.h>
-#include <stddef.h>
-#include <fcntl.h>
-#include <io.h> /* _open_osfhandle */
-
-#ifndef MIN
-#define MIN(a,b) (((a)<(b)) ? (a) : (b))
-#endif
-
-#define STDIN_FILENO 0
-#define STDOUT_FILENO 1
-#define STDERR_FILENO 2
-
-#ifdef WIN64
-#define KREG_WOW6432 KEY_WOW64_32KEY
-#else
-#define KREG_WOW6432 0
-#endif
-
-typedef struct APX_STDWRAP {
- LPCWSTR szLogPath;
- LPCWSTR szStdOutFilename;
- LPCWSTR szStdErrFilename;
- HANDLE hStdOutFile;
- HANDLE hStdErrFile;
- FILE *fpStdOutFile;
- FILE *fpStdErrFile;
- FILE fpStdOutSave;
- FILE fpStdErrSave;
-} APX_STDWRAP;
-
-/* Use static variables instead of #defines */
-static LPCWSTR PRSRV_AUTO = L"auto";
-static LPCWSTR PRSRV_JAVA = L"java";
-static LPCWSTR PRSRV_JVM = L"jvm";
-static LPCWSTR PRSRV_MANUAL = L"manual";
-static LPCWSTR PRSRV_JBIN = L"\\bin\\java.exe";
-static LPCWSTR PRSRV_SIGNAL = L"SIGNAL";
-
-static LPCWSTR STYPE_INTERACTIVE = L"interactive";
-
-static LPWSTR _service_name = NULL;
-/* Allowed procrun commands */
-static LPCWSTR _commands[] = {
- L"TS", /* 1 Run Service as console application (default)*/
- L"RS", /* 2 Run Service */
- L"SS", /* 3 Stop Service */
- L"US", /* 4 Update Service parameters */
- L"IS", /* 5 Install Service */
- L"DS", /* 6 Delete Service */
- NULL
-};
-
-/* Allowed procrun parameters */
-static APXCMDLINEOPT _options[] = {
-
-/* 0 */ { L"Description", L"Description", NULL, APXCMDOPT_STR | APXCMDOPT_SRV, NULL, 0},
-/* 1 */ { L"DisplayName", L"DisplayName", NULL, APXCMDOPT_STR | APXCMDOPT_SRV, NULL, 0},
-/* 2 */ { L"Install", L"ImagePath", NULL, APXCMDOPT_STE | APXCMDOPT_SRV, NULL, 0},
-/* 3 */ { L"ServiceUser", L"ServiceUser", NULL, APXCMDOPT_STR | APXCMDOPT_SRV, NULL, 0},
-/* 4 */ { L"ServicePassword", L"ServicePassword", NULL, APXCMDOPT_STR | APXCMDOPT_SRV, NULL, 0},
-/* 5 */ { L"Startup", L"Startup", NULL, APXCMDOPT_STR | APXCMDOPT_SRV, NULL, 0},
-/* 6 */ { L"Type", L"Type", NULL, APXCMDOPT_STR | APXCMDOPT_SRV, NULL, 0},
-
-/* 7 */ { L"DependsOn", L"DependsOn", NULL, APXCMDOPT_MSZ | APXCMDOPT_REG, NULL, 0},
-/* 8 */ { L"Environment", L"Environment", NULL, APXCMDOPT_MSZ | APXCMDOPT_REG, NULL, 0},
-/* 9 */ { L"User", L"User", NULL, APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
-/* 10 */ { L"Password", L"Password", NULL, APXCMDOPT_BIN | APXCMDOPT_REG, NULL, 0},
-
-/* 11 */ { L"JavaHome", L"JavaHome", L"Java", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
-/* 12 */ { L"Jvm", L"Jvm", L"Java", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
-/* 13 */ { L"JvmOptions", L"Options", L"Java", APXCMDOPT_MSZ | APXCMDOPT_REG, NULL, 0},
-/* 14 */ { L"Classpath", L"Classpath", L"Java", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
-/* 15 */ { L"JvmMs", L"JvmMs", L"Java", APXCMDOPT_INT | APXCMDOPT_REG, NULL, 0},
-/* 16 */ { L"JvmMx", L"JvmMx", L"Java", APXCMDOPT_INT | APXCMDOPT_REG, NULL, 0},
-/* 17 */ { L"JvmSs", L"JvmSs", L"Java", APXCMDOPT_INT | APXCMDOPT_REG, NULL, 0},
-
-/* 18 */ { L"StopImage", L"Image", L"Stop", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
-/* 19 */ { L"StopPath", L"WorkingPath", L"Stop", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
-/* 20 */ { L"StopClass", L"Class", L"Stop", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
-/* 21 */ { L"StopParams", L"Params", L"Stop", APXCMDOPT_MSZ | APXCMDOPT_REG, NULL, 0},
-/* 22 */ { L"StopMethod", L"Method", L"Stop", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
-/* 23 */ { L"StopMode", L"Mode", L"Stop", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
-/* 24 */ { L"StopTimeout", L"Timeout", L"Stop", APXCMDOPT_INT | APXCMDOPT_REG, NULL, 0},
-
-/* 25 */ { L"StartImage", L"Image", L"Start", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
-/* 26 */ { L"StartPath", L"WorkingPath", L"Start", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
-/* 27 */ { L"StartClass", L"Class", L"Start", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
-/* 28 */ { L"StartParams", L"Params", L"Start", APXCMDOPT_MSZ | APXCMDOPT_REG, NULL, 0},
-/* 29 */ { L"StartMethod", L"Method", L"Start", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
-/* 30 */ { L"StartMode", L"Mode", L"Start", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
-
-/* 31 */ { L"LogPath", L"Path", L"Log", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
-/* 32 */ { L"LogPrefix", L"Prefix", L"Log", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
-/* 33 */ { L"LogLevel", L"Level", L"Log", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
-/* 34 */ { L"StdError", L"StdError", L"Log", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
-/* 35 */ { L"StdOutput", L"StdOutput", L"Log", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
- /* NULL terminate the array */
- { NULL }
-};
-
-#define GET_OPT_V(x) _options[x].szValue
-#define GET_OPT_I(x) _options[x].dwValue
-#define GET_OPT_T(x) _options[x].dwType
-
-#define ST_DESCRIPTION GET_OPT_T(0)
-#define ST_DISPLAYNAME GET_OPT_T(1)
-#define ST_INSTALL GET_OPT_T(2)
-#define ST_SUSER GET_OPT_T(3)
-#define ST_SPASSWORD GET_OPT_T(4)
-#define ST_STARTUP GET_OPT_T(5)
-#define ST_TYPE GET_OPT_T(6)
-
-#define SO_DESCRIPTION GET_OPT_V(0)
-#define SO_DISPLAYNAME GET_OPT_V(1)
-#define SO_INSTALL GET_OPT_V(2)
-#define SO_SUSER GET_OPT_V(3)
-#define SO_SPASSWORD GET_OPT_V(4)
-#define SO_STARTUP GET_OPT_V(5)
-#define SO_TYPE GET_OPT_V(6)
-
-#define SO_DEPENDSON GET_OPT_V(7)
-#define SO_ENVIRONMENT GET_OPT_V(8)
-
-#define SO_USER GET_OPT_V(9)
-#define SO_PASSWORD GET_OPT_V(10)
-
-#define SO_JAVAHOME GET_OPT_V(11)
-#define SO_JVM GET_OPT_V(12)
-#define SO_JVMOPTIONS GET_OPT_V(13)
-#define SO_CLASSPATH GET_OPT_V(14)
-#define SO_JVMMS GET_OPT_I(15)
-#define SO_JVMMX GET_OPT_I(16)
-#define SO_JVMSS GET_OPT_I(17)
-
-#define SO_STOPIMAGE GET_OPT_V(18)
-#define SO_STOPPATH GET_OPT_V(19)
-#define SO_STOPCLASS GET_OPT_V(20)
-#define SO_STOPPARAMS GET_OPT_V(21)
-#define SO_STOPMETHOD GET_OPT_V(22)
-#define SO_STOPMODE GET_OPT_V(23)
-#define SO_STOPTIMEOUT GET_OPT_I(24)
-
-#define SO_STARTIMAGE GET_OPT_V(25)
-#define SO_STARTPATH GET_OPT_V(26)
-#define SO_STARTCLASS GET_OPT_V(27)
-#define SO_STARTPARAMS GET_OPT_V(28)
-#define SO_STARTMETHOD GET_OPT_V(29)
-#define SO_STARTMODE GET_OPT_V(30)
-
-#define SO_LOGPATH GET_OPT_V(31)
-#define SO_LOGPREFIX GET_OPT_V(32)
-#define SO_LOGLEVEL GET_OPT_V(33)
-
-#define SO_STDERROR GET_OPT_V(34)
-#define SO_STDOUTPUT GET_OPT_V(35)
-
-/* Main servic table entry
- * filled at run-time
- */
-static SERVICE_TABLE_ENTRYW _service_table[] = {
- {NULL, NULL},
- {NULL, NULL}
-};
-
-static SERVICE_STATUS _service_status;
-static SERVICE_STATUS_HANDLE _service_status_handle = NULL;
-/* Set if launched by SCM */
-static BOOL _service_mode = FALSE;
-/* JVM used as worker */
-static BOOL _jni_startup = FALSE;
-/* JVM used for shutdown */
-static BOOL _jni_shutdown = FALSE;
-/* Global variables and objects */
-static APXHANDLE gPool;
-static APXHANDLE gWorker;
-static APX_STDWRAP gStdwrap; /* stdio/stderr redirection */
-
-static LPWSTR _jni_jvmpath = NULL; /* Path to jvm dll */
-static LPSTR _jni_jvmoptions = NULL; /* Path to jvm options */
-
-static LPSTR _jni_classpath = NULL;
-static LPCWSTR _jni_rparam = NULL; /* Startup arguments */
-static LPCWSTR _jni_sparam = NULL; /* Shutdown arguments */
-static LPSTR _jni_rmethod = NULL; /* Startup arguments */
-static LPSTR _jni_smethod = NULL; /* Shutdown arguments */
-static CHAR _jni_rclass[SIZ_RESLEN] = {'\0'}; /* Startup class */
-static CHAR _jni_sclass[SIZ_RESLEN] = {'\0'}; /* Shutdown class */
-
-static HANDLE gShutdownEvent = NULL;
-static HANDLE gSignalEvent = NULL;
-static HANDLE gSignalThread = NULL;
-static BOOL gSignalValid = TRUE;
-
-DWORD WINAPI eventThread(LPVOID lpParam)
-{
- for (;;) {
- DWORD dw = WaitForSingleObject(gSignalEvent, INFINITE);
- if (dw == WAIT_OBJECT_0 && gSignalValid) {
- if (!GenerateConsoleCtrlEvent(CTRL_BREAK_EVENT, 0))
- apxLogWrite(APXLOG_MARK_SYSERR);
- ResetEvent(gSignalEvent);
- }
- else
- break;
- }
- ExitThread(0);
- return 0;
-}
-
-/* redirect console stdout/stderr to files
- * so that java messages can get logged
- * If stderrfile is not specified it will
- * go to stdoutfile.
- */
-static BOOL redirectStdStreams(APX_STDWRAP *lpWrapper)
-{
- BOOL aErr = FALSE;
- BOOL aOut = FALSE;
-
- /* Clear up the handles */
- lpWrapper->fpStdErrFile = NULL;
- lpWrapper->fpStdOutFile = NULL;
-
- /* Save the original streams */
- lpWrapper->fpStdOutSave = *stdout;
- lpWrapper->fpStdErrSave = *stderr;
-
- /* redirect to file or console */
- if (lpWrapper->szStdOutFilename) {
- if (lstrcmpiW(lpWrapper->szStdOutFilename, PRSRV_AUTO) == 0) {
- aOut = TRUE;
- lpWrapper->szStdOutFilename = apxLogFile(gPool,
- lpWrapper->szLogPath,
- NULL,
- L"stdout_");
- }
- /* Delete the file if not in append mode
- * XXX: See if we can use the params instead of that.
- */
- if (!aOut)
- DeleteFileW(lpWrapper->szStdOutFilename);
- lpWrapper->hStdOutFile = CreateFileW(lpWrapper->szStdOutFilename,
- GENERIC_WRITE | GENERIC_READ,
- FILE_SHARE_READ | FILE_SHARE_WRITE,
- NULL,
- OPEN_ALWAYS,
- FILE_ATTRIBUTE_NORMAL,
- NULL);
- if (IS_INVALID_HANDLE(lpWrapper->hStdOutFile))
- return FALSE;
- /* Allways move to the end of file */
- SetFilePointer(lpWrapper->hStdOutFile, 0, NULL, FILE_END);
- }
- else {
- lpWrapper->hStdOutFile = CreateFileW(L"CONOUT$",
- GENERIC_READ | GENERIC_WRITE,
- FILE_SHARE_READ | FILE_SHARE_WRITE,
- NULL,
- OPEN_EXISTING,
- 0,
- NULL);
- if (IS_INVALID_HANDLE(lpWrapper->hStdOutFile))
- return FALSE;
- }
- if (lpWrapper->szStdErrFilename) {
- if (lstrcmpiW(lpWrapper->szStdErrFilename, PRSRV_AUTO) == 0) {
- aErr = TRUE;
- lpWrapper->szStdErrFilename = apxLogFile(gPool,
- lpWrapper->szLogPath,
- NULL,
- L"stderr_");
- }
- if (!aErr)
- DeleteFileW(lpWrapper->szStdErrFilename);
- lpWrapper->hStdErrFile = CreateFileW(lpWrapper->szStdErrFilename,
- GENERIC_WRITE | GENERIC_READ,
- FILE_SHARE_READ | FILE_SHARE_WRITE,
- NULL,
- OPEN_ALWAYS,
- FILE_ATTRIBUTE_NORMAL,
- NULL);
- if (IS_INVALID_HANDLE(lpWrapper->hStdErrFile))
- return FALSE;
- SetFilePointer(lpWrapper->hStdErrFile, 0, NULL, FILE_END);
- }
- else if (lpWrapper->szStdOutFilename) {
- /* Use the same file handle for stderr as for stdout */
- lpWrapper->szStdErrFilename = lpWrapper->szStdOutFilename;
- lpWrapper->hStdErrFile = lpWrapper->hStdOutFile;
- }
- else {
- lpWrapper->hStdErrFile = lpWrapper->hStdOutFile;
- }
- /* Open the stream buffers
- * This will redirect all printf to go to the redirected files.
- * It is used for JNI vprintf functionality.
- */
- lpWrapper->fpStdOutFile = _fdopen(_open_osfhandle(
- (intptr_t)lpWrapper->hStdOutFile,
- _O_TEXT), "w");
- lpWrapper->fpStdErrFile = _fdopen(_open_osfhandle(
- (intptr_t)lpWrapper->hStdErrFile,
- _O_TEXT), "w");
- if (lpWrapper->fpStdOutFile) {
- *stdout = *lpWrapper->fpStdOutFile;
- setvbuf(stdout, NULL, _IONBF, 0);
- }
- if (lpWrapper->fpStdErrFile) {
- *stderr = *lpWrapper->fpStdErrFile;
- setvbuf(stderr, NULL, _IONBF, 0);
- }
- return TRUE;
-}
-
-static void cleanupStdStreams(APX_STDWRAP *lpWrapper)
-{
- /* Close the redirectied streams */
- if (lpWrapper->fpStdOutFile) {
- fclose(lpWrapper->fpStdOutFile);
- *stdout = lpWrapper->fpStdOutSave;
- }
- if (lpWrapper->fpStdErrFile) {
- fclose(lpWrapper->fpStdErrFile);
- *stderr = lpWrapper->fpStdErrSave;
- }
-}
-
-/* Debugging functions */
-static void printUsage(LPAPXCMDLINE lpCmdline)
-{
-#ifdef _DEBUG
- int i = 0;
- fwprintf(stderr, L"Usage: %s //CMD//Servce [--options]\n",
- lpCmdline->szExecutable);
- fwprintf(stderr, L" Commands:\n");
- fwprintf(stderr, L" //IS//ServiceName Install Service\n");
- fwprintf(stderr, L" //US//ServiceName Update Service parameters\n");
- fwprintf(stderr, L" //DS//ServiceName Delete Service\n");
- fwprintf(stderr, L" //RS//ServiceName Run Service\n");
- fwprintf(stderr, L" //SS//ServiceName Stop Service\n");
- fwprintf(stderr,
- L" //TS//ServiceName Run Service as console application\n");
- fwprintf(stderr, L" Options:\n");
- while (_options[i].szName) {
- fwprintf(stderr, L" --%s\n", _options[i].szName);
- ++i;
- }
-#endif
-}
-/* Display configuration parameters */
-static void dumpCmdline()
-{
- int i = 0;
- while (_options[i].szName) {
- if (_options[i].dwType & APXCMDOPT_INT)
- fwprintf(stderr, L"--%-16s %d\n", _options[i].szName,
- _options[i].dwValue);
- else if (_options[i].szValue)
- fwprintf(stderr, L"--%-16s %s\n", _options[i].szName,
- _options[i].szValue);
- else
- fwprintf(stderr, L"--%-16s <NULL>\n", _options[i].szName);
- ++i;
- }
-}
-
-static void setInprocEnvironment()
-{
- LPWSTR p, e;
-
- if (!SO_ENVIRONMENT)
- return; /* Nothing to do */
-
- for (p = SO_ENVIRONMENT; *p; p++) {
- e = apxExpandStrW(gPool, p);
- _wputenv(e);
- apxFree(e);
- while (*p)
- p++;
- }
-}
-
-/* Load the configuration from Registry
- * loads only nonspecified items
- */
-static BOOL loadConfiguration(LPAPXCMDLINE lpCmdline)
-{
- APXHANDLE hRegistry;
- int i = 0;
-
- SetLastError(ERROR_SUCCESS);
- hRegistry = apxCreateRegistryW(gPool, KEY_READ | KREG_WOW6432,
- PRG_REGROOT,
- lpCmdline->szApplication,
- APXREG_SOFTWARE | APXREG_SERVICE);
- if (IS_INVALID_HANDLE(hRegistry)) {
- apxLogWrite(APXLOG_MARK_SYSERR);
- return FALSE;
- }
- /* browse through options */
- while (_options[i].szName) {
- DWORD dwFrom;
-
- dwFrom = (_options[i].dwType & APXCMDOPT_REG) ? APXREG_PARAMSOFTWARE : APXREG_SERVICE;
- if (!(_options[i].dwType & APXCMDOPT_FOUND)) {
- if (_options[i].dwType & APXCMDOPT_STR) {
- _options[i].szValue = apxRegistryGetStringW(hRegistry,
- dwFrom,
- _options[i].szSubkey,
- _options[i].szRegistry);
- /* Expand environment variables */
- if (_options[i].szValue && (_options[i].dwType & APXCMDOPT_STE)) {
- LPWSTR exp = apxExpandStrW(gPool, _options[i].szValue);
- if (exp != _options[i].szValue)
- apxFree(_options[i].szValue);
- _options[i].szValue = exp;
- }
- }
- else if (_options[i].dwType & APXCMDOPT_INT) {
- _options[i].dwValue = apxRegistryGetNumberW(hRegistry,
- dwFrom,
- _options[i].szSubkey,
- _options[i].szRegistry);
- }
- else if (_options[i].dwType & APXCMDOPT_MSZ) {
- _options[i].szValue = apxRegistryGetMzStrW(hRegistry,
- dwFrom,
- _options[i].szSubkey,
- _options[i].szRegistry,
- NULL,
- &(_options[i].dwValue));
- }
- }
- /* Merge the command line options with registry */
- else if (_options[i].dwType & APXCMDOPT_ADD) {
- LPWSTR cv = _options[i].szValue;
- LPWSTR ov = NULL;
- if (_options[i].dwType & APXCMDOPT_MSZ) {
- ov = apxRegistryGetMzStrW(hRegistry, dwFrom,
- _options[i].szSubkey,
- _options[i].szRegistry,
- NULL,
- &(_options[i].dwValue));
- _options[i].szValue = apxMultiSzCombine(gPool, ov, cv,
- &(_options[i].dwValue));
- if (ov)
- apxFree(ov);
- }
- }
- ++i;
- }
- apxCloseHandle(hRegistry);
-#ifdef _DEBUG
- dumpCmdline();
-#endif
- return TRUE;
-}
-
-/* Save changed configuration to registry
- */
-static BOOL saveConfiguration(LPAPXCMDLINE lpCmdline)
-{
- APXHANDLE hRegistry;
- int i = 0;
- hRegistry = apxCreateRegistryW(gPool, KEY_WRITE | KREG_WOW6432,
- PRG_REGROOT,
- lpCmdline->szApplication,
- APXREG_SOFTWARE | APXREG_SERVICE);
- if (IS_INVALID_HANDLE(hRegistry))
- return FALSE;
- /* TODO: Use array size */
- while (_options[i].szName) {
- /* Skip the service params */
- if ((_options[i].dwType & APXCMDOPT_SRV) ||
- !(_options[i].dwType & APXCMDOPT_FOUND)) {
- /* Skip non-modified version */
- }
- /* Update only modified params */
- else if (_options[i].dwType & APXCMDOPT_STR)
- apxRegistrySetStrW(hRegistry, APXREG_PARAMSOFTWARE,
- _options[i].szSubkey,
- _options[i].szRegistry,
- _options[i].szValue);
- else if (_options[i].dwType & APXCMDOPT_INT)
- apxRegistrySetNumW(hRegistry, APXREG_PARAMSOFTWARE,
- _options[i].szSubkey,
- _options[i].szRegistry,
- _options[i].dwValue);
- else if (_options[i].dwType & APXCMDOPT_MSZ)
- apxRegistrySetMzStrW(hRegistry, APXREG_PARAMSOFTWARE,
- _options[i].szSubkey,
- _options[i].szRegistry,
- _options[i].szValue,
- _options[i].dwValue);
- ++i;
- }
- apxCloseHandle(hRegistry);
- return TRUE;
-}
-
-/* Operations */
-static BOOL docmdInstallService(LPAPXCMDLINE lpCmdline)
-{
- APXHANDLE hService;
- BOOL rv;
- DWORD dwStart = SERVICE_DEMAND_START;
- DWORD dwType = SERVICE_WIN32_OWN_PROCESS;
- WCHAR szImage[SIZ_HUGLEN];
-
- apxLogWrite(APXLOG_MARK_DEBUG "Installing service...");
- hService = apxCreateService(gPool, SC_MANAGER_CREATE_SERVICE, FALSE);
- if (IS_INVALID_HANDLE(hService)) {
- apxLogWrite(APXLOG_MARK_ERROR "Unable to open the Service Manager");
- return FALSE;
- }
- /* Check the startup mode */
- if ((ST_STARTUP & APXCMDOPT_FOUND) &&
- lstrcmpiW(SO_STARTUP, PRSRV_AUTO) == 0)
- dwStart = SERVICE_AUTO_START;
- /* Check the service type */
- if ((ST_TYPE & APXCMDOPT_FOUND) &&
- lstrcmpiW(SO_TYPE, STYPE_INTERACTIVE) == 0)
- dwType |= SERVICE_INTERACTIVE_PROCESS;
-
- /* Check if --Install is provided */
- if (!SO_INSTALL) {
- lstrcpyW(szImage, lpCmdline->szExePath);
- lstrcatW(szImage, L"\\");
- lstrcatW(szImage, lpCmdline->szExecutable);
- lstrcatW(szImage, L".exe");
- }
- else
- lstrcpyW(szImage, SO_INSTALL);
- /* Replace not needed qoutes */
- apxStrQuoteInplaceW(szImage);
- /* Add run-service command line option */
- lstrcatW(szImage, L" //RS//");
- lstrcatW(szImage, lpCmdline->szApplication);
- SO_INSTALL = apxPoolStrdupW(gPool, szImage);
- /* Ensure that option gets saved in the registry */
- ST_INSTALL |= APXCMDOPT_FOUND;
-#ifdef _DEBUG
- /* Display configured options */
- dumpCmdline();
-#endif
- apxLogWrite(APXLOG_MARK_INFO "Service %S name %S", lpCmdline->szApplication,
- SO_DISPLAYNAME);
- rv = apxServiceInstall(hService,
- lpCmdline->szApplication,
- SO_DISPLAYNAME, /* --DisplayName */
- SO_INSTALL,
- SO_DEPENDSON, /* --DependendsOn */
- dwType,
- dwStart);
- /* Set the --Description */
- if (rv) {
- LPCWSTR sd = NULL;
- LPCWSTR su = NULL;
- LPCWSTR sp = NULL;
- if (ST_DESCRIPTION & APXCMDOPT_FOUND) {
- sd = SO_DESCRIPTION;
- apxLogWrite(APXLOG_MARK_DEBUG "Setting service description %S",
- SO_DESCRIPTION);
- }
- if (ST_SUSER & APXCMDOPT_FOUND) {
- su = SO_SUSER;
- apxLogWrite(APXLOG_MARK_DEBUG "Setting service user %S",
- SO_SUSER);
- }
- if (ST_SPASSWORD & APXCMDOPT_FOUND) {
- sp = SO_SPASSWORD;
- apxLogWrite(APXLOG_MARK_DEBUG "Setting service password %S",
- SO_SPASSWORD);
- }
- apxServiceSetNames(hService, NULL, NULL, sd, su, sp);
- }
- apxCloseHandle(hService);
- if (rv) {
- saveConfiguration(lpCmdline);
- apxLogWrite(APXLOG_MARK_INFO "Service %S installed",
- lpCmdline->szApplication);
- }
- else
- apxLogWrite(APXLOG_MARK_ERROR "Failed installing %S service",
- lpCmdline->szApplication);
-
- return rv;
-}
-
-static BOOL docmdDeleteService(LPAPXCMDLINE lpCmdline)
-{
- APXHANDLE hService;
- BOOL rv = FALSE;
-
- apxLogWrite(APXLOG_MARK_INFO "Deleting service...");
- hService = apxCreateService(gPool, SC_MANAGER_CONNECT, FALSE);
- if (IS_INVALID_HANDLE(hService)) {
- apxLogWrite(APXLOG_MARK_ERROR "Unable to open the Service Manager");
- return FALSE;
- }
- /* Delete service will stop the service if running */
- if (apxServiceOpen(hService, lpCmdline->szApplication, SERVICE_ALL_ACCESS)) {
- WCHAR szWndManagerClass[SIZ_RESLEN];
- HANDLE hWndManager = NULL;
- lstrcpyW(szWndManagerClass, lpCmdline->szApplication);
- lstrcatW(szWndManagerClass, L"_CLASS");
- /* Close the monitor application if running */
- if ((hWndManager = FindWindowW(szWndManagerClass, NULL)) != NULL) {
- SendMessage(hWndManager, WM_CLOSE, 0, 0);
- }
- rv = apxServiceDelete(hService);
- }
- if (rv) {
- /* Delete all service registry settings */
- apxDeleteRegistryW(PRG_REGROOT, lpCmdline->szApplication, TRUE);
- apxLogWrite(APXLOG_MARK_DEBUG "Service %S deleted",
- lpCmdline->szApplication);
- }
- else {
- apxDisplayError(TRUE, NULL, 0, "Unable to delete %S service",
- lpCmdline->szApplication);
- }
- apxCloseHandle(hService);
- apxLogWrite(APXLOG_MARK_INFO "Delete service finished.");
- return rv;
-}
-
-static BOOL docmdStopService(LPAPXCMDLINE lpCmdline)
-{
- APXHANDLE hService;
- BOOL rv = FALSE;
-
- apxLogWrite(APXLOG_MARK_INFO "Stopping service...");
- hService = apxCreateService(gPool, GENERIC_ALL, FALSE);
- if (IS_INVALID_HANDLE(hService)) {
- apxLogWrite(APXLOG_MARK_ERROR "Unable to open the Service Manager");
- return FALSE;
- }
-
- SetLastError(ERROR_SUCCESS);
- /* Open the service */
- if (apxServiceOpen(hService, lpCmdline->szApplication,
- GENERIC_READ | GENERIC_EXECUTE)) {
- rv = apxServiceControl(hService,
- SERVICE_CONTROL_STOP,
- 0,
- NULL,
- NULL);
- if (rv)
- apxLogWrite(APXLOG_MARK_INFO "Service %S stopped",
- lpCmdline->szApplication);
- else
- apxLogWrite(APXLOG_MARK_ERROR "Failed to stop %S service",
- lpCmdline->szApplication);
-
- }
- else
- apxDisplayError(TRUE, NULL, 0, "Unable to open %S service",
- lpCmdline->szApplication);
- apxCloseHandle(hService);
- apxLogWrite(APXLOG_MARK_INFO "Stop service finished.");
- return rv;
-}
-
-static BOOL docmdUpdateService(LPAPXCMDLINE lpCmdline)
-{
- APXHANDLE hService;
- BOOL rv = FALSE;
-
- apxLogWrite(APXLOG_MARK_INFO "Updating service...");
-
- hService = apxCreateService(gPool, SC_MANAGER_CREATE_SERVICE, FALSE);
- if (IS_INVALID_HANDLE(hService)) {
- apxLogWrite(APXLOG_MARK_ERROR "Unable to open the Service Manager");
- return FALSE;
- }
- SetLastError(0);
- /* Open the service */
- if (apxServiceOpen(hService, lpCmdline->szApplication, SERVICE_ALL_ACCESS)) {
- DWORD dwStart = SERVICE_NO_CHANGE;
- DWORD dwType = SERVICE_NO_CHANGE;
- LPCWSTR su = NULL;
- LPCWSTR sp = NULL;
- if (ST_SUSER & APXCMDOPT_FOUND) {
- su = SO_SUSER;
- apxLogWrite(APXLOG_MARK_DEBUG "Setting service user %S",
- SO_SUSER);
- }
- if (ST_SPASSWORD & APXCMDOPT_FOUND) {
- sp = SO_SPASSWORD;
- apxLogWrite(APXLOG_MARK_DEBUG "Setting service password %S",
- SO_SPASSWORD);
- }
- apxServiceSetNames(hService,
- NULL, /* Never update the ImagePath */
- SO_DISPLAYNAME,
- SO_DESCRIPTION,
- su,
- sp);
- /* Update the --Startup mode */
- if (ST_STARTUP & APXCMDOPT_FOUND) {
- if (!lstrcmpiW(SO_STARTUP, PRSRV_AUTO))
- dwStart = SERVICE_AUTO_START;
- else if (!lstrcmpiW(SO_STARTUP, PRSRV_MANUAL))
- dwStart = SERVICE_DEMAND_START;
- }
- if (ST_TYPE & APXCMDOPT_FOUND) {
- if (!lstrcmpiW(SO_TYPE, STYPE_INTERACTIVE))
- dwType = SERVICE_WIN32_OWN_PROCESS | SERVICE_INTERACTIVE_PROCESS;
- }
- apxServiceSetOptions(hService,
- dwType,
- dwStart,
- SERVICE_NO_CHANGE);
-
- apxLogWrite(APXLOG_MARK_INFO "Service %S updated",
- lpCmdline->szApplication);
-
- saveConfiguration(lpCmdline);
- }
- else
- apxDisplayError(TRUE, NULL, 0, "Unable to open %S service",
- lpCmdline->szApplication);
- apxCloseHandle(hService);
- apxLogWrite(APXLOG_MARK_INFO "Update service finished.");
- return rv;
-}
-
-
-/* Report the service status to the SCM
- */
-int reportServiceStatus(DWORD dwCurrentState,
- DWORD dwWin32ExitCode,
- DWORD dwWaitHint)
-{
- static DWORD dwCheckPoint = 1;
- BOOL fResult = TRUE;
-
- if (_service_mode && _service_status_handle) {
- if (dwCurrentState == SERVICE_START_PENDING)
- _service_status.dwControlsAccepted = 0;
- else
- _service_status.dwControlsAccepted = SERVICE_ACCEPT_STOP;
-
- _service_status.dwCurrentState = dwCurrentState;
- _service_status.dwWin32ExitCode = dwWin32ExitCode;
- _service_status.dwWaitHint = dwWaitHint;
-
- if ((dwCurrentState == SERVICE_RUNNING) ||
- (dwCurrentState == SERVICE_STOPPED))
- _service_status.dwCheckPoint = 0;
- else
- _service_status.dwCheckPoint = dwCheckPoint++;
- fResult = SetServiceStatus(_service_status_handle, &_service_status);
- if (!fResult) {
- /* TODO: Deal with error */
- }
- }
- return fResult;
-}
-
-
-BOOL child_callback(APXHANDLE hObject, UINT uMsg,
- WPARAM wParam, LPARAM lParam)
-{
- /* TODO: Make stdout and stderr buffers
- * to prevent streams intermixing when there
- * is no separate file for each stream
- */
- if (uMsg == WM_CHAR) {
- int ch = LOWORD(wParam);
- if (lParam)
- fputc(ch, stderr);
- else
- fputc(ch, stdout);
- }
- return TRUE;
-}
-
-/* Executed when the service receives stop event */
-static DWORD WINAPI serviceStop(LPVOID lpParameter)
-{
- APXHANDLE hWorker = NULL;
- DWORD rv = 0;
- BOOL wait_to_die = FALSE;
- DWORD timeout = SO_STOPTIMEOUT * 1000;
- DWORD dwCtrlType = (DWORD)((BYTE *)lpParameter - (BYTE *)0);
-
- apxLogWrite(APXLOG_MARK_INFO "Stopping service...");
-
- if (IS_INVALID_HANDLE(gWorker)) {
- apxLogWrite(APXLOG_MARK_INFO "Worker is not defined");
- return TRUE; /* Nothing to do */
- }
- if (_jni_shutdown) {
- if (!SO_STARTPATH && SO_STOPPATH) {
- /* If the Working path is specified change the current directory
- * but only if the start path wasn't specified already.
- */
- SetCurrentDirectoryW(SO_STARTPATH);
- }
- hWorker = apxCreateJava(gPool, _jni_jvmpath);
- if (IS_INVALID_HANDLE(hWorker)) {
- apxLogWrite(APXLOG_MARK_ERROR "Failed creating java %S", _jni_jvmpath);
- return 1;
- }
- if (!apxJavaInitialize(hWorker, _jni_classpath, _jni_jvmoptions,
- SO_JVMMS, SO_JVMMX, SO_JVMSS)) {
- rv = 2;
- apxLogWrite(APXLOG_MARK_ERROR "Failed initializing java %s", _jni_classpath);
- goto cleanup;
- }
- if (!apxJavaLoadMainClass(hWorker, _jni_sclass, _jni_smethod, _jni_sparam)) {
- rv = 2;
- apxLogWrite(APXLOG_MARK_ERROR "Failed loading main %s class %s",
- _jni_rclass, _jni_classpath);
- goto cleanup;
- }
- /* Create sutdown event */
- gShutdownEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
- if (!apxJavaStart(hWorker)) {
- apxLogWrite(APXLOG_MARK_ERROR "Failed starting java");
- rv = 3;
- }
- else {
- apxLogWrite(APXLOG_MARK_DEBUG "Waiting for java jni stop worker to finish...");
- apxJavaWait(hWorker, INFINITE, FALSE);
- apxLogWrite(APXLOG_MARK_DEBUG "Java jni stop worker finished.");
- }
- wait_to_die = TRUE;
- }
- else if (SO_STOPMODE) { /* Only in case we have a stop mode */
- DWORD nArgs;
- LPWSTR *pArgs;
- /* Redirect process */
- hWorker = apxCreateProcessW(gPool,
- 0,
- child_callback,
- SO_USER,
- SO_PASSWORD,
- FALSE);
- if (IS_INVALID_HANDLE(hWorker)) {
- apxLogWrite(APXLOG_MARK_ERROR "Failed creating process");
- return 1;
- }
- if (!apxProcessSetExecutableW(hWorker, SO_STOPIMAGE)) {
- apxLogWrite(APXLOG_MARK_ERROR "Failed setting process executable %S",
- SO_STARTIMAGE);
- rv = 2;
- goto cleanup;
- }
- /* Assemble the command line */
- nArgs = apxMultiSzToArrayW(gPool, SO_STOPPARAMS, &pArgs);
- /* Pass the argv to child process */
- if (!apxProcessSetCommandArgsW(hWorker, SO_STOPIMAGE,
- nArgs, pArgs)) {
- rv = 3;
- apxLogWrite(APXLOG_MARK_ERROR "Failed setting process arguments (argc=%d)",
- nArgs);
- goto cleanup;
- }
- /* Set the working path */
- if (!apxProcessSetWorkingPathW(hWorker, SO_STOPPATH)) {
- rv = 4;
- apxLogWrite(APXLOG_MARK_ERROR "Failed setting process working path to %S",
- SO_STOPPATH);
- goto cleanup;
- }
- /* Finally execute the child process
- */
- if (!apxProcessExecute(hWorker)) {
- rv = 5;
- apxLogWrite(APXLOG_MARK_ERROR "Failed executing process");
- goto cleanup;
- } else {
- apxLogWrite(APXLOG_MARK_DEBUG "Waiting stop worker to finish...");
- apxHandleWait(hWorker, INFINITE, FALSE);
- apxLogWrite(APXLOG_MARK_DEBUG "Stop worker finished.");
- }
- wait_to_die = TRUE;
- }
-cleanup:
- /* Close Java JNI handle or stop worker
- * If this is the single JVM instance it will unload
- * the JVM dll too.
- * The worker will be closed on service exit.
- */
- if (!IS_INVALID_HANDLE(hWorker))
- apxCloseHandle(hWorker);
- if (gSignalEvent) {
- gSignalValid = FALSE;
- SetEvent(gSignalEvent);
- WaitForSingleObject(gSignalThread, 1000);
- CloseHandle(gSignalEvent);
- CloseHandle(gSignalThread);
- gSignalEvent = NULL;
- }
- SetEvent(gShutdownEvent);
- if (timeout > 0x7FFFFFFF)
- timeout = INFINITE; /* If the timeout was '-1' wait forewer */
- if (wait_to_die && !timeout)
- timeout = 300 * 1000; /* Use the 5 minute default shutdown */
-
- if (dwCtrlType == SERVICE_CONTROL_SHUTDOWN)
- timeout = MIN(timeout, apxGetMaxServiceTimeout(gPool));
- reportServiceStatus(SERVICE_STOP_PENDING, NO_ERROR, timeout);
-
- if (timeout) {
- FILETIME fts, fte;
- ULARGE_INTEGER s, e;
- DWORD nms;
- /* Wait to give it a chance to die naturally, then kill it. */
- apxLogWrite(APXLOG_MARK_DEBUG "Waiting for worker to die naturally...");
- GetSystemTimeAsFileTime(&fts);
- rv = apxHandleWait(gWorker, timeout, TRUE);
- GetSystemTimeAsFileTime(&fte);
- s.LowPart = fts.dwLowDateTime;
- s.HighPart = fts.dwHighDateTime;
- e.LowPart = fte.dwLowDateTime;
- e.HighPart = fte.dwHighDateTime;
- nms = (DWORD)((e.QuadPart - s.QuadPart) / 10000);
- if (rv == WAIT_OBJECT_0) {
- rv = 0;
- apxLogWrite(APXLOG_MARK_DEBUG "Worker finished gracefully in %d ms.", nms);
- }
- else
- apxLogWrite(APXLOG_MARK_DEBUG "Worker was killed in %d ms.", nms);
- }
- else {
- apxLogWrite(APXLOG_MARK_DEBUG "Sending WM_CLOSE to worker");
- apxHandleSendMessage(gWorker, WM_CLOSE, 0, 0);
- }
-
- apxLogWrite(APXLOG_MARK_INFO "Service stopped.");
- reportServiceStatus(SERVICE_STOPPED, NO_ERROR, 0);
- return rv;
-}
-
-/* Executed when the service receives start event */
-static DWORD serviceStart()
-{
- DWORD rv = 0;
- DWORD nArgs;
- LPWSTR *pArgs;
- FILETIME fts;
-
- apxLogWrite(APXLOG_MARK_INFO "Starting service...");
-
- if (!IS_INVALID_HANDLE(gWorker)) {
- apxLogWrite(APXLOG_MARK_INFO "Worker is not defined");
- return TRUE; /* Nothing to do */
- }
- GetSystemTimeAsFileTime(&fts);
- if (_jni_startup) {
- if (SO_STARTPATH) {
- /* If the Working path is specified change the current directory */
- SetCurrentDirectoryW(SO_STARTPATH);
- }
- /* Set the environment using putenv, so JVM can use it */
- setInprocEnvironment();
- /* Create the JVM glbal worker */
- gWorker = apxCreateJava(gPool, _jni_jvmpath);
- if (IS_INVALID_HANDLE(gWorker)) {
- apxLogWrite(APXLOG_MARK_ERROR "Failed creating java %S", _jni_jvmpath);
- return 1;
- }
- if (!apxJavaInitialize(gWorker, _jni_classpath, _jni_jvmoptions,
- SO_JVMMS, SO_JVMMX, SO_JVMSS)) {
- rv = 2;
- apxLogWrite(APXLOG_MARK_ERROR "Failed initializing java %s", _jni_classpath);
- goto cleanup;
- }
- if (!apxJavaLoadMainClass(gWorker, _jni_rclass, _jni_rmethod, _jni_rparam)) {
- rv = 3;
- apxLogWrite(APXLOG_MARK_ERROR "Failed loading main %s class %s", _jni_rclass, _jni_classpath);
- goto cleanup;
- }
- apxJavaSetOut(gWorker, TRUE, gStdwrap.szStdErrFilename);
- apxJavaSetOut(gWorker, FALSE, gStdwrap.szStdOutFilename);
- if (!apxJavaStart(gWorker)) {
- rv = 4;
- apxLogWrite(APXLOG_MARK_ERROR "Failed starting Java");
- goto cleanup;
- }
- apxLogWrite(APXLOG_MARK_DEBUG "Java started %s", _jni_rclass);
- }
- else {
- /* Redirect process */
- gWorker = apxCreateProcessW(gPool,
- 0,
- child_callback,
- SO_USER,
- SO_PASSWORD,
- FALSE);
- if (IS_INVALID_HANDLE(gWorker)) {
- apxLogWrite(APXLOG_MARK_ERROR "Failed creating process");
- return 1;
- }
- if (!apxProcessSetExecutableW(gWorker, SO_STARTIMAGE)) {
- apxLogWrite(APXLOG_MARK_ERROR "Failed setting process executable %S",
- SO_STARTIMAGE);
- rv = 2;
- goto cleanup;
- }
- /* Assemble the command line */
- nArgs = apxMultiSzToArrayW(gPool, SO_STARTPARAMS, &pArgs);
- /* Pass the argv to child process */
- if (!apxProcessSetCommandArgsW(gWorker, SO_STARTIMAGE,
- nArgs, pArgs)) {
- rv = 3;
- apxLogWrite(APXLOG_MARK_ERROR "Failed setting process arguments (argc=%d)",
- nArgs);
- goto cleanup;
- }
- /* Set the working path */
- if (!apxProcessSetWorkingPathW(gWorker, SO_STARTPATH)) {
- rv = 4;
- apxLogWrite(APXLOG_MARK_ERROR "Failed setting process working path to %S",
- SO_STARTPATH);
- goto cleanup;
- }
- /* Finally execute the child process
- */
- if (!apxProcessExecute(gWorker)) {
- rv = 5;
- apxLogWrite(APXLOG_MARK_ERROR "Failed executing process");
- goto cleanup;
- }
- }
- if (rv == 0) {
- FILETIME fte;
- ULARGE_INTEGER s, e;
- DWORD nms;
- GetSystemTimeAsFileTime(&fte);
- s.LowPart = fts.dwLowDateTime;
- s.HighPart = fts.dwHighDateTime;
- e.LowPart = fte.dwLowDateTime;
- e.HighPart = fte.dwHighDateTime;
- nms = (DWORD)((e.QuadPart - s.QuadPart) / 10000);
- apxLogWrite(APXLOG_MARK_INFO "Service started in %d ms.", nms);
- }
- return rv;
-cleanup:
- if (!IS_INVALID_HANDLE(gWorker))
- apxCloseHandle(gWorker); /* Close the worker handle */
- gWorker = NULL;
- return rv;
-}
-
-/* Service control handler
- */
-void WINAPI service_ctrl_handler(DWORD dwCtrlCode)
-{
- DWORD threadId;
- HANDLE stopThread;
-
- switch (dwCtrlCode) {
- case SERVICE_CONTROL_STOP:
- reportServiceStatus(SERVICE_STOP_PENDING, NO_ERROR, 3000);
- /* Stop the service asynchronously */
- stopThread = CreateThread(NULL, 0,
- serviceStop,
- (LPVOID)SERVICE_CONTROL_STOP,
- 0, &threadId);
- WaitForSingleObject(stopThread, INFINITE);
- CloseHandle(stopThread);
-
- return;
- case SERVICE_CONTROL_SHUTDOWN:
- apxLogWrite(APXLOG_MARK_INFO "Service SHUTDOWN signaled");
- reportServiceStatus(SERVICE_STOP_PENDING, NO_ERROR, 3000);
- /* Stop the service asynchronously */
- stopThread = CreateThread(NULL, 0,
- serviceStop,
- (LPVOID)SERVICE_CONTROL_SHUTDOWN,
- 0, &threadId);
- WaitForSingleObject(stopThread, INFINITE);
- CloseHandle(stopThread);
- return;
- case SERVICE_CONTROL_INTERROGATE:
- reportServiceStatus(_service_status.dwCurrentState,
- _service_status.dwWin32ExitCode,
- _service_status.dwWaitHint);
- return;
- default:
- break;
- }
-}
-
-/* Console control handler
- *
- */
-BOOL WINAPI console_handler(DWORD dwCtrlType)
-{
- switch (dwCtrlType) {
- case CTRL_BREAK_EVENT:
- apxLogWrite(APXLOG_MARK_INFO "Console CTRL+BREAK event signaled");
- return FALSE;
- case CTRL_C_EVENT:
- apxLogWrite(APXLOG_MARK_INFO "Console CTRL+C event signaled");
- serviceStop((LPVOID)SERVICE_CONTROL_STOP);
- return TRUE;
- case CTRL_CLOSE_EVENT:
- apxLogWrite(APXLOG_MARK_INFO "Console CTRL+CLOSE event signaled");
- serviceStop((LPVOID)SERVICE_CONTROL_STOP);
- return TRUE;
- case CTRL_SHUTDOWN_EVENT:
- apxLogWrite(APXLOG_MARK_INFO "Console SHUTDOWN event signaled");
- serviceStop((LPVOID)SERVICE_CONTROL_SHUTDOWN);
- return TRUE;
- case CTRL_LOGOFF_EVENT:
- apxLogWrite(APXLOG_MARK_INFO "Console LOGOFF event signaled");
- if (!_service_mode) {
- serviceStop((LPVOID)SERVICE_CONTROL_STOP);
- }
- return TRUE;
- break;
-
- }
- return FALSE;
-}
-
-/* Main service execution loop */
-void WINAPI serviceMain(DWORD argc, LPTSTR *argv)
-{
- DWORD rc;
- _service_status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
- _service_status.dwCurrentState = SERVICE_START_PENDING;
- _service_status.dwControlsAccepted = SERVICE_ACCEPT_STOP |
- SERVICE_ACCEPT_PAUSE_CONTINUE;
- _service_status.dwWin32ExitCode = 0;
- _service_status.dwCheckPoint = 0;
- _service_status.dwWaitHint = 0;
- _service_status.dwServiceSpecificExitCode = 0;
-
- apxLogWrite(APXLOG_MARK_DEBUG "Inside ServiceMain...");
-
- if (_service_name) {
- WCHAR en[SIZ_DESLEN];
- int i;
- PSECURITY_ATTRIBUTES sa = GetNullACL();
- lstrcpyW(en, L"Global\\");
- lstrcatW(en, _service_name);
- lstrcatW(en, PRSRV_SIGNAL);
- for (i = 7; i < lstrlenW(en); i++) {
- if (en[i] >= L'a' && en[i] <= L'z')
- en[i] = en[i] - 32;
- }
- gSignalEvent = CreateEventW(sa, TRUE, FALSE, en);
- CleanNullACL((void *)sa);
-
- if (gSignalEvent) {
- DWORD tid;
- gSignalThread = CreateThread(NULL, 0, eventThread, NULL, 0, &tid);
- }
- }
- /* Check the StartMode */
- if (SO_STARTMODE) {
- if (!lstrcmpiW(SO_STARTMODE, PRSRV_JVM)) {
- _jni_startup = TRUE;
- WideToAscii(SO_STARTCLASS, _jni_rclass);
- /* Exchange all dots with slashes */
- apxStrCharReplaceA(_jni_rclass, '.', '/');
- _jni_rparam = SO_STARTPARAMS;
- }
- else if (!lstrcmpiW(SO_STARTMODE, PRSRV_JAVA)) {
- LPWSTR jx = NULL, szJH = apxGetJavaSoftHome(gPool, FALSE);
- if (szJH) {
- jx = apxPoolAlloc(gPool, (lstrlenW(szJH) + 16) * sizeof(WCHAR));
- lstrcpyW(jx, szJH);
- lstrcatW(jx, PRSRV_JBIN);
- SO_STARTPATH = szJH;
- }
- /* StartImage now contains the full path to the java.exe */
- SO_STARTIMAGE = jx;
- }
- }
- /* Check the StopMode */
- if (SO_STOPMODE) {
- if (!lstrcmpiW(SO_STOPMODE, PRSRV_JVM)) {
- _jni_shutdown = TRUE;
- WideToAscii(SO_STOPCLASS, _jni_sclass);
- apxStrCharReplaceA(_jni_sclass, '.', '/');
- _jni_sparam = SO_STOPPARAMS;
- }
- else if (!lstrcmpiW(SO_STOPMODE, PRSRV_JAVA)) {
- LPWSTR jx = NULL, szJH = apxGetJavaSoftHome(gPool, FALSE);
- if (szJH) {
- jx = apxPoolAlloc(gPool, (lstrlenW(szJH) + 16) * sizeof(WCHAR));
- lstrcpyW(jx, szJH);
- lstrcatW(jx, PRSRV_JBIN);
- SO_STOPPATH = szJH;
- }
- /* StopImage now contains the full path to the java.exe */
- SO_STOPIMAGE = jx;
- }
- }
- /* Find the classpath */
- if (_jni_shutdown || _jni_startup) {
- if (SO_JVM) {
- if (lstrcmpW(SO_JVM, PRSRV_AUTO))
- _jni_jvmpath = SO_JVM;
- }
- if (SO_CLASSPATH)
- _jni_classpath = WideToUTF8(SO_CLASSPATH);
- if (SO_STARTMETHOD)
- _jni_rmethod = WideToAscii(SO_STARTMETHOD, (LPSTR)SO_STARTMETHOD);
- if (SO_STOPMETHOD)
- _jni_smethod = WideToAscii(SO_STOPMETHOD, (LPSTR)SO_STOPMETHOD);
- _jni_jvmoptions = MzWideToUTF8(SO_JVMOPTIONS);
- }
- if (_service_mode) {
- /* Register Service Control handler */
- _service_status_handle = RegisterServiceCtrlHandlerW(_service_name,
- service_ctrl_handler);
- if (IS_INVALID_HANDLE(_service_status_handle)) {
- apxLogWrite(APXLOG_MARK_ERROR "Failed to register Service Control for %S",
- _service_name);
- goto cleanup;
- }
- /* Allocate console so that events gets processed */
- AllocConsole();
- }
- reportServiceStatus(SERVICE_START_PENDING, NO_ERROR, 3000);
- if ((rc = serviceStart()) == 0) {
- /* Service is started */
- DWORD rv;
- reportServiceStatus(SERVICE_RUNNING, NO_ERROR, 0);
- apxLogWrite(APXLOG_MARK_DEBUG "Waiting worker to finish...");
- /* Set console handler to capture CTRL events */
- SetConsoleCtrlHandler((PHANDLER_ROUTINE)console_handler, TRUE);
-
- rv = apxHandleWait(gWorker, INFINITE, FALSE);
- apxLogWrite(APXLOG_MARK_DEBUG "Worker finished.");
- reportServiceStatus(SERVICE_STOP_PENDING, NO_ERROR, 0);
- fflush(stdout);
- }
- else {
- apxLogWrite(APXLOG_MARK_ERROR "ServiceStart returned %d", rc);
- goto cleanup;
- }
- if (gShutdownEvent) {
- /* Ensure that shutdown thread exits before us */
- apxLogWrite(APXLOG_MARK_DEBUG "Waiting for ShutdownEvent");
- WaitForSingleObject(gShutdownEvent, 60 * 1000);
- apxLogWrite(APXLOG_MARK_DEBUG "ShutdownEvent signaled");
- CloseHandle(gShutdownEvent);
- }
- reportServiceStatus(SERVICE_STOPPED, NO_ERROR, 0);
-
- return;
-cleanup:
- /* Cleanup */
- reportServiceStatus(SERVICE_STOPPED, ERROR_SERVICE_SPECIFIC_ERROR, 0);
- return;
-}
-
-
-/* Run the service in the debug mode */
-BOOL docmdDebugService(LPAPXCMDLINE lpCmdline)
-{
- BOOL rv = FALSE;
-
- _service_mode = FALSE;
- _service_name = lpCmdline->szApplication;
- apxLogWrite(APXLOG_MARK_INFO "Debugging Service...");
- serviceMain(0, NULL);
- apxLogWrite(APXLOG_MARK_INFO "Debug service finished.");
-
- return rv;
-}
-
-BOOL docmdRunService(LPAPXCMDLINE lpCmdline)
-{
- BOOL rv = FALSE;
- _service_mode = TRUE;
-
- apxLogWrite(APXLOG_MARK_INFO "Running Service...");
- _service_name = lpCmdline->szApplication;
- _service_table[0].lpServiceName = lpCmdline->szApplication;
- _service_table[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTIONW)serviceMain;
- rv = (StartServiceCtrlDispatcherW(_service_table) == FALSE);
- apxLogWrite(APXLOG_MARK_INFO "Run service finished.");
- return rv;
-}
-
-void __cdecl main(int argc, char **argv)
-{
- UINT rv = 0;
-
- LPAPXCMDLINE lpCmdline;
-
- apxHandleManagerInitialize();
- /* Create the main Pool */
- gPool = apxPoolCreate(NULL, 0);
-
- /* Parse the command line */
- if ((lpCmdline = apxCmdlineParse(gPool, _options, _commands)) == NULL) {
- apxLogWrite(APXLOG_MARK_ERROR "Invalid command line arguments");
- rv = 1;
- goto cleanup;
- }
- apxCmdlineLoadEnvVars(lpCmdline);
- if (lpCmdline->dwCmdIndex < 5 &&
- !loadConfiguration(lpCmdline)) {
- apxLogWrite(APXLOG_MARK_ERROR "Load configuration failed");
- rv = 2;
- goto cleanup;
- }
-
- apxLogOpen(gPool, SO_LOGPATH, SO_LOGPREFIX);
- apxLogLevelSetW(NULL, SO_LOGLEVEL);
- apxLogWrite(APXLOG_MARK_DEBUG "Procrun log initialized");
- apxLogWrite(APXLOG_MARK_INFO "Procrun (%s) started", PRG_VERSION);
-
- AplZeroMemory(&gStdwrap, sizeof(APX_STDWRAP));
-
- gStdwrap.szLogPath = SO_LOGPATH;
- /* In debug mode allways use console */
- if (lpCmdline->dwCmdIndex != 1) {
- gStdwrap.szStdOutFilename = SO_STDOUTPUT;
- gStdwrap.szStdErrFilename = SO_STDERROR;
- }
- redirectStdStreams(&gStdwrap);
- switch (lpCmdline->dwCmdIndex) {
- case 1: /* Run Service as console application */
- if (!docmdDebugService(lpCmdline))
- rv = 3;
- break;
- case 2: /* Run Service */
- if (!docmdRunService(lpCmdline))
- rv = 4;
- break;
- case 3: /* Stop Service */
- if (!docmdStopService(lpCmdline))
- rv = 5;
- break;
- case 4: /* Update Service parameters */
- if (!docmdUpdateService(lpCmdline))
- rv = 6;
- break;
- case 5: /* Install Service */
- if (!docmdInstallService(lpCmdline))
- rv = 7;
- break;
- case 6: /* Delete Service */
- if (!docmdDeleteService(lpCmdline))
- rv = 8;
- break;
- default:
- /* Unknown command option */
- apxLogWrite(APXLOG_MARK_ERROR "Unknown command line option");
- printUsage(lpCmdline);
- rv = 99;
- break;
- }
-
-cleanup:
- apxLogWrite(APXLOG_MARK_INFO "Procrun finished.");
- if (lpCmdline)
- apxCmdlineFree(lpCmdline);
- if (_service_status_handle)
- CloseHandle(_service_status_handle);
- _service_status_handle = NULL;
- apxLogClose(NULL);
- apxHandleManagerDestroy();
- cleanupStdStreams(&gStdwrap);
- ExitProcess(rv);
-}
Copied: tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.c (from rev 2689, trunk/utils/windows/native/service/procrun/apps/prunsrv/prunsrv.c)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.c (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,1491 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/* ====================================================================
+ * prunsrv -- Service Runner.
+ * Contributed by Mladen Turk <mturk at apache.org>
+ * 05 Aug 2003
+ * ====================================================================
+ */
+
+/* Force the JNI vprintf functions */
+#define _DEBUG_JNI 1
+#include "apxwin.h"
+#include "prunsrv.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <stddef.h>
+#include <fcntl.h>
+#include <io.h> /* _open_osfhandle */
+
+#ifndef MIN
+#define MIN(a,b) (((a)<(b)) ? (a) : (b))
+#endif
+
+#define STDIN_FILENO 0
+#define STDOUT_FILENO 1
+#define STDERR_FILENO 2
+#define ONE_MINUTE (60 * 1000)
+
+#ifdef WIN64
+#define KREG_WOW6432 KEY_WOW64_32KEY
+#else
+#define KREG_WOW6432 0
+#endif
+
+typedef struct APX_STDWRAP {
+ LPCWSTR szLogPath;
+ LPCWSTR szStdOutFilename;
+ LPCWSTR szStdErrFilename;
+ FILE *fpStdOutFile;
+ FILE *fpStdErrFile;
+} APX_STDWRAP;
+
+/* Use static variables instead of #defines */
+static LPCWSTR PRSRV_AUTO = L"auto";
+static LPCWSTR PRSRV_JAVA = L"java";
+static LPCWSTR PRSRV_JVM = L"jvm";
+static LPCWSTR PRSRV_MANUAL = L"manual";
+static LPCWSTR PRSRV_JBIN = L"\\bin\\java.exe";
+static LPCWSTR PRSRV_SIGNAL = L"SIGNAL";
+
+static LPCWSTR STYPE_INTERACTIVE = L"interactive";
+
+static LPWSTR _service_name = NULL;
+/* Allowed procrun commands */
+static LPCWSTR _commands[] = {
+ L"TS", /* 1 Run Service as console application (default)*/
+ L"RS", /* 2 Run Service */
+ L"SS", /* 3 Stop Service */
+ L"US", /* 4 Update Service parameters */
+ L"IS", /* 5 Install Service */
+ L"DS", /* 6 Delete Service */
+ L"?", /* 7 Help */
+ L"VS", /* 8 Version */
+ NULL
+};
+
+/* Allowed procrun parameters */
+static APXCMDLINEOPT _options[] = {
+
+/* 0 */ { L"Description", L"Description", NULL, APXCMDOPT_STR | APXCMDOPT_SRV, NULL, 0},
+/* 1 */ { L"DisplayName", L"DisplayName", NULL, APXCMDOPT_STR | APXCMDOPT_SRV, NULL, 0},
+/* 2 */ { L"Install", L"ImagePath", NULL, APXCMDOPT_STE | APXCMDOPT_SRV, NULL, 0},
+/* 3 */ { L"ServiceUser", L"ServiceUser", NULL, APXCMDOPT_STR | APXCMDOPT_SRV, NULL, 0},
+/* 4 */ { L"ServicePassword", L"ServicePassword", NULL, APXCMDOPT_STR | APXCMDOPT_SRV, NULL, 0},
+/* 5 */ { L"Startup", L"Startup", NULL, APXCMDOPT_STR | APXCMDOPT_SRV, NULL, 0},
+/* 6 */ { L"Type", L"Type", NULL, APXCMDOPT_STR | APXCMDOPT_SRV, NULL, 0},
+
+/* 7 */ { L"DependsOn", L"DependsOn", NULL, APXCMDOPT_MSZ | APXCMDOPT_REG, NULL, 0},
+/* 8 */ { L"Environment", L"Environment", NULL, APXCMDOPT_MSZ | APXCMDOPT_REG, NULL, 0},
+/* 9 */ { L"User", L"User", NULL, APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
+/* 10 */ { L"Password", L"Password", NULL, APXCMDOPT_BIN | APXCMDOPT_REG, NULL, 0},
+
+/* 11 */ { L"JavaHome", L"JavaHome", L"Java", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
+/* 12 */ { L"Jvm", L"Jvm", L"Java", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
+/* 13 */ { L"JvmOptions", L"Options", L"Java", APXCMDOPT_MSZ | APXCMDOPT_REG, NULL, 0},
+/* 14 */ { L"Classpath", L"Classpath", L"Java", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
+/* 15 */ { L"JvmMs", L"JvmMs", L"Java", APXCMDOPT_INT | APXCMDOPT_REG, NULL, 0},
+/* 16 */ { L"JvmMx", L"JvmMx", L"Java", APXCMDOPT_INT | APXCMDOPT_REG, NULL, 0},
+/* 17 */ { L"JvmSs", L"JvmSs", L"Java", APXCMDOPT_INT | APXCMDOPT_REG, NULL, 0},
+
+/* 18 */ { L"StopImage", L"Image", L"Stop", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
+/* 19 */ { L"StopPath", L"WorkingPath", L"Stop", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
+/* 20 */ { L"StopClass", L"Class", L"Stop", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
+/* 21 */ { L"StopParams", L"Params", L"Stop", APXCMDOPT_MSZ | APXCMDOPT_REG, NULL, 0},
+/* 22 */ { L"StopMethod", L"Method", L"Stop", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
+/* 23 */ { L"StopMode", L"Mode", L"Stop", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
+/* 24 */ { L"StopTimeout", L"Timeout", L"Stop", APXCMDOPT_INT | APXCMDOPT_REG, NULL, 0},
+
+/* 25 */ { L"StartImage", L"Image", L"Start", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
+/* 26 */ { L"StartPath", L"WorkingPath", L"Start", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
+/* 27 */ { L"StartClass", L"Class", L"Start", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
+/* 28 */ { L"StartParams", L"Params", L"Start", APXCMDOPT_MSZ | APXCMDOPT_REG, NULL, 0},
+/* 29 */ { L"StartMethod", L"Method", L"Start", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
+/* 30 */ { L"StartMode", L"Mode", L"Start", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
+
+/* 31 */ { L"LogPath", L"Path", L"Log", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
+/* 32 */ { L"LogPrefix", L"Prefix", L"Log", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
+/* 33 */ { L"LogLevel", L"Level", L"Log", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
+/* 34 */ { L"StdError", L"StdError", L"Log", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
+/* 35 */ { L"StdOutput", L"StdOutput", L"Log", APXCMDOPT_STE | APXCMDOPT_REG, NULL, 0},
+/* 36 */ { L"LogJniMessages", L"LogJniMessages", L"Log", APXCMDOPT_INT | APXCMDOPT_REG, NULL, 1},
+/* 37 */ { L"PidFile", L"PidFile", L"Log", APXCMDOPT_STR | APXCMDOPT_REG, NULL, 0},
+ /* NULL terminate the array */
+ { NULL }
+};
+
+#define GET_OPT_V(x) _options[x].szValue
+#define GET_OPT_I(x) _options[x].dwValue
+#define GET_OPT_T(x) _options[x].dwType
+
+#define ST_DESCRIPTION GET_OPT_T(0)
+#define ST_DISPLAYNAME GET_OPT_T(1)
+#define ST_INSTALL GET_OPT_T(2)
+#define ST_SUSER GET_OPT_T(3)
+#define ST_SPASSWORD GET_OPT_T(4)
+#define ST_STARTUP GET_OPT_T(5)
+#define ST_TYPE GET_OPT_T(6)
+
+#define SO_DESCRIPTION GET_OPT_V(0)
+#define SO_DISPLAYNAME GET_OPT_V(1)
+#define SO_INSTALL GET_OPT_V(2)
+#define SO_SUSER GET_OPT_V(3)
+#define SO_SPASSWORD GET_OPT_V(4)
+#define SO_STARTUP GET_OPT_V(5)
+#define SO_TYPE GET_OPT_V(6)
+
+#define SO_DEPENDSON GET_OPT_V(7)
+#define SO_ENVIRONMENT GET_OPT_V(8)
+
+#define SO_USER GET_OPT_V(9)
+#define SO_PASSWORD GET_OPT_V(10)
+
+#define SO_JAVAHOME GET_OPT_V(11)
+#define SO_JVM GET_OPT_V(12)
+#define SO_JVMOPTIONS GET_OPT_V(13)
+#define SO_CLASSPATH GET_OPT_V(14)
+#define SO_JVMMS GET_OPT_I(15)
+#define SO_JVMMX GET_OPT_I(16)
+#define SO_JVMSS GET_OPT_I(17)
+
+#define SO_STOPIMAGE GET_OPT_V(18)
+#define SO_STOPPATH GET_OPT_V(19)
+#define SO_STOPCLASS GET_OPT_V(20)
+#define SO_STOPPARAMS GET_OPT_V(21)
+#define SO_STOPMETHOD GET_OPT_V(22)
+#define SO_STOPMODE GET_OPT_V(23)
+#define SO_STOPTIMEOUT GET_OPT_I(24)
+
+#define SO_STARTIMAGE GET_OPT_V(25)
+#define SO_STARTPATH GET_OPT_V(26)
+#define SO_STARTCLASS GET_OPT_V(27)
+#define SO_STARTPARAMS GET_OPT_V(28)
+#define SO_STARTMETHOD GET_OPT_V(29)
+#define SO_STARTMODE GET_OPT_V(30)
+
+#define SO_LOGPATH GET_OPT_V(31)
+#define SO_LOGPREFIX GET_OPT_V(32)
+#define SO_LOGLEVEL GET_OPT_V(33)
+
+#define SO_STDERROR GET_OPT_V(34)
+#define SO_STDOUTPUT GET_OPT_V(35)
+#define SO_JNIVFPRINTF GET_OPT_I(36)
+#define SO_PIDFILE GET_OPT_V(37)
+
+/* Main service table entry
+ * filled at run-time
+ */
+static SERVICE_TABLE_ENTRYW _service_table[] = {
+ {NULL, NULL},
+ {NULL, NULL}
+};
+
+static SERVICE_STATUS _service_status;
+static SERVICE_STATUS_HANDLE _service_status_handle = NULL;
+/* Set if launched by SCM */
+static BOOL _service_mode = FALSE;
+/* JVM used as worker */
+static BOOL _jni_startup = FALSE;
+/* JVM used for shutdown */
+static BOOL _jni_shutdown = FALSE;
+/* Java used as worker */
+static BOOL _java_startup = FALSE;
+/* Java used for shutdown */
+static BOOL _java_shutdown = FALSE;
+/* Global variables and objects */
+static APXHANDLE gPool;
+static APXHANDLE gWorker;
+static APX_STDWRAP gStdwrap; /* stdio/stderr redirection */
+
+static LPWSTR _jni_jvmpath = NULL; /* Path to jvm dll */
+static LPSTR _jni_jvmoptions = NULL; /* Path to jvm options */
+
+static LPSTR _jni_classpath = NULL;
+static LPCWSTR _jni_rparam = NULL; /* Startup arguments */
+static LPCWSTR _jni_sparam = NULL; /* Shutdown arguments */
+static LPSTR _jni_rmethod = NULL; /* Startup method */
+static LPSTR _jni_smethod = NULL; /* Shutdown method */
+static LPSTR _jni_rclass = NULL; /* Startup class */
+static LPSTR _jni_sclass = NULL; /* Shutdown class */
+
+static HANDLE gShutdownEvent = NULL;
+static HANDLE gSignalEvent = NULL;
+static HANDLE gSignalThread = NULL;
+static HANDLE gPidfileHandle = NULL;
+static LPWSTR gPidfileName = NULL;
+static BOOL gSignalValid = TRUE;
+
+DWORD WINAPI eventThread(LPVOID lpParam)
+{
+ for (;;) {
+ DWORD dw = WaitForSingleObject(gSignalEvent, INFINITE);
+ if (dw == WAIT_OBJECT_0 && gSignalValid) {
+ if (!GenerateConsoleCtrlEvent(CTRL_BREAK_EVENT, 0))
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ ResetEvent(gSignalEvent);
+ }
+ else
+ break;
+ }
+ ExitThread(0);
+ return 0;
+}
+
+/* redirect console stdout/stderr to files
+ * so that java messages can get logged
+ * If stderrfile is not specified it will
+ * go to stdoutfile.
+ */
+static BOOL redirectStdStreams(APX_STDWRAP *lpWrapper)
+{
+ BOOL aErr = FALSE;
+ BOOL aOut = FALSE;
+
+ if (lpWrapper->szStdOutFilename || lpWrapper->szStdErrFilename)
+ AllocConsole();
+ /* redirect to file or console */
+ if (lpWrapper->szStdOutFilename) {
+ if (lstrcmpiW(lpWrapper->szStdOutFilename, PRSRV_AUTO) == 0) {
+ aOut = TRUE;
+ lpWrapper->szStdOutFilename = apxLogFile(gPool,
+ lpWrapper->szLogPath,
+ L"service-stdout",
+ NULL, TRUE);
+ }
+ /* Delete the file if not in append mode
+ * XXX: See if we can use the params instead of that.
+ */
+ if (!aOut)
+ DeleteFileW(lpWrapper->szStdOutFilename);
+ if ((lpWrapper->fpStdOutFile = _wfopen(lpWrapper->szStdOutFilename,
+ L"a"))) {
+ _dup2(_fileno(lpWrapper->fpStdOutFile), 1);
+ *stdout = *lpWrapper->fpStdOutFile;
+ setvbuf(stdout, NULL, _IONBF, 0);
+ }
+ }
+ if (lpWrapper->szStdErrFilename) {
+ if (lstrcmpiW(lpWrapper->szStdErrFilename, PRSRV_AUTO) == 0) {
+ aErr = TRUE;
+ lpWrapper->szStdErrFilename = apxLogFile(gPool,
+ lpWrapper->szLogPath,
+ L"service-stderr",
+ NULL, TRUE);
+ }
+ if (!aErr)
+ DeleteFileW(lpWrapper->szStdErrFilename);
+ if ((lpWrapper->fpStdErrFile = _wfopen(lpWrapper->szStdErrFilename,
+ L"a"))) {
+ _dup2(_fileno(lpWrapper->fpStdErrFile), 2);
+ *stderr = *lpWrapper->fpStdErrFile;
+ setvbuf(stderr, NULL, _IONBF, 0);
+ }
+ }
+ else if (lpWrapper->fpStdOutFile) {
+ _dup2(_fileno(lpWrapper->fpStdOutFile), 2);
+ *stderr = *lpWrapper->fpStdOutFile;
+ setvbuf(stderr, NULL, _IONBF, 0);
+ }
+ return TRUE;
+}
+
+/* Debugging functions */
+static void printUsage(LPAPXCMDLINE lpCmdline, BOOL isHelp)
+{
+ int i = 0;
+ fwprintf(stderr, L"Usage: %s //CMD//Service [--options]\n",
+ lpCmdline->szExecutable);
+ fwprintf(stderr, L" Commands:\n");
+ if (isHelp)
+ fwprintf(stderr, L" //? This page\n");
+ fwprintf(stderr, L" //IS[//ServiceName] Install Service\n");
+ fwprintf(stderr, L" //US[//ServiceName] Update Service parameters\n");
+ fwprintf(stderr, L" //DS[//ServiceName] Delete Service\n");
+ fwprintf(stderr, L" //RS[//ServiceName] Run Service\n");
+ fwprintf(stderr, L" //SS[//ServiceName] Stop Service\n");
+ fwprintf(stderr, L" //TS[//ServiceName] Run Service as console application\n");
+ fwprintf(stderr, L" //PP[//Num Seconds] Sleep for n Seconds (defaults to 60)\n");
+ fwprintf(stderr, L" //VS Display version\n");
+ fwprintf(stderr, L" Options:\n");
+ while (_options[i].szName) {
+ fwprintf(stderr, L" --%s\n", _options[i].szName);
+ ++i;
+ }
+}
+
+static void printVersion(void)
+{
+#ifdef _WIN64
+ int b = 64;
+#else
+ int b = 32;
+#endif
+ fwprintf(stderr, L"Commons Daemon Service Runner version %S/Win%d (%S)\n",
+ PRG_VERSION, b, __DATE__);
+ fwprintf(stderr, L"Copyright (c) 2000-2010 The Apache Software Foundation.\n");
+}
+
+/* Display configuration parameters */
+static void dumpCmdline()
+{
+ int i = 0;
+ while (_options[i].szName) {
+ if (_options[i].dwType & APXCMDOPT_INT)
+ fwprintf(stderr, L"--%-16s %d\n", _options[i].szName,
+ _options[i].dwValue);
+ else if (_options[i].szValue)
+ fwprintf(stderr, L"--%-16s %s\n", _options[i].szName,
+ _options[i].szValue);
+ else
+ fwprintf(stderr, L"--%-16s <NULL>\n", _options[i].szName);
+ ++i;
+ }
+}
+
+static void setInprocEnvironment()
+{
+ LPWSTR p, e;
+
+ if (!SO_ENVIRONMENT)
+ return; /* Nothing to do */
+
+ for (p = SO_ENVIRONMENT; *p; p++) {
+ e = apxExpandStrW(gPool, p);
+ _wputenv(e);
+ apxFree(e);
+ while (*p)
+ p++;
+ }
+}
+
+/* Load the configuration from Registry
+ * loads only nonspecified items
+ */
+static BOOL loadConfiguration(LPAPXCMDLINE lpCmdline)
+{
+ APXHANDLE hRegistry;
+ int i = 0;
+
+ SetLastError(ERROR_SUCCESS);
+ hRegistry = apxCreateRegistryW(gPool, KEY_READ | KREG_WOW6432,
+ PRG_REGROOT,
+ lpCmdline->szApplication,
+ APXREG_SOFTWARE | APXREG_SERVICE);
+ if (IS_INVALID_HANDLE(hRegistry)) {
+ if (GetLastError() == ERROR_FILE_NOT_FOUND)
+ apxLogWrite(APXLOG_MARK_WARN "The system cannot find the Registry key for service '%S'",
+ lpCmdline->szApplication);
+ else
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return FALSE;
+ }
+ /* browse through options */
+ while (_options[i].szName) {
+ DWORD dwFrom;
+
+ dwFrom = (_options[i].dwType & APXCMDOPT_REG) ? APXREG_PARAMSOFTWARE : APXREG_SERVICE;
+ if (!(_options[i].dwType & APXCMDOPT_FOUND)) {
+ if (_options[i].dwType & APXCMDOPT_STR) {
+ _options[i].szValue = apxRegistryGetStringW(hRegistry,
+ dwFrom,
+ _options[i].szSubkey,
+ _options[i].szRegistry);
+ /* Expand environment variables */
+ if (_options[i].szValue && (_options[i].dwType & APXCMDOPT_STE)) {
+ LPWSTR exp = apxExpandStrW(gPool, _options[i].szValue);
+ if (exp != _options[i].szValue)
+ apxFree(_options[i].szValue);
+ _options[i].szValue = exp;
+ }
+ }
+ else if (_options[i].dwType & APXCMDOPT_INT) {
+ _options[i].dwValue = apxRegistryGetNumberW(hRegistry,
+ dwFrom,
+ _options[i].szSubkey,
+ _options[i].szRegistry);
+ }
+ else if (_options[i].dwType & APXCMDOPT_MSZ) {
+ _options[i].szValue = apxRegistryGetMzStrW(hRegistry,
+ dwFrom,
+ _options[i].szSubkey,
+ _options[i].szRegistry,
+ NULL,
+ &(_options[i].dwValue));
+ }
+ }
+ /* Merge the command line options with registry */
+ else if (_options[i].dwType & APXCMDOPT_ADD) {
+ LPWSTR cv = _options[i].szValue;
+ LPWSTR ov = NULL;
+ if (_options[i].dwType & APXCMDOPT_MSZ) {
+ ov = apxRegistryGetMzStrW(hRegistry, dwFrom,
+ _options[i].szSubkey,
+ _options[i].szRegistry,
+ NULL,
+ &(_options[i].dwValue));
+ _options[i].szValue = apxMultiSzCombine(gPool, ov, cv,
+ &(_options[i].dwValue));
+ if (ov)
+ apxFree(ov);
+ }
+ }
+ ++i;
+ }
+ apxCloseHandle(hRegistry);
+#ifdef _DEBUG
+ dumpCmdline();
+#endif
+ return TRUE;
+}
+
+/* Save changed configuration to registry
+ */
+static BOOL saveConfiguration(LPAPXCMDLINE lpCmdline)
+{
+ APXHANDLE hRegistry;
+ int i = 0;
+ hRegistry = apxCreateRegistryW(gPool, KEY_WRITE | KREG_WOW6432,
+ PRG_REGROOT,
+ lpCmdline->szApplication,
+ APXREG_SOFTWARE | APXREG_SERVICE);
+ if (IS_INVALID_HANDLE(hRegistry))
+ return FALSE;
+ /* TODO: Use array size */
+ while (_options[i].szName) {
+ /* Skip the service params */
+ if ((_options[i].dwType & APXCMDOPT_SRV) ||
+ !(_options[i].dwType & APXCMDOPT_FOUND)) {
+ /* Skip non-modified version */
+ }
+ /* Update only modified params */
+ else if (_options[i].dwType & APXCMDOPT_STR)
+ apxRegistrySetStrW(hRegistry, APXREG_PARAMSOFTWARE,
+ _options[i].szSubkey,
+ _options[i].szRegistry,
+ _options[i].szValue);
+ else if (_options[i].dwType & APXCMDOPT_INT)
+ apxRegistrySetNumW(hRegistry, APXREG_PARAMSOFTWARE,
+ _options[i].szSubkey,
+ _options[i].szRegistry,
+ _options[i].dwValue);
+ else if (_options[i].dwType & APXCMDOPT_MSZ)
+ apxRegistrySetMzStrW(hRegistry, APXREG_PARAMSOFTWARE,
+ _options[i].szSubkey,
+ _options[i].szRegistry,
+ _options[i].szValue,
+ _options[i].dwValue);
+ ++i;
+ }
+ apxCloseHandle(hRegistry);
+ return TRUE;
+}
+
+/* Operations */
+static BOOL docmdInstallService(LPAPXCMDLINE lpCmdline)
+{
+ APXHANDLE hService;
+ BOOL rv;
+ DWORD dwStart = SERVICE_DEMAND_START;
+ DWORD dwType = SERVICE_WIN32_OWN_PROCESS;
+ WCHAR szImage[SIZ_HUGLEN];
+
+ apxLogWrite(APXLOG_MARK_DEBUG "Installing service...");
+ hService = apxCreateService(gPool, SC_MANAGER_CREATE_SERVICE, FALSE);
+ if (IS_INVALID_HANDLE(hService)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Unable to open the Service Manager");
+ return FALSE;
+ }
+ /* Check the startup mode */
+ if ((ST_STARTUP & APXCMDOPT_FOUND) &&
+ lstrcmpiW(SO_STARTUP, PRSRV_AUTO) == 0)
+ dwStart = SERVICE_AUTO_START;
+ /* Check the service type */
+ if ((ST_TYPE & APXCMDOPT_FOUND) &&
+ lstrcmpiW(SO_TYPE, STYPE_INTERACTIVE) == 0)
+ dwType |= SERVICE_INTERACTIVE_PROCESS;
+
+ /* Check if --Install is provided */
+ if (!SO_INSTALL) {
+ lstrlcpyW(szImage, SIZ_HUGLEN, lpCmdline->szExePath);
+ lstrlcatW(szImage, SIZ_HUGLEN, L"\\");
+ lstrlcatW(szImage, SIZ_HUGLEN, lpCmdline->szExecutable);
+ lstrlcatW(szImage, SIZ_HUGLEN, L".exe");
+ }
+ else
+ lstrcpyW(szImage, SO_INSTALL);
+ /* Replace not needed qoutes */
+ apxStrQuoteInplaceW(szImage);
+ /* Add run-service command line option */
+ lstrlcatW(szImage, SIZ_HUGLEN, L" //RS//");
+ lstrlcatW(szImage, SIZ_HUGLEN, lpCmdline->szApplication);
+ SO_INSTALL = apxPoolStrdupW(gPool, szImage);
+ /* Ensure that option gets saved in the registry */
+ ST_INSTALL |= APXCMDOPT_FOUND;
+#ifdef _DEBUG
+ /* Display configured options */
+ dumpCmdline();
+#endif
+ apxLogWrite(APXLOG_MARK_INFO "Service %S name %S", lpCmdline->szApplication,
+ SO_DISPLAYNAME);
+ rv = apxServiceInstall(hService,
+ lpCmdline->szApplication,
+ SO_DISPLAYNAME, /* --DisplayName */
+ SO_INSTALL,
+ SO_DEPENDSON, /* --DependendsOn */
+ dwType,
+ dwStart);
+ /* Set the --Description */
+ if (rv) {
+ LPCWSTR sd = NULL;
+ LPCWSTR su = NULL;
+ LPCWSTR sp = NULL;
+ if (ST_DESCRIPTION & APXCMDOPT_FOUND) {
+ sd = SO_DESCRIPTION;
+ apxLogWrite(APXLOG_MARK_DEBUG "Setting service description %S",
+ SO_DESCRIPTION);
+ }
+ if (ST_SUSER & APXCMDOPT_FOUND) {
+ su = SO_SUSER;
+ apxLogWrite(APXLOG_MARK_DEBUG "Setting service user %S",
+ SO_SUSER);
+ }
+ if (ST_SPASSWORD & APXCMDOPT_FOUND) {
+ sp = SO_SPASSWORD;
+ apxLogWrite(APXLOG_MARK_DEBUG "Setting service password %S",
+ SO_SPASSWORD);
+ }
+ apxServiceSetNames(hService, NULL, NULL, sd, su, sp);
+ }
+ apxCloseHandle(hService);
+ if (rv) {
+ saveConfiguration(lpCmdline);
+ apxLogWrite(APXLOG_MARK_INFO "Service '%S' installed",
+ lpCmdline->szApplication);
+ }
+ else
+ apxLogWrite(APXLOG_MARK_ERROR "Failed installing '%S' service",
+ lpCmdline->szApplication);
+
+ return rv;
+}
+
+static BOOL docmdDeleteService(LPAPXCMDLINE lpCmdline)
+{
+ APXHANDLE hService;
+ BOOL rv = FALSE;
+
+ apxLogWrite(APXLOG_MARK_INFO "Deleting service...");
+ hService = apxCreateService(gPool, SC_MANAGER_CONNECT, FALSE);
+ if (IS_INVALID_HANDLE(hService)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Unable to open the Service Manager");
+ return FALSE;
+ }
+ /* Delete service will stop the service if running */
+ if (apxServiceOpen(hService, lpCmdline->szApplication, SERVICE_ALL_ACCESS)) {
+ WCHAR szWndManagerClass[SIZ_RESLEN];
+ HANDLE hWndManager = NULL;
+ lstrlcpyW(szWndManagerClass, SIZ_RESLEN, lpCmdline->szApplication);
+ lstrlcatW(szWndManagerClass, SIZ_RESLEN, L"_CLASS");
+ /* Close the monitor application if running */
+ if ((hWndManager = FindWindowW(szWndManagerClass, NULL)) != NULL) {
+ SendMessage(hWndManager, WM_CLOSE, 0, 0);
+ }
+ rv = apxServiceDelete(hService);
+ }
+ if (rv) {
+ /* Delete all service registry settings */
+ apxDeleteRegistryW(PRG_REGROOT, lpCmdline->szApplication, KREG_WOW6432, TRUE);
+ apxLogWrite(APXLOG_MARK_DEBUG "Service '%S' deleted",
+ lpCmdline->szApplication);
+ }
+ else {
+ apxDisplayError(FALSE, NULL, 0, "Unable to delete '%S' service",
+ lpCmdline->szApplication);
+ }
+ apxCloseHandle(hService);
+ apxLogWrite(APXLOG_MARK_INFO "Delete service finished.");
+ return rv;
+}
+
+static BOOL docmdStopService(LPAPXCMDLINE lpCmdline)
+{
+ APXHANDLE hService;
+ BOOL rv = FALSE;
+
+ apxLogWrite(APXLOG_MARK_INFO "Stopping service '%S' ...",
+ lpCmdline->szApplication);
+ hService = apxCreateService(gPool, GENERIC_ALL, FALSE);
+ if (IS_INVALID_HANDLE(hService)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Unable to open the Service Manager");
+ return FALSE;
+ }
+
+ SetLastError(ERROR_SUCCESS);
+ /* Open the service */
+ if (apxServiceOpen(hService, lpCmdline->szApplication,
+ GENERIC_READ | GENERIC_EXECUTE)) {
+ rv = apxServiceControl(hService,
+ SERVICE_CONTROL_STOP,
+ 0,
+ NULL,
+ NULL);
+ if (rv)
+ apxLogWrite(APXLOG_MARK_INFO "Service '%S' stopped",
+ lpCmdline->szApplication);
+ else
+ apxLogWrite(APXLOG_MARK_ERROR "Failed to stop '%S' service",
+ lpCmdline->szApplication);
+
+ }
+ else
+ apxDisplayError(FALSE, NULL, 0, "Unable to open '%S' service",
+ lpCmdline->szApplication);
+ apxCloseHandle(hService);
+ apxLogWrite(APXLOG_MARK_INFO "Stop service finished.");
+ return rv;
+}
+
+static BOOL docmdUpdateService(LPAPXCMDLINE lpCmdline)
+{
+ APXHANDLE hService;
+ BOOL rv = TRUE;
+
+ apxLogWrite(APXLOG_MARK_INFO "Updating service...");
+
+ hService = apxCreateService(gPool, SC_MANAGER_CREATE_SERVICE, FALSE);
+ if (IS_INVALID_HANDLE(hService)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Unable to open the Service Manager");
+ return FALSE;
+ }
+ SetLastError(0);
+ /* Open the service */
+ if (!apxServiceOpen(hService, lpCmdline->szApplication, SERVICE_ALL_ACCESS)) {
+ /* Close the existing manager handler.
+ * It will be reopened inside install.
+ */
+ apxCloseHandle(hService);
+ /* In case service doesn't exist try to install it.
+ * Install will fail if there is no minimum parameters required.
+ */
+ return docmdInstallService(lpCmdline);
+ }
+ else {
+ DWORD dwStart = SERVICE_NO_CHANGE;
+ DWORD dwType = SERVICE_NO_CHANGE;
+ LPCWSTR su = NULL;
+ LPCWSTR sp = NULL;
+ if (ST_SUSER & APXCMDOPT_FOUND) {
+ su = SO_SUSER;
+ apxLogWrite(APXLOG_MARK_DEBUG "Setting service user %S",
+ SO_SUSER);
+ }
+ if (ST_SPASSWORD & APXCMDOPT_FOUND) {
+ sp = SO_SPASSWORD;
+ apxLogWrite(APXLOG_MARK_DEBUG "Setting service password %S",
+ SO_SPASSWORD);
+ }
+ rv = (rv && apxServiceSetNames(hService,
+ NULL, /* Never update the ImagePath */
+ SO_DISPLAYNAME,
+ SO_DESCRIPTION,
+ su,
+ sp));
+ /* Update the --Startup mode */
+ if (ST_STARTUP & APXCMDOPT_FOUND) {
+ if (!lstrcmpiW(SO_STARTUP, PRSRV_AUTO))
+ dwStart = SERVICE_AUTO_START;
+ else if (!lstrcmpiW(SO_STARTUP, PRSRV_MANUAL))
+ dwStart = SERVICE_DEMAND_START;
+ }
+ if (ST_TYPE & APXCMDOPT_FOUND) {
+ if (!lstrcmpiW(SO_TYPE, STYPE_INTERACTIVE))
+ dwType = SERVICE_WIN32_OWN_PROCESS | SERVICE_INTERACTIVE_PROCESS;
+ }
+ rv = (rv && apxServiceSetOptions(hService,
+ dwType,
+ dwStart,
+ SERVICE_NO_CHANGE));
+
+ apxLogWrite(APXLOG_MARK_INFO "Service '%S' updated",
+ lpCmdline->szApplication);
+
+ rv = (rv && saveConfiguration(lpCmdline));
+ }
+ apxCloseHandle(hService);
+ if (rv)
+ apxLogWrite(APXLOG_MARK_INFO "Update service finished.");
+ else
+ apxLogWrite(APXLOG_MARK_INFO "Update service '%S' failed.",
+ lpCmdline->szApplication);
+ return rv;
+}
+
+
+/* Report the service status to the SCM
+ */
+int reportServiceStatus(DWORD dwCurrentState,
+ DWORD dwWin32ExitCode,
+ DWORD dwWaitHint)
+{
+ static DWORD dwCheckPoint = 1;
+ BOOL fResult = TRUE;
+
+ if (_service_mode && _service_status_handle) {
+ if (dwCurrentState == SERVICE_START_PENDING)
+ _service_status.dwControlsAccepted = 0;
+ else
+ _service_status.dwControlsAccepted = SERVICE_ACCEPT_STOP;
+
+ _service_status.dwCurrentState = dwCurrentState;
+ _service_status.dwWin32ExitCode = dwWin32ExitCode;
+ _service_status.dwWaitHint = dwWaitHint;
+
+ if ((dwCurrentState == SERVICE_RUNNING) ||
+ (dwCurrentState == SERVICE_STOPPED))
+ _service_status.dwCheckPoint = 0;
+ else
+ _service_status.dwCheckPoint = dwCheckPoint++;
+ fResult = SetServiceStatus(_service_status_handle, &_service_status);
+ if (!fResult) {
+ /* TODO: Deal with error */
+ }
+ }
+ return fResult;
+}
+
+
+BOOL child_callback(APXHANDLE hObject, UINT uMsg,
+ WPARAM wParam, LPARAM lParam)
+{
+ /* TODO: Make stdout and stderr buffers
+ * to prevent streams intermixing when there
+ * is no separate file for each stream
+ */
+ if (uMsg == WM_CHAR) {
+ int ch = LOWORD(wParam);
+ if (lParam)
+ fputc(ch, stderr);
+ else
+ fputc(ch, stdout);
+ }
+ return TRUE;
+}
+
+/* Executed when the service receives stop event */
+static DWORD WINAPI serviceStop(LPVOID lpParameter)
+{
+ APXHANDLE hWorker = NULL;
+ DWORD rv = 0;
+ BOOL wait_to_die = FALSE;
+ DWORD timeout = SO_STOPTIMEOUT * 1000;
+ DWORD dwCtrlType = (DWORD)((BYTE *)lpParameter - (BYTE *)0);
+
+ apxLogWrite(APXLOG_MARK_INFO "Stopping service...");
+
+ if (IS_INVALID_HANDLE(gWorker)) {
+ apxLogWrite(APXLOG_MARK_INFO "Worker is not defined");
+ return TRUE; /* Nothing to do */
+ }
+ if (_jni_shutdown) {
+ if (!SO_STARTPATH && SO_STOPPATH) {
+ /* If the Working path is specified change the current directory
+ * but only if the start path wasn't specified already.
+ */
+ SetCurrentDirectoryW(SO_STARTPATH);
+ }
+ hWorker = apxCreateJava(gPool, _jni_jvmpath);
+ if (IS_INVALID_HANDLE(hWorker)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Failed creating java %S", _jni_jvmpath);
+ return 1;
+ }
+ if (!apxJavaInitialize(hWorker, _jni_classpath, _jni_jvmoptions,
+ SO_JVMMS, SO_JVMMX, SO_JVMSS, SO_JNIVFPRINTF)) {
+ rv = 2;
+ apxLogWrite(APXLOG_MARK_ERROR "Failed initializing java %s", _jni_classpath);
+ goto cleanup;
+ }
+ if (!apxJavaLoadMainClass(hWorker, _jni_sclass, _jni_smethod, _jni_sparam)) {
+ rv = 2;
+ apxLogWrite(APXLOG_MARK_ERROR "Failed loading main %s class %s",
+ _jni_rclass, _jni_classpath);
+ goto cleanup;
+ }
+ /* Create sutdown event */
+ gShutdownEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
+ if (!apxJavaStart(hWorker)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Failed starting java");
+ rv = 3;
+ }
+ else {
+ apxLogWrite(APXLOG_MARK_DEBUG "Waiting for java jni stop worker to finish...");
+ apxJavaWait(hWorker, INFINITE, FALSE);
+ apxLogWrite(APXLOG_MARK_DEBUG "Java jni stop worker finished.");
+ }
+ wait_to_die = TRUE;
+ }
+ else if (SO_STOPMODE) { /* Only in case we have a stop mode */
+ DWORD nArgs;
+ LPWSTR *pArgs;
+ /* Redirect process */
+ hWorker = apxCreateProcessW(gPool,
+ 0,
+ child_callback,
+ SO_USER,
+ SO_PASSWORD,
+ FALSE);
+ if (IS_INVALID_HANDLE(hWorker)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Failed creating process");
+ return 1;
+ }
+ if (!apxProcessSetExecutableW(hWorker, SO_STOPIMAGE)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Failed setting process executable %S",
+ SO_STARTIMAGE);
+ rv = 2;
+ goto cleanup;
+ }
+ /* Assemble the command line */
+ if (_java_shutdown) {
+ nArgs = apxJavaCmdInitialize(gPool, SO_CLASSPATH, SO_STOPCLASS,
+ SO_JVMOPTIONS, SO_JVMMS, SO_JVMMX,
+ SO_JVMSS, SO_STOPPARAMS, &pArgs);
+ }
+ else {
+ nArgs = apxMultiSzToArrayW(gPool, SO_STOPPARAMS, &pArgs);
+ }
+
+ /* Pass the argv to child process */
+ if (!apxProcessSetCommandArgsW(hWorker, SO_STOPIMAGE,
+ nArgs, pArgs)) {
+ rv = 3;
+ apxLogWrite(APXLOG_MARK_ERROR "Failed setting process arguments (argc=%d)",
+ nArgs);
+ goto cleanup;
+ }
+ /* Set the working path */
+ if (!apxProcessSetWorkingPathW(hWorker, SO_STOPPATH)) {
+ rv = 4;
+ apxLogWrite(APXLOG_MARK_ERROR "Failed setting process working path to %S",
+ SO_STOPPATH);
+ goto cleanup;
+ }
+ /* Finally execute the child process
+ */
+ if (!apxProcessExecute(hWorker)) {
+ rv = 5;
+ apxLogWrite(APXLOG_MARK_ERROR "Failed executing process");
+ goto cleanup;
+ } else {
+ apxLogWrite(APXLOG_MARK_DEBUG "Waiting for stop worker to finish...");
+ apxHandleWait(hWorker, INFINITE, FALSE);
+ apxLogWrite(APXLOG_MARK_DEBUG "Stop worker finished.");
+ }
+ wait_to_die = TRUE;
+ }
+cleanup:
+ /* Close Java JNI handle or stop worker
+ * If this is the single JVM instance it will unload
+ * the JVM dll too.
+ * The worker will be closed on service exit.
+ */
+ if (!IS_INVALID_HANDLE(hWorker))
+ apxCloseHandle(hWorker);
+ if (gSignalEvent) {
+ gSignalValid = FALSE;
+ SetEvent(gSignalEvent);
+ WaitForSingleObject(gSignalThread, 1000);
+ CloseHandle(gSignalEvent);
+ CloseHandle(gSignalThread);
+ gSignalEvent = NULL;
+ }
+ SetEvent(gShutdownEvent);
+ if (timeout > 0x7FFFFFFF)
+ timeout = INFINITE; /* If the timeout was '-1' wait forewer */
+ if (wait_to_die && !timeout)
+ timeout = 300 * 1000; /* Use the 5 minute default shutdown */
+
+ if (dwCtrlType == SERVICE_CONTROL_SHUTDOWN)
+ timeout = MIN(timeout, apxGetMaxServiceTimeout(gPool));
+ reportServiceStatus(SERVICE_STOP_PENDING, NO_ERROR, timeout);
+
+ if (timeout) {
+ FILETIME fts, fte;
+ ULARGE_INTEGER s, e;
+ DWORD nms;
+ /* Wait to give it a chance to die naturally, then kill it. */
+ apxLogWrite(APXLOG_MARK_DEBUG "Waiting for worker to die naturally...");
+ GetSystemTimeAsFileTime(&fts);
+ rv = apxHandleWait(gWorker, timeout, TRUE);
+ GetSystemTimeAsFileTime(&fte);
+ s.LowPart = fts.dwLowDateTime;
+ s.HighPart = fts.dwHighDateTime;
+ e.LowPart = fte.dwLowDateTime;
+ e.HighPart = fte.dwHighDateTime;
+ nms = (DWORD)((e.QuadPart - s.QuadPart) / 10000);
+ if (rv == WAIT_OBJECT_0) {
+ rv = 0;
+ apxLogWrite(APXLOG_MARK_DEBUG "Worker finished gracefully in %d ms.", nms);
+ }
+ else
+ apxLogWrite(APXLOG_MARK_DEBUG "Worker was killed in %d ms.", nms);
+ }
+ else {
+ apxLogWrite(APXLOG_MARK_DEBUG "Sending WM_CLOSE to worker");
+ apxHandleSendMessage(gWorker, WM_CLOSE, 0, 0);
+ }
+
+ apxLogWrite(APXLOG_MARK_INFO "Service stopped.");
+ reportServiceStatus(SERVICE_STOPPED, NO_ERROR, 0);
+ return rv;
+}
+
+/* Executed when the service receives start event */
+static DWORD serviceStart()
+{
+ DWORD rv = 0;
+ DWORD nArgs;
+ LPWSTR *pArgs;
+ FILETIME fts;
+
+ apxLogWrite(APXLOG_MARK_INFO "Starting service...");
+
+ if (!IS_INVALID_HANDLE(gWorker)) {
+ apxLogWrite(APXLOG_MARK_INFO "Worker is not defined");
+ return TRUE; /* Nothing to do */
+ }
+ if (SO_PIDFILE) {
+ gPidfileName = apxLogFile(gPool, SO_LOGPATH, SO_PIDFILE, NULL, FALSE);
+ if (GetFileAttributesW(gPidfileName) != INVALID_FILE_ATTRIBUTES) {
+ /* Pid file exists */
+ apxLogWrite(APXLOG_MARK_ERROR "Pid file '%S' exists", gPidfileName);
+ return 1;
+ }
+ }
+ GetSystemTimeAsFileTime(&fts);
+ if (_jni_startup) {
+ if (SO_STARTPATH) {
+ /* If the Working path is specified change the current directory */
+ SetCurrentDirectoryW(SO_STARTPATH);
+ }
+ /* Set the environment using putenv, so JVM can use it */
+ setInprocEnvironment();
+ /* Create the JVM glbal worker */
+ gWorker = apxCreateJava(gPool, _jni_jvmpath);
+ if (IS_INVALID_HANDLE(gWorker)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Failed creating java %S", _jni_jvmpath);
+ return 1;
+ }
+ if (!apxJavaInitialize(gWorker, _jni_classpath, _jni_jvmoptions,
+ SO_JVMMS, SO_JVMMX, SO_JVMSS, SO_JNIVFPRINTF)) {
+ rv = 2;
+ apxLogWrite(APXLOG_MARK_ERROR "Failed initializing java %s", _jni_classpath);
+ goto cleanup;
+ }
+ if (!apxJavaLoadMainClass(gWorker, _jni_rclass, _jni_rmethod, _jni_rparam)) {
+ rv = 3;
+ apxLogWrite(APXLOG_MARK_ERROR "Failed loading main %s class %s", _jni_rclass, _jni_classpath);
+ goto cleanup;
+ }
+ apxJavaSetOut(gWorker, TRUE, gStdwrap.szStdErrFilename);
+ apxJavaSetOut(gWorker, FALSE, gStdwrap.szStdOutFilename);
+ if (!apxJavaStart(gWorker)) {
+ rv = 4;
+ apxLogWrite(APXLOG_MARK_ERROR "Failed to start Java");
+ goto cleanup;
+ }
+ apxLogWrite(APXLOG_MARK_DEBUG "Java started %s", _jni_rclass);
+ }
+ else {
+ /* Redirect process */
+ gWorker = apxCreateProcessW(gPool,
+ 0,
+ child_callback,
+ SO_USER,
+ SO_PASSWORD,
+ FALSE);
+ if (IS_INVALID_HANDLE(gWorker)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Failed to create process");
+ return 1;
+ }
+ if (!apxProcessSetExecutableW(gWorker, SO_STARTIMAGE)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Failed setting process executable %S",
+ SO_STARTIMAGE);
+ rv = 2;
+ goto cleanup;
+ }
+ /* Assemble the command line */
+ if (_java_startup) {
+ nArgs = apxJavaCmdInitialize(gPool, SO_CLASSPATH, SO_STARTCLASS,
+ SO_JVMOPTIONS, SO_JVMMS, SO_JVMMX,
+ SO_JVMSS, SO_STARTPARAMS, &pArgs);
+ }
+ else {
+ nArgs = apxMultiSzToArrayW(gPool, SO_STARTPARAMS, &pArgs);
+ }
+
+ /* Pass the argv to child process */
+ if (!apxProcessSetCommandArgsW(gWorker, SO_STARTIMAGE,
+ nArgs, pArgs)) {
+ rv = 3;
+ apxLogWrite(APXLOG_MARK_ERROR "Failed setting process arguments (argc=%d)",
+ nArgs);
+ goto cleanup;
+ }
+ /* Set the working path */
+ if (!apxProcessSetWorkingPathW(gWorker, SO_STARTPATH)) {
+ rv = 4;
+ apxLogWrite(APXLOG_MARK_ERROR "Failed setting process working path to %S",
+ SO_STARTPATH);
+ goto cleanup;
+ }
+ /* Finally execute the child process
+ */
+ if (!apxProcessExecute(gWorker)) {
+ rv = 5;
+ apxLogWrite(APXLOG_MARK_ERROR "Failed to execute process");
+ goto cleanup;
+ }
+ }
+ if (rv == 0) {
+ FILETIME fte;
+ ULARGE_INTEGER s, e;
+ DWORD nms;
+ /* Create pidfile */
+ if (gPidfileName) {
+ char pids[32];
+ gPidfileHandle = CreateFileW(gPidfileName,
+ GENERIC_READ | GENERIC_WRITE,
+ FILE_SHARE_READ,
+ NULL,
+ CREATE_NEW,
+ FILE_ATTRIBUTE_NORMAL,
+ NULL);
+
+ if (gPidfileHandle != INVALID_HANDLE_VALUE) {
+ DWORD wr = 0;
+ if (_jni_startup)
+ _snprintf(pids, 32, "%d\r\n", GetCurrentProcessId());
+ else
+ _snprintf(pids, 32, "%d\r\n", apxProcessGetPid(gWorker));
+ WriteFile(gPidfileHandle, pids, (DWORD)strlen(pids), &wr, NULL);
+ FlushFileBuffers(gPidfileName);
+ }
+ }
+ GetSystemTimeAsFileTime(&fte);
+ s.LowPart = fts.dwLowDateTime;
+ s.HighPart = fts.dwHighDateTime;
+ e.LowPart = fte.dwLowDateTime;
+ e.HighPart = fte.dwHighDateTime;
+ nms = (DWORD)((e.QuadPart - s.QuadPart) / 10000);
+ apxLogWrite(APXLOG_MARK_INFO "Service started in %d ms.", nms);
+ }
+ return rv;
+cleanup:
+ if (!IS_INVALID_HANDLE(gWorker))
+ apxCloseHandle(gWorker); /* Close the worker handle */
+ gWorker = NULL;
+ return rv;
+}
+
+/* Service control handler
+ */
+void WINAPI service_ctrl_handler(DWORD dwCtrlCode)
+{
+ DWORD threadId;
+ HANDLE stopThread;
+
+ switch (dwCtrlCode) {
+ case SERVICE_CONTROL_STOP:
+ reportServiceStatus(SERVICE_STOP_PENDING, NO_ERROR, 3000);
+ /* Stop the service asynchronously */
+ stopThread = CreateThread(NULL, 0,
+ serviceStop,
+ (LPVOID)SERVICE_CONTROL_STOP,
+ 0, &threadId);
+ WaitForSingleObject(stopThread, INFINITE);
+ CloseHandle(stopThread);
+
+ return;
+ case SERVICE_CONTROL_SHUTDOWN:
+ apxLogWrite(APXLOG_MARK_INFO "Service SHUTDOWN signaled");
+ reportServiceStatus(SERVICE_STOP_PENDING, NO_ERROR, 3000);
+ /* Stop the service asynchronously */
+ stopThread = CreateThread(NULL, 0,
+ serviceStop,
+ (LPVOID)SERVICE_CONTROL_SHUTDOWN,
+ 0, &threadId);
+ WaitForSingleObject(stopThread, INFINITE);
+ CloseHandle(stopThread);
+ return;
+ case SERVICE_CONTROL_INTERROGATE:
+ reportServiceStatus(_service_status.dwCurrentState,
+ _service_status.dwWin32ExitCode,
+ _service_status.dwWaitHint);
+ return;
+ default:
+ break;
+ }
+}
+
+/* Console control handler
+ *
+ */
+BOOL WINAPI console_handler(DWORD dwCtrlType)
+{
+ switch (dwCtrlType) {
+ case CTRL_BREAK_EVENT:
+ apxLogWrite(APXLOG_MARK_INFO "Console CTRL+BREAK event signaled");
+ return FALSE;
+ case CTRL_C_EVENT:
+ apxLogWrite(APXLOG_MARK_INFO "Console CTRL+C event signaled");
+ serviceStop((LPVOID)SERVICE_CONTROL_STOP);
+ return TRUE;
+ case CTRL_CLOSE_EVENT:
+ apxLogWrite(APXLOG_MARK_INFO "Console CTRL+CLOSE event signaled");
+ serviceStop((LPVOID)SERVICE_CONTROL_STOP);
+ return TRUE;
+ case CTRL_SHUTDOWN_EVENT:
+ apxLogWrite(APXLOG_MARK_INFO "Console SHUTDOWN event signaled");
+ serviceStop((LPVOID)SERVICE_CONTROL_SHUTDOWN);
+ return TRUE;
+ case CTRL_LOGOFF_EVENT:
+ apxLogWrite(APXLOG_MARK_INFO "Console LOGOFF event signaled");
+ if (!_service_mode) {
+ serviceStop((LPVOID)SERVICE_CONTROL_STOP);
+ }
+ return TRUE;
+ break;
+
+ }
+ return FALSE;
+}
+
+/* Main service execution loop */
+void WINAPI serviceMain(DWORD argc, LPTSTR *argv)
+{
+ DWORD rc;
+ _service_status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
+ _service_status.dwCurrentState = SERVICE_START_PENDING;
+ _service_status.dwControlsAccepted = SERVICE_ACCEPT_STOP |
+ SERVICE_ACCEPT_PAUSE_CONTINUE;
+ _service_status.dwWin32ExitCode = 0;
+ _service_status.dwCheckPoint = 0;
+ _service_status.dwWaitHint = 0;
+ _service_status.dwServiceSpecificExitCode = 0;
+
+ apxLogWrite(APXLOG_MARK_DEBUG "Inside ServiceMain...");
+
+ if (_service_name) {
+ WCHAR en[SIZ_HUGLEN];
+ int i;
+ PSECURITY_ATTRIBUTES sa = GetNullACL();
+ lstrlcpyW(en, SIZ_DESLEN, L"Global\\");
+ lstrlcatW(en, SIZ_DESLEN, _service_name);
+ lstrlcatW(en, SIZ_DESLEN, PRSRV_SIGNAL);
+ for (i = 7; i < lstrlenW(en); i++) {
+ if (en[i] >= L'a' && en[i] <= L'z')
+ en[i] = en[i] - 32;
+ }
+ gSignalEvent = CreateEventW(sa, TRUE, FALSE, en);
+ CleanNullACL((void *)sa);
+
+ if (gSignalEvent) {
+ DWORD tid;
+ gSignalThread = CreateThread(NULL, 0, eventThread, NULL, 0, &tid);
+ }
+ }
+ /* Check the StartMode */
+ if (SO_STARTMODE) {
+ if (!lstrcmpiW(SO_STARTMODE, PRSRV_JVM)) {
+ _jni_startup = TRUE;
+ _jni_rclass = WideToUTF8(SO_STARTCLASS);
+ /* Exchange all dots with slashes */
+ apxStrCharReplaceA(_jni_rclass, '.', '/');
+ _jni_rparam = SO_STARTPARAMS;
+ }
+ else if (!lstrcmpiW(SO_STARTMODE, PRSRV_JAVA)) {
+ LPWSTR jx = NULL, szJH = SO_JAVAHOME;
+ if (!szJH)
+ szJH = apxGetJavaSoftHome(gPool, FALSE);
+ if (szJH) {
+ jx = apxPoolAlloc(gPool, (lstrlenW(szJH) + 16) * sizeof(WCHAR));
+ lstrcpyW(jx, szJH);
+ lstrcatW(jx, PRSRV_JBIN);
+ SO_STARTPATH = szJH;
+ }
+ else {
+ apxLogWrite(APXLOG_MARK_ERROR "Unable to find Java Runtime Environment.");
+ goto cleanup;
+ }
+ _java_startup = TRUE;
+ /* StartImage now contains the full path to the java.exe */
+ SO_STARTIMAGE = jx;
+ }
+ }
+ /* Check the StopMode */
+ if (SO_STOPMODE) {
+ if (!lstrcmpiW(SO_STOPMODE, PRSRV_JVM)) {
+ _jni_shutdown = TRUE;
+ _jni_sclass = WideToUTF8(SO_STOPCLASS);
+ apxStrCharReplaceA(_jni_sclass, '.', '/');
+ _jni_sparam = SO_STOPPARAMS;
+ }
+ else if (!lstrcmpiW(SO_STOPMODE, PRSRV_JAVA)) {
+ LPWSTR jx = NULL, szJH = SO_JAVAHOME;
+ if (!szJH)
+ szJH = apxGetJavaSoftHome(gPool, FALSE);
+ if (szJH) {
+ jx = apxPoolAlloc(gPool, (lstrlenW(szJH) + 16) * sizeof(WCHAR));
+ lstrcpyW(jx, szJH);
+ lstrcatW(jx, PRSRV_JBIN);
+ SO_STOPPATH = szJH;
+ }
+ else {
+ apxLogWrite(APXLOG_MARK_ERROR "Unable to find Java Runtime Environment.");
+ goto cleanup;
+ }
+ _java_shutdown = TRUE;
+ /* StopImage now contains the full path to the java.exe */
+ SO_STOPIMAGE = jx;
+ }
+ }
+ /* Find the classpath */
+ if (_jni_shutdown || _jni_startup) {
+ if (SO_JVM) {
+ if (lstrcmpW(SO_JVM, PRSRV_AUTO))
+ _jni_jvmpath = SO_JVM;
+ }
+ if (SO_CLASSPATH)
+ _jni_classpath = WideToUTF8(SO_CLASSPATH);
+ if (SO_STARTMETHOD)
+ _jni_rmethod = WideToAscii(SO_STARTMETHOD, (LPSTR)SO_STARTMETHOD);
+ if (SO_STOPMETHOD)
+ _jni_smethod = WideToAscii(SO_STOPMETHOD, (LPSTR)SO_STOPMETHOD);
+ _jni_jvmoptions = MzWideToUTF8(SO_JVMOPTIONS);
+ }
+ if (_service_mode) {
+ /* Register Service Control handler */
+ _service_status_handle = RegisterServiceCtrlHandlerW(_service_name,
+ service_ctrl_handler);
+ if (IS_INVALID_HANDLE(_service_status_handle)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Failed to register Service Control for %S",
+ _service_name);
+ goto cleanup;
+ }
+ /* Allocate console so that events gets processed */
+ AllocConsole();
+ }
+ reportServiceStatus(SERVICE_START_PENDING, NO_ERROR, 3000);
+ if ((rc = serviceStart()) == 0) {
+ /* Service is started */
+ DWORD rv;
+ reportServiceStatus(SERVICE_RUNNING, NO_ERROR, 0);
+ apxLogWrite(APXLOG_MARK_DEBUG "Waiting for worker to finish...");
+ /* Set console handler to capture CTRL events */
+ SetConsoleCtrlHandler((PHANDLER_ROUTINE)console_handler, TRUE);
+
+ rv = apxHandleWait(gWorker, INFINITE, FALSE);
+ apxLogWrite(APXLOG_MARK_DEBUG "Worker finished.");
+ }
+ else {
+ apxLogWrite(APXLOG_MARK_ERROR "ServiceStart returned %d", rc);
+ goto cleanup;
+ }
+ if (gShutdownEvent) {
+ reportServiceStatus(SERVICE_STOP_PENDING, NO_ERROR, 0);
+ /* Ensure that shutdown thread exits before us */
+ apxLogWrite(APXLOG_MARK_DEBUG "Waiting for ShutdownEvent");
+ WaitForSingleObject(gShutdownEvent, ONE_MINUTE);
+ apxLogWrite(APXLOG_MARK_DEBUG "ShutdownEvent signaled");
+ CloseHandle(gShutdownEvent);
+ /* This will cause to wait for all threads to exit
+ */
+ apxLogWrite(APXLOG_MARK_DEBUG "Waiting 1 minute for all threads to exit");
+ apxDestroyJvm(ONE_MINUTE);
+ }
+ else {
+ /* We came here without shutdown event
+ * Probably because main() returned without ensuring all threads
+ * have finished
+ */
+ apxLogWrite(APXLOG_MARK_DEBUG "Waiting for all threads to exit");
+ apxDestroyJvm(INFINITE);
+ reportServiceStatus(SERVICE_STOP_PENDING, NO_ERROR, 0);
+ }
+ apxLogWrite(APXLOG_MARK_DEBUG "JVM destroyed.");
+ reportServiceStatus(SERVICE_STOPPED, NO_ERROR, 0);
+
+ return;
+cleanup:
+ /* Cleanup */
+ reportServiceStatus(SERVICE_STOPPED, ERROR_SERVICE_SPECIFIC_ERROR, 0);
+ return;
+}
+
+
+/* Run the service in the debug mode */
+BOOL docmdDebugService(LPAPXCMDLINE lpCmdline)
+{
+ BOOL rv = FALSE;
+
+ _service_mode = FALSE;
+ _service_name = lpCmdline->szApplication;
+ apxLogWrite(APXLOG_MARK_INFO "Debugging '%S' Service...", _service_name);
+ serviceMain(0, NULL);
+ apxLogWrite(APXLOG_MARK_INFO "Debug service finished.");
+ SAFE_CLOSE_HANDLE(gPidfileHandle);
+ if (gPidfileName)
+ DeleteFileW(gPidfileName);
+ return rv;
+}
+
+BOOL docmdRunService(LPAPXCMDLINE lpCmdline)
+{
+ BOOL rv = FALSE;
+
+ _service_mode = TRUE;
+ _service_name = lpCmdline->szApplication;
+ apxLogWrite(APXLOG_MARK_INFO "Running '%S' Service...", _service_name);
+ _service_table[0].lpServiceName = lpCmdline->szApplication;
+ _service_table[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTIONW)serviceMain;
+ rv = (StartServiceCtrlDispatcherW(_service_table) != 0);
+ apxLogWrite(APXLOG_MARK_INFO "Run service finished.");
+ SAFE_CLOSE_HANDLE(gPidfileHandle);
+ if (gPidfileName)
+ DeleteFileW(gPidfileName);
+ return rv;
+}
+
+void __cdecl main(int argc, char **argv)
+{
+ UINT rv = 0;
+
+ LPAPXCMDLINE lpCmdline;
+
+ if (argc > 1 && strncmp(argv[1], "//PP", 4) == 0) {
+ /* Handy sleep routine defaulting to 1 minute */
+ DWORD ss = 60;
+ if (argv[1][4] && argv[1][5] && argv[1][6]) {
+ int us = atoi(argv[1] + 6);
+ if (us > 0)
+ ss = (DWORD)us;
+ }
+ Sleep(ss * 1000);
+ ExitProcess(0);
+ return;
+ }
+ apxHandleManagerInitialize();
+ /* Create the main Pool */
+ gPool = apxPoolCreate(NULL, 0);
+
+ /* Parse the command line */
+ if ((lpCmdline = apxCmdlineParse(gPool, _options, _commands)) == NULL) {
+ apxLogWrite(APXLOG_MARK_ERROR "Invalid command line arguments");
+ rv = 1;
+ goto cleanup;
+ }
+ apxCmdlineLoadEnvVars(lpCmdline);
+ if (lpCmdline->dwCmdIndex < 5) {
+ if (!loadConfiguration(lpCmdline) &&
+ lpCmdline->dwCmdIndex < 4) {
+ apxLogWrite(APXLOG_MARK_ERROR "Load configuration failed");
+ rv = 2;
+ goto cleanup;
+ }
+ }
+
+ apxLogOpen(gPool, SO_LOGPATH, SO_LOGPREFIX);
+ apxLogLevelSetW(NULL, SO_LOGLEVEL);
+ apxLogWrite(APXLOG_MARK_DEBUG "Commons Daemon procrun log initialized");
+ apxLogWrite(APXLOG_MARK_INFO "Commons Daemon procrun (%s) started", PRG_VERSION);
+
+ AplZeroMemory(&gStdwrap, sizeof(APX_STDWRAP));
+
+ gStdwrap.szLogPath = SO_LOGPATH;
+ /* In debug mode allways use console */
+ if (lpCmdline->dwCmdIndex != 1) {
+ gStdwrap.szStdOutFilename = SO_STDOUTPUT;
+ gStdwrap.szStdErrFilename = SO_STDERROR;
+ }
+ redirectStdStreams(&gStdwrap);
+ if (lpCmdline->dwCmdIndex == 2) {
+ SYSTEMTIME t;
+ GetLocalTime(&t);
+ fprintf(stdout, "\n%d-%02d-%02d %02d:%02d:%02d "
+ "Commons Daemon procrun stdout initialized\n",
+ t.wYear, t.wMonth, t.wDay,
+ t.wHour, t.wMinute, t.wSecond);
+ fprintf(stderr, "\n%d-%02d-%02d %02d:%02d:%02d "
+ "Commons Daemon procrun stderr initialized\n",
+ t.wYear, t.wMonth, t.wDay,
+ t.wHour, t.wMinute, t.wSecond);
+ }
+ switch (lpCmdline->dwCmdIndex) {
+ case 1: /* Run Service as console application */
+ if (!docmdDebugService(lpCmdline))
+ rv = 3;
+ break;
+ case 2: /* Run Service */
+ if (!docmdRunService(lpCmdline))
+ rv = 4;
+ break;
+ case 3: /* Stop Service */
+ if (!docmdStopService(lpCmdline))
+ rv = 5;
+ break;
+ case 4: /* Update Service parameters */
+ if (!docmdUpdateService(lpCmdline))
+ rv = 6;
+ break;
+ case 5: /* Install Service */
+ if (!docmdInstallService(lpCmdline))
+ rv = 7;
+ break;
+ case 6: /* Delete Service */
+ if (!docmdDeleteService(lpCmdline))
+ rv = 8;
+ break;
+ case 7: /* Print Usage and exit */
+ printUsage(lpCmdline, TRUE);
+ break;
+ case 8: /* Print version and exit */
+ printVersion();
+ break;
+ default:
+ /* Unknown command option */
+ apxLogWrite(APXLOG_MARK_ERROR "Unknown command line option");
+ printUsage(lpCmdline, FALSE);
+ rv = 99;
+ break;
+ }
+
+cleanup:
+ if (rv)
+ apxLogWrite(APXLOG_MARK_ERROR "Commons Daemon procrun failed "
+ "with exit value: %d", rv);
+ else
+ apxLogWrite(APXLOG_MARK_INFO "Commons Daemon procrun finished");
+ if (lpCmdline)
+ apxCmdlineFree(lpCmdline);
+ if (_service_status_handle)
+ CloseHandle(_service_status_handle);
+ _service_status_handle = NULL;
+ _flushall();
+ apxLogClose(NULL);
+ apxHandleManagerDestroy();
+ ExitProcess(rv);
+}
Deleted: tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.h
===================================================================
--- trunk/utils/windows/native/service/procrun/apps/prunsrv/prunsrv.h 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.h 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,33 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-/* ====================================================================
- * jar2exe -- convert .jar file to WIN32 executable.
- * Contributed by Mladen Turk <mturk at apache.org>
- * 05 Aug 2003
- * ====================================================================
- */
-
-#ifndef _PRUNSRV_H
-#define _PRUNSRV_H
-
-#undef PRG_VERSION
-#define PRG_VERSION "2.0.4.1"
-#define CSV_VERSION 2,0,4,1
-#define PRG_REGROOT L"JBoss\\JBoss Web 2"
-
-#endif /* _PRUNSRV_H */
-
Copied: tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.h (from rev 2689, trunk/utils/windows/native/service/procrun/apps/prunsrv/prunsrv.h)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.h (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.h 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,33 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/* ====================================================================
+ * jar2exe -- convert .jar file to WIN32 executable.
+ * Contributed by Mladen Turk <mturk at apache.org>
+ * 05 Aug 2003
+ * ====================================================================
+ */
+
+#ifndef _PRUNSRV_H
+#define _PRUNSRV_H
+
+#undef PRG_VERSION
+#define PRG_VERSION "2.0.5.1"
+#define CSV_VERSION 2,0,5,1
+#define PRG_REGROOT L"JBoss\\JBoss Web 2"
+
+#endif /* _PRUNSRV_H */
+
Deleted: tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.rc
===================================================================
--- trunk/utils/windows/native/service/procrun/apps/prunsrv/prunsrv.rc 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.rc 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,65 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#include "apxwin.h"
-#include "prunsrv.h"
-
-#define STR_COPYRIGHT "Copyright © 2007 Red Hat Middleware, LLC. " \
- "or its licensors, as applicable."
-
-#define STR_LICENSE "Distributable under LGPL license. " \
- "See terms of license at gnu.org."
-
-#define STR_COMPANY "Red Hat®, Inc."
-#define STR_TRADEMARK "™ Red Hat Inc."
-#define STR_PRODUCT "JBoss Web Service wrapper"
-
-IDI_MAINICON ICON "../../resources/jboss.ico"
-IDI_REDHATICON ICON "../../resources/jboss.ico"
-
-1 VERSIONINFO
- FILEVERSION CSV_VERSION
- PRODUCTVERSION CSV_VERSION
- FILEFLAGSMASK 0x3fL
-#if defined(_DEBUG)
- FILEFLAGS 0x03L
-#else
- FILEFLAGS 0x02L
-#endif
- FILEOS 0x40004L
- FILETYPE 0x1L
- FILESUBTYPE 0x0L
-BEGIN
- BLOCK "StringFileInfo"
- BEGIN
- BLOCK "040904b0"
- BEGIN
- VALUE "Comments", STR_COPYRIGHT "\0"
- VALUE "CompanyName", STR_COMPANY "\0"
- VALUE "FileDescription", STR_PRODUCT "\0"
- VALUE "FileVersion", PRG_VERSION
- VALUE "InternalName", STR_PRODUCT "\0"
- VALUE "LegalCopyright", STR_COPYRIGHT "\0"
- VALUE "OriginalFilename", "jbossweb.exe\0"
- VALUE "ProductName", STR_PRODUCT "\0"
- VALUE "ProductVersion", PRG_VERSION
- END
- END
- BLOCK "VarFileInfo"
- BEGIN
- VALUE "Translation", 0x409, 1200
- END
-END
Copied: tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.rc (from rev 2689, trunk/utils/windows/native/service/procrun/apps/prunsrv/prunsrv.rc)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.rc (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/apps/prunsrv/prunsrv.rc 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,65 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "apxwin.h"
+#include "prunsrv.h"
+
+#define STR_COPYRIGHT "Copyright � 2010 Red Hat, Inc. " \
+ "or its licensors, as applicable."
+
+#define STR_LICENSE "Distributable under LGPL license. " \
+ "See terms of license at gnu.org."
+
+#define STR_COMPANY "Red Hat�, Inc."
+#define STR_TRADEMARK "� Red Hat Inc."
+#define STR_PRODUCT "JBoss Web Service wrapper"
+
+IDI_MAINICON ICON "../../resources/jboss.ico"
+IDI_REDHATICON ICON "../../resources/jboss.ico"
+
+1 VERSIONINFO
+ FILEVERSION CSV_VERSION
+ PRODUCTVERSION CSV_VERSION
+ FILEFLAGSMASK 0x3fL
+#if defined(_DEBUG)
+ FILEFLAGS 0x03L
+#else
+ FILEFLAGS 0x02L
+#endif
+ FILEOS 0x40004L
+ FILETYPE 0x1L
+ FILESUBTYPE 0x0L
+BEGIN
+ BLOCK "StringFileInfo"
+ BEGIN
+ BLOCK "040904b0"
+ BEGIN
+ VALUE "Comments", STR_COPYRIGHT "\0"
+ VALUE "CompanyName", STR_COMPANY "\0"
+ VALUE "FileDescription", STR_PRODUCT "\0"
+ VALUE "FileVersion", PRG_VERSION
+ VALUE "InternalName", STR_PRODUCT "\0"
+ VALUE "LegalCopyright", STR_COPYRIGHT "\0"
+ VALUE "OriginalFilename", "jbossweb.exe\0"
+ VALUE "ProductName", STR_PRODUCT "\0"
+ VALUE "ProductVersion", PRG_VERSION
+ END
+ END
+ BLOCK "VarFileInfo"
+ BEGIN
+ VALUE "Translation", 0x409, 1200
+ END
+END
Deleted: tags/jbosssvc/2.0.8/procrun/src/cmdline.c
===================================================================
--- trunk/utils/windows/native/service/procrun/src/cmdline.c 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/src/cmdline.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,295 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#include "apxwin.h"
-#include "private.h"
-
-#define EXE_SUFFIX L".EXE"
-#define EXE_SUFFIXLEN (sizeof(EXE_SUFFIX) / sizeof(WCHAR) - 1)
-
-#define X86_SUFFIX L".X86"
-#define X64_SUFFIX L".X64"
-#define A64_SUFFIX L".I64"
-
-/* Those two are declared in handles.c */
-extern LPWSTR *_st_sys_argvw;
-extern int _st_sys_argc;
-
-static WCHAR _st_sys_appexe[MAX_PATH];
-/*
- * argv parsing.
- * Parse the argv[0] and split to ExePath and
- * Executable name. Strip the extension ('.exe').
- * Check for command in argv[1] //CMD//Application
- * Parse the options --option value or --option==value
- * break on first argument that doesn't start with '--'
- */
-LPAPXCMDLINE apxCmdlineParse(
- APXHANDLE hPool,
- APXCMDLINEOPT *lpOptions,
- LPCWSTR *lpszCommands)
-{
-
- LPAPXCMDLINE lpCmdline = NULL;
- DWORD l, i = 0, s = 1;
- LPWSTR p;
- WCHAR cmd[4];
- DWORD match;
- BOOL add;
- if (_st_sys_argc < 1)
- return NULL;
-
- l = lstrlenW(_st_sys_argvw[0]);
- if (l < EXE_SUFFIXLEN)
- return NULL;
- if (!(lpCmdline = (LPAPXCMDLINE)apxPoolCalloc(hPool, sizeof(APXCMDLINE))))
- return NULL;
- lpCmdline->hPool = hPool;
- lpCmdline->lpOptions = lpOptions;
- lpCmdline->szExePath = _st_sys_argvw[0];
- p = lpCmdline->szExePath + l;
-
- while (p > lpCmdline->szExePath) {
- if (*p == L'\\') {
- *p++ = L'\0';
- break;
- }
- p--;
- }
- /* Find the path if it wasn't specified in the argv[0] */
- if (p == lpCmdline->szExePath) {
- WCHAR mh[4096];
- LPWSTR m;
- if (GetModuleFileNameW(GetModuleHandle(NULL), mh, 4096)) {
- lpCmdline->szExePath = apxPoolStrdupW(hPool, mh);
- m = lpCmdline->szExePath + lstrlenW(lpCmdline->szExePath);
- while (m > lpCmdline->szExePath) {
- if (*(m--) == L'\\') {
- *(++m) = L'\0';
- break;
- }
- }
- }
- }
- lpCmdline->szExecutable = p;
- p = _st_sys_argvw[0] + l - EXE_SUFFIXLEN;
- if (lstrcmpiW(p, EXE_SUFFIX) == 0)
- *p = L'\0';
- /* Strip CPU specific suffixes */
- l = lstrlenW(_st_sys_argvw[0]);
- if (l > EXE_SUFFIXLEN) {
- p = _st_sys_argvw[0] + l - EXE_SUFFIXLEN;
- if (lstrcmpiW(p, X86_SUFFIX) == 0) {
- *p = L'\0';
- }
- else if (lstrcmpiW(p, X64_SUFFIX) == 0) {
- *p = L'\0';
- }
- else if (lstrcmpiW(p, A64_SUFFIX) == 0) {
- *p = L'\0';
- }
- }
- if (lpszCommands && _st_sys_argc > 1 && lstrlenW(_st_sys_argvw[1]) > 5) {
- if (_st_sys_argvw[1][0] == L'/' &&
- _st_sys_argvw[1][1] == L'/' &&
- _st_sys_argvw[1][5] == L'/') { /* allow max tree char command */
- l = 2;
- while (_st_sys_argvw[1][l] != L'/') {
- cmd[l - 2] = _st_sys_argvw[1][l];
- if (l++ > 3)
- break;
- }
- cmd[l - 2] = L'\0';
- while (lpszCommands[i]) {
- if (lstrcmpW(lpszCommands[i++], cmd) == 0) {
- lpCmdline->dwCmdIndex = i;
- s = 2;
- break;
- }
- }
- if (lpCmdline->dwCmdIndex) {
- _st_sys_argvw[1][l+1] = L'\0';
- lpCmdline->szApplication = &(_st_sys_argvw[1][l+2]);
- if (!lstrlenW(lpCmdline->szApplication))
- lpCmdline->szApplication = _st_sys_argvw[0];
- }
- else {
- apxLogWrite(APXLOG_MARK_ERROR "Unrecognized cmd option %s", cmd);
- s = 2;
- }
- }
- }
- else {
- lpCmdline->szApplication = lpCmdline->szExecutable;
- lpCmdline->dwCmdIndex = 1;
- return lpCmdline;
- }
- for (i = s; i < (DWORD)_st_sys_argc; i++) {
- LPWSTR e = NULL;
- if (_st_sys_argvw[i][0] != L'-' ||
- _st_sys_argvw[i][1] != L'-') {
- if (_st_sys_argvw[i][0] != L'+' ||
- _st_sys_argvw[i][1] != L'+')
- break;
- else
- add = TRUE;
- }
- else
- add = FALSE;
- p = &(_st_sys_argvw[i][2]);
- /* Find if the option has '=' char
- * for --option==value or --option value cases.
- */
- while (*p) {
- if (*p == L'=') {
- *p = L'\0';
- e = p + 1;
- break;
- }
- else
- p++;
- }
- match = 0;
- for (l = 0; lpOptions[l].szName; l++) {
- if (lstrcmpW(lpOptions[l].szName, &(_st_sys_argvw[i][2])) == 0) {
- LPWSTR val;
- /* check if arg is needed */
- if (e)
- val = e;
- else if ((i + 1) < (DWORD)_st_sys_argc)
- val = _st_sys_argvw[++i];
- else {
- lpOptions[l].dwValue = 0;
- lpOptions[l].szValue = NULL;
- lpOptions[l].dwType |= APXCMDOPT_FOUND;
- break;
- }
- if (lpOptions[l].dwType & APXCMDOPT_STR)
- lpOptions[l].szValue = val;
- else if (lpOptions[l].dwType & APXCMDOPT_INT)
- lpOptions[l].dwValue = (DWORD)apxAtoulW(val);
- else if (lpOptions[l].dwType & APXCMDOPT_MSZ) {
- LPWSTR pp;
- BOOL insquote = FALSE, indquote=FALSE;
- DWORD sp = 0;
- LPWSTR ov = lpOptions[l].szValue;
- if (lpOptions[l].dwValue > 2) {
- sp = (lpOptions[l].dwValue - sizeof(WCHAR)) / sizeof(WCHAR);
- }
- lpOptions[l].dwValue = (sp + lstrlenW(val) + 2) * sizeof(WCHAR);
- lpOptions[l].szValue = (LPWSTR)apxPoolCalloc(hPool,
- lpOptions[l].dwValue);
- if (sp) {
- AplMoveMemory(lpOptions[l].szValue, ov, sp * sizeof(WCHAR));
- apxFree(ov);
- }
- pp = val;
- while(*pp) {
- if (*pp == L'\'')
- insquote = !insquote;
- else if (*pp == L'"') {
- indquote = !indquote;
- lpOptions[l].szValue[sp++] = L'"';
- }
- else if ((*pp == L'#' || *pp == L';') && !insquote && !indquote)
- lpOptions[l].szValue[sp++] = L'\0';
- else
- lpOptions[l].szValue[sp++] = *pp;
- pp++;
- }
- }
- lpOptions[l].dwType |= APXCMDOPT_FOUND;
- if (add)
- lpOptions[l].dwType |= APXCMDOPT_ADD;
- match = l + 1;
- break;
- }
- }
- }
- if (i < (DWORD)_st_sys_argc) {
- lpCmdline->dwArgc = _st_sys_argc - i;
- lpCmdline->lpArgvw = &_st_sys_argvw[i];
- }
- return lpCmdline;
-}
-
-/* Used for future expansion */
-void apxCmdlineFree(
- LPAPXCMDLINE lpCmdline)
-{
-
- apxFree(lpCmdline);
-}
-
-/*
- * Environment variables parsing
- * Each variable is prfixed with PR_
- * for example 'set PR_JVM=auto' has a same meaning as providing '--Jvm auto'
- * on the command line.
- * Multistring varisables are added to the present conf.
- */
-void apxCmdlineLoadEnvVars(
- LPAPXCMDLINE lpCmdline)
-{
- WCHAR szEnv[64];
- int i = 0;
- if (!lpCmdline || !lpCmdline->lpOptions)
- return;
-
- while (lpCmdline->lpOptions[i].szName) {
- DWORD l;
- WCHAR szVar[SIZ_HUGLEN];
- lstrcpyW(szEnv, L"PR_");
- lstrcatW(szEnv, lpCmdline->lpOptions[i].szName);
- l = GetEnvironmentVariableW(szEnv, szVar, SIZ_HUGMAX);
- if (l == 0 && GetLastError() != ERROR_ENVVAR_NOT_FOUND) {
- apxLogWrite(APXLOG_MARK_ERROR "Error geting environment variable %S",
- szEnv);
- ++i;
- continue;
- }
- if (l && (lpCmdline->lpOptions[i].dwType & APXCMDOPT_STR)) {
- lpCmdline->lpOptions[i].szValue = apxPoolStrdupW(lpCmdline->hPool, szVar);
- lpCmdline->lpOptions[i].dwType |= APXCMDOPT_FOUND;
- }
- else if (l && (lpCmdline->lpOptions[i].dwType & APXCMDOPT_MSZ)) {
- LPWSTR pp;
- BOOL insquote = FALSE, indquote = FALSE;
- DWORD sp = 0;
- lpCmdline->lpOptions[i].szValue = apxPoolCalloc(lpCmdline->hPool,
- (lstrlenW(szVar) + 2) *
- sizeof(WCHAR));
- lstrcpyW(lpCmdline->lpOptions[i].szValue, szVar);
- pp = szVar;
- while(*pp) {
- if (*pp == L'\'')
- insquote = !insquote;
- else if (*pp == L'"') {
- indquote = !indquote;
- lpCmdline->lpOptions[i].szValue[sp++] = L'"';
- }
- else if ((*pp == L'#' || *pp == L';') && !insquote && !indquote)
- lpCmdline->lpOptions[i].szValue[sp++] = L'\0';
- else
- lpCmdline->lpOptions[i].szValue[sp++] = *pp;
- pp++;
- }
- lpCmdline->lpOptions[i].dwType |= APXCMDOPT_FOUND | APXCMDOPT_ADD;
- }
- ++i;
- }
-
-}
\ No newline at end of file
Copied: tags/jbosssvc/2.0.8/procrun/src/cmdline.c (from rev 2689, trunk/utils/windows/native/service/procrun/src/cmdline.c)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/src/cmdline.c (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/src/cmdline.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,311 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "apxwin.h"
+#include "private.h"
+
+#define EXE_SUFFIX L".EXE"
+#define EXE_SUFFIXLEN (sizeof(EXE_SUFFIX) / sizeof(WCHAR) - 1)
+
+#define X86_SUFFIX L".X86"
+#define X64_SUFFIX L".X64"
+#define A64_SUFFIX L".I64"
+
+/* Those two are declared in handles.c */
+extern LPWSTR *_st_sys_argvw;
+extern int _st_sys_argc;
+
+static WCHAR _st_sys_appexe[MAX_PATH];
+/*
+ * argv parsing.
+ * Parse the argv[0] and split to ExePath and
+ * Executable name. Strip the extension ('.exe').
+ * Check for command in argv[1] //CMD//Application
+ * Parse the options --option value or --option==value
+ * break on first argument that doesn't start with '--'
+ */
+LPAPXCMDLINE apxCmdlineParse(
+ APXHANDLE hPool,
+ APXCMDLINEOPT *lpOptions,
+ LPCWSTR *lpszCommands)
+{
+
+ LPAPXCMDLINE lpCmdline = NULL;
+ DWORD l, i = 0, s = 1;
+ LPWSTR p;
+ WCHAR cmd[4];
+ DWORD match;
+ BOOL add;
+ if (_st_sys_argc < 1)
+ return NULL;
+
+ l = lstrlenW(_st_sys_argvw[0]);
+ if (l < EXE_SUFFIXLEN)
+ return NULL;
+ if (!(lpCmdline = (LPAPXCMDLINE)apxPoolCalloc(hPool, sizeof(APXCMDLINE))))
+ return NULL;
+ lpCmdline->hPool = hPool;
+ lpCmdline->lpOptions = lpOptions;
+ lpCmdline->szExePath = _st_sys_argvw[0];
+ p = lpCmdline->szExePath + l;
+
+ while (p > lpCmdline->szExePath) {
+ if (*p == L'\\') {
+ *p++ = L'\0';
+ break;
+ }
+ p--;
+ }
+ /* Find the path if it wasn't specified in the argv[0] */
+ if (p == lpCmdline->szExePath) {
+ WCHAR mh[4096];
+ LPWSTR m;
+ if (GetModuleFileNameW(GetModuleHandle(NULL), mh, 4096)) {
+ lpCmdline->szExePath = apxPoolStrdupW(hPool, mh);
+ m = lpCmdline->szExePath + lstrlenW(lpCmdline->szExePath);
+ while (m > lpCmdline->szExePath) {
+ if (*(m--) == L'\\') {
+ *(++m) = L'\0';
+ break;
+ }
+ }
+ }
+ }
+ lpCmdline->szExecutable = p;
+ p = _st_sys_argvw[0] + l - EXE_SUFFIXLEN;
+ if (lstrcmpiW(p, EXE_SUFFIX) == 0)
+ *p = L'\0';
+ /* Strip CPU specific suffixes */
+ l = lstrlenW(_st_sys_argvw[0]);
+ if (l > EXE_SUFFIXLEN) {
+ p = _st_sys_argvw[0] + l - EXE_SUFFIXLEN;
+ if (lstrcmpiW(p, X86_SUFFIX) == 0) {
+ *p = L'\0';
+ }
+ else if (lstrcmpiW(p, X64_SUFFIX) == 0) {
+ *p = L'\0';
+ }
+ else if (lstrcmpiW(p, A64_SUFFIX) == 0) {
+ *p = L'\0';
+ }
+ }
+ if (lpszCommands && _st_sys_argc > 1 && lstrlenW(_st_sys_argvw[1]) > 2) {
+ LPWSTR ca = _st_sys_argvw[1];
+ if (ca[0] == L'/' && ca[1] == L'/') {
+ l = 0;
+ ca += 2;
+ while (*ca && *ca != L'/') {
+ cmd[l] = *ca;
+ if (l++ > 3)
+ break;
+ ca++;
+ }
+ cmd[l] = L'\0';
+ while (lpszCommands[i]) {
+ if (lstrcmpW(lpszCommands[i++], cmd) == 0) {
+ lpCmdline->dwCmdIndex = i;
+ break;
+ }
+ }
+ if (lpCmdline->dwCmdIndex) {
+ while (*ca == '/')
+ *(ca++) = L'\0';
+ if (*ca == '\0')
+ lpCmdline->szApplication = _st_sys_argvw[0];
+ else
+ lpCmdline->szApplication = ca;
+ }
+ else {
+ apxLogWrite(APXLOG_MARK_ERROR "Unrecognized cmd option %S",
+ _st_sys_argvw[1]);
+ return NULL;
+ }
+ s = 2;
+ }
+ }
+ else {
+ lpCmdline->szApplication = lpCmdline->szExecutable;
+ lpCmdline->dwCmdIndex = 1;
+ return lpCmdline;
+ }
+ for (i = s; i < (DWORD)_st_sys_argc; i++) {
+ LPWSTR e = NULL;
+ if (_st_sys_argvw[i][0] != L'-' ||
+ _st_sys_argvw[i][1] != L'-') {
+ if (_st_sys_argvw[i][0] != L'+' ||
+ _st_sys_argvw[i][1] != L'+')
+ break;
+ else
+ add = TRUE;
+ }
+ else
+ add = FALSE;
+ p = &(_st_sys_argvw[i][2]);
+ /* Find if the option has '=' char
+ * for --option==value or --option value cases.
+ */
+ while (*p) {
+ if (*p == L'=') {
+ *p = L'\0';
+ e = p + 1;
+ break;
+ }
+ else
+ p++;
+ }
+ match = 0;
+ for (l = 0; lpOptions[l].szName; l++) {
+ if (lstrcmpW(lpOptions[l].szName, &(_st_sys_argvw[i][2])) == 0) {
+ LPWSTR val;
+ /* check if arg is needed */
+ if (e)
+ val = e;
+ else if ((i + 1) < (DWORD)_st_sys_argc)
+ val = _st_sys_argvw[++i];
+ else {
+ lpOptions[l].dwValue = 0;
+ lpOptions[l].szValue = NULL;
+ lpOptions[l].dwType |= APXCMDOPT_FOUND;
+ break;
+ }
+ if (lpOptions[l].dwType & APXCMDOPT_STR)
+ lpOptions[l].szValue = val;
+ else if (lpOptions[l].dwType & APXCMDOPT_INT)
+ lpOptions[l].dwValue = (DWORD)apxAtoulW(val);
+ else if (lpOptions[l].dwType & APXCMDOPT_MSZ) {
+ LPWSTR pp;
+ BOOL insquote = FALSE, indquote=FALSE;
+ DWORD sp = 0;
+ LPWSTR ov = lpOptions[l].szValue;
+ if (lpOptions[l].dwValue > 2) {
+ sp = (lpOptions[l].dwValue - sizeof(WCHAR)) / sizeof(WCHAR);
+ }
+ lpOptions[l].dwValue = (sp + lstrlenW(val) + 2) * sizeof(WCHAR);
+ lpOptions[l].szValue = (LPWSTR)apxPoolCalloc(hPool,
+ lpOptions[l].dwValue);
+ if (sp) {
+ AplMoveMemory(lpOptions[l].szValue, ov, sp * sizeof(WCHAR));
+ apxFree(ov);
+ }
+ pp = val;
+ while(*pp) {
+ if (*pp == L'\'')
+ insquote = !insquote;
+ else if (*pp == L'"') {
+ indquote = !indquote;
+ lpOptions[l].szValue[sp++] = L'"';
+ }
+ else if ((*pp == L'#' || *pp == L';') && !insquote && !indquote)
+ lpOptions[l].szValue[sp++] = L'\0';
+ else
+ lpOptions[l].szValue[sp++] = *pp;
+ pp++;
+ }
+ }
+ lpOptions[l].dwType |= APXCMDOPT_FOUND;
+ if (add)
+ lpOptions[l].dwType |= APXCMDOPT_ADD;
+ match = l + 1;
+ break;
+ }
+ }
+ if (match == 0) {
+ /* --unknown option
+ *
+ */
+ apxLogWrite(APXLOG_MARK_ERROR "Unrecognized program option %S",
+ _st_sys_argvw[i]);
+ return NULL;
+ }
+ }
+ if (i < (DWORD)_st_sys_argc) {
+ lpCmdline->dwArgc = _st_sys_argc - i;
+ lpCmdline->lpArgvw = &_st_sys_argvw[i];
+ }
+ return lpCmdline;
+}
+
+/* Used for future expansion */
+void apxCmdlineFree(
+ LPAPXCMDLINE lpCmdline)
+{
+
+ apxFree(lpCmdline);
+}
+
+/*
+ * Environment variables parsing
+ * Each variable is prfixed with PR_
+ * for example 'set PR_JVM=auto' has a same meaning as providing '--Jvm auto'
+ * on the command line.
+ * Multistring varisables are added to the present conf.
+ */
+void apxCmdlineLoadEnvVars(
+ LPAPXCMDLINE lpCmdline)
+{
+ WCHAR szEnv[64];
+ int i = 0;
+ if (!lpCmdline || !lpCmdline->lpOptions)
+ return;
+
+ while (lpCmdline->lpOptions[i].szName) {
+ DWORD l;
+ WCHAR szVar[SIZ_HUGLEN];
+ lstrlcpyW(szEnv, SIZ_HUGLEN, L"PR_");
+ lstrlcatW(szEnv, SIZ_HUGLEN, lpCmdline->lpOptions[i].szName);
+ l = GetEnvironmentVariableW(szEnv, szVar, SIZ_HUGMAX);
+ if (l == 0 && GetLastError() != ERROR_ENVVAR_NOT_FOUND) {
+ apxLogWrite(APXLOG_MARK_ERROR "Error geting environment variable %S",
+ szEnv);
+ ++i;
+ continue;
+ }
+ if (l && (lpCmdline->lpOptions[i].dwType & APXCMDOPT_STR)) {
+ lpCmdline->lpOptions[i].szValue = apxPoolStrdupW(lpCmdline->hPool, szVar);
+ lpCmdline->lpOptions[i].dwType |= APXCMDOPT_FOUND;
+ }
+ else if (l && (lpCmdline->lpOptions[i].dwType & APXCMDOPT_INT)) {
+ lpCmdline->lpOptions[i].dwValue = (DWORD)apxAtoulW(szVar);
+ lpCmdline->lpOptions[i].dwType |= APXCMDOPT_FOUND;
+ }
+ else if (l && (lpCmdline->lpOptions[i].dwType & APXCMDOPT_MSZ)) {
+ LPWSTR pp;
+ BOOL insquote = FALSE, indquote = FALSE;
+ DWORD sp = 0;
+ lpCmdline->lpOptions[i].dwValue = (lstrlenW(szVar) + 2) * sizeof(WCHAR);
+ lpCmdline->lpOptions[i].szValue = apxPoolCalloc(lpCmdline->hPool,
+ lpCmdline->lpOptions[i].dwValue);
+ pp = szVar;
+ while(*pp) {
+ if (*pp == L'\'')
+ insquote = !insquote;
+ else if (*pp == L'"') {
+ indquote = !indquote;
+ lpCmdline->lpOptions[i].szValue[sp++] = L'"';
+ }
+ else if ((*pp == L'#' || *pp == L';') && !insquote && !indquote)
+ lpCmdline->lpOptions[i].szValue[sp++] = L'\0';
+ else
+ lpCmdline->lpOptions[i].szValue[sp++] = *pp;
+ pp++;
+ }
+ lpCmdline->lpOptions[i].dwType |= APXCMDOPT_FOUND | APXCMDOPT_ADD;
+ }
+ ++i;
+ }
+
+}
+
Deleted: tags/jbosssvc/2.0.8/procrun/src/handles.c
===================================================================
--- trunk/utils/windows/native/service/procrun/src/handles.c 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/src/handles.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,686 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#include "apxwin.h"
-#include "private.h"
-
-#define ALLOCBLOCK_INVALID 0xdeadbeef
-
-typedef struct APXPOOL APXPOOL;
-typedef APXPOOL* LPAPXPOOL;
-
-typedef struct ALLOCBLOCK {
- DWORD dwSize;
- APXHANDLE lpPool;
- APXMEMWORD lpAlign;
-} ALLOCBLOCK, *LPALLOCBLOCK;
-
-struct APXPOOL {
- TAILQ_HEAD(_lHandles, stAPXHANDLE) lHandles;
- TAILQ_HEAD(_lPools, stAPXHANDLE) lPools;
-};
-
-static SYSTEM_INFO _st_sys_info;
-static APXHANDLE _st_sys_pool = NULL;
-static int _st_sys_init = 0;
-static LPVOID _st_sys_page = NULL;
-LPWSTR *_st_sys_argvw = NULL;
-int _st_sys_argc = 0;
-
-#ifdef _DEBUG
-static INT _heap_count = 0;
-static INT _heap_alloc_count = 0;
-static INT _heap_realloc_count = 0;
-
-HANDLE HeapCREATE(DWORD flOptions, SIZE_T dwInitialSize, SIZE_T dwMaximumSize)
-{
- _heap_count++;
- return HeapCreate(flOptions, dwInitialSize, dwMaximumSize);
-}
-
-BOOL HeapDESTROY(HANDLE hHeap)
-{
- _heap_count--;
- return HeapDestroy(hHeap);
-}
-
-
-LPVOID HeapALLOC(HANDLE hHeap, DWORD dwFlags, SIZE_T nSize)
-{
- _heap_alloc_count++;
- return HeapAlloc(hHeap, dwFlags, nSize);
-}
-
-BOOL HeapFREE(HANDLE hHeap, DWORD dwFlags, LPVOID lpMem)
-{
- _heap_alloc_count--;
- return HeapFree(hHeap, dwFlags, lpMem);
-}
-
-LPVOID HeapREALLOC(HANDLE hHeap, DWORD dwFlags, LPVOID lpMem, SIZE_T dwBytes)
-{
-
- _heap_realloc_count++;
- return HeapReAlloc(hHeap, dwFlags, lpMem, dwBytes);
-}
-#endif
-
-static LPVOID __apxPoolAllocCore(APXHANDLE hPool,
- DWORD dwSize, DWORD dwOptions)
-{
- DWORD dwPhysicalSize;
- LPALLOCBLOCK lpBlock;
-
- if (!hPool)
- hPool = _st_sys_pool;
- dwPhysicalSize = APX_ALIGN_DEFAULT(dwSize + sizeof(ALLOCBLOCK));
- lpBlock = HeapALLOC(hPool->hHeap, dwOptions, dwPhysicalSize);
- lpBlock->dwSize = dwPhysicalSize;
- lpBlock->lpPool = hPool;
-
- return ((char *)lpBlock + sizeof(ALLOCBLOCK));
-}
-
-static LPVOID __apxPoolReallocCore(APXHANDLE hPool, LPVOID lpMem,
- DWORD dwSize, DWORD dwOptions)
-{
- DWORD dwPhysicalSize;
- LPALLOCBLOCK lpBlock;
- LPALLOCBLOCK lpOrg;
-
- if (!lpMem)
- return __apxPoolAllocCore(hPool, dwSize, dwOptions);
- lpOrg = (LPALLOCBLOCK)((char *)lpMem - sizeof(ALLOCBLOCK));
- if (!hPool)
- hPool = _st_sys_pool;
- /* Trying to realloc something that isn't valid */
- if (lpOrg->lpPool == APXHANDLE_INVALID ||
- lpOrg->lpPool != hPool)
- return NULL;
- dwPhysicalSize = APX_ALIGN_DEFAULT(dwSize + sizeof(ALLOCBLOCK));
- lpBlock = HeapREALLOC(hPool->hHeap, dwOptions, lpOrg, dwPhysicalSize);
- lpBlock->dwSize = dwPhysicalSize;
- lpBlock->lpPool = hPool;
-
- return ((char *)lpBlock + sizeof(ALLOCBLOCK));
-}
-
-static void __apxPoolFreeCore(LPVOID lpMem)
-{
- APXHANDLE hPool;
- LPALLOCBLOCK lpBlock = (LPALLOCBLOCK)((char *)lpMem - sizeof(ALLOCBLOCK));
-
- if (lpBlock->lpPool != APXHANDLE_INVALID) {
- hPool = lpBlock->lpPool;
- lpBlock->lpPool = APXHANDLE_INVALID;
- }
- else
- return;
- HeapFREE(hPool->hHeap, 0, lpBlock);
-}
-/*
- *
- */
-static DWORD WINAPI __apxHandleEventThread(LPVOID lpParameter)
-{
- APXHANDLE hHandle = (APXHANDLE)lpParameter;
- DWORD rv = 0;
- while (hHandle->dwType != APXHANDLE_TYPE_INVALID) {
- DWORD dwState;
- dwState = WaitForSingleObject(hHandle->hEventHandle, INFINITE);
- /* the flags can be changed to invalid meaning we are killing
- * this event.
- */
- if (dwState == WAIT_OBJECT_0 &&
- hHandle->dwType != APXHANDLE_TYPE_INVALID) {
- if (hHandle->uMsg && (hHandle->wParam || hHandle->lParam)) {
- APXCALLHOOK *lpCall;
- rv = (*hHandle->fnCallback)(hHandle, hHandle->uMsg,
- hHandle->wParam, hHandle->lParam);
- TAILQ_FOREACH(lpCall, &hHandle->lCallbacks, queue) {
- (*lpCall->fnCallback)(hHandle, hHandle->uMsg,
- hHandle->wParam, hHandle->lParam);
- }
- hHandle->uMsg = 0;
- if (!rv)
- break;
- }
- ResetEvent(hHandle->hEventHandle);
- SwitchToThread();
- }
- else
- break;
- }
- ResetEvent(hHandle->hEventHandle);
- /* This will rise the Thread waiting function */
- return 0;
-}
-
-static BOOL __apxPoolCallback(APXHANDLE hObject, UINT uMsg,
- WPARAM wParam, LPARAM lParam)
-{
- LPAPXPOOL lpPool;
- APXHANDLE hCur;
- if (hObject->dwType != APXHANDLE_TYPE_POOL)
- return FALSE;
- lpPool = APXHANDLE_DATA(hObject);
-
- /* recurse the subpools */
- TAILQ_FOREACH(hCur, &lpPool->lPools, queue) {
- __apxPoolCallback(hCur, uMsg, 0, 0);
- }
- /* call the handles callback */
- for(hCur = TAILQ_FIRST(&lpPool->lHandles) ;
- hCur != NULL ;
- hCur = TAILQ_FIRST(&lpPool->lHandles)) {
- apxCloseHandle(hCur);
- }
- /* if we are closing this pool destroy the private Heap */
- if (uMsg == WM_CLOSE) {
- if (hObject->dwFlags & APXHANDLE_HAS_HEAP)
- HeapDESTROY(hObject->hHeap);
- hObject->dwSize = 0;
- }
- else if (uMsg == WM_CLEAR)
- hObject->dwSize = 0;
-
- return TRUE;
-}
-
-static BOOL __apxHandleCallback(APXHANDLE hObject, UINT uMsg,
- WPARAM wParam, LPARAM lParam)
-{
- BOOL rv = FALSE;
- if (hObject->dwType == APXHANDLE_TYPE_INVALID)
- return FALSE;
- /* Default handler handles only close event */
- if (uMsg != WM_CLOSE)
- return FALSE;
- if (hObject->dwType == APXHANDLE_TYPE_WINHANDLE &&
- !(IS_INVALID_HANDLE(hObject->uData.hWinHandle))) {
- rv = CloseHandle(hObject->uData.hWinHandle);
- hObject->uData.hWinHandle = NULL;
- }
- /* Invalidate the handle */
- hObject->dwType = APXHANDLE_TYPE_INVALID;
-
- if (hObject->dwFlags & APXHANDLE_HAS_EVENT) {
- DWORD dwState;
- /* Signal the EventThread to exit */
- SetEvent(hObject->hEventHandle);
- /* Wait for EventThread to Exit */
- dwState = WaitForSingleObject(hObject->hEventThread, 1000);
- SAFE_CLOSE_HANDLE(hObject->hEventHandle);
- if (dwState == WAIT_TIMEOUT)
- TerminateThread(hObject->hEventThread, 0);
- SAFE_CLOSE_HANDLE(hObject->hEventThread);
- /* Reset the evant flag */
- hObject->dwFlags &= ~APXHANDLE_HAS_EVENT;
- }
- return rv;
-}
-
-static BOOL __apxCreateSystemPool()
-{
- LPAPXPOOL lpPool;
- HANDLE hHeap;
-
- GetSystemInfo(&_st_sys_info);
- apxGetOsLevel();
- /* First create the shared data segment */
- _st_sys_page = VirtualAlloc(NULL, _st_sys_info.dwAllocationGranularity,
- MEM_RESERVE, PAGE_NOACCESS);
- if (!_st_sys_page)
- return FALSE;
- _st_sys_page = VirtualAlloc(_st_sys_page, _st_sys_info.dwAllocationGranularity,
- MEM_COMMIT, PAGE_READWRITE);
-
- /* Create the main Heap */
- hHeap = HeapCREATE(0, _st_sys_info.dwAllocationGranularity, 0);
- _st_sys_pool = HeapALLOC(hHeap, HEAP_ZERO_MEMORY,
- APX_ALIGN_DEFAULT(APXHANDLE_SZ + sizeof(APXPOOL)));
- _st_sys_pool->hHeap = hHeap;
- _st_sys_pool->dwType = APXHANDLE_TYPE_INVALID;
- if (IS_INVALID_HANDLE(_st_sys_pool->hHeap))
- return FALSE;
- _st_sys_pool->fnCallback = __apxPoolCallback;
- lpPool = APXHANDLE_DATA(_st_sys_pool);
- /* Initialize the pool and object lists */
- TAILQ_INIT(&lpPool->lHandles);
- TAILQ_INIT(&lpPool->lPools);
- _st_sys_pool->dwType = APXHANDLE_TYPE_POOL;
-
- /** TODO: For each unsupported function make a surrogate */
- _st_sys_argvw = CommandLineToArgvW(GetCommandLineW(), &_st_sys_argc);
-
- return TRUE;
-}
-
-BOOL
-apxHandleManagerInitialize()
-{
- BOOL rv;
- if (_st_sys_init++)
- return TRUE;
- rv = __apxCreateSystemPool();
-
- return rv;
-}
-
-BOOL
-apxHandleManagerDestroy()
-{
- HANDLE hHeap;
-
- if (--_st_sys_init == 0) {
- hHeap = _st_sys_pool->hHeap;
- apxCloseHandle(_st_sys_pool);
- /* Destroy the main Heap */
- HeapDESTROY(hHeap);
- _st_sys_pool = NULL;
- VirtualFree(_st_sys_page, 0, MEM_RELEASE);
- GlobalFree(_st_sys_argvw);
- _st_sys_argvw = NULL;
- _st_sys_argc = 0;
-#ifdef _DEBUG
- apxLogWrite(APXLOG_MARK_DEBUG "Alloc Count %d", _heap_alloc_count);
- apxLogWrite(APXLOG_MARK_DEBUG "Realloc Count %d", _heap_realloc_count);
- apxLogWrite(APXLOG_MARK_DEBUG "Heap Count %d", _heap_count);
-#endif
- return TRUE;
- }
-
- return FALSE;
-}
-
-APXHANDLE
-apxPoolCreate(APXHANDLE hParent, DWORD dwOptions)
-{
- APXHANDLE hHandle;
- LPAPXPOOL lpPool;
- HANDLE hHeap;
-
- if (IS_INVALID_HANDLE(hParent))
- hParent = _st_sys_pool;
- if (hParent->dwType != APXHANDLE_TYPE_POOL) {
- apxLogWrite(APXLOG_MARK_ERROR "Parent Handle type is not POOL %d",
- hParent->dwType);
- return INVALID_HANDLE_VALUE;
- }
- /* Allocate the handle from the parent */
- hHandle = __apxPoolAllocCore(hParent, APXHANDLE_SZ + sizeof(APXPOOL),
- HEAP_ZERO_MEMORY);
-
- if (dwOptions & APXHANDLE_HAS_HEAP) {
- /* Create the private Heap */
- hHeap = HeapCREATE(0, _st_sys_info.dwAllocationGranularity, 0);
- hHandle->dwFlags |= APXHANDLE_HAS_HEAP;
- }
- else
- hHeap = hParent->hHeap;
- hHandle->hHeap = hHeap;
- hHandle->dwType = APXHANDLE_TYPE_POOL;
- hHandle->hPool = hParent;
- hHandle->fnCallback = __apxPoolCallback;
- lpPool = APXHANDLE_DATA(hHandle);
- TAILQ_INIT(&lpPool->lHandles);
- TAILQ_INIT(&lpPool->lPools);
-
- /* Insert the pool to the head of parent pool */
- lpPool = APXHANDLE_DATA(hParent);
- APXHANDLE_SPINLOCK(hParent);
- TAILQ_INSERT_HEAD(&lpPool->lPools, hHandle, queue);
- ++hParent->dwSize;
- APXHANDLE_SPINUNLOCK(hParent);
-
- return hHandle;
-}
-
-LPVOID
-apxPoolAlloc(APXHANDLE hPool, DWORD dwSize)
-{
- if (IS_INVALID_HANDLE(hPool) ||
- (hPool->dwType != APXHANDLE_TYPE_POOL))
- hPool = _st_sys_pool;
- return __apxPoolAllocCore(hPool, dwSize, 0);
-}
-
-LPVOID
-apxPoolCalloc(APXHANDLE hPool, DWORD dwSize)
-{
- if (IS_INVALID_HANDLE(hPool) ||
- (hPool->dwType != APXHANDLE_TYPE_POOL))
- hPool = _st_sys_pool;
- return __apxPoolAllocCore(hPool, dwSize, HEAP_ZERO_MEMORY);
-}
-
-LPVOID
-apxPoolRealloc(APXHANDLE hPool, LPVOID lpMem, DWORD dwNewSize)
-{
- if (IS_INVALID_HANDLE(hPool) ||
- (hPool->dwType != APXHANDLE_TYPE_POOL))
- hPool = _st_sys_pool;
- return __apxPoolReallocCore(hPool, lpMem, dwNewSize, HEAP_ZERO_MEMORY);
-}
-
-LPVOID
-apxAlloc(DWORD dwSize)
-{
- return __apxPoolAllocCore(_st_sys_pool, dwSize, 0);
-}
-
-LPVOID
-apxCalloc(DWORD dwSize)
-{
- return __apxPoolAllocCore(_st_sys_pool, dwSize, HEAP_ZERO_MEMORY);
-}
-
-LPVOID
-apxRealloc(LPVOID lpMem, DWORD dwNewSize)
-{
- return __apxPoolReallocCore(_st_sys_pool, lpMem, dwNewSize, HEAP_ZERO_MEMORY);
-}
-
-VOID
-apxFree(LPVOID lpMem)
-{
- if (lpMem)
- __apxPoolFreeCore(lpMem);
-
-}
-
-LPWSTR
-apxPoolWStrdupA(APXHANDLE hPool, LPCSTR szSource)
-{
- if (szSource) {
- LPWSTR szDest;
- DWORD l = lstrlenA(szSource);
- szDest = apxPoolAlloc(hPool, (l + 1) * sizeof(WCHAR));
- AsciiToWide(szSource, szDest);
- return szDest;
- }
- else
- return NULL;
-}
-
-LPWSTR apxWStrdupA(LPCTSTR szSource)
-{
- return apxPoolWStrdup(_st_sys_pool, szSource);
-}
-
-LPSTR
-apxPoolStrdupA(APXHANDLE hPool, LPCSTR szSource)
-{
- if (szSource) {
- LPSTR szDest;
- DWORD l = lstrlenA(szSource);
- szDest = apxPoolAlloc(hPool, l + 1);
- lstrcpyA(szDest, szSource);
- return szDest;
- }
- else
- return NULL;
-}
-
-LPWSTR
-apxPoolStrdupW(APXHANDLE hPool, LPCWSTR szSource)
-{
- if (szSource) {
- LPWSTR szDest;
- DWORD l = lstrlenW(szSource);
- szDest = apxPoolAlloc(hPool, (l + 1) * sizeof(WCHAR));
- lstrcpyW(szDest, szSource);
- return szDest;
- }
- else
- return NULL;
-}
-
-LPSTR
-apxStrdupA(LPCSTR szSource)
-{
- return apxPoolStrdupA(_st_sys_pool, szSource);
-}
-
-LPWSTR
-apxStrdupW(LPCWSTR szSource)
-{
- return apxPoolStrdupW(_st_sys_pool, szSource);
-}
-
-APXHANDLE
-apxHandleCreate(APXHANDLE hPool, DWORD dwFlags,
- LPVOID lpData, DWORD dwDataSize,
- LPAPXFNCALLBACK fnCallback)
-{
- APXHANDLE hHandle;
- LPAPXPOOL lpPool;
-
- if (IS_INVALID_HANDLE(hPool))
- hPool = _st_sys_pool;
- if (hPool->dwType != APXHANDLE_TYPE_POOL) {
- apxLogWrite(APXLOG_MARK_ERROR "Parent Handle type is not POOL %d",
- hPool->dwType);
- return INVALID_HANDLE_VALUE;
- }
- hHandle = __apxPoolAllocCore(hPool, APXHANDLE_SZ + dwDataSize,
- HEAP_ZERO_MEMORY);
-
- hHandle->hPool = hPool;
- if (fnCallback)
- hHandle->fnCallback = fnCallback;
- else
- hHandle->fnCallback = __apxHandleCallback;
-
- if (dwFlags & APXHANDLE_TYPE_WINHANDLE) {
- hHandle->dwFlags |= APXHANDLE_HAS_USERDATA;
- hHandle->dwFlags |= APXHANDLE_TYPE_WINHANDLE;
- hHandle->uData.hWinHandle = lpData;
- }
- else if (dwFlags & APXHANDLE_TYPE_LPTR) {
- hHandle->dwFlags |= APXHANDLE_HAS_USERDATA;
- hHandle->dwFlags |= APXHANDLE_TYPE_LPTR;
- hHandle->uData.lpPtr = lpData;
- }
- else if (dwDataSize && lpData) {
- hHandle->dwFlags |= APXHANDLE_HAS_USERDATA;
- AplCopyMemory(APXHANDLE_DATA(hHandle), lpData, dwDataSize);
- hHandle->dwSize = dwDataSize;
- }
-
- if (dwFlags & APXHANDLE_HAS_EVENT) {
- /* Create the message event and message wathcing thread */
- hHandle->hEventHandle = CreateEvent(NULL, TRUE, FALSE, NULL);
- hHandle->hEventThread = CreateThread(NULL, 0, __apxHandleEventThread,
- hHandle, 0,
- &(hHandle->hEventThreadId));
- if (IS_INVALID_HANDLE(hHandle->hEventThread)) {
- SAFE_CLOSE_HANDLE(hHandle->hEventHandle);
- }
- else
- hHandle->dwFlags |= APXHANDLE_HAS_EVENT;
- }
- TAILQ_INIT(&hHandle->lCallbacks);
- /* Add the handle to the pool's object list */
- lpPool = APXHANDLE_DATA(hPool);
- APXHANDLE_SPINLOCK(hPool);
- TAILQ_INSERT_HEAD(&lpPool->lHandles, hHandle, queue);
- ++hPool->dwSize;
- APXHANDLE_SPINUNLOCK(hPool);
-
- return hHandle;
-}
-
-BOOL
-apxCloseHandle(APXHANDLE hObject)
-{
- LPAPXPOOL lpPool;
- APXCALLHOOK *lpCall;
-
- if (IS_INVALID_HANDLE(hObject) || hObject->dwType == APXHANDLE_TYPE_INVALID)
- return FALSE;
- /* Call the user callback first */
- (*hObject->fnCallback)(hObject, WM_CLOSE, 0, 0);
- /* Now go through the callback chain */
- TAILQ_FOREACH(lpCall, &hObject->lCallbacks, queue) {
- (*lpCall->fnCallback)(hObject, WM_CLOSE, 0, 0);
- TAILQ_REMOVE(&hObject->lCallbacks, lpCall, queue);
- __apxPoolFreeCore(lpCall);
- }
-
- hObject->dwType = APXHANDLE_TYPE_INVALID;
- if (hObject->dwFlags & APXHANDLE_HAS_EVENT) {
- DWORD dwState;
-
- SetEvent(hObject->hEventHandle);
- dwState = WaitForSingleObject(hObject->hEventThread, 1000);
- SAFE_CLOSE_HANDLE(hObject->hEventHandle);
- if (dwState == WAIT_TIMEOUT)
- TerminateThread(hObject->hEventThread, 0);
- SAFE_CLOSE_HANDLE(hObject->hEventThread);
-
- hObject->dwFlags &= ~APXHANDLE_HAS_EVENT;
- }
-
- /* finaly remove the object from the pool's object list */
- if (!IS_INVALID_HANDLE(hObject->hPool)) {
- lpPool = APXHANDLE_DATA(hObject->hPool);
- APXHANDLE_SPINLOCK(hObject->hPool);
- TAILQ_REMOVE(&lpPool->lHandles, hObject, queue);
- hObject->hPool->dwSize--;
- APXHANDLE_SPINUNLOCK(hObject->hPool);
- __apxPoolFreeCore(hObject);
- }
- return TRUE;
-}
-
-LPVOID
-apxHandleGetUserData(APXHANDLE hObject)
-{
- if (hObject->dwType == APXHANDLE_TYPE_INVALID)
- return NULL;
- if (hObject->dwFlags & APXHANDLE_HAS_USERDATA)
- return APXHANDLE_DATA(hObject);
- else
- return hObject->uData.lpPtr;
-}
-
-LPVOID
-apxHandleSetUserData(APXHANDLE hObject, LPVOID lpData, DWORD dwDataSize)
-{
- if (hObject->dwType == APXHANDLE_TYPE_INVALID)
- return NULL;
- if (hObject->dwFlags & APXHANDLE_HAS_USERDATA && hObject->dwSize > 0) {
- AplCopyMemory(APXHANDLE_DATA(hObject), lpData,
- MIN(hObject->dwSize, dwDataSize));
- return APXHANDLE_DATA(hObject);
- }
- else {
- LPVOID lpOrg = hObject->uData.lpPtr;
- hObject->uData.lpPtr = lpData;
- return lpOrg;
- }
-}
-
-BOOL apxHandleSendMessage(APXHANDLE hObject, UINT uMsg, WPARAM wParam, LPARAM lParam)
-{
- BOOL rv = TRUE;
- APXCALLHOOK *lpCall;
- if (hObject->dwType == APXHANDLE_TYPE_INVALID)
- return FALSE;
- /* Serialize requests to the callback */
- APXHANDLE_SPINLOCK(hObject);
- if (hObject->fnCallback)
- rv = (*hObject->fnCallback)(hObject, uMsg, wParam, lParam);
- TAILQ_FOREACH(lpCall, &hObject->lCallbacks, queue) {
- (*lpCall->fnCallback)(hObject, uMsg, wParam, lParam);
- }
- APXHANDLE_SPINUNLOCK(hObject);
-
- return rv;
-}
-
-BOOL apxHandlePostMessage(APXHANDLE hObject, UINT uMsg, WPARAM wParam, LPARAM lParam)
-{
- if (hObject->dwType == APXHANDLE_TYPE_INVALID)
- return FALSE;
- if (hObject->dwFlags & APXHANDLE_HAS_EVENT) {
- /* TODO: Create a thread message queue
- * Right now wait while the event gets nonsignaled
- */
- while (WaitForSingleObject(hObject->hEventHandle, 0) == WAIT_OBJECT_0)
- SwitchToThread();
- APXHANDLE_SPINLOCK(hObject);
- SuspendThread(hObject->hEventThread);
- hObject->uMsg = uMsg;
- hObject->wParam = wParam;
- hObject->lParam = lParam;
- /* Signal the event thread to call the user callback */
- SetEvent(hObject->hEventHandle);
- ResumeThread(hObject->hEventThread);
- APXHANDLE_SPINUNLOCK(hObject);
-
- return TRUE;
- }
- return FALSE;
-}
-
-BOOL apxHandleLock(APXHANDLE hObject, BOOL bLock)
-{
- if (bLock)
- APXHANDLE_SPINLOCK(hObject);
- else
- APXHANDLE_SPINUNLOCK(hObject);
-
- return TRUE;
-}
-
-BOOL apxHandleAddHook(APXHANDLE hObject, DWORD dwWhere,
- LPAPXFNCALLBACK fnCallback)
-{
- APXCALLHOOK *lpCall;
-
- if (hObject->dwType == APXHANDLE_TYPE_INVALID || !fnCallback)
- return FALSE;
- lpCall = (APXCALLHOOK *)__apxPoolAllocCore(hObject->hPool,
- sizeof(APXCALLHOOK), 0);
- if (!lpCall)
- return FALSE;
- lpCall->fnCallback = fnCallback;
- APXHANDLE_SPINLOCK(hObject);
- if (dwWhere == APXHANDLE_HOOK_FIRST) {
- TAILQ_INSERT_HEAD(&hObject->lCallbacks, lpCall, queue);
- }
- else {
- TAILQ_INSERT_TAIL(&hObject->lCallbacks, lpCall, queue);
- }
- APXHANDLE_SPINUNLOCK(hObject);
-
- return TRUE;
-}
-
-DWORD apxHandleWait(APXHANDLE hHandle, DWORD dwMilliseconds, BOOL bKill)
-{
- if (IS_INVALID_HANDLE(hHandle))
- return WAIT_ABANDONED;
- if (hHandle->dwType == APXHANDLE_TYPE_JVM)
- return apxJavaWait(hHandle, dwMilliseconds, bKill);
- else if (hHandle->dwType == APXHANDLE_TYPE_PROCESS)
- return apxProcessWait(hHandle, dwMilliseconds, bKill);
- else
- return WAIT_ABANDONED;
-}
-
Copied: tags/jbosssvc/2.0.8/procrun/src/handles.c (from rev 2689, trunk/utils/windows/native/service/procrun/src/handles.c)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/src/handles.c (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/src/handles.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,689 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "apxwin.h"
+#include "private.h"
+
+#define ALLOCBLOCK_INVALID 0xdeadbeef
+
+typedef struct APXPOOL APXPOOL;
+typedef APXPOOL* LPAPXPOOL;
+
+typedef struct ALLOCBLOCK {
+ DWORD dwSize;
+ APXHANDLE lpPool;
+ APXMEMWORD lpAlign;
+} ALLOCBLOCK, *LPALLOCBLOCK;
+
+struct APXPOOL {
+ TAILQ_HEAD(_lHandles, stAPXHANDLE) lHandles;
+ TAILQ_HEAD(_lPools, stAPXHANDLE) lPools;
+};
+
+static SYSTEM_INFO _st_sys_info;
+static APXHANDLE _st_sys_pool = NULL;
+static int _st_sys_init = 0;
+static LPVOID _st_sys_page = NULL;
+LPWSTR *_st_sys_argvw = NULL;
+int _st_sys_argc = 0;
+
+#ifdef _DEBUG
+static INT _heap_count = 0;
+static INT _heap_alloc_count = 0;
+static INT _heap_realloc_count = 0;
+
+HANDLE HeapCREATE(DWORD flOptions, SIZE_T dwInitialSize, SIZE_T dwMaximumSize)
+{
+ _heap_count++;
+ return HeapCreate(flOptions, dwInitialSize, dwMaximumSize);
+}
+
+BOOL HeapDESTROY(HANDLE hHeap)
+{
+ _heap_count--;
+ return HeapDestroy(hHeap);
+}
+
+
+LPVOID HeapALLOC(HANDLE hHeap, DWORD dwFlags, SIZE_T nSize)
+{
+ _heap_alloc_count++;
+ return HeapAlloc(hHeap, dwFlags, nSize);
+}
+
+BOOL HeapFREE(HANDLE hHeap, DWORD dwFlags, LPVOID lpMem)
+{
+ _heap_alloc_count--;
+ return HeapFree(hHeap, dwFlags, lpMem);
+}
+
+LPVOID HeapREALLOC(HANDLE hHeap, DWORD dwFlags, LPVOID lpMem, SIZE_T dwBytes)
+{
+
+ _heap_realloc_count++;
+ return HeapReAlloc(hHeap, dwFlags, lpMem, dwBytes);
+}
+#endif
+
+static LPVOID __apxPoolAllocCore(APXHANDLE hPool,
+ DWORD dwSize, DWORD dwOptions)
+{
+ DWORD dwPhysicalSize;
+ LPALLOCBLOCK lpBlock;
+
+ if (!hPool)
+ hPool = _st_sys_pool;
+ dwPhysicalSize = APX_ALIGN_DEFAULT(dwSize + sizeof(ALLOCBLOCK));
+ lpBlock = HeapALLOC(hPool->hHeap, dwOptions, dwPhysicalSize);
+ lpBlock->dwSize = dwPhysicalSize;
+ lpBlock->lpPool = hPool;
+
+ return ((char *)lpBlock + sizeof(ALLOCBLOCK));
+}
+
+static LPVOID __apxPoolReallocCore(APXHANDLE hPool, LPVOID lpMem,
+ DWORD dwSize, DWORD dwOptions)
+{
+ DWORD dwPhysicalSize;
+ LPALLOCBLOCK lpBlock;
+ LPALLOCBLOCK lpOrg;
+
+ if (!lpMem)
+ return __apxPoolAllocCore(hPool, dwSize, dwOptions);
+ lpOrg = (LPALLOCBLOCK)((char *)lpMem - sizeof(ALLOCBLOCK));
+ if (!hPool)
+ hPool = _st_sys_pool;
+ /* Trying to realloc something that isn't valid */
+ if (lpOrg->lpPool == APXHANDLE_INVALID ||
+ lpOrg->lpPool != hPool)
+ return NULL;
+ dwPhysicalSize = APX_ALIGN_DEFAULT(dwSize + sizeof(ALLOCBLOCK));
+ lpBlock = HeapREALLOC(hPool->hHeap, dwOptions, lpOrg, dwPhysicalSize);
+ lpBlock->dwSize = dwPhysicalSize;
+ lpBlock->lpPool = hPool;
+
+ return ((char *)lpBlock + sizeof(ALLOCBLOCK));
+}
+
+static void __apxPoolFreeCore(LPVOID lpMem)
+{
+ APXHANDLE hPool;
+ LPALLOCBLOCK lpBlock = (LPALLOCBLOCK)((char *)lpMem - sizeof(ALLOCBLOCK));
+
+ if (lpBlock->lpPool != APXHANDLE_INVALID) {
+ hPool = lpBlock->lpPool;
+ lpBlock->lpPool = APXHANDLE_INVALID;
+ }
+ else
+ return;
+ HeapFREE(hPool->hHeap, 0, lpBlock);
+}
+/*
+ *
+ */
+static DWORD WINAPI __apxHandleEventThread(LPVOID lpParameter)
+{
+ APXHANDLE hHandle = (APXHANDLE)lpParameter;
+ DWORD rv = 0;
+ while (hHandle->dwType != APXHANDLE_TYPE_INVALID) {
+ DWORD dwState;
+ dwState = WaitForSingleObject(hHandle->hEventHandle, INFINITE);
+ /* the flags can be changed to invalid meaning we are killing
+ * this event.
+ */
+ if (dwState == WAIT_OBJECT_0 &&
+ hHandle->dwType != APXHANDLE_TYPE_INVALID) {
+ if (hHandle->uMsg && (hHandle->wParam || hHandle->lParam)) {
+ APXCALLHOOK *lpCall;
+ rv = (*hHandle->fnCallback)(hHandle, hHandle->uMsg,
+ hHandle->wParam, hHandle->lParam);
+ TAILQ_FOREACH(lpCall, &hHandle->lCallbacks, queue) {
+ (*lpCall->fnCallback)(hHandle, hHandle->uMsg,
+ hHandle->wParam, hHandle->lParam);
+ }
+ hHandle->uMsg = 0;
+ if (!rv)
+ break;
+ }
+ ResetEvent(hHandle->hEventHandle);
+ SwitchToThread();
+ }
+ else
+ break;
+ }
+ ResetEvent(hHandle->hEventHandle);
+ /* This will rise the Thread waiting function */
+ return 0;
+}
+
+static BOOL __apxPoolCallback(APXHANDLE hObject, UINT uMsg,
+ WPARAM wParam, LPARAM lParam)
+{
+ LPAPXPOOL lpPool;
+ APXHANDLE hCur;
+ if (hObject->dwType != APXHANDLE_TYPE_POOL)
+ return FALSE;
+ lpPool = APXHANDLE_DATA(hObject);
+
+ /* recurse the subpools */
+ TAILQ_FOREACH(hCur, &lpPool->lPools, queue) {
+ __apxPoolCallback(hCur, uMsg, 0, 0);
+ }
+ /* call the handles callback */
+ for(hCur = TAILQ_FIRST(&lpPool->lHandles) ;
+ hCur != NULL ;
+ hCur = TAILQ_FIRST(&lpPool->lHandles)) {
+ apxCloseHandle(hCur);
+ }
+ /* if we are closing this pool destroy the private Heap */
+ if (uMsg == WM_CLOSE) {
+ if (hObject->dwFlags & APXHANDLE_HAS_HEAP)
+ HeapDESTROY(hObject->hHeap);
+ hObject->dwSize = 0;
+ }
+ else if (uMsg == WM_CLEAR)
+ hObject->dwSize = 0;
+
+ return TRUE;
+}
+
+static BOOL __apxHandleCallback(APXHANDLE hObject, UINT uMsg,
+ WPARAM wParam, LPARAM lParam)
+{
+ BOOL rv = FALSE;
+ if (hObject->dwType == APXHANDLE_TYPE_INVALID)
+ return FALSE;
+ /* Default handler handles only close event */
+ if (uMsg != WM_CLOSE)
+ return FALSE;
+ if (hObject->dwType == APXHANDLE_TYPE_WINHANDLE &&
+ !(IS_INVALID_HANDLE(hObject->uData.hWinHandle))) {
+ rv = CloseHandle(hObject->uData.hWinHandle);
+ hObject->uData.hWinHandle = NULL;
+ }
+ /* Invalidate the handle */
+ hObject->dwType = APXHANDLE_TYPE_INVALID;
+
+ if (hObject->dwFlags & APXHANDLE_HAS_EVENT) {
+ DWORD dwState;
+ /* Signal the EventThread to exit */
+ SetEvent(hObject->hEventHandle);
+ /* Wait for EventThread to Exit */
+ dwState = WaitForSingleObject(hObject->hEventThread, 1000);
+ SAFE_CLOSE_HANDLE(hObject->hEventHandle);
+ if (dwState == WAIT_TIMEOUT)
+ TerminateThread(hObject->hEventThread, 0);
+ SAFE_CLOSE_HANDLE(hObject->hEventThread);
+ /* Reset the evant flag */
+ hObject->dwFlags &= ~APXHANDLE_HAS_EVENT;
+ }
+ return rv;
+}
+
+static BOOL __apxCreateSystemPool()
+{
+ LPAPXPOOL lpPool;
+ HANDLE hHeap;
+
+ GetSystemInfo(&_st_sys_info);
+ apxGetOsLevel();
+ /* First create the shared data segment */
+ _st_sys_page = VirtualAlloc(NULL, _st_sys_info.dwAllocationGranularity,
+ MEM_RESERVE, PAGE_NOACCESS);
+ if (!_st_sys_page)
+ return FALSE;
+ _st_sys_page = VirtualAlloc(_st_sys_page, _st_sys_info.dwAllocationGranularity,
+ MEM_COMMIT, PAGE_READWRITE);
+
+ /* Create the main Heap */
+ hHeap = HeapCREATE(0, _st_sys_info.dwAllocationGranularity, 0);
+ _st_sys_pool = HeapALLOC(hHeap, HEAP_ZERO_MEMORY,
+ APX_ALIGN_DEFAULT(APXHANDLE_SZ + sizeof(APXPOOL)));
+ _st_sys_pool->hHeap = hHeap;
+ _st_sys_pool->dwType = APXHANDLE_TYPE_INVALID;
+ if (IS_INVALID_HANDLE(_st_sys_pool->hHeap))
+ return FALSE;
+ _st_sys_pool->fnCallback = __apxPoolCallback;
+ lpPool = APXHANDLE_DATA(_st_sys_pool);
+ /* Initialize the pool and object lists */
+ TAILQ_INIT(&lpPool->lHandles);
+ TAILQ_INIT(&lpPool->lPools);
+ _st_sys_pool->dwType = APXHANDLE_TYPE_POOL;
+
+ /** TODO: For each unsupported function make a surrogate */
+ _st_sys_argvw = CommandLineToArgvW(GetCommandLineW(), &_st_sys_argc);
+
+ return TRUE;
+}
+
+BOOL
+apxHandleManagerInitialize()
+{
+ BOOL rv;
+ if (_st_sys_init++)
+ return TRUE;
+ rv = __apxCreateSystemPool();
+
+ return rv;
+}
+
+BOOL
+apxHandleManagerDestroy()
+{
+ HANDLE hHeap;
+
+ if (--_st_sys_init == 0) {
+ hHeap = _st_sys_pool->hHeap;
+ apxCloseHandle(_st_sys_pool);
+ /* Destroy the main Heap */
+ HeapDESTROY(hHeap);
+ _st_sys_pool = NULL;
+ VirtualFree(_st_sys_page, 0, MEM_RELEASE);
+ GlobalFree(_st_sys_argvw);
+ _st_sys_argvw = NULL;
+ _st_sys_argc = 0;
+#ifdef _DEBUG
+ apxLogWrite(APXLOG_MARK_DEBUG "Alloc Count %d", _heap_alloc_count);
+ apxLogWrite(APXLOG_MARK_DEBUG "Realloc Count %d", _heap_realloc_count);
+ apxLogWrite(APXLOG_MARK_DEBUG "Heap Count %d", _heap_count);
+#endif
+ return TRUE;
+ }
+
+ return FALSE;
+}
+
+APXHANDLE
+apxPoolCreate(APXHANDLE hParent, DWORD dwOptions)
+{
+ APXHANDLE hHandle;
+ LPAPXPOOL lpPool;
+ HANDLE hHeap;
+
+ if (IS_INVALID_HANDLE(hParent))
+ hParent = _st_sys_pool;
+ if (hParent->dwType != APXHANDLE_TYPE_POOL) {
+ apxLogWrite(APXLOG_MARK_ERROR "Parent Handle type is not POOL %d",
+ hParent->dwType);
+ return INVALID_HANDLE_VALUE;
+ }
+ /* Allocate the handle from the parent */
+ hHandle = __apxPoolAllocCore(hParent, APXHANDLE_SZ + sizeof(APXPOOL),
+ HEAP_ZERO_MEMORY);
+
+ if (dwOptions & APXHANDLE_HAS_HEAP) {
+ /* Create the private Heap */
+ hHeap = HeapCREATE(0, _st_sys_info.dwAllocationGranularity, 0);
+ hHandle->dwFlags |= APXHANDLE_HAS_HEAP;
+ }
+ else
+ hHeap = hParent->hHeap;
+ hHandle->hHeap = hHeap;
+ hHandle->dwType = APXHANDLE_TYPE_POOL;
+ hHandle->hPool = hParent;
+ hHandle->fnCallback = __apxPoolCallback;
+ lpPool = APXHANDLE_DATA(hHandle);
+ TAILQ_INIT(&lpPool->lHandles);
+ TAILQ_INIT(&lpPool->lPools);
+
+ /* Insert the pool to the head of parent pool */
+ lpPool = APXHANDLE_DATA(hParent);
+ APXHANDLE_SPINLOCK(hParent);
+ TAILQ_INSERT_HEAD(&lpPool->lPools, hHandle, queue);
+ ++hParent->dwSize;
+ APXHANDLE_SPINUNLOCK(hParent);
+
+ return hHandle;
+}
+
+LPVOID
+apxPoolAlloc(APXHANDLE hPool, DWORD dwSize)
+{
+ if (IS_INVALID_HANDLE(hPool) ||
+ (hPool->dwType != APXHANDLE_TYPE_POOL))
+ hPool = _st_sys_pool;
+ return __apxPoolAllocCore(hPool, dwSize, 0);
+}
+
+LPVOID
+apxPoolCalloc(APXHANDLE hPool, DWORD dwSize)
+{
+ if (IS_INVALID_HANDLE(hPool) ||
+ (hPool->dwType != APXHANDLE_TYPE_POOL))
+ hPool = _st_sys_pool;
+ return __apxPoolAllocCore(hPool, dwSize, HEAP_ZERO_MEMORY);
+}
+
+LPVOID
+apxPoolRealloc(APXHANDLE hPool, LPVOID lpMem, DWORD dwNewSize)
+{
+ if (IS_INVALID_HANDLE(hPool) ||
+ (hPool->dwType != APXHANDLE_TYPE_POOL))
+ hPool = _st_sys_pool;
+ return __apxPoolReallocCore(hPool, lpMem, dwNewSize, HEAP_ZERO_MEMORY);
+}
+
+LPVOID
+apxAlloc(DWORD dwSize)
+{
+ return __apxPoolAllocCore(_st_sys_pool, dwSize, 0);
+}
+
+LPVOID
+apxCalloc(DWORD dwSize)
+{
+ return __apxPoolAllocCore(_st_sys_pool, dwSize, HEAP_ZERO_MEMORY);
+}
+
+LPVOID
+apxRealloc(LPVOID lpMem, DWORD dwNewSize)
+{
+ return __apxPoolReallocCore(_st_sys_pool, lpMem, dwNewSize, HEAP_ZERO_MEMORY);
+}
+
+VOID
+apxFree(LPVOID lpMem)
+{
+ if (lpMem)
+ __apxPoolFreeCore(lpMem);
+
+}
+
+LPWSTR
+apxPoolWStrdupA(APXHANDLE hPool, LPCSTR szSource)
+{
+ if (szSource) {
+ LPWSTR szDest;
+ int cch = MultiByteToWideChar(CP_UTF8, 0, szSource, -1, NULL, 0);
+ szDest = (LPWSTR)apxPoolAlloc(hPool, cch * sizeof(WCHAR));
+ if (!MultiByteToWideChar(CP_UTF8, 0, szSource, -1, szDest, cch)) {
+ apxFree(szDest);
+ return NULL;
+ }
+ return szDest;
+ }
+ else
+ return NULL;
+}
+
+LPWSTR apxWStrdupA(LPCTSTR szSource)
+{
+ return apxPoolWStrdup(_st_sys_pool, szSource);
+}
+
+LPSTR
+apxPoolStrdupA(APXHANDLE hPool, LPCSTR szSource)
+{
+ if (szSource) {
+ LPSTR szDest;
+ DWORD l = lstrlenA(szSource);
+ szDest = apxPoolAlloc(hPool, l + 1);
+ lstrcpyA(szDest, szSource);
+ return szDest;
+ }
+ else
+ return NULL;
+}
+
+LPWSTR
+apxPoolStrdupW(APXHANDLE hPool, LPCWSTR szSource)
+{
+ if (szSource) {
+ LPWSTR szDest;
+ DWORD l = lstrlenW(szSource);
+ szDest = apxPoolAlloc(hPool, (l + 1) * sizeof(WCHAR));
+ lstrcpyW(szDest, szSource);
+ return szDest;
+ }
+ else
+ return NULL;
+}
+
+LPSTR
+apxStrdupA(LPCSTR szSource)
+{
+ return apxPoolStrdupA(_st_sys_pool, szSource);
+}
+
+LPWSTR
+apxStrdupW(LPCWSTR szSource)
+{
+ return apxPoolStrdupW(_st_sys_pool, szSource);
+}
+
+APXHANDLE
+apxHandleCreate(APXHANDLE hPool, DWORD dwFlags,
+ LPVOID lpData, DWORD dwDataSize,
+ LPAPXFNCALLBACK fnCallback)
+{
+ APXHANDLE hHandle;
+ LPAPXPOOL lpPool;
+
+ if (IS_INVALID_HANDLE(hPool))
+ hPool = _st_sys_pool;
+ if (hPool->dwType != APXHANDLE_TYPE_POOL) {
+ apxLogWrite(APXLOG_MARK_ERROR "Parent Handle type is not POOL %d",
+ hPool->dwType);
+ return INVALID_HANDLE_VALUE;
+ }
+ hHandle = __apxPoolAllocCore(hPool, APXHANDLE_SZ + dwDataSize,
+ HEAP_ZERO_MEMORY);
+
+ hHandle->hPool = hPool;
+ if (fnCallback)
+ hHandle->fnCallback = fnCallback;
+ else
+ hHandle->fnCallback = __apxHandleCallback;
+
+ if (dwFlags & APXHANDLE_TYPE_WINHANDLE) {
+ hHandle->dwFlags |= APXHANDLE_HAS_USERDATA;
+ hHandle->dwFlags |= APXHANDLE_TYPE_WINHANDLE;
+ hHandle->uData.hWinHandle = lpData;
+ }
+ else if (dwFlags & APXHANDLE_TYPE_LPTR) {
+ hHandle->dwFlags |= APXHANDLE_HAS_USERDATA;
+ hHandle->dwFlags |= APXHANDLE_TYPE_LPTR;
+ hHandle->uData.lpPtr = lpData;
+ }
+ else if (dwDataSize && lpData) {
+ hHandle->dwFlags |= APXHANDLE_HAS_USERDATA;
+ AplCopyMemory(APXHANDLE_DATA(hHandle), lpData, dwDataSize);
+ hHandle->dwSize = dwDataSize;
+ }
+
+ if (dwFlags & APXHANDLE_HAS_EVENT) {
+ /* Create the message event and message wathcing thread */
+ hHandle->hEventHandle = CreateEvent(NULL, TRUE, FALSE, NULL);
+ hHandle->hEventThread = CreateThread(NULL, 0, __apxHandleEventThread,
+ hHandle, 0,
+ &(hHandle->hEventThreadId));
+ if (IS_INVALID_HANDLE(hHandle->hEventThread)) {
+ SAFE_CLOSE_HANDLE(hHandle->hEventHandle);
+ }
+ else
+ hHandle->dwFlags |= APXHANDLE_HAS_EVENT;
+ }
+ TAILQ_INIT(&hHandle->lCallbacks);
+ /* Add the handle to the pool's object list */
+ lpPool = APXHANDLE_DATA(hPool);
+ APXHANDLE_SPINLOCK(hPool);
+ TAILQ_INSERT_HEAD(&lpPool->lHandles, hHandle, queue);
+ ++hPool->dwSize;
+ APXHANDLE_SPINUNLOCK(hPool);
+
+ return hHandle;
+}
+
+BOOL
+apxCloseHandle(APXHANDLE hObject)
+{
+ LPAPXPOOL lpPool;
+ APXCALLHOOK *lpCall;
+
+ if (IS_INVALID_HANDLE(hObject) || hObject->dwType == APXHANDLE_TYPE_INVALID)
+ return FALSE;
+ /* Call the user callback first */
+ (*hObject->fnCallback)(hObject, WM_CLOSE, 0, 0);
+ /* Now go through the callback chain */
+ TAILQ_FOREACH(lpCall, &hObject->lCallbacks, queue) {
+ (*lpCall->fnCallback)(hObject, WM_CLOSE, 0, 0);
+ TAILQ_REMOVE(&hObject->lCallbacks, lpCall, queue);
+ __apxPoolFreeCore(lpCall);
+ }
+
+ hObject->dwType = APXHANDLE_TYPE_INVALID;
+ if (hObject->dwFlags & APXHANDLE_HAS_EVENT) {
+ DWORD dwState;
+
+ SetEvent(hObject->hEventHandle);
+ dwState = WaitForSingleObject(hObject->hEventThread, 1000);
+ SAFE_CLOSE_HANDLE(hObject->hEventHandle);
+ if (dwState == WAIT_TIMEOUT)
+ TerminateThread(hObject->hEventThread, 0);
+ SAFE_CLOSE_HANDLE(hObject->hEventThread);
+
+ hObject->dwFlags &= ~APXHANDLE_HAS_EVENT;
+ }
+
+ /* finaly remove the object from the pool's object list */
+ if (!IS_INVALID_HANDLE(hObject->hPool)) {
+ lpPool = APXHANDLE_DATA(hObject->hPool);
+ APXHANDLE_SPINLOCK(hObject->hPool);
+ TAILQ_REMOVE(&lpPool->lHandles, hObject, queue);
+ hObject->hPool->dwSize--;
+ APXHANDLE_SPINUNLOCK(hObject->hPool);
+ __apxPoolFreeCore(hObject);
+ }
+ return TRUE;
+}
+
+LPVOID
+apxHandleGetUserData(APXHANDLE hObject)
+{
+ if (hObject->dwType == APXHANDLE_TYPE_INVALID)
+ return NULL;
+ if (hObject->dwFlags & APXHANDLE_HAS_USERDATA)
+ return APXHANDLE_DATA(hObject);
+ else
+ return hObject->uData.lpPtr;
+}
+
+LPVOID
+apxHandleSetUserData(APXHANDLE hObject, LPVOID lpData, DWORD dwDataSize)
+{
+ if (hObject->dwType == APXHANDLE_TYPE_INVALID)
+ return NULL;
+ if (hObject->dwFlags & APXHANDLE_HAS_USERDATA && hObject->dwSize > 0) {
+ AplCopyMemory(APXHANDLE_DATA(hObject), lpData,
+ MIN(hObject->dwSize, dwDataSize));
+ return APXHANDLE_DATA(hObject);
+ }
+ else {
+ LPVOID lpOrg = hObject->uData.lpPtr;
+ hObject->uData.lpPtr = lpData;
+ return lpOrg;
+ }
+}
+
+BOOL apxHandleSendMessage(APXHANDLE hObject, UINT uMsg, WPARAM wParam, LPARAM lParam)
+{
+ BOOL rv = TRUE;
+ APXCALLHOOK *lpCall;
+ if (hObject->dwType == APXHANDLE_TYPE_INVALID)
+ return FALSE;
+ /* Serialize requests to the callback */
+ APXHANDLE_SPINLOCK(hObject);
+ if (hObject->fnCallback)
+ rv = (*hObject->fnCallback)(hObject, uMsg, wParam, lParam);
+ TAILQ_FOREACH(lpCall, &hObject->lCallbacks, queue) {
+ (*lpCall->fnCallback)(hObject, uMsg, wParam, lParam);
+ }
+ APXHANDLE_SPINUNLOCK(hObject);
+
+ return rv;
+}
+
+BOOL apxHandlePostMessage(APXHANDLE hObject, UINT uMsg, WPARAM wParam, LPARAM lParam)
+{
+ if (hObject->dwType == APXHANDLE_TYPE_INVALID)
+ return FALSE;
+ if (hObject->dwFlags & APXHANDLE_HAS_EVENT) {
+ /* TODO: Create a thread message queue
+ * Right now wait while the event gets nonsignaled
+ */
+ while (WaitForSingleObject(hObject->hEventHandle, 0) == WAIT_OBJECT_0)
+ SwitchToThread();
+ APXHANDLE_SPINLOCK(hObject);
+ SuspendThread(hObject->hEventThread);
+ hObject->uMsg = uMsg;
+ hObject->wParam = wParam;
+ hObject->lParam = lParam;
+ /* Signal the event thread to call the user callback */
+ SetEvent(hObject->hEventHandle);
+ ResumeThread(hObject->hEventThread);
+ APXHANDLE_SPINUNLOCK(hObject);
+
+ return TRUE;
+ }
+ return FALSE;
+}
+
+BOOL apxHandleLock(APXHANDLE hObject, BOOL bLock)
+{
+ if (bLock)
+ APXHANDLE_SPINLOCK(hObject);
+ else
+ APXHANDLE_SPINUNLOCK(hObject);
+
+ return TRUE;
+}
+
+BOOL apxHandleAddHook(APXHANDLE hObject, DWORD dwWhere,
+ LPAPXFNCALLBACK fnCallback)
+{
+ APXCALLHOOK *lpCall;
+
+ if (hObject->dwType == APXHANDLE_TYPE_INVALID || !fnCallback)
+ return FALSE;
+ lpCall = (APXCALLHOOK *)__apxPoolAllocCore(hObject->hPool,
+ sizeof(APXCALLHOOK), 0);
+ if (!lpCall)
+ return FALSE;
+ lpCall->fnCallback = fnCallback;
+ APXHANDLE_SPINLOCK(hObject);
+ if (dwWhere == APXHANDLE_HOOK_FIRST) {
+ TAILQ_INSERT_HEAD(&hObject->lCallbacks, lpCall, queue);
+ }
+ else {
+ TAILQ_INSERT_TAIL(&hObject->lCallbacks, lpCall, queue);
+ }
+ APXHANDLE_SPINUNLOCK(hObject);
+
+ return TRUE;
+}
+
+DWORD apxHandleWait(APXHANDLE hHandle, DWORD dwMilliseconds, BOOL bKill)
+{
+ if (IS_INVALID_HANDLE(hHandle))
+ return WAIT_ABANDONED;
+ if (hHandle->dwType == APXHANDLE_TYPE_JVM)
+ return apxJavaWait(hHandle, dwMilliseconds, bKill);
+ else if (hHandle->dwType == APXHANDLE_TYPE_PROCESS)
+ return apxProcessWait(hHandle, dwMilliseconds, bKill);
+ else
+ return WAIT_ABANDONED;
+}
+
Deleted: tags/jbosssvc/2.0.8/procrun/src/javajni.c
===================================================================
--- trunk/utils/windows/native/service/procrun/src/javajni.c 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/src/javajni.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,784 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#include "apxwin.h"
-#include "private.h"
-
-#include <jni.h>
-
-#ifndef JNI_VERSION_1_2
-#error -------------------------------------------------------
-#error JAVA 1.1 IS NO LONGER SUPPORTED
-#error -------------------------------------------------------
-#endif
-
-#ifdef JNI_VERSION_1_4
-#define JNI_VERSION_DEFAULT JNI_VERSION_1_4
-#else
-#define JNI_VERSION_DEFAULT JNI_VERSION_1_2
-#endif
-
-/* Standard jvm.dll prototypes
- * since only single jvm can exist per process
- * make those global
- */
-
-DYNOLAD_TYPE_DECLARE(JNI_GetDefaultJavaVMInitArgs, JNICALL, jint)(void *);
-static DYNLOAD_FPTR_DECLARE(JNI_GetDefaultJavaVMInitArgs) = NULL;
-
-DYNOLAD_TYPE_DECLARE(JNI_CreateJavaVM, JNICALL, jint)(JavaVM **, void **, void *);
-static DYNLOAD_FPTR_DECLARE(JNI_CreateJavaVM) = NULL;
-
-DYNOLAD_TYPE_DECLARE(JNI_GetCreatedJavaVMs, JNICALL, jint)(JavaVM **, jsize, jsize *);
-static DYNLOAD_FPTR_DECLARE(JNI_GetCreatedJavaVMs) = NULL;
-
-static HANDLE _st_sys_jvmDllHandle = NULL;
-
-#define JVM_DELETE_CLAZZ(jvm, cl) \
- APXMACRO_BEGIN \
- if ((jvm)->lpEnv && (jvm)->##cl.jClazz) { \
- (*((jvm)->lpEnv))->DeleteGlobalRef((jvm)->lpEnv, (jvm)->##cl.jClazz); \
- (jvm)->##cl.jClazz = NULL; \
- } APXMACRO_END
-
-#define JVM_EXCEPTION_CHECK(jvm) \
- ((*((jvm)->lpEnv))->ExceptionCheck((jvm)->lpEnv) != JNI_OK)
-
-#ifdef _DEBUG
-#define JVM_EXCEPTION_CLEAR(jvm) \
- APXMACRO_BEGIN \
- if ((jvm)->lpEnv) { \
- if ((*((jvm)->lpEnv))->ExceptionCheck((jvm)->lpEnv)) { \
- (*((jvm)->lpEnv))->ExceptionDescribe((jvm)->lpEnv); \
- (*((jvm)->lpEnv))->ExceptionClear((jvm)->lpEnv); \
- } \
- } APXMACRO_END
-#else
-#define JVM_EXCEPTION_CLEAR(jvm) \
- APXMACRO_BEGIN \
- if ((jvm)->lpEnv) { \
- if ((*((jvm)->lpEnv))->ExceptionCheck((jvm)->lpEnv)) { \
- (*((jvm)->lpEnv))->ExceptionClear((jvm)->lpEnv); \
- } \
- } APXMACRO_END
-#endif
-
-#define JNI_LOCAL_UNREF(obj) \
- (*(lpJava->lpEnv))->DeleteLocalRef(lpJava->lpEnv, obj)
-
-#define JNICALL_0(fName) \
- ((*(lpJava->lpEnv))->##fName(lpJava->lpEnv))
-
-#define JNICALL_1(fName, a1) \
- ((*(lpJava->lpEnv))->##fName(lpJava->lpEnv, (a1)))
-
-#define JNICALL_2(fName, a1, a2) \
- ((*(lpJava->lpEnv))->##fName(lpJava->lpEnv, (a1), (a2)))
-
-#define JNICALL_3(fName, a1, a2, a3) \
- ((*(lpJava->lpEnv))->##fName(lpJava->lpEnv, (a1), (a2), (a3)))
-
-#define JNICALL_4(fName, a1, a2, a3, a4) \
- ((*(lpJava->lpEnv))->##fName(lpJava->lpEnv, (a1), (a2), (a3), (a4)))
-
-typedef struct APXJAVASTDCLAZZ {
- jclass jClazz;
- jmethodID jMethod;
- jobject jObject;
- jarray jArgs;
-} APXJAVASTDCLAZZ, *LPAPXJAVASTDCLAZZ;
-
-typedef struct APXJAVAVM {
- DWORD dwOptions;
- APXJAVASTDCLAZZ clString;
- APXJAVASTDCLAZZ clWorker;
- jint iVersion;
- jsize iVmCount;
- JNIEnv *lpEnv;
- JavaVM *lpJvm;
- /* JVM worker thread info */
- HANDLE hWorkerThread;
- DWORD iWorkerThread;
- DWORD dwWorkerStatus;
-
-} APXJAVAVM, *LPAPXJAVAVM;
-
-#define JAVA_CLASSPATH "-Djava.class.path="
-#define JAVA_CLASSSTRING "java/lang/String"
-
-static __inline BOOL __apxJvmAttach(LPAPXJAVAVM lpJava)
-{
- jint _iStatus = (*(lpJava->lpJvm))->GetEnv(lpJava->lpJvm,
- (void **)&(lpJava->lpEnv),
- lpJava->iVersion);
- if (_iStatus != JNI_OK) {
- if (_iStatus == JNI_EDETACHED)
- _iStatus = (*(lpJava->lpJvm))->AttachCurrentThread(lpJava->lpJvm,
- (void **)&(lpJava->lpEnv), NULL);
- }
- if (_iStatus != JNI_OK) {
- lpJava->lpEnv = NULL;
- return FALSE;
- }
- else
- return TRUE;
-}
-
-static __inline BOOL __apxJvmDetach(LPAPXJAVAVM lpJava)
-{
- jint _iStatus = (*(lpJava->lpJvm))->DetachCurrentThread(lpJava->lpJvm);
- if (_iStatus != JNI_OK) {
- lpJava->lpEnv = NULL;
- return FALSE;
- }
- else
- return TRUE;
-}
-
-static BOOL __apxLoadJvmDll(LPCWSTR szJvmDllPath)
-{
- UINT errMode;
- LPWSTR dllJvmPath = (LPWSTR)szJvmDllPath;
-
- if (!IS_INVALID_HANDLE(_st_sys_jvmDllHandle))
- return TRUE; /* jvm.dll is already loaded */
-
- if (!dllJvmPath || *dllJvmPath == L'\0')
- dllJvmPath = apxGetJavaSoftRuntimeLib(NULL);
- if (!dllJvmPath)
- return FALSE;
- /* Suppress the not found system popup message */
- errMode = SetErrorMode(SEM_FAILCRITICALERRORS);
-
- _st_sys_jvmDllHandle = LoadLibraryExW(dllJvmPath, NULL, 0);
- /* This shuldn't happen, but try to search in %PATH% */
- if (IS_INVALID_HANDLE(_st_sys_jvmDllHandle))
- _st_sys_jvmDllHandle = LoadLibraryExW(dllJvmPath, NULL,
- LOAD_WITH_ALTERED_SEARCH_PATH);
- /* Restore the error mode signalization */
- SetErrorMode(errMode);
- if (IS_INVALID_HANDLE(_st_sys_jvmDllHandle)) {
- apxLogWrite(APXLOG_MARK_SYSERR);
- return FALSE;
- }
- DYNLOAD_FPTR_LOAD(JNI_GetDefaultJavaVMInitArgs, _st_sys_jvmDllHandle);
- DYNLOAD_FPTR_LOAD(JNI_CreateJavaVM, _st_sys_jvmDllHandle);
- DYNLOAD_FPTR_LOAD(JNI_GetCreatedJavaVMs, _st_sys_jvmDllHandle);
-
- if (!DYNLOAD_FPTR(JNI_GetDefaultJavaVMInitArgs) ||
- !DYNLOAD_FPTR(JNI_CreateJavaVM) ||
- !DYNLOAD_FPTR(JNI_GetCreatedJavaVMs)) {
- apxLogWrite(APXLOG_MARK_SYSERR);
- FreeLibrary(_st_sys_jvmDllHandle);
- _st_sys_jvmDllHandle = NULL;
- return FALSE;
- }
-
- /* Real voodo ... */
- return TRUE;
-}
-
-static BOOL __apxJavaJniCallback(APXHANDLE hObject, UINT uMsg,
- WPARAM wParam, LPARAM lParam)
-{
- LPAPXJAVAVM lpJava;
- DWORD dwJvmRet = 0;
-
- lpJava = APXHANDLE_DATA(hObject);
- switch (uMsg) {
- case WM_CLOSE:
- if (lpJava->lpJvm) {
- if (!IS_INVALID_HANDLE(lpJava->hWorkerThread)) {
- if (GetExitCodeThread(lpJava->hWorkerThread, &dwJvmRet) &&
- dwJvmRet == STILL_ACTIVE) {
- TerminateThread(lpJava->hWorkerThread, 5);
- }
- }
- SAFE_CLOSE_HANDLE(lpJava->hWorkerThread);
- __apxJvmAttach(lpJava);
- JVM_DELETE_CLAZZ(lpJava, clWorker);
- JVM_DELETE_CLAZZ(lpJava, clString);
- __apxJvmDetach(lpJava);
- /* Check if this is the jvm loader */
- if (!lpJava->iVmCount && _st_sys_jvmDllHandle) {
-#if 0
- /* Do not destroy if we terminated the worker thread */
- if (dwJvmRet != STILL_ACTIVE)
- (*(lpJava->lpJvm))->DestroyJavaVM(lpJava->lpJvm);
-#endif
- /* Unload JVM dll */
- FreeLibrary(_st_sys_jvmDllHandle);
- _st_sys_jvmDllHandle = NULL;
- }
- lpJava->lpJvm = NULL;
- }
- break;
- default:
- break;
- }
- return TRUE;
-}
-
-APXHANDLE
-apxCreateJava(APXHANDLE hPool, LPCWSTR szJvmDllPath)
-{
-
- APXHANDLE hJava;
- LPAPXJAVAVM lpJava;
- jsize iVmCount;
- JavaVM *lpJvm = NULL;
-
- if (!__apxLoadJvmDll(szJvmDllPath))
- return NULL;
- /*
- */
- if (DYNLOAD_FPTR(JNI_GetCreatedJavaVMs)(&lpJvm, 1, &iVmCount) != JNI_OK) {
- return NULL;
- }
- if (iVmCount && !lpJvm)
- return NULL;
-
- hJava = apxHandleCreate(hPool, 0,
- NULL, sizeof(APXJAVAVM),
- __apxJavaJniCallback);
- if (IS_INVALID_HANDLE(hJava))
- return NULL;
- hJava->dwType = APXHANDLE_TYPE_JVM;
- lpJava = APXHANDLE_DATA(hJava);
- lpJava->lpJvm = lpJvm;
- lpJava->iVmCount = iVmCount;
- return hJava;
-}
-
-static DWORD __apxMultiSzToJvmOptions(APXHANDLE hPool,
- LPCSTR lpString,
- JavaVMOption **lppArray,
- DWORD nExtra)
-{
- DWORD i, n = 0, l = 0;
- char *buff;
- LPSTR p;
-
- if (lpString) {
- l = __apxGetMultiSzLengthA(lpString, &n);
- }
- n += nExtra;
- if (IS_INVALID_HANDLE(hPool))
- buff = apxPoolAlloc(hPool, (n + 1) * sizeof(JavaVMOption) + (l + 1));
- else
- buff = apxAlloc((n + 1) * sizeof(JavaVMOption) + (l + 1));
-
- *lppArray = (JavaVMOption *)buff;
- p = (LPSTR)(buff + (n + 1) * sizeof(JavaVMOption));
- if (lpString)
- AplCopyMemory(p, lpString, l + 1);
- for (i = 0; i < (n - nExtra); i++) {
- DWORD qr = apxStrUnQuoteInplaceA(p);
- (*lppArray)[i].optionString = p;
- while (*p)
- p++;
- p++;
- p += qr;
- }
-
- return n;
-}
-
-/* a hook for a function that redirects all VM messages. */
-static jint JNICALL __apxJniVfprintf(FILE *fp, const char *format, va_list args)
-{
- jint rv;
- CHAR sBuf[1024+16];
- rv = wvsprintfA(sBuf, format, args);
- if (apxLogWrite(APXLOG_MARK_INFO "%s", sBuf) == 0)
- fputs(sBuf, stdout);
- return rv;
-}
-
-
-/* ANSI version only */
-BOOL
-apxJavaInitialize(APXHANDLE hJava, LPCSTR szClassPath,
- LPCVOID lpOptions, DWORD dwMs, DWORD dwMx,
- DWORD dwSs)
-{
- LPAPXJAVAVM lpJava;
- JavaVMInitArgs vmArgs;
- JavaVMOption *lpJvmOptions;
- DWORD i, nOptions, sOptions = 2;
- BOOL rv = FALSE;
- if (hJava->dwType != APXHANDLE_TYPE_JVM)
- return FALSE;
-
- lpJava = APXHANDLE_DATA(hJava);
-
- if (lpJava->iVmCount) {
- if (!lpJava->lpEnv && !__apxJvmAttach(lpJava)) {
- if (lpJava->iVersion == JNI_VERSION_1_2) {
- apxLogWrite(APXLOG_MARK_ERROR "Unable To Attach the JVM");
- return FALSE;
- }
- else
- lpJava->iVersion = JNI_VERSION_1_2;
- if (!__apxJvmAttach(lpJava)) {
- apxLogWrite(APXLOG_MARK_ERROR "Unable To Attach the JVM");
- return FALSE;
- }
- }
- lpJava->iVersion = JNICALL_0(GetVersion);
- if (lpJava->iVersion < JNI_VERSION_1_2) {
- apxLogWrite(APXLOG_MARK_ERROR "Unsupported JNI version %#08x", lpJava->iVersion);
- return FALSE;
- }
- rv = TRUE;
- }
- else {
- CHAR iB[3][64];
- LPSTR szCp;
- lpJava->iVersion = JNI_VERSION_DEFAULT;
- if (dwMs)
- ++sOptions;
- if (dwMx)
- ++sOptions;
- if (dwSs)
- ++sOptions;
- nOptions = __apxMultiSzToJvmOptions(hJava->hPool, lpOptions,
- &lpJvmOptions, sOptions);
- szCp = apxPoolAlloc(hJava->hPool, sizeof(JAVA_CLASSPATH) + lstrlenA(szClassPath));
- lstrcpyA(szCp, JAVA_CLASSPATH);
- lstrcatA(szCp, szClassPath);
- lpJvmOptions[nOptions - sOptions].optionString = szCp;
- --sOptions;
- /* default JNI error printer */
- lpJvmOptions[nOptions - sOptions].optionString = "vfprintf";
- lpJvmOptions[nOptions - sOptions].extraInfo = __apxJniVfprintf;
- --sOptions;
- if (dwMs) {
- wsprintfA(iB[0], "-Xms%dm", dwMs);
- lpJvmOptions[nOptions - sOptions].optionString = iB[0];
- --sOptions;
- }
- if (dwMx) {
- wsprintfA(iB[1], "-Xmx%dm", dwMx);
- lpJvmOptions[nOptions - sOptions].optionString = iB[1];
- --sOptions;
- }
- if (dwSs) {
- wsprintfA(iB[2], "-Xss%dk", dwSs);
- lpJvmOptions[nOptions - sOptions].optionString = iB[2];
- --sOptions;
- }
- for (i = 0; i < nOptions; i++) {
- apxLogWrite(APXLOG_MARK_DEBUG "Jvm Option[%d] %s", i,
- lpJvmOptions[i].optionString);
- }
- vmArgs.options = lpJvmOptions;
- vmArgs.nOptions = nOptions;
- vmArgs.version = lpJava->iVersion;
- vmArgs.ignoreUnrecognized = JNI_FALSE;
- if (DYNLOAD_FPTR(JNI_CreateJavaVM)(&(lpJava->lpJvm),
- (void **)&(lpJava->lpEnv),
- &vmArgs) != JNI_OK) {
- apxLogWrite(APXLOG_MARK_ERROR "CreateJavaVM Failed");
- rv = FALSE;
- }
- else
- rv = TRUE;
- apxFree(szCp);
- apxFree(lpJvmOptions);
- }
- /* Load standard classes */
- if (rv) {
- jclass jClazz = JNICALL_1(FindClass, JAVA_CLASSSTRING);
- if (!jClazz) {
- apxLogWrite(APXLOG_MARK_ERROR "FindClass " JAVA_CLASSSTRING " failed");
- goto cleanup;
- }
- lpJava->clString.jClazz = JNICALL_1(NewGlobalRef, jClazz);
- JNI_LOCAL_UNREF(jClazz);
-
- return TRUE;
- }
- else
- return FALSE;
-
-cleanup:
- JVM_EXCEPTION_CLEAR(lpJava);
- return FALSE;
-}
-
-BOOL
-apxJavaLoadMainClass(APXHANDLE hJava, LPCSTR szClassName,
- LPCSTR szMethodName,
- LPCVOID lpArguments)
-{
- LPWSTR *lpArgs = NULL;
- DWORD nArgs;
- LPAPXJAVAVM lpJava;
- jclass jClazz;
-
- if (hJava->dwType != APXHANDLE_TYPE_JVM)
- return FALSE;
- lpJava = APXHANDLE_DATA(hJava);
- if (!__apxJvmAttach(lpJava))
- return FALSE;
-
- /* Find the class */
- jClazz = JNICALL_1(FindClass, szClassName);
- if (!jClazz) {
- JVM_EXCEPTION_CLEAR(lpJava);
- apxLogWrite(APXLOG_MARK_ERROR "FindClass %s failed", szClassName);
- return FALSE;
- }
- /* Make the class global so that worker thread can attach */
- lpJava->clWorker.jClazz = JNICALL_1(NewGlobalRef, jClazz);
- JNI_LOCAL_UNREF(jClazz);
-
- if (szMethodName)
- lpJava->clWorker.jMethod = JNICALL_3(GetStaticMethodID,
- lpJava->clWorker.jClazz,
- szMethodName, "([Ljava/lang/String;)V");
- else
- lpJava->clWorker.jMethod = JNICALL_3(GetStaticMethodID,
- lpJava->clWorker.jClazz,
- "main", "([Ljava/lang/String;)V");
- if (!lpJava->clWorker.jMethod) {
- JVM_EXCEPTION_CLEAR(lpJava);
- apxLogWrite(APXLOG_MARK_ERROR "Static method 'void main(String[])' in Class %s not found", szClassName);
- return FALSE;
- }
- nArgs = apxMultiSzToArrayW(hJava->hPool, lpArguments, &lpArgs);
- if (nArgs) {
- DWORD i;
- lpJava->clWorker.jArgs = JNICALL_3(NewObjectArray, nArgs,
- lpJava->clString.jClazz, NULL);
- for (i = 0; i < nArgs; i++) {
- jstring arg = JNICALL_2(NewString, lpArgs[i], lstrlenW(lpArgs[i]));
- JNICALL_3(SetObjectArrayElement, lpJava->clWorker.jArgs, i, arg);
- apxLogWrite(APXLOG_MARK_DEBUG "argv[%d] = %S", i, lpArgs[i]);
- }
- }
- apxFree(lpArgs);
- return TRUE;
-}
-
-
-/* Main java application worker thread
- * It will launch Java main and wait until
- * it finishes.
- */
-static DWORD WINAPI __apxJavaWorkerThread(LPVOID lpParameter)
-{
-#define WORKER_EXIT(x) { rv = x; goto finished; }
- DWORD rv = 0;
- LPAPXJAVAVM lpJava;
- APXHANDLE hJava = (APXHANDLE)lpParameter;
- /* This shouldn't happen */
- if (hJava->dwType != APXHANDLE_TYPE_JVM)
- WORKER_EXIT(0);
- lpJava = APXHANDLE_DATA(hJava);
- /* Check if we have a class and a method */
- if (!lpJava->clWorker.jClazz || !lpJava->clWorker.jMethod)
- WORKER_EXIT(2);
- if (!__apxJvmAttach(lpJava))
- WORKER_EXIT(3);
- lpJava->dwWorkerStatus = 1;
- JNICALL_3(CallStaticVoidMethod,
- lpJava->clWorker.jClazz,
- lpJava->clWorker.jMethod,
- lpJava->clWorker.jArgs);
-
- JVM_EXCEPTION_CLEAR(lpJava);
- __apxJvmDetach(lpJava);
-finished:
- lpJava->dwWorkerStatus = 0;
- apxLogWrite(APXLOG_MARK_DEBUG "Java Worker thread finished");
- ExitThread(rv);
- /* never gets here but keep the compiler happy */
- return 0;
-}
-
-
-BOOL
-apxJavaStart(APXHANDLE hJava)
-{
-
- LPAPXJAVAVM lpJava;
-
- if (hJava->dwType != APXHANDLE_TYPE_JVM)
- return FALSE;
- lpJava = APXHANDLE_DATA(hJava);
-
- lpJava->hWorkerThread = CreateThread(NULL, 0, __apxJavaWorkerThread,
- hJava, CREATE_SUSPENDED,
- &lpJava->iWorkerThread);
- if (IS_INVALID_HANDLE(lpJava->hWorkerThread)) {
- apxLogWrite(APXLOG_MARK_SYSERR);
- return FALSE;
- }
- ResumeThread(lpJava->hWorkerThread);
- /* Give some time to initialize the thread */
- Sleep(1000);
- return TRUE;
-}
-
-DWORD
-apxJavaWait(APXHANDLE hJava, DWORD dwMilliseconds, BOOL bKill)
-{
- DWORD rv;
- LPAPXJAVAVM lpJava;
-
- if (hJava->dwType != APXHANDLE_TYPE_JVM)
- return FALSE;
- lpJava = APXHANDLE_DATA(hJava);
-
- if (!lpJava->dwWorkerStatus && lpJava->hWorkerThread)
- return WAIT_OBJECT_0;
- rv = WaitForSingleObject(lpJava->hWorkerThread, dwMilliseconds);
- if (rv == WAIT_TIMEOUT && bKill) {
- __apxJavaJniCallback(hJava, WM_CLOSE, 0, 0);
- }
-
- return rv;
-}
-
-LPVOID
-apxJavaCreateClassV(APXHANDLE hJava, LPCSTR szClassName,
- LPCSTR szSignature, va_list lpArgs)
-{
- LPAPXJAVAVM lpJava;
- jclass clazz;
- jmethodID ccont;
- jobject cinst;
-
- if (hJava->dwType != APXHANDLE_TYPE_JVM)
- return NULL;
- lpJava = APXHANDLE_DATA(hJava);
- if (!__apxJvmAttach(lpJava))
- return NULL;
-
- clazz = JNICALL_1(FindClass, szClassName);
- if (clazz == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
- JVM_EXCEPTION_CLEAR(lpJava);
- apxLogWrite(APXLOG_MARK_ERROR "Could not FindClass %s", szClassName);
- return NULL;
- }
-
- ccont = JNICALL_3(GetMethodID, clazz, "<init>", szSignature);
- if (ccont == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
- JVM_EXCEPTION_CLEAR(lpJava);
- apxLogWrite(APXLOG_MARK_ERROR "Could not find Constructor %s for %s",
- szSignature, szClassName);
- return NULL;
- }
-
- cinst = JNICALL_3(NewObjectV, clazz, ccont, lpArgs);
- if (cinst == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
- JVM_EXCEPTION_CLEAR(lpJava);
- apxLogWrite(APXLOG_MARK_ERROR "Could not create instance of %s",
- szClassName);
- return NULL;
- }
-
- return cinst;
-}
-
-LPVOID
-apxJavaCreateClass(APXHANDLE hJava, LPCSTR szClassName,
- LPCSTR szSignature, ...)
-{
- LPVOID rv;
- va_list args;
-
- va_start(args, szSignature);
- rv = apxJavaCreateClassV(hJava, szClassName, szSignature, args);
- va_end(args);
-
- return rv;
-}
-
-LPVOID
-apxJavaCreateStringA(APXHANDLE hJava, LPCSTR szString)
-{
- LPAPXJAVAVM lpJava;
- jstring str;
-
- if (hJava->dwType != APXHANDLE_TYPE_JVM)
- return NULL;
- lpJava = APXHANDLE_DATA(hJava);
-
- str = JNICALL_1(NewStringUTF, szString);
- if (str == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
- JVM_EXCEPTION_CLEAR(lpJava);
- apxLogWrite(APXLOG_MARK_ERROR "Could not create string for %s",
- szString);
- return NULL;
- }
-
- return str;
-}
-
-LPVOID
-apxJavaCreateStringW(APXHANDLE hJava, LPCWSTR szString)
-{
- LPAPXJAVAVM lpJava;
- jstring str;
-
- if (hJava->dwType != APXHANDLE_TYPE_JVM)
- return NULL;
- lpJava = APXHANDLE_DATA(hJava);
-
- str = JNICALL_2(NewString, szString, lstrlenW(szString));
- if (str == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
- JVM_EXCEPTION_CLEAR(lpJava);
- apxLogWrite(APXLOG_MARK_ERROR "Could not create string for %S",
- szString);
- return NULL;
- }
-
- return str;
-}
-
-jvalue
-apxJavaCallStaticMethodV(APXHANDLE hJava, jclass lpClass, LPCSTR szMethodName,
- LPCSTR szSignature, va_list lpArgs)
-{
- LPAPXJAVAVM lpJava;
- jmethodID method;
- jvalue rv;
- LPCSTR s = szSignature;
- rv.l = 0;
- if (hJava->dwType != APXHANDLE_TYPE_JVM)
- return rv;
- lpJava = APXHANDLE_DATA(hJava);
-
- while (*s && *s != ')')
- ++s;
- if (*s != ')') {
- return rv;
- }
- else
- ++s;
- method = JNICALL_3(GetStaticMethodID, lpClass, szMethodName, szSignature);
- if (method == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
- JVM_EXCEPTION_CLEAR(lpJava);
- apxLogWrite(APXLOG_MARK_ERROR "Could not find method %s with signature %s",
- szMethodName, szSignature);
- return rv;
- }
- switch (*s) {
- case 'V':
- JNICALL_3(CallStaticVoidMethodV, lpClass, method, lpArgs);
- break;
- case 'L':
- case '[':
- rv.l = JNICALL_3(CallStaticObjectMethodV, lpClass, method, lpArgs);
- break;
- case 'Z':
- rv.z = JNICALL_3(CallStaticBooleanMethodV, lpClass, method, lpArgs);
- break;
- case 'B':
- rv.b = JNICALL_3(CallStaticByteMethodV, lpClass, method, lpArgs);
- break;
- case 'C':
- rv.c = JNICALL_3(CallStaticCharMethodV, lpClass, method, lpArgs);
- break;
- case 'S':
- rv.i = JNICALL_3(CallStaticShortMethodV, lpClass, method, lpArgs);
- break;
- case 'I':
- rv.i = JNICALL_3(CallStaticIntMethodV, lpClass, method, lpArgs);
- break;
- case 'J':
- rv.j = JNICALL_3(CallStaticLongMethodV, lpClass, method, lpArgs);
- break;
- case 'F':
- rv.f = JNICALL_3(CallStaticFloatMethodV, lpClass, method, lpArgs);
- break;
- case 'D':
- rv.d = JNICALL_3(CallStaticDoubleMethodV, lpClass, method, lpArgs);
- break;
- default:
- apxLogWrite(APXLOG_MARK_ERROR "Invalid signature %s for method %s",
- szSignature, szMethodName);
- return rv;
- break;
- }
-
- return rv;
-}
-
-jvalue
-apxJavaCallStaticMethod(APXHANDLE hJava, jclass lpClass, LPCSTR szMethodName,
- LPCSTR szSignature, ...)
-{
- jvalue rv;
- va_list args;
-
- va_start(args, szSignature);
- rv = apxJavaCallStaticMethodV(hJava, lpClass, szMethodName, szSignature, args);
- va_end(args);
-
- return rv;
-}
-
-/* Call the Java:
- * System.setOut(new PrintStream(new FileOutputStream(filename)));
- */
-BOOL
-apxJavaSetOut(APXHANDLE hJava, BOOL setErrorOrOut, LPCWSTR szFilename)
-{
- LPAPXJAVAVM lpJava;
- jobject fs;
- jobject ps;
- jstring fn;
- jclass sys;
-
- if (hJava->dwType != APXHANDLE_TYPE_JVM || !szFilename)
- return FALSE;
- lpJava = APXHANDLE_DATA(hJava);
- if (!__apxJvmAttach(lpJava))
- return FALSE;
-
- if ((fn = apxJavaCreateStringW(hJava, szFilename)) == NULL)
- return FALSE;
- if ((fs = apxJavaCreateClass(hJava, "java/io/FileOutputStream",
- "(Ljava/lang/String;Z)V", fn, JNI_TRUE)) == NULL)
- return FALSE;
- if ((ps = apxJavaCreateClass(hJava, "java/io/PrintStream",
- "(Ljava/io/OutputStream;)V", fs)) == NULL)
- return FALSE;
- sys = JNICALL_1(FindClass, "java/lang/System");
- if (sys == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
- JVM_EXCEPTION_CLEAR(lpJava);
- apxLogWrite(APXLOG_MARK_ERROR "Could not FindClass java/lang/System");
- return FALSE;
- }
-
- if (setErrorOrOut)
- apxJavaCallStaticMethod(hJava, sys, "setErr", "(Ljava/io/PrintStream;)V", ps);
- else
- apxJavaCallStaticMethod(hJava, sys, "setOut", "(Ljava/io/PrintStream;)V", ps);
-
- if (JVM_EXCEPTION_CHECK(lpJava)) {
- JVM_EXCEPTION_CLEAR(lpJava);
- apxLogWrite(APXLOG_MARK_ERROR "Error calling set method for java/lang/System");
- return FALSE;
- }
- else
- return TRUE;
-
-}
Copied: tags/jbosssvc/2.0.8/procrun/src/javajni.c (from rev 2689, trunk/utils/windows/native/service/procrun/src/javajni.c)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/src/javajni.c (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/src/javajni.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,970 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "apxwin.h"
+#include "private.h"
+
+#include <jni.h>
+
+#ifndef JNI_VERSION_1_2
+#error -------------------------------------------------------
+#error JAVA 1.1 IS NO LONGER SUPPORTED
+#error -------------------------------------------------------
+#endif
+
+#ifdef JNI_VERSION_1_4
+#define JNI_VERSION_DEFAULT JNI_VERSION_1_4
+#else
+#define JNI_VERSION_DEFAULT JNI_VERSION_1_2
+#endif
+
+/* Standard jvm.dll prototypes
+ * since only single jvm can exist per process
+ * make those global
+ */
+
+DYNOLAD_TYPE_DECLARE(JNI_GetDefaultJavaVMInitArgs, JNICALL, jint)(void *);
+static DYNLOAD_FPTR_DECLARE(JNI_GetDefaultJavaVMInitArgs) = NULL;
+
+DYNOLAD_TYPE_DECLARE(JNI_CreateJavaVM, JNICALL, jint)(JavaVM **, void **, void *);
+static DYNLOAD_FPTR_DECLARE(JNI_CreateJavaVM) = NULL;
+
+DYNOLAD_TYPE_DECLARE(JNI_GetCreatedJavaVMs, JNICALL, jint)(JavaVM **, jsize, jsize *);
+static DYNLOAD_FPTR_DECLARE(JNI_GetCreatedJavaVMs) = NULL;
+
+static HANDLE _st_sys_jvmDllHandle = NULL;
+static JavaVM *_st_sys_jvm = NULL;
+
+DYNOLAD_TYPE_DECLARE(SetDllDirectoryW, WINAPI, BOOL)(LPCWSTR);
+static DYNLOAD_FPTR_DECLARE(SetDllDirectoryW) = NULL;
+
+#define JVM_DELETE_CLAZZ(jvm, cl) \
+ APXMACRO_BEGIN \
+ if ((jvm)->lpEnv && (jvm)->cl.jClazz) { \
+ (*((jvm)->lpEnv))->DeleteGlobalRef((jvm)->lpEnv, (jvm)->cl.jClazz); \
+ (jvm)->cl.jClazz = NULL; \
+ } APXMACRO_END
+
+#define JVM_EXCEPTION_CHECK(jvm) \
+ ((*((jvm)->lpEnv))->ExceptionCheck((jvm)->lpEnv) != JNI_OK)
+
+#ifdef _DEBUG
+#define JVM_EXCEPTION_CLEAR(jvm) \
+ APXMACRO_BEGIN \
+ if ((jvm)->lpEnv) { \
+ if ((*((jvm)->lpEnv))->ExceptionCheck((jvm)->lpEnv)) { \
+ (*((jvm)->lpEnv))->ExceptionDescribe((jvm)->lpEnv); \
+ (*((jvm)->lpEnv))->ExceptionClear((jvm)->lpEnv); \
+ } \
+ } APXMACRO_END
+#else
+#define JVM_EXCEPTION_CLEAR(jvm) \
+ APXMACRO_BEGIN \
+ if ((jvm)->lpEnv) { \
+ if ((*((jvm)->lpEnv))->ExceptionCheck((jvm)->lpEnv)) { \
+ (*((jvm)->lpEnv))->ExceptionClear((jvm)->lpEnv); \
+ } \
+ } APXMACRO_END
+#endif
+
+#define JNI_LOCAL_UNREF(obj) \
+ (*(lpJava->lpEnv))->DeleteLocalRef(lpJava->lpEnv, obj)
+
+#define JNICALL_0(fName) \
+ ((*(lpJava->lpEnv))->fName(lpJava->lpEnv))
+
+#define JNICALL_1(fName, a1) \
+ ((*(lpJava->lpEnv))->fName(lpJava->lpEnv, (a1)))
+
+#define JNICALL_2(fName, a1, a2) \
+ ((*(lpJava->lpEnv))->fName(lpJava->lpEnv, (a1), (a2)))
+
+#define JNICALL_3(fName, a1, a2, a3) \
+ ((*(lpJava->lpEnv))->fName(lpJava->lpEnv, (a1), (a2), (a3)))
+
+#define JNICALL_4(fName, a1, a2, a3, a4) \
+ ((*(lpJava->lpEnv))->fName(lpJava->lpEnv, (a1), (a2), (a3), (a4)))
+
+typedef struct APXJAVASTDCLAZZ {
+ CHAR sClazz[1024];
+ CHAR sMethod[512];
+ jclass jClazz;
+ jmethodID jMethod;
+ jobject jObject;
+ jarray jArgs;
+} APXJAVASTDCLAZZ, *LPAPXJAVASTDCLAZZ;
+
+typedef struct APXJAVAVM {
+ DWORD dwOptions;
+ APXJAVASTDCLAZZ clString;
+ APXJAVASTDCLAZZ clWorker;
+ jint iVersion;
+ jsize iVmCount;
+ JNIEnv *lpEnv;
+ JavaVM *lpJvm;
+ /* JVM worker thread info */
+ HANDLE hWorkerThread;
+ DWORD iWorkerThread;
+ DWORD dwWorkerStatus;
+
+} APXJAVAVM, *LPAPXJAVAVM;
+
+#define JAVA_CLASSPATH "-Djava.class.path="
+#define JAVA_CLASSPATH_W L"-Djava.class.path="
+#define JAVA_CLASSSTRING "java/lang/String"
+
+static __inline BOOL __apxJvmAttach(LPAPXJAVAVM lpJava)
+{
+ jint _iStatus;
+
+ if (!_st_sys_jvm)
+ return FALSE;
+ _iStatus = (*(lpJava->lpJvm))->GetEnv(lpJava->lpJvm,
+ (void **)&(lpJava->lpEnv),
+ lpJava->iVersion);
+ if (_iStatus != JNI_OK) {
+ if (_iStatus == JNI_EDETACHED)
+ _iStatus = (*(lpJava->lpJvm))->AttachCurrentThread(lpJava->lpJvm,
+ (void **)&(lpJava->lpEnv), NULL);
+ }
+ if (_iStatus != JNI_OK) {
+ lpJava->lpEnv = NULL;
+ return FALSE;
+ }
+ else
+ return TRUE;
+}
+
+static __inline BOOL __apxJvmDetach(LPAPXJAVAVM lpJava)
+{
+ if (!_st_sys_jvm)
+ return FALSE;
+ if ((*(lpJava->lpJvm))->DetachCurrentThread(lpJava->lpJvm) != JNI_OK) {
+ lpJava->lpEnv = NULL;
+ return FALSE;
+ }
+ else
+ return TRUE;
+}
+
+static BOOL __apxLoadJvmDll(LPCWSTR szJvmDllPath)
+{
+ UINT errMode;
+ LPWSTR dllJvmPath = (LPWSTR)szJvmDllPath;
+ DYNLOAD_FPTR_DECLARE(SetDllDirectoryW);
+
+ if (!IS_INVALID_HANDLE(_st_sys_jvmDllHandle))
+ return TRUE; /* jvm.dll is already loaded */
+
+ if (!dllJvmPath || *dllJvmPath == L'\0')
+ dllJvmPath = apxGetJavaSoftRuntimeLib(NULL);
+ if (!dllJvmPath)
+ return FALSE;
+ /* Suppress the not found system popup message */
+ errMode = SetErrorMode(SEM_FAILCRITICALERRORS);
+
+ _st_sys_jvmDllHandle = LoadLibraryExW(dllJvmPath, NULL, 0);
+ /* This shuldn't happen, but try to search in %PATH% */
+ if (IS_INVALID_HANDLE(_st_sys_jvmDllHandle))
+ _st_sys_jvmDllHandle = LoadLibraryExW(dllJvmPath, NULL,
+ LOAD_WITH_ALTERED_SEARCH_PATH);
+
+ if (IS_INVALID_HANDLE(_st_sys_jvmDllHandle)) {
+ WCHAR jreBinPath[SIZ_PATHLEN];
+ DWORD i, l = 0;
+
+ lstrlcpyW(jreBinPath, SIZ_PATHLEN, dllJvmPath);
+ DYNLOAD_FPTR_ADDRESS(SetDllDirectoryW, KERNEL32);
+ for (i = lstrlenW(jreBinPath); i > 0, l < 2; i--) {
+ if (jreBinPath[i] == L'\\' || jreBinPath[i] == L'/') {
+ jreBinPath[i] = L'\0';
+ DYNLOAD_CALL(SetDllDirectoryW)(jreBinPath);
+ l++;
+ }
+ }
+ _st_sys_jvmDllHandle = LoadLibraryExW(dllJvmPath, NULL, 0);
+ if (IS_INVALID_HANDLE(_st_sys_jvmDllHandle))
+ _st_sys_jvmDllHandle = LoadLibraryExW(dllJvmPath, NULL,
+ LOAD_WITH_ALTERED_SEARCH_PATH);
+ }
+ /* Restore the error mode signalization */
+ SetErrorMode(errMode);
+ if (IS_INVALID_HANDLE(_st_sys_jvmDllHandle)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return FALSE;
+ }
+ DYNLOAD_FPTR_LOAD(JNI_GetDefaultJavaVMInitArgs, _st_sys_jvmDllHandle);
+ DYNLOAD_FPTR_LOAD(JNI_CreateJavaVM, _st_sys_jvmDllHandle);
+ DYNLOAD_FPTR_LOAD(JNI_GetCreatedJavaVMs, _st_sys_jvmDllHandle);
+
+ if (!DYNLOAD_FPTR(JNI_GetDefaultJavaVMInitArgs) ||
+ !DYNLOAD_FPTR(JNI_CreateJavaVM) ||
+ !DYNLOAD_FPTR(JNI_GetCreatedJavaVMs)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ FreeLibrary(_st_sys_jvmDllHandle);
+ _st_sys_jvmDllHandle = NULL;
+ return FALSE;
+ }
+
+ /* Real voodo ... */
+ return TRUE;
+}
+
+static BOOL __apxJavaJniCallback(APXHANDLE hObject, UINT uMsg,
+ WPARAM wParam, LPARAM lParam)
+{
+ LPAPXJAVAVM lpJava;
+ DWORD dwJvmRet = 0;
+
+ lpJava = APXHANDLE_DATA(hObject);
+ switch (uMsg) {
+ case WM_CLOSE:
+ if (_st_sys_jvm && lpJava->lpJvm) {
+ if (!IS_INVALID_HANDLE(lpJava->hWorkerThread)) {
+ if (GetExitCodeThread(lpJava->hWorkerThread, &dwJvmRet) &&
+ dwJvmRet == STILL_ACTIVE) {
+ TerminateThread(lpJava->hWorkerThread, 5);
+ }
+ }
+ SAFE_CLOSE_HANDLE(lpJava->hWorkerThread);
+ __apxJvmAttach(lpJava);
+ JVM_DELETE_CLAZZ(lpJava, clWorker);
+ JVM_DELETE_CLAZZ(lpJava, clString);
+ __apxJvmDetach(lpJava);
+ /* Check if this is the jvm loader */
+ if (!lpJava->iVmCount && _st_sys_jvmDllHandle) {
+ /* Unload JVM dll */
+ FreeLibrary(_st_sys_jvmDllHandle);
+ _st_sys_jvmDllHandle = NULL;
+ }
+ lpJava->lpJvm = NULL;
+ }
+ break;
+ default:
+ break;
+ }
+ return TRUE;
+}
+
+APXHANDLE
+apxCreateJava(APXHANDLE hPool, LPCWSTR szJvmDllPath)
+{
+
+ APXHANDLE hJava;
+ LPAPXJAVAVM lpJava;
+ jsize iVmCount;
+ JavaVM *lpJvm = NULL;
+
+ if (!__apxLoadJvmDll(szJvmDllPath))
+ return NULL;
+ /*
+ */
+ if (DYNLOAD_FPTR(JNI_GetCreatedJavaVMs)(&lpJvm, 1, &iVmCount) != JNI_OK) {
+ return NULL;
+ }
+ if (iVmCount && !lpJvm)
+ return NULL;
+
+ hJava = apxHandleCreate(hPool, 0,
+ NULL, sizeof(APXJAVAVM),
+ __apxJavaJniCallback);
+ if (IS_INVALID_HANDLE(hJava))
+ return NULL;
+ hJava->dwType = APXHANDLE_TYPE_JVM;
+ lpJava = APXHANDLE_DATA(hJava);
+ lpJava->lpJvm = lpJvm;
+ lpJava->iVmCount = iVmCount;
+ if (!_st_sys_jvm)
+ _st_sys_jvm = lpJvm;
+ return hJava;
+}
+
+static DWORD WINAPI __apxJavaDestroyThread(LPVOID lpParameter)
+{
+ JavaVM *lpJvm = (JavaVM *)lpParameter;
+ (*lpJvm)->DestroyJavaVM(lpJvm);
+ return 0;
+}
+
+BOOL
+apxDestroyJvm(DWORD dwTimeout)
+{
+ if (_st_sys_jvm) {
+ DWORD tid;
+ HANDLE hWaiter;
+ BOOL rv = FALSE;
+ JavaVM *lpJvm = _st_sys_jvm;
+
+ _st_sys_jvm = NULL;
+ (*lpJvm)->DetachCurrentThread(lpJvm);
+ hWaiter = CreateThread(NULL, 0, __apxJavaDestroyThread,
+ (void *)lpJvm, 0, &tid);
+ if (IS_INVALID_HANDLE(hWaiter)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return FALSE;
+ }
+ if (WaitForSingleObject(hWaiter, dwTimeout) == WAIT_OBJECT_0)
+ rv = TRUE;
+ CloseHandle(hWaiter);
+ return rv;
+ }
+ else
+ return FALSE;
+}
+
+static DWORD __apxMultiSzToJvmOptions(APXHANDLE hPool,
+ LPCSTR lpString,
+ JavaVMOption **lppArray,
+ DWORD nExtra)
+{
+ DWORD i, n = 0, l = 0;
+ char *buff;
+ LPSTR p;
+
+ if (lpString) {
+ l = __apxGetMultiSzLengthA(lpString, &n);
+ }
+ n += nExtra;
+ if (IS_INVALID_HANDLE(hPool))
+ buff = apxPoolAlloc(hPool, (n + 1) * sizeof(JavaVMOption) + (l + 1));
+ else
+ buff = apxAlloc((n + 1) * sizeof(JavaVMOption) + (l + 1));
+
+ *lppArray = (JavaVMOption *)buff;
+ p = (LPSTR)(buff + (n + 1) * sizeof(JavaVMOption));
+ if (lpString)
+ AplCopyMemory(p, lpString, l + 1);
+ for (i = 0; i < (n - nExtra); i++) {
+ DWORD qr = apxStrUnQuoteInplaceA(p);
+ (*lppArray)[i].optionString = p;
+ while (*p)
+ p++;
+ p++;
+ p += qr;
+ }
+
+ return n;
+}
+
+/* a hook for a function that redirects all VM messages. */
+static jint JNICALL __apxJniVfprintf(FILE *fp, const char *format, va_list args)
+{
+ jint rv;
+ CHAR sBuf[1024+16];
+ rv = wvsprintfA(sBuf, format, args);
+ if (apxLogWrite(APXLOG_MARK_INFO "%s", sBuf) == 0)
+ fputs(sBuf, stdout);
+ return rv;
+}
+
+
+/* ANSI version only */
+BOOL
+apxJavaInitialize(APXHANDLE hJava, LPCSTR szClassPath,
+ LPCVOID lpOptions, DWORD dwMs, DWORD dwMx,
+ DWORD dwSs, DWORD bJniVfprintf)
+{
+ LPAPXJAVAVM lpJava;
+ JavaVMInitArgs vmArgs;
+ JavaVMOption *lpJvmOptions;
+ DWORD i, nOptions, sOptions = 0;
+ BOOL rv = FALSE;
+ if (hJava->dwType != APXHANDLE_TYPE_JVM)
+ return FALSE;
+
+ lpJava = APXHANDLE_DATA(hJava);
+
+ if (lpJava->iVmCount) {
+ if (!lpJava->lpEnv && !__apxJvmAttach(lpJava)) {
+ if (lpJava->iVersion == JNI_VERSION_1_2) {
+ apxLogWrite(APXLOG_MARK_ERROR "Unable To Attach the JVM");
+ return FALSE;
+ }
+ else
+ lpJava->iVersion = JNI_VERSION_1_2;
+ if (!__apxJvmAttach(lpJava)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Unable To Attach the JVM");
+ return FALSE;
+ }
+ }
+ lpJava->iVersion = JNICALL_0(GetVersion);
+ if (lpJava->iVersion < JNI_VERSION_1_2) {
+ apxLogWrite(APXLOG_MARK_ERROR "Unsupported JNI version %#08x", lpJava->iVersion);
+ return FALSE;
+ }
+ rv = TRUE;
+ }
+ else {
+ CHAR iB[3][64];
+ LPSTR szCp;
+ lpJava->iVersion = JNI_VERSION_DEFAULT;
+ if (dwMs)
+ ++sOptions;
+ if (dwMx)
+ ++sOptions;
+ if (dwSs)
+ ++sOptions;
+ if (bJniVfprintf)
+ ++sOptions;
+ if (szClassPath && *szClassPath)
+ ++sOptions;
+ nOptions = __apxMultiSzToJvmOptions(hJava->hPool, lpOptions,
+ &lpJvmOptions, sOptions);
+ if (szClassPath && *szClassPath) {
+ szCp = apxPoolAlloc(hJava->hPool, sizeof(JAVA_CLASSPATH) + lstrlenA(szClassPath));
+ lstrcpyA(szCp, JAVA_CLASSPATH);
+ lstrcatA(szCp, szClassPath);
+ lpJvmOptions[nOptions - sOptions].optionString = szCp;
+ --sOptions;
+ }
+ if (bJniVfprintf) {
+ /* default JNI error printer */
+ lpJvmOptions[nOptions - sOptions].optionString = "vfprintf";
+ lpJvmOptions[nOptions - sOptions].extraInfo = __apxJniVfprintf;
+ --sOptions;
+ }
+ if (dwMs) {
+ wsprintfA(iB[0], "-Xms%dm", dwMs);
+ lpJvmOptions[nOptions - sOptions].optionString = iB[0];
+ --sOptions;
+ }
+ if (dwMx) {
+ wsprintfA(iB[1], "-Xmx%dm", dwMx);
+ lpJvmOptions[nOptions - sOptions].optionString = iB[1];
+ --sOptions;
+ }
+ if (dwSs) {
+ wsprintfA(iB[2], "-Xss%dk", dwSs);
+ lpJvmOptions[nOptions - sOptions].optionString = iB[2];
+ --sOptions;
+ }
+ for (i = 0; i < nOptions; i++) {
+ apxLogWrite(APXLOG_MARK_DEBUG "Jvm Option[%d] %s", i,
+ lpJvmOptions[i].optionString);
+ }
+ vmArgs.options = lpJvmOptions;
+ vmArgs.nOptions = nOptions;
+ vmArgs.version = lpJava->iVersion;
+ vmArgs.ignoreUnrecognized = JNI_FALSE;
+ if (DYNLOAD_FPTR(JNI_CreateJavaVM)(&(lpJava->lpJvm),
+ (void **)&(lpJava->lpEnv),
+ &vmArgs) != JNI_OK) {
+ apxLogWrite(APXLOG_MARK_ERROR "CreateJavaVM Failed");
+ rv = FALSE;
+ }
+ else {
+ rv = TRUE;
+ if (!_st_sys_jvm)
+ _st_sys_jvm = lpJava->lpJvm;
+ }
+ apxFree(szCp);
+ apxFree(lpJvmOptions);
+ }
+ /* Load standard classes */
+ if (rv) {
+ jclass jClazz = JNICALL_1(FindClass, JAVA_CLASSSTRING);
+ if (!jClazz) {
+ apxLogWrite(APXLOG_MARK_ERROR "FindClass " JAVA_CLASSSTRING " failed");
+ goto cleanup;
+ }
+ lpJava->clString.jClazz = JNICALL_1(NewGlobalRef, jClazz);
+ JNI_LOCAL_UNREF(jClazz);
+
+ return TRUE;
+ }
+ else
+ return FALSE;
+
+cleanup:
+ JVM_EXCEPTION_CLEAR(lpJava);
+ return FALSE;
+}
+
+/* ANSI version only */
+DWORD
+apxJavaCmdInitialize(APXHANDLE hPool, LPCWSTR szClassPath, LPCWSTR szClass,
+ LPCWSTR szOptions, DWORD dwMs, DWORD dwMx,
+ DWORD dwSs, LPCWSTR szCmdArgs, LPWSTR **lppArray)
+{
+
+ DWORD i, nJVM, nCmd, nTotal, lJVM, lCmd;
+ LPWSTR p;
+
+ /* Calculate the number of all arguments */
+ nTotal = 0;
+ if (szClassPath)
+ ++nTotal;
+ if (szClass)
+ ++nTotal;
+ lJVM = __apxGetMultiSzLengthW(szOptions, &nJVM);
+ nTotal += nJVM;
+ lCmd = __apxGetMultiSzLengthW(szCmdArgs, &nCmd);
+ nTotal += nCmd;
+ if (dwMs)
+ ++nTotal;
+ if (dwMx)
+ ++nTotal;
+ if (dwSs)
+ ++nTotal;
+
+ if (nTotal == 0)
+ return 0;
+
+ /* Allocate the array to store all arguments' pointers
+ */
+ *lppArray = (LPWSTR *)apxPoolAlloc(hPool, (nTotal + 2) * sizeof(LPWSTR));
+
+ /* Process JVM options */
+ if (nJVM && lJVM) {
+ p = (LPWSTR)apxPoolAlloc(hPool, (lJVM + 1) * sizeof(WCHAR));
+ AplCopyMemory(p, szOptions, (lJVM + 1) * sizeof(WCHAR) + sizeof(WCHAR));
+ for (i = 0; i < nJVM; i++) {
+ (*lppArray)[i] = p;
+ while (*p)
+ p++;
+ p++;
+ }
+ }
+
+ /* Process the 3 extra JVM options */
+ if (dwMs) {
+ p = (LPWSTR)apxPoolAlloc(hPool, 64 * sizeof(WCHAR));
+ wsprintfW(p, L"-Xms%dm", dwMs);
+ (*lppArray)[i++] = p;
+ }
+ if (dwMx) {
+ p = (LPWSTR)apxPoolAlloc(hPool, 64 * sizeof(WCHAR));
+ wsprintfW(p, L"-Xmx%dm", dwMx);
+ (*lppArray)[i++] = p;
+ }
+ if (dwSs) {
+ p = (LPWSTR)apxPoolAlloc(hPool, 64 * sizeof(WCHAR));
+ wsprintfW(p, L"-Xss%dk", dwSs);
+ (*lppArray)[i++] = p;
+ }
+
+ /* Process the classpath and class */
+ if (szClassPath) {
+ p = (LPWSTR)apxPoolAlloc(hPool, (lstrlenW(JAVA_CLASSPATH_W) + lstrlenW(szClassPath)) * sizeof(WCHAR));
+ lstrcpyW(p, JAVA_CLASSPATH_W);
+ lstrcatW(p, szClassPath);
+ (*lppArray)[i++] = p;
+ }
+ if (szClass) {
+ p = (LPWSTR)apxPoolAlloc(hPool, (lstrlenW(szClass)) * sizeof(WCHAR));
+ lstrcpyW(p, szClass);
+ (*lppArray)[i++] = p;
+ }
+
+ /* Process command arguments */
+ if (nCmd && lCmd) {
+ p = (LPWSTR)apxPoolAlloc(hPool, (lCmd + 1) * sizeof(WCHAR));
+ AplCopyMemory(p, szCmdArgs, (lCmd + 1) * sizeof(WCHAR) + sizeof(WCHAR));
+ for (; i < nTotal; i++) {
+ (*lppArray)[i] = p;
+ while (*p)
+ p++;
+ p++;
+ }
+ }
+
+ (*lppArray)[++i] = NULL;
+
+ return nTotal;
+}
+
+
+BOOL
+apxJavaLoadMainClass(APXHANDLE hJava, LPCSTR szClassName,
+ LPCSTR szMethodName,
+ LPCVOID lpArguments)
+{
+ LPWSTR *lpArgs = NULL;
+ DWORD nArgs;
+ LPAPXJAVAVM lpJava;
+ jclass jClazz;
+
+ if (hJava->dwType != APXHANDLE_TYPE_JVM)
+ return FALSE;
+ lpJava = APXHANDLE_DATA(hJava);
+ if (!__apxJvmAttach(lpJava))
+ return FALSE;
+
+ /* Find the class */
+ jClazz = JNICALL_1(FindClass, szClassName);
+ if (!jClazz) {
+ JVM_EXCEPTION_CLEAR(lpJava);
+ apxLogWrite(APXLOG_MARK_ERROR "FindClass %s failed", szClassName);
+ return FALSE;
+ }
+ /* Make the class global so that worker thread can attach */
+ lpJava->clWorker.jClazz = JNICALL_1(NewGlobalRef, jClazz);
+ JNI_LOCAL_UNREF(jClazz);
+
+ if (!szMethodName)
+ szMethodName = "main";
+ lstrlcpyA(lpJava->clWorker.sClazz, 1024, szClassName);
+ lstrlcpyA(lpJava->clWorker.sMethod, 512, szMethodName);
+ lpJava->clWorker.jMethod = JNICALL_3(GetStaticMethodID,
+ lpJava->clWorker.jClazz,
+ szMethodName, "([Ljava/lang/String;)V");
+ if (!lpJava->clWorker.jMethod) {
+ JVM_EXCEPTION_CLEAR(lpJava);
+ apxLogWrite(APXLOG_MARK_ERROR "Method 'static void %s(String[])' not found in Class %s",
+ szMethodName, szClassName);
+ return FALSE;
+ }
+ nArgs = apxMultiSzToArrayW(hJava->hPool, lpArguments, &lpArgs);
+ lpJava->clWorker.jArgs = JNICALL_3(NewObjectArray, nArgs,
+ lpJava->clString.jClazz, NULL);
+ if (nArgs) {
+ DWORD i;
+ for (i = 0; i < nArgs; i++) {
+ jstring arg = JNICALL_2(NewString, lpArgs[i], lstrlenW(lpArgs[i]));
+ JNICALL_3(SetObjectArrayElement, lpJava->clWorker.jArgs, i, arg);
+ apxLogWrite(APXLOG_MARK_DEBUG "argv[%d] = %S", i, lpArgs[i]);
+ }
+ }
+ apxFree(lpArgs);
+ return TRUE;
+}
+
+/* Main java application worker thread
+ * It will launch Java main and wait until
+ * it finishes.
+ */
+static DWORD WINAPI __apxJavaWorkerThread(LPVOID lpParameter)
+{
+#define WORKER_EXIT(x) { rv = x; goto finished; }
+ DWORD rv = 0;
+ LPAPXJAVAVM lpJava;
+ APXHANDLE hJava = (APXHANDLE)lpParameter;
+ /* This shouldn't happen */
+ if (hJava->dwType != APXHANDLE_TYPE_JVM)
+ WORKER_EXIT(0);
+ lpJava = APXHANDLE_DATA(hJava);
+ /* Check if we have a class and a method */
+ if (!lpJava->clWorker.jClazz || !lpJava->clWorker.jMethod)
+ WORKER_EXIT(2);
+ if (!__apxJvmAttach(lpJava))
+ WORKER_EXIT(3);
+ apxLogWrite(APXLOG_MARK_DEBUG "Java Worker thread started %s:%s",
+ lpJava->clWorker.sClazz, lpJava->clWorker.sMethod);
+ lpJava->dwWorkerStatus = 1;
+ JNICALL_3(CallStaticVoidMethod,
+ lpJava->clWorker.jClazz,
+ lpJava->clWorker.jMethod,
+ lpJava->clWorker.jArgs);
+
+ JVM_EXCEPTION_CLEAR(lpJava);
+ __apxJvmDetach(lpJava);
+ apxLogWrite(APXLOG_MARK_DEBUG "Java Worker thread %s:%s finished",
+ lpJava->clWorker.sClazz, lpJava->clWorker.sMethod);
+finished:
+ lpJava->dwWorkerStatus = 0;
+ apxLogWrite(APXLOG_MARK_DEBUG "Java Worker thread finished %s:%s",
+ lpJava->clWorker.sClazz, lpJava->clWorker.sMethod);
+ ExitThread(rv);
+ /* never gets here but keep the compiler happy */
+ return 0;
+}
+
+
+BOOL
+apxJavaStart(APXHANDLE hJava)
+{
+
+ LPAPXJAVAVM lpJava;
+
+ if (hJava->dwType != APXHANDLE_TYPE_JVM)
+ return FALSE;
+ lpJava = APXHANDLE_DATA(hJava);
+
+ lpJava->hWorkerThread = CreateThread(NULL, 0, __apxJavaWorkerThread,
+ hJava, CREATE_SUSPENDED,
+ &lpJava->iWorkerThread);
+ if (IS_INVALID_HANDLE(lpJava->hWorkerThread)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return FALSE;
+ }
+ ResumeThread(lpJava->hWorkerThread);
+ /* Give some time to initialize the thread */
+ Sleep(1000);
+ return TRUE;
+}
+
+DWORD
+apxJavaSetOptions(APXHANDLE hJava, DWORD dwOptions)
+{
+ DWORD dwOrgOptions;
+ LPAPXJAVAVM lpJava;
+
+ if (hJava->dwType != APXHANDLE_TYPE_JVM)
+ return 0;
+ lpJava = APXHANDLE_DATA(hJava);
+ dwOrgOptions = lpJava->dwOptions;
+ lpJava->dwOptions = dwOptions;
+ return dwOrgOptions;
+}
+
+DWORD
+apxJavaWait(APXHANDLE hJava, DWORD dwMilliseconds, BOOL bKill)
+{
+ DWORD rv;
+ LPAPXJAVAVM lpJava;
+
+ if (hJava->dwType != APXHANDLE_TYPE_JVM)
+ return FALSE;
+ lpJava = APXHANDLE_DATA(hJava);
+
+ if (!lpJava->dwWorkerStatus && lpJava->hWorkerThread)
+ return WAIT_OBJECT_0;
+ rv = WaitForSingleObject(lpJava->hWorkerThread, dwMilliseconds);
+ if (rv == WAIT_TIMEOUT && bKill) {
+ __apxJavaJniCallback(hJava, WM_CLOSE, 0, 0);
+ }
+
+ return rv;
+}
+
+LPVOID
+apxJavaCreateClassV(APXHANDLE hJava, LPCSTR szClassName,
+ LPCSTR szSignature, va_list lpArgs)
+{
+ LPAPXJAVAVM lpJava;
+ jclass clazz;
+ jmethodID ccont;
+ jobject cinst;
+
+ if (hJava->dwType != APXHANDLE_TYPE_JVM)
+ return NULL;
+ lpJava = APXHANDLE_DATA(hJava);
+ if (!__apxJvmAttach(lpJava))
+ return NULL;
+
+ clazz = JNICALL_1(FindClass, szClassName);
+ if (clazz == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
+ JVM_EXCEPTION_CLEAR(lpJava);
+ apxLogWrite(APXLOG_MARK_ERROR "Could not FindClass %s", szClassName);
+ return NULL;
+ }
+
+ ccont = JNICALL_3(GetMethodID, clazz, "<init>", szSignature);
+ if (ccont == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
+ JVM_EXCEPTION_CLEAR(lpJava);
+ apxLogWrite(APXLOG_MARK_ERROR "Could not find Constructor %s for %s",
+ szSignature, szClassName);
+ return NULL;
+ }
+
+ cinst = JNICALL_3(NewObjectV, clazz, ccont, lpArgs);
+ if (cinst == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
+ JVM_EXCEPTION_CLEAR(lpJava);
+ apxLogWrite(APXLOG_MARK_ERROR "Could not create instance of %s",
+ szClassName);
+ return NULL;
+ }
+
+ return cinst;
+}
+
+LPVOID
+apxJavaCreateClass(APXHANDLE hJava, LPCSTR szClassName,
+ LPCSTR szSignature, ...)
+{
+ LPVOID rv;
+ va_list args;
+
+ va_start(args, szSignature);
+ rv = apxJavaCreateClassV(hJava, szClassName, szSignature, args);
+ va_end(args);
+
+ return rv;
+}
+
+LPVOID
+apxJavaCreateStringA(APXHANDLE hJava, LPCSTR szString)
+{
+ LPAPXJAVAVM lpJava;
+ jstring str;
+
+ if (hJava->dwType != APXHANDLE_TYPE_JVM)
+ return NULL;
+ lpJava = APXHANDLE_DATA(hJava);
+
+ str = JNICALL_1(NewStringUTF, szString);
+ if (str == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
+ JVM_EXCEPTION_CLEAR(lpJava);
+ apxLogWrite(APXLOG_MARK_ERROR "Could not create string for %s",
+ szString);
+ return NULL;
+ }
+
+ return str;
+}
+
+LPVOID
+apxJavaCreateStringW(APXHANDLE hJava, LPCWSTR szString)
+{
+ LPAPXJAVAVM lpJava;
+ jstring str;
+
+ if (hJava->dwType != APXHANDLE_TYPE_JVM)
+ return NULL;
+ lpJava = APXHANDLE_DATA(hJava);
+
+ str = JNICALL_2(NewString, szString, lstrlenW(szString));
+ if (str == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
+ JVM_EXCEPTION_CLEAR(lpJava);
+ apxLogWrite(APXLOG_MARK_ERROR "Could not create string for %S",
+ szString);
+ return NULL;
+ }
+
+ return str;
+}
+
+jvalue
+apxJavaCallStaticMethodV(APXHANDLE hJava, jclass lpClass, LPCSTR szMethodName,
+ LPCSTR szSignature, va_list lpArgs)
+{
+ LPAPXJAVAVM lpJava;
+ jmethodID method;
+ jvalue rv;
+ LPCSTR s = szSignature;
+ rv.l = 0;
+ if (hJava->dwType != APXHANDLE_TYPE_JVM)
+ return rv;
+ lpJava = APXHANDLE_DATA(hJava);
+
+ while (*s && *s != ')')
+ ++s;
+ if (*s != ')') {
+ return rv;
+ }
+ else
+ ++s;
+ method = JNICALL_3(GetStaticMethodID, lpClass, szMethodName, szSignature);
+ if (method == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
+ JVM_EXCEPTION_CLEAR(lpJava);
+ apxLogWrite(APXLOG_MARK_ERROR "Could not find method %s with signature %s",
+ szMethodName, szSignature);
+ return rv;
+ }
+ switch (*s) {
+ case 'V':
+ JNICALL_3(CallStaticVoidMethodV, lpClass, method, lpArgs);
+ break;
+ case 'L':
+ case '[':
+ rv.l = JNICALL_3(CallStaticObjectMethodV, lpClass, method, lpArgs);
+ break;
+ case 'Z':
+ rv.z = JNICALL_3(CallStaticBooleanMethodV, lpClass, method, lpArgs);
+ break;
+ case 'B':
+ rv.b = JNICALL_3(CallStaticByteMethodV, lpClass, method, lpArgs);
+ break;
+ case 'C':
+ rv.c = JNICALL_3(CallStaticCharMethodV, lpClass, method, lpArgs);
+ break;
+ case 'S':
+ rv.i = JNICALL_3(CallStaticShortMethodV, lpClass, method, lpArgs);
+ break;
+ case 'I':
+ rv.i = JNICALL_3(CallStaticIntMethodV, lpClass, method, lpArgs);
+ break;
+ case 'J':
+ rv.j = JNICALL_3(CallStaticLongMethodV, lpClass, method, lpArgs);
+ break;
+ case 'F':
+ rv.f = JNICALL_3(CallStaticFloatMethodV, lpClass, method, lpArgs);
+ break;
+ case 'D':
+ rv.d = JNICALL_3(CallStaticDoubleMethodV, lpClass, method, lpArgs);
+ break;
+ default:
+ apxLogWrite(APXLOG_MARK_ERROR "Invalid signature %s for method %s",
+ szSignature, szMethodName);
+ return rv;
+ break;
+ }
+
+ return rv;
+}
+
+jvalue
+apxJavaCallStaticMethod(APXHANDLE hJava, jclass lpClass, LPCSTR szMethodName,
+ LPCSTR szSignature, ...)
+{
+ jvalue rv;
+ va_list args;
+
+ va_start(args, szSignature);
+ rv = apxJavaCallStaticMethodV(hJava, lpClass, szMethodName, szSignature, args);
+ va_end(args);
+
+ return rv;
+}
+
+/* Call the Java:
+ * System.setOut(new PrintStream(new FileOutputStream(filename)));
+ */
+BOOL
+apxJavaSetOut(APXHANDLE hJava, BOOL setErrorOrOut, LPCWSTR szFilename)
+{
+ LPAPXJAVAVM lpJava;
+ jobject fs;
+ jobject ps;
+ jstring fn;
+ jclass sys;
+
+ if (hJava->dwType != APXHANDLE_TYPE_JVM || !szFilename)
+ return FALSE;
+ lpJava = APXHANDLE_DATA(hJava);
+ if (!__apxJvmAttach(lpJava))
+ return FALSE;
+
+ if ((fn = apxJavaCreateStringW(hJava, szFilename)) == NULL)
+ return FALSE;
+ if ((fs = apxJavaCreateClass(hJava, "java/io/FileOutputStream",
+ "(Ljava/lang/String;Z)V", fn, JNI_TRUE)) == NULL)
+ return FALSE;
+ if ((ps = apxJavaCreateClass(hJava, "java/io/PrintStream",
+ "(Ljava/io/OutputStream;)V", fs)) == NULL)
+ return FALSE;
+ sys = JNICALL_1(FindClass, "java/lang/System");
+ if (sys == NULL || (JVM_EXCEPTION_CHECK(lpJava))) {
+ JVM_EXCEPTION_CLEAR(lpJava);
+ apxLogWrite(APXLOG_MARK_ERROR "Could not FindClass java/lang/System");
+ return FALSE;
+ }
+
+ if (setErrorOrOut)
+ apxJavaCallStaticMethod(hJava, sys, "setErr", "(Ljava/io/PrintStream;)V", ps);
+ else
+ apxJavaCallStaticMethod(hJava, sys, "setOut", "(Ljava/io/PrintStream;)V", ps);
+
+ if (JVM_EXCEPTION_CHECK(lpJava)) {
+ JVM_EXCEPTION_CLEAR(lpJava);
+ apxLogWrite(APXLOG_MARK_ERROR "Error calling set method for java/lang/System");
+ return FALSE;
+ }
+ else
+ return TRUE;
+
+}
+
Deleted: tags/jbosssvc/2.0.8/procrun/src/log.c
===================================================================
--- trunk/utils/windows/native/service/procrun/src/log.c 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/src/log.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,410 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#include "apxwin.h"
-#include "private.h"
-
-#define LINE_SEP "\n"
-
-static LPCSTR _log_level[] = {
- "[debug] ",
- "[info] ",
- "[warn] ",
- "[error] ",
- NULL
-};
-
-typedef struct apx_logfile_st {
- HANDLE hFile;
- DWORD dwLogLevel;
- BOOL bRotate;
- SYSTEMTIME sysTime;
- WCHAR szPath[MAX_PATH + 1];
- WCHAR szPrefix[MAX_PATH];
-} apx_logfile_st;
-
-/* Per-application master log file */
-static apx_logfile_st *_st_sys_loghandle = NULL;
-
-static apx_logfile_st _st_sys_errhandle = { NULL, APXLOG_LEVEL_WARN, FALSE};
-
-
-LPWSTR apxLogFile(
- APXHANDLE hPool,
- LPCWSTR szPath,
- LPCWSTR szPrefix,
- LPCWSTR szName)
-{
- LPWSTR sRet;
- WCHAR sPath[MAX_PATH+1];
- WCHAR sName[MAX_PATH+1];
- SYSTEMTIME sysTime;
-
- GetLocalTime(&sysTime);
- if (!szPath) {
- if (GetSystemDirectoryW(sPath, MAX_PATH) == 0)
- return INVALID_HANDLE_VALUE;
- lstrcatW(sPath, L"\\LogFiles\\");
- if (!szPrefix)
- lstrcatW(sPath, L"Apache");
- else
- lstrcatW(sPath, szPrefix);
- wsprintfW(sName, L"\\%s%04d%02d%02d.log",
- szName,
- sysTime.wYear,
- sysTime.wMonth,
- sysTime.wDay);
- }
- else {
- lstrcpyW(sPath, szPath);
- if (szPrefix)
- wsprintfW(sName, L"\\%s", szPrefix);
- else
- wsprintfW(sName, L"\\%s%04d%02d%02d.log",
- szName,
- sysTime.wYear,
- sysTime.wMonth,
- sysTime.wDay);
- }
- sRet = apxPoolAlloc(hPool, (MAX_PATH + 1) * sizeof(WCHAR));
- /* Set default level to info */
- CreateDirectoryW(sPath, NULL);
-
- lstrcpyW(sRet, sPath);
- lstrcatW(sRet, sName);
-
- return sRet;
-}
-
-/* Open the log file
- * TODO: format like standard apache error.log
- * Add the EventLogger
- */
-HANDLE apxLogOpen(
- APXHANDLE hPool,
- LPCWSTR szPath,
- LPCWSTR szPrefix)
-{
-
- WCHAR sPath[MAX_PATH+1];
- WCHAR sName[MAX_PATH+1];
- SYSTEMTIME sysTime;
- apx_logfile_st *h;
-
- GetLocalTime(&sysTime);
- if (!szPath) {
- if (GetSystemDirectoryW(sPath, MAX_PATH) == 0)
- return INVALID_HANDLE_VALUE;
- lstrcatW(sPath, L"\\LogFiles\\");
- if (!szPrefix)
- lstrcatW(sPath, L"Apache");
- else
- lstrcatW(sPath, szPrefix);
- wsprintfW(sName, L"\\%04d%02d%02d.log",
- sysTime.wYear,
- sysTime.wMonth,
- sysTime.wDay);
- }
- else {
- lstrcpyW(sPath, szPath);
- if (szPrefix)
- wsprintfW(sName, L"\\%s", szPrefix);
- else
- wsprintfW(sName, L"\\jakarta_service_%04d%02d%02d.log",
- sysTime.wYear,
- sysTime.wMonth,
- sysTime.wDay);
- }
- if (!(h = (apx_logfile_st *)apxPoolCalloc(hPool, sizeof(apx_logfile_st))))
- return NULL;
- /* Set default level to info */
- h->dwLogLevel = APXLOG_LEVEL_INFO;
- CreateDirectoryW(sPath, NULL);
-
- h->sysTime = sysTime;
- lstrcpyW(h->szPath, sPath);
- lstrcatW(sPath, sName);
- if (szPrefix)
- lstrcpyW(h->szPrefix, szPrefix);
-
- h->hFile = CreateFileW(sPath,
- GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE,
- NULL,
- OPEN_ALWAYS,
- FILE_ATTRIBUTE_NORMAL | FILE_FLAG_WRITE_THROUGH,
- NULL);
- /* Set this file as system log file */
- if (!_st_sys_loghandle)
- _st_sys_loghandle = h;
-
- return (HANDLE)h;
-}
-
-void apxLogLevelSet(HANDLE hFile, DWORD dwLevel)
-{
- apx_logfile_st *lf = (apx_logfile_st *)hFile;
- if (IS_INVALID_HANDLE(lf))
- lf = _st_sys_loghandle;
- if (IS_INVALID_HANDLE(lf))
- return;
- if (dwLevel < 4)
- lf->dwLogLevel = dwLevel;
-}
-
-void apxLogRotateSet(HANDLE hFile, BOOL doRotate)
-{
- apx_logfile_st *lf = (apx_logfile_st *)hFile;
- if (IS_INVALID_HANDLE(lf))
- lf = _st_sys_loghandle;
- if (IS_INVALID_HANDLE(lf))
- return;
- lf->bRotate = doRotate;
-}
-
-void apxLogLevelSetW(HANDLE hFile,
- LPCWSTR szLevel)
-{
- apx_logfile_st *lf = (apx_logfile_st *)hFile;
-
- if (IS_INVALID_HANDLE(lf))
- lf = _st_sys_loghandle;
- if (IS_INVALID_HANDLE(lf))
- return;
- if (szLevel) {
- if (!lstrcmpiW(szLevel, L"error"))
- lf->dwLogLevel = APXLOG_LEVEL_ERROR;
- else if (!lstrcmpiW(szLevel, L"warn"))
- lf->dwLogLevel = APXLOG_LEVEL_WARN;
- else if (!lstrcmpiW(szLevel, L"info"))
- lf->dwLogLevel = APXLOG_LEVEL_INFO;
- else if (!lstrcmpiW(szLevel, L"debug"))
- lf->dwLogLevel = APXLOG_LEVEL_DEBUG;
- }
-}
-
-static BOOL apx_log_rotate(apx_logfile_st *l,
- LPSYSTEMTIME lpCtime)
-{
- WCHAR sPath[MAX_PATH+1];
-
- /* rotate on daily basis */
- if (l->sysTime.wDay == lpCtime->wDay)
- return TRUE;
- FlushFileBuffers(l->hFile);
- CloseHandle(l->hFile);
- l->sysTime = *lpCtime;
-
- wsprintfW(sPath, L"%s\\%s%04d%02d%02d.log",
- l->szPath,
- l->szPrefix,
- l->sysTime.wYear,
- l->sysTime.wMonth,
- l->sysTime.wDay);
- l->hFile = CreateFileW(sPath,
- GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE,
- NULL,
- OPEN_ALWAYS,
- FILE_ATTRIBUTE_NORMAL | FILE_FLAG_WRITE_THROUGH,
- NULL);
- if (IS_INVALID_HANDLE(l->hFile))
- return FALSE;
- else
- return TRUE;
-}
-
-int
-apxLogWrite(
- HANDLE hFile,
- DWORD dwLevel,
- BOOL bTimeStamp,
- LPCSTR szFile,
- DWORD dwLine,
- LPCSTR szFormat,
- ...)
-{
- va_list args;
- CHAR buffer[1024+32];
- LPSTR szBp;
- int len = 0;
- LPCSTR f = szFile;
- CHAR sb[MAX_PATH+1];
- DWORD wr;
- DWORD err;
- BOOL dolock = TRUE;
- apx_logfile_st *lf = (apx_logfile_st *)hFile;
-
- if (IS_INVALID_HANDLE(lf))
- lf = _st_sys_loghandle;
- if (IS_INVALID_HANDLE(lf)) {
- lf = &_st_sys_errhandle;
- lf->hFile = GetStdHandle(STD_ERROR_HANDLE);
- dolock = FALSE;
- }
- if (dwLevel < lf->dwLogLevel)
- return 0;
- err = GetLastError(); /* save the last Error code */
- if (f) {
- f = (szFile + lstrlenA(szFile) - 1);
- while(f != szFile && '\\' != *f && '/' != *f)
- f--;
- if(f != szFile)
- f++;
- }
- lstrcpyA(buffer, _log_level[dwLevel]);
- if (!dolock)
- lstrcatA(buffer, "\n");
- szBp = &buffer[lstrlenA(buffer)];
- if (!szFormat) {
- FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM |
- FORMAT_MESSAGE_IGNORE_INSERTS,
- NULL,
- err,
- MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
- szBp,
- 1000,
- NULL);
- }
- else {
- va_start(args, szFormat);
- wvsprintfA(szBp, szFormat, args);
- va_end(args);
- }
- len = lstrlenA(buffer);
- if (len > 0) {
- /* Remove trailing line separator */
- if (buffer[len - 1] == '\n') {
- buffer[len - 1] = '\0';
- --len;
- }
- if (!IS_INVALID_HANDLE(lf->hFile)) {
- SYSTEMTIME t;
- GetLocalTime(&t);
- if (lf->bRotate) {
- if (!apx_log_rotate(lf, &t))
- return 0;
- }
- if (dolock) {
- APX_LOGLOCK(lf->hFile);
- }
- if (bTimeStamp) {
- wsprintfA(sb, "[%d-%02d-%02d %02d:%02d:%02d] ",
- t.wYear, t.wMonth, t.wDay,
- t.wHour, t.wMinute, t.wSecond);
- WriteFile(lf->hFile, sb, lstrlenA(sb), &wr, NULL);
- }
- if (f) {
- wsprintfA(sb, "[%-4d %s] ", dwLine, f);
- WriteFile(lf->hFile, sb, lstrlenA(sb), &wr, NULL);
- }
-
- WriteFile(lf->hFile, buffer, len, &wr, NULL);
- /* Terminate the line */
- WriteFile(lf->hFile, LINE_SEP, sizeof(LINE_SEP) - 1, &wr, NULL);
-#ifdef _DEBUG_FULL
- FlushFileBuffers(lf->hFile);
-#endif
- if (dolock) {
- APX_LOGUNLOCK(lf->hFile);
- }
- }
-#ifdef _DEBUG_FULL
- {
- char tid[1024 + 16];
- wsprintfA(tid, "[%04X] %s", GetCurrentThreadId(), buffer);
- OutputDebugStringA(tid);
- }
-#endif
- }
- /* Restore the last Error code */
- SetLastError(err);
- return len;
-}
-
-void apxLogClose(
- HANDLE hFile)
-{
- apx_logfile_st *lf = (apx_logfile_st *)hFile;
-
- if (IS_INVALID_HANDLE(lf))
- lf = _st_sys_loghandle;
- if (IS_INVALID_HANDLE(lf))
- return;
-
- FlushFileBuffers(lf->hFile);
- CloseHandle(lf->hFile);
- if (lf == _st_sys_loghandle)
- _st_sys_loghandle = NULL;
- apxFree(lf);
-}
-
-int
-apxDisplayError(
- BOOL bDisplay,
- LPCSTR szFile,
- DWORD dwLine,
- LPCSTR szFormat,
- ...)
-{
- va_list args;
- CHAR buffer[1024+16];
- CHAR sysbuf[2048];
- int len = 0, nRet;
- LPCSTR f = szFile;
- DWORD err = GetLastError(); /* save the last Error code */
- if (f) {
- f = (szFile + lstrlenA(szFile) - 1);
- while(f != szFile && '\\' != *f && '/' != *f)
- f--;
- if(f != szFile)
- f++;
- }
- sysbuf[0] = '\0';
- if (err != ERROR_SUCCESS) {
- len = FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM |
- FORMAT_MESSAGE_IGNORE_INSERTS,
- NULL,
- err,
- MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
- sysbuf,
- 1000,
- NULL);
- sysbuf[len] = 0;
- }
- if (szFormat) {
- va_start(args, szFormat);
- wvsprintfA(buffer, szFormat, args);
- va_end(args);
- if (f) {
- CHAR sb[MAX_PATH+1];
- wsprintfA(sb, "\n%s (%d)", f, dwLine);
- lstrcatA(sysbuf, sb);
- }
- lstrcatA(sysbuf, "\n");
- lstrcatA(sysbuf, buffer);
- }
- len = lstrlenA(sysbuf);
-#ifdef _DEBUG_FULL
- OutputDebugStringA(sysbuf);
-#endif
- if (len > 0 && bDisplay) {
- nRet = MessageBoxA(NULL, sysbuf,
- "Application System Error",
- MB_ICONERROR | MB_OK | MB_SYSTEMMODAL);
- }
- /* Restore the last Error code */
- SetLastError(err);
- return len;
-}
Copied: tags/jbosssvc/2.0.8/procrun/src/log.c (from rev 2689, trunk/utils/windows/native/service/procrun/src/log.c)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/src/log.c (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/src/log.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,419 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "apxwin.h"
+#include "private.h"
+#include <stdio.h>
+
+#define LINE_SEP "\n"
+#define LOGF_EXT L".%04d-%02d-%02d.log"
+
+static LPCSTR _log_level[] = {
+ "[debug] ",
+ "[info] ",
+ "[warn] ",
+ "[error] ",
+ NULL
+};
+
+typedef struct apx_logfile_st {
+ HANDLE hFile;
+ DWORD dwLogLevel;
+ BOOL bRotate;
+ SYSTEMTIME sysTime;
+ WCHAR szPath[SIZ_PATHLEN];
+ WCHAR szPrefix[MAX_PATH];
+} apx_logfile_st;
+
+/* Per-application master log file */
+static apx_logfile_st *_st_sys_loghandle = NULL;
+
+static apx_logfile_st _st_sys_errhandle = { NULL, APXLOG_LEVEL_WARN, FALSE};
+
+
+LPWSTR apxLogFile(
+ APXHANDLE hPool,
+ LPCWSTR szPath,
+ LPCWSTR szPrefix,
+ LPCWSTR szName,
+ BOOL bTimeStamp)
+{
+ LPWSTR sRet;
+ WCHAR sPath[SIZ_PATHLEN];
+ WCHAR sName[SIZ_PATHLEN];
+ SYSTEMTIME sysTime;
+
+ GetLocalTime(&sysTime);
+ if (!szPath) {
+ if (GetSystemDirectoryW(sPath, MAX_PATH) == 0)
+ return INVALID_HANDLE_VALUE;
+ lstrlcatW(sPath, MAX_PATH, L"\\LogFiles\\Apache");
+ }
+ else {
+ lstrlcpyW(sPath, MAX_PATH, szPath);
+ }
+ if (!szPrefix)
+ szPrefix = L"";
+ if (!szName)
+ szName = L"";
+ if (bTimeStamp)
+ wsprintfW(sName,
+ L"\\%s%s" LOGF_EXT,
+ szPrefix,
+ szName,
+ sysTime.wYear,
+ sysTime.wMonth,
+ sysTime.wDay);
+ else
+ wsprintfW(sName,
+ L"\\%s%s",
+ szPrefix,
+ szName);
+
+ sRet = apxPoolAlloc(hPool, (SIZ_PATHLEN) * sizeof(WCHAR));
+ /* Set default level to info */
+ CreateDirectoryW(sPath, NULL);
+
+ lstrlcpyW(sRet, SIZ_PATHMAX, sPath);
+ lstrlcatW(sRet, SIZ_PATHMAX, sName);
+
+ return sRet;
+}
+
+/* Open the log file
+ * TODO: format like standard apache error.log
+ * Add the EventLogger
+ */
+HANDLE apxLogOpen(
+ APXHANDLE hPool,
+ LPCWSTR szPath,
+ LPCWSTR szPrefix)
+{
+
+ WCHAR sPath[SIZ_PATHLEN];
+ WCHAR sName[SIZ_PATHLEN];
+ SYSTEMTIME sysTime;
+ apx_logfile_st *h;
+
+ GetLocalTime(&sysTime);
+ if (!szPath) {
+ if (GetSystemDirectoryW(sPath, MAX_PATH) == 0)
+ return INVALID_HANDLE_VALUE;
+ lstrlcatW(sPath, MAX_PATH, L"\\LogFiles\\Apache");
+ }
+ else {
+ lstrlcpyW(sPath, MAX_PATH, szPath);
+ }
+ if (!szPrefix)
+ szPrefix = L"commons-daemon";
+ wsprintfW(sName, L"\\%s" LOGF_EXT,
+ szPrefix,
+ sysTime.wYear,
+ sysTime.wMonth,
+ sysTime.wDay);
+ if (!(h = (apx_logfile_st *)apxPoolCalloc(hPool, sizeof(apx_logfile_st))))
+ return NULL;
+ /* Set default level to info */
+ h->dwLogLevel = APXLOG_LEVEL_INFO;
+ CreateDirectoryW(sPath, NULL);
+
+ h->sysTime = sysTime;
+ lstrlcpyW(h->szPath, MAX_PATH, sPath);
+ lstrlcatW(sPath, MAX_PATH, sName);
+ lstrlcpyW(h->szPrefix, MAX_PATH, szPrefix);
+
+ h->hFile = CreateFileW(sPath,
+ GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE,
+ NULL,
+ OPEN_ALWAYS,
+ FILE_ATTRIBUTE_NORMAL | FILE_FLAG_WRITE_THROUGH | FILE_FLAG_SEQUENTIAL_SCAN,
+ NULL);
+ /* Set this file as system log file */
+ if (!_st_sys_loghandle)
+ _st_sys_loghandle = h;
+
+ return (HANDLE)h;
+}
+
+void apxLogLevelSet(HANDLE hFile, DWORD dwLevel)
+{
+ apx_logfile_st *lf = (apx_logfile_st *)hFile;
+ if (IS_INVALID_HANDLE(lf))
+ lf = _st_sys_loghandle;
+ if (IS_INVALID_HANDLE(lf))
+ return;
+ if (dwLevel < 4)
+ lf->dwLogLevel = dwLevel;
+}
+
+void apxLogRotateSet(HANDLE hFile, BOOL doRotate)
+{
+ apx_logfile_st *lf = (apx_logfile_st *)hFile;
+ if (IS_INVALID_HANDLE(lf))
+ lf = _st_sys_loghandle;
+ if (IS_INVALID_HANDLE(lf))
+ return;
+ lf->bRotate = doRotate;
+}
+
+void apxLogLevelSetW(HANDLE hFile,
+ LPCWSTR szLevel)
+{
+ apx_logfile_st *lf = (apx_logfile_st *)hFile;
+
+ if (IS_INVALID_HANDLE(lf))
+ lf = _st_sys_loghandle;
+ if (IS_INVALID_HANDLE(lf))
+ return;
+ if (szLevel) {
+ if (!lstrcmpiW(szLevel, L"error"))
+ lf->dwLogLevel = APXLOG_LEVEL_ERROR;
+ else if (!lstrcmpiW(szLevel, L"warn"))
+ lf->dwLogLevel = APXLOG_LEVEL_WARN;
+ else if (!lstrcmpiW(szLevel, L"info"))
+ lf->dwLogLevel = APXLOG_LEVEL_INFO;
+ else if (!lstrcmpiW(szLevel, L"debug"))
+ lf->dwLogLevel = APXLOG_LEVEL_DEBUG;
+ }
+}
+
+static BOOL apx_log_rotate(apx_logfile_st *l,
+ LPSYSTEMTIME lpCtime)
+{
+ WCHAR sPath[SIZ_PATHLEN];
+
+ /* rotate on daily basis */
+ if (l->sysTime.wDay == lpCtime->wDay)
+ return TRUE;
+ FlushFileBuffers(l->hFile);
+ CloseHandle(l->hFile);
+ l->sysTime = *lpCtime;
+
+ wsprintfW(sPath, L"%s\\%s" LOGF_EXT,
+ l->szPath,
+ l->szPrefix,
+ l->sysTime.wYear,
+ l->sysTime.wMonth,
+ l->sysTime.wDay);
+ l->hFile = CreateFileW(sPath,
+ GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE,
+ NULL,
+ OPEN_ALWAYS,
+ FILE_FLAG_NO_BUFFERING | FILE_FLAG_SEQUENTIAL_SCAN,
+ NULL);
+ if (IS_INVALID_HANDLE(l->hFile))
+ return FALSE;
+ else
+ return TRUE;
+}
+
+int
+apxLogWrite(
+ HANDLE hFile,
+ DWORD dwLevel,
+ BOOL bTimeStamp,
+ LPCSTR szFile,
+ DWORD dwLine,
+ LPCSTR szFormat,
+ ...)
+{
+ va_list args;
+ CHAR buffer[1024+32];
+ LPSTR szBp;
+ int len = 0;
+ LPCSTR f = szFile;
+ CHAR sb[SIZ_PATHLEN];
+ DWORD wr;
+ DWORD err;
+ BOOL dolock = TRUE;
+ apx_logfile_st *lf = (apx_logfile_st *)hFile;
+
+ err = GetLastError(); /* save the last Error code */
+ if (IS_INVALID_HANDLE(lf))
+ lf = _st_sys_loghandle;
+ if (IS_INVALID_HANDLE(lf)) {
+ lf = &_st_sys_errhandle;
+ lf->hFile = GetStdHandle(STD_ERROR_HANDLE);
+ dolock = FALSE;
+ }
+ if (dwLevel < lf->dwLogLevel)
+ return 0;
+ if (f) {
+ f = (szFile + lstrlenA(szFile) - 1);
+ while(f != szFile && '\\' != *f && '/' != *f)
+ f--;
+ if(f != szFile)
+ f++;
+ }
+ szBp = &buffer[0];
+ if (!szFormat) {
+ FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM |
+ FORMAT_MESSAGE_IGNORE_INSERTS,
+ NULL,
+ err,
+ MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
+ szBp,
+ 1000,
+ NULL);
+ }
+ else {
+ va_start(args, szFormat);
+ wvsprintfA(szBp, szFormat, args);
+ va_end(args);
+ }
+ len = lstrlenA(buffer);
+ if (len > 0) {
+ /* Remove trailing line separator */
+ if (buffer[len - 1] == '\n') {
+ buffer[len - 1] = '\0';
+ --len;
+ }
+ if (!IS_INVALID_HANDLE(lf->hFile)) {
+ SYSTEMTIME t;
+ GetLocalTime(&t);
+ if (lf->bRotate) {
+ if (!apx_log_rotate(lf, &t))
+ return 0;
+ }
+ if (dolock) {
+ APX_LOGLOCK(lf->hFile);
+ }
+ if (bTimeStamp) {
+ wsprintfA(sb, "[%d-%02d-%02d %02d:%02d:%02d] ",
+ t.wYear, t.wMonth, t.wDay,
+ t.wHour, t.wMinute, t.wSecond);
+ WriteFile(lf->hFile, sb, lstrlenA(sb), &wr, NULL);
+ }
+ WriteFile(lf->hFile, _log_level[dwLevel],
+ strlen(_log_level[dwLevel]), &wr, NULL);
+ if (f && lf->dwLogLevel == APXLOG_LEVEL_DEBUG) {
+ wsprintfA(sb, "(%10s:%-4d) ", f, dwLine);
+ WriteFile(lf->hFile, sb, lstrlenA(sb), &wr, NULL);
+ }
+ WriteFile(lf->hFile, buffer, len, &wr, NULL);
+
+ /* Terminate the line */
+ WriteFile(lf->hFile, LINE_SEP, sizeof(LINE_SEP) - 1, &wr, NULL);
+#ifdef _DEBUG_FULL
+ FlushFileBuffers(lf->hFile);
+#endif
+ if (dolock) {
+ APX_LOGUNLOCK(lf->hFile);
+ }
+ }
+#ifdef _DEBUG_FULL
+ {
+ char tid[1024 + 16];
+ wsprintfA(tid, "[%04d] %s", GetCurrentThreadId(), buffer);
+ OutputDebugStringA(tid);
+ }
+#endif
+ }
+ /* Restore the last Error code */
+ SetLastError(err);
+ return len;
+}
+
+void apxLogClose(
+ HANDLE hFile)
+{
+ apx_logfile_st *lf = (apx_logfile_st *)hFile;
+
+ if (IS_INVALID_HANDLE(lf))
+ lf = _st_sys_loghandle;
+ if (IS_INVALID_HANDLE(lf))
+ return;
+
+ FlushFileBuffers(lf->hFile);
+ CloseHandle(lf->hFile);
+ if (lf == _st_sys_loghandle)
+ _st_sys_loghandle = NULL;
+ apxFree(lf);
+}
+
+int
+apxDisplayError(
+ BOOL bDisplay,
+ LPCSTR szFile,
+ DWORD dwLine,
+ LPCSTR szFormat,
+ ...)
+{
+ va_list args;
+ CHAR buffer[SIZ_HUGLEN];
+ CHAR sysbuf[SIZ_HUGLEN];
+ int len = 0, nRet;
+ LPCSTR f = szFile;
+ DWORD err = GetLastError(); /* save the last Error code */
+ if (f) {
+ f = (szFile + lstrlenA(szFile) - 1);
+ while(f != szFile && '\\' != *f && '/' != *f)
+ f--;
+ if(f != szFile)
+ f++;
+ }
+ sysbuf[0] = '\0';
+ if (err != ERROR_SUCCESS) {
+ len = FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM |
+ FORMAT_MESSAGE_IGNORE_INSERTS,
+ NULL,
+ err,
+ MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
+ sysbuf,
+ SIZ_DESLEN,
+ NULL);
+ if (len > 0) {
+ sysbuf[len] = '\0';
+ if (sysbuf[len - 1] == '\n') {
+ sysbuf[len - 1] = '\0';
+ --len;
+ }
+ }
+ else
+ sysbuf[0] = '\0';
+ }
+ if (szFormat) {
+ va_start(args, szFormat);
+ wvsprintfA(buffer, szFormat, args);
+ va_end(args);
+ if (f) {
+ CHAR sb[SIZ_PATHLEN];
+ wsprintfA(sb, "%s (%d)", f, dwLine);
+ lstrcatA(sysbuf, sb);
+ }
+ lstrlcatA(sysbuf, SIZ_HUGLEN, "\n");
+ lstrlcatA(sysbuf, SIZ_HUGLEN, buffer);
+ }
+ len = lstrlenA(sysbuf);
+#ifdef _DEBUG_FULL
+ OutputDebugStringA(sysbuf);
+#endif
+ if (len > 0) {
+ if (bDisplay) {
+ nRet = MessageBoxA(NULL, sysbuf,
+ "Application System Error",
+ MB_ICONERROR | MB_OK | MB_SYSTEMMODAL);
+ }
+ else {
+ fputs(sysbuf, stderr);
+ fputc('\n', stderr);
+ fflush(stderr);
+ }
+ }
+ /* Restore the last Error code */
+ SetLastError(err);
+ return len;
+}
Deleted: tags/jbosssvc/2.0.8/procrun/src/mclib.c
===================================================================
--- trunk/utils/windows/native/service/procrun/src/mclib.c 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/src/mclib.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,294 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-/*
- * Copyright (c) 1994
- * The Regents of the University of California. All rights reserved.
- *
- * This code is derived from software contributed to Berkeley by
- * Andrew Hume of AT&T Bell Laboratories.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- * 1. Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
- * 3. Neither the name of the University nor the names of its contributors
- * may be used to endorse or promote products derived from this software
- * without specific prior written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
- * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
- * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
- * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
- * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- * SUCH DAMAGE.
- */
-
-#include "apxwin.h"
-#include "private.h"
-
-typedef UINT word; /* "word" used for optimal copy speed */
-#define wsize sizeof(UINT)
-#define wmask (wsize - 1)
-
-LPVOID AplFillMemory(PVOID Destination, SIZE_T Length, BYTE Fill)
-{
-
- SIZE_T t;
-#ifdef WIN64
- UINT64 c;
-#else
- UINT c;
-#endif
- LPBYTE dst;
-
- dst = Destination;
- /*
- * If not enough words, just fill bytes. A length >= 2 words
- * guarantees that at least one of them is `complete' after
- * any necessary alignment. For instance:
- *
- * |-----------|-----------|-----------|
- * |00|01|02|03|04|05|06|07|08|09|0A|00|
- * ^---------------------^
- * dst dst+length-1
- *
- * but we use a minimum of 3 here since the overhead of the code
- * to do word writes is substantial.
- */
- if (Length < 3 * wsize) {
- while (Length != 0) {
- *dst++ = Fill;
- --Length;
- }
- return (Destination);
- }
-
- if ((c = Fill) != 0) { /* Fill the word. */
- c = (c << 8) | c; /* u_int is 16 bits. */
- c = (c << 16) | c; /* u_int is 32 bits. */
-#ifdef WIN64
- c = (c << 32) | c; /* u_int is 64 bits. */
-#endif
- }
- /* Align destination by filling in bytes. */
- if ((t = (SIZE_T)dst & wmask) != 0) {
- t = wsize - t;
- Length -= t;
- do {
- *dst++ = Fill;
- } while (--t != 0);
- }
-
- /* Fill words. Length was >= 2*words so we know t >= 1 here. */
- t = Length / wsize;
- do {
-#ifdef WIN64
- *(UINT64 *)dst = c;
-#else
- *(UINT *)dst = c;
-#endif
- dst += wsize;
- } while (--t != 0);
-
- /* Mop up trailing bytes, if any. */
- t = Length & wmask;
- if (t != 0)
- do {
- *dst++ = Fill;
- } while (--t != 0);
- return (Destination);
-}
-
-void AplZeroMemory(PVOID Destination, SIZE_T Length)
-{
-
- SIZE_T t;
- LPBYTE dst;
-
- dst = Destination;
- /*
- * If not enough words, just fill bytes. A length >= 2 words
- * guarantees that at least one of them is `complete' after
- * any necessary alignment. For instance:
- *
- * |-----------|-----------|-----------|
- * |00|01|02|03|04|05|06|07|08|09|0A|00|
- * ^---------------------^
- * dst dst+length-1
- *
- * but we use a minimum of 3 here since the overhead of the code
- * to do word writes is substantial.
- */
- if (Length < 3 * wsize) {
- while (Length != 0) {
- *dst++ = 0;
- --Length;
- }
- return;
- }
-
- /* Align destination by filling in bytes. */
- if ((t = (SIZE_T)dst & wmask) != 0) {
- t = wsize - t;
- Length -= t;
- do {
- *dst++ = 0;
- } while (--t != 0);
- }
-
- /* Fill words. Length was >= 2*words so we know t >= 1 here. */
- t = Length / wsize;
- do {
- *(UINT *)dst = 0;
- dst += wsize;
- } while (--t != 0);
-
- /* Mop up trailing bytes, if any. */
- t = Length & wmask;
- if (t != 0)
- do {
- *dst++ = 0;
- } while (--t != 0);
-}
-
-LPVOID AplCopyMemory(PVOID Destination, const VOID* Source, SIZE_T Length)
-{
- char *dst = Destination;
- const char *src = Source;
- SIZE_T t;
-
- if (Length == 0 || dst == src) /* nothing to do */
- goto done;
-
- /*
- * Macros: loop-t-times; and loop-t-times, t>0
- */
-#define TLOOP(s) if (t) TLOOP1(s)
-#define TLOOP1(s) do { s; } while (--t)
-
- if ((SIZE_T)dst < (SIZE_T)src) {
- /*
- * Copy forward.
- */
- t = (int)(SIZE_T)src; /* only need low bits */
- if ((t | (int)(SIZE_T)dst) & wmask) {
- /*
- * Try to align operands. This cannot be done
- * unless the low bits match.
- */
- if ((t ^ (int)(SIZE_T)dst) & wmask || Length < wsize)
- t = Length;
- else
- t = wsize - (t & wmask);
- Length -= t;
- TLOOP1(*dst++ = *src++);
- }
- /*
- * Copy whole words, then mop up any trailing bytes.
- */
- t = Length / wsize;
- TLOOP(*(word *)dst = *(word *)src; src += wsize; dst += wsize);
- t = Length & wmask;
- TLOOP(*dst++ = *src++);
- } else {
- /*
- * Copy backwards. Otherwise essentially the same.
- * Alignment works as before, except that it takes
- * (t&wmask) bytes to align, not wsize-(t&wmask).
- */
- src += Length;
- dst += Length;
- t = (int)(SIZE_T)src;
- if ((t | (int)(SIZE_T)dst) & wmask) {
- if ((t ^ (int)(SIZE_T)dst) & wmask || Length <= wsize)
- t = Length;
- else
- t &= wmask;
- Length -= t;
- TLOOP1(*--dst = *--src);
- }
- t = Length / wsize;
- TLOOP(src -= wsize; dst -= wsize; *(word *)dst = *(word *)src);
- t = Length & wmask;
- TLOOP(*--dst = *--src);
- }
-done:
- return (Destination);
-}
-
-
-INT
-AplMemCmp(LPCVOID lpA, LPCVOID lpB, SIZE_T nBytes)
-{
- if (nBytes != 0) {
- const BYTE *p1 = lpA, *p2 = lpB;
-
- do {
- if (*p1++ != *p2++)
- return (*--p1 - *--p2);
- } while (--nBytes != 0);
- }
- return 0;
-}
-
-/*
- * Find the first occurrence of lpFind in lpMem.
- * dwLen: The length of lpFind
- * dwSize: The length of lpMem
- */
-LPBYTE
-ApcMemSearch(LPCVOID lpMem, LPCVOID lpFind, SIZE_T dwLen, SIZE_T dwSize)
-{
- BYTE c, sc;
- SIZE_T cnt = 0;
- const BYTE *s = lpMem, *find = lpFind;
-
- if ((c = *find++) != 0) {
- do {
- do {
- sc = *s++;
- if (cnt++ > dwSize)
- return NULL;
- } while (sc != c);
- } while (AplMemCmp(s, find, dwLen - 1) != 0);
- s--;
- }
- return (LPBYTE)s;
-}
-
-LPSTR
-AplRindexA(LPCSTR lpStr, int ch)
-{
- LPSTR save;
-
- for (save = NULL;; ++lpStr) {
- if (*lpStr == ch)
- save = (LPSTR)lpStr;
- if (!*lpStr)
- return save;
- }
- /* NOTREACHED */
-}
Copied: tags/jbosssvc/2.0.8/procrun/src/mclib.c (from rev 2689, trunk/utils/windows/native/service/procrun/src/mclib.c)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/src/mclib.c (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/src/mclib.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,409 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/*
+ * Copyright (c) 1994
+ * The Regents of the University of California. All rights reserved.
+ *
+ * This code is derived from software contributed to Berkeley by
+ * Andrew Hume of AT&T Bell Laboratories.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ * 3. Neither the name of the University nor the names of its contributors
+ * may be used to endorse or promote products derived from this software
+ * without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ */
+
+#include "apxwin.h"
+#include "private.h"
+
+typedef UINT word; /* "word" used for optimal copy speed */
+#define wsize sizeof(UINT)
+#define wmask (wsize - 1)
+
+LPVOID AplFillMemory(PVOID Destination, SIZE_T Length, BYTE Fill)
+{
+
+ SIZE_T t;
+#ifdef WIN64
+ UINT64 c;
+#else
+ UINT c;
+#endif
+ LPBYTE dst;
+
+ dst = Destination;
+ /*
+ * If not enough words, just fill bytes. A length >= 2 words
+ * guarantees that at least one of them is `complete' after
+ * any necessary alignment. For instance:
+ *
+ * |-----------|-----------|-----------|
+ * |00|01|02|03|04|05|06|07|08|09|0A|00|
+ * ^---------------------^
+ * dst dst+length-1
+ *
+ * but we use a minimum of 3 here since the overhead of the code
+ * to do word writes is substantial.
+ */
+ if (Length < 3 * wsize) {
+ while (Length != 0) {
+ *dst++ = Fill;
+ --Length;
+ }
+ return (Destination);
+ }
+
+ if ((c = Fill) != 0) { /* Fill the word. */
+ c = (c << 8) | c; /* u_int is 16 bits. */
+ c = (c << 16) | c; /* u_int is 32 bits. */
+#ifdef WIN64
+ c = (c << 32) | c; /* u_int is 64 bits. */
+#endif
+ }
+ /* Align destination by filling in bytes. */
+ if ((t = (SIZE_T)dst & wmask) != 0) {
+ t = wsize - t;
+ Length -= t;
+ do {
+ *dst++ = Fill;
+ } while (--t != 0);
+ }
+
+ /* Fill words. Length was >= 2*words so we know t >= 1 here. */
+ t = Length / wsize;
+ do {
+#ifdef WIN64
+ *(UINT64 *)dst = c;
+#else
+ *(UINT *)dst = c;
+#endif
+ dst += wsize;
+ } while (--t != 0);
+
+ /* Mop up trailing bytes, if any. */
+ t = Length & wmask;
+ if (t != 0)
+ do {
+ *dst++ = Fill;
+ } while (--t != 0);
+ return (Destination);
+}
+
+void AplZeroMemory(PVOID Destination, SIZE_T Length)
+{
+
+ SIZE_T t;
+ LPBYTE dst;
+
+ dst = Destination;
+ /*
+ * If not enough words, just fill bytes. A length >= 2 words
+ * guarantees that at least one of them is `complete' after
+ * any necessary alignment. For instance:
+ *
+ * |-----------|-----------|-----------|
+ * |00|01|02|03|04|05|06|07|08|09|0A|00|
+ * ^---------------------^
+ * dst dst+length-1
+ *
+ * but we use a minimum of 3 here since the overhead of the code
+ * to do word writes is substantial.
+ */
+ if (Length < 3 * wsize) {
+ while (Length != 0) {
+ *dst++ = 0;
+ --Length;
+ }
+ return;
+ }
+
+ /* Align destination by filling in bytes. */
+ if ((t = (SIZE_T)dst & wmask) != 0) {
+ t = wsize - t;
+ Length -= t;
+ do {
+ *dst++ = 0;
+ } while (--t != 0);
+ }
+
+ /* Fill words. Length was >= 2*words so we know t >= 1 here. */
+ t = Length / wsize;
+ do {
+ *(UINT *)dst = 0;
+ dst += wsize;
+ } while (--t != 0);
+
+ /* Mop up trailing bytes, if any. */
+ t = Length & wmask;
+ if (t != 0)
+ do {
+ *dst++ = 0;
+ } while (--t != 0);
+}
+
+LPVOID AplCopyMemory(PVOID Destination, const VOID* Source, SIZE_T Length)
+{
+ char *dst = Destination;
+ const char *src = Source;
+ SIZE_T t;
+
+ if (Length == 0 || dst == src) /* nothing to do */
+ goto done;
+
+ /*
+ * Macros: loop-t-times; and loop-t-times, t>0
+ */
+#define TLOOP(s) if (t) TLOOP1(s)
+#define TLOOP1(s) do { s; } while (--t)
+
+ if ((SIZE_T)dst < (SIZE_T)src) {
+ /*
+ * Copy forward.
+ */
+ t = (int)(SIZE_T)src; /* only need low bits */
+ if ((t | (int)(SIZE_T)dst) & wmask) {
+ /*
+ * Try to align operands. This cannot be done
+ * unless the low bits match.
+ */
+ if ((t ^ (int)(SIZE_T)dst) & wmask || Length < wsize)
+ t = Length;
+ else
+ t = wsize - (t & wmask);
+ Length -= t;
+ TLOOP1(*dst++ = *src++);
+ }
+ /*
+ * Copy whole words, then mop up any trailing bytes.
+ */
+ t = Length / wsize;
+ TLOOP(*(word *)dst = *(word *)src; src += wsize; dst += wsize);
+ t = Length & wmask;
+ TLOOP(*dst++ = *src++);
+ } else {
+ /*
+ * Copy backwards. Otherwise essentially the same.
+ * Alignment works as before, except that it takes
+ * (t&wmask) bytes to align, not wsize-(t&wmask).
+ */
+ src += Length;
+ dst += Length;
+ t = (int)(SIZE_T)src;
+ if ((t | (int)(SIZE_T)dst) & wmask) {
+ if ((t ^ (int)(SIZE_T)dst) & wmask || Length <= wsize)
+ t = Length;
+ else
+ t &= wmask;
+ Length -= t;
+ TLOOP1(*--dst = *--src);
+ }
+ t = Length / wsize;
+ TLOOP(src -= wsize; dst -= wsize; *(word *)dst = *(word *)src);
+ t = Length & wmask;
+ TLOOP(*--dst = *--src);
+ }
+done:
+ return (Destination);
+}
+
+
+INT
+AplMemCmp(LPCVOID lpA, LPCVOID lpB, SIZE_T nBytes)
+{
+ if (nBytes != 0) {
+ const BYTE *p1 = lpA, *p2 = lpB;
+
+ do {
+ if (*p1++ != *p2++)
+ return (*--p1 - *--p2);
+ } while (--nBytes != 0);
+ }
+ return 0;
+}
+
+/*
+ * Find the first occurrence of lpFind in lpMem.
+ * dwLen: The length of lpFind
+ * dwSize: The length of lpMem
+ */
+LPBYTE
+ApcMemSearch(LPCVOID lpMem, LPCVOID lpFind, SIZE_T dwLen, SIZE_T dwSize)
+{
+ BYTE c, sc;
+ SIZE_T cnt = 0;
+ const BYTE *s = lpMem, *find = lpFind;
+
+ if ((c = *find++) != 0) {
+ do {
+ do {
+ sc = *s++;
+ if (cnt++ > dwSize)
+ return NULL;
+ } while (sc != c);
+ } while (AplMemCmp(s, find, dwLen - 1) != 0);
+ s--;
+ }
+ return (LPBYTE)s;
+}
+
+LPSTR
+AplRindexA(LPCSTR lpStr, int ch)
+{
+ LPSTR save;
+
+ for (save = NULL;; ++lpStr) {
+ if (*lpStr == ch)
+ save = (LPSTR)lpStr;
+ if (!*lpStr)
+ return save;
+ }
+ /* NOTREACHED */
+}
+
+/*
+ * Appends src to string dst of size siz (unlike strncat, siz is the
+ * full size of dst, not space left). At most siz-1 characters
+ * will be copied. Always NUL terminates (unless siz <= strlen(dst)).
+ * Returns strlen(src) + MIN(siz, strlen(initial dst)).
+ * If retval >= siz, truncation occurred.
+ */
+LPSTR
+lstrlcatA(LPSTR dst, int siz, LPCSTR src)
+{
+ LPSTR d = dst;
+ LPCSTR s = src;
+ int n = siz;
+ int dlen;
+
+ /* Find the end of dst and adjust bytes left but don't go past end */
+ while (n-- != 0 && *d != '\0')
+ d++;
+ dlen = d - dst;
+ n = siz - dlen;
+
+ if (n == 0)
+ return NULL;
+ while (*s != '\0') {
+ if (n != 1) {
+ *d++ = *s;
+ n--;
+ }
+ s++;
+ }
+ *d = '\0';
+
+ return dst;
+}
+
+LPWSTR
+lstrlcatW(LPWSTR dst, int siz, LPCWSTR src)
+{
+ LPWSTR d = dst;
+ LPCWSTR s = src;
+ int n = siz;
+ int dlen;
+
+ /* Find the end of dst and adjust bytes left but don't go past end */
+ while (n-- != 0 && *d != '\0')
+ d++;
+ dlen = d - dst;
+ n = siz - dlen;
+
+ if (n == 0)
+ return NULL;
+ while (*s != L'\0') {
+ if (n != 1) {
+ *d++ = *s;
+ n--;
+ }
+ s++;
+ }
+ *d = L'\0';
+
+ return dst;
+}
+
+LPSTR
+lstrlcpyA(LPSTR dst, int siz, LPCSTR src)
+{
+ LPSTR d = dst;
+ LPCSTR s = src;
+ int n = siz;
+
+ /* Copy as many bytes as will fit */
+ if (n != 0) {
+ while (--n != 0) {
+ if ((*d++ = *s++) == '\0')
+ break;
+ }
+ }
+
+ /* Not enough room in dst, add NUL and traverse rest of src */
+ if (n == 0) {
+ if (siz != 0)
+ *d = '\0'; /* NUL-terminate dst */
+ while (*s++)
+ ;
+ }
+
+ return d;
+}
+
+LPWSTR
+lstrlcpyW(LPWSTR dst, int siz, LPCWSTR src)
+{
+ LPWSTR d = dst;
+ LPCWSTR s = src;
+ int n = siz;
+
+ /* Copy as many bytes as will fit */
+ if (n != 0) {
+ while (--n != 0) {
+ if ((*d++ = *s++) == L'\0')
+ break;
+ }
+ }
+
+ /* Not enough room in dst, add NUL and traverse rest of src */
+ if (n == 0) {
+ if (siz != 0)
+ *d = L'\0'; /* NUL-terminate dst */
+ while (*s++)
+ ;
+ }
+
+ return d;
+}
Deleted: tags/jbosssvc/2.0.8/procrun/src/private.h
===================================================================
--- trunk/utils/windows/native/service/procrun/src/private.h 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/src/private.h 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,274 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#ifndef _PRIVATE_H_INCLUDED_
-#define _PRIVATE_H_INCLUDED_
-
-#include "mclib.h"
-
-#ifdef _DEBUG
-
-HANDLE HeapCREATE(DWORD flOptions, SIZE_T dwInitialSize, SIZE_T dwMaximumSize);
-BOOL HeapDESTROY(HANDLE hHeap);
-
-LPVOID HeapALLOC(HANDLE hHeap, DWORD dwFlags, SIZE_T nSize);
-BOOL HeapFREE(HANDLE hHeap, DWORD dwFlags, LPVOID lpMem);
-LPVOID HeapREALLOC(HANDLE hHeap, DWORD dwFlags, LPVOID lpMem, SIZE_T dwBytes);
-
-#else
-
-#define HeapCREATE HeapCreate
-#define HeapDESTROY HeapDestroy
-#define HeapALLOC HeapAlloc
-#define HeapFREE HeapFree
-#define HeapREALLOC HeapReAlloc
-
-#endif
-
-/*
- * Tail queue declarations.
- */
-#define TAILQ_HEAD(name, type) \
-struct name { \
- struct type *tqh_first; /* first element */ \
- struct type **tqh_last; /* addr of last next element */ \
-}
-
-#define TAILQ_HEAD_INITIALIZER(head) \
- { NULL, &(head).tqh_first }
-
-#define TAILQ_ENTRY(type) \
-struct { \
- struct type *tqe_next; /* next element */ \
- struct type **tqe_prev; /* address of previous next element */ \
-}
-
-/*
- * Tail queue functions.
- */
-#define TAILQ_CONCAT(head1, head2, field) do { \
- if (!TAILQ_EMPTY(head2)) { \
- *(head1)->tqh_last = (head2)->tqh_first; \
- (head2)->tqh_first->field.tqe_prev = (head1)->tqh_last; \
- (head1)->tqh_last = (head2)->tqh_last; \
- TAILQ_INIT((head2)); \
- } \
-} while (0)
-
-#define TAILQ_EMPTY(head) ((head)->tqh_first == NULL)
-
-#define TAILQ_FIRST(head) ((head)->tqh_first)
-
-#define TAILQ_FOREACH(var, head, field) \
- for ((var) = TAILQ_FIRST((head)); \
- (var); \
- (var) = TAILQ_NEXT((var), field))
-
-#define TAILQ_FOREACH_REVERSE(var, head, headname, field) \
- for ((var) = TAILQ_LAST((head), headname); \
- (var); \
- (var) = TAILQ_PREV((var), headname, field))
-
-#define TAILQ_INIT(head) do { \
- TAILQ_FIRST((head)) = NULL; \
- (head)->tqh_last = &TAILQ_FIRST((head)); \
-} while (0)
-
-#define TAILQ_INSERT_AFTER(head, listelm, elm, field) do { \
- if ((TAILQ_NEXT((elm), field) = TAILQ_NEXT((listelm), field)) != NULL)\
- TAILQ_NEXT((elm), field)->field.tqe_prev = \
- &TAILQ_NEXT((elm), field); \
- else { \
- (head)->tqh_last = &TAILQ_NEXT((elm), field); \
- } \
- TAILQ_NEXT((listelm), field) = (elm); \
- (elm)->field.tqe_prev = &TAILQ_NEXT((listelm), field); \
-} while (0)
-
-#define TAILQ_INSERT_BEFORE(listelm, elm, field) do { \
- (elm)->field.tqe_prev = (listelm)->field.tqe_prev; \
- TAILQ_NEXT((elm), field) = (listelm); \
- *(listelm)->field.tqe_prev = (elm); \
- (listelm)->field.tqe_prev = &TAILQ_NEXT((elm), field); \
-} while (0)
-
-#define TAILQ_INSERT_HEAD(head, elm, field) do { \
- if ((TAILQ_NEXT((elm), field) = TAILQ_FIRST((head))) != NULL) \
- TAILQ_FIRST((head))->field.tqe_prev = \
- &TAILQ_NEXT((elm), field); \
- else \
- (head)->tqh_last = &TAILQ_NEXT((elm), field); \
- TAILQ_FIRST((head)) = (elm); \
- (elm)->field.tqe_prev = &TAILQ_FIRST((head)); \
-} while (0)
-
-#define TAILQ_INSERT_TAIL(head, elm, field) do { \
- TAILQ_NEXT((elm), field) = NULL; \
- (elm)->field.tqe_prev = (head)->tqh_last; \
- *(head)->tqh_last = (elm); \
- (head)->tqh_last = &TAILQ_NEXT((elm), field); \
-} while (0)
-
-#define TAILQ_LAST(head, headname) \
- (*(((struct headname *)((head)->tqh_last))->tqh_last))
-
-#define TAILQ_NEXT(elm, field) ((elm)->field.tqe_next)
-
-#define TAILQ_PREV(elm, headname, field) \
- (*(((struct headname *)((elm)->field.tqe_prev))->tqh_last))
-
-#define TAILQ_REMOVE(head, elm, field) do { \
- if ((TAILQ_NEXT((elm), field)) != NULL) \
- TAILQ_NEXT((elm), field)->field.tqe_prev = \
- (elm)->field.tqe_prev; \
- else { \
- (head)->tqh_last = (elm)->field.tqe_prev; \
- } \
- *(elm)->field.tqe_prev = TAILQ_NEXT((elm), field); \
-} while (0)
-
-/** Some usefull macros */
-
-#define APXHANDLE_SPINLOCK(h) \
- APXMACRO_BEGIN \
- while (InterlockedCompareExchange(&((h)->lvSpin), 1, 0) != 0) { \
- Sleep(10); \
- SwitchToThread(); \
- } \
- APXMACRO_END
-
-#define APXHANDLE_SPINUNLOCK(h) \
- APXMACRO_BEGIN \
- InterlockedExchange(&((h)->lvSpin), 0); \
- APXMACRO_END
-
-#define APX_SPINLOCK(lock) \
- APXMACRO_BEGIN \
- while (InterlockedCompareExchange(&(lock), 1, 0) != 0) \
- SwitchToThread(); \
- APXMACRO_END
-
-#define APX_SPINUNLOCK(lock) \
- APXMACRO_BEGIN \
- InterlockedExchange(&(lock), 0); \
- APXMACRO_END
-
-/*
- * Define a union with types which are likely to have the longest
- * *relevant* CPU-specific memory word alignment restrictions...
- */
-typedef union APXMEMWORD {
- void *vp;
- void (*fp)(void);
- char *cp;
- long l;
- double d;
-} APXMEMWORD;
-
-typedef struct APXCALLHOOK APXCALLHOOK;
-
-struct APXCALLHOOK {
-
- LPAPXFNCALLBACK fnCallback;
- TAILQ_ENTRY(APXCALLHOOK) queue;
-};
-
-struct stAPXHANDLE {
- /** The type of the handle */
- DWORD dwType;
- /** Handle Flags */
- DWORD dwFlags;
- /** Handle user data size */
- DWORD dwSize;
- /** parameters for event callback */
- WPARAM wParam;
- LPARAM lParam;
- UINT uMsg;
- /** main callback function (using default if not specified) */
- LPAPXFNCALLBACK fnCallback;
- /** callback functions hook list */
- TAILQ_HEAD(_lCallbacks, APXCALLHOOK) lCallbacks;
- /** allocation pool */
- APXHANDLE hPool;
- /** interlocking value */
- LONG volatile lvSpin;
-
- /** message event handle */
- HANDLE hEventHandle;
- /** message event thread */
- HANDLE hEventThread;
- /** message event thread id */
- DWORD hEventThreadId;
- /** private local heap */
- HANDLE hHeap;
- /** list enty for pool */
- TAILQ_ENTRY(stAPXHANDLE) queue;
- /** small userdata pointer */
- union {
- LPVOID lpPtr;
- HANDLE hWinHandle;
- double dValue;
- void (*fpValue)();
- } uData;
-
- APXMEMWORD stAlign;
-};
-
-#define APXHANDLE_DATA(h) ((void *)((char*)(h) + sizeof(stAPXHANDLE)))
-#define APXHANDLE_SZ sizeof(stAPXHANDLE)
-
-extern APX_OSLEVEL _st_apx_oslevel;
-
-#define APX_GET_OSLEVEL() ((_st_apx_oslevel == APX_WINVER_UNK) ? apxGetOsLevel() : _st_apx_oslevel)
-
-#ifdef _DEBUG
-void ErrorMessage(LPCTSTR szError, BOOL bFatal);
-#else
-#define ErrorMessage(szError, bFatal) ((void)0)
-#endif
-/* zero separated, double zero terminated string */
-struct APXMULTISZ {
- DWORD dwAllocated; /* length including terminators */
- DWORD dwInsert; /* next insert position */
-};
-
-typedef struct APXREGENUM {
- HKEY hServicesKey;
- DWORD dwIndex; /* current enum index */
- DWORD cSubKeys; /* number of subkeys */
- DWORD cbMaxSubKey; /* longest subkey size */
- DWORD cchMaxClass; /* longest class string */
- DWORD cValues; /* number of values for key */
- DWORD cchMaxValue; /* longest value name */
- DWORD cbMaxValueData; /* longest value data */
-
-} APXREGENUM, *LPAPXREGENUM;
-
-BOOL apxRegistryEnumServices(LPAPXREGENUM lpEnum, LPAPXSERVENTRY lpEntry);
-
-BOOL apxSetServiceDescriptionW(LPCWSTR szServiceName, LPCWSTR szDescription);
-BOOL apxGetServiceDescriptionW(LPCWSTR szServiceName, LPWSTR szDescription,
- DWORD dwDescriptionLength);
-BOOL apxGetServiceUserW(LPCWSTR szServiceName, LPWSTR szUser,
- DWORD dwUserLength);
-
-DWORD __apxGetMultiSzLengthA(LPCSTR lpStr, LPDWORD lpdwCount);
-DWORD __apxGetMultiSzLengthW(LPCWSTR lpStr, LPDWORD lpdwCount);
-LPSTR __apxGetEnvironmentVariableA(APXHANDLE hPool, LPCSTR szName);
-LPWSTR __apxGetEnvironmentVariableW(APXHANDLE hPool, LPCWSTR wsName);
-
-
-#endif /* _PRIVATE_H_INCLUDED_ */
Copied: tags/jbosssvc/2.0.8/procrun/src/private.h (from rev 2689, trunk/utils/windows/native/service/procrun/src/private.h)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/src/private.h (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/src/private.h 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,269 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef _PRIVATE_H_INCLUDED_
+#define _PRIVATE_H_INCLUDED_
+
+#include "mclib.h"
+
+#ifdef _DEBUG
+
+HANDLE HeapCREATE(DWORD flOptions, SIZE_T dwInitialSize, SIZE_T dwMaximumSize);
+BOOL HeapDESTROY(HANDLE hHeap);
+
+LPVOID HeapALLOC(HANDLE hHeap, DWORD dwFlags, SIZE_T nSize);
+BOOL HeapFREE(HANDLE hHeap, DWORD dwFlags, LPVOID lpMem);
+LPVOID HeapREALLOC(HANDLE hHeap, DWORD dwFlags, LPVOID lpMem, SIZE_T dwBytes);
+
+#else
+
+#define HeapCREATE HeapCreate
+#define HeapDESTROY HeapDestroy
+#define HeapALLOC HeapAlloc
+#define HeapFREE HeapFree
+#define HeapREALLOC HeapReAlloc
+
+#endif
+
+/*
+ * Tail queue declarations.
+ */
+#define TAILQ_HEAD(name, type) \
+struct name { \
+ struct type *tqh_first; /* first element */ \
+ struct type **tqh_last; /* addr of last next element */ \
+}
+
+#define TAILQ_HEAD_INITIALIZER(head) \
+ { NULL, &(head).tqh_first }
+
+#define TAILQ_ENTRY(type) \
+struct { \
+ struct type *tqe_next; /* next element */ \
+ struct type **tqe_prev; /* address of previous next element */ \
+}
+
+/*
+ * Tail queue functions.
+ */
+#define TAILQ_CONCAT(head1, head2, field) do { \
+ if (!TAILQ_EMPTY(head2)) { \
+ *(head1)->tqh_last = (head2)->tqh_first; \
+ (head2)->tqh_first->field.tqe_prev = (head1)->tqh_last; \
+ (head1)->tqh_last = (head2)->tqh_last; \
+ TAILQ_INIT((head2)); \
+ } \
+} while (0)
+
+#define TAILQ_EMPTY(head) ((head)->tqh_first == NULL)
+
+#define TAILQ_FIRST(head) ((head)->tqh_first)
+
+#define TAILQ_FOREACH(var, head, field) \
+ for ((var) = TAILQ_FIRST((head)); \
+ (var); \
+ (var) = TAILQ_NEXT((var), field))
+
+#define TAILQ_FOREACH_REVERSE(var, head, headname, field) \
+ for ((var) = TAILQ_LAST((head), headname); \
+ (var); \
+ (var) = TAILQ_PREV((var), headname, field))
+
+#define TAILQ_INIT(head) do { \
+ TAILQ_FIRST((head)) = NULL; \
+ (head)->tqh_last = &TAILQ_FIRST((head)); \
+} while (0)
+
+#define TAILQ_INSERT_AFTER(head, listelm, elm, field) do { \
+ if ((TAILQ_NEXT((elm), field) = TAILQ_NEXT((listelm), field)) != NULL)\
+ TAILQ_NEXT((elm), field)->field.tqe_prev = \
+ &TAILQ_NEXT((elm), field); \
+ else { \
+ (head)->tqh_last = &TAILQ_NEXT((elm), field); \
+ } \
+ TAILQ_NEXT((listelm), field) = (elm); \
+ (elm)->field.tqe_prev = &TAILQ_NEXT((listelm), field); \
+} while (0)
+
+#define TAILQ_INSERT_BEFORE(listelm, elm, field) do { \
+ (elm)->field.tqe_prev = (listelm)->field.tqe_prev; \
+ TAILQ_NEXT((elm), field) = (listelm); \
+ *(listelm)->field.tqe_prev = (elm); \
+ (listelm)->field.tqe_prev = &TAILQ_NEXT((elm), field); \
+} while (0)
+
+#define TAILQ_INSERT_HEAD(head, elm, field) do { \
+ if ((TAILQ_NEXT((elm), field) = TAILQ_FIRST((head))) != NULL) \
+ TAILQ_FIRST((head))->field.tqe_prev = \
+ &TAILQ_NEXT((elm), field); \
+ else \
+ (head)->tqh_last = &TAILQ_NEXT((elm), field); \
+ TAILQ_FIRST((head)) = (elm); \
+ (elm)->field.tqe_prev = &TAILQ_FIRST((head)); \
+} while (0)
+
+#define TAILQ_INSERT_TAIL(head, elm, field) do { \
+ TAILQ_NEXT((elm), field) = NULL; \
+ (elm)->field.tqe_prev = (head)->tqh_last; \
+ *(head)->tqh_last = (elm); \
+ (head)->tqh_last = &TAILQ_NEXT((elm), field); \
+} while (0)
+
+#define TAILQ_LAST(head, headname) \
+ (*(((struct headname *)((head)->tqh_last))->tqh_last))
+
+#define TAILQ_NEXT(elm, field) ((elm)->field.tqe_next)
+
+#define TAILQ_PREV(elm, headname, field) \
+ (*(((struct headname *)((elm)->field.tqe_prev))->tqh_last))
+
+#define TAILQ_REMOVE(head, elm, field) do { \
+ if ((TAILQ_NEXT((elm), field)) != NULL) \
+ TAILQ_NEXT((elm), field)->field.tqe_prev = \
+ (elm)->field.tqe_prev; \
+ else { \
+ (head)->tqh_last = (elm)->field.tqe_prev; \
+ } \
+ *(elm)->field.tqe_prev = TAILQ_NEXT((elm), field); \
+} while (0)
+
+/** Some usefull macros */
+
+#define APXHANDLE_SPINLOCK(h) \
+ APXMACRO_BEGIN \
+ while (InterlockedCompareExchange(&((h)->lvSpin), 1, 0) != 0) { \
+ Sleep(10); \
+ SwitchToThread(); \
+ } \
+ APXMACRO_END
+
+#define APXHANDLE_SPINUNLOCK(h) \
+ APXMACRO_BEGIN \
+ InterlockedExchange(&((h)->lvSpin), 0); \
+ APXMACRO_END
+
+#define APX_SPINLOCK(lock) \
+ APXMACRO_BEGIN \
+ while (InterlockedCompareExchange(&(lock), 1, 0) != 0) \
+ SwitchToThread(); \
+ APXMACRO_END
+
+#define APX_SPINUNLOCK(lock) \
+ APXMACRO_BEGIN \
+ InterlockedExchange(&(lock), 0); \
+ APXMACRO_END
+
+/*
+ * Define a union with types which are likely to have the longest
+ * *relevant* CPU-specific memory word alignment restrictions...
+ */
+typedef union APXMEMWORD {
+ void *vp;
+ void (*fp)(void);
+ char *cp;
+ long l;
+ double d;
+} APXMEMWORD;
+
+typedef struct APXCALLHOOK APXCALLHOOK;
+
+struct APXCALLHOOK {
+
+ LPAPXFNCALLBACK fnCallback;
+ TAILQ_ENTRY(APXCALLHOOK) queue;
+};
+
+struct stAPXHANDLE {
+ /** The type of the handle */
+ DWORD dwType;
+ /** Handle Flags */
+ DWORD dwFlags;
+ /** Handle user data size */
+ DWORD dwSize;
+ /** parameters for event callback */
+ WPARAM wParam;
+ LPARAM lParam;
+ UINT uMsg;
+ /** main callback function (using default if not specified) */
+ LPAPXFNCALLBACK fnCallback;
+ /** callback functions hook list */
+ TAILQ_HEAD(_lCallbacks, APXCALLHOOK) lCallbacks;
+ /** allocation pool */
+ APXHANDLE hPool;
+ /** interlocking value */
+ LONG volatile lvSpin;
+
+ /** message event handle */
+ HANDLE hEventHandle;
+ /** message event thread */
+ HANDLE hEventThread;
+ /** message event thread id */
+ DWORD hEventThreadId;
+ /** private local heap */
+ HANDLE hHeap;
+ /** list enty for pool */
+ TAILQ_ENTRY(stAPXHANDLE) queue;
+ /** small userdata pointer */
+ union {
+ LPVOID lpPtr;
+ HANDLE hWinHandle;
+ double dValue;
+ void (*fpValue)();
+ } uData;
+
+ APXMEMWORD stAlign;
+};
+
+#define APXHANDLE_DATA(h) ((void *)((char*)(h) + sizeof(stAPXHANDLE)))
+#define APXHANDLE_SZ sizeof(stAPXHANDLE)
+
+extern APX_OSLEVEL _st_apx_oslevel;
+
+#define APX_GET_OSLEVEL() ((_st_apx_oslevel == APX_WINVER_UNK) ? apxGetOsLevel() : _st_apx_oslevel)
+
+/* zero separated, double zero terminated string */
+struct APXMULTISZ {
+ DWORD dwAllocated; /* length including terminators */
+ DWORD dwInsert; /* next insert position */
+};
+
+typedef struct APXREGENUM {
+ HKEY hServicesKey;
+ DWORD dwIndex; /* current enum index */
+ DWORD cSubKeys; /* number of subkeys */
+ DWORD cbMaxSubKey; /* longest subkey size */
+ DWORD cchMaxClass; /* longest class string */
+ DWORD cValues; /* number of values for key */
+ DWORD cchMaxValue; /* longest value name */
+ DWORD cbMaxValueData; /* longest value data */
+
+} APXREGENUM, *LPAPXREGENUM;
+
+BOOL apxRegistryEnumServices(LPAPXREGENUM lpEnum, LPAPXSERVENTRY lpEntry);
+
+BOOL apxSetServiceDescriptionW(LPCWSTR szServiceName, LPCWSTR szDescription);
+BOOL apxGetServiceDescriptionW(LPCWSTR szServiceName, LPWSTR szDescription,
+ DWORD dwDescriptionLength);
+BOOL apxGetServiceUserW(LPCWSTR szServiceName, LPWSTR szUser,
+ DWORD dwUserLength);
+
+DWORD __apxGetMultiSzLengthA(LPCSTR lpStr, LPDWORD lpdwCount);
+DWORD __apxGetMultiSzLengthW(LPCWSTR lpStr, LPDWORD lpdwCount);
+LPSTR __apxGetEnvironmentVariableA(APXHANDLE hPool, LPCSTR szName);
+LPWSTR __apxGetEnvironmentVariableW(APXHANDLE hPool, LPCWSTR wsName);
+
+
+#endif /* _PRIVATE_H_INCLUDED_ */
Deleted: tags/jbosssvc/2.0.8/procrun/src/registry.c
===================================================================
--- trunk/utils/windows/native/service/procrun/src/registry.c 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/src/registry.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,1112 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#include "apxwin.h"
-#include "private.h"
-
-static LPCWSTR REGSERVICE_ROOT = L"SYSTEM\\CurrentControlSet\\Services\\";
-static LPCWSTR REGSOFTWARE_ROOT = L"SOFTWARE\\";
-static LPCWSTR REGSERVICE_START = L"Start";
-static LPCWSTR REGSERVICE_USER = L"ObjectName";
-static LPCWSTR REGPARAMS = L"Parameters";
-static LPCWSTR REGDESCRIPTION = L"Description";
-static LPCWSTR REGSEPARATOR = L"\\";
-static LPCWSTR REGAPACHE_ROOT = L"Apache Software Foundation";
-/* predefined java keys */
-static LPCWSTR JRE_REGKEY = L"SOFTWARE\\JavaSoft\\Java Runtime Environment\\";
-static LPCWSTR JDK_REGKEY = L"SOFTWARE\\JavaSoft\\Java Development Kit\\";
-static LPCWSTR JAVA_CURRENT = L"CurrentVersion";
-static LPCWSTR JAVA_RUNTIME = L"RuntimeLib";
-static LPCWSTR JAVA_HOME = L"JAVA_HOME";
-static LPCWSTR JAVAHOME = L"JavaHome";
-static LPCWSTR CONTROL_REGKEY = L"SYSTEM\\CurrentControlSet\\Control";
-static LPCWSTR REGTIMEOUT = L"WaitToKillServiceTimeout";
-
-#define REG_CAN_CREATE(r) \
- ((r)->samOptions & KEY_CREATE_SUB_KEY)
-
-#define REG_CAN_WRITE(r) \
- ((r)->samOptions & KEY_SET_VALUE)
-
-
-#define REG_GET_KEY(r, w, k) \
- APXMACRO_BEGIN \
- switch(w) { \
- case APXREG_SOFTWARE: k = (r)->hRootKey; break; \
- case APXREG_PARAMSOFTWARE: k = (r)->hRparamKey; break; \
- case APXREG_SERVICE: k = (r)->hServKey; break; \
- case APXREG_PARAMSERVICE: k = (r)->hSparamKey; break; \
- case APXREG_USER: k = (r)->hUserKey; break; \
- case APXREG_PARAMUSER: k = (r)->hUparamKey; break; \
- default: k = NULL; break; \
- } APXMACRO_END
-
-
-typedef struct APXREGISTRY APXREGISTRY;
-typedef APXREGISTRY* LPAPXREGISTRY;
-typedef struct APXREGSUBKEY APXREGSUBKEY;
-
-struct APXREGSUBKEY {
- APXHANDLE hRegistry;
- HKEY hKey;
- LPCTSTR syKeyName;
- TAILQ_ENTRY(APXREGSUBKEY);
-};
-
-struct APXREGISTRY {
- HKEY hRootKey; /* root key */
- HKEY hServKey; /* service key */
- HKEY hUserKey; /* user key */
- HKEY hCurrKey; /* Current opened key */
- LPVOID pCurrVal; /* Current value, overwitten on a next call */
- HKEY hRparamKey; /* root\\Parameters */
- HKEY hSparamKey; /* service\\Parameters */
- HKEY hUparamKey; /* service\\Parameters */
- REGSAM samOptions;
- /** list enty for opened subkeys */
- TAILQ_HEAD(_lSubkeys, APXREGSUBKEY) lSubkeys;
-
-};
-
-#define SAFE_CLOSE_KEY(k) \
- if ((k) != NULL && (k) != INVALID_HANDLE_VALUE) { \
- RegCloseKey((k)); \
- (k) = NULL; \
- }
-
-static BOOL __apxRegistryCallback(APXHANDLE hObject, UINT uMsg,
- WPARAM wParam, LPARAM lParam)
-{
- LPAPXREGISTRY lpReg;
-
- lpReg = APXHANDLE_DATA(hObject);
- switch (uMsg) {
- case WM_CLOSE:
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- SAFE_CLOSE_KEY(lpReg->hRparamKey);
- SAFE_CLOSE_KEY(lpReg->hSparamKey);
- SAFE_CLOSE_KEY(lpReg->hUparamKey);
- SAFE_CLOSE_KEY(lpReg->hRootKey);
- SAFE_CLOSE_KEY(lpReg->hServKey);
- SAFE_CLOSE_KEY(lpReg->hUserKey);
- break;
- default:
- break;
- }
- return TRUE;
-}
-
-LPSTR __apxGetRegistrySzA(APXHANDLE hPool, HKEY hKey, LPCSTR szValueName)
-{
- LPSTR szRet;
- DWORD rc;
- DWORD dwType;
- DWORD dwSize;
-
- rc = RegQueryValueExA(hKey, szValueName, NULL, &dwType, NULL, &dwSize);
- if (rc != ERROR_SUCCESS || dwType != REG_SZ) {
- return NULL;
- }
- if (!(szRet = apxPoolAlloc(hPool, dwSize)))
- return NULL;
- RegQueryValueExA(hKey, szValueName, NULL, &dwType, (LPBYTE)szRet, &dwSize);
-
- return szRet;
-}
-
-LPWSTR __apxGetRegistrySzW(APXHANDLE hPool, HKEY hKey, LPCWSTR wsValueName)
-{
- LPWSTR wsRet;
- DWORD rc;
- DWORD dwType;
- DWORD dwSize;
-
- rc = RegQueryValueExW(hKey, wsValueName, NULL, &dwType, NULL, &dwSize);
- if (rc != ERROR_SUCCESS || dwType != REG_SZ) {
- return NULL;
- }
- if (!(wsRet = apxPoolAlloc(hPool, dwSize * sizeof(WCHAR))))
- return NULL;
- RegQueryValueExW(hKey, wsValueName, NULL, &dwType, (LPBYTE)wsRet, &dwSize);
-
- return wsRet;
-}
-
-BOOL __apxGetRegistryStrW(APXHANDLE hPool, HKEY hKey, LPCWSTR wsValueName,
- LPWSTR lpRetval, DWORD dwMaxLen)
-{
- DWORD rc;
- DWORD dwType;
- DWORD dwSize = dwMaxLen;
-
- rc = RegQueryValueExW(hKey, wsValueName, NULL, &dwType, (LPBYTE)lpRetval, &dwSize);
- if (rc != ERROR_SUCCESS || dwType != REG_SZ) {
- lpRetval = L'\0';
- return FALSE;
- }
- else
- return TRUE;
-}
-
-LPBYTE __apxGetRegistryBinaryA(APXHANDLE hPool, HKEY hKey, LPCSTR szValueName,
- LPDWORD lpdwLength)
-{
- LPBYTE lpRet;
- DWORD rc;
- DWORD dwType;
- DWORD dwSize;
-
- rc = RegQueryValueExA(hKey, szValueName, NULL, &dwType, NULL, &dwSize);
- if (rc != ERROR_SUCCESS || dwSize == 0) {
- return NULL;
- }
- if (!(lpRet = apxPoolAlloc(hPool, dwSize)))
- return NULL;
- RegQueryValueExA(hKey, szValueName, NULL, &dwType, lpRet, &dwSize);
- if (lpdwLength)
- *lpdwLength = dwSize;
- return lpRet;
-}
-
-LPBYTE __apxGetRegistryBinaryW(APXHANDLE hPool, HKEY hKey, LPCWSTR wsValueName,
- LPDWORD lpdwLength)
-{
- LPBYTE lpRet;
- DWORD rc;
- DWORD dwType;
- DWORD dwSize;
-
- rc = RegQueryValueExW(hKey, wsValueName, NULL, &dwType, NULL, &dwSize);
- if (rc != ERROR_SUCCESS || dwSize == 0) {
- return NULL;
- }
- if (!(lpRet = apxPoolAlloc(hPool, dwSize)))
- return NULL;
- RegQueryValueExW(hKey, wsValueName, NULL, &dwType, lpRet, &dwSize);
- if (lpdwLength)
- *lpdwLength = dwSize;
-
- return lpRet;
-}
-
-DWORD __apxGetRegistryDwordW(APXHANDLE hPool, HKEY hKey, LPCWSTR wsValueName)
-{
- DWORD dwRet;
- DWORD rc;
- DWORD dwType;
- DWORD dwSize = sizeof(DWORD);
-
- rc = RegQueryValueExW(hKey, wsValueName, NULL, &dwType, (LPBYTE)&dwRet, &dwSize);
- if (rc != ERROR_SUCCESS || dwType != REG_DWORD) {
- return 0xFFFFFFFF;
- }
-
- return dwRet;
-}
-
-
-APXHANDLE
-apxCreateRegistryW(APXHANDLE hPool, REGSAM samDesired,
- LPCWSTR szRoot,
- LPCWSTR szKeyName,
- DWORD dwOptions)
-{
- APXHANDLE hRegistry;
- LPAPXREGISTRY lpReg;
- /* maximum key length is 512 characters. */
- WCHAR buff[SIZ_BUFLEN];
- LONG rc = ERROR_SUCCESS;
- HKEY hRootKey = NULL;
- HKEY hUserKey = NULL;
- HKEY hServKey = NULL;
- HKEY hRparamKey = NULL;
- HKEY hSparamKey = NULL;
- HKEY hUparamKey = NULL;
-
- if (!szKeyName || lstrlenW(szKeyName) > SIZ_RESMAX)
- return NULL;
- if (szRoot && lstrlenW(szRoot) > SIZ_RESMAX)
- return NULL;
-
- /* make the HKLM\\SOFTWARE key */
- lstrcpyW(buff, REGSOFTWARE_ROOT);
- if (szRoot)
- lstrcatW(buff, szRoot);
- else
- lstrcatW(buff, REGAPACHE_ROOT);
- lstrcatW(buff, REGSEPARATOR);
- lstrcatW(buff, szKeyName);
- /* Open or create the root key */
- if (dwOptions & APXREG_SOFTWARE) {
- if (samDesired & KEY_CREATE_SUB_KEY)
- rc = RegCreateKeyExW(HKEY_LOCAL_MACHINE, buff, 0, NULL, 0,
- samDesired, NULL,
- &hRootKey, NULL);
- else
- rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE, buff, 0,
- samDesired, &hRootKey);
- if (rc != ERROR_SUCCESS) {
- hRootKey = NULL;
- goto cleanup;
- }
- /* Open or create the root parameters key */
- if (samDesired & KEY_CREATE_SUB_KEY)
- rc = RegCreateKeyExW(hRootKey, REGPARAMS, 0, NULL, 0,
- samDesired, NULL,
- &hRparamKey, NULL);
- else
- rc = RegOpenKeyExW(hRootKey, REGPARAMS, 0,
- samDesired, &hRparamKey);
- if (rc != ERROR_SUCCESS) {
- hRparamKey = NULL;
- goto cleanup;
- }
- }
-
- if (dwOptions & APXREG_USER) {
- /* Open or create the users root key */
- if (samDesired & KEY_CREATE_SUB_KEY)
- rc = RegCreateKeyExW(HKEY_CURRENT_USER, buff, 0, NULL, 0,
- samDesired, NULL,
- &hUserKey, NULL);
- else
- rc = RegOpenKeyExW(HKEY_CURRENT_USER, buff, 0,
- samDesired, &hUserKey);
- if (rc != ERROR_SUCCESS) {
- hUserKey = NULL;
- goto cleanup;
- }
- /* Open or create the users parameters key */
- if (samDesired & KEY_CREATE_SUB_KEY)
- rc = RegCreateKeyExW(hUserKey, REGPARAMS, 0, NULL, 0,
- samDesired, NULL,
- &hUparamKey, NULL);
- else
- rc = RegOpenKeyExW(hUserKey, REGPARAMS, 0,
- samDesired, &hUparamKey);
- if (rc != ERROR_SUCCESS) {
- hUparamKey = NULL;
- goto cleanup;
- }
- }
- /* Check if we need a service key */
- if (dwOptions & APXREG_SERVICE) {
- lstrcpyW(buff, REGSERVICE_ROOT);
- lstrcatW(buff, szKeyName);
- /* Service has to be created allready */
- rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE, buff, 0,
- samDesired, &hServKey);
- if (rc != ERROR_SUCCESS) {
- hServKey = NULL;
- goto cleanup;
- }
- /* Open or create the root parameters key */
- if (samDesired & KEY_CREATE_SUB_KEY)
- rc = RegCreateKeyExW(hServKey, REGPARAMS, 0, NULL, 0,
- samDesired, NULL,
- &hSparamKey, NULL);
- else
- rc = RegOpenKeyExW(hServKey, REGPARAMS, 0,
- samDesired, &hSparamKey);
- if (rc != ERROR_SUCCESS) {
- hSparamKey = NULL;
- goto cleanup;
- }
- }
- hRegistry = apxHandleCreate(hPool, 0,
- NULL, sizeof(APXREGISTRY),
- __apxRegistryCallback);
- if (IS_INVALID_HANDLE(hRegistry))
- return NULL;
- hRegistry->dwType = APXHANDLE_TYPE_REGISTRY;
- lpReg = APXHANDLE_DATA(hRegistry);
- lpReg->samOptions = samDesired;
- lpReg->hRootKey = hRootKey;
- lpReg->hUserKey = hUserKey;
- lpReg->hServKey = hServKey;
- lpReg->hRparamKey = hRparamKey;
- lpReg->hUparamKey = hUparamKey;
- lpReg->hSparamKey = hSparamKey;
- TAILQ_INIT(&lpReg->lSubkeys);
-
- SetLastError(rc);
- return hRegistry;
-
-cleanup:
- SAFE_CLOSE_KEY(hRparamKey);
- SAFE_CLOSE_KEY(hSparamKey);
- SAFE_CLOSE_KEY(hUparamKey);
- SAFE_CLOSE_KEY(hRootKey);
- SAFE_CLOSE_KEY(hServKey);
- SAFE_CLOSE_KEY(hUserKey);
-
- SetLastError(rc);
- return NULL;
-}
-
-APXHANDLE
-apxCreateRegistryA(APXHANDLE hPool, REGSAM samDesired,
- LPCSTR szRoot,
- LPCSTR szKeyName,
- DWORD dwOptions)
-{
- WCHAR wcRoot[SIZ_RESLEN];
- WCHAR wcKey[SIZ_RESLEN];
- LPWSTR wsRoot = NULL;
- if (szRoot) {
- AsciiToWide(szRoot, wcRoot);
- wsRoot = wcRoot;
- }
- AsciiToWide(szKeyName, wcKey);
- return apxCreateRegistryW(hPool, samDesired, wsRoot, wcKey, dwOptions);
-}
-
-LPSTR
-apxRegistryGetStringA(APXHANDLE hRegistry, DWORD dwFrom,
- LPCSTR szSubkey, LPCSTR szValueName)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return NULL;
- lpReg = APXHANDLE_DATA(hRegistry);
-
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return NULL;
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegOpenKeyExA(hKey, szSubkey, 0,
- lpReg->samOptions, &hSub) != ERROR_SUCCESS)
- return NULL;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- lpReg->pCurrVal = __apxGetRegistrySzA(hRegistry->hPool, hKey, szValueName);
-
- return lpReg->pCurrVal;
-}
-
-LPWSTR
-apxRegistryGetStringW(APXHANDLE hRegistry, DWORD dwFrom,
- LPCWSTR szSubkey, LPCWSTR szValueName)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return NULL;
- lpReg = APXHANDLE_DATA(hRegistry);
-
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return NULL;
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegOpenKeyExW(hKey, szSubkey, 0,
- lpReg->samOptions, &hSub) != ERROR_SUCCESS)
- return NULL;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- lpReg->pCurrVal = __apxGetRegistrySzW(hRegistry->hPool, hKey, szValueName);
-
- return lpReg->pCurrVal;
-}
-
-LPBYTE
-apxRegistryGetBinaryA(APXHANDLE hRegistry, DWORD dwFrom,
- LPCSTR szSubkey, LPCSTR szValueName,
- LPBYTE lpData, LPDWORD lpdwLength)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return NULL;
- lpReg = APXHANDLE_DATA(hRegistry);
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return NULL;
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegOpenKeyExA(hKey, szSubkey, 0,
- lpReg->samOptions, &hSub) != ERROR_SUCCESS)
- return NULL;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- if (lpData && lpdwLength && *lpdwLength) {
- DWORD rc, dwType = REG_BINARY;
- rc = RegQueryValueExA(hKey, szValueName, NULL, &dwType, lpData, lpdwLength);
- if (rc != ERROR_SUCCESS || dwType != REG_BINARY) {
- ErrorMessage(NULL, FALSE);
- return NULL;
- }
- lpReg->pCurrVal = lpData;
- }
- else {
- lpReg->pCurrVal = __apxGetRegistryBinaryA(hRegistry->hPool, hKey, szValueName, lpdwLength);
- }
-
- return lpReg->pCurrVal;
-}
-
-LPBYTE
-apxRegistryGetBinaryW(APXHANDLE hRegistry, DWORD dwFrom,
- LPCWSTR szSubkey, LPCWSTR szValueName,
- LPBYTE lpData, LPDWORD lpdwLength)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return NULL;
- lpReg = APXHANDLE_DATA(hRegistry);
-
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return NULL;
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegOpenKeyExW(hKey, szSubkey, 0,
- lpReg->samOptions, &hSub) != ERROR_SUCCESS)
- return NULL;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- if (lpData && lpdwLength && *lpdwLength) {
- DWORD rc, dwType = REG_BINARY;
- rc = RegQueryValueExW(hKey, szValueName, NULL, &dwType, lpData, lpdwLength);
- if (rc != ERROR_SUCCESS || dwType != REG_BINARY) {
- return NULL;
- }
- lpReg->pCurrVal = lpData;
- }
- else {
- lpReg->pCurrVal = __apxGetRegistryBinaryW(hRegistry->hPool, hKey, szValueName, lpdwLength);
- }
- return lpReg->pCurrVal;
-}
-
-DWORD
-apxRegistryGetNumberW(APXHANDLE hRegistry, DWORD dwFrom,
- LPCWSTR szSubkey, LPCWSTR szValueName)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- DWORD dwRval, rl;
- DWORD rc, dwType = REG_DWORD;
-
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return 0;
- lpReg = APXHANDLE_DATA(hRegistry);
-
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return 0;
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegOpenKeyExW(hKey, szSubkey, 0,
- lpReg->samOptions, &hSub) != ERROR_SUCCESS)
- return 0;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- rl = sizeof(DWORD);
- rc = RegQueryValueExW(hKey, szValueName, NULL, &dwType, (LPBYTE)&dwRval, &rl);
- if (rc != ERROR_SUCCESS || dwType != REG_DWORD)
- return 0;
- else
- return dwRval;
-}
-
-LPWSTR
-apxRegistryGetMzStrW(APXHANDLE hRegistry, DWORD dwFrom,
- LPCWSTR szSubkey, LPCWSTR szValueName,
- LPWSTR lpData, LPDWORD lpdwLength)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return NULL;
- lpReg = APXHANDLE_DATA(hRegistry);
-
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return NULL;
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegOpenKeyExW(hKey, szSubkey, 0,
- lpReg->samOptions, &hSub) != ERROR_SUCCESS)
- return NULL;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- if (lpData && lpdwLength && *lpdwLength) {
- DWORD rc, dwType = REG_MULTI_SZ;
- rc = RegQueryValueExW(hKey, szValueName, NULL, &dwType, (BYTE *)lpData, lpdwLength);
- if (rc != ERROR_SUCCESS || dwType != REG_MULTI_SZ) {
- return NULL;
- }
- lpReg->pCurrVal = lpData;
- }
- else {
- lpReg->pCurrVal = __apxGetRegistryBinaryW(hRegistry->hPool, hKey, szValueName, lpdwLength);
- if (lpReg->pCurrVal && lpdwLength)
- *lpdwLength = *lpdwLength * sizeof(WCHAR);
- }
- return lpReg->pCurrVal;
-}
-
-BOOL
-apxRegistrySetBinaryA(APXHANDLE hRegistry, DWORD dwFrom,
- LPCSTR szSubkey, LPCSTR szValueName,
- const LPBYTE lpData, DWORD dwLength)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- DWORD dwType = REG_BINARY;
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return FALSE;
- lpReg = APXHANDLE_DATA(hRegistry);
-
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return FALSE;
-
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegCreateKeyExA(hKey, szSubkey, 0,
- NULL, 0, lpReg->samOptions,
- NULL, &hSub, NULL) != ERROR_SUCCESS)
- return FALSE;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- if (RegSetValueExA(hKey, szValueName, 0, dwType,
- lpData, dwLength) != ERROR_SUCCESS)
- return FALSE;
-
- return TRUE;
-}
-
-BOOL
-apxRegistrySetBinaryW(APXHANDLE hRegistry, DWORD dwFrom,
- LPCWSTR szSubkey, LPCWSTR szValueName,
- const LPBYTE lpData, DWORD dwLength)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- DWORD dwType = REG_BINARY;
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return FALSE;
- lpReg = APXHANDLE_DATA(hRegistry);
-
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return FALSE;
-
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegCreateKeyExW(hKey, szSubkey, 0,
- NULL, 0, lpReg->samOptions,
- NULL, &hSub, NULL) != ERROR_SUCCESS)
- return FALSE;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- if (RegSetValueExW(hKey, szValueName, 0, dwType,
- lpData, dwLength) != ERROR_SUCCESS)
- return FALSE;
-
- return TRUE;
-}
-
-BOOL
-apxRegistrySetMzStrW(APXHANDLE hRegistry, DWORD dwFrom,
- LPCWSTR szSubkey, LPCWSTR szValueName,
- LPCWSTR lpData, DWORD dwLength)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- DWORD dwType = REG_MULTI_SZ;
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return FALSE;
- lpReg = APXHANDLE_DATA(hRegistry);
-
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return FALSE;
-
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegCreateKeyExW(hKey, szSubkey, 0,
- NULL, 0, lpReg->samOptions,
- NULL, &hSub, NULL) != ERROR_SUCCESS)
- return FALSE;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- if (RegSetValueExW(hKey, szValueName, 0, dwType,
- (const BYTE *)lpData, dwLength) != ERROR_SUCCESS)
- return FALSE;
-
- return TRUE;
-}
-
-BOOL
-apxRegistrySetStrA(APXHANDLE hRegistry, DWORD dwFrom,
- LPCSTR szSubkey, LPCSTR szValueName,
- LPCSTR szValue)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- DWORD dwType = REG_SZ;
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return FALSE;
- lpReg = APXHANDLE_DATA(hRegistry);
-
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return FALSE;
-
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegCreateKeyExA(hKey, szSubkey, 0,
- NULL, 0, lpReg->samOptions,
- NULL, &hSub, NULL) != ERROR_SUCCESS)
- return FALSE;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- if (!szValue || !lstrlenA(szValue)) {
- if (RegDeleteValueA(hKey, szValueName) != ERROR_SUCCESS)
- return FALSE;
- }
- else if (RegSetValueExA(hKey, szValueName, 0, dwType,
- (LPBYTE)szValue, lstrlenA(szValue)) != ERROR_SUCCESS)
- return FALSE;
-
- return TRUE;
-}
-
-BOOL
-apxRegistrySetStrW(APXHANDLE hRegistry, DWORD dwFrom,
- LPCWSTR szSubkey, LPCWSTR szValueName,
- LPCWSTR szValue)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- DWORD dwType = REG_SZ;
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return FALSE;
- lpReg = APXHANDLE_DATA(hRegistry);
-
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return FALSE;
-
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegCreateKeyExW(hKey, szSubkey, 0,
- NULL, 0, lpReg->samOptions,
- NULL, &hSub, NULL) != ERROR_SUCCESS)
- return FALSE;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- if (!szValue || !lstrlenW(szValue)) {
- if (RegDeleteValueW(hKey, szValueName) != ERROR_SUCCESS)
- return FALSE;
- }
- else if (RegSetValueExW(hKey, szValueName, 0, dwType,
- (LPBYTE)szValue,
- lstrlenW(szValue) * sizeof(WCHAR)) != ERROR_SUCCESS)
- return FALSE;
-
- return TRUE;
-}
-
-BOOL
-apxRegistrySetNumW(APXHANDLE hRegistry, DWORD dwFrom,
- LPCWSTR szSubkey, LPCWSTR szValueName,
- DWORD dwValue)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- DWORD dwType = REG_DWORD;
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return FALSE;
- lpReg = APXHANDLE_DATA(hRegistry);
-
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return FALSE;
-
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegCreateKeyExW(hKey, szSubkey, 0,
- NULL, 0, lpReg->samOptions,
- NULL, &hSub, NULL) != ERROR_SUCCESS)
- return FALSE;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- if (RegSetValueExW(hKey, szValueName, 0, dwType,
- (LPBYTE)&dwValue,
- sizeof(DWORD)) != ERROR_SUCCESS)
- return FALSE;
-
- return TRUE;
-}
-
-
-BOOL
-apxRegistryDeleteW(APXHANDLE hRegistry, DWORD dwFrom,
- LPCWSTR szSubkey, LPCWSTR szValueName)
-{
- LPAPXREGISTRY lpReg;
- HKEY hKey, hSub = NULL;
- DWORD dwType = REG_SZ;
- if (IS_INVALID_HANDLE(hRegistry) ||
- hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
- return FALSE;
- lpReg = APXHANDLE_DATA(hRegistry);
-
- REG_GET_KEY(lpReg, dwFrom, hKey);
- if (!hKey)
- return FALSE;
-
- if (szSubkey) {
- SAFE_CLOSE_KEY(lpReg->hCurrKey);
- if (RegOpenKeyExW(hKey, szSubkey, 0,
- lpReg->samOptions, &hSub) != ERROR_SUCCESS)
- return FALSE;
- lpReg->hCurrKey = hSub;
- hKey = hSub;
- }
- if (RegDeleteValueW(hKey, szValueName) != ERROR_SUCCESS)
- return FALSE;
-
- return TRUE;
-}
-
-
-BOOL
-apxDeleteRegistryW(LPCWSTR szRoot,
- LPCWSTR szKeyName,
- BOOL bDeleteEmpty)
-{
- WCHAR buff[SIZ_BUFLEN];
- BOOL rv;
-
- if (!szKeyName || lstrlenW(szKeyName) > SIZ_RESMAX)
- return FALSE;
- if (szRoot && lstrlenW(szRoot) > SIZ_RESMAX)
- return FALSE;
-
- lstrcpyW(buff, REGSOFTWARE_ROOT);
- if (szRoot)
- lstrcatW(buff, szRoot);
- else
- lstrcatW(buff, REGAPACHE_ROOT);
- lstrcatW(buff, REGSEPARATOR);
- lstrcatW(buff, szKeyName);
-
- rv = SHDeleteKeyW(HKEY_LOCAL_MACHINE, buff);
- rv += SHDeleteKeyW(HKEY_CURRENT_USER, buff);
-
- if (bDeleteEmpty) {
- lstrcpyW(buff, REGSOFTWARE_ROOT);
- if (szRoot)
- lstrcatW(buff, szRoot);
- else
- lstrcatW(buff, REGAPACHE_ROOT);
-
- SHDeleteEmptyKeyW(HKEY_LOCAL_MACHINE, buff);
- SHDeleteEmptyKeyW(HKEY_CURRENT_USER, buff);
- }
- return rv;
-}
-
-BOOL
-apxDeleteRegistryA(LPCSTR szRoot,
- LPCSTR szKeyName,
- BOOL bDeleteEmpty)
-{
- WCHAR wcRoot[SIZ_RESLEN];
- WCHAR wcKey[SIZ_RESLEN];
- LPWSTR wsRoot = NULL;
- if (szRoot) {
- AsciiToWide(szRoot, wcRoot);
- wsRoot = wcRoot;
- }
- AsciiToWide(szKeyName, wcKey);
-
- return apxDeleteRegistryW(wsRoot, wcKey, bDeleteEmpty);
-}
-
-
-LPWSTR apxGetJavaSoftHome(APXHANDLE hPool, BOOL bPreferJre)
-{
- LPWSTR wsJhome, off;
- DWORD err, dwLen;
- HKEY hKey;
- WCHAR wsBuf[SIZ_BUFLEN];
- WCHAR wsKey[SIZ_RESLEN];
-#if 1 /* XXX: Add that customizable using function call arg */
- if (!bPreferJre && (wsJhome = __apxGetEnvironmentVariableW(hPool, JAVA_HOME)))
- return wsJhome;
-#endif
- lstrcpyW(wsKey, JAVA_CURRENT);
- if (bPreferJre)
- lstrcpyW(wsBuf, JRE_REGKEY);
- else
- lstrcpyW(wsBuf, JDK_REGKEY);
- dwLen = lstrlenW(wsBuf);
- off = &wsBuf[dwLen];
- dwLen = SIZ_RESMAX;
- if ((err = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wsBuf,
- 0, KEY_READ, &hKey)) != ERROR_SUCCESS) {
- return NULL;
- }
- if ((err = RegQueryValueExW(hKey, JAVA_CURRENT, NULL, NULL,
- (LPBYTE)off,
- &dwLen)) != ERROR_SUCCESS) {
- RegCloseKey(hKey);
- return NULL;
- }
- RegCloseKey(hKey);
- if ((err = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wsBuf,
- 0, KEY_READ, &hKey)) != ERROR_SUCCESS) {
- return NULL;
- }
- wsJhome = __apxGetRegistrySzW(hPool, hKey, JAVAHOME);
- if (wsJhome)
- SetEnvironmentVariableW(JAVA_HOME, wsJhome);
- RegCloseKey(hKey);
-
- return wsJhome;
-}
-
-LPWSTR apxGetJavaSoftRuntimeLib(APXHANDLE hPool)
-{
- LPWSTR wsRtlib, off;
- DWORD err, dwLen = SIZ_RESLEN;
- HKEY hKey;
- WCHAR wsBuf[SIZ_BUFLEN];
-
- lstrcpyW(wsBuf, JRE_REGKEY);
-
- dwLen = lstrlenW(wsBuf);
- off = &wsBuf[dwLen];
- dwLen = SIZ_RESLEN;
- if ((err = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wsBuf,
- 0, KEY_READ, &hKey)) != ERROR_SUCCESS) {
- return NULL;
- }
- if ((err = RegQueryValueExW(hKey, JAVA_CURRENT, NULL, NULL,
- (LPBYTE)off,
- &dwLen)) != ERROR_SUCCESS) {
- RegCloseKey(hKey);
- return NULL;
- }
- RegCloseKey(hKey);
- if ((err = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wsBuf,
- 0, KEY_READ, &hKey)) != ERROR_SUCCESS) {
- return NULL;
- }
- wsRtlib = __apxGetRegistrySzW(hPool, hKey, JAVA_RUNTIME);
- RegCloseKey(hKey);
-
- return wsRtlib;
-}
-
-/* Service Registry helper functions */
-
-BOOL apxRegistryEnumServices(LPAPXREGENUM lpEnum, LPAPXSERVENTRY lpEntry)
-{
- DWORD rc, dwLength = SIZ_RESLEN;
-
- if (IS_INVALID_HANDLE(lpEnum->hServicesKey)) {
- rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE, REGSERVICE_ROOT, 0,
- KEY_READ, &(lpEnum->hServicesKey));
- if (rc != ERROR_SUCCESS) {
- return FALSE;
- }
- rc = RegQueryInfoKeyW(lpEnum->hServicesKey,
- NULL,
- NULL,
- NULL,
- &lpEnum->cSubKeys,
- &lpEnum->cbMaxSubKey,
- &lpEnum->cchMaxClass,
- &lpEnum->cValues,
- &lpEnum->cchMaxValue,
- &lpEnum->cbMaxValueData,
- NULL,
- NULL);
- /* TODO: add dynamic maxsubkey length */
- if (rc != ERROR_SUCCESS || lpEnum->cbMaxSubKey > SIZ_RESLEN) {
- SAFE_CLOSE_KEY(lpEnum->hServicesKey);
- return FALSE;
- }
- }
- if (lpEnum->dwIndex >= lpEnum->cSubKeys) {
- SAFE_CLOSE_KEY(lpEnum->hServicesKey);
- return FALSE;
- }
- rc = RegEnumKeyExW(lpEnum->hServicesKey,
- lpEnum->dwIndex++,
- lpEntry->szServiceName,
- &dwLength,
- NULL,
- NULL,
- NULL,
- NULL);
- if (rc != ERROR_SUCCESS) {
- SAFE_CLOSE_KEY(lpEnum->hServicesKey);
- return FALSE;
- }
- else {
- HKEY hKey;
- rc = RegOpenKeyExW(lpEnum->hServicesKey, lpEntry->szServiceName,
- 0, KEY_READ, &hKey);
- if (rc != ERROR_SUCCESS) {
- SAFE_CLOSE_KEY(lpEnum->hServicesKey);
- return FALSE;
- }
- __apxGetRegistryStrW(NULL, hKey, REGDESCRIPTION,
- lpEntry->szServiceDescription, SIZ_DESLEN);
- __apxGetRegistryStrW(NULL, hKey, REGSERVICE_USER,
- lpEntry->szObjectName, SIZ_RESLEN);
- lpEntry->dwStart = __apxGetRegistryDwordW(NULL, hKey, REGSERVICE_START);
- RegCloseKey(hKey);
-
- }
- return TRUE;
-}
-
-/* We could use the ChangeServiceConfig2 on WIN2K+
- * For now use the registry.
- */
-BOOL apxSetServiceDescriptionW(LPCWSTR szServiceName, LPCWSTR szDescription)
-{
- HKEY hKey;
- WCHAR wcName[SIZ_RESLEN];
- DWORD rc;
-
- if (lstrlenW(szServiceName) > SIZ_RESMAX)
- return FALSE;
- lstrcpyW(wcName, REGSERVICE_ROOT);
- lstrcatW(wcName, szServiceName);
-
- rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wcName, 0, KEY_WRITE, &hKey);
- if (rc != ERROR_SUCCESS) {
- return FALSE;
- }
-
- rc = RegSetValueExW(hKey, REGDESCRIPTION, 0, REG_SZ, (CONST BYTE *)szDescription,
- lstrlenW(szDescription) * sizeof(WCHAR));
- SAFE_CLOSE_KEY(hKey);
-
- return rc = ERROR_SUCCESS;
-}
-
-BOOL apxGetServiceDescriptionW(LPCWSTR szServiceName, LPWSTR szDescription,
- DWORD dwDescriptionLength)
-{
- HKEY hKey;
- WCHAR wcName[SIZ_RESLEN];
- DWORD rc, l = dwDescriptionLength * sizeof(WCHAR);
- DWORD t = REG_SZ;
- if (lstrlenW(szServiceName) > SIZ_RESMAX)
- return FALSE;
- lstrcpyW(wcName, REGSERVICE_ROOT);
- lstrcatW(wcName, szServiceName);
-
- rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wcName, 0, KEY_READ, &hKey);
- if (rc != ERROR_SUCCESS) {
- return FALSE;
- }
- rc = RegQueryValueExW(hKey, REGDESCRIPTION, NULL, &t, (BYTE *)szDescription,
- &l);
- SAFE_CLOSE_KEY(hKey);
- if (rc == ERROR_SUCCESS && t == REG_SZ)
- return TRUE;
- else
- return FALSE;
-}
-
-BOOL apxGetServiceUserW(LPCWSTR szServiceName, LPWSTR szUser,
- DWORD dwUserLength)
-{
- HKEY hKey;
- WCHAR wcName[SIZ_RESLEN];
- DWORD rc, l = dwUserLength * sizeof(WCHAR);
- DWORD t = REG_SZ;
- if (lstrlenW(szServiceName) > SIZ_RESMAX)
- return FALSE;
- lstrcpyW(wcName, REGSERVICE_ROOT);
- lstrcatW(wcName, szServiceName);
-
- rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wcName, 0, KEY_READ, &hKey);
- if (rc != ERROR_SUCCESS) {
- return FALSE;
- }
- rc = RegQueryValueExW(hKey, REGSERVICE_USER, NULL, &t, (BYTE *)szUser,
- &l);
- SAFE_CLOSE_KEY(hKey);
- if (rc == ERROR_SUCCESS && t == REG_SZ)
- return TRUE;
- else
- return FALSE;
-}
-
-DWORD apxGetMaxServiceTimeout(APXHANDLE hPool)
-{
- DWORD maxTimeout = 20000;
- LPWSTR wsMaxTimeout;
- DWORD err;
- HKEY hKey;
- WCHAR wsBuf[SIZ_BUFLEN];
-
- lstrcpyW(wsBuf, CONTROL_REGKEY);
-
- if ((err = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wsBuf,
- 0, KEY_READ, &hKey)) != ERROR_SUCCESS) {
- return maxTimeout;
- }
- wsMaxTimeout = __apxGetRegistrySzW(hPool, hKey, REGTIMEOUT);
- RegCloseKey(hKey);
-
- if (wsMaxTimeout[0]) {
- maxTimeout = (DWORD)apxAtoulW(wsMaxTimeout);
- }
- apxFree(wsMaxTimeout);
-
- return maxTimeout;
-}
Copied: tags/jbosssvc/2.0.8/procrun/src/registry.c (from rev 2689, trunk/utils/windows/native/service/procrun/src/registry.c)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/src/registry.c (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/src/registry.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,1160 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "apxwin.h"
+#include "private.h"
+
+static LPCWSTR REGSERVICE_ROOT = L"SYSTEM\\CurrentControlSet\\Services\\";
+static LPCWSTR REGSOFTWARE_ROOT = L"SOFTWARE\\";
+static LPCWSTR REGSERVICE_START = L"Start";
+static LPCWSTR REGSERVICE_USER = L"ObjectName";
+static LPCWSTR REGPARAMS = L"Parameters";
+static LPCWSTR REGDESCRIPTION = L"Description";
+static LPCWSTR REGSEPARATOR = L"\\";
+static LPCWSTR REGAPACHE_ROOT = L"Apache Software Foundation";
+/* predefined java keys */
+static LPCWSTR JRE_REGKEY = L"SOFTWARE\\JavaSoft\\Java Runtime Environment\\";
+static LPCWSTR JDK_REGKEY = L"SOFTWARE\\JavaSoft\\Java Development Kit\\";
+static LPCWSTR JAVA_CURRENT = L"CurrentVersion";
+static LPCWSTR JAVA_RUNTIME = L"RuntimeLib";
+static LPCWSTR JAVA_HOME = L"JAVA_HOME";
+static LPCWSTR JAVAHOME = L"JavaHome";
+static LPCWSTR CONTROL_REGKEY = L"SYSTEM\\CurrentControlSet\\Control";
+static LPCWSTR REGTIMEOUT = L"WaitToKillServiceTimeout";
+
+#define REG_CAN_CREATE(r) \
+ ((r)->samOptions & KEY_CREATE_SUB_KEY)
+
+#define REG_CAN_WRITE(r) \
+ ((r)->samOptions & KEY_SET_VALUE)
+
+
+#define REG_GET_KEY(r, w, k) \
+ APXMACRO_BEGIN \
+ switch(w) { \
+ case APXREG_SOFTWARE: k = (r)->hRootKey; break; \
+ case APXREG_PARAMSOFTWARE: k = (r)->hRparamKey; break; \
+ case APXREG_SERVICE: k = (r)->hServKey; break; \
+ case APXREG_PARAMSERVICE: k = (r)->hSparamKey; break; \
+ case APXREG_USER: k = (r)->hUserKey; break; \
+ case APXREG_PARAMUSER: k = (r)->hUparamKey; break; \
+ default: k = NULL; break; \
+ } APXMACRO_END
+
+
+typedef struct APXREGISTRY APXREGISTRY;
+typedef APXREGISTRY* LPAPXREGISTRY;
+typedef struct APXREGSUBKEY APXREGSUBKEY;
+
+struct APXREGSUBKEY {
+ APXHANDLE hRegistry;
+ HKEY hKey;
+ LPCTSTR syKeyName;
+ TAILQ_ENTRY(APXREGSUBKEY);
+};
+
+struct APXREGISTRY {
+ HKEY hRootKey; /* root key */
+ HKEY hServKey; /* service key */
+ HKEY hUserKey; /* user key */
+ HKEY hCurrKey; /* Current opened key */
+ LPVOID pCurrVal; /* Current value, overwitten on a next call */
+ HKEY hRparamKey; /* root\\Parameters */
+ HKEY hSparamKey; /* service\\Parameters */
+ HKEY hUparamKey; /* service\\Parameters */
+ REGSAM samOptions;
+ /** list enty for opened subkeys */
+ TAILQ_HEAD(_lSubkeys, APXREGSUBKEY) lSubkeys;
+
+};
+
+#define SAFE_CLOSE_KEY(k) \
+ if ((k) != NULL && (k) != INVALID_HANDLE_VALUE) { \
+ RegCloseKey((k)); \
+ (k) = NULL; \
+ }
+
+static BOOL __apxRegistryCallback(APXHANDLE hObject, UINT uMsg,
+ WPARAM wParam, LPARAM lParam)
+{
+ LPAPXREGISTRY lpReg;
+
+ lpReg = APXHANDLE_DATA(hObject);
+ switch (uMsg) {
+ case WM_CLOSE:
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ SAFE_CLOSE_KEY(lpReg->hRparamKey);
+ SAFE_CLOSE_KEY(lpReg->hSparamKey);
+ SAFE_CLOSE_KEY(lpReg->hUparamKey);
+ SAFE_CLOSE_KEY(lpReg->hRootKey);
+ SAFE_CLOSE_KEY(lpReg->hServKey);
+ SAFE_CLOSE_KEY(lpReg->hUserKey);
+ break;
+ default:
+ break;
+ }
+ return TRUE;
+}
+
+LPSTR __apxGetRegistrySzA(APXHANDLE hPool, HKEY hKey, LPCSTR szValueName)
+{
+ LPSTR szRet;
+ DWORD rc;
+ DWORD dwType;
+ DWORD dwSize;
+
+ rc = RegQueryValueExA(hKey, szValueName, NULL, &dwType, NULL, &dwSize);
+ if (rc != ERROR_SUCCESS || dwType != REG_SZ) {
+ return NULL;
+ }
+ if (!(szRet = apxPoolAlloc(hPool, dwSize)))
+ return NULL;
+ RegQueryValueExA(hKey, szValueName, NULL, &dwType, (LPBYTE)szRet, &dwSize);
+
+ return szRet;
+}
+
+LPWSTR __apxGetRegistrySzW(APXHANDLE hPool, HKEY hKey, LPCWSTR wsValueName)
+{
+ LPWSTR wsRet;
+ DWORD rc;
+ DWORD dwType;
+ DWORD dwSize;
+
+ rc = RegQueryValueExW(hKey, wsValueName, NULL, &dwType, NULL, &dwSize);
+ if (rc != ERROR_SUCCESS || dwType != REG_SZ) {
+ return NULL;
+ }
+ if (!(wsRet = apxPoolAlloc(hPool, dwSize * sizeof(WCHAR))))
+ return NULL;
+ RegQueryValueExW(hKey, wsValueName, NULL, &dwType, (LPBYTE)wsRet, &dwSize);
+
+ return wsRet;
+}
+
+BOOL __apxGetRegistryStrW(APXHANDLE hPool, HKEY hKey, LPCWSTR wsValueName,
+ LPWSTR lpRetval, DWORD dwMaxLen)
+{
+ DWORD rc;
+ DWORD dwType;
+ DWORD dwSize = dwMaxLen;
+
+ rc = RegQueryValueExW(hKey, wsValueName, NULL, &dwType, (LPBYTE)lpRetval, &dwSize);
+ if (rc != ERROR_SUCCESS || dwType != REG_SZ) {
+ lpRetval = L'\0';
+ return FALSE;
+ }
+ else
+ return TRUE;
+}
+
+LPBYTE __apxGetRegistryBinaryA(APXHANDLE hPool, HKEY hKey, LPCSTR szValueName,
+ LPDWORD lpdwLength)
+{
+ LPBYTE lpRet;
+ DWORD rc;
+ DWORD dwType;
+ DWORD dwSize;
+
+ rc = RegQueryValueExA(hKey, szValueName, NULL, &dwType, NULL, &dwSize);
+ if (rc != ERROR_SUCCESS || dwSize == 0) {
+ return NULL;
+ }
+ if (!(lpRet = apxPoolAlloc(hPool, dwSize)))
+ return NULL;
+ RegQueryValueExA(hKey, szValueName, NULL, &dwType, lpRet, &dwSize);
+ if (lpdwLength)
+ *lpdwLength = dwSize;
+ return lpRet;
+}
+
+LPBYTE __apxGetRegistryBinaryW(APXHANDLE hPool, HKEY hKey, LPCWSTR wsValueName,
+ LPDWORD lpdwLength)
+{
+ LPBYTE lpRet;
+ DWORD rc;
+ DWORD dwType;
+ DWORD dwSize;
+
+ rc = RegQueryValueExW(hKey, wsValueName, NULL, &dwType, NULL, &dwSize);
+ if (rc != ERROR_SUCCESS || dwSize == 0) {
+ return NULL;
+ }
+ if (!(lpRet = apxPoolAlloc(hPool, dwSize)))
+ return NULL;
+ RegQueryValueExW(hKey, wsValueName, NULL, &dwType, lpRet, &dwSize);
+ if (lpdwLength)
+ *lpdwLength = dwSize;
+
+ return lpRet;
+}
+
+DWORD __apxGetRegistryDwordW(APXHANDLE hPool, HKEY hKey, LPCWSTR wsValueName)
+{
+ DWORD dwRet;
+ DWORD rc;
+ DWORD dwType;
+ DWORD dwSize = sizeof(DWORD);
+
+ rc = RegQueryValueExW(hKey, wsValueName, NULL, &dwType, (LPBYTE)&dwRet, &dwSize);
+ if (rc != ERROR_SUCCESS || dwType != REG_DWORD) {
+ return 0xFFFFFFFF;
+ }
+
+ return dwRet;
+}
+
+
+APXHANDLE
+apxCreateRegistryW(APXHANDLE hPool, REGSAM samDesired,
+ LPCWSTR szRoot,
+ LPCWSTR szKeyName,
+ DWORD dwOptions)
+{
+ APXHANDLE hRegistry;
+ LPAPXREGISTRY lpReg;
+ /* maximum key length is 512 characters. */
+ WCHAR buff[SIZ_BUFLEN];
+ LONG rc = ERROR_SUCCESS;
+ HKEY hRootKey = NULL;
+ HKEY hUserKey = NULL;
+ HKEY hServKey = NULL;
+ HKEY hRparamKey = NULL;
+ HKEY hSparamKey = NULL;
+ HKEY hUparamKey = NULL;
+
+ if (!szKeyName || lstrlenW(szKeyName) > SIZ_RESMAX)
+ return NULL;
+ if (szRoot && lstrlenW(szRoot) > SIZ_RESMAX)
+ return NULL;
+
+ /* make the HKLM\\SOFTWARE key */
+ lstrcpyW(buff, REGSOFTWARE_ROOT);
+ if (szRoot)
+ lstrcatW(buff, szRoot);
+ else
+ lstrcatW(buff, REGAPACHE_ROOT);
+ lstrcatW(buff, REGSEPARATOR);
+ lstrcatW(buff, szKeyName);
+ /* Open or create the root key */
+ if (dwOptions & APXREG_SOFTWARE) {
+ if (samDesired & KEY_CREATE_SUB_KEY)
+ rc = RegCreateKeyExW(HKEY_LOCAL_MACHINE, buff, 0, NULL, 0,
+ samDesired, NULL,
+ &hRootKey, NULL);
+ else
+ rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE, buff, 0,
+ samDesired, &hRootKey);
+ if (rc != ERROR_SUCCESS) {
+ hRootKey = NULL;
+ goto cleanup;
+ }
+ /* Open or create the root parameters key */
+ if (samDesired & KEY_CREATE_SUB_KEY)
+ rc = RegCreateKeyExW(hRootKey, REGPARAMS, 0, NULL, 0,
+ samDesired, NULL,
+ &hRparamKey, NULL);
+ else
+ rc = RegOpenKeyExW(hRootKey, REGPARAMS, 0,
+ samDesired, &hRparamKey);
+ if (rc != ERROR_SUCCESS) {
+ hRparamKey = NULL;
+ goto cleanup;
+ }
+ }
+
+ if (dwOptions & APXREG_USER) {
+ /* Open or create the users root key */
+ if (samDesired & KEY_CREATE_SUB_KEY)
+ rc = RegCreateKeyExW(HKEY_CURRENT_USER, buff, 0, NULL, 0,
+ samDesired, NULL,
+ &hUserKey, NULL);
+ else
+ rc = RegOpenKeyExW(HKEY_CURRENT_USER, buff, 0,
+ samDesired, &hUserKey);
+ if (rc != ERROR_SUCCESS) {
+ hUserKey = NULL;
+ goto cleanup;
+ }
+ /* Open or create the users parameters key */
+ if (samDesired & KEY_CREATE_SUB_KEY)
+ rc = RegCreateKeyExW(hUserKey, REGPARAMS, 0, NULL, 0,
+ samDesired, NULL,
+ &hUparamKey, NULL);
+ else
+ rc = RegOpenKeyExW(hUserKey, REGPARAMS, 0,
+ samDesired, &hUparamKey);
+ if (rc != ERROR_SUCCESS) {
+ hUparamKey = NULL;
+ goto cleanup;
+ }
+ }
+ /* Check if we need a service key */
+ if (dwOptions & APXREG_SERVICE) {
+ lstrcpyW(buff, REGSERVICE_ROOT);
+ lstrcatW(buff, szKeyName);
+ /* Service has to be created allready */
+ rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE, buff, 0,
+ samDesired, &hServKey);
+ if (rc != ERROR_SUCCESS) {
+ hServKey = NULL;
+ goto cleanup;
+ }
+ /* Open or create the root parameters key */
+ if (samDesired & KEY_CREATE_SUB_KEY)
+ rc = RegCreateKeyExW(hServKey, REGPARAMS, 0, NULL, 0,
+ samDesired, NULL,
+ &hSparamKey, NULL);
+ else
+ rc = RegOpenKeyExW(hServKey, REGPARAMS, 0,
+ samDesired, &hSparamKey);
+ if (rc != ERROR_SUCCESS) {
+ hSparamKey = NULL;
+ goto cleanup;
+ }
+ }
+ hRegistry = apxHandleCreate(hPool, 0,
+ NULL, sizeof(APXREGISTRY),
+ __apxRegistryCallback);
+ if (IS_INVALID_HANDLE(hRegistry))
+ return NULL;
+ hRegistry->dwType = APXHANDLE_TYPE_REGISTRY;
+ lpReg = APXHANDLE_DATA(hRegistry);
+ lpReg->samOptions = samDesired;
+ lpReg->hRootKey = hRootKey;
+ lpReg->hUserKey = hUserKey;
+ lpReg->hServKey = hServKey;
+ lpReg->hRparamKey = hRparamKey;
+ lpReg->hUparamKey = hUparamKey;
+ lpReg->hSparamKey = hSparamKey;
+ TAILQ_INIT(&lpReg->lSubkeys);
+
+ SetLastError(rc);
+ return hRegistry;
+
+cleanup:
+ SAFE_CLOSE_KEY(hRparamKey);
+ SAFE_CLOSE_KEY(hSparamKey);
+ SAFE_CLOSE_KEY(hUparamKey);
+ SAFE_CLOSE_KEY(hRootKey);
+ SAFE_CLOSE_KEY(hServKey);
+ SAFE_CLOSE_KEY(hUserKey);
+
+ SetLastError(rc);
+ return NULL;
+}
+
+APXHANDLE
+apxCreateRegistryA(APXHANDLE hPool, REGSAM samDesired,
+ LPCSTR szRoot,
+ LPCSTR szKeyName,
+ DWORD dwOptions)
+{
+ WCHAR wcRoot[SIZ_RESLEN];
+ WCHAR wcKey[SIZ_RESLEN];
+ LPWSTR wsRoot = NULL;
+ if (szRoot) {
+ AsciiToWide(szRoot, wcRoot);
+ wsRoot = wcRoot;
+ }
+ AsciiToWide(szKeyName, wcKey);
+ return apxCreateRegistryW(hPool, samDesired, wsRoot, wcKey, dwOptions);
+}
+
+LPSTR
+apxRegistryGetStringA(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCSTR szSubkey, LPCSTR szValueName)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return NULL;
+ lpReg = APXHANDLE_DATA(hRegistry);
+
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return NULL;
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegOpenKeyExA(hKey, szSubkey, 0,
+ lpReg->samOptions, &hSub) != ERROR_SUCCESS)
+ return NULL;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ lpReg->pCurrVal = __apxGetRegistrySzA(hRegistry->hPool, hKey, szValueName);
+
+ return lpReg->pCurrVal;
+}
+
+LPWSTR
+apxRegistryGetStringW(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCWSTR szSubkey, LPCWSTR szValueName)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return NULL;
+ lpReg = APXHANDLE_DATA(hRegistry);
+
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return NULL;
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegOpenKeyExW(hKey, szSubkey, 0,
+ lpReg->samOptions, &hSub) != ERROR_SUCCESS)
+ return NULL;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ lpReg->pCurrVal = __apxGetRegistrySzW(hRegistry->hPool, hKey, szValueName);
+
+ return lpReg->pCurrVal;
+}
+
+LPBYTE
+apxRegistryGetBinaryA(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCSTR szSubkey, LPCSTR szValueName,
+ LPBYTE lpData, LPDWORD lpdwLength)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return NULL;
+ lpReg = APXHANDLE_DATA(hRegistry);
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return NULL;
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegOpenKeyExA(hKey, szSubkey, 0,
+ lpReg->samOptions, &hSub) != ERROR_SUCCESS)
+ return NULL;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ if (lpData && lpdwLength && *lpdwLength) {
+ DWORD rc, dwType = REG_BINARY;
+ rc = RegQueryValueExA(hKey, szValueName, NULL, &dwType, lpData, lpdwLength);
+ if (rc != ERROR_SUCCESS || dwType != REG_BINARY) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return NULL;
+ }
+ lpReg->pCurrVal = lpData;
+ }
+ else {
+ lpReg->pCurrVal = __apxGetRegistryBinaryA(hRegistry->hPool, hKey, szValueName, lpdwLength);
+ }
+
+ return lpReg->pCurrVal;
+}
+
+LPBYTE
+apxRegistryGetBinaryW(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCWSTR szSubkey, LPCWSTR szValueName,
+ LPBYTE lpData, LPDWORD lpdwLength)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return NULL;
+ lpReg = APXHANDLE_DATA(hRegistry);
+
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return NULL;
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegOpenKeyExW(hKey, szSubkey, 0,
+ lpReg->samOptions, &hSub) != ERROR_SUCCESS)
+ return NULL;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ if (lpData && lpdwLength && *lpdwLength) {
+ DWORD rc, dwType = REG_BINARY;
+ rc = RegQueryValueExW(hKey, szValueName, NULL, &dwType, lpData, lpdwLength);
+ if (rc != ERROR_SUCCESS || dwType != REG_BINARY) {
+ return NULL;
+ }
+ lpReg->pCurrVal = lpData;
+ }
+ else {
+ lpReg->pCurrVal = __apxGetRegistryBinaryW(hRegistry->hPool, hKey, szValueName, lpdwLength);
+ }
+ return lpReg->pCurrVal;
+}
+
+DWORD
+apxRegistryGetNumberW(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCWSTR szSubkey, LPCWSTR szValueName)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ DWORD dwRval, rl;
+ DWORD rc, dwType = REG_DWORD;
+
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return 0;
+ lpReg = APXHANDLE_DATA(hRegistry);
+
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return 0;
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegOpenKeyExW(hKey, szSubkey, 0,
+ lpReg->samOptions, &hSub) != ERROR_SUCCESS)
+ return 0;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ rl = sizeof(DWORD);
+ rc = RegQueryValueExW(hKey, szValueName, NULL, &dwType, (LPBYTE)&dwRval, &rl);
+ if (rc != ERROR_SUCCESS || dwType != REG_DWORD)
+ return 0;
+ else
+ return dwRval;
+}
+
+LPWSTR
+apxRegistryGetMzStrW(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCWSTR szSubkey, LPCWSTR szValueName,
+ LPWSTR lpData, LPDWORD lpdwLength)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return NULL;
+ lpReg = APXHANDLE_DATA(hRegistry);
+
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return NULL;
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegOpenKeyExW(hKey, szSubkey, 0,
+ lpReg->samOptions, &hSub) != ERROR_SUCCESS)
+ return NULL;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ if (lpData && lpdwLength && *lpdwLength) {
+ DWORD rc, dwType = REG_MULTI_SZ;
+ rc = RegQueryValueExW(hKey, szValueName, NULL, &dwType, (BYTE *)lpData, lpdwLength);
+ if (rc != ERROR_SUCCESS || dwType != REG_MULTI_SZ) {
+ return NULL;
+ }
+ lpReg->pCurrVal = lpData;
+ }
+ else {
+ lpReg->pCurrVal = __apxGetRegistryBinaryW(hRegistry->hPool, hKey, szValueName, lpdwLength);
+ if (lpReg->pCurrVal && lpdwLength)
+ *lpdwLength = *lpdwLength * sizeof(WCHAR);
+ }
+ return lpReg->pCurrVal;
+}
+
+BOOL
+apxRegistrySetBinaryA(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCSTR szSubkey, LPCSTR szValueName,
+ const LPBYTE lpData, DWORD dwLength)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ DWORD dwType = REG_BINARY;
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return FALSE;
+ lpReg = APXHANDLE_DATA(hRegistry);
+
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return FALSE;
+
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegCreateKeyExA(hKey, szSubkey, 0,
+ NULL, 0, lpReg->samOptions,
+ NULL, &hSub, NULL) != ERROR_SUCCESS)
+ return FALSE;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ if (RegSetValueExA(hKey, szValueName, 0, dwType,
+ lpData, dwLength) != ERROR_SUCCESS)
+ return FALSE;
+
+ return TRUE;
+}
+
+BOOL
+apxRegistrySetBinaryW(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCWSTR szSubkey, LPCWSTR szValueName,
+ const LPBYTE lpData, DWORD dwLength)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ DWORD dwType = REG_BINARY;
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return FALSE;
+ lpReg = APXHANDLE_DATA(hRegistry);
+
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return FALSE;
+
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegCreateKeyExW(hKey, szSubkey, 0,
+ NULL, 0, lpReg->samOptions,
+ NULL, &hSub, NULL) != ERROR_SUCCESS)
+ return FALSE;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ if (RegSetValueExW(hKey, szValueName, 0, dwType,
+ lpData, dwLength) != ERROR_SUCCESS)
+ return FALSE;
+
+ return TRUE;
+}
+
+BOOL
+apxRegistrySetMzStrW(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCWSTR szSubkey, LPCWSTR szValueName,
+ LPCWSTR lpData, DWORD dwLength)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ DWORD dwType = REG_MULTI_SZ;
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return FALSE;
+ lpReg = APXHANDLE_DATA(hRegistry);
+
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return FALSE;
+
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegCreateKeyExW(hKey, szSubkey, 0,
+ NULL, 0, lpReg->samOptions,
+ NULL, &hSub, NULL) != ERROR_SUCCESS)
+ return FALSE;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ if (RegSetValueExW(hKey, szValueName, 0, dwType,
+ (const BYTE *)lpData, dwLength) != ERROR_SUCCESS)
+ return FALSE;
+
+ return TRUE;
+}
+
+BOOL
+apxRegistrySetStrA(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCSTR szSubkey, LPCSTR szValueName,
+ LPCSTR szValue)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ DWORD dwType = REG_SZ;
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return FALSE;
+ lpReg = APXHANDLE_DATA(hRegistry);
+
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return FALSE;
+
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegCreateKeyExA(hKey, szSubkey, 0,
+ NULL, 0, lpReg->samOptions,
+ NULL, &hSub, NULL) != ERROR_SUCCESS)
+ return FALSE;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ if (!szValue || !lstrlenA(szValue)) {
+ if (RegDeleteValueA(hKey, szValueName) != ERROR_SUCCESS)
+ return FALSE;
+ }
+ else if (RegSetValueExA(hKey, szValueName, 0, dwType,
+ (LPBYTE)szValue, lstrlenA(szValue) + 1) != ERROR_SUCCESS)
+ return FALSE;
+
+ return TRUE;
+}
+
+BOOL
+apxRegistrySetStrW(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCWSTR szSubkey, LPCWSTR szValueName,
+ LPCWSTR szValue)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ DWORD dwType = REG_SZ;
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return FALSE;
+ lpReg = APXHANDLE_DATA(hRegistry);
+
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return FALSE;
+
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegCreateKeyExW(hKey, szSubkey, 0,
+ NULL, 0, lpReg->samOptions,
+ NULL, &hSub, NULL) != ERROR_SUCCESS)
+ return FALSE;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ if (!szValue || !lstrlenW(szValue)) {
+ if (RegDeleteValueW(hKey, szValueName) != ERROR_SUCCESS)
+ return FALSE;
+ }
+ else if (RegSetValueExW(hKey, szValueName, 0, dwType,
+ (LPBYTE)szValue,
+ (lstrlenW(szValue) + 1) * sizeof(WCHAR)) != ERROR_SUCCESS)
+ return FALSE;
+
+ return TRUE;
+}
+
+BOOL
+apxRegistrySetNumW(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCWSTR szSubkey, LPCWSTR szValueName,
+ DWORD dwValue)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ DWORD dwType = REG_DWORD;
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return FALSE;
+ lpReg = APXHANDLE_DATA(hRegistry);
+
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return FALSE;
+
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegCreateKeyExW(hKey, szSubkey, 0,
+ NULL, 0, lpReg->samOptions,
+ NULL, &hSub, NULL) != ERROR_SUCCESS)
+ return FALSE;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ if (RegSetValueExW(hKey, szValueName, 0, dwType,
+ (LPBYTE)&dwValue,
+ sizeof(DWORD)) != ERROR_SUCCESS)
+ return FALSE;
+
+ return TRUE;
+}
+
+
+BOOL
+apxRegistryDeleteW(APXHANDLE hRegistry, DWORD dwFrom,
+ LPCWSTR szSubkey, LPCWSTR szValueName)
+{
+ LPAPXREGISTRY lpReg;
+ HKEY hKey, hSub = NULL;
+ DWORD dwType = REG_SZ;
+ if (IS_INVALID_HANDLE(hRegistry) ||
+ hRegistry->dwType != APXHANDLE_TYPE_REGISTRY)
+ return FALSE;
+ lpReg = APXHANDLE_DATA(hRegistry);
+
+ REG_GET_KEY(lpReg, dwFrom, hKey);
+ if (!hKey)
+ return FALSE;
+
+ if (szSubkey) {
+ SAFE_CLOSE_KEY(lpReg->hCurrKey);
+ if (RegOpenKeyExW(hKey, szSubkey, 0,
+ lpReg->samOptions, &hSub) != ERROR_SUCCESS)
+ return FALSE;
+ lpReg->hCurrKey = hSub;
+ hKey = hSub;
+ }
+ if (RegDeleteValueW(hKey, szValueName) != ERROR_SUCCESS)
+ return FALSE;
+
+ return TRUE;
+}
+
+
+LONG apxDeleteRegistryRecursive(HKEY hKeyRoot, LPCWSTR szSubKey) {
+ LONG rc = ERROR_SUCCESS;
+ DWORD dwSize = 0;
+ WCHAR szName[SIZ_BUFLEN];
+ HKEY hKey = NULL;
+
+ if (ERROR_SUCCESS == RegDeleteKey(hKeyRoot, szSubKey)) {
+ return ERROR_SUCCESS;
+ }
+
+ rc = RegOpenKeyExW(hKeyRoot, szSubKey, 0, KEY_ENUMERATE_SUB_KEYS | DELETE, &hKey);
+ if (rc != ERROR_SUCCESS) {
+ if (rc == ERROR_FILE_NOT_FOUND) {
+ return ERROR_SUCCESS;
+ } else {
+ return rc;
+ }
+ }
+ while (rc == ERROR_SUCCESS) {
+ dwSize = SIZ_BUFLEN;
+ rc = RegEnumKeyExW(hKey, 0, szName, &dwSize, NULL, NULL, NULL, NULL );
+
+ if (rc == ERROR_NO_MORE_ITEMS) {
+ rc = RegDeleteKeyW(hKeyRoot, szSubKey);
+ break;
+ } else {
+ rc = apxDeleteRegistryRecursive(hKey, szName);
+ if (rc != ERROR_SUCCESS) {
+ break; // abort when we start failing
+ }
+ }
+ }
+ RegCloseKey(hKey);
+ return rc;
+}
+
+
+BOOL
+apxDeleteRegistryW(LPCWSTR szRoot,
+ LPCWSTR szKeyName,
+ REGSAM samDesired,
+ BOOL bDeleteEmptyRoot)
+{
+ WCHAR buff[SIZ_BUFLEN];
+ LONG rc = ERROR_SUCCESS;
+ HKEY hKey = NULL;
+ BOOL rv = TRUE;
+ HKEY hives[] = {HKEY_LOCAL_MACHINE, HKEY_CURRENT_USER, NULL}, *hive = NULL;
+
+ if (!szKeyName || lstrlenW(szKeyName) > SIZ_RESMAX)
+ return FALSE;
+ if (szRoot && lstrlenW(szRoot) > SIZ_RESMAX)
+ return FALSE;
+
+ if (szRoot)
+ lstrcpyW(buff, szRoot);
+ else
+ lstrcpyW(buff, REGAPACHE_ROOT);
+ lstrcatW(buff, REGSEPARATOR);
+
+ for (hive = &hives[0]; *hive; hive++) {
+ HKEY hkeySoftware = NULL;
+
+ rc = RegOpenKeyExW(*hive, REGSOFTWARE_ROOT, 0, KEY_READ | samDesired, &hkeySoftware);
+ if (rc != ERROR_SUCCESS) {
+ rv = FALSE;
+ } else {
+ rc = RegOpenKeyExW(hkeySoftware, buff, 0, samDesired | KEY_ENUMERATE_SUB_KEYS | DELETE, &hKey);
+ if (rc == ERROR_SUCCESS) {
+ rc = apxDeleteRegistryRecursive(hKey, szKeyName);
+ RegCloseKey(hKey);
+ hKey = NULL;
+ rv |= (rc == ERROR_SUCCESS);
+ }
+ if (bDeleteEmptyRoot) {
+ // will fail if there are subkeys, just like we want
+ RegDeleteKeyW(hkeySoftware, buff);
+ }
+ RegCloseKey(hkeySoftware);
+ }
+ }
+ return rv;
+}
+
+BOOL
+apxDeleteRegistryA(LPCSTR szRoot,
+ LPCSTR szKeyName,
+ REGSAM samDesired,
+ BOOL bDeleteEmptyRoot)
+{
+ WCHAR wcRoot[SIZ_RESLEN];
+ WCHAR wcKey[SIZ_RESLEN];
+ LPWSTR wsRoot = NULL;
+ if (szRoot) {
+ AsciiToWide(szRoot, wcRoot);
+ wsRoot = wcRoot;
+ }
+ AsciiToWide(szKeyName, wcKey);
+
+ return apxDeleteRegistryW(wsRoot, wcKey, samDesired, bDeleteEmptyRoot);
+}
+
+
+LPWSTR apxGetJavaSoftHome(APXHANDLE hPool, BOOL bPreferJre)
+{
+ LPWSTR wsJhome, off;
+ DWORD err, dwLen;
+ HKEY hKey;
+ WCHAR wsBuf[SIZ_BUFLEN];
+ WCHAR wsKey[SIZ_RESLEN];
+#if 1 /* XXX: Add that customizable using function call arg */
+ if (!bPreferJre && (wsJhome = __apxGetEnvironmentVariableW(hPool, JAVA_HOME)))
+ return wsJhome;
+#endif
+ lstrcpyW(wsKey, JAVA_CURRENT);
+ if (bPreferJre)
+ lstrcpyW(wsBuf, JRE_REGKEY);
+ else
+ lstrcpyW(wsBuf, JDK_REGKEY);
+ dwLen = lstrlenW(wsBuf);
+ off = &wsBuf[dwLen];
+ dwLen = SIZ_RESMAX;
+ if ((err = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wsBuf,
+ 0, KEY_READ, &hKey)) != ERROR_SUCCESS) {
+ return NULL;
+ }
+ if ((err = RegQueryValueExW(hKey, JAVA_CURRENT, NULL, NULL,
+ (LPBYTE)off,
+ &dwLen)) != ERROR_SUCCESS) {
+ RegCloseKey(hKey);
+ return NULL;
+ }
+ RegCloseKey(hKey);
+ if ((err = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wsBuf,
+ 0, KEY_READ, &hKey)) != ERROR_SUCCESS) {
+ return NULL;
+ }
+ wsJhome = __apxGetRegistrySzW(hPool, hKey, JAVAHOME);
+ if (wsJhome)
+ SetEnvironmentVariableW(JAVA_HOME, wsJhome);
+ RegCloseKey(hKey);
+
+ return wsJhome;
+}
+
+LPWSTR apxGetJavaSoftRuntimeLib(APXHANDLE hPool)
+{
+ LPWSTR wsRtlib, off;
+ DWORD err, dwLen = SIZ_RESLEN;
+ HKEY hKey;
+ WCHAR wsBuf[SIZ_BUFLEN];
+
+ lstrcpyW(wsBuf, JRE_REGKEY);
+
+ dwLen = lstrlenW(wsBuf);
+ off = &wsBuf[dwLen];
+ dwLen = SIZ_RESLEN;
+ if ((err = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wsBuf,
+ 0, KEY_READ, &hKey)) != ERROR_SUCCESS) {
+ return NULL;
+ }
+ if ((err = RegQueryValueExW(hKey, JAVA_CURRENT, NULL, NULL,
+ (LPBYTE)off,
+ &dwLen)) != ERROR_SUCCESS) {
+ RegCloseKey(hKey);
+ return NULL;
+ }
+ RegCloseKey(hKey);
+ if ((err = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wsBuf,
+ 0, KEY_READ, &hKey)) != ERROR_SUCCESS) {
+ return NULL;
+ }
+ wsRtlib = __apxGetRegistrySzW(hPool, hKey, JAVA_RUNTIME);
+ RegCloseKey(hKey);
+
+ return wsRtlib;
+}
+
+/* Service Registry helper functions */
+
+BOOL apxRegistryEnumServices(LPAPXREGENUM lpEnum, LPAPXSERVENTRY lpEntry)
+{
+ DWORD rc, dwLength = SIZ_RESLEN;
+
+ if (IS_INVALID_HANDLE(lpEnum->hServicesKey)) {
+ rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE, REGSERVICE_ROOT, 0,
+ KEY_READ, &(lpEnum->hServicesKey));
+ if (rc != ERROR_SUCCESS) {
+ return FALSE;
+ }
+ rc = RegQueryInfoKeyW(lpEnum->hServicesKey,
+ NULL,
+ NULL,
+ NULL,
+ &lpEnum->cSubKeys,
+ &lpEnum->cbMaxSubKey,
+ &lpEnum->cchMaxClass,
+ &lpEnum->cValues,
+ &lpEnum->cchMaxValue,
+ &lpEnum->cbMaxValueData,
+ NULL,
+ NULL);
+ /* TODO: add dynamic maxsubkey length */
+ if (rc != ERROR_SUCCESS || lpEnum->cbMaxSubKey > SIZ_RESLEN) {
+ SAFE_CLOSE_KEY(lpEnum->hServicesKey);
+ return FALSE;
+ }
+ }
+ if (lpEnum->dwIndex >= lpEnum->cSubKeys) {
+ SAFE_CLOSE_KEY(lpEnum->hServicesKey);
+ return FALSE;
+ }
+ rc = RegEnumKeyExW(lpEnum->hServicesKey,
+ lpEnum->dwIndex++,
+ lpEntry->szServiceName,
+ &dwLength,
+ NULL,
+ NULL,
+ NULL,
+ NULL);
+ if (rc != ERROR_SUCCESS) {
+ SAFE_CLOSE_KEY(lpEnum->hServicesKey);
+ return FALSE;
+ }
+ else {
+ HKEY hKey;
+ rc = RegOpenKeyExW(lpEnum->hServicesKey, lpEntry->szServiceName,
+ 0, KEY_READ, &hKey);
+ if (rc != ERROR_SUCCESS) {
+ SAFE_CLOSE_KEY(lpEnum->hServicesKey);
+ return FALSE;
+ }
+ __apxGetRegistryStrW(NULL, hKey, REGDESCRIPTION,
+ lpEntry->szServiceDescription, SIZ_DESLEN);
+ __apxGetRegistryStrW(NULL, hKey, REGSERVICE_USER,
+ lpEntry->szObjectName, SIZ_RESLEN);
+ lpEntry->dwStart = __apxGetRegistryDwordW(NULL, hKey, REGSERVICE_START);
+ RegCloseKey(hKey);
+
+ }
+ return TRUE;
+}
+
+/* We could use the ChangeServiceConfig2 on WIN2K+
+ * For now use the registry.
+ */
+BOOL apxSetServiceDescriptionW(LPCWSTR szServiceName, LPCWSTR szDescription)
+{
+ HKEY hKey;
+ WCHAR wcName[SIZ_RESLEN];
+ DWORD rc;
+
+ if (lstrlenW(szServiceName) > SIZ_RESMAX)
+ return FALSE;
+ lstrcpyW(wcName, REGSERVICE_ROOT);
+ lstrcatW(wcName, szServiceName);
+
+ rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wcName, 0, KEY_WRITE, &hKey);
+ if (rc != ERROR_SUCCESS) {
+ return FALSE;
+ }
+
+ rc = RegSetValueExW(hKey, REGDESCRIPTION, 0, REG_SZ, (CONST BYTE *)szDescription,
+ (lstrlenW(szDescription) + 1) * sizeof(WCHAR));
+ SAFE_CLOSE_KEY(hKey);
+
+ return rc == ERROR_SUCCESS;
+}
+
+BOOL apxGetServiceDescriptionW(LPCWSTR szServiceName, LPWSTR szDescription,
+ DWORD dwDescriptionLength)
+{
+ HKEY hKey;
+ WCHAR wcName[SIZ_RESLEN];
+ DWORD rc, l = dwDescriptionLength * sizeof(WCHAR);
+ DWORD t = REG_SZ;
+ if (lstrlenW(szServiceName) > SIZ_RESMAX)
+ return FALSE;
+ lstrcpyW(wcName, REGSERVICE_ROOT);
+ lstrcatW(wcName, szServiceName);
+
+ rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wcName, 0, KEY_READ, &hKey);
+ if (rc != ERROR_SUCCESS) {
+ return FALSE;
+ }
+ rc = RegQueryValueExW(hKey, REGDESCRIPTION, NULL, &t, (BYTE *)szDescription,
+ &l);
+ SAFE_CLOSE_KEY(hKey);
+ if (rc == ERROR_SUCCESS && t == REG_SZ)
+ return TRUE;
+ else
+ return FALSE;
+}
+
+BOOL apxGetServiceUserW(LPCWSTR szServiceName, LPWSTR szUser,
+ DWORD dwUserLength)
+{
+ HKEY hKey;
+ WCHAR wcName[SIZ_RESLEN];
+ DWORD rc, l = dwUserLength * sizeof(WCHAR);
+ DWORD t = REG_SZ;
+ if (lstrlenW(szServiceName) > SIZ_RESMAX)
+ return FALSE;
+ lstrcpyW(wcName, REGSERVICE_ROOT);
+ lstrcatW(wcName, szServiceName);
+
+ rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wcName, 0, KEY_READ, &hKey);
+ if (rc != ERROR_SUCCESS) {
+ return FALSE;
+ }
+ rc = RegQueryValueExW(hKey, REGSERVICE_USER, NULL, &t, (BYTE *)szUser,
+ &l);
+ SAFE_CLOSE_KEY(hKey);
+ if (rc == ERROR_SUCCESS && t == REG_SZ)
+ return TRUE;
+ else
+ return FALSE;
+}
+
+DWORD apxGetMaxServiceTimeout(APXHANDLE hPool)
+{
+ DWORD maxTimeout = 20000;
+ LPWSTR wsMaxTimeout;
+ DWORD err;
+ HKEY hKey;
+ WCHAR wsBuf[SIZ_BUFLEN];
+
+ lstrcpyW(wsBuf, CONTROL_REGKEY);
+
+ if ((err = RegOpenKeyExW(HKEY_LOCAL_MACHINE, wsBuf,
+ 0, KEY_READ, &hKey)) != ERROR_SUCCESS) {
+ return maxTimeout;
+ }
+ wsMaxTimeout = __apxGetRegistrySzW(hPool, hKey, REGTIMEOUT);
+ RegCloseKey(hKey);
+
+ if (wsMaxTimeout[0]) {
+ maxTimeout = (DWORD)apxAtoulW(wsMaxTimeout);
+ }
+ apxFree(wsMaxTimeout);
+
+ return maxTimeout;
+}
Deleted: tags/jbosssvc/2.0.8/procrun/src/rprocess.c
===================================================================
--- trunk/utils/windows/native/service/procrun/src/rprocess.c 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/src/rprocess.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,939 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#include "apxwin.h"
-#include "private.h"
-
-#define CHILD_RUNNING 0x0001
-#define CHILD_INITIALIZED 0x0002
-#define CHILD_MAINTREAD_FINISHED 0x0004
-#define PROC_INITIALIZED 0x0008
-#define CHILD_TERMINATE_CODE 19640323 /* Could be any value like my birthday ;-)*/
-
-DYNOLAD_TYPE_DECLARE(CreateRemoteThread,
- __stdcall, HANDLE)(HANDLE, LPSECURITY_ATTRIBUTES,
- DWORD, LPTHREAD_START_ROUTINE,
- LPVOID, DWORD, LPDWORD);
-
-DYNOLAD_TYPE_DECLARE(ExitProcess, __stdcall, void)(UINT);
-
-#define CHECK_IF_ACTIVE(proc) \
- APXMACRO_BEGIN \
- DWORD __st; \
- if (!GetExitCodeProcess((proc)->stProcInfo.hProcess, &__st) || \
- (__st != STILL_ACTIVE)) \
- goto cleanup; \
- APXMACRO_END
-
-#define SAVE_STD_HANDLES(p) \
- APXMACRO_BEGIN \
- if ((p)->bSaveHandles) { \
- (p)->hParentStdSave[0] = GetStdHandle(STD_INPUT_HANDLE); \
- (p)->hParentStdSave[1] = GetStdHandle(STD_OUTPUT_HANDLE); \
- (p)->hParentStdSave[2] = GetStdHandle(STD_ERROR_HANDLE); \
- } APXMACRO_END
-
-#define RESTORE_STD_HANDLES(p) \
- APXMACRO_BEGIN \
- if ((p)->bSaveHandles) { \
- SetStdHandle(STD_INPUT_HANDLE, (p)->hParentStdSave[0]); \
- SetStdHandle(STD_OUTPUT_HANDLE, (p)->hParentStdSave[1]); \
- SetStdHandle(STD_ERROR_HANDLE, (p)->hParentStdSave[2]); \
- } APXMACRO_END
-
-#define REDIRECT_STD_HANDLES(p) \
- APXMACRO_BEGIN \
- if ((p)->bSaveHandles) { \
- SetStdHandle(STD_INPUT_HANDLE, (p)->hChildStdInp); \
- SetStdHandle(STD_OUTPUT_HANDLE, (p)->hChildStdOut); \
- SetStdHandle(STD_ERROR_HANDLE, (p)->hChildStdErr); \
- } APXMACRO_END
-
-typedef struct APXPROCESS {
- DWORD dwChildStatus;
- DWORD dwOptions;
- PROCESS_INFORMATION stProcInfo;
- /* Size of chars for ANSI/Unicode programs */
- DWORD chSize;
- /* application working path */
- LPWSTR szWorkingPath;
- /* executable name */
- LPWSTR szApplicationExec;
- /* command line (first arg is program name for argv[0]) */
- LPWSTR szCommandLine;
- LPWSTR lpEnvironment;
- /* set of child inherited pipes */
- HANDLE hChildStdInp;
- HANDLE hChildStdOut;
- HANDLE hChildStdErr;
- /* parent ends of child pipes */
- HANDLE hChildInpWr;
- HANDLE hChildOutRd;
- HANDLE hChildErrRd;
- /* Saved console pipes */
- HANDLE hParentStdSave[3];
- HANDLE hWorkerThreads[3];
- HANDLE hUserToken;
- HANDLE hCurrentProcess;
- BOOL bSaveHandles;
- /** callback function */
- LPAPXFNCALLBACK fnUserCallback;
- LPSECURITY_ATTRIBUTES lpSA;
- LPVOID lpSD;
- BYTE bSD[SECURITY_DESCRIPTOR_MIN_LENGTH];
- BYTE bSA[sizeof(SECURITY_ATTRIBUTES)];
-
-} APXPROCESS, *LPAPXPROCESS;
-
-/** Process worker thread
- * Monitors the process thread
- */
-static DWORD WINAPI __apxProcWorkerThread(LPVOID lpParameter)
-{
- APXHANDLE hProcess = (APXHANDLE)lpParameter;
- LPAPXPROCESS lpProc;
- DWORD dwExitCode = 0;
-
- lpProc = APXHANDLE_DATA(hProcess);
- /* Wait util a process has finished its initialization.
- */
- WaitForInputIdle(lpProc->stProcInfo.hProcess, INFINITE);
- lpProc->dwChildStatus |= CHILD_INITIALIZED;
- /* Wait until the child process exits */
- if (WaitForSingleObject(lpProc->stProcInfo.hProcess,
- INFINITE) == WAIT_OBJECT_0) {
- lpProc->dwChildStatus |= CHILD_MAINTREAD_FINISHED;
-#if 0
- if (hProcess->fnCallback)
- (*hProcess->fnCallback)(hProcess, WM_QUIT, (WPARAM)dwExitCode, 0);
- lpProc->dwChildStatus &= ~CHILD_RUNNING;
- GetExitCodeProcess(lpProc->stProcInfo.hProcess, &dwExitCode);
-
- /* Close that handle */
-#endif
- }
- ExitThread(0);
- return 0;
-}
-
-static DWORD WINAPI __apxProcStdoutThread(LPVOID lpParameter)
-{
- APXHANDLE hProcess = (APXHANDLE)lpParameter;
- LPAPXPROCESS lpProc;
- APXCALLHOOK *lpCall;
- INT ch;
- DWORD dwReaded;
- lpProc = APXHANDLE_DATA(hProcess);
- while (lpProc->dwChildStatus & CHILD_RUNNING) {
- ch = 0;
- if (!ReadFile(lpProc->hChildOutRd, &ch, lpProc->chSize,
- &dwReaded, NULL) || !dwReaded) {
-
- break;
- }
- if (lpProc->fnUserCallback)
- (*lpProc->fnUserCallback)(hProcess, WM_CHAR, (WPARAM)ch, (LPARAM)0);
- TAILQ_FOREACH(lpCall, &hProcess->lCallbacks, queue) {
- (*lpCall->fnCallback)(hProcess, WM_CHAR, (WPARAM)ch, (LPARAM)0);
- }
- dwReaded = 0;
- SwitchToThread();
- }
- ExitThread(0);
- return 0;
-}
-
-static DWORD WINAPI __apxProcStderrThread(LPVOID lpParameter)
-{
- APXHANDLE hProcess = (APXHANDLE)lpParameter;
- LPAPXPROCESS lpProc;
- APXCALLHOOK *lpCall;
- INT ch;
- DWORD dwReaded;
- lpProc = APXHANDLE_DATA(hProcess);
- while (lpProc->dwChildStatus & CHILD_RUNNING) {
- if (!ReadFile(lpProc->hChildErrRd, &ch, lpProc->chSize,
- &dwReaded, NULL) || !dwReaded) {
-
- break;
- }
- if (lpProc->fnUserCallback)
- (*lpProc->fnUserCallback)(hProcess, WM_CHAR, (WPARAM)ch, (LPARAM)1);
- TAILQ_FOREACH(lpCall, &hProcess->lCallbacks, queue) {
- (*lpCall->fnCallback)(hProcess, WM_CHAR, (WPARAM)ch, (LPARAM)1);
- }
-
- dwReaded = 0;
- SwitchToThread();
- }
-
- ExitThread(0);
- return 0;
-}
-
-static DWORD __apxProcessPutc(LPAPXPROCESS lpProc, INT ch, DWORD dwSize)
-{
- if (lpProc->dwChildStatus & CHILD_RUNNING) {
- DWORD wr = 0;
- if (WriteFile(lpProc->hChildInpWr, &ch, dwSize, &wr, NULL) &&
- wr == dwSize) {
- return 1;
- }
- }
-
- return 0;
-}
-
-static DWORD __apxProcessPuts(LPAPXPROCESS lpProc, LPCTSTR szString)
-{
- DWORD l, n = 0;
- l = lstrlen(szString) * lpProc->chSize;
- if (lpProc->dwChildStatus & CHILD_RUNNING && l) {
- DWORD wr = 0;
- while (TRUE) {
- if (WriteFile(lpProc->hChildInpWr, szString, l,
- &wr, NULL)) {
- n += wr;
- if (wr < l) {
- l -= wr;
- szString += wr;
- }
- else {
- /* Flush the buffer */
- FlushFileBuffers(lpProc->hChildInpWr);
- break;
- }
- }
- else
- break;
- }
- }
-
- return n;
-}
-
-static DWORD __apxProcessWrite(LPAPXPROCESS lpProc, LPCVOID lpData, DWORD dwLen)
-{
- LPBYTE buf = (LPBYTE)lpData;
- DWORD n = 0;
- if (!lpData || !dwLen)
- return 0;
-
- if (lpProc->dwChildStatus & CHILD_RUNNING) {
- DWORD wr = 0;
- while (lpProc->dwChildStatus & CHILD_RUNNING) {
- if (WriteFile(lpProc->hChildInpWr, buf, dwLen,
- &wr, NULL)) {
- n += wr;
- if (wr < dwLen) {
- dwLen -= wr;
- buf += wr;
- }
- else
- break;
- }
- else
- break;
- }
- }
-
- return n;
-}
-
-/** Helper functions */
-static BOOL __apxProcCreateChildPipes(LPAPXPROCESS lpProc)
-{
- BOOL rv = FALSE;
-
- if (!CreatePipe(&(lpProc->hChildStdInp),
- &(lpProc->hChildInpWr),
- lpProc->lpSA, 0)) {
- ErrorMessage(NULL, FALSE);
- goto cleanup;
- }
- if (!SetHandleInformation(lpProc->hChildInpWr,
- HANDLE_FLAG_INHERIT, 0)) {
- ErrorMessage(NULL, FALSE);
- goto cleanup;
- }
-
- if (!CreatePipe(&(lpProc->hChildOutRd),
- &(lpProc->hChildStdOut),
- lpProc->lpSA, 0)) {
- ErrorMessage(NULL, FALSE);
- goto cleanup;
- }
-
- if (!SetHandleInformation(lpProc->hChildOutRd,
- HANDLE_FLAG_INHERIT, 0)) {
- ErrorMessage(NULL, FALSE);
- goto cleanup;
- }
-
- if (!CreatePipe(&(lpProc->hChildErrRd),
- &(lpProc->hChildStdErr),
- lpProc->lpSA, 0)) {
- ErrorMessage(NULL, FALSE);
- goto cleanup;
- }
-
- if (!SetHandleInformation(lpProc->hChildErrRd,
- HANDLE_FLAG_INHERIT, 0)) {
- ErrorMessage(NULL, FALSE);
- goto cleanup;
- }
- rv = TRUE;
-cleanup:
-
- return rv;
-}
-
-/* Close the process.
- * Create the remote thread and call the ExitProcess
- * Terminate the process, if all of the above fails.
- */
-static BOOL __apxProcessClose(APXHANDLE hProcess)
-{
- LPAPXPROCESS lpProc;
- DYNLOAD_FPTR_DECLARE(CreateRemoteThread);
- DYNLOAD_FPTR_DECLARE(ExitProcess);
-
- UINT uExitCode = CHILD_TERMINATE_CODE; /* Could be any value like my birthday ;-)*/
- HANDLE hDup, hRemote;
-
- lpProc = APXHANDLE_DATA(hProcess);
- CHECK_IF_ACTIVE(lpProc);
-
- /* Try to close the child's stdin first */
- SAFE_CLOSE_HANDLE(lpProc->hChildInpWr);
- /* Wait 1 sec for child process to
- * recognize that the stdin has been closed.
- */
- if (WaitForSingleObject(lpProc->stProcInfo.hProcess, 1000) == WAIT_OBJECT_0)
- goto cleanup;
-
- CHECK_IF_ACTIVE(lpProc);
-
- /* Try to create the remote thread in the child address space */
- DYNLOAD_FPTR_ADDRESS(CreateRemoteThread, KERNEL32);
- if (DuplicateHandle(lpProc->hCurrentProcess,
- lpProc->stProcInfo.hProcess,
- lpProc->hCurrentProcess,
- &hDup,
- PROCESS_ALL_ACCESS,
- FALSE, 0)) {
- DYNLOAD_FPTR_ADDRESS(ExitProcess, KERNEL32);
- /* Now call the ExitProcess from inside the client
- * This will safely unload all the dll's.
- */
- hRemote = DYNLOAD_CALL(CreateRemoteThread)(hDup,
- NULL, 0,
- (LPTHREAD_START_ROUTINE)DYNLOAD_FPTR(ExitProcess),
- (PVOID)&uExitCode, 0, NULL);
- if (!IS_INVALID_HANDLE(hRemote)) {
- if (WaitForSingleObject(lpProc->stProcInfo.hProcess,
- 2000) == WAIT_OBJECT_0) {
-
- }
- else {
- TerminateProcess(lpProc->stProcInfo.hProcess, CHILD_TERMINATE_CODE);
- }
- CloseHandle(hRemote);
- }
- CloseHandle(hDup);
- goto cleanup;
- }
-
- TerminateProcess(lpProc->stProcInfo.hProcess, CHILD_TERMINATE_CODE);
-
-cleanup:
- /* Close the process handle */
- SAFE_CLOSE_HANDLE(lpProc->stProcInfo.hProcess);
- lpProc->dwChildStatus &= ~CHILD_RUNNING;
- return TRUE;
-}
-
-static BOOL __apxProcessCallback(APXHANDLE hObject, UINT uMsg,
- WPARAM wParam, LPARAM lParam)
-{
- LPAPXPROCESS lpProc;
-
- lpProc = APXHANDLE_DATA(hObject);
- /* Call the user supplied callback first */
- if (lpProc->fnUserCallback)
- (*lpProc->fnUserCallback)(hObject, uMsg, wParam, lParam);
- switch (uMsg) {
- case WM_CLOSE:
- if (lpProc->dwChildStatus & CHILD_RUNNING) {
- __apxProcessClose(hObject);
- /* Wait for all worker threads to exit */
- WaitForMultipleObjects(3, lpProc->hWorkerThreads, TRUE, INFINITE);
- }
- SAFE_CLOSE_HANDLE(lpProc->stProcInfo.hProcess);
-
- /* Close parent side of the pipes */
- SAFE_CLOSE_HANDLE(lpProc->hChildInpWr);
- SAFE_CLOSE_HANDLE(lpProc->hChildOutRd);
- SAFE_CLOSE_HANDLE(lpProc->hChildErrRd);
-
- SAFE_CLOSE_HANDLE(lpProc->hWorkerThreads[0]);
- SAFE_CLOSE_HANDLE(lpProc->hWorkerThreads[1]);
- SAFE_CLOSE_HANDLE(lpProc->hWorkerThreads[2]);
- SAFE_CLOSE_HANDLE(lpProc->hUserToken);
- apxFree(lpProc->szApplicationExec);
- apxFree(lpProc->szCommandLine);
- apxFree(lpProc->szWorkingPath);
- RESTORE_STD_HANDLES(lpProc);
- SAFE_CLOSE_HANDLE(lpProc->hCurrentProcess);
- if (lpProc->lpEnvironment)
- FreeEnvironmentStringsW(lpProc->lpEnvironment);
-
- case WM_QUIT:
- /* The process has finished
- * This is a WorkerThread message
- */
- lpProc->dwChildStatus &= ~CHILD_RUNNING;
- break;
- case WM_CHAR:
- __apxProcessPutc(lpProc, (INT)lParam, lpProc->chSize);
- break;
- case WM_SETTEXT:
- if (wParam)
- __apxProcessWrite(lpProc, (LPCVOID)lParam, (DWORD)wParam);
- else
- __apxProcessPuts(lpProc, (LPCTSTR)lParam);
- break;
- default:
- break;
- }
-
- return TRUE;
-}
-
-/** Create the process but don't start it
- *
- APXHANDLE
-apxCreateProcessW(APXHANDLE hPool, DWORD dwOptions,
- LPAPXFNCALLBACK fnCallback,
- LPCWSTR szUsername, LPCWSTR szPassword,
- BOOL bLogonAsService)
-{
- APXHANDLE hProcess;
- LPAPXPROCESS lpProc;
- HANDLE hUserToken = NULL;
- if (szUsername && (APX_GET_OSLEVEL() >= APX_WINVER_NT_4)) {
- HANDLE hUser;
-#ifndef _UNICODE
- WCHAR wsUsername[256];
- WCHAR wsPassword[256];
- AsciiToWide(szUsername, wsUsername);
- AsciiToWide(szPassword, wsPassword);
-#else
- LPCWSTR wsUsername = szUsername;
- LPCWSTR wsPassword = szPassword;
-#endif
-
- */
-
-
- APXHANDLE
-apxCreateProcessA(APXHANDLE hPool, DWORD dwOptions,
- LPAPXFNCALLBACK fnCallback,
- LPCSTR szUsername, LPCSTR szPassword,
- BOOL bLogonAsService)
-{
-
- if (szUsername && szPassword) {
- WCHAR wsUsername[256];
- WCHAR wsPassword[256];
- AsciiToWide(szUsername, wsUsername);
- AsciiToWide(szPassword, wsPassword);
- return apxCreateProcessW(hPool, dwOptions, fnCallback,
- wsUsername, wsPassword, bLogonAsService);
- }
- else
- return apxCreateProcessW(hPool, dwOptions, fnCallback,
- NULL, NULL, bLogonAsService);
-}
-
-APXHANDLE
-apxCreateProcessW(APXHANDLE hPool, DWORD dwOptions,
- LPAPXFNCALLBACK fnCallback,
- LPCWSTR szUsername, LPCWSTR szPassword,
- BOOL bLogonAsService)
-{
- APXHANDLE hProcess;
- LPAPXPROCESS lpProc;
- HANDLE hUserToken = NULL;
- /* CreateProcessAsUser is supported only on NT */
- if (szUsername && (APX_GET_OSLEVEL() >= APX_WINVER_NT_4)) {
- HANDLE hUser;
- if (!LogonUserW(szUsername,
- NULL,
- szPassword,
- bLogonAsService ? LOGON32_LOGON_SERVICE : LOGON32_LOGON_NETWORK,
- LOGON32_PROVIDER_DEFAULT,
- &hUser)) {
- /* Logon Failed */
- ErrorMessage(NULL, TRUE);
- return NULL;
- }
- if (!DuplicateTokenEx(hUser,
- TOKEN_QUERY | TOKEN_DUPLICATE | TOKEN_ASSIGN_PRIMARY,
- NULL,
- SecurityImpersonation,
- TokenPrimary,
- &hUserToken)) {
- CloseHandle(hUser);
- /* Failed to duplicate the user token */
- ErrorMessage(NULL, TRUE);
- return NULL;
- }
- if (!ImpersonateLoggedOnUser(hUserToken)) {
- CloseHandle(hUser);
- CloseHandle(hUserToken);
- /* failed to impersonate the logged user */
- ErrorMessage(NULL, TRUE);
- return NULL;
- }
- CloseHandle(hUser);
- }
-
- hProcess = apxHandleCreate(hPool, APXHANDLE_HAS_EVENT,
- NULL, sizeof(APXPROCESS),
- __apxProcessCallback);
- if (IS_INVALID_HANDLE(hProcess))
- return NULL;
- hProcess->dwType = APXHANDLE_TYPE_PROCESS;
- lpProc = APXHANDLE_DATA(hProcess);
- lpProc->dwOptions = dwOptions;
- lpProc->fnUserCallback = fnCallback;
- lpProc->hUserToken = hUserToken;
- /* set the CHAR length */
- if (dwOptions & CREATE_UNICODE_ENVIRONMENT)
- lpProc->chSize = sizeof(WCHAR);
- else
- lpProc->chSize = sizeof(CHAR);
-#if 1
- DuplicateHandle(GetCurrentProcess(),
- GetCurrentProcess(),
- GetCurrentProcess(),
- &lpProc->hCurrentProcess,
- PROCESS_ALL_ACCESS,
- FALSE,
- 0);
-#else
- lpProc->hCurrentProcess = GetCurrentProcess();
-#endif
- lpProc->lpSD = &lpProc->bSD;
-
- InitializeSecurityDescriptor(lpProc->lpSD, SECURITY_DESCRIPTOR_REVISION);
- SetSecurityDescriptorDacl(lpProc->lpSD, -1, 0, 0);
-
- lpProc->lpSA = (LPSECURITY_ATTRIBUTES)&lpProc->bSA[0];
- lpProc->lpSA->nLength = sizeof (SECURITY_ATTRIBUTES);
- lpProc->lpSA->lpSecurityDescriptor = lpProc->lpSD;
- lpProc->lpSA->bInheritHandle = TRUE;
-
- return hProcess;
-}
-
-static WCHAR _desktop_name[] =
- {'W', 'i', 'n', 's', 't', 'a', '0', '\\', 'D', 'e', 'f', 'a', 'u', 'l', 't', 0};
-
-BOOL
-apxProcessExecute(APXHANDLE hProcess)
-{
- LPAPXPROCESS lpProc;
- STARTUPINFOW si;
- DWORD id;
- BOOL bS = FALSE;
-
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return FALSE;
-
- lpProc = APXHANDLE_DATA(hProcess);
- /* don't allow multiple execute calls on the same object */
- if (lpProc->dwChildStatus & PROC_INITIALIZED)
- return FALSE;
- lpProc->bSaveHandles = TRUE;
- SAVE_STD_HANDLES(lpProc);
- if (!__apxProcCreateChildPipes(lpProc))
- goto cleanup;
- REDIRECT_STD_HANDLES(lpProc);
- AplZeroMemory(&si, sizeof(STARTUPINFO));
-
- si.cb = sizeof(STARTUPINFO);
- si.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
- si.wShowWindow = SW_HIDE;
- /* Set the redirected handles */
- si.hStdOutput = lpProc->hChildStdOut;
- si.hStdError = lpProc->hChildStdErr;
- si.hStdInput = lpProc->hChildStdInp;
-
- if (lpProc->lpEnvironment)
- FreeEnvironmentStringsW(lpProc->lpEnvironment);
- lpProc->lpEnvironment = GetEnvironmentStringsW();
-
- if (!IS_INVALID_HANDLE(lpProc->hUserToken)) {
- si.lpDesktop = _desktop_name;
- bS = CreateProcessAsUserW(lpProc->hUserToken,
- lpProc->szApplicationExec,
- lpProc->szCommandLine,
- lpProc->lpSA,
- NULL,
- TRUE,
- CREATE_SUSPENDED | CREATE_UNICODE_ENVIRONMENT | lpProc->dwOptions,
- lpProc->lpEnvironment,
- lpProc->szWorkingPath,
- &si,
- &(lpProc->stProcInfo));
- }
- else {
- OutputDebugStringW(lpProc->szApplicationExec);
- OutputDebugStringW(lpProc->szCommandLine);
-
- bS = CreateProcessW(lpProc->szApplicationExec,
- lpProc->szCommandLine,
- lpProc->lpSA,
- NULL,
- TRUE,
- CREATE_SUSPENDED | CREATE_UNICODE_ENVIRONMENT | lpProc->dwOptions,
- lpProc->lpEnvironment,
- lpProc->szWorkingPath,
- &si,
- &(lpProc->stProcInfo));
- }
- /* Close unused sides of pipes */
- SAFE_CLOSE_HANDLE(lpProc->hChildStdInp);
- SAFE_CLOSE_HANDLE(lpProc->hChildStdOut);
- SAFE_CLOSE_HANDLE(lpProc->hChildStdErr);
- if (!bS)
- goto cleanup;
- /* Set the running flag */
- lpProc->dwChildStatus |= (CHILD_RUNNING | PROC_INITIALIZED);
-
- lpProc->hWorkerThreads[0] = CreateThread(NULL, 0, __apxProcStdoutThread,
- hProcess, 0, &id);
- lpProc->hWorkerThreads[1] = CreateThread(NULL, 0, __apxProcStderrThread,
- hProcess, 0, &id);
- ResumeThread(lpProc->stProcInfo.hThread);
- lpProc->hWorkerThreads[2] = CreateThread(NULL, 0, __apxProcWorkerThread,
- hProcess, 0, &id);
-
- SAFE_CLOSE_HANDLE(lpProc->stProcInfo.hThread);
- /* Close child handles first */
- return TRUE;
-cleanup:
- /* Close parent side of the pipes */
- SAFE_CLOSE_HANDLE(lpProc->hChildInpWr);
- SAFE_CLOSE_HANDLE(lpProc->hChildOutRd);
- SAFE_CLOSE_HANDLE(lpProc->hChildErrRd);
-
- return FALSE;
-}
-
-BOOL
-apxProcessSetExecutableA(APXHANDLE hProcess, LPCSTR szName)
-{
- LPAPXPROCESS lpProc;
-
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return FALSE;
-
- lpProc = APXHANDLE_DATA(hProcess);
- apxFree(lpProc->szApplicationExec);
- lpProc->szApplicationExec = apxPoolWStrdupA(hProcess->hPool, szName);
- OutputDebugStringW(lpProc->szApplicationExec);
- return lpProc->szApplicationExec != NULL;
-}
-
-BOOL
-apxProcessSetExecutableW(APXHANDLE hProcess, LPCWSTR szName)
-{
- LPAPXPROCESS lpProc;
-
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return FALSE;
-
- lpProc = APXHANDLE_DATA(hProcess);
- apxFree(lpProc->szApplicationExec);
- lpProc->szApplicationExec = apxPoolStrdupW(hProcess->hPool, szName);
- OutputDebugStringW(lpProc->szApplicationExec);
- return lpProc->szApplicationExec != NULL;
-}
-
-BOOL
-apxProcessSetCommandLineA(APXHANDLE hProcess, LPCSTR szCmdline)
-{
- LPAPXPROCESS lpProc;
-
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return FALSE;
-
- lpProc = APXHANDLE_DATA(hProcess);
- apxFree(lpProc->szCommandLine);
- lpProc->szCommandLine = apxPoolWStrdupA(hProcess->hPool, szCmdline);
-
- return lpProc->szCommandLine != NULL;
-}
-
-BOOL
-apxProcessSetCommandLineW(APXHANDLE hProcess, LPCWSTR szCmdline)
-{
- LPAPXPROCESS lpProc;
-
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return FALSE;
-
- lpProc = APXHANDLE_DATA(hProcess);
- apxFree(lpProc->szCommandLine);
- lpProc->szCommandLine = apxPoolStrdupW(hProcess->hPool, szCmdline);
-
- return lpProc->szCommandLine != NULL;
-}
-
-BOOL
-apxProcessSetWorkingPathA(APXHANDLE hProcess, LPCSTR szPath)
-{
- LPAPXPROCESS lpProc;
-
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return FALSE;
-
- lpProc = APXHANDLE_DATA(hProcess);
- apxFree(lpProc->szWorkingPath);
- lpProc->szWorkingPath = apxPoolWStrdupA(hProcess->hPool, szPath);
-
- return lpProc->szWorkingPath != NULL;
-}
-
-BOOL
-apxProcessSetWorkingPathW(APXHANDLE hProcess, LPCWSTR szPath)
-{
- LPAPXPROCESS lpProc;
-
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return FALSE;
-
- lpProc = APXHANDLE_DATA(hProcess);
- apxFree(lpProc->szWorkingPath);
- if (!szPath) {
- /* Clear the WorkingPath */
- lpProc->szWorkingPath = NULL;
- return TRUE;
- }
- lpProc->szWorkingPath = apxPoolWStrdupW(hProcess->hPool, szPath);
-
- return lpProc->szWorkingPath != NULL;
-}
-
-DWORD
-apxProcessPutcA(APXHANDLE hProcess, INT ch)
-{
- LPAPXPROCESS lpProc;
-
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return 0;
-
- lpProc = APXHANDLE_DATA(hProcess);
- return __apxProcessPutc(lpProc, ch, sizeof(CHAR));
-}
-
-DWORD
-apxProcessPutcW(APXHANDLE hProcess, INT ch)
-{
- LPAPXPROCESS lpProc;
-
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return 0;
-
- lpProc = APXHANDLE_DATA(hProcess);
- return __apxProcessPutc(lpProc, ch, sizeof(WCHAR));
-}
-
-DWORD
-apxProcessPutsA(APXHANDLE hProcess, LPCSTR szString)
-{
- LPAPXPROCESS lpProc;
- DWORD len;
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS || !szString)
- return 0;
- lpProc = APXHANDLE_DATA(hProcess);
- len = lstrlenA(szString);
- len = __apxProcessWrite(lpProc, szString, len);
- if (len) {
- FlushFileBuffers(lpProc->hChildInpWr);
- return len;
- }
- else
- return 0;
-}
-
-DWORD
-apxProcessPutsW(APXHANDLE hProcess, LPCWSTR szString)
-{
- LPAPXPROCESS lpProc;
- DWORD len;
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS || !szString)
- return 0;
- lpProc = APXHANDLE_DATA(hProcess);
- len = lstrlenW(szString);
- len = __apxProcessWrite(lpProc, szString, len * 2);
- if (len) {
- FlushFileBuffers(lpProc->hChildInpWr);
- return (len / sizeof(WCHAR));
- }
- else
- return 0;
-}
-
-DWORD
-apxProcessWrite(APXHANDLE hProcess, LPCVOID lpData, DWORD dwLen)
-{
- LPAPXPROCESS lpProc;
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return 0;
-
- lpProc = APXHANDLE_DATA(hProcess);
-
- return __apxProcessWrite(lpProc, lpData, dwLen);
-}
-
-BOOL
-apxProcessFlushStdin(APXHANDLE hProcess)
-{
- LPAPXPROCESS lpProc;
-
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return FALSE;
-
- lpProc = APXHANDLE_DATA(hProcess);
-
- if (lpProc->dwChildStatus & CHILD_RUNNING) {
- return FlushFileBuffers(lpProc->hChildInpWr);
- }
-
- return FALSE;
-}
-
-VOID
-apxProcessCloseInputStream(APXHANDLE hProcess)
-{
- if (hProcess->dwType == APXHANDLE_TYPE_PROCESS) {
- LPAPXPROCESS lpProc = APXHANDLE_DATA(hProcess);
- if (lpProc->dwChildStatus & CHILD_RUNNING)
- SAFE_CLOSE_HANDLE(lpProc->hChildInpWr);
- }
-}
-
-DWORD
-apxProcessWait(APXHANDLE hProcess, DWORD dwMilliseconds, BOOL bKill)
-{
- LPAPXPROCESS lpProc;
-
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return WAIT_ABANDONED;
-
- lpProc = APXHANDLE_DATA(hProcess);
-
- if (lpProc->dwChildStatus & CHILD_RUNNING) {
- DWORD rv = WaitForMultipleObjects(3, lpProc->hWorkerThreads,
- TRUE, dwMilliseconds);
- if (rv == WAIT_TIMEOUT && bKill) {
- __apxProcessCallback(hProcess, WM_CLOSE, 0, 0);
- }
- return rv;
- }
- else
- return WAIT_OBJECT_0;
-}
-
-BOOL
-apxProcessRunning(APXHANDLE hProcess)
-{
- LPAPXPROCESS lpProc;
-
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return FALSE;
-
- lpProc = APXHANDLE_DATA(hProcess);
-
- return (lpProc->dwChildStatus & CHILD_RUNNING);
-}
-
-static LPWSTR __apxStrQuote(LPWSTR lpDest, LPCWSTR szSrc)
-{
- LPWSTR p;
- BOOL q = FALSE;
-
- /* Find if string has embeded spaces
- */
- for (p = (LPWSTR)szSrc; *p; p++) {
- if (*p == L' ' || *p == '\t' ||
- *p == '"' || *p == '\\') {
- q = TRUE;
- break;
- }
- }
- p = lpDest;
- if (q) *p++ = L'"';
- while (*szSrc) {
- if (*szSrc == '"' || *szSrc == '\\')
- *p++ = L'\\';
- *p++ = *szSrc++;
- }
- if (q) *p++ = L'"';
- return p;
-}
-
-BOOL
-apxProcessSetCommandArgsW(APXHANDLE hProcess, LPCWSTR szTitle,
- DWORD dwArgc, LPCWSTR *lpArgs)
-{
- LPAPXPROCESS lpProc;
- DWORD i, l = 0;
- LPWSTR p;
- if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
- return FALSE;
-
- lpProc = APXHANDLE_DATA(hProcess);
- apxFree(lpProc->szCommandLine);
-
- l = lstrlenW(szTitle) + 3;
- for (i = 0; i < (DWORD)lstrlenW(szTitle); i++) {
- if (szTitle[i] == L'"')
- ++l;
- }
- for (i = 0; i < dwArgc; i++) {
- int x, q = 0;
- l += (lstrlenW(lpArgs[i]) + 3);
- for (x = 0; x < lstrlenW(lpArgs[i]); x++) {
- if (lpArgs[i][x] == L'"')
- ++q;
- }
- l += q;
- }
- p = lpProc->szCommandLine = apxPoolAlloc(hProcess->hPool, l * sizeof(WCHAR));
- p = __apxStrQuote(p, szTitle);
- for (i = 0; i < dwArgc; i++) {
- *p++ = L' ';
- p = __apxStrQuote(p, lpArgs[i]);
- }
- *p = L'\0';
- OutputDebugStringW(lpProc->szCommandLine);
- return lpProc->szCommandLine != NULL;
-}
Copied: tags/jbosssvc/2.0.8/procrun/src/rprocess.c (from rev 2689, trunk/utils/windows/native/service/procrun/src/rprocess.c)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/src/rprocess.c (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/src/rprocess.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,943 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "apxwin.h"
+#include "private.h"
+
+#define CHILD_RUNNING 0x0001
+#define CHILD_INITIALIZED 0x0002
+#define CHILD_MAINTREAD_FINISHED 0x0004
+#define PROC_INITIALIZED 0x0008
+#define CHILD_TERMINATE_CODE 19640323 /* Could be any value like my birthday ;-)*/
+
+DYNOLAD_TYPE_DECLARE(CreateRemoteThread,
+ __stdcall, HANDLE)(HANDLE, LPSECURITY_ATTRIBUTES,
+ DWORD, LPTHREAD_START_ROUTINE,
+ LPVOID, DWORD, LPDWORD);
+
+DYNOLAD_TYPE_DECLARE(ExitProcess, __stdcall, void)(UINT);
+
+#define CHECK_IF_ACTIVE(proc) \
+ APXMACRO_BEGIN \
+ DWORD __st; \
+ if (!GetExitCodeProcess((proc)->stProcInfo.hProcess, &__st) || \
+ (__st != STILL_ACTIVE)) \
+ goto cleanup; \
+ APXMACRO_END
+
+#define SAVE_STD_HANDLES(p) \
+ APXMACRO_BEGIN \
+ if ((p)->bSaveHandles) { \
+ (p)->hParentStdSave[0] = GetStdHandle(STD_INPUT_HANDLE); \
+ (p)->hParentStdSave[1] = GetStdHandle(STD_OUTPUT_HANDLE); \
+ (p)->hParentStdSave[2] = GetStdHandle(STD_ERROR_HANDLE); \
+ } APXMACRO_END
+
+#define RESTORE_STD_HANDLES(p) \
+ APXMACRO_BEGIN \
+ if ((p)->bSaveHandles) { \
+ SetStdHandle(STD_INPUT_HANDLE, (p)->hParentStdSave[0]); \
+ SetStdHandle(STD_OUTPUT_HANDLE, (p)->hParentStdSave[1]); \
+ SetStdHandle(STD_ERROR_HANDLE, (p)->hParentStdSave[2]); \
+ } APXMACRO_END
+
+#define REDIRECT_STD_HANDLES(p) \
+ APXMACRO_BEGIN \
+ if ((p)->bSaveHandles) { \
+ SetStdHandle(STD_INPUT_HANDLE, (p)->hChildStdInp); \
+ SetStdHandle(STD_OUTPUT_HANDLE, (p)->hChildStdOut); \
+ SetStdHandle(STD_ERROR_HANDLE, (p)->hChildStdErr); \
+ } APXMACRO_END
+
+typedef struct APXPROCESS {
+ DWORD dwChildStatus;
+ DWORD dwOptions;
+ PROCESS_INFORMATION stProcInfo;
+ /* Size of chars for ANSI/Unicode programs */
+ DWORD chSize;
+ /* application working path */
+ LPWSTR szWorkingPath;
+ /* executable name */
+ LPWSTR szApplicationExec;
+ /* command line (first arg is program name for argv[0]) */
+ LPWSTR szCommandLine;
+ LPWSTR lpEnvironment;
+ /* set of child inherited pipes */
+ HANDLE hChildStdInp;
+ HANDLE hChildStdOut;
+ HANDLE hChildStdErr;
+ /* parent ends of child pipes */
+ HANDLE hChildInpWr;
+ HANDLE hChildOutRd;
+ HANDLE hChildErrRd;
+ /* Saved console pipes */
+ HANDLE hParentStdSave[3];
+ HANDLE hWorkerThreads[3];
+ HANDLE hUserToken;
+ HANDLE hCurrentProcess;
+ BOOL bSaveHandles;
+ /** callback function */
+ LPAPXFNCALLBACK fnUserCallback;
+ LPSECURITY_ATTRIBUTES lpSA;
+ LPVOID lpSD;
+ BYTE bSD[SECURITY_DESCRIPTOR_MIN_LENGTH];
+ BYTE bSA[sizeof(SECURITY_ATTRIBUTES)];
+
+} APXPROCESS, *LPAPXPROCESS;
+
+/** Process worker thread
+ * Monitors the process thread
+ */
+static DWORD WINAPI __apxProcWorkerThread(LPVOID lpParameter)
+{
+ APXHANDLE hProcess = (APXHANDLE)lpParameter;
+ LPAPXPROCESS lpProc;
+ DWORD dwExitCode = 0;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+ /* Wait util a process has finished its initialization.
+ */
+ WaitForInputIdle(lpProc->stProcInfo.hProcess, INFINITE);
+ lpProc->dwChildStatus |= CHILD_INITIALIZED;
+ /* Wait until the child process exits */
+ if (WaitForSingleObject(lpProc->stProcInfo.hProcess,
+ INFINITE) == WAIT_OBJECT_0) {
+ lpProc->dwChildStatus |= CHILD_MAINTREAD_FINISHED;
+#if 0
+ if (hProcess->fnCallback)
+ (*hProcess->fnCallback)(hProcess, WM_QUIT, (WPARAM)dwExitCode, 0);
+ lpProc->dwChildStatus &= ~CHILD_RUNNING;
+ GetExitCodeProcess(lpProc->stProcInfo.hProcess, &dwExitCode);
+
+ /* Close that handle */
+#endif
+ }
+ ExitThread(0);
+ return 0;
+}
+
+static DWORD WINAPI __apxProcStdoutThread(LPVOID lpParameter)
+{
+ APXHANDLE hProcess = (APXHANDLE)lpParameter;
+ LPAPXPROCESS lpProc;
+ APXCALLHOOK *lpCall;
+ INT ch;
+ DWORD dwReaded;
+ lpProc = APXHANDLE_DATA(hProcess);
+ while (lpProc->dwChildStatus & CHILD_RUNNING) {
+ ch = 0;
+ if (!ReadFile(lpProc->hChildOutRd, &ch, lpProc->chSize,
+ &dwReaded, NULL) || !dwReaded) {
+
+ break;
+ }
+ if (lpProc->fnUserCallback)
+ (*lpProc->fnUserCallback)(hProcess, WM_CHAR, (WPARAM)ch, (LPARAM)0);
+ TAILQ_FOREACH(lpCall, &hProcess->lCallbacks, queue) {
+ (*lpCall->fnCallback)(hProcess, WM_CHAR, (WPARAM)ch, (LPARAM)0);
+ }
+ dwReaded = 0;
+ SwitchToThread();
+ }
+ ExitThread(0);
+ return 0;
+}
+
+static DWORD WINAPI __apxProcStderrThread(LPVOID lpParameter)
+{
+ APXHANDLE hProcess = (APXHANDLE)lpParameter;
+ LPAPXPROCESS lpProc;
+ APXCALLHOOK *lpCall;
+ INT ch;
+ DWORD dwReaded;
+ lpProc = APXHANDLE_DATA(hProcess);
+ while (lpProc->dwChildStatus & CHILD_RUNNING) {
+ if (!ReadFile(lpProc->hChildErrRd, &ch, lpProc->chSize,
+ &dwReaded, NULL) || !dwReaded) {
+
+ break;
+ }
+ if (lpProc->fnUserCallback)
+ (*lpProc->fnUserCallback)(hProcess, WM_CHAR, (WPARAM)ch, (LPARAM)1);
+ TAILQ_FOREACH(lpCall, &hProcess->lCallbacks, queue) {
+ (*lpCall->fnCallback)(hProcess, WM_CHAR, (WPARAM)ch, (LPARAM)1);
+ }
+
+ dwReaded = 0;
+ SwitchToThread();
+ }
+
+ ExitThread(0);
+ return 0;
+}
+
+static DWORD __apxProcessPutc(LPAPXPROCESS lpProc, INT ch, DWORD dwSize)
+{
+ if (lpProc->dwChildStatus & CHILD_RUNNING) {
+ DWORD wr = 0;
+ if (WriteFile(lpProc->hChildInpWr, &ch, dwSize, &wr, NULL) &&
+ wr == dwSize) {
+ return 1;
+ }
+ }
+
+ return 0;
+}
+
+static DWORD __apxProcessPuts(LPAPXPROCESS lpProc, LPCTSTR szString)
+{
+ DWORD l, n = 0;
+ l = lstrlen(szString) * lpProc->chSize;
+ if (lpProc->dwChildStatus & CHILD_RUNNING && l) {
+ DWORD wr = 0;
+ while (TRUE) {
+ if (WriteFile(lpProc->hChildInpWr, szString, l,
+ &wr, NULL)) {
+ n += wr;
+ if (wr < l) {
+ l -= wr;
+ szString += wr;
+ }
+ else {
+ /* Flush the buffer */
+ FlushFileBuffers(lpProc->hChildInpWr);
+ break;
+ }
+ }
+ else
+ break;
+ }
+ }
+
+ return n;
+}
+
+static DWORD __apxProcessWrite(LPAPXPROCESS lpProc, LPCVOID lpData, DWORD dwLen)
+{
+ LPBYTE buf = (LPBYTE)lpData;
+ DWORD n = 0;
+ if (!lpData || !dwLen)
+ return 0;
+
+ if (lpProc->dwChildStatus & CHILD_RUNNING) {
+ DWORD wr = 0;
+ while (lpProc->dwChildStatus & CHILD_RUNNING) {
+ if (WriteFile(lpProc->hChildInpWr, buf, dwLen,
+ &wr, NULL)) {
+ n += wr;
+ if (wr < dwLen) {
+ dwLen -= wr;
+ buf += wr;
+ }
+ else
+ break;
+ }
+ else
+ break;
+ }
+ }
+
+ return n;
+}
+
+/** Helper functions */
+static BOOL __apxProcCreateChildPipes(LPAPXPROCESS lpProc)
+{
+ BOOL rv = FALSE;
+
+ if (!CreatePipe(&(lpProc->hChildStdInp),
+ &(lpProc->hChildInpWr),
+ lpProc->lpSA, 0)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ goto cleanup;
+ }
+ if (!SetHandleInformation(lpProc->hChildInpWr,
+ HANDLE_FLAG_INHERIT, 0)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ goto cleanup;
+ }
+
+ if (!CreatePipe(&(lpProc->hChildOutRd),
+ &(lpProc->hChildStdOut),
+ lpProc->lpSA, 0)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ goto cleanup;
+ }
+
+ if (!SetHandleInformation(lpProc->hChildOutRd,
+ HANDLE_FLAG_INHERIT, 0)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ goto cleanup;
+ }
+
+ if (!CreatePipe(&(lpProc->hChildErrRd),
+ &(lpProc->hChildStdErr),
+ lpProc->lpSA, 0)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ goto cleanup;
+ }
+
+ if (!SetHandleInformation(lpProc->hChildErrRd,
+ HANDLE_FLAG_INHERIT, 0)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ goto cleanup;
+ }
+ rv = TRUE;
+cleanup:
+
+ return rv;
+}
+
+/* Close the process.
+ * Create the remote thread and call the ExitProcess
+ * Terminate the process, if all of the above fails.
+ */
+static BOOL __apxProcessClose(APXHANDLE hProcess)
+{
+ LPAPXPROCESS lpProc;
+ DYNLOAD_FPTR_DECLARE(CreateRemoteThread);
+ DYNLOAD_FPTR_DECLARE(ExitProcess);
+
+ UINT uExitCode = CHILD_TERMINATE_CODE; /* Could be any value like my birthday ;-)*/
+ HANDLE hDup, hRemote;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+ CHECK_IF_ACTIVE(lpProc);
+
+ /* Try to close the child's stdin first */
+ SAFE_CLOSE_HANDLE(lpProc->hChildInpWr);
+ /* Wait 1 sec for child process to
+ * recognize that the stdin has been closed.
+ */
+ if (WaitForSingleObject(lpProc->stProcInfo.hProcess, 1000) == WAIT_OBJECT_0)
+ goto cleanup;
+
+ CHECK_IF_ACTIVE(lpProc);
+
+ /* Try to create the remote thread in the child address space */
+ DYNLOAD_FPTR_ADDRESS(CreateRemoteThread, KERNEL32);
+ if (DuplicateHandle(lpProc->hCurrentProcess,
+ lpProc->stProcInfo.hProcess,
+ lpProc->hCurrentProcess,
+ &hDup,
+ PROCESS_ALL_ACCESS,
+ FALSE, 0)) {
+ DYNLOAD_FPTR_ADDRESS(ExitProcess, KERNEL32);
+ /* Now call the ExitProcess from inside the client
+ * This will safely unload all the dll's.
+ */
+ hRemote = DYNLOAD_CALL(CreateRemoteThread)(hDup,
+ NULL, 0,
+ (LPTHREAD_START_ROUTINE)DYNLOAD_FPTR(ExitProcess),
+ (PVOID)&uExitCode, 0, NULL);
+ if (!IS_INVALID_HANDLE(hRemote)) {
+ if (WaitForSingleObject(lpProc->stProcInfo.hProcess,
+ 2000) == WAIT_OBJECT_0) {
+
+ }
+ else {
+ TerminateProcess(lpProc->stProcInfo.hProcess, CHILD_TERMINATE_CODE);
+ }
+ CloseHandle(hRemote);
+ }
+ CloseHandle(hDup);
+ goto cleanup;
+ }
+
+ TerminateProcess(lpProc->stProcInfo.hProcess, CHILD_TERMINATE_CODE);
+
+cleanup:
+ /* Close the process handle */
+ SAFE_CLOSE_HANDLE(lpProc->stProcInfo.hProcess);
+ lpProc->dwChildStatus &= ~CHILD_RUNNING;
+ return TRUE;
+}
+
+static BOOL __apxProcessCallback(APXHANDLE hObject, UINT uMsg,
+ WPARAM wParam, LPARAM lParam)
+{
+ LPAPXPROCESS lpProc;
+
+ lpProc = APXHANDLE_DATA(hObject);
+ /* Call the user supplied callback first */
+ if (lpProc->fnUserCallback)
+ (*lpProc->fnUserCallback)(hObject, uMsg, wParam, lParam);
+ switch (uMsg) {
+ case WM_CLOSE:
+ if (lpProc->dwChildStatus & CHILD_RUNNING) {
+ __apxProcessClose(hObject);
+ /* Wait for all worker threads to exit */
+ WaitForMultipleObjects(3, lpProc->hWorkerThreads, TRUE, INFINITE);
+ }
+ SAFE_CLOSE_HANDLE(lpProc->stProcInfo.hProcess);
+
+ /* Close parent side of the pipes */
+ SAFE_CLOSE_HANDLE(lpProc->hChildInpWr);
+ SAFE_CLOSE_HANDLE(lpProc->hChildOutRd);
+ SAFE_CLOSE_HANDLE(lpProc->hChildErrRd);
+
+ SAFE_CLOSE_HANDLE(lpProc->hWorkerThreads[0]);
+ SAFE_CLOSE_HANDLE(lpProc->hWorkerThreads[1]);
+ SAFE_CLOSE_HANDLE(lpProc->hWorkerThreads[2]);
+ SAFE_CLOSE_HANDLE(lpProc->hUserToken);
+ apxFree(lpProc->szApplicationExec);
+ apxFree(lpProc->szCommandLine);
+ apxFree(lpProc->szWorkingPath);
+ RESTORE_STD_HANDLES(lpProc);
+ SAFE_CLOSE_HANDLE(lpProc->hCurrentProcess);
+ if (lpProc->lpEnvironment)
+ FreeEnvironmentStringsW(lpProc->lpEnvironment);
+
+ case WM_QUIT:
+ /* The process has finished
+ * This is a WorkerThread message
+ */
+ lpProc->dwChildStatus &= ~CHILD_RUNNING;
+ break;
+ case WM_CHAR:
+ __apxProcessPutc(lpProc, (INT)lParam, lpProc->chSize);
+ break;
+ case WM_SETTEXT:
+ if (wParam)
+ __apxProcessWrite(lpProc, (LPCVOID)lParam, (DWORD)wParam);
+ else
+ __apxProcessPuts(lpProc, (LPCTSTR)lParam);
+ break;
+ default:
+ break;
+ }
+
+ return TRUE;
+}
+
+/** Create the process but don't start it
+ *
+ APXHANDLE
+apxCreateProcessW(APXHANDLE hPool, DWORD dwOptions,
+ LPAPXFNCALLBACK fnCallback,
+ LPCWSTR szUsername, LPCWSTR szPassword,
+ BOOL bLogonAsService)
+{
+ APXHANDLE hProcess;
+ LPAPXPROCESS lpProc;
+ HANDLE hUserToken = NULL;
+ if (szUsername && (APX_GET_OSLEVEL() >= APX_WINVER_NT_4)) {
+ HANDLE hUser;
+#ifndef _UNICODE
+ WCHAR wsUsername[256];
+ WCHAR wsPassword[256];
+ AsciiToWide(szUsername, wsUsername);
+ AsciiToWide(szPassword, wsPassword);
+#else
+ LPCWSTR wsUsername = szUsername;
+ LPCWSTR wsPassword = szPassword;
+#endif
+
+ */
+
+
+ APXHANDLE
+apxCreateProcessA(APXHANDLE hPool, DWORD dwOptions,
+ LPAPXFNCALLBACK fnCallback,
+ LPCSTR szUsername, LPCSTR szPassword,
+ BOOL bLogonAsService)
+{
+
+ if (szUsername && szPassword) {
+ WCHAR wsUsername[256];
+ WCHAR wsPassword[256];
+ AsciiToWide(szUsername, wsUsername);
+ AsciiToWide(szPassword, wsPassword);
+ return apxCreateProcessW(hPool, dwOptions, fnCallback,
+ wsUsername, wsPassword, bLogonAsService);
+ }
+ else
+ return apxCreateProcessW(hPool, dwOptions, fnCallback,
+ NULL, NULL, bLogonAsService);
+}
+
+APXHANDLE
+apxCreateProcessW(APXHANDLE hPool, DWORD dwOptions,
+ LPAPXFNCALLBACK fnCallback,
+ LPCWSTR szUsername, LPCWSTR szPassword,
+ BOOL bLogonAsService)
+{
+ APXHANDLE hProcess;
+ LPAPXPROCESS lpProc;
+ HANDLE hUserToken = NULL;
+ /* CreateProcessAsUser is supported only on NT */
+ if (szUsername && (APX_GET_OSLEVEL() >= APX_WINVER_NT_4)) {
+ HANDLE hUser;
+ if (!LogonUserW(szUsername,
+ NULL,
+ szPassword,
+ bLogonAsService ? LOGON32_LOGON_SERVICE : LOGON32_LOGON_NETWORK,
+ LOGON32_PROVIDER_DEFAULT,
+ &hUser)) {
+ /* Logon Failed */
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return NULL;
+ }
+ if (!DuplicateTokenEx(hUser,
+ TOKEN_QUERY | TOKEN_DUPLICATE | TOKEN_ASSIGN_PRIMARY,
+ NULL,
+ SecurityImpersonation,
+ TokenPrimary,
+ &hUserToken)) {
+ CloseHandle(hUser);
+ /* Failed to duplicate the user token */
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return NULL;
+ }
+ if (!ImpersonateLoggedOnUser(hUserToken)) {
+ CloseHandle(hUser);
+ CloseHandle(hUserToken);
+ /* failed to impersonate the logged user */
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return NULL;
+ }
+ CloseHandle(hUser);
+ }
+
+ hProcess = apxHandleCreate(hPool, APXHANDLE_HAS_EVENT,
+ NULL, sizeof(APXPROCESS),
+ __apxProcessCallback);
+ if (IS_INVALID_HANDLE(hProcess))
+ return NULL;
+ hProcess->dwType = APXHANDLE_TYPE_PROCESS;
+ lpProc = APXHANDLE_DATA(hProcess);
+ lpProc->dwOptions = dwOptions;
+ lpProc->fnUserCallback = fnCallback;
+ lpProc->hUserToken = hUserToken;
+ /* set the CHAR length */
+ if (dwOptions & CREATE_UNICODE_ENVIRONMENT)
+ lpProc->chSize = sizeof(WCHAR);
+ else
+ lpProc->chSize = sizeof(CHAR);
+#if 1
+ DuplicateHandle(GetCurrentProcess(),
+ GetCurrentProcess(),
+ GetCurrentProcess(),
+ &lpProc->hCurrentProcess,
+ PROCESS_ALL_ACCESS,
+ FALSE,
+ 0);
+#else
+ lpProc->hCurrentProcess = GetCurrentProcess();
+#endif
+ lpProc->lpSD = &lpProc->bSD;
+
+ InitializeSecurityDescriptor(lpProc->lpSD, SECURITY_DESCRIPTOR_REVISION);
+ SetSecurityDescriptorDacl(lpProc->lpSD, -1, 0, 0);
+
+ lpProc->lpSA = (LPSECURITY_ATTRIBUTES)&lpProc->bSA[0];
+ lpProc->lpSA->nLength = sizeof (SECURITY_ATTRIBUTES);
+ lpProc->lpSA->lpSecurityDescriptor = lpProc->lpSD;
+ lpProc->lpSA->bInheritHandle = TRUE;
+
+ return hProcess;
+}
+
+static WCHAR _desktop_name[] =
+ {'W', 'i', 'n', 's', 't', 'a', '0', '\\', 'D', 'e', 'f', 'a', 'u', 'l', 't', 0};
+
+BOOL
+apxProcessExecute(APXHANDLE hProcess)
+{
+ LPAPXPROCESS lpProc;
+ STARTUPINFOW si;
+ DWORD id;
+ BOOL bS = FALSE;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return FALSE;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+ /* don't allow multiple execute calls on the same object */
+ if (lpProc->dwChildStatus & PROC_INITIALIZED)
+ return FALSE;
+ lpProc->bSaveHandles = TRUE;
+ SAVE_STD_HANDLES(lpProc);
+ if (!__apxProcCreateChildPipes(lpProc))
+ goto cleanup;
+ REDIRECT_STD_HANDLES(lpProc);
+ AplZeroMemory(&si, sizeof(STARTUPINFO));
+
+ si.cb = sizeof(STARTUPINFO);
+ si.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
+ si.wShowWindow = SW_HIDE;
+ /* Set the redirected handles */
+ si.hStdOutput = lpProc->hChildStdOut;
+ si.hStdError = lpProc->hChildStdErr;
+ si.hStdInput = lpProc->hChildStdInp;
+
+ if (lpProc->lpEnvironment)
+ FreeEnvironmentStringsW(lpProc->lpEnvironment);
+ lpProc->lpEnvironment = GetEnvironmentStringsW();
+
+ if (!IS_INVALID_HANDLE(lpProc->hUserToken)) {
+ si.lpDesktop = _desktop_name;
+ bS = CreateProcessAsUserW(lpProc->hUserToken,
+ lpProc->szApplicationExec,
+ lpProc->szCommandLine,
+ lpProc->lpSA,
+ NULL,
+ TRUE,
+ CREATE_SUSPENDED | CREATE_UNICODE_ENVIRONMENT | lpProc->dwOptions,
+ lpProc->lpEnvironment,
+ lpProc->szWorkingPath,
+ &si,
+ &(lpProc->stProcInfo));
+ }
+ else {
+ OutputDebugStringW(lpProc->szApplicationExec);
+ OutputDebugStringW(lpProc->szCommandLine);
+
+ bS = CreateProcessW(lpProc->szApplicationExec,
+ lpProc->szCommandLine,
+ lpProc->lpSA,
+ NULL,
+ TRUE,
+ CREATE_SUSPENDED | CREATE_UNICODE_ENVIRONMENT | lpProc->dwOptions,
+ lpProc->lpEnvironment,
+ lpProc->szWorkingPath,
+ &si,
+ &(lpProc->stProcInfo));
+ }
+ /* Close unused sides of pipes */
+ SAFE_CLOSE_HANDLE(lpProc->hChildStdInp);
+ SAFE_CLOSE_HANDLE(lpProc->hChildStdOut);
+ SAFE_CLOSE_HANDLE(lpProc->hChildStdErr);
+ if (!bS)
+ goto cleanup;
+ /* Set the running flag */
+ lpProc->dwChildStatus |= (CHILD_RUNNING | PROC_INITIALIZED);
+
+ lpProc->hWorkerThreads[0] = CreateThread(NULL, 0, __apxProcStdoutThread,
+ hProcess, 0, &id);
+ lpProc->hWorkerThreads[1] = CreateThread(NULL, 0, __apxProcStderrThread,
+ hProcess, 0, &id);
+ ResumeThread(lpProc->stProcInfo.hThread);
+ lpProc->hWorkerThreads[2] = CreateThread(NULL, 0, __apxProcWorkerThread,
+ hProcess, 0, &id);
+
+ SAFE_CLOSE_HANDLE(lpProc->stProcInfo.hThread);
+ /* Close child handles first */
+ return TRUE;
+cleanup:
+ /* Close parent side of the pipes */
+ SAFE_CLOSE_HANDLE(lpProc->hChildInpWr);
+ SAFE_CLOSE_HANDLE(lpProc->hChildOutRd);
+ SAFE_CLOSE_HANDLE(lpProc->hChildErrRd);
+
+ return FALSE;
+}
+
+BOOL
+apxProcessSetExecutableA(APXHANDLE hProcess, LPCSTR szName)
+{
+ LPAPXPROCESS lpProc;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return FALSE;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+ apxFree(lpProc->szApplicationExec);
+ lpProc->szApplicationExec = apxPoolWStrdupA(hProcess->hPool, szName);
+ OutputDebugStringW(lpProc->szApplicationExec);
+ return lpProc->szApplicationExec != NULL;
+}
+
+BOOL
+apxProcessSetExecutableW(APXHANDLE hProcess, LPCWSTR szName)
+{
+ LPAPXPROCESS lpProc;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return FALSE;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+ apxFree(lpProc->szApplicationExec);
+ lpProc->szApplicationExec = apxPoolStrdupW(hProcess->hPool, szName);
+ OutputDebugStringW(lpProc->szApplicationExec);
+ return lpProc->szApplicationExec != NULL;
+}
+
+BOOL
+apxProcessSetCommandLineA(APXHANDLE hProcess, LPCSTR szCmdline)
+{
+ LPAPXPROCESS lpProc;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return FALSE;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+ apxFree(lpProc->szCommandLine);
+ lpProc->szCommandLine = apxPoolWStrdupA(hProcess->hPool, szCmdline);
+
+ return lpProc->szCommandLine != NULL;
+}
+
+BOOL
+apxProcessSetCommandLineW(APXHANDLE hProcess, LPCWSTR szCmdline)
+{
+ LPAPXPROCESS lpProc;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return FALSE;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+ apxFree(lpProc->szCommandLine);
+ lpProc->szCommandLine = apxPoolStrdupW(hProcess->hPool, szCmdline);
+
+ return lpProc->szCommandLine != NULL;
+}
+
+BOOL
+apxProcessSetWorkingPathA(APXHANDLE hProcess, LPCSTR szPath)
+{
+ LPAPXPROCESS lpProc;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return FALSE;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+ apxFree(lpProc->szWorkingPath);
+ lpProc->szWorkingPath = apxPoolWStrdupA(hProcess->hPool, szPath);
+
+ return lpProc->szWorkingPath != NULL;
+}
+
+BOOL
+apxProcessSetWorkingPathW(APXHANDLE hProcess, LPCWSTR szPath)
+{
+ LPAPXPROCESS lpProc;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return FALSE;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+ apxFree(lpProc->szWorkingPath);
+ if (!szPath) {
+ /* Clear the WorkingPath */
+ lpProc->szWorkingPath = NULL;
+ return TRUE;
+ }
+ lpProc->szWorkingPath = apxPoolWStrdupW(hProcess->hPool, szPath);
+
+ return lpProc->szWorkingPath != NULL;
+}
+
+DWORD
+apxProcessPutcA(APXHANDLE hProcess, INT ch)
+{
+ LPAPXPROCESS lpProc;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return 0;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+ return __apxProcessPutc(lpProc, ch, sizeof(CHAR));
+}
+
+DWORD
+apxProcessPutcW(APXHANDLE hProcess, INT ch)
+{
+ LPAPXPROCESS lpProc;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return 0;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+ return __apxProcessPutc(lpProc, ch, sizeof(WCHAR));
+}
+
+DWORD
+apxProcessPutsA(APXHANDLE hProcess, LPCSTR szString)
+{
+ LPAPXPROCESS lpProc;
+ DWORD len;
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS || !szString)
+ return 0;
+ lpProc = APXHANDLE_DATA(hProcess);
+ len = lstrlenA(szString);
+ len = __apxProcessWrite(lpProc, szString, len);
+ if (len) {
+ FlushFileBuffers(lpProc->hChildInpWr);
+ return len;
+ }
+ else
+ return 0;
+}
+
+DWORD
+apxProcessPutsW(APXHANDLE hProcess, LPCWSTR szString)
+{
+ LPAPXPROCESS lpProc;
+ DWORD len;
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS || !szString)
+ return 0;
+ lpProc = APXHANDLE_DATA(hProcess);
+ len = lstrlenW(szString);
+ len = __apxProcessWrite(lpProc, szString, len * 2);
+ if (len) {
+ FlushFileBuffers(lpProc->hChildInpWr);
+ return (len / sizeof(WCHAR));
+ }
+ else
+ return 0;
+}
+
+DWORD
+apxProcessWrite(APXHANDLE hProcess, LPCVOID lpData, DWORD dwLen)
+{
+ LPAPXPROCESS lpProc;
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return 0;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+
+ return __apxProcessWrite(lpProc, lpData, dwLen);
+}
+
+BOOL
+apxProcessFlushStdin(APXHANDLE hProcess)
+{
+ LPAPXPROCESS lpProc;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return FALSE;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+
+ if (lpProc->dwChildStatus & CHILD_RUNNING) {
+ return FlushFileBuffers(lpProc->hChildInpWr);
+ }
+
+ return FALSE;
+}
+
+VOID
+apxProcessCloseInputStream(APXHANDLE hProcess)
+{
+ if (hProcess->dwType == APXHANDLE_TYPE_PROCESS) {
+ LPAPXPROCESS lpProc = APXHANDLE_DATA(hProcess);
+ if (lpProc->dwChildStatus & CHILD_RUNNING)
+ SAFE_CLOSE_HANDLE(lpProc->hChildInpWr);
+ }
+}
+
+DWORD
+apxProcessWait(APXHANDLE hProcess, DWORD dwMilliseconds, BOOL bKill)
+{
+ LPAPXPROCESS lpProc;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return WAIT_ABANDONED;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+
+ if (lpProc->dwChildStatus & CHILD_RUNNING) {
+ DWORD rv = WaitForMultipleObjects(3, lpProc->hWorkerThreads,
+ TRUE, dwMilliseconds);
+ if (rv == WAIT_TIMEOUT && bKill) {
+ __apxProcessCallback(hProcess, WM_CLOSE, 0, 0);
+ }
+ return rv;
+ }
+ else
+ return WAIT_OBJECT_0;
+}
+
+BOOL
+apxProcessRunning(APXHANDLE hProcess)
+{
+ LPAPXPROCESS lpProc;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return FALSE;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+
+ return (lpProc->dwChildStatus & CHILD_RUNNING);
+}
+
+DWORD
+apxProcessGetPid(APXHANDLE hProcess)
+{
+ LPAPXPROCESS lpProc;
+
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return 0;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+
+ return lpProc->stProcInfo.dwProcessId;
+}
+
+static LPWSTR __apxStrQuote(LPWSTR lpDest, LPCWSTR szSrc)
+{
+ LPWSTR p;
+ BOOL space = FALSE, quote = FALSE;
+
+ /* Find if string has embeded spaces, add quotes only if no quotes found
+ */
+ for (p = (LPWSTR)szSrc; *p; p++) {
+ if (*p == L' ' || *p == '\t') {
+ space = TRUE;
+ } else if (*p == L'"') {
+ quote = TRUE;
+ }
+ }
+ p = lpDest;
+ if (space && !quote) *p++ = L'"';
+ while (*szSrc) {
+ *p++ = *szSrc++;
+ }
+ if (space && !quote) *p++ = L'"';
+ return p;
+}
+
+BOOL
+apxProcessSetCommandArgsW(APXHANDLE hProcess, LPCWSTR szTitle,
+ DWORD dwArgc, LPCWSTR *lpArgs)
+{
+ LPAPXPROCESS lpProc;
+ DWORD i, l = 0;
+ LPWSTR p;
+ if (hProcess->dwType != APXHANDLE_TYPE_PROCESS)
+ return FALSE;
+
+ lpProc = APXHANDLE_DATA(hProcess);
+ apxFree(lpProc->szCommandLine);
+
+ l = lstrlenW(szTitle) + 3;
+ for (i = 0; i < dwArgc; i++) {
+ int q = 0;
+ l += (lstrlenW(lpArgs[i]) + 3);
+ l += q;
+ }
+ p = lpProc->szCommandLine = apxPoolAlloc(hProcess->hPool, l * sizeof(WCHAR));
+ p = __apxStrQuote(p, szTitle);
+ for (i = 0; i < dwArgc; i++) {
+ *p++ = L' ';
+ p = __apxStrQuote(p, lpArgs[i]);
+ }
+ *p = L'\0';
+ OutputDebugStringW(lpProc->szCommandLine);
+ return lpProc->szCommandLine != NULL;
+}
+
Deleted: tags/jbosssvc/2.0.8/procrun/src/service.c
===================================================================
--- trunk/utils/windows/native/service/procrun/src/service.c 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/src/service.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,565 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#include "apxwin.h"
-#include "private.h"
-
-#define SAFE_CLOSE_SCH(h) \
- if ((h) != NULL && (h) != INVALID_HANDLE_VALUE) { \
- CloseServiceHandle((h)); \
- (h) = NULL; \
- }
-
-#define CHANGE_SERVICE(h, b, u, p, d) \
- ChangeServiceConfigW(h, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, \
- SERVICE_NO_CHANGE,b,NULL,NULL,NULL, \
- u,p,d)
-
-typedef struct APXSERVICE {
- /* Are we a service manager or we are the service itself */
- BOOL bManagerMode;
- /* Handle to the current service */
- SC_HANDLE hService;
- /* Handle of the Service manager */
- SC_HANDLE hManager;
- APXSERVENTRY stServiceEntry;
-
-} APXSERVICE, *LPAPXSERVICE;
-
-static BOOL __apxIsValidServiceName(LPCWSTR szServiceName)
-{
- do {
- if (!IsCharAlphaNumericW(*szServiceName)) {
- apxDisplayError(TRUE, NULL, 0, "NonAlpha %d", *szServiceName);
- return FALSE;
- }
- } while( *(++szServiceName));
- return TRUE;
-}
-
-static BOOL __apxServiceCallback(APXHANDLE hObject, UINT uMsg,
- WPARAM wParam, LPARAM lParam)
-{
- LPAPXSERVICE lpService;
-
- lpService = APXHANDLE_DATA(hObject);
- switch (uMsg) {
- case WM_CLOSE:
- apxFree(lpService->stServiceEntry.lpConfig);
- lpService->stServiceEntry.lpConfig = NULL;
- SAFE_CLOSE_SCH(lpService->hService);
- SAFE_CLOSE_SCH(lpService->hManager);
- break;
- default:
- break;
- }
- return TRUE;
-}
-
-APXHANDLE
-apxCreateService(APXHANDLE hPool, DWORD dwOptions, BOOL bManagerMode)
-{
- APXHANDLE hService;
- LPAPXSERVICE lpService;
- SC_HANDLE hManager;
-
- if (!(hManager = OpenSCManager(NULL, NULL, dwOptions))) {
- apxLogWrite(APXLOG_MARK_SYSERR);
- return NULL;
- }
- hService = apxHandleCreate(hPool, 0,
- NULL, sizeof(APXSERVICE),
- __apxServiceCallback);
- if (IS_INVALID_HANDLE(hService)) {
- apxLogWrite(APXLOG_MARK_ERROR "Failed to Create Handle for Service");
- return NULL;
- }
- hService->dwType = APXHANDLE_TYPE_SERVICE;
- lpService = APXHANDLE_DATA(hService);
- lpService->hManager = hManager;
- lpService->bManagerMode = bManagerMode;
-
- return hService;
-}
-
-BOOL
-apxServiceOpen(APXHANDLE hService, LPCWSTR szServiceName, DWORD dwOptions)
-{
- LPAPXSERVICE lpService;
- DWORD dwNeeded;
-
- if (hService->dwType != APXHANDLE_TYPE_SERVICE)
- return FALSE;
-
- lpService = APXHANDLE_DATA(hService);
- /* Manager mode cannot handle services */
- if (lpService->bManagerMode)
- return FALSE;
-
- /* Close any previous instance
- * Same handle can manage multiple services
- */
- SAFE_CLOSE_SCH(lpService->hService);
- *lpService->stServiceEntry.szServiceDescription = L'\0';
- *lpService->stServiceEntry.szObjectName = L'\0';
- apxFree(lpService->stServiceEntry.lpConfig);
- lpService->stServiceEntry.lpConfig = NULL;
- /* Open the service */
- lpService->hService = OpenServiceW(lpService->hManager,
- szServiceName,
- dwOptions);
-
- if (IS_INVALID_HANDLE(lpService->hService)) {
- apxLogWrite(APXLOG_MARK_SYSERR);
- return FALSE;
- }
- lstrcpyW(lpService->stServiceEntry.szServiceName, szServiceName);
- if (!apxGetServiceDescriptionW(szServiceName,
- lpService->stServiceEntry.szServiceDescription,
- SIZ_DESLEN)) {
- apxLogWrite(APXLOG_MARK_WARN "Failed obtaing service description");
- lpService->stServiceEntry.szServiceDescription[0] = L'\0';
- }
- if (!apxGetServiceUserW(szServiceName,
- lpService->stServiceEntry.szObjectName,
- SIZ_RESLEN)) {
- apxLogWrite(APXLOG_MARK_WARN "Failed obtaing service user name");
- lpService->stServiceEntry.szObjectName[0] = L'\0';
- }
- if (!QueryServiceConfigW(lpService->hService, NULL, 0, &dwNeeded)) {
- if (GetLastError() != ERROR_INSUFFICIENT_BUFFER)
- apxLogWrite(APXLOG_MARK_SYSERR);
- }
- /* TODO: Check GetLastError ERROR_INSUFFICIENT_BUFFER */
- lpService->stServiceEntry.lpConfig = (LPQUERY_SERVICE_CONFIGW)apxPoolAlloc(hService->hPool,
- dwNeeded);
- return QueryServiceConfigW(lpService->hService,
- lpService->stServiceEntry.lpConfig,
- dwNeeded, &dwNeeded);
-}
-
-LPAPXSERVENTRY
-apxServiceEntry(APXHANDLE hService, BOOL bRequeryStatus)
-{
- LPAPXSERVICE lpService;
-
- if (hService->dwType != APXHANDLE_TYPE_SERVICE)
- return NULL;
-
- lpService = APXHANDLE_DATA(hService);
- /* Manager mode cannot handle services */
- if (lpService->bManagerMode)
- return NULL;
-
- if (bRequeryStatus && !QueryServiceStatus(lpService->hService,
- &(lpService->stServiceEntry.stServiceStatus))) {
- apxLogWrite(APXLOG_MARK_SYSERR);
- return NULL;
- }
-
- return &lpService->stServiceEntry;
-}
-
-/* Set the service names etc...
- * If the ImagePath contains a space, it must be quoted
- */
-BOOL
-apxServiceSetNames(APXHANDLE hService,
- LPCWSTR szImagePath,
- LPCWSTR szDisplayName,
- LPCWSTR szDescription,
- LPCWSTR szUsername,
- LPCWSTR szPassword)
-{
- LPAPXSERVICE lpService;
-
- if (hService->dwType != APXHANDLE_TYPE_SERVICE)
- return FALSE;
-
- lpService = APXHANDLE_DATA(hService);
- /* Manager mode cannot handle services */
- if (lpService->bManagerMode)
- return FALSE;
- /* Check if the ServiceOpen has been called */
- if (IS_INVALID_HANDLE(lpService->hService))
- return FALSE;
- if (!CHANGE_SERVICE(lpService->hService,
- szImagePath,
- szUsername,
- szPassword,
- szDisplayName)) {
- apxLogWrite(APXLOG_MARK_SYSERR);
- return FALSE;
- }
- if (szDescription) {
- return apxSetServiceDescriptionW(lpService->stServiceEntry.szServiceName,
- szDescription);
- }
- return TRUE;
-}
-
-BOOL
-apxServiceSetOptions(APXHANDLE hService,
- DWORD dwServiceType,
- DWORD dwStartType,
- DWORD dwErrorControl)
-{
- LPAPXSERVICE lpService;
-
- if (hService->dwType != APXHANDLE_TYPE_SERVICE)
- return FALSE;
-
- lpService = APXHANDLE_DATA(hService);
- /* Manager mode cannot handle services */
- if (lpService->bManagerMode)
- return FALSE;
- /* Check if the ServixeOpen has been called */
- if (IS_INVALID_HANDLE(lpService->hService))
- return FALSE;
- return ChangeServiceConfig(lpService->hService, dwServiceType,
- dwStartType, dwErrorControl,
- NULL, NULL, NULL, NULL, NULL,
- NULL, NULL);
-}
-
-BOOL
-apxServiceControl(APXHANDLE hService, DWORD dwControl, UINT uMsg,
- LPAPXFNCALLBACK fnControlCallback,
- LPVOID lpCbData)
-{
- LPAPXSERVICE lpService;
- SERVICE_STATUS stStatus;
- DWORD dwPending = 0;
- DWORD dwState = 0;
- DWORD dwTick = 0;
- DWORD dwWait, dwCheck, dwStart;
- BOOL bStatus;
-
- if (hService->dwType != APXHANDLE_TYPE_SERVICE)
- return FALSE;
-
- lpService = APXHANDLE_DATA(hService);
- /* Manager mode cannot handle services */
- if (lpService->bManagerMode)
- return FALSE;
- /* Check if the ServixeOpen has been called */
- if (IS_INVALID_HANDLE(lpService->hService))
- return FALSE;
- switch (dwControl) {
- case SERVICE_CONTROL_CONTINUE:
- dwPending = SERVICE_START_PENDING;
- dwState = SERVICE_RUNNING;
- break;
- case SERVICE_CONTROL_STOP:
- dwPending = SERVICE_STOP_PENDING;
- dwState = SERVICE_STOPPED;
- break;
- case SERVICE_CONTROL_PAUSE:
- dwPending = SERVICE_PAUSE_PENDING;
- dwState = SERVICE_PAUSED;
- break;
- default:
- break;
- }
- /* user defined controls */
- if (dwControl > 127 && dwControl < 224) {
- /* 128 ... 159 start signals
- * 160 ... 191 stop signals
- * 192 ... 223 pause signals
- */
- switch (dwControl & 0xE0) {
- case 0x80:
- case 0x90:
- dwPending = SERVICE_START_PENDING;
- dwState = SERVICE_RUNNING;
- break;
- case 0xA0:
- case 0xB0:
- dwPending = SERVICE_STOP_PENDING;
- dwState = SERVICE_STOPPED;
- break;
- case 0xC0:
- case 0xD0:
- dwPending = SERVICE_PAUSE_PENDING;
- dwState = SERVICE_PAUSED;
- break;
- default:
- break;
- }
- }
- if (!dwPending && !dwState)
- return FALSE;
- /* Now lets control */
- if (!QueryServiceStatus(lpService->hService, &stStatus)) {
- apxLogWrite(APXLOG_MARK_SYSERR);
- return FALSE;
- }
- /* signal that we are about to control the service */
- if (fnControlCallback)
- (*fnControlCallback)(lpCbData, uMsg, (WPARAM)1, (LPARAM)dwState);
- if (dwControl == SERVICE_CONTROL_CONTINUE &&
- stStatus.dwCurrentState != SERVICE_PAUSED)
- bStatus = StartService(lpService->hService, 0, NULL);
- else
- bStatus = ControlService(lpService->hService, dwControl, &stStatus);
- dwStart = GetTickCount();
- dwCheck = stStatus.dwCheckPoint;
- if (bStatus) {
- Sleep(100); /* Initial Sleep period */
- while (QueryServiceStatus(lpService->hService, &stStatus)) {
- if (stStatus.dwCurrentState == dwPending) {
- /* Do not wait longer than the wait hint. A good interval is
- * one tenth the wait hint, but no less than 1 second and no
- * more than 10 seconds.
- */
- dwWait = stStatus.dwWaitHint / 10;
-
- if( dwWait < 1000 )
- dwWait = 1000;
- else if ( dwWait > 10000 )
- dwWait = 10000;
- /* Signal to the callback that we are pending
- * break if callback returns false.
- */
- if (fnControlCallback) {
- if (!(*fnControlCallback)(lpCbData, uMsg, (WPARAM)2,
- (LPARAM)dwTick++))
- break;
- }
- Sleep(dwWait);
- if (stStatus.dwCheckPoint > dwCheck) {
- /* The service is making progress. */
- dwStart = GetTickCount();
- dwCheck = stStatus.dwCheckPoint;
- }
- else {
- if(GetTickCount() - dwStart > stStatus.dwWaitHint) {
- /* No progress made within the wait hint */
- break;
- }
- }
- }
- else
- break;
- }
- }
- /* signal that we are done with controling the service */
- if (fnControlCallback)
- (*fnControlCallback)(lpCbData, uMsg, (WPARAM)3, (LPARAM)0);
- /* Check if we are in the desired state */
- Sleep(1000);
-
- if (QueryServiceStatus(lpService->hService, &stStatus)) {
- if (fnControlCallback)
- (*fnControlCallback)(lpCbData, uMsg, (WPARAM)4,
- (LPARAM)&stStatus);
- if (stStatus.dwCurrentState == dwState)
- return TRUE;
-
- }
-
- return FALSE;
-}
-
-BOOL
-apxServiceInstall(APXHANDLE hService, LPCWSTR szServiceName,
- LPCWSTR szDisplayName, LPCWSTR szImagePath,
- LPCWSTR lpDependencies, DWORD dwServiceType,
- DWORD dwStartType)
-{
- LPAPXSERVICE lpService;
-
- if (hService->dwType != APXHANDLE_TYPE_SERVICE)
- return FALSE;
-
- lpService = APXHANDLE_DATA(hService);
- /* Manager mode cannot handle services */
- if (lpService->bManagerMode)
- return FALSE;
- if (IS_INVALID_HANDLE(lpService->hManager))
- return FALSE;
- if (!__apxIsValidServiceName(szServiceName))
- return FALSE;
- /* Close any previous instance
- * Same handle can install multiple services
- */
- SAFE_CLOSE_SCH(lpService->hService);
-
- apxFree(lpService->stServiceEntry.lpConfig);
- lpService->stServiceEntry.lpConfig = NULL;
- AplZeroMemory(&lpService->stServiceEntry, sizeof(APXSERVENTRY));
-
- lpService->hService = CreateServiceW(lpService->hManager,
- szServiceName,
- szDisplayName,
- SERVICE_ALL_ACCESS,
- dwServiceType,
- dwStartType,
- SERVICE_ERROR_NORMAL,
- szImagePath,
- NULL,
- NULL,
- lpDependencies,
- NULL,
- NULL);
-
- if (IS_INVALID_HANDLE(lpService->hService)) {
- apxLogWrite(APXLOG_MARK_SYSERR);
-
- ErrorMessage(NULL, FALSE);
- return FALSE;
- }
- else {
- lstrcpyW(lpService->stServiceEntry.szServiceName, szServiceName);
- lpService->stServiceEntry.dwStart = dwStartType;
- return TRUE;
- }
-}
-
-BOOL
-apxServiceDelete(APXHANDLE hService)
-{
- LPAPXSERVICE lpService;
- SERVICE_STATUS stStatus;
- if (hService->dwType != APXHANDLE_TYPE_SERVICE)
- return FALSE;
-
- lpService = APXHANDLE_DATA(hService);
- /* Manager mode cannot handle services */
- if (lpService->bManagerMode)
- return FALSE;
- /* Check if the ServiceOpen has been called */
- if (IS_INVALID_HANDLE(lpService->hService))
- return FALSE;
- if (QueryServiceStatus(lpService->hService, &stStatus)) {
- BOOL rv;
- if (stStatus.dwCurrentState != SERVICE_STOPPED)
- apxServiceControl(hService, SERVICE_CONTROL_STOP, 0, NULL, NULL);
- rv = DeleteService(lpService->hService);
- SAFE_CLOSE_SCH(lpService->hService);
- SAFE_CLOSE_SCH(lpService->hManager);
-
- return rv;
- }
- return FALSE;
-}
-
-/* Browse the services */
-DWORD
-apxServiceBrowse(APXHANDLE hService,
- LPCWSTR szIncludeNamePattern,
- LPCWSTR szIncludeImagePattern,
- LPCWSTR szExcludeNamePattern,
- LPCWSTR szExcludeImagePattern,
- UINT uMsg,
- LPAPXFNCALLBACK fnDisplayCallback,
- LPVOID lpCbData)
-{
- DWORD nFound = 0;
- APXREGENUM stEnum;
- LPAPXSERVICE lpService;
- SC_LOCK hLock;
- if (hService->dwType != APXHANDLE_TYPE_SERVICE || !fnDisplayCallback)
- return 0;
-
- lpService = APXHANDLE_DATA(hService);
- /* Only the manager mode can browse services */
- if (!lpService->bManagerMode ||
- IS_INVALID_HANDLE(lpService->hManager))
- return 0;
- hLock = LockServiceDatabase(lpService->hManager);
- if (IS_INVALID_HANDLE(hLock)) {
- apxLogWrite(APXLOG_MARK_SYSERR);
-
- ErrorMessage(NULL, FALSE);
- return 0;
- }
- AplZeroMemory(&stEnum, sizeof(APXREGENUM));
-
- while (TRUE) {
- APXSERVENTRY stEntry;
- BOOL rv;
- AplZeroMemory(&stEntry, sizeof(APXSERVENTRY));
- rv = apxRegistryEnumServices(&stEnum, &stEntry);
-
- if (rv) {
- INT fm = -1;
- SC_HANDLE hSrv = NULL;
- DWORD dwNeeded = 0;
- hSrv = OpenServiceW(lpService->hManager,
- stEntry.szServiceName,
- GENERIC_READ);
- if (!IS_INVALID_HANDLE(hSrv)) {
- QueryServiceConfigW(hSrv, NULL, 0, &dwNeeded);
- stEntry.lpConfig = (LPQUERY_SERVICE_CONFIGW)apxPoolAlloc(hService->hPool,
- dwNeeded);
- /* Call the QueryServiceConfig againg with allocated config */
- if (QueryServiceConfigW(hSrv, stEntry.lpConfig, dwNeeded, &dwNeeded)) {
- /* Make that customizable so that kernel mode drivers can be
- * displayed and maintained. For now skip the
- * filesystem and device drivers.
- * XXX: Do we need that customizable after all?
- */
- if ((stEntry.lpConfig->dwServiceType &
- ~SERVICE_INTERACTIVE_PROCESS) & SERVICE_WIN32)
- fm = 0;
-
- if (!fm && szIncludeNamePattern) {
- fm = apxMultiStrMatchW(stEntry.szServiceName,
- szIncludeNamePattern, L';', TRUE);
- }
- if (!fm && szExcludeNamePattern) {
- fm = !apxMultiStrMatchW(stEntry.szServiceName,
- szExcludeNamePattern, L';', TRUE);
- }
- if (!fm && szIncludeImagePattern) {
- fm = apxMultiStrMatchW(stEntry.lpConfig->lpBinaryPathName,
- szIncludeImagePattern, L';', TRUE);
- }
- if (!fm && szExcludeImagePattern) {
- fm = !apxMultiStrMatchW(stEntry.szServiceName,
- szExcludeImagePattern, L';', TRUE);
- }
- if (!fm) {
- QueryServiceStatus(hSrv, &(stEntry.stServiceStatus));
- /* WIN2K + extended service info */
- if (_st_apx_oslevel >= 4) {
- DWORD dwNeed;
- QueryServiceStatusEx(hSrv, SC_STATUS_PROCESS_INFO,
- (LPBYTE)(&(stEntry.stStatusProcess)),
- sizeof(SERVICE_STATUS_PROCESS),
- &dwNeed);
- }
- /* finaly call the provided callback */
- rv = (*fnDisplayCallback)(lpCbData, uMsg,
- (WPARAM)&stEntry,
- (LPARAM)nFound++);
- }
- }
- /* release the skipped service config */
- if (fm) {
- apxFree(stEntry.lpConfig);
- }
- }
- SAFE_CLOSE_SCH(hSrv);
- }
- if (!rv)
- break;
- }
-
- UnlockServiceDatabase(hLock);
- return nFound;
-}
Copied: tags/jbosssvc/2.0.8/procrun/src/service.c (from rev 2689, trunk/utils/windows/native/service/procrun/src/service.c)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/src/service.c (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/src/service.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,669 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "apxwin.h"
+#include "private.h"
+
+#define SAFE_CLOSE_SCH(h) \
+ if ((h) != NULL && (h) != INVALID_HANDLE_VALUE) { \
+ CloseServiceHandle((h)); \
+ (h) = NULL; \
+ }
+
+#define CHANGE_SERVICE(h, b, u, p, d) \
+ ChangeServiceConfigW(h, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, \
+ SERVICE_NO_CHANGE,b,NULL,NULL,NULL, \
+ u,p,d)
+
+typedef struct APXSERVICE {
+ /* Are we a service manager or we are the service itself */
+ BOOL bManagerMode;
+ /* Handle to the current service */
+ SC_HANDLE hService;
+ /* Handle of the Service manager */
+ SC_HANDLE hManager;
+ APXSERVENTRY stServiceEntry;
+
+} APXSERVICE, *LPAPXSERVICE;
+
+static BOOL __apxIsValidServiceName(LPCWSTR szServiceName)
+{
+ do {
+ if (!IsCharAlphaNumericW(*szServiceName)) {
+ apxDisplayError(FALSE, NULL, 0, "NonAlpha %d", *szServiceName);
+ return FALSE;
+ }
+ } while( *(++szServiceName));
+ return TRUE;
+}
+
+static BOOL __apxServiceCallback(APXHANDLE hObject, UINT uMsg,
+ WPARAM wParam, LPARAM lParam)
+{
+ LPAPXSERVICE lpService;
+
+ lpService = APXHANDLE_DATA(hObject);
+ switch (uMsg) {
+ case WM_CLOSE:
+ apxFree(lpService->stServiceEntry.lpConfig);
+ lpService->stServiceEntry.lpConfig = NULL;
+ SAFE_CLOSE_SCH(lpService->hService);
+ SAFE_CLOSE_SCH(lpService->hManager);
+ break;
+ default:
+ break;
+ }
+ return TRUE;
+}
+
+APXHANDLE
+apxCreateService(APXHANDLE hPool, DWORD dwOptions, BOOL bManagerMode)
+{
+ APXHANDLE hService;
+ LPAPXSERVICE lpService;
+ SC_HANDLE hManager;
+
+ if (!(hManager = OpenSCManager(NULL, NULL, dwOptions))) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return NULL;
+ }
+ hService = apxHandleCreate(hPool, 0,
+ NULL, sizeof(APXSERVICE),
+ __apxServiceCallback);
+ if (IS_INVALID_HANDLE(hService)) {
+ apxLogWrite(APXLOG_MARK_ERROR "Failed to Create Handle for Service");
+ return NULL;
+ }
+ hService->dwType = APXHANDLE_TYPE_SERVICE;
+ lpService = APXHANDLE_DATA(hService);
+ lpService->hManager = hManager;
+ lpService->bManagerMode = bManagerMode;
+
+ return hService;
+}
+
+BOOL
+apxServiceOpen(APXHANDLE hService, LPCWSTR szServiceName, DWORD dwOptions)
+{
+ LPAPXSERVICE lpService;
+ DWORD dwNeeded;
+
+ if (hService->dwType != APXHANDLE_TYPE_SERVICE)
+ return FALSE;
+
+ lpService = APXHANDLE_DATA(hService);
+ /* Manager mode cannot handle services */
+ if (lpService->bManagerMode)
+ return FALSE;
+
+ /* Close any previous instance
+ * Same handle can manage multiple services
+ */
+ SAFE_CLOSE_SCH(lpService->hService);
+ *lpService->stServiceEntry.szServiceDescription = L'\0';
+ *lpService->stServiceEntry.szObjectName = L'\0';
+ apxFree(lpService->stServiceEntry.lpConfig);
+ lpService->stServiceEntry.lpConfig = NULL;
+ /* Open the service */
+ lpService->hService = OpenServiceW(lpService->hManager,
+ szServiceName,
+ dwOptions);
+
+ if (IS_INVALID_HANDLE(lpService->hService)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return FALSE;
+ }
+ lstrlcpyW(lpService->stServiceEntry.szServiceName, SIZ_RESLEN, szServiceName);
+ if (!apxGetServiceDescriptionW(szServiceName,
+ lpService->stServiceEntry.szServiceDescription,
+ SIZ_DESLEN)) {
+ apxLogWrite(APXLOG_MARK_WARN "Failed to obtain service description");
+ lpService->stServiceEntry.szServiceDescription[0] = L'\0';
+ }
+ if (!apxGetServiceUserW(szServiceName,
+ lpService->stServiceEntry.szObjectName,
+ SIZ_RESLEN)) {
+ apxLogWrite(APXLOG_MARK_WARN "Failed to obtain service user name");
+ lpService->stServiceEntry.szObjectName[0] = L'\0';
+ }
+ if (!QueryServiceConfigW(lpService->hService, NULL, 0, &dwNeeded)) {
+ if (GetLastError() != ERROR_INSUFFICIENT_BUFFER)
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ }
+ /* TODO: Check GetLastError ERROR_INSUFFICIENT_BUFFER */
+ lpService->stServiceEntry.lpConfig = (LPQUERY_SERVICE_CONFIGW)apxPoolAlloc(hService->hPool,
+ dwNeeded);
+ return QueryServiceConfigW(lpService->hService,
+ lpService->stServiceEntry.lpConfig,
+ dwNeeded, &dwNeeded);
+}
+
+LPAPXSERVENTRY
+apxServiceEntry(APXHANDLE hService, BOOL bRequeryStatus)
+{
+ LPAPXSERVICE lpService;
+
+ if (hService->dwType != APXHANDLE_TYPE_SERVICE)
+ return NULL;
+
+ lpService = APXHANDLE_DATA(hService);
+ /* Manager mode cannot handle services */
+ if (lpService->bManagerMode)
+ return NULL;
+
+ if (bRequeryStatus && !QueryServiceStatus(lpService->hService,
+ &(lpService->stServiceEntry.stServiceStatus))) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return NULL;
+ }
+
+ return &lpService->stServiceEntry;
+}
+
+/* Set the service names etc...
+ * If the ImagePath contains a space, it must be quoted
+ */
+BOOL
+apxServiceSetNames(APXHANDLE hService,
+ LPCWSTR szImagePath,
+ LPCWSTR szDisplayName,
+ LPCWSTR szDescription,
+ LPCWSTR szUsername,
+ LPCWSTR szPassword)
+{
+ LPAPXSERVICE lpService;
+
+ if (hService->dwType != APXHANDLE_TYPE_SERVICE)
+ return FALSE;
+
+ lpService = APXHANDLE_DATA(hService);
+ /* Manager mode cannot handle services */
+ if (lpService->bManagerMode)
+ return FALSE;
+ /* Check if the ServiceOpen has been called */
+ if (IS_INVALID_HANDLE(lpService->hService))
+ return FALSE;
+ if (!CHANGE_SERVICE(lpService->hService,
+ szImagePath,
+ szUsername,
+ szPassword,
+ szDisplayName)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return FALSE;
+ }
+ if (szDescription) {
+ return apxSetServiceDescriptionW(lpService->stServiceEntry.szServiceName,
+ szDescription);
+ }
+ return TRUE;
+}
+
+BOOL
+apxServiceSetOptions(APXHANDLE hService,
+ DWORD dwServiceType,
+ DWORD dwStartType,
+ DWORD dwErrorControl)
+{
+ LPAPXSERVICE lpService;
+
+ if (hService->dwType != APXHANDLE_TYPE_SERVICE)
+ return FALSE;
+
+ lpService = APXHANDLE_DATA(hService);
+ /* Manager mode cannot handle services */
+ if (lpService->bManagerMode)
+ return FALSE;
+ /* Check if the ServixeOpen has been called */
+ if (IS_INVALID_HANDLE(lpService->hService))
+ return FALSE;
+ return ChangeServiceConfig(lpService->hService, dwServiceType,
+ dwStartType, dwErrorControl,
+ NULL, NULL, NULL, NULL, NULL,
+ NULL, NULL);
+}
+
+static BOOL
+__apxStopDependentServices(LPAPXSERVICE lpService)
+{
+ DWORD i;
+ DWORD dwBytesNeeded;
+ DWORD dwCount;
+
+ LPENUM_SERVICE_STATUS lpDependencies = NULL;
+ ENUM_SERVICE_STATUS ess;
+ SC_HANDLE hDepService;
+ SERVICE_STATUS_PROCESS ssp;
+
+ DWORD dwStartTime = GetTickCount();
+ /* Use the 30-second time-out */
+ DWORD dwTimeout = 30000;
+
+ /* Pass a zero-length buffer to get the required buffer size.
+ */
+ if (EnumDependentServices(lpService->hService,
+ SERVICE_ACTIVE,
+ lpDependencies, 0,
+ &dwBytesNeeded,
+ &dwCount)) {
+ /* If the Enum call succeeds, then there are no dependent
+ * services, so do nothing.
+ */
+ return TRUE;
+ }
+ else {
+ if (GetLastError() != ERROR_MORE_DATA)
+ return FALSE; // Unexpected error
+
+ /* Allocate a buffer for the dependencies.
+ */
+ lpDependencies = (LPENUM_SERVICE_STATUS) HeapAlloc(GetProcessHeap(),
+ HEAP_ZERO_MEMORY,
+ dwBytesNeeded);
+ if (!lpDependencies)
+ return FALSE;
+
+ __try {
+ /* Enumerate the dependencies. */
+ if (!EnumDependentServices(lpService->hService,
+ SERVICE_ACTIVE,
+ lpDependencies,
+ dwBytesNeeded,
+ &dwBytesNeeded,
+ &dwCount))
+ return FALSE;
+
+ for (i = 0; i < dwCount; i++) {
+ ess = *(lpDependencies + i);
+ /* Open the service. */
+ hDepService = OpenService(lpService->hManager,
+ ess.lpServiceName,
+ SERVICE_STOP | SERVICE_QUERY_STATUS);
+
+ if (!hDepService)
+ return FALSE;
+
+ __try {
+ /* Send a stop code. */
+ if (!ControlService(hDepService,
+ SERVICE_CONTROL_STOP,
+ (LPSERVICE_STATUS) &ssp))
+ return FALSE;
+
+ /* Wait for the service to stop. */
+ while (ssp.dwCurrentState != SERVICE_STOPPED) {
+ Sleep(ssp.dwWaitHint);
+ if (!QueryServiceStatusEx(hDepService,
+ SC_STATUS_PROCESS_INFO,
+ (LPBYTE)&ssp,
+ sizeof(SERVICE_STATUS_PROCESS),
+ &dwBytesNeeded))
+ return FALSE;
+
+ if (ssp.dwCurrentState == SERVICE_STOPPED)
+ break;
+
+ if (GetTickCount() - dwStartTime > dwTimeout)
+ return FALSE;
+ }
+ }
+ __finally {
+ /* Always release the service handle. */
+ CloseServiceHandle(hDepService);
+ }
+ }
+ }
+ __finally {
+ /* Always free the enumeration buffer. */
+ HeapFree(GetProcessHeap(), 0, lpDependencies);
+ }
+ }
+ return TRUE;
+}
+
+BOOL
+apxServiceControl(APXHANDLE hService, DWORD dwControl, UINT uMsg,
+ LPAPXFNCALLBACK fnControlCallback,
+ LPVOID lpCbData)
+{
+ LPAPXSERVICE lpService;
+ SERVICE_STATUS stStatus;
+ DWORD dwPending = 0;
+ DWORD dwState = 0;
+ DWORD dwTick = 0;
+ DWORD dwWait, dwCheck, dwStart;
+ BOOL bStatus;
+
+ if (hService->dwType != APXHANDLE_TYPE_SERVICE)
+ return FALSE;
+
+ lpService = APXHANDLE_DATA(hService);
+ /* Manager mode cannot handle services */
+ if (lpService->bManagerMode)
+ return FALSE;
+ /* Check if the ServiceOpen has been called */
+ if (IS_INVALID_HANDLE(lpService->hService))
+ return FALSE;
+ switch (dwControl) {
+ case SERVICE_CONTROL_CONTINUE:
+ dwPending = SERVICE_START_PENDING;
+ dwState = SERVICE_RUNNING;
+ break;
+ case SERVICE_CONTROL_STOP:
+ dwPending = SERVICE_STOP_PENDING;
+ dwState = SERVICE_STOPPED;
+ break;
+ case SERVICE_CONTROL_PAUSE:
+ dwPending = SERVICE_PAUSE_PENDING;
+ dwState = SERVICE_PAUSED;
+ break;
+ default:
+ break;
+ }
+ /* user defined controls */
+ if (dwControl > 127 && dwControl < 224) {
+ /* 128 ... 159 start signals
+ * 160 ... 191 stop signals
+ * 192 ... 223 pause signals
+ */
+ switch (dwControl & 0xE0) {
+ case 0x80:
+ case 0x90:
+ dwPending = SERVICE_START_PENDING;
+ dwState = SERVICE_RUNNING;
+ break;
+ case 0xA0:
+ case 0xB0:
+ dwPending = SERVICE_STOP_PENDING;
+ dwState = SERVICE_STOPPED;
+ break;
+ case 0xC0:
+ case 0xD0:
+ dwPending = SERVICE_PAUSE_PENDING;
+ dwState = SERVICE_PAUSED;
+ break;
+ default:
+ break;
+ }
+ }
+ if (!dwPending && !dwState)
+ return FALSE;
+ /* Now lets control */
+ if (!QueryServiceStatus(lpService->hService, &stStatus)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return FALSE;
+ }
+ /* signal that we are about to control the service */
+ if (fnControlCallback)
+ (*fnControlCallback)(lpCbData, uMsg, (WPARAM)1, (LPARAM)dwState);
+ if (dwControl == SERVICE_CONTROL_CONTINUE &&
+ stStatus.dwCurrentState != SERVICE_PAUSED)
+ bStatus = StartService(lpService->hService, 0, NULL);
+ else {
+ bStatus = TRUE;
+ if (dwControl == SERVICE_CONTROL_STOP) {
+ /* First stop dependent services
+ */
+ bStatus = __apxStopDependentServices(lpService);
+ }
+ if (bStatus)
+ bStatus = ControlService(lpService->hService, dwControl, &stStatus);
+ }
+ dwStart = GetTickCount();
+ dwCheck = stStatus.dwCheckPoint;
+ if (bStatus) {
+ Sleep(100); /* Initial Sleep period */
+ while (QueryServiceStatus(lpService->hService, &stStatus)) {
+ if (stStatus.dwCurrentState == dwPending) {
+ /* Do not wait longer than the wait hint. A good interval is
+ * one tenth the wait hint, but no less than 1 second and no
+ * more than 10 seconds.
+ */
+ dwWait = stStatus.dwWaitHint / 10;
+
+ if( dwWait < 1000 )
+ dwWait = 1000;
+ else if ( dwWait > 10000 )
+ dwWait = 10000;
+ /* Signal to the callback that we are pending
+ * break if callback returns false.
+ */
+ if (fnControlCallback) {
+ if (!(*fnControlCallback)(lpCbData, uMsg, (WPARAM)2,
+ (LPARAM)dwTick++))
+ break;
+ }
+ Sleep(dwWait);
+ if (stStatus.dwCheckPoint > dwCheck) {
+ /* The service is making progress. */
+ dwStart = GetTickCount();
+ dwCheck = stStatus.dwCheckPoint;
+ }
+ else {
+ if(GetTickCount() - dwStart > stStatus.dwWaitHint) {
+ /* No progress made within the wait hint */
+ break;
+ }
+ }
+ }
+ else
+ break;
+ }
+ }
+ /* signal that we are done with controling the service */
+ if (fnControlCallback)
+ (*fnControlCallback)(lpCbData, uMsg, (WPARAM)3, (LPARAM)0);
+ /* Check if we are in the desired state */
+ Sleep(1000);
+
+ if (QueryServiceStatus(lpService->hService, &stStatus)) {
+ if (fnControlCallback)
+ (*fnControlCallback)(lpCbData, uMsg, (WPARAM)4,
+ (LPARAM)&stStatus);
+ if (stStatus.dwCurrentState == dwState)
+ return TRUE;
+
+ }
+
+ return FALSE;
+}
+
+BOOL
+apxServiceInstall(APXHANDLE hService, LPCWSTR szServiceName,
+ LPCWSTR szDisplayName, LPCWSTR szImagePath,
+ LPCWSTR lpDependencies, DWORD dwServiceType,
+ DWORD dwStartType)
+{
+ LPAPXSERVICE lpService;
+
+ if (hService->dwType != APXHANDLE_TYPE_SERVICE)
+ return FALSE;
+
+ lpService = APXHANDLE_DATA(hService);
+ /* Manager mode cannot handle services */
+ if (lpService->bManagerMode)
+ return FALSE;
+ if (IS_INVALID_HANDLE(lpService->hManager))
+ return FALSE;
+ if (!__apxIsValidServiceName(szServiceName))
+ return FALSE;
+ /* Close any previous instance
+ * Same handle can install multiple services
+ */
+ SAFE_CLOSE_SCH(lpService->hService);
+
+ apxFree(lpService->stServiceEntry.lpConfig);
+ lpService->stServiceEntry.lpConfig = NULL;
+ AplZeroMemory(&lpService->stServiceEntry, sizeof(APXSERVENTRY));
+
+ lpService->hService = CreateServiceW(lpService->hManager,
+ szServiceName,
+ szDisplayName,
+ SERVICE_ALL_ACCESS,
+ dwServiceType,
+ dwStartType,
+ SERVICE_ERROR_NORMAL,
+ szImagePath,
+ NULL,
+ NULL,
+ lpDependencies,
+ NULL,
+ NULL);
+
+ if (IS_INVALID_HANDLE(lpService->hService)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ return FALSE;
+ }
+ else {
+ lstrlcpyW(lpService->stServiceEntry.szServiceName,
+ SIZ_RESLEN, szServiceName);
+ lpService->stServiceEntry.dwStart = dwStartType;
+ return TRUE;
+ }
+}
+
+BOOL
+apxServiceDelete(APXHANDLE hService)
+{
+ LPAPXSERVICE lpService;
+ SERVICE_STATUS stStatus;
+ if (hService->dwType != APXHANDLE_TYPE_SERVICE)
+ return FALSE;
+
+ lpService = APXHANDLE_DATA(hService);
+ /* Manager mode cannot handle services */
+ if (lpService->bManagerMode)
+ return FALSE;
+ /* Check if the ServiceOpen has been called */
+ if (IS_INVALID_HANDLE(lpService->hService))
+ return FALSE;
+ if (QueryServiceStatus(lpService->hService, &stStatus)) {
+ BOOL rv;
+ if (stStatus.dwCurrentState != SERVICE_STOPPED)
+ apxServiceControl(hService, SERVICE_CONTROL_STOP, 0, NULL, NULL);
+ rv = DeleteService(lpService->hService);
+ SAFE_CLOSE_SCH(lpService->hService);
+ SAFE_CLOSE_SCH(lpService->hManager);
+
+ return rv;
+ }
+ return FALSE;
+}
+
+/* Browse the services */
+DWORD
+apxServiceBrowse(APXHANDLE hService,
+ LPCWSTR szIncludeNamePattern,
+ LPCWSTR szIncludeImagePattern,
+ LPCWSTR szExcludeNamePattern,
+ LPCWSTR szExcludeImagePattern,
+ UINT uMsg,
+ LPAPXFNCALLBACK fnDisplayCallback,
+ LPVOID lpCbData)
+{
+ DWORD nFound = 0;
+ APXREGENUM stEnum;
+ LPAPXSERVICE lpService;
+ SC_LOCK hLock;
+ if (hService->dwType != APXHANDLE_TYPE_SERVICE || !fnDisplayCallback)
+ return 0;
+
+ lpService = APXHANDLE_DATA(hService);
+ /* Only the manager mode can browse services */
+ if (!lpService->bManagerMode ||
+ IS_INVALID_HANDLE(lpService->hManager))
+ return 0;
+ hLock = LockServiceDatabase(lpService->hManager);
+ if (IS_INVALID_HANDLE(hLock)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+
+ return 0;
+ }
+ AplZeroMemory(&stEnum, sizeof(APXREGENUM));
+
+ while (TRUE) {
+ APXSERVENTRY stEntry;
+ BOOL rv;
+ AplZeroMemory(&stEntry, sizeof(APXSERVENTRY));
+ rv = apxRegistryEnumServices(&stEnum, &stEntry);
+
+ if (rv) {
+ INT fm = -1;
+ SC_HANDLE hSrv = NULL;
+ DWORD dwNeeded = 0;
+ hSrv = OpenServiceW(lpService->hManager,
+ stEntry.szServiceName,
+ GENERIC_READ);
+ if (!IS_INVALID_HANDLE(hSrv)) {
+ QueryServiceConfigW(hSrv, NULL, 0, &dwNeeded);
+ stEntry.lpConfig = (LPQUERY_SERVICE_CONFIGW)apxPoolAlloc(hService->hPool,
+ dwNeeded);
+ /* Call the QueryServiceConfig againg with allocated config */
+ if (QueryServiceConfigW(hSrv, stEntry.lpConfig, dwNeeded, &dwNeeded)) {
+ /* Make that customizable so that kernel mode drivers can be
+ * displayed and maintained. For now skip the
+ * filesystem and device drivers.
+ * XXX: Do we need that customizable after all?
+ */
+ if ((stEntry.lpConfig->dwServiceType &
+ ~SERVICE_INTERACTIVE_PROCESS) & SERVICE_WIN32)
+ fm = 0;
+
+ if (!fm && szIncludeNamePattern) {
+ fm = apxMultiStrMatchW(stEntry.szServiceName,
+ szIncludeNamePattern, L';', TRUE);
+ }
+ if (!fm && szExcludeNamePattern) {
+ fm = !apxMultiStrMatchW(stEntry.szServiceName,
+ szExcludeNamePattern, L';', TRUE);
+ }
+ if (!fm && szIncludeImagePattern) {
+ fm = apxMultiStrMatchW(stEntry.lpConfig->lpBinaryPathName,
+ szIncludeImagePattern, L';', TRUE);
+ }
+ if (!fm && szExcludeImagePattern) {
+ fm = !apxMultiStrMatchW(stEntry.szServiceName,
+ szExcludeImagePattern, L';', TRUE);
+ }
+ if (!fm) {
+ QueryServiceStatus(hSrv, &(stEntry.stServiceStatus));
+ /* WIN2K + extended service info */
+ if (_st_apx_oslevel >= 4) {
+ DWORD dwNeed;
+ QueryServiceStatusEx(hSrv, SC_STATUS_PROCESS_INFO,
+ (LPBYTE)(&(stEntry.stStatusProcess)),
+ sizeof(SERVICE_STATUS_PROCESS),
+ &dwNeed);
+ }
+ /* finaly call the provided callback */
+ rv = (*fnDisplayCallback)(lpCbData, uMsg,
+ (WPARAM)&stEntry,
+ (LPARAM)nFound++);
+ }
+ }
+ /* release the skipped service config */
+ if (fm) {
+ apxFree(stEntry.lpConfig);
+ }
+ }
+ SAFE_CLOSE_SCH(hSrv);
+ }
+ if (!rv)
+ break;
+ }
+
+ UnlockServiceDatabase(hLock);
+ return nFound;
+}
Deleted: tags/jbosssvc/2.0.8/procrun/src/utils.c
===================================================================
--- trunk/utils/windows/native/service/procrun/src/utils.c 2010-06-16 10:02:46 UTC (rev 2688)
+++ tags/jbosssvc/2.0.8/procrun/src/utils.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -1,1037 +0,0 @@
-/* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#include "apxwin.h"
-#include "private.h"
-
-APX_OSLEVEL _st_apx_oslevel = APX_WINVER_UNK;
-
-/* Apache's APR stripped Os level detection */
-APX_OSLEVEL apxGetOsLevel()
-{
- if (_st_apx_oslevel == APX_WINVER_UNK) {
- static OSVERSIONINFO oslev;
- oslev.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
- GetVersionEx(&oslev);
-
- if (oslev.dwPlatformId == VER_PLATFORM_WIN32_NT) {
- if (oslev.dwMajorVersion < 4)
- _st_apx_oslevel = APX_WINVER_UNSUP;
- else if (oslev.dwMajorVersion == 4)
- _st_apx_oslevel = APX_WINVER_NT_4;
- else if (oslev.dwMajorVersion == 5) {
- if (oslev.dwMinorVersion == 0)
- _st_apx_oslevel = APX_WINVER_2000;
- else
- _st_apx_oslevel = APX_WINVER_XP;
- }
- else
- _st_apx_oslevel = APX_WINVER_XP;
- }
-#ifndef WINNT
- else if (oslev.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS) {
- if (oslev.dwMinorVersion < 10)
- _st_apx_oslevel = APX_WINVER_95;
- else if (oslev.dwMinorVersion < 90)
- _st_apx_oslevel = APX_WINVER_98;
- else
- _st_apx_oslevel = APX_WINVER_ME;
- }
-#endif
-#ifdef _WIN32_WCE
- else if (oslev.dwPlatformId == VER_PLATFORM_WIN32_CE) {
- if (oslev.dwMajorVersion < 3)
- _st_apx_oslevel = APX_WINVER_UNSUP;
- else
- _st_apx_oslevel = APX_WINVER_CE_3;
- }
-#endif
- else
- _st_apx_oslevel = APX_WINVER_UNSUP;
- }
-
- if (_st_apx_oslevel < APX_WINVER_UNSUP)
- return APX_WINVER_UNK;
- else
- return _st_apx_oslevel;
-}
-
-LPWSTR __apxGetEnvironmentVariableW(APXHANDLE hPool, LPCWSTR wsName)
-{
- LPWSTR wsRet;
- DWORD rc;
-
- rc = GetEnvironmentVariableW(wsName, NULL, 0);
- if (rc == 0 && GetLastError() == ERROR_ENVVAR_NOT_FOUND)
- return NULL;
-
- if (!(wsRet = apxPoolAlloc(hPool, (rc + 1) * sizeof(WCHAR))))
- return NULL;
- if (!GetEnvironmentVariableW(wsName, wsRet, rc)) {
- apxFree(wsRet);
- ErrorMessage(NULL, FALSE);
- return NULL;
- }
- return wsRet;
-}
-
-LPSTR __apxGetEnvironmentVariableA(APXHANDLE hPool, LPCSTR szName)
-{
- LPSTR szRet;
- DWORD rc;
-
- rc = GetEnvironmentVariableA(szName, NULL, 0);
- if (rc == 0 && GetLastError() == ERROR_ENVVAR_NOT_FOUND)
- return NULL;
-
- if (!(szRet = apxPoolAlloc(hPool, rc + 1)))
- return NULL;
- if (!GetEnvironmentVariableA(szName, szRet, rc)) {
- apxFree(szRet);
- ErrorMessage(NULL, FALSE);
- return NULL;
- }
- return szRet;
-}
-
-
-LPWSTR AsciiToWide(LPCSTR s, LPWSTR ws)
-{
- LPWSTR pszSave = ws;
-
- if (!s) {
- *ws = L'\0';
- return pszSave;
- }
- do {
- *ws++ = (WCHAR)*s;
- } while (*s++);
- return pszSave;
-}
-
-LPSTR WideToAscii(LPCWSTR ws, LPSTR s)
-{
- LPSTR pszSave = s;
-
- if (!ws) {
- *s = '\0';
- return pszSave;
- }
- do {
- *s++ = (CHAR)*ws;
- } while( *ws++);
- return pszSave;
-}
-
-LPSTR WideToUTF8(LPCWSTR ws)
-{
-
- LPSTR s;
- int cch = WideCharToMultiByte(CP_UTF8, 0, ws, -1, NULL, 0, NULL, NULL);
- s = (LPSTR)apxAlloc(cch);
- if (!WideCharToMultiByte(CP_UTF8, 0, ws, -1, s, cch, NULL, NULL)) {
- apxFree(s);
- return NULL;
- }
- return s;
-}
-
-LPSTR MzWideToAscii(LPCWSTR ws, LPSTR s)
-{
- LPSTR pszSave = s;
-
- if (ws) {
- do {
- *s++ = (CHAR)*ws;
- ws++;
- } while( *ws || *(ws + 1));
- }
- /* double terminate */
- *s++ = '\0';
- *s = '\0';
- return pszSave;
-}
-
-LPSTR MzWideToUTF8(LPCWSTR ws)
-{
- LPSTR str;
- LPSTR s;
- LPCWSTR p = ws;
- int cch = 0;
-
- for ( ; p && *p; p++) {
- int len = WideCharToMultiByte(CP_UTF8, 0, p, -1, NULL, 0, NULL, NULL);
- if (len > 0)
- cch += len;
- while (*p)
- p++;
- }
- cch ++;
- str = s = (LPSTR)apxAlloc(cch + 1);
-
- p = ws;
- for ( ; p && *p; p++) {
- int len = WideCharToMultiByte(CP_UTF8, 0, p, -1, s, cch, NULL, NULL);
- if (len > 0) {
- s = s + len;
- cch -= len;
- }
- while (*p)
- p++;
- }
- /* double terminate */
- *s = '\0';
- return str;
-}
-
-#ifdef _DEBUG
-
-void ErrorMessage(LPCTSTR szError, BOOL bFatal)
-{
- LPVOID lpMsgBuf = NULL;
- UINT nType;
- int nRet;
- DWORD dwErr = GetLastError();
- if (bFatal)
- nType = MB_ICONERROR | MB_ABORTRETRYIGNORE | MB_SYSTEMMODAL;
- else
- nType = MB_ICONEXCLAMATION | MB_OK;
- if (szError) {
- nRet = MessageBox(NULL, szError, TEXT("Application Error"), nType);
- }
- else {
- FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |
- FORMAT_MESSAGE_FROM_SYSTEM |
- FORMAT_MESSAGE_IGNORE_INSERTS,
- NULL,
- dwErr,
- MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),
- (LPTSTR) &lpMsgBuf, 0, NULL);
- nRet = MessageBox(NULL, (LPCTSTR)lpMsgBuf,
- TEXT("Application System Error"), nType);
- LocalFree(lpMsgBuf);
- }
- if (bFatal && (nRet == IDCANCEL || nRet == IDABORT)) {
- ExitProcess(dwErr);
- }
-}
-#endif /* _DEBUG */
-
-DWORD __apxGetMultiSzLengthA(LPCSTR lpStr, LPDWORD lpdwCount)
-{
- LPCSTR p = lpStr;
- if (lpdwCount)
- *lpdwCount = 0;
- for ( ; p && *p; p++) {
- if (lpdwCount)
- *lpdwCount += 1;
- while (*p)
- p++;
- }
- return (DWORD)(p - lpStr);
-}
-
-DWORD __apxGetMultiSzLengthW(LPCWSTR lpStr, LPDWORD lpdwCount)
-{
- LPCWSTR p = lpStr;
- if (lpdwCount)
- *lpdwCount = 0;
- for ( ; p && *p; p++) {
- if (lpdwCount)
- *lpdwCount += 1;
- while (*p)
- p++;
- }
- return (DWORD)((p - lpStr));
-}
-
-LPWSTR apxMultiSzCombine(APXHANDLE hPool, LPCWSTR lpStrA, LPCWSTR lpStrB,
- LPDWORD lpdwLength)
-{
- LPWSTR rv;
- DWORD la = 0, lb = 0;
- if (!lpStrA && !lpStrB)
- return NULL; /* Nothing to do if both are NULL */
-
- la = __apxGetMultiSzLengthW(lpStrA, NULL);
- lb = __apxGetMultiSzLengthW(lpStrB, NULL);
-
- rv = apxPoolCalloc(hPool, (la + lb + 1) * sizeof(WCHAR));
- if (la) {
- AplMoveMemory(rv, lpStrA, la * sizeof(WCHAR));
- }
- if (lb) {
- AplMoveMemory(&rv[la], lpStrB, lb * sizeof(WCHAR));
- }
- if (*lpdwLength)
- *lpdwLength = (la + lb + 1) * sizeof(WCHAR);
- return rv;
-}
-
-BOOL
-apxSetEnvironmentVariable(APXHANDLE hPool, LPCTSTR szName, LPCTSTR szValue,
- BOOL bAppend)
-{
- LPTSTR szNew = (LPTSTR)szValue;
-
- if (bAppend) {
- DWORD l = GetEnvironmentVariable(szName, NULL, 0);
- if (l > 0) {
- BOOL rv;
- if (IS_INVALID_HANDLE(hPool))
- szNew = apxAlloc(l + lstrlen(szValue) + 3);
- else
- szNew = apxPoolAlloc(hPool, l + lstrlen(szValue) + 3);
- GetEnvironmentVariable(szName, szNew, l + 1);
- lstrcat(szNew, TEXT(";"));
- lstrcat(szNew, szValue);
- rv = SetEnvironmentVariable(szName, szNew);
- apxFree(szNew);
- return rv;
- }
- }
- return SetEnvironmentVariable(szName, szNew);
-}
-
-
-/** Convert null separated double null terimated string to LPTSTR array)
- * returns array size
- */
-DWORD
-apxMultiSzToArrayW(APXHANDLE hPool, LPCWSTR lpString, LPWSTR **lppArray)
-{
- DWORD i, n, l;
- char *buff;
- LPWSTR p;
-
- l = __apxGetMultiSzLengthW(lpString, &n);
- if (!n || !l)
- return 0;
- if (IS_INVALID_HANDLE(hPool))
- buff = apxPoolAlloc(hPool, (n + 2) * sizeof(LPWSTR) + (l + 1) * sizeof(WCHAR));
- else
- buff = apxAlloc((n + 2) * sizeof(LPWSTR) + (l + 1) * sizeof(WCHAR));
-
- *lppArray = (LPWSTR *)buff;
- p = (LPWSTR)(buff + (n + 2) * sizeof(LPWSTR));
- AplCopyMemory(p, lpString, (l + 1) * sizeof(WCHAR) + sizeof(WCHAR));
- for (i = 0; i < n; i++) {
- (*lppArray)[i] = p;
- while (*p)
- p++;
- p++;
- }
- (*lppArray)[++i] = NULL;
-
- return n;
-}
-
-DWORD
-apxMultiSzToArrayA(APXHANDLE hPool, LPCSTR lpString, LPSTR **lppArray)
-{
- DWORD i, n, l;
- char *buff;
- LPSTR p;
-
- l = __apxGetMultiSzLengthA(lpString, &n);
- if (!n || !l)
- return 0;
- if (IS_INVALID_HANDLE(hPool))
- buff = apxPoolAlloc(hPool, (n + 2) * sizeof(LPTSTR) + (l + 1));
- else
- buff = apxAlloc((n + 2) * sizeof(LPSTR) + (l + 1) * sizeof(CHAR));
-
- *lppArray = (LPSTR *)buff;
- p = (LPSTR)(buff + (n + 2) * sizeof(LPSTR));
- AplCopyMemory(p, lpString, (l + 1) * sizeof(CHAR) + sizeof(CHAR));
- for (i = 0; i < n; i++) {
- (*lppArray)[i] = p;
- while (*p)
- p++;
- p++;
- }
- (*lppArray)[++i] = NULL;
-
- return n;
-}
-
-#define QSTR_BOUNDARY 127
-#define QSTR_ALIGN(size) \
- (((size) + QSTR_BOUNDARY + sizeof(APXMULTISZ) + 2) & ~(QSTR_BOUNDARY))
-
-#define QSTR_SIZE(size) \
- ((QSTR_ALIGN(size)) - sizeof(APXMULTISZ))
-
-#define QSTR_DATA(q) ((char *)(q) + sizeof(APXMULTISZ))
-
-#if 0
-LPAPXMULTISZ apxMultiSzStrdup(LPCTSTR szSrc)
-{
- LPAPXMULTISZ q;
-
- if (szSrc) {
- DWORD l = lstrlen(szSrc);
- q = (LPAPXMULTISZ)apxAlloc(QSTR_ALIGN(l));
- q->dwAllocated = QSTR_SIZE(l);
- q->dwInsert = l + 1;
- AplMoveMemory(QSTR_DATA(q), szSrc, l);
- RtlZeroMemory(QSTR_DATA(q) + l, q->dwAllocated - l);
- }
- else {
- q = (LPAPXMULTISZ)apxCalloc(QSTR_ALIGN(0));
- q->dwAllocated = QSTR_SIZE(0);
- }
-
- return q;
-}
-
-LPTSTR apxMultiSzStrcat(LPAPXMULTISZ lpmSz, LPCTSTR szSrc)
-{
- DWORD l = lstrlen(szSrc);
- LPTSTR p;
-
- if (lpmSz->dwInsert + l + 2 > lpmSz->dwAllocated) {
- if ((lpmSz = (LPAPXMULTISZ )apxRealloc(lpmSz, QSTR_ALIGN(lpmSz->dwInsert + l))) == NULL)
- return NULL;
-
- lpmSz->dwAllocated = QSTR_SIZE(lpmSz->dwInsert + l);
- AplZeroMemory(QSTR_DATA(lpmSz) + lpmSz->dwInsert + l,
- lpmSz->dwAllocated - (lpmSz->dwInsert + l));
- }
- p = (LPTSTR)QSTR_DATA(lpmSz) + lpmSz->dwInsert;
- AplMoveMemory(p, szSrc, l);
-
- lpmSz->dwInsert += (l + 1);
- return p;
-}
-
-DWORD apxMultiSzLen(LPAPXMULTISZ lpmSz)
-{
- if (lpmSz->dwInsert)
- return lpmSz->dwInsert - 1;
- else
- return 0;
-}
-
-LPCTSTR apxMultiSzGet(LPAPXMULTISZ lpmSz)
-{
- return (LPCTSTR)QSTR_DATA(lpmSz);
-}
-#endif
-
-LPTSTR apxStrCharRemove(LPTSTR szString, TCHAR chSkip)
-{
- LPTSTR p = szString;
- LPTSTR q = szString;
- while (*p) {
- if(*p != chSkip)
- *q++ = *p;
- ++p;
- }
- *q = TEXT('\0');
-
- return szString;
-}
-
-DWORD apxStrCharRemoveA(LPSTR szString, CHAR chSkip)
-{
- LPSTR p = szString;
- LPSTR q = szString;
- DWORD c = 0;
- while (*p) {
- if(*p != chSkip)
- *q++ = *p;
- else
- ++c;
- ++p;
- }
- *q = '\0';
-
- return c;
-}
-
-DWORD apxStrCharRemoveW(LPWSTR szString, WCHAR chSkip)
-{
- LPWSTR p = szString;
- LPWSTR q = szString;
- DWORD c = 0;
- while (*p) {
- if(*p != chSkip)
- *q++ = *p;
- else
- ++c;
- ++p;
- }
- *q = L'\0';
-
- return c;
-}
-
-void
-apxStrCharReplaceA(LPSTR szString, CHAR chReplace, CHAR chReplaceWith)
-{
- LPSTR p = szString;
- LPSTR q = szString;
- while (*p) {
- if(*p == chReplace)
- *q++ = chReplaceWith;
- else
- *q++ = *p;
- ++p;
- }
- *q = '\0';
-}
-
-void
-apxStrCharReplaceW(LPWSTR szString, WCHAR chReplace, WCHAR chReplaceWith)
-{
- LPWSTR p = szString;
- LPWSTR q = szString;
- while (*p) {
- if(*p == chReplace)
- *q++ = chReplaceWith;
- else
- *q++ = *p;
- ++p;
- }
- *q = L'\0';
-}
-
-static const LPCTSTR _st_hex = TEXT("0123456789abcdef");
-#define XTOABUFFER_SIZE (sizeof(ULONG) * 2 + 2)
-#define UTOABUFFER_SIZE (sizeof(ULONG_PTR) * 2 + 2)
-#define LO_NIBLE(x) ((BYTE)((x) & 0x0000000F))
-
-BOOL apxUltohex(ULONG n, LPTSTR lpBuff, DWORD dwBuffLength)
-{
- LPTSTR p;
- DWORD i;
- *lpBuff = 0;
- if (dwBuffLength < XTOABUFFER_SIZE)
- return FALSE;
- p = lpBuff + XTOABUFFER_SIZE;
- *p-- = 0;
- for (i = 0; i < sizeof(ULONG) * 2; i++) {
- *p-- = _st_hex[LO_NIBLE(n)];
- n = n >> 4;
- }
- *p-- = TEXT('x');
- *p = TEXT('0');
- return TRUE;
-}
-
-BOOL apxUptohex(ULONG_PTR n, LPTSTR lpBuff, DWORD dwBuffLength)
-{
- LPTSTR p;
- DWORD i;
- *lpBuff = 0;
- if (dwBuffLength < UTOABUFFER_SIZE)
- return FALSE;
- p = lpBuff + UTOABUFFER_SIZE;
- *p-- = 0;
- for (i = 0; i < sizeof(ULONG_PTR) * 2; i++) {
- *p-- = _st_hex[LO_NIBLE(n)];
- n = n >> 4;
- }
- *p-- = TEXT('x');
- *p = TEXT('0');
- return TRUE;
-}
-
-ULONG apxStrToul(LPCTSTR szNum)
-{
- ULONG rv = 0;
- DWORD sh = 0;
- LPCTSTR p = szNum;
- ++p;
- while (*p && (*p != TEXT('x')) && (*(p - 1) != TEXT('0')))
- p++;
- if (*p != 'x')
- return 0;
- /* go to the last digit */
- while (*(p + 1)) p++;
-
- /* go back to 'x' */
- while (*p != TEXT('x')) {
- ULONG v = 0;
- switch (*p--) {
- case TEXT('0'): v = 0UL; break;
- case TEXT('1'): v = 1UL; break;
- case TEXT('2'): v = 2UL; break;
- case TEXT('3'): v = 3UL; break;
- case TEXT('4'): v = 4UL; break;
- case TEXT('5'): v = 5UL; break;
- case TEXT('6'): v = 6UL; break;
- case TEXT('7'): v = 7UL; break;
- case TEXT('8'): v = 8UL; break;
- case TEXT('9'): v = 9UL; break;
- case TEXT('a'): case TEXT('A'): v = 10UL; break;
- case TEXT('b'): case TEXT('B'): v = 11UL; break;
- case TEXT('c'): case TEXT('C'): v = 12UL; break;
- case TEXT('d'): case TEXT('D'): v = 13UL; break;
- case TEXT('e'): case TEXT('E'): v = 14UL; break;
- case TEXT('f'): case TEXT('F'): v = 15UL; break;
- default:
- return 0;
- break;
- }
- rv |= rv + (v << sh);
- sh += 4;
- }
- return rv;
-}
-
-ULONG apxStrToulW(LPCWSTR szNum)
-{
- ULONG rv = 0;
- DWORD sh = 0;
- LPCWSTR p = szNum;
- ++p;
- while (*p && (*p != L'x') && (*(p - 1) != L'0'))
- p++;
- if (*p != L'x')
- return 0;
- /* go to the last digit */
- while (*(p + 1)) p++;
-
- /* go back to 'x' */
- while (*p != L'x') {
- ULONG v = 0;
- switch (*p--) {
- case L'0': v = 0UL; break;
- case L'1': v = 1UL; break;
- case L'2': v = 2UL; break;
- case L'3': v = 3UL; break;
- case L'4': v = 4UL; break;
- case L'5': v = 5UL; break;
- case L'6': v = 6UL; break;
- case L'7': v = 7UL; break;
- case L'8': v = 8UL; break;
- case L'9': v = 9UL; break;
- case L'a': case L'A': v = 10UL; break;
- case L'b': case L'B': v = 11UL; break;
- case L'c': case L'C': v = 12UL; break;
- case L'd': case L'D': v = 13UL; break;
- case L'e': case L'E': v = 14UL; break;
- case L'f': case L'F': v = 15UL; break;
- default:
- return 0;
- break;
- }
- rv |= rv + (v << sh);
- sh += 4;
- }
- return rv;
-}
-
-ULONG apxAtoulW(LPCWSTR szNum)
-{
- ULONG rv = 0;
- DWORD sh = 1;
- int s = 1;
- LPCWSTR p = szNum;
-
- /* go to the last digit */
- if (!p || !*p)
- return 0;
- if (*p == L'-') {
- s = -1;
- ++p;
- }
- while (*(p + 1)) p++;
-
- /* go back */
- while (p >= szNum) {
- ULONG v = 0;
- switch (*p--) {
- case L'0': v = 0UL; break;
- case L'1': v = 1UL; break;
- case L'2': v = 2UL; break;
- case L'3': v = 3UL; break;
- case L'4': v = 4UL; break;
- case L'5': v = 5UL; break;
- case L'6': v = 6UL; break;
- case L'7': v = 7UL; break;
- case L'8': v = 8UL; break;
- case L'9': v = 9UL; break;
- default:
- return rv * s;
- break;
- }
- rv = rv + (v * sh);
- sh = sh * 10;
- }
- return rv * s;
-}
-
-/* Make the unique system resource name from prefix and process id
- *
- */
-BOOL
-apxMakeResourceName(LPCTSTR szPrefix, LPTSTR lpBuff, DWORD dwBuffLength)
-{
- DWORD pl = lstrlen(szPrefix);
- if (dwBuffLength < (pl + 11))
- return FALSE;
- lstrcpy(lpBuff, szPrefix);
- return apxUltohex(GetCurrentProcessId(), lpBuff + pl, dwBuffLength - pl);
-}
-/** apxStrMatchA ANSI string pattern matching
- * Match = 0, NoMatch = 1, Abort = -1
- * Based loosely on sections of wildmat.c by Rich Salz
- */
-INT apxStrMatchA(LPCSTR szString, LPCSTR szPattern, BOOL bIgnoreCase)
-{
- int x, y;
-
- for (x = 0, y = 0; szPattern[y]; ++y, ++x) {
- if (!szPattern[x] && (szPattern[y] != '*' || szPattern[y] != '?'))
- return -1;
- if (szPattern[y] == '*') {
- while (szPattern[++y] == '*');
- if (!szPattern[y])
- return 0;
- while (szString[x]) {
- INT rc;
- if ((rc = apxStrMatchA(&szString[x++], &szPattern[y],
- bIgnoreCase)) != 1)
- return rc;
- }
- return -1;
- }
- else if (szPattern[y] != '?') {
- if (bIgnoreCase) {
- if (CharLowerA((LPSTR)((SIZE_T)szString[x])) !=
- CharLowerA((LPSTR)((SIZE_T)szPattern[y])))
- return 1;
- }
- else {
- if (szString[x] != szPattern[y])
- return 1;
- }
- }
- }
- return (szString[x] != '\0');
-}
-
-INT apxStrMatchW(LPCWSTR szString, LPCWSTR szPattern, BOOL bIgnoreCase)
-{
- int x, y;
-
- for (x = 0, y = 0; szPattern[y]; ++y, ++x) {
- if (!szPattern[x] && (szPattern[y] != L'*' || szPattern[y] != L'?'))
- return -1;
- if (szPattern[y] == L'*') {
- while (szPattern[++y] == L'*');
- if (!szPattern[y])
- return 0;
- while (szString[x]) {
- INT rc;
- if ((rc = apxStrMatchW(&szString[x++], &szPattern[y],
- bIgnoreCase)) != 1)
- return rc;
- }
- return -1;
- }
- else if (szPattern[y] != L'?') {
- if (bIgnoreCase) {
- if (CharLowerW((LPWSTR)((SIZE_T)szString[x])) !=
- CharLowerW((LPWSTR)((SIZE_T)szPattern[y])))
- return 1;
- }
- else {
- if (szString[x] != szPattern[y])
- return 1;
- }
- }
- }
- return (szString[x] != L'\0');
-}
-
-INT apxMultiStrMatchW(LPCWSTR szString, LPCWSTR szPattern,
- WCHAR chSeparator, BOOL bIgnoreCase)
-{
- WCHAR szM[SIZ_HUGLEN];
- DWORD i = 0;
- LPCWSTR p = szPattern;
- INT m = -1;
-
- if (chSeparator == 0)
- return apxStrMatchW(szString, szPattern, bIgnoreCase);
- while (*p != L'\0') {
- if (*p == chSeparator) {
- m = apxStrMatchW(szString, szM, bIgnoreCase);
- if (m == 0)
- return 0;
- p++;
- i = 0;
- szM[0] = L'\0';
- }
- else {
- if (i < SIZ_HUGMAX)
- szM[i++] = *p++;
- else
- return -1;
- }
- }
- szM[i] = L'\0';
- if (szM[0])
- return apxStrMatchW(szString, szM, bIgnoreCase);
- else
- return m;
-}
-
-LPSTR apxArrayToMultiSzA(APXHANDLE hPool, DWORD nArgs, LPCSTR *lpArgs)
-{
- DWORD i, l = 0;
- LPSTR lpSz, p;
- if (!nArgs)
- return NULL;
- for (i = 0; i < nArgs; i++)
- l += lstrlenA(lpArgs[i]);
- l += (nArgs + 2);
-
- p = lpSz = (LPSTR)apxPoolAlloc(hPool, l);
- for (i = 0; i < nArgs; i++) {
- lstrcpyA(p, lpArgs[i]);
- p += lstrlenA(lpArgs[i]);
- *p++ = '\0';
- }
- *p++ = '\0';
- *p++ = '\0';
- return lpSz;
-}
-
-void apxStrQuoteInplaceW(LPWSTR szString)
-{
- LPWSTR p = szString;
- BOOL needsQuote = FALSE;
- while (*p) {
- if (*p++ == L' ') {
- needsQuote = TRUE;
- break;
- }
- }
- if (needsQuote) {
- DWORD l = lstrlenW(szString);
- AplMoveMemory(&szString[1], szString, l * sizeof(WCHAR));
- szString[0] = L'"';
- szString[++l] = L'"';
- szString[++l] = L'\0';
- }
-}
-
-DWORD apxStrUnQuoteInplaceA(LPSTR szString)
-{
- LPSTR p = szString;
- BOOL needsQuote = FALSE;
- BOOL inQuote = FALSE;
- while (*p) {
- if (*p == '"') {
- if (inQuote)
- break;
- else
- inQuote = TRUE;
- }
- else if (*p == ' ') {
- if (inQuote) {
- needsQuote = TRUE;
- break;
- }
- }
- ++p;
- }
- if (!needsQuote)
- return apxStrCharRemoveA(szString, '"');
- else
- return 0;
-}
-
-DWORD apxStrUnQuoteInplaceW(LPWSTR szString)
-{
- LPWSTR p = szString;
- BOOL needsQuote = FALSE;
- BOOL inQuote = FALSE;
- while (*p) {
- if (*p == L'"') {
- if (inQuote)
- break;
- else
- inQuote = TRUE;
- }
- else if (*p == L' ') {
- if (inQuote) {
- needsQuote = TRUE;
- break;
- }
- }
- ++p;
- }
- if (!needsQuote)
- return apxStrCharRemoveW(szString, L'"');
- else
- return 0;
-}
-
-LPWSTR
-apxMszToCRLFW(APXHANDLE hPool, LPCWSTR szStr)
-{
- DWORD l, c;
- LPWSTR rv, b;
- LPCWSTR p = szStr;
-
- l = __apxGetMultiSzLengthW(szStr, &c);
- b = rv = apxPoolCalloc(hPool, (l + c + 2) * sizeof(WCHAR));
-
- while (c > 0) {
- if (*p)
- *b++ = *p;
- else {
- *b++ = L'\r';
- *b++ = L'\n';
- c--;
- }
- p++;
- }
- return rv;
-}
-
-LPWSTR
-apxCRLFToMszW(APXHANDLE hPool, LPCWSTR szStr, LPDWORD lpdwBytes)
-{
- DWORD l, c, n = 0;
- LPWSTR rv, b;
-
- l = lstrlenW(szStr);
- b = rv = apxPoolCalloc(hPool, (l + 2) * sizeof(WCHAR));
- for (c = 0; c < l; c++) {
- if (szStr[c] == L'\r') {
- *b++ = '\0';
- n++;
- }
- else if (szStr[c] != L'\n') {
- *b++ = szStr[c];
- n++;
- }
- }
- if (lpdwBytes)
- *lpdwBytes = (n + 2) * sizeof(WCHAR);
- return rv;
-}
-
-LPSTR
-apxExpandStrA(APXHANDLE hPool, LPCSTR szString)
-{
- LPCSTR p = szString;
- while (*p) {
- if (*p == '%') {
- p = szString;
- break;
- }
- ++p;
- }
- if (p != szString)
- return apxPoolStrdupA(hPool, szString);
- else {
- DWORD l = ExpandEnvironmentStringsA(szString, NULL, 0);
- if (l) {
- LPSTR rv = apxPoolAlloc(hPool, l);
- l = ExpandEnvironmentStringsA(szString, rv, l);
- if (l)
- return rv;
- else {
- apxFree(rv);
- return NULL;
- }
- }
- else
- return NULL;
- }
-}
-
-LPWSTR
-apxExpandStrW(APXHANDLE hPool, LPCWSTR szString)
-{
- LPCWSTR p = szString;
- while (*p) {
- if (*p == L'%') {
- p = szString;
- break;
- }
- ++p;
- }
- if (p != szString)
- return apxPoolStrdupW(hPool, szString);
- else {
- DWORD l = ExpandEnvironmentStringsW(szString, NULL, 0);
- if (l) {
- LPWSTR rv = apxPoolAlloc(hPool, l * sizeof(WCHAR));
- l = ExpandEnvironmentStringsW(szString, rv, l);
- if (l)
- return rv;
- else {
- apxFree(rv);
- return NULL;
- }
- }
- else
- return NULL;
- }
-}
-
-/* To share the semaphores with other processes, we need a NULL ACL
- * Code from MS KB Q106387
- */
-PSECURITY_ATTRIBUTES GetNullACL()
-{
- PSECURITY_DESCRIPTOR pSD;
- PSECURITY_ATTRIBUTES sa;
-
- sa = (PSECURITY_ATTRIBUTES) LocalAlloc(LPTR, sizeof(SECURITY_ATTRIBUTES));
- sa->nLength = sizeof(sizeof(SECURITY_ATTRIBUTES));
-
- pSD = (PSECURITY_DESCRIPTOR) LocalAlloc(LPTR, SECURITY_DESCRIPTOR_MIN_LENGTH);
- sa->lpSecurityDescriptor = pSD;
-
- if (pSD == NULL || sa == NULL) {
- return NULL;
- }
- SetLastError(0);
- if (!InitializeSecurityDescriptor(pSD, SECURITY_DESCRIPTOR_REVISION)
- || GetLastError()) {
- LocalFree( pSD );
- LocalFree( sa );
- return NULL;
- }
- if (!SetSecurityDescriptorDacl(pSD, TRUE, (PACL) NULL, FALSE)
- || GetLastError()) {
- LocalFree( pSD );
- LocalFree( sa );
- return NULL;
- }
-
- sa->bInheritHandle = FALSE;
- return sa;
-}
-
-
-void CleanNullACL(void *sa)
-{
- if (sa) {
- LocalFree(((PSECURITY_ATTRIBUTES)sa)->lpSecurityDescriptor);
- LocalFree(sa);
- }
-}
Copied: tags/jbosssvc/2.0.8/procrun/src/utils.c (from rev 2689, trunk/utils/windows/native/service/procrun/src/utils.c)
===================================================================
--- tags/jbosssvc/2.0.8/procrun/src/utils.c (rev 0)
+++ tags/jbosssvc/2.0.8/procrun/src/utils.c 2010-07-21 20:14:05 UTC (rev 2690)
@@ -0,0 +1,1017 @@
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "apxwin.h"
+#include "private.h"
+
+APX_OSLEVEL _st_apx_oslevel = APX_WINVER_UNK;
+
+/* Apache's APR stripped Os level detection */
+APX_OSLEVEL apxGetOsLevel()
+{
+ if (_st_apx_oslevel == APX_WINVER_UNK) {
+ static OSVERSIONINFO oslev;
+ oslev.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
+ GetVersionEx(&oslev);
+
+ if (oslev.dwPlatformId == VER_PLATFORM_WIN32_NT) {
+ if (oslev.dwMajorVersion < 4)
+ _st_apx_oslevel = APX_WINVER_UNSUP;
+ else if (oslev.dwMajorVersion == 4)
+ _st_apx_oslevel = APX_WINVER_NT_4;
+ else if (oslev.dwMajorVersion == 5) {
+ if (oslev.dwMinorVersion == 0)
+ _st_apx_oslevel = APX_WINVER_2000;
+ else
+ _st_apx_oslevel = APX_WINVER_XP;
+ }
+ else
+ _st_apx_oslevel = APX_WINVER_XP;
+ }
+#ifndef WINNT
+ else if (oslev.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS) {
+ if (oslev.dwMinorVersion < 10)
+ _st_apx_oslevel = APX_WINVER_95;
+ else if (oslev.dwMinorVersion < 90)
+ _st_apx_oslevel = APX_WINVER_98;
+ else
+ _st_apx_oslevel = APX_WINVER_ME;
+ }
+#endif
+#ifdef _WIN32_WCE
+ else if (oslev.dwPlatformId == VER_PLATFORM_WIN32_CE) {
+ if (oslev.dwMajorVersion < 3)
+ _st_apx_oslevel = APX_WINVER_UNSUP;
+ else
+ _st_apx_oslevel = APX_WINVER_CE_3;
+ }
+#endif
+ else
+ _st_apx_oslevel = APX_WINVER_UNSUP;
+ }
+
+ if (_st_apx_oslevel < APX_WINVER_UNSUP)
+ return APX_WINVER_UNK;
+ else
+ return _st_apx_oslevel;
+}
+
+LPWSTR __apxGetEnvironmentVariableW(APXHANDLE hPool, LPCWSTR wsName)
+{
+ LPWSTR wsRet;
+ DWORD rc;
+
+ rc = GetEnvironmentVariableW(wsName, NULL, 0);
+ if (rc == 0 && GetLastError() == ERROR_ENVVAR_NOT_FOUND)
+ return NULL;
+
+ if (!(wsRet = apxPoolAlloc(hPool, (rc + 1) * sizeof(WCHAR))))
+ return NULL;
+ if (!GetEnvironmentVariableW(wsName, wsRet, rc)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ apxFree(wsRet);
+ return NULL;
+ }
+ return wsRet;
+}
+
+LPSTR __apxGetEnvironmentVariableA(APXHANDLE hPool, LPCSTR szName)
+{
+ LPSTR szRet;
+ DWORD rc;
+
+ rc = GetEnvironmentVariableA(szName, NULL, 0);
+ if (rc == 0 && GetLastError() == ERROR_ENVVAR_NOT_FOUND)
+ return NULL;
+
+ if (!(szRet = apxPoolAlloc(hPool, rc + 1)))
+ return NULL;
+ if (!GetEnvironmentVariableA(szName, szRet, rc)) {
+ apxLogWrite(APXLOG_MARK_SYSERR);
+ apxFree(szRet);
+ return NULL;
+ }
+ return szRet;
+}
+
+
+LPWSTR AsciiToWide(LPCSTR s, LPWSTR ws)
+{
+ LPWSTR pszSave = ws;
+
+ if (!s) {
+ *ws = L'\0';
+ return pszSave;
+ }
+ do {
+ *ws++ = (WCHAR)*s;
+ } while (*s++);
+ return pszSave;
+}
+
+LPSTR WideToAscii(LPCWSTR ws, LPSTR s)
+{
+ LPSTR pszSave = s;
+
+ if (!ws) {
+ *s = '\0';
+ return pszSave;
+ }
+ do {
+ *s++ = (CHAR)*ws;
+ } while( *ws++);
+ return pszSave;
+}
+
+LPSTR WideToUTF8(LPCWSTR ws)
+{
+
+ LPSTR s;
+ int cch = WideCharToMultiByte(CP_UTF8, 0, ws, -1, NULL, 0, NULL, NULL);
+ s = (LPSTR)apxAlloc(cch);
+ if (!WideCharToMultiByte(CP_UTF8, 0, ws, -1, s, cch, NULL, NULL)) {
+ apxFree(s);
+ return NULL;
+ }
+ return s;
+}
+
+LPWSTR UTF8ToWide(LPCSTR cs)
+{
+
+ LPWSTR s;
+ int cch = MultiByteToWideChar(CP_UTF8, 0, cs, -1, NULL, 0);
+ s = (LPWSTR)apxAlloc(cch * sizeof(WCHAR));
+ if (!MultiByteToWideChar(CP_UTF8, 0, cs, -1, s, cch)) {
+ apxFree(s);
+ return NULL;
+ }
+ return s;
+}
+
+LPSTR MzWideToAscii(LPCWSTR ws, LPSTR s)
+{
+ LPSTR pszSave = s;
+
+ if (ws) {
+ do {
+ *s++ = (CHAR)*ws;
+ ws++;
+ } while( *ws || *(ws + 1));
+ }
+ /* double terminate */
+ *s++ = '\0';
+ *s = '\0';
+ return pszSave;
+}
+
+LPSTR MzWideToUTF8(LPCWSTR ws)
+{
+ LPSTR str;
+ LPSTR s;
+ LPCWSTR p = ws;
+ int cch = 0;
+
+ for ( ; p && *p; p++) {
+ int len = WideCharToMultiByte(CP_UTF8, 0, p, -1, NULL, 0, NULL, NULL);
+ if (len > 0)
+ cch += len;
+ while (*p)
+ p++;
+ }
+ cch ++;
+ str = s = (LPSTR)apxAlloc(cch + 1);
+
+ p = ws;
+ for ( ; p && *p; p++) {
+ int len = WideCharToMultiByte(CP_UTF8, 0, p, -1, s, cch, NULL, NULL);
+ if (len > 0) {
+ s = s + len;
+ cch -= len;
+ }
+ while (*p)
+ p++;
+ }
+ /* double terminate */
+ *s = '\0';
+ return str;
+}
+
+DWORD __apxGetMultiSzLengthA(LPCSTR lpStr, LPDWORD lpdwCount)
+{
+ LPCSTR p = lpStr;
+ if (lpdwCount)
+ *lpdwCount = 0;
+ for ( ; p && *p; p++) {
+ if (lpdwCount)
+ *lpdwCount += 1;
+ while (*p)
+ p++;
+ }
+ return (DWORD)(p - lpStr);
+}
+
+DWORD __apxGetMultiSzLengthW(LPCWSTR lpStr, LPDWORD lpdwCount)
+{
+ LPCWSTR p = lpStr;
+ if (lpdwCount)
+ *lpdwCount = 0;
+ for ( ; p && *p; p++) {
+ if (lpdwCount)
+ *lpdwCount += 1;
+ while (*p)
+ p++;
+ }
+ return (DWORD)((p - lpStr));
+}
+
+LPWSTR apxMultiSzCombine(APXHANDLE hPool, LPCWSTR lpStrA, LPCWSTR lpStrB,
+ LPDWORD lpdwLength)
+{
+ LPWSTR rv;
+ DWORD la = 0, lb = 0;
+ if (!lpStrA && !lpStrB)
+ return NULL; /* Nothing to do if both are NULL */
+
+ la = __apxGetMultiSzLengthW(lpStrA, NULL);
+ lb = __apxGetMultiSzLengthW(lpStrB, NULL);
+
+ rv = apxPoolCalloc(hPool, (la + lb + 1) * sizeof(WCHAR));
+ if (la) {
+ AplMoveMemory(rv, lpStrA, la * sizeof(WCHAR));
+ }
+ if (lb) {
+ AplMoveMemory(&rv[la], lpStrB, lb * sizeof(WCHAR));
+ }
+ if (*lpdwLength)
+ *lpdwLength = (la + lb + 1) * sizeof(WCHAR);
+ return rv;
+}
+
+BOOL
+apxSetEnvironmentVariable(APXHANDLE hPool, LPCTSTR szName, LPCTSTR szValue,
+ BOOL bAppend)
+{
+ LPTSTR szNew = (LPTSTR)szValue;
+
+ if (bAppend) {
+ DWORD l = GetEnvironmentVariable(szName, NULL, 0);
+ if (l > 0) {
+ BOOL rv;
+ if (IS_INVALID_HANDLE(hPool))
+ szNew = apxAlloc(l + lstrlen(szValue) + 3);
+ else
+ szNew = apxPoolAlloc(hPool, l + lstrlen(szValue) + 3);
+ GetEnvironmentVariable(szName, szNew, l + 1);
+ lstrcat(szNew, TEXT(";"));
+ lstrcat(szNew, szValue);
+ rv = SetEnvironmentVariable(szName, szNew);
+ apxFree(szNew);
+ return rv;
+ }
+ }
+ return SetEnvironmentVariable(szName, szNew);
+}
+
+
+/** Convert null separated double null terimated string to LPTSTR array)
+ * returns array size
+ */
+DWORD
+apxMultiSzToArrayW(APXHANDLE hPool, LPCWSTR lpString, LPWSTR **lppArray)
+{
+ DWORD i, n, l;
+ char *buff;
+ LPWSTR p;
+
+ l = __apxGetMultiSzLengthW(lpString, &n);
+ if (!n || !l)
+ return 0;
+ if (IS_INVALID_HANDLE(hPool))
+ buff = apxPoolAlloc(hPool, (n + 2) * sizeof(LPWSTR) + (l + 1) * sizeof(WCHAR));
+ else
+ buff = apxAlloc((n + 2) * sizeof(LPWSTR) + (l + 1) * sizeof(WCHAR));
+
+ *lppArray = (LPWSTR *)buff;
+ p = (LPWSTR)(buff + (n + 2) * sizeof(LPWSTR));
+ AplCopyMemory(p, lpString, (l + 1) * sizeof(WCHAR) + sizeof(WCHAR));
+ for (i = 0; i < n; i++) {
+ (*lppArray)[i] = p;
+ while (*p)
+ p++;
+ p++;
+ }
+ (*lppArray)[++i] = NULL;
+
+ return n;
+}
+
+DWORD
+apxMultiSzToArrayA(APXHANDLE hPool, LPCSTR lpString, LPSTR **lppArray)
+{
+ DWORD i, n, l;
+ char *buff;
+ LPSTR p;
+
+ l = __apxGetMultiSzLengthA(lpString, &n);
+ if (!n || !l)
+ return 0;
+ if (IS_INVALID_HANDLE(hPool))
+ buff = apxPoolAlloc(hPool, (n + 2) * sizeof(LPTSTR) + (l + 1));
+ else
+ buff = apxAlloc((n + 2) * sizeof(LPSTR) + (l + 1) * sizeof(CHAR));
+
+ *lppArray = (LPSTR *)buff;
+ p = (LPSTR)(buff + (n + 2) * sizeof(LPSTR));
+ AplCopyMemory(p, lpString, (l + 1) * sizeof(CHAR) + sizeof(CHAR));
+ for (i = 0; i < n; i++) {
+ (*lppArray)[i] = p;
+ while (*p)
+ p++;
+ p++;
+ }
+ (*lppArray)[++i] = NULL;
+
+ return n;
+}
+
+#define QSTR_BOUNDARY 127
+#define QSTR_ALIGN(size) \
+ (((size) + QSTR_BOUNDARY + sizeof(APXMULTISZ) + 2) & ~(QSTR_BOUNDARY))
+
+#define QSTR_SIZE(size) \
+ ((QSTR_ALIGN(size)) - sizeof(APXMULTISZ))
+
+#define QSTR_DATA(q) ((char *)(q) + sizeof(APXMULTISZ))
+
+#if 0
+LPAPXMULTISZ apxMultiSzStrdup(LPCTSTR szSrc)
+{
+ LPAPXMULTISZ q;
+
+ if (szSrc) {
+ DWORD l = lstrlen(szSrc);
+ q = (LPAPXMULTISZ)apxAlloc(QSTR_ALIGN(l));
+ q->dwAllocated = QSTR_SIZE(l);
+ q->dwInsert = l + 1;
+ AplMoveMemory(QSTR_DATA(q), szSrc, l);
+ RtlZeroMemory(QSTR_DATA(q) + l, q->dwAllocated - l);
+ }
+ else {
+ q = (LPAPXMULTISZ)apxCalloc(QSTR_ALIGN(0));
+ q->dwAllocated = QSTR_SIZE(0);
+ }
+
+ return q;
+}
+
+LPTSTR apxMultiSzStrcat(LPAPXMULTISZ lpmSz, LPCTSTR szSrc)
+{
+ DWORD l = lstrlen(szSrc);
+ LPTSTR p;
+
+ if (lpmSz->dwInsert + l + 2 > lpmSz->dwAllocated) {
+ if ((lpmSz = (LPAPXMULTISZ )apxRealloc(lpmSz, QSTR_ALIGN(lpmSz->dwInsert + l))) == NULL)
+ return NULL;
+
+ lpmSz->dwAllocated = QSTR_SIZE(lpmSz->dwInsert + l);
+ AplZeroMemory(QSTR_DATA(lpmSz) + lpmSz->dwInsert + l,
+ lpmSz->dwAllocated - (lpmSz->dwInsert + l));
+ }
+ p = (LPTSTR)QSTR_DATA(lpmSz) + lpmSz->dwInsert;
+ AplMoveMemory(p, szSrc, l);
+
+ lpmSz->dwInsert += (l + 1);
+ return p;
+}
+
+DWORD apxMultiSzLen(LPAPXMULTISZ lpmSz)
+{
+ if (lpmSz->dwInsert)
+ return lpmSz->dwInsert - 1;
+ else
+ return 0;
+}
+
+LPCTSTR apxMultiSzGet(LPAPXMULTISZ lpmSz)
+{
+ return (LPCTSTR)QSTR_DATA(lpmSz);
+}
+#endif
+
+LPTSTR apxStrCharRemove(LPTSTR szString, TCHAR chSkip)
+{
+ LPTSTR p = szString;
+ LPTSTR q = szString;
+ while (*p) {
+ if(*p != chSkip)
+ *q++ = *p;
+ ++p;
+ }
+ *q = TEXT('\0');
+
+ return szString;
+}
+
+DWORD apxStrCharRemoveA(LPSTR szString, CHAR chSkip)
+{
+ LPSTR p = szString;
+ LPSTR q = szString;
+ DWORD c = 0;
+ while (*p) {
+ if(*p != chSkip)
+ *q++ = *p;
+ else
+ ++c;
+ ++p;
+ }
+ *q = '\0';
+
+ return c;
+}
+
+DWORD apxStrCharRemoveW(LPWSTR szString, WCHAR chSkip)
+{
+ LPWSTR p = szString;
+ LPWSTR q = szString;
+ DWORD c = 0;
+ while (*p) {
+ if(*p != chSkip)
+ *q++ = *p;
+ else
+ ++c;
+ ++p;
+ }
+ *q = L'\0';
+
+ return c;
+}
+
+void
+apxStrCharReplaceA(LPSTR szString, CHAR chReplace, CHAR chReplaceWith)
+{
+ LPSTR p = szString;
+ LPSTR q = szString;
+ while (*p) {
+ if(*p == chReplace)
+ *q++ = chReplaceWith;
+ else
+ *q++ = *p;
+ ++p;
+ }
+ *q = '\0';
+}
+
+void
+apxStrCharReplaceW(LPWSTR szString, WCHAR chReplace, WCHAR chReplaceWith)
+{
+ LPWSTR p = szString;
+ LPWSTR q = szString;
+ while (*p) {
+ if(*p == chReplace)
+ *q++ = chReplaceWith;
+ else
+ *q++ = *p;
+ ++p;
+ }
+ *q = L'\0';
+}
+
+static const LPCTSTR _st_hex = TEXT("0123456789abcdef");
+#define XTOABUFFER_SIZE (sizeof(ULONG) * 2 + 2)
+#define UTOABUFFER_SIZE (sizeof(ULONG_PTR) * 2 + 2)
+#define LO_NIBLE(x) ((BYTE)((x) & 0x0000000F))
+
+BOOL apxUltohex(ULONG n, LPTSTR lpBuff, DWORD dwBuffLength)
+{
+ LPTSTR p;
+ DWORD i;
+ *lpBuff = 0;
+ if (dwBuffLength < XTOABUFFER_SIZE)
+ return FALSE;
+ p = lpBuff + XTOABUFFER_SIZE;
+ *p-- = 0;
+ for (i = 0; i < sizeof(ULONG) * 2; i++) {
+ *p-- = _st_hex[LO_NIBLE(n)];
+ n = n >> 4;
+ }
+ *p-- = TEXT('x');
+ *p = TEXT('0');
+ return TRUE;
+}
+
+BOOL apxUptohex(ULONG_PTR n, LPTSTR lpBuff, DWORD dwBuffLength)
+{
+ LPTSTR p;
+ DWORD i;
+ *lpBuff = 0;
+ if (dwBuffLength < UTOABUFFER_SIZE)
+ return FALSE;
+ p = lpBuff + UTOABUFFER_SIZE;
+ *p-- = 0;
+ for (i = 0; i < sizeof(ULONG_PTR) * 2; i++) {
+ *p-- = _st_hex[LO_NIBLE(n)];
+ n = n >> 4;
+ }
+ *p-- = TEXT('x');
+ *p = TEXT('0');
+ return TRUE;
+}
+
+ULONG apxStrToul(LPCTSTR szNum)
+{
+ ULONG rv = 0;
+ DWORD sh = 0;
+ LPCTSTR p = szNum;
+ ++p;
+ while (*p && (*p != TEXT('x')) && (*(p - 1) != TEXT('0')))
+ p++;
+ if (*p != 'x')
+ return 0;
+ /* go to the last digit */
+ while (*(p + 1)) p++;
+
+ /* go back to 'x' */
+ while (*p != TEXT('x')) {
+ ULONG v = 0;
+ switch (*p--) {
+ case TEXT('0'): v = 0UL; break;
+ case TEXT('1'): v = 1UL; break;
+ case TEXT('2'): v = 2UL; break;
+ case TEXT('3'): v = 3UL; break;
+ case TEXT('4'): v = 4UL; break;
+ case TEXT('5'): v = 5UL; break;
+ case TEXT('6'): v = 6UL; break;
+ case TEXT('7'): v = 7UL; break;
+ case TEXT('8'): v = 8UL; break;
+ case TEXT('9'): v = 9UL; break;
+ case TEXT('a'): case TEXT('A'): v = 10UL; break;
+ case TEXT('b'): case TEXT('B'): v = 11UL; break;
+ case TEXT('c'): case TEXT('C'): v = 12UL; break;
+ case TEXT('d'): case TEXT('D'): v = 13UL; break;
+ case TEXT('e'): case TEXT('E'): v = 14UL; break;
+ case TEXT('f'): case TEXT('F'): v = 15UL; break;
+ default:
+ return 0;
+ break;
+ }
+ rv |= rv + (v << sh);
+ sh += 4;
+ }
+ return rv;
+}
+
+ULONG apxStrToulW(LPCWSTR szNum)
+{
+ ULONG rv = 0;
+ DWORD sh = 0;
+ LPCWSTR p = szNum;
+ ++p;
+ while (*p && (*p != L'x') && (*(p - 1) != L'0'))
+ p++;
+ if (*p != L'x')
+ return 0;
+ /* go to the last digit */
+ while (*(p + 1)) p++;
+
+ /* go back to 'x' */
+ while (*p != L'x') {
+ ULONG v = 0;
+ switch (*p--) {
+ case L'0': v = 0UL; break;
+ case L'1': v = 1UL; break;
+ case L'2': v = 2UL; break;
+ case L'3': v = 3UL; break;
+ case L'4': v = 4UL; break;
+ case L'5': v = 5UL; break;
+ case L'6': v = 6UL; break;
+ case L'7': v = 7UL; break;
+ case L'8': v = 8UL; break;
+ case L'9': v = 9UL; break;
+ case L'a': case L'A': v = 10UL; break;
+ case L'b': case L'B': v = 11UL; break;
+ case L'c': case L'C': v = 12UL; break;
+ case L'd': case L'D': v = 13UL; break;
+ case L'e': case L'E': v = 14UL; break;
+ case L'f': case L'F': v = 15UL; break;
+ default:
+ return 0;
+ break;
+ }
+ rv |= rv + (v << sh);
+ sh += 4;
+ }
+ return rv;
+}
+
+ULONG apxAtoulW(LPCWSTR szNum)
+{
+ ULONG rv = 0;
+ DWORD sh = 1;
+ int s = 1;
+ LPCWSTR p = szNum;
+
+ /* go to the last digit */
+ if (!p || !*p)
+ return 0;
+ if (*p == L'-') {
+ s = -1;
+ ++p;
+ }
+ while (*(p + 1)) p++;
+
+ /* go back */
+ while (p >= szNum) {
+ ULONG v = 0;
+ switch (*p--) {
+ case L'0': v = 0UL; break;
+ case L'1': v = 1UL; break;
+ case L'2': v = 2UL; break;
+ case L'3': v = 3UL; break;
+ case L'4': v = 4UL; break;
+ case L'5': v = 5UL; break;
+ case L'6': v = 6UL; break;
+ case L'7': v = 7UL; break;
+ case L'8': v = 8UL; break;
+ case L'9': v = 9UL; break;
+ default:
+ return rv * s;
+ break;
+ }
+ rv = rv + (v * sh);
+ sh = sh * 10;
+ }
+ return rv * s;
+}
+
+/* Make the unique system resource name from prefix and process id
+ *
+ */
+BOOL
+apxMakeResourceName(LPCTSTR szPrefix, LPTSTR lpBuff, DWORD dwBuffLength)
+{
+ DWORD pl = lstrlen(szPrefix);
+ if (dwBuffLength < (pl + 11))
+ return FALSE;
+ lstrcpy(lpBuff, szPrefix);
+ return apxUltohex(GetCurrentProcessId(), lpBuff + pl, dwBuffLength - pl);
+}
+/** apxStrMatchA ANSI string pattern matching
+ * Match = 0, NoMatch = 1, Abort = -1
+ * Based loosely on sections of wildmat.c by Rich Salz
+ */
+INT apxStrMatchA(LPCSTR szString, LPCSTR szPattern, BOOL bIgnoreCase)
+{
+ int x, y;
+
+ for (x = 0, y = 0; szPattern[y]; ++y, ++x) {
+ if (!szPattern[x] && (szPattern[y] != '*' || szPattern[y] != '?'))
+ return -1;
+ if (szPattern[y] == '*') {
+ while (szPattern[++y] == '*');
+ if (!szPattern[y])
+ return 0;
+ while (szString[x]) {
+ INT rc;
+ if ((rc = apxStrMatchA(&szString[x++], &szPattern[y],
+ bIgnoreCase)) != 1)
+ return rc;
+ }
+ return -1;
+ }
+ else if (szPattern[y] != '?') {
+ if (bIgnoreCase) {
+ if (CharLowerA((LPSTR)((SIZE_T)szString[x])) !=
+ CharLowerA((LPSTR)((SIZE_T)szPattern[y])))
+ return 1;
+ }
+ else {
+ if (szString[x] != szPattern[y])
+ return 1;
+ }
+ }
+ }
+ return (szString[x] != '\0');
+}
+
+INT apxStrMatchW(LPCWSTR szString, LPCWSTR szPattern, BOOL bIgnoreCase)
+{
+ int x, y;
+
+ for (x = 0, y = 0; szPattern[y]; ++y, ++x) {
+ if (!szPattern[x] && (szPattern[y] != L'*' || szPattern[y] != L'?'))
+ return -1;
+ if (szPattern[y] == L'*') {
+ while (szPattern[++y] == L'*');
+ if (!szPattern[y])
+ return 0;
+ while (szString[x]) {
+ INT rc;
+ if ((rc = apxStrMatchW(&szString[x++], &szPattern[y],
+ bIgnoreCase)) != 1)
+ return rc;
+ }
+ return -1;
+ }
+ else if (szPattern[y] != L'?') {
+ if (bIgnoreCase) {
+ if (CharLowerW((LPWSTR)((SIZE_T)szString[x])) !=
+ CharLowerW((LPWSTR)((SIZE_T)szPattern[y])))
+ return 1;
+ }
+ else {
+ if (szString[x] != szPattern[y])
+ return 1;
+ }
+ }
+ }
+ return (szString[x] != L'\0');
+}
+
+INT apxMultiStrMatchW(LPCWSTR szString, LPCWSTR szPattern,
+ WCHAR chSeparator, BOOL bIgnoreCase)
+{
+ WCHAR szM[SIZ_HUGLEN];
+ DWORD i = 0;
+ LPCWSTR p = szPattern;
+ INT m = -1;
+
+ if (chSeparator == 0)
+ return apxStrMatchW(szString, szPattern, bIgnoreCase);
+ while (*p != L'\0') {
+ if (*p == chSeparator) {
+ m = apxStrMatchW(szString, szM, bIgnoreCase);
+ if (m == 0)
+ return 0;
+ p++;
+ i = 0;
+ szM[0] = L'\0';
+ }
+ else {
+ if (i < SIZ_HUGMAX)
+ szM[i++] = *p++;
+ else
+ return -1;
+ }
+ }
+ szM[i] = L'\0';
+ if (szM[0])
+ return apxStrMatchW(szString, szM, bIgnoreCase);
+ else
+ return m;
+}
+
+LPSTR apxArrayToMultiSzA(APXHANDLE hPool, DWORD nArgs, LPCSTR *lpArgs)
+{
+ DWORD i, l = 0;
+ LPSTR lpSz, p;
+ if (!nArgs)
+ return NULL;
+ for (i = 0; i < nArgs; i++)
+ l += lstrlenA(lpArgs[i]);
+ l += (nArgs + 2);
+
+ p = lpSz = (LPSTR)apxPoolAlloc(hPool, l);
+ for (i = 0; i < nArgs; i++) {
+ lstrcpyA(p, lpArgs[i]);
+ p += lstrlenA(lpArgs[i]);
+ *p++ = '\0';
+ }
+ *p++ = '\0';
+ *p++ = '\0';
+ return lpSz;
+}
+
+void apxStrQuoteInplaceW(LPWSTR szString)
+{
+ LPWSTR p = szString;
+ BOOL needsQuote = FALSE;
+ while (*p) {
+ if (*p++ == L' ') {
+ needsQuote = TRUE;
+ break;
+ }
+ }
+ if (needsQuote) {
+ DWORD l = lstrlenW(szString);
+ AplMoveMemory(&szString[1], szString, l * sizeof(WCHAR));
+ szString[0] = L'"';
+ szString[++l] = L'"';
+ szString[++l] = L'\0';
+ }
+}
+
+DWORD apxStrUnQuoteInplaceA(LPSTR szString)
+{
+ LPSTR p = szString;
+ BOOL needsQuote = FALSE;
+ BOOL inQuote = FALSE;
+ while (*p) {
+ if (*p == '"') {
+ if (inQuote)
+ break;
+ else
+ inQuote = TRUE;
+ }
+ else if (*p == ' ') {
+ if (inQuote) {
+ needsQuote = TRUE;
+ break;
+ }
+ }
+ ++p;
+ }
+ if (!needsQuote)
+ return apxStrCharRemoveA(szString, '"');
+ else
+ return 0;
+}
+
+DWORD apxStrUnQuoteInplaceW(LPWSTR szString)
+{
+ LPWSTR p = szString;
+ BOOL needsQuote = FALSE;
+ BOOL inQuote = FALSE;
+ while (*p) {
+ if (*p == L'"') {
+ if (inQuote)
+ break;
+ else
+ inQuote = TRUE;
+ }
+ else if (*p == L' ') {
+ if (inQuote) {
+ needsQuote = TRUE;
+ break;
+ }
+ }
+ ++p;
+ }
+ if (!needsQuote)
+ return apxStrCharRemoveW(szString, L'"');
+ else
+ return 0;
+}
+
+LPWSTR
+apxMszToCRLFW(APXHANDLE hPool, LPCWSTR szStr)
+{
+ DWORD l, c;
+ LPWSTR rv, b;
+ LPCWSTR p = szStr;
+
+ l = __apxGetMultiSzLengthW(szStr, &c);
+ b = rv = apxPoolCalloc(hPool, (l + c + 2) * sizeof(WCHAR));
+
+ while (c > 0) {
+ if (*p)
+ *b++ = *p;
+ else {
+ *b++ = L'\r';
+ *b++ = L'\n';
+ c--;
+ }
+ p++;
+ }
+ return rv;
+}
+
+LPWSTR
+apxCRLFToMszW(APXHANDLE hPool, LPCWSTR szStr, LPDWORD lpdwBytes)
+{
+ DWORD l, c, n = 0;
+ LPWSTR rv, b;
+
+ l = lstrlenW(szStr);
+ b = rv = apxPoolCalloc(hPool, (l + 2) * sizeof(WCHAR));
+ for (c = 0; c < l; c++) {
+ if (szStr[c] == L'\r') {
+ *b++ = '\0';
+ n++;
+ }
+ else if (szStr[c] != L'\n') {
+ *b++ = szStr[c];
+ n++;
+ }
+ }
+ if (lpdwBytes)
+ *lpdwBytes = (n + 2) * sizeof(WCHAR);
+ return rv;
+}
+
+LPSTR
+apxExpandStrA(APXHANDLE hPool, LPCSTR szString)
+{
+ LPCSTR p = szString;
+ while (*p) {
+ if (*p == '%') {
+ p = szString;
+ break;
+ }
+ ++p;
+ }
+ if (p != szString)
+ return apxPoolStrdupA(hPool, szString);
+ else {
+ DWORD l = ExpandEnvironmentStringsA(szString, NULL, 0);
+ if (l) {
+ LPSTR rv = apxPoolAlloc(hPool, l);
+ l = ExpandEnvironmentStringsA(szString, rv, l);
+ if (l)
+ return rv;
+ else {
+ apxFree(rv);
+ return NULL;
+ }
+ }
+ else
+ return NULL;
+ }
+}
+
+LPWSTR
+apxExpandStrW(APXHANDLE hPool, LPCWSTR szString)
+{
+ LPCWSTR p = szString;
+ while (*p) {
+ if (*p == L'%') {
+ p = szString;
+ break;
+ }
+ ++p;
+ }
+ if (p != szString)
+ return apxPoolStrdupW(hPool, szString);
+ else {
+ DWORD l = ExpandEnvironmentStringsW(szString, NULL, 0);
+ if (l) {
+ LPWSTR rv = apxPoolAlloc(hPool, l * sizeof(WCHAR));
+ l = ExpandEnvironmentStringsW(szString, rv, l);
+ if (l)
+ return rv;
+ else {
+ apxFree(rv);
+ return NULL;
+ }
+ }
+ else
+ return NULL;
+ }
+}
+
+/* To share the semaphores with other processes, we need a NULL ACL
+ * Code from MS KB Q106387
+ */
+PSECURITY_ATTRIBUTES GetNullACL()
+{
+ PSECURITY_DESCRIPTOR pSD;
+ PSECURITY_ATTRIBUTES sa;
+
+ sa = (PSECURITY_ATTRIBUTES) LocalAlloc(LPTR, sizeof(SECURITY_ATTRIBUTES));
+ sa->nLength = sizeof(sizeof(SECURITY_ATTRIBUTES));
+
+ pSD = (PSECURITY_DESCRIPTOR) LocalAlloc(LPTR, SECURITY_DESCRIPTOR_MIN_LENGTH);
+ sa->lpSecurityDescriptor = pSD;
+
+ if (pSD == NULL || sa == NULL) {
+ return NULL;
+ }
+ SetLastError(0);
+ if (!InitializeSecurityDescriptor(pSD, SECURITY_DESCRIPTOR_REVISION)
+ || GetLastError()) {
+ LocalFree( pSD );
+ LocalFree( sa );
+ return NULL;
+ }
+ if (!SetSecurityDescriptorDacl(pSD, TRUE, (PACL) NULL, FALSE)
+ || GetLastError()) {
+ LocalFree( pSD );
+ LocalFree( sa );
+ return NULL;
+ }
+
+ sa->bInheritHandle = FALSE;
+ return sa;
+}
+
+
+void CleanNullACL(void *sa)
+{
+ if (sa) {
+ LocalFree(((PSECURITY_ATTRIBUTES)sa)->lpSecurityDescriptor);
+ LocalFree(sa);
+ }
+}
More information about the jbossnative-commits
mailing list