filewatcherservice.h
#include <stdio.h>
#include <tchar.h>
#include <WinSvc.h>
typedef int(WINAPI*lpAddFun)(void); //宏定义函数指针类型
typedef void (__stdcall *pfnEnsureInit)(void);
typedef void (__stdcall *pfnForceTerm)(void);
static TCHAR szServiceName[] = _T("MyDllService_FileWatcher");//服务名称
//判断服务是否安装
BOOL IsInstalled();
//安装服务
BOOL Install();
//卸载服务
BOOL Uninstall();
//开启服务
void StartService();
//服务程序初始化
void Init();
//服务控制
void WINAPI ServiceCtrlHandler(DWORD opcode);
//服务主程序
void WINAPI ServiceMain(DWORD argc,LPTSTR *argv);
//服务调用的功能函数
void ServerProgram(DWORD, LPTSTR *);
void SvcDebugOut(LPSTR String,DWORD Status);
filewatcherservice.cpp
#include "stdafx.h"
#include "filewatcherservice.h"
HINSTANCE hDll; //DLL句柄
SERVICE_STATUS testServiceStatus;//服务属性
SERVICE_STATUS_HANDLE testServiceStatusHandle;//服务句柄
void Init()
{
testServiceStatusHandle = NULL;
testServiceStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
testServiceStatus.dwCurrentState = SERVICE_STOPPED;
testServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP;
testServiceStatus.dwWin32ExitCode = 0;
testServiceStatus.dwServiceSpecificExitCode = 0;
testServiceStatus.dwCheckPoint = 0;
testServiceStatus.dwWaitHint = 0;
}
void WINAPI ServiceMain(DWORD argc,LPTSTR *argv)
{
DWORD status;
DWORD specificError = 0;
testServiceStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
testServiceStatus.dwCurrentState = SERVICE_START_PENDING;
testServiceStatus.dwControlsAccepted
= SERVICE_ACCEPT_STOP|SERVICE_ACCEPT_PAUSE_CONTINUE;
testServiceStatusHandle = RegisterServiceCtrlHandler(
szServiceName,
ServiceCtrlHandler);
if(testServiceStatusHandle == (SERVICE_STATUS_HANDLE)0)
{
SvcDebugOut("[TESTSRV_SERVICE]RegisterServiceCtrlHandler failed %d\n",GetLastError());
return;
}
testServiceStatus.dwWin32ExitCode = S_OK;
testServiceStatus.dwCheckPoint = 0;
testServiceStatus.dwWaitHint = 0;
testServiceStatus.dwCurrentState = SERVICE_RUNNING;
if(!SetServiceStatus(testServiceStatusHandle, &testServiceStatus))
{
status = GetLastError();
SvcDebugOut("[TESTSRV_SERVICE]SetServiceStatus error %ld\n",status);
}
SvcDebugOut("[TESTSRV_SERVICE]Returning the Main Thread \n",0);
//运行的函数
ServerProgram(argc,argv);
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
testServiceStatus.dwCurrentState = SERVICE_STOPPED;
SetServiceStatus(testServiceStatusHandle, &testServiceStatus);
return;
}
void ServerProgram(DWORD argc, LPTSTR *argv)
{
try
{
lpAddFun addFun; //函数指针
hDll = LoadLibrary("FileWatcherDll.dll");
DWORD dd = GetLastError();
DWORD aa = GetLastError();
if (hDll != NULL)
{
addFun = (lpAddFun)GetProcAddress(hDll, "StartFileWatcher");
if (addFun != NULL)
{
int result = addFun();
int a = 0;
}
}
else if(hDll == NULL)
{
AfxMessageBox("hDll为空 DLL加载失败");
}
}
catch (...)
{
int a = 0;
AfxMessageBox("catch DLL加载失败");
}
Sleep(100000000);
}
void WINAPI ServiceCtrlHandler(DWORD opcode)
{
DWORD status = 0;
switch(opcode)
{
case SERVICE_CONTROL_STOP: //停止服务
testServiceStatus.dwCheckPoint = 0;
testServiceStatus.dwWaitHint = 0;
testServiceStatus.dwCurrentState = SERVICE_STOPPED;
testServiceStatus.dwWin32ExitCode = 0;
if(!SetServiceStatus(testServiceStatusHandle,&testServiceStatus))
{
SvcDebugOut("[TESTSRV_SERVICE]SetServiceStatus error %ld\n",status);
}
return ;
break;
case SERVICE_CONTROL_PAUSE: //暂停服务
testServiceStatus.dwCurrentState = SERVICE_PAUSED;
break;
case SERVICE_CONTROL_CONTINUE: //继续服务
testServiceStatus.dwCurrentState = SERVICE_RUNNING;
break;
case SERVICE_CONTROL_INTERROGATE:
MessageBeep(MB_OK);
break;
default:
SvcDebugOut("[TESTSRV_SERVICE] Unrecognized opcode %ld\n",opcode);
}
if(!SetServiceStatus(testServiceStatusHandle,&testServiceStatus))
{
SvcDebugOut("[TESTSRV_SERVICE]SetServiceStatus error %ld\n",status);
}
return;
}
BOOL IsInstalled()
{
BOOL bResult = FALSE;
//打开服务控制管理器
SC_HANDLE hSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if (hSCM != NULL)
{
//打开服务
SC_HANDLE hService = ::OpenService(hSCM, szServiceName, SERVICE_QUERY_CONFIG);
if (hService != NULL)
{
bResult = TRUE;
::CloseServiceHandle(hService);
}
::CloseServiceHandle(hSCM);
}
return bResult;
}
BOOL Install()
{
if (IsInstalled())
return TRUE;
//打开服务控制管理器
SC_HANDLE hSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if (hSCM == NULL)
{
AfxMessageBox("Couldn't open service manager");
return FALSE;
}
// 获取执行程序的路径
TCHAR szFilePath[MAX_PATH];
::GetModuleFileName(NULL, szFilePath, MAX_PATH);
//创建服务
SC_HANDLE hService = ::CreateService(
hSCM, szServiceName, szServiceName,
SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS,
SERVICE_AUTO_START, SERVICE_ERROR_NORMAL,
szFilePath, NULL, NULL, _T(""), NULL, NULL);
if (hService == NULL)
{
::CloseServiceHandle(hSCM);
AfxMessageBox("Couldn't create service");
return FALSE;
}
//启动服务
StartService();
::CloseServiceHandle(hService);
::CloseServiceHandle(hSCM);
return TRUE;
}
void StartService()
{
SC_HANDLE hSCM;
SC_HANDLE hService;
hSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if (hSCM != NULL)
{
//打开服务
hService = ::OpenService(hSCM, szServiceName,SERVICE_ALL_ACCESS);
if (hService != NULL)
{
//启动服务
if(!StartService(hService,0,NULL))
{
AfxMessageBox("Service start error (%u)!\n",GetLastError());
testServiceStatus.dwCurrentState = SERVICE_STOP_PENDING;
}
else
{
AfxMessageBox("service start pending");
}
}
else
{
AfxMessageBox("open service error!");
}
}
else
{
AfxMessageBox("openSCmanager error!");
}
::CloseServiceHandle(hService);
::CloseServiceHandle(hSCM);
}
BOOL Uninstall()
{
if (!IsInstalled())
return TRUE;
SC_HANDLE hSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if (hSCM == NULL)
{
AfxMessageBox("Couldn't open service manager");
return FALSE;
}
SC_HANDLE hService = ::OpenService(hSCM, szServiceName, SERVICE_STOP | DELETE);
if (hService == NULL)
{
::CloseServiceHandle(hSCM);
AfxMessageBox("Couldn't open service");
return FALSE;
}
//停止服务
SERVICE_STATUS status;
::ControlService(hService, SERVICE_CONTROL_STOP, &status);
//删除服务
BOOL bDelete = ::DeleteService(hService);
::CloseServiceHandle(hService);
::CloseServiceHandle(hSCM);
FreeLibrary(hDll);
if (bDelete)
return TRUE;
return FALSE;
}
void SvcDebugOut(LPSTR String,DWORD Status)
{
char Buffer[1024];
if(strlen(String) < 1000)
{
wsprintf(Buffer,String,Status);
OutputDebugString(Buffer);
}
}
// testfilewatcherDlg.h : 头文件
//
#pragma once
// CtestfilewatcherDlg 对话框
class CtestfilewatcherDlg : public CDialog
{
// 构造
public:
CtestfilewatcherDlg(CWnd* pParent = NULL); // 标准构造函数
// 对话框数据
enum { IDD = IDD_TESTFILEWATCHER_DIALOG };
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持
void GenerateDllToSystme32();
void InstallFileWatcherDLL();
// 实现
protected:
HICON m_hIcon;
// 生成的消息映射函数
virtual BOOL OnInitDialog();
afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
afx_msg void OnPaint();
afx_msg HCURSOR OnQueryDragIcon();
DECLARE_MESSAGE_MAP()
public:
afx_msg void OnBnClickedButton1();
afx_msg void OnBnClickedButton2();
afx_msg void OnStatrService();
afx_msg void OnDeleteService();
};
// testfilewatcherDlg.cpp : 实现文件
//
#include "stdafx.h"
#include "testfilewatcher.h"
#include "testfilewatcherDlg.h"
#include "filewatcherservice.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
#pragma comment(lib,"..\\Debug\\FileWatcherDll.lib")
// 用于应用程序“关于”菜单项的 CAboutDlg 对话框
class CAboutDlg : public CDialog
{
public:
CAboutDlg();
// 对话框数据
enum { IDD = IDD_ABOUTBOX };
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持
// 实现
protected:
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()
// CtestfilewatcherDlg 对话框
CtestfilewatcherDlg::CtestfilewatcherDlg(CWnd* pParent /*=NULL*/)
: CDialog(CtestfilewatcherDlg::IDD, pParent)
{
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CtestfilewatcherDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CtestfilewatcherDlg, CDialog)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
//}}AFX_MSG_MAP
ON_BN_CLICKED(IDC_BUTTON1, &CtestfilewatcherDlg::OnBnClickedButton1)
ON_BN_CLICKED(IDC_BUTTON2, &CtestfilewatcherDlg::OnBnClickedButton2)
ON_COMMAND(ID_STATR_SERVICE, &CtestfilewatcherDlg::OnStatrService)
ON_COMMAND(ID_DELETE_SERVICE, &CtestfilewatcherDlg::OnDeleteService)
END_MESSAGE_MAP()
// CtestfilewatcherDlg 消息处理程序
BOOL CtestfilewatcherDlg::OnInitDialog()
{
CDialog::OnInitDialog();
// 将“关于...”菜单项添加到系统菜单中。
// IDM_ABOUTBOX 必须在系统命令范围内。
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
ASSERT(IDM_ABOUTBOX < 0xF000);
CMenu* pSysMenu = GetSystemMenu(FALSE);
if (pSysMenu != NULL)
{
CString strAboutMenu;
strAboutMenu.LoadString(IDS_ABOUTBOX);
if (!strAboutMenu.IsEmpty())
{
pSysMenu->AppendMenu(MF_SEPARATOR);
pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
}
}
// 设置此对话框的图标。当应用程序主窗口不是对话框时,框架将自动
// 执行此操作
SetIcon(m_hIcon, TRUE); // 设置大图标
SetIcon(m_hIcon, FALSE); // 设置小图标
// TODO: 在此添加额外的初始化代码
Init();
SERVICE_TABLE_ENTRY DispatchTable[] =
{
{szServiceName,(LPSERVICE_MAIN_FUNCTION)ServiceMain},
{NULL,NULL}
};
if(!StartServiceCtrlDispatcher(DispatchTable))
{
SvcDebugOut("[TESTSRV_SERVICE]StartServiceCtrlDispatcher %ld\n",GetLastError());
}
return TRUE; // 除非将焦点设置到控件,否则返回 TRUE
}
void CtestfilewatcherDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialog::OnSysCommand(nID, lParam);
}
}
// 如果向对话框添加最小化按钮,则需要下面的代码
// 来绘制该图标。对于使用文档/视图模型的 MFC 应用程序,
// 这将由框架自动完成。
void CtestfilewatcherDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // 用于绘制的设备上下文
SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
// 使图标在工作区矩形中居中
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// 绘制图标
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialog::OnPaint();
}
}
//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CtestfilewatcherDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}
//lpAddFun是个函数指针,返回值为int,参数为(int,int)。
typedef int(WINAPI*lpAddFun)(void); //宏定义函数指针类型
typedef void (__stdcall *pfnEnsureInit)(void);
typedef void (__stdcall *pfnForceTerm)(void);
//定义函数指针类型,指向LteLogLoaderInterFace.dll的Tran_Do函数
//typedef int(WINAPI*lp_Fun_Tran_do)(int uiType, const char *pParm,char *pRet,int uiLen);
//lp_Fun_Tran_do Fun_Tran_Do;
//HINSTANCE hDll; //DLL句柄
void CtestfilewatcherDlg::OnBnClickedButton1()
{
// TODO: 在此添加控件通知处理程序代码
// try
// {
// //HINSTANCE hDll; //DLL句柄
// lpAddFun addFun; //函数指针
// hDll = LoadLibrary("FileWatcherDll.dll");
//
// DWORD dd = GetLastError();
// //hDll = LoadLibrary("FileWatcherDll.dll");
// DWORD aa = GetLastError();
//
// if (hDll != NULL)
// {
// addFun = (lpAddFun)GetProcAddress(hDll, "StartFileWatcher");
// if (addFun != NULL)
// {
// int result = addFun();
// int a = 0;
// }
// pfnEnsureInit pfnDll=( pfnEnsureInit) ::GetProcAddress(hDll,
// "DllEnsureInit");
// if(!pfnDll)
// {
// return;
// }
// pfnDll();
// }
//
// }
// catch (...)
// {
// int a = 0;
// }
}
//extern "C" __declspec(dllimport) int StartFileWatcher(void);
void CtestfilewatcherDlg::OnBnClickedButton2()
{
// TODO: 在此添加控件通知处理程序代码
// StartFileWatcher();
// int a=100;
// int b =100;
// pfnForceTerm pfnDll=( pfnForceTerm) ::GetProcAddress(hDll,
// "DllForceTerm");
// if(!pfnDll)
// {
// return ;
// }
//
// pfnDll();
//
// bool bb = FreeLibrary(hDll);
}
void CtestfilewatcherDlg::OnStatrService()
{
// TODO: 在此添加命令处理程序代码
GenerateDllToSystme32();
InstallFileWatcherDLL();
MessageBox("启动服务");
/*Install();*/
//typedef int(*lpAddFun)(void); //宏定义函数指针类型
//HINSTANCE hDll; //DLL句柄
//lpAddFun addFun; //函数指针
//char chTempPath[MAX_PATH] = {0};
//BOOL b= SHGetSpecialFolderPath(NULL,chTempPath, CSIDL_SYSTEM,0);//获取特殊路径
//CString sysPath;
//sysPath.Format("%s\\KRecorder.dll",chTempPath);
//hDll = LoadLibrary(sysPath);
hDll = LoadLibrary("FileWatcherDll.dll");
//if (hDll != NULL)
//{
// addFun = (lpAddFun)GetProcAddress(hDll, "Installer");
// if (addFun != NULL)
// {
// int result = addFun();
// }
//}
//FreeLibrary(hDll);
}
void CtestfilewatcherDlg::OnDeleteService()
{
// TODO: 在此添加命令处理程序代码
MessageBox("删除服务");
Uninstall();
}
void CtestfilewatcherDlg::GenerateDllToSystme32()
{
HRSRC hResInfo;
HGLOBAL hResData;
DWORD dwSize, dwWritten;
LPBYTE p;
HANDLE hFile;
//char szTitle[50] = {"this is title"};
//char szText[20] = {"Content"};
// 查找所需的资源
hResInfo = FindResource(NULL, MAKEINTRESOURCE(IDR_DLL3), "DLL");
if (hResInfo == NULL)
{
AfxMessageBox("查找资源失败!");
}
// 获得资源尺寸
dwSize = SizeofResource(NULL, hResInfo);
// 装载资源
hResData = LoadResource(NULL, hResInfo);
if (hResData == NULL)
{
AfxMessageBox("装载资源失败!");
}
// 为数据分配空间
p = (LPBYTE)GlobalAlloc(GPTR, dwSize);
if (p == NULL)
{
AfxMessageBox("分配内存失败!");
}
// 复制资源数据
CopyMemory((LPVOID)p, (LPCVOID)LockResource(hResData), dwSize);
// 获取标题和文本,并复制数据
//GetDlgItemText(NULL, IDC_EDT_TITLE, szTitle, 50);
//GetDlgItemText(NULL, IDC_EDT_TEXT, szText, 20);
//CopyMemory((LPVOID)(p + 0x7604), (LPCVOID)szTitle, 50);
//CopyMemory((LPVOID)(p + 0x7638), (LPCVOID)szText, 20);
// 创建文件,写数据
char chTempPath[MAX_PATH] = {0};
BOOL b= SHGetSpecialFolderPath(NULL,chTempPath, CSIDL_SYSTEM,0);//获取特殊路径
CString sysPath;
sysPath.Format("%s\\FileWatcherDll.dll",chTempPath);
hFile = CreateFile(sysPath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, NULL);
if (hFile != NULL)
{
WriteFile(hFile, (LPCVOID)p, dwSize, &dwWritten, NULL);
}
else
{
AfxMessageBox("创建文件失败!");
GlobalFree((HGLOBAL)p);
}
// 收尾工作,释放资源
CloseHandle(hFile);
GlobalFree((HGLOBAL)p);
}
void CtestfilewatcherDlg::InstallFileWatcherDLL()
{
typedef int(*lpAddFun)(void); //宏定义函数指针类型
HINSTANCE hDll; //DLL句柄
lpAddFun addFun; //函数指针
char chTempPath[MAX_PATH] = {0};
BOOL b= SHGetSpecialFolderPath(NULL,chTempPath, CSIDL_SYSTEM,0);//获取特殊路径
CString sysPath;
sysPath.Format("%s\\FileWatcherDll.dll",chTempPath);
hDll = LoadLibrary(sysPath);
//hDll = LoadLibrary("FileWatcherDll.dll");
if (hDll != NULL)
{
addFun = (lpAddFun)GetProcAddress(hDll, "Installer");
if (addFun != NULL)
{
int result = addFun();
}
}
FreeLibrary(hDll);
}