【发布时间】:2012-04-19 17:27:23
【问题描述】:
我一直在努力用 Haskell 编写一个 Windows 服务应用程序。
背景
服务应用程序由 Windows 服务控制管理器执行。在启动时,它会对StartServiceCtrlDispatcher 进行阻塞调用,并提供一个回调用作service's main function。
服务的主函数应该注册第二个回调来处理传入的命令,例如启动、停止、继续等。它通过调用 RegisterServiceCtrlHandler 来实现。
问题
我可以编写一个注册服务主函数的程序。然后我可以将该程序安装为 Windows 服务并从服务管理控制台启动它。该服务能够启动,报告自己正在运行,然后等待传入的请求。
问题是我无法调用我的service handler function。查询服务状态显示它正在运行,但只要我发送它一个“停止”命令窗口就会弹出一条消息说:
Windows could not stop the Test service on Local Computer.
Error 1061: The service cannot accept control messages at this time.
根据MSDN documentation,StartServiceCtrlDispatcher 函数会阻塞,直到所有服务报告它们已停止。在服务主函数被调用后,调度线程应该等到服务控制管理器发送命令,此时该线程应该调用处理函数。
详情
下面是我正在尝试做的一个非常简化的版本,但它演示了我的处理函数没有被调用的问题。
首先,一些名称和导入:
module Main where
import Control.Applicative
import Foreign
import System.Win32
wIN32_OWN_PROCESS :: DWORD
wIN32_OWN_PROCESS = 0x00000010
sTART_PENDING, rUNNING :: DWORD
sTART_PENDING = 0x00000002
rUNNING = 0x00000004
aCCEPT_STOP, aCCEPT_NONE :: DWORD
aCCEPT_STOP = 0x00000001
aCCEPT_NONE = 0x00000000
nO_ERROR :: DWORD
nO_ERROR = 0x00000000
type HANDLER_FUNCTION = DWORD -> IO ()
type MAIN_FUNCTION = DWORD -> Ptr LPTSTR -> IO ()
我需要用可存储实例定义一些特殊的数据类型用于数据编组:
data TABLE_ENTRY = TABLE_ENTRY LPTSTR (FunPtr MAIN_FUNCTION)
instance Storable TABLE_ENTRY where
sizeOf _ = 8
alignment _ = 4
peek ptr = TABLE_ENTRY <$> peek (castPtr ptr) <*> peek (castPtr ptr `plusPtr` 4)
poke ptr (TABLE_ENTRY name proc) = do
poke (castPtr ptr) name
poke (castPtr ptr `plusPtr` 4) proc
data STATUS = STATUS DWORD DWORD DWORD DWORD DWORD DWORD DWORD
instance Storable STATUS where
sizeOf _ = 28
alignment _ = 4
peek ptr = STATUS
<$> peek (castPtr ptr)
<*> peek (castPtr ptr `plusPtr` 4)
<*> peek (castPtr ptr `plusPtr` 8)
<*> peek (castPtr ptr `plusPtr` 12)
<*> peek (castPtr ptr `plusPtr` 16)
<*> peek (castPtr ptr `plusPtr` 20)
<*> peek (castPtr ptr `plusPtr` 24)
poke ptr (STATUS a b c d e f g) = do
poke (castPtr ptr) a
poke (castPtr ptr `plusPtr` 4) b
poke (castPtr ptr `plusPtr` 8) c
poke (castPtr ptr `plusPtr` 12) d
poke (castPtr ptr `plusPtr` 16) e
poke (castPtr ptr `plusPtr` 20) f
poke (castPtr ptr `plusPtr` 24) g
只需要制造三个外国进口产品。我将提供给 Win32 的两个回调有一个“包装器”导入:
foreign import stdcall "wrapper"
smfToFunPtr :: MAIN_FUNCTION -> IO (FunPtr MAIN_FUNCTION)
foreign import stdcall "wrapper"
handlerToFunPtr :: HANDLER_FUNCTION -> IO (FunPtr HANDLER_FUNCTION)
foreign import stdcall "windows.h RegisterServiceCtrlHandlerW"
c_RegisterServiceCtrlHandler
:: LPCTSTR -> FunPtr HANDLER_FUNCTION -> IO HANDLE
foreign import stdcall "windows.h SetServiceStatus"
c_SetServiceStatus :: HANDLE -> Ptr STATUS -> IO BOOL
foreign import stdcall "windows.h StartServiceCtrlDispatcherW"
c_StartServiceCtrlDispatcher :: Ptr TABLE_ENTRY -> IO BOOL
主程序
最后,这里是主要的服务应用:
main :: IO ()
main =
withTString "Test" $ \name ->
smfToFunPtr svcMain >>= \fpMain ->
withArray [TABLE_ENTRY name fpMain, TABLE_ENTRY nullPtr nullFunPtr] $ \ste ->
c_StartServiceCtrlDispatcher ste >> return ()
svcMain :: MAIN_FUNCTION
svcMain argc argv = do
appendFile "c:\\log.txt" "svcMain: svcMain here!\n"
args <- peekArray (fromIntegral argc) argv
fpHandler <- handlerToFunPtr svcHandler
h <- c_RegisterServiceCtrlHandler (head args) fpHandler
_ <- setServiceStatus h running
appendFile "c:\\log.txt" "svcMain: exiting\n"
svcHandler :: DWORD -> IO ()
svcHandler _ = appendFile "c:\\log.txt" "svcCtrlHandler: received.\n"
setServiceStatus :: HANDLE -> STATUS -> IO BOOL
setServiceStatus h status = with status $ c_SetServiceStatus h
running :: STATUS
running = STATUS wIN32_OWN_PROCESS rUNNING aCCEPT_STOP nO_ERROR 0 0 3000
输出
我之前使用sc create Test binPath= c:\Main.exe 安装了该服务。
这是编译程序的输出:
C:\path>ghc -threaded --make Main.hs
[1 of 1] Compiling Main ( Main.hs, Main.o )
Linking Main.exe ...
C:\path>
然后我从服务控制监视器启动服务。这是我对 SetServiceStatus 的调用被接受的证明:
C:\Path>sc query Test
SERVICE_NAME: Test
TYPE : 10 WIN32_OWN_PROCESS
STATE : 4 RUNNING
(STOPPABLE, NOT_PAUSABLE, IGNORES_SHUTDOWN)
WIN32_EXIT_CODE : 0 (0x0)
SERVICE_EXIT_CODE : 0 (0x0)
CHECKPOINT : 0x0
WAIT_HINT : 0x0
C:\Path>
这里是log.txt的内容,证明我的第一个回调svcMain被调用了:
svcMain: svcMain here!
svcMain: exiting
一旦我使用服务控制管理器发送停止命令,我就会收到错误消息。我的处理程序函数应该在日志文件中添加一行,但这不会发生。然后我的服务出现在停止状态:
C:\Path>sc query Test
SERVICE_NAME: Test
TYPE : 10 WIN32_OWN_PROCESS
STATE : 1 STOPPED
WIN32_EXIT_CODE : 0 (0x0)
SERVICE_EXIT_CODE : 0 (0x0)
CHECKPOINT : 0x0
WAIT_HINT : 0x0
C:\Path>
问题
有没有人知道我可以尝试什么来调用我的处理函数?
更新 20130306
我在 Windows 7 64 位上遇到了这个问题,但在 Windows XP 上没有。其他版本的 Windows 尚未经过测试。当我将编译后的可执行文件复制到多台机器上并执行相同的步骤时,我会得到不同的结果。
【问题讨论】:
-
+1,非常好的问题。考虑到赏金的增加,我认为这还没有解决。稍后我会看看这个。
-
抱歉,我没有看到您的
svcHandler实现调用SetServiceStatus来报告服务的SERVICE_STOPPED状态。另一个问题:您是否在日志文件中看到消息“svcCtrlHandler: received.\n”?顺便说一句,我建议您使用sc interrogate Test来验证svcHandler是否正确注册。如果svcHandler接收SERVICE_CONTROL_INTERROGATE(4) 作为输入,它应该调用SetServiceStatus以与svcMain相同的方式使用报告状态rUNNING(SERVICE_RUNNING) 并控制它接受(@ 987654352@). -
@Oleg 这个例子的处理程序没有做任何事情来保持简单。据我所知,它甚至没有被调用。我没有尝试手动询问服务,但会这样做。我的日志文件不包含预期的输出。
-
我回来并处理了您的代码,重现了您遇到的完全相同的问题。
sc interrogate Test抛出 "The service cannot accept control messages at this time",这是一个很大的线索。基于this worked example from Microsoft,需要在SERVICE_RUNNING之前设置初始SERVICE_START_PENDING状态。我相信当你纠正状态转换时,代码将开始按预期工作——其他一切都检查出来了。 -
唷。经过几天的闲逛,看看我是否能在您的实施中找到一些问题,我终于找到了答案。这一切都根据您挂钩的代码的返回值进行检查,但它不工作。我认为这种事情会为Raymond Chen 提出一个绝妙的问题。
标签: winapi haskell callback windows-services ffi