如何解决Windows服务未在ASIO阻止套接字中停止
我有一个Windows服务,改编自
可以在执行简单任务时正常启动/停止,例如 像本例中一样,在while循环中写入文本文件
#include <windows.h>
#include <string>
#include "service.hh"
#include "log.hh"
LPSTR service_name = "_service_test";
std::string log_program_name("M:\\" + std::string(service_name));
VOID WINAPI service_main(DWORD,LPTSTR*);
void write_txt_file(const std::string& file_name,const std::string& input);
SERVICE_STATUS_HANDLE service_handle = NULL;
HANDLE service_stop_event = INVALID_HANDLE_VALUE;
DWORD current_state = 0;
/////////////////////////////////////////////////////////////////////////////////////////////////////
//main
//create service in administrator shell with
//sc create _service_test binPath= "L:\net\build\Debug\test_service.exe"
/////////////////////////////////////////////////////////////////////////////////////////////////////
int main(int argc,char* argv[])
{
events::start_log();
//A service process has a SERVICE_TABLE_ENTRY structure for each service that it can start.
//The structure specifies the service name and a pointer to the service main function for that service.
//The main function of a service program calls the StartServiceCtrlDispatcher
//function to connect to the service control manager (SCM)
SERVICE_TABLE_ENTRY service_table[] =
{
{ service_name,(LPSERVICE_MAIN_FUNCTION)service_main },{ NULL,NULL }
};
if (StartServiceCtrlDispatcher(service_table))
{
return 0;
}
else
{
return 1;
}
}
void WINAPI service_main(DWORD argc,LPTSTR* argv)
{
service_handle = RegisterServiceCtrlHandler(service_name,service_handler);
if (service_handle == NULL)
{
return;
}
report_status(SERVICE_START_PENDING);
events::log("service start pending..." + std::to_string(current_state));
service_stop_event = CreateEvent(NULL,TRUE,FALSE,NULL);
if (service_stop_event == NULL)
{
return;
}
report_status(SERVICE_RUNNING);
events::log("service running..." + std::to_string(current_state));
int i = 0;
while (WaitForSingleObject(service_stop_event,1000) != WAIT_OBJECT_0)
{
write_txt_file("M:\\log.txt","writing...#" + std::to_string(i));
Sleep(1000);
i++;
}
report_status(SERVICE_STOP_PENDING);
events::log("service stop pending..." + std::to_string(current_state));
CloseHandle(service_stop_event);
report_status(SERVICE_STOPPED);
events::log("service stopped..." + std::to_string(current_state));
}
void write_txt_file(const std::string& file_name,const std::string& input)
{
FILE* f = fopen(file_name.c_str(),"a+");
fprintf(f,"%s\n",input.c_str());
fclose(f);
}
这将产生日志输出
2020-09-05,14:18:03 service start pending...2
2020-09-05,14:18:03 service running...4
2020-09-05,14:18:20 service stop pending state...4
2020-09-05,14:18:21 service stop pending...3
2020-09-05,14:18:21 service stopped...1
但是,当在阻塞套接字上将该服务用作ASIO服务器时,当请求服务停止时,它永远不会到达停止状态,只有挂起的停止状态
服务器代码缩写
asio::io_service io_service;
tcp::acceptor acceptor(io_service,tcp::endpoint(tcp::v4(),port_num));
while (WaitForSingleObject(service_stop_event,1000) != WAIT_OBJECT_0)
{
tcp::socket sock(io_service);
acceptor.accept(sock); /// blocking call
}
服务代码功能
/////////////////////////////////////////////////////////////////////////////////////////////////////
//handles
/////////////////////////////////////////////////////////////////////////////////////////////////////
extern SERVICE_STATUS_HANDLE service_handle;
extern HANDLE service_stop_event;
extern DWORD current_state;
/////////////////////////////////////////////////////////////////////////////////////////////////////
//service_handler
//handles stop service
/////////////////////////////////////////////////////////////////////////////////////////////////////
VOID WINAPI service_handler(DWORD code)
{
switch (code)
{
case SERVICE_CONTROL_SHUTDOWN:
case SERVICE_CONTROL_STOP:
events::log("service stop pending state..." + std::to_string(current_state));
report_status(SERVICE_STOP_PENDING);
SetEvent(service_stop_event);
break;
default:
events::log("service running state..." + std::to_string(current_state));
report_status(current_state);
break;
}
return;
}
/////////////////////////////////////////////////////////////////////////////////////////////////////
//report_status
/////////////////////////////////////////////////////////////////////////////////////////////////////
void report_status(DWORD state)
{
current_state = state;
SERVICE_STATUS service_status = {
SERVICE_WIN32_OWN_PROCESS,current_state,state == SERVICE_START_PENDING ? 0 : SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN,NO_ERROR,};
if (SetServiceStatus(service_handle,&service_status) == FALSE)
{
return;
}
}
解决方法
通过在服务器上发送“虚拟”自我请求解决,该请求解除了对acceptor.accept(sock)的阻止;
SetEvent(service_stop_event);
send_close(port);
关闭呼叫的位置
/////////////////////////////////////////////////////////////////////////////////////////////////////
//temporary client connection to self on localhost to unblock acceptor.accept()
/////////////////////////////////////////////////////////////////////////////////////////////////////
void http::send_close(const std::string& port_num)
{
asio::io_service io_service;
asio::ip::tcp::resolver resolver(io_service);
asio::ip::tcp::socket socket(io_service);
asio::connect(socket,resolver.resolve("127.0.0.1",port_num));
std::string request("GET /stop_service HTTP/1.1\r\n\r\n");
asio::write(socket,asio::buffer(request,request.size()));
socket.shutdown(asio::ip::tcp::socket::shutdown_both);
socket.close();
events::log("sent GET /stop_service");
}
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。