140 lines
5.1 KiB
C
140 lines
5.1 KiB
C
// ----------------------------------------------------------------------------
|
|
//
|
|
// _UMClnt.h
|
|
//
|
|
// Client definition for Utility Manager
|
|
//
|
|
// Author: J. Eckhardt, ECO Kommunikation
|
|
// (c) 1997-99 Microsoft
|
|
//
|
|
// History: created oct-98 by JE
|
|
// JE nov-15-98: removed any code related to key hook
|
|
// JE nov-15-98: changed UMDialog message to be a service control message
|
|
// JE nov-15 98: changed "umc_machine_ts" to save memory
|
|
// JE nov-15 98: changed "umc_machine_ts" to support launch specific client
|
|
// JE nov-15 98: changed "umclient_ts" for multiple instances support
|
|
// YX jun-01 99: added DisplayName member to the umc_machine_ts
|
|
// YX jun-23 99: added IsAdmin function
|
|
// ----------------------------------------------------------------------------
|
|
#ifndef __UMCLNT_H_
|
|
#define __UMCLNT_H_
|
|
#include "UtilMan.h"
|
|
// ---------------------------------
|
|
// HKLM\Software\Microsoft\Windows NT\CurrentVersion\Accessibility\Utility Manager\[Application Name]
|
|
// ---------------------------------
|
|
#define ACC_KEY_NONE -1
|
|
typedef struct
|
|
{
|
|
WCHAR ApplicationName[MAX_APPLICATION_NAME_LEN];
|
|
WCHAR DisplayName[MAX_APPLICATION_NAME_LEN]; // YX: added for localization purposes
|
|
DWORD ApplicationType;//APPLICATION_TYPE_xxx
|
|
DWORD WontRespondTimeout;//NO_WONTRESPONDTIMEOUT or up to MAX_WONTRESPONDTIMEOUT (sec)
|
|
DWORD MaxRunCount;// instances (only a byte in registry)
|
|
DWORD ClientControlCode;//JE nov-15 98
|
|
WPARAM AcceleratorKey; // micw - the accelerator key for this applet
|
|
} umc_machine_ts,*umc_machine_tsp;
|
|
// ---------------------------------
|
|
// HKCU\Software\Microsoft\Windows NT\CurrentVersion\Accessibility\Utility Manager\[Application Name]
|
|
typedef struct
|
|
{
|
|
BOOL fCanRunSecure;
|
|
BOOL fStartWithUtilityManager;
|
|
BOOL fStartAtLogon;
|
|
BOOL fStartOnLockDesktop;
|
|
BOOL fRestartOnDefaultDesk;
|
|
} umc_user_ts, *umc_user_tsp;
|
|
// ---------------------------------
|
|
// internal client struct (for each instance)
|
|
#define UM_CLIENT_NOT_RUNNING 0
|
|
#define UM_CLIENT_RUNNING 1
|
|
#define UM_CLIENT_NOT_RESPONDING 2
|
|
typedef struct
|
|
{
|
|
umc_machine_ts machine;
|
|
umc_user_ts user;
|
|
DWORD runCount;// number of instance
|
|
DWORD state;
|
|
DWORD processID[MAX_APP_RUNCOUNT];
|
|
HANDLE hProcess[MAX_APP_RUNCOUNT];
|
|
DWORD mainThreadID[MAX_APP_RUNCOUNT];
|
|
DWORD lastResponseTime[MAX_APP_RUNCOUNT];
|
|
} umclient_ts, *umclient_tsp;
|
|
// ---------------------------------
|
|
// header structure
|
|
#define START_BY_OTHER 0x0
|
|
#define START_BY_HOTKEY 0x1
|
|
#define START_BY_MENU 0x2
|
|
|
|
typedef struct
|
|
{
|
|
DWORD numberOfClients; // number of applets being managed
|
|
DWORD dwStartMode; // one of START_BY_HOTKEY, START_BY_MENU, or START_BY_OTHER
|
|
BOOL fShowWarningAgain; // flag for showing warning dlg when started via Start menu
|
|
} umc_header_ts, *umc_header_tsp;
|
|
// ---------------------------------
|
|
// memory mapped files
|
|
#define UMC_HEADER_FILE _TEXT("UtilityManagerClientHeaderFile")
|
|
// sizeof(umc_header_ts)
|
|
#define UMC_CLIENT_FILE _TEXT("UtilityManagerClientDataFile")
|
|
// sizeof(umclient_ts) * (umc_header_tsp)->numberOfClients
|
|
// ---------------------------------
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
BOOL StartClient(HWND hParent,umclient_tsp client);
|
|
BOOL StopClient(umclient_tsp client);
|
|
BOOL StartApplication(LPTSTR pszPath, LPTSTR pszArg, BOOL fIsTrusted,
|
|
DWORD *pdwProcessId, HANDLE *phProcess, DWORD *pdwThreadId);
|
|
BOOL GetClientApplicationPath(LPTSTR ApplicationName, LPTSTR ApplicationPath,DWORD len);
|
|
BOOL GetClientErrorOnLaunch(LPTSTR ApplicationName, LPTSTR ErrorOnLaunch,DWORD len);
|
|
BOOL CheckStatus(umclient_tsp c, DWORD cClients);
|
|
BOOL IsAdmin();
|
|
BOOL IsInteractiveUser();
|
|
BOOL IsSystem();
|
|
HANDLE GetUserAccessToken(BOOL fNeedImpersonationToken, BOOL *fError);
|
|
BOOL TestServiceClientRuns(umclient_tsp client,SERVICE_STATUS *ssStatus);
|
|
|
|
// Helpers to start up the utilman instance that displays UI
|
|
|
|
extern HANDLE g_hUIProcess;
|
|
|
|
__inline void OpenUManDialogOutOfProc()
|
|
{
|
|
TCHAR szUtilmanPath[_MAX_PATH+64] = {0};
|
|
if (GetModuleFileName(NULL, szUtilmanPath, _MAX_PATH+64))
|
|
{
|
|
// This function is called (when there is an interactive user) to bring up
|
|
// the utilman UI in the user's security context. This avoids the problem
|
|
// where a non-trusted application could send a message to utilman and cause
|
|
// some process to start as SYSTEM. In this context, utilman is not considered
|
|
// trusted; it must start as the interactive user or not at all.
|
|
StartApplication(szUtilmanPath, TEXT("/start"), FALSE, NULL, &g_hUIProcess, NULL);
|
|
}
|
|
}
|
|
__inline HANDLE GetUIUtilman()
|
|
{
|
|
return g_hUIProcess;
|
|
}
|
|
__inline BOOL ResetUIUtilman()
|
|
{
|
|
// This process detected the switch and should quit on its own
|
|
if (g_hUIProcess)
|
|
{
|
|
CloseHandle(g_hUIProcess);
|
|
g_hUIProcess = 0;
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
__inline void SetUIUtilman(HANDLE hProcess)
|
|
{
|
|
ResetUIUtilman();
|
|
g_hUIProcess = hProcess;
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
#endif __UMCLNT_H_
|