You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
2.1 KiB
74 lines
2.1 KiB
// stdafx.h : include file for standard system include files,
|
|
// or project specific include files that are used frequently,
|
|
// but are changed infrequently
|
|
|
|
#if !defined(AFX_STDAFX_H__985A2EB9_ECE0_4406_AB2C_1B7B05DCDA32__INCLUDED_)
|
|
#define AFX_STDAFX_H__985A2EB9_ECE0_4406_AB2C_1B7B05DCDA32__INCLUDED_
|
|
|
|
#if _MSC_VER > 1000
|
|
#pragma once
|
|
#endif // _MSC_VER > 1000
|
|
|
|
#define STRICT
|
|
#ifndef _WIN32_WINNT
|
|
#define _WIN32_WINNT 0x0400
|
|
#endif
|
|
#define _ATL_APARTMENT_THREADED
|
|
|
|
#include <atlbase.h>
|
|
//You may derive a class from CComModule and use it if you want to override
|
|
//something, but do not change the name of _Module
|
|
|
|
class CServiceModule : public CComModule
|
|
{
|
|
public:
|
|
HRESULT RegisterServer(BOOL bRegTypeLib, BOOL bService);
|
|
HRESULT UnregisterServer();
|
|
void Init(_ATL_OBJMAP_ENTRY* p, HINSTANCE h, UINT nServiceNameID, const GUID* plibid = NULL);
|
|
void Start();
|
|
void ServiceMain(DWORD dwArgc, LPTSTR* lpszArgv);
|
|
void Handler(DWORD dwOpcode);
|
|
void Run();
|
|
BOOL IsInstalled();
|
|
BOOL Install();
|
|
BOOL Uninstall();
|
|
LONG Unlock();
|
|
void LogEvent(LPCTSTR pszFormat, ...);
|
|
void SetServiceStatus(DWORD dwState);
|
|
void SetupAsLocalServer();
|
|
BOOL GetRegData();
|
|
|
|
//Implementation
|
|
private:
|
|
static void WINAPI _ServiceMain(DWORD dwArgc, LPTSTR* lpszArgv);
|
|
static void WINAPI _Handler(DWORD dwOpcode);
|
|
|
|
// data members
|
|
public:
|
|
TCHAR m_szServiceName[256];
|
|
SERVICE_STATUS_HANDLE m_hServiceStatus;
|
|
SERVICE_STATUS m_status;
|
|
DWORD dwThreadID;
|
|
BOOL m_bService;
|
|
|
|
private:
|
|
|
|
void RenameCabFile(TCHAR * tFileName);
|
|
void SearchRootDirectory(void);
|
|
int SearchDirectory(TCHAR * tzDirectory);
|
|
BOOL GetResponseURL(wchar_t *RemoteFileName);
|
|
};
|
|
|
|
BOOL
|
|
RenameFile(TCHAR *CurrentPath,
|
|
TCHAR *CurrentName,
|
|
TCHAR *NewName);
|
|
|
|
#define s_cszStopEvent _T("StressSvc_Event")
|
|
extern CServiceModule _Module;
|
|
#include <atlcom.h>
|
|
|
|
//{{AFX_INSERT_LOCATION}}
|
|
// Microsoft Visual C++ will insert additional declarations immediately before the previous line.
|
|
|
|
#endif // !defined(AFX_STDAFX_H__985A2EB9_ECE0_4406_AB2C_1B7B05DCDA32__INCLUDED)
|