95 lines
3.1 KiB
C++
95 lines
3.1 KiB
C++
|
#include <stdio.h>
|
|||
|
|
#include <stdlib.h>
|
|||
|
|
|
|||
|
|
#include "monitor.h"
|
|||
|
|
#include "rpc_server.h"
|
|||
|
|
#include "send_msg.h"
|
|||
|
|
#include "spy_types.h"
|
|||
|
|
|
|||
|
|
#include "../Rpc/rpc_h.h"
|
|||
|
|
#pragma comment(lib, "Rpcrt4.lib")
|
|||
|
|
|
|||
|
|
extern HANDLE g_hEvent;
|
|||
|
|
extern MsgQueue_t g_MsgQueue;
|
|||
|
|
|
|||
|
|
int server_IsLogin() { return IsLogin(); }
|
|||
|
|
|
|||
|
|
void server_EnableReceiveMsg()
|
|||
|
|
{
|
|||
|
|
unsigned long ulCode = 0;
|
|||
|
|
|
|||
|
|
RpcTryExcept
|
|||
|
|
{
|
|||
|
|
// 调用客户端的回调函数
|
|||
|
|
while (true) {
|
|||
|
|
// 中断式，兼顾及时性和CPU使用率
|
|||
|
|
WaitForSingleObject(g_hEvent, INFINITE); // 等待消息
|
|||
|
|
while (!g_MsgQueue.empty()) {
|
|||
|
|
client_ReceiveMsg((RpcMessage_t *)&g_MsgQueue.front()); // 调用接收消息回调
|
|||
|
|
g_MsgQueue.pop();
|
|||
|
|
}
|
|||
|
|
ResetEvent(g_hEvent);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
RpcExcept(1)
|
|||
|
|
{
|
|||
|
|
ulCode = RpcExceptionCode();
|
|||
|
|
printf("Runtime reported exception 0x%lx = %ld\n", ulCode, ulCode);
|
|||
|
|
}
|
|||
|
|
RpcEndExcept
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int server_SendTextMsg(const wchar_t *wxid, const wchar_t *at_wxid, const wchar_t *msg)
|
|||
|
|
{
|
|||
|
|
SendTextMessage(wxid, at_wxid, msg);
|
|||
|
|
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
RPC_STATUS CALLBACK SecurityCallback(RPC_IF_HANDLE /*hInterface*/, void * /*pBindingHandle*/)
|
|||
|
|
{
|
|||
|
|
return RPC_S_OK; // Always allow anyone.
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int RpcStartServer(HMODULE hModule)
|
|||
|
|
{
|
|||
|
|
RPC_STATUS status;
|
|||
|
|
// Uses the protocol combined with the endpoint for receiving
|
|||
|
|
// remote procedure calls.
|
|||
|
|
status = RpcServerUseProtseqEp(reinterpret_cast<RPC_WSTR>((RPC_WSTR)L"ncalrpc"), // Use TCP/IP protocol
|
|||
|
|
RPC_C_LISTEN_MAX_CALLS_DEFAULT, // Backlog queue length for TCP/IP.
|
|||
|
|
reinterpret_cast<RPC_WSTR>((RPC_WSTR)L"tmp_endpoint"), // TCP/IP port to use
|
|||
|
|
NULL // No security
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
if (status)
|
|||
|
|
return status;
|
|||
|
|
|
|||
|
|
// Registers the interface and auto listen
|
|||
|
|
// Equal to RpcServerRegisterIf + RpcServerListen
|
|||
|
|
status = RpcServerRegisterIf2(server_ISpy_v1_0_s_ifspec, // Interface to register.
|
|||
|
|
NULL, // Use the MIDL generated entry-point vector.
|
|||
|
|
NULL, // Use the MIDL generated entry-point vector.
|
|||
|
|
RPC_IF_ALLOW_LOCAL_ONLY | RPC_IF_AUTOLISTEN, // Forces use of security callback.
|
|||
|
|
RPC_C_LISTEN_MAX_CALLS_DEFAULT, // Use default number of concurrent calls.
|
|||
|
|
(unsigned)-1, // Infinite max size of incoming data blocks.
|
|||
|
|
SecurityCallback); // Naive security callback.
|
|||
|
|
|
|||
|
|
while (1) {
|
|||
|
|
Sleep(10000); // 休眠，释放CPU
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int RpcStopServer(void)
|
|||
|
|
{
|
|||
|
|
RPC_STATUS status;
|
|||
|
|
status = RpcMgmtStopServerListening(NULL);
|
|||
|
|
if (status)
|
|||
|
|
return status;
|
|||
|
|
|
|||
|
|
status = RpcServerUnregisterIf(NULL, NULL, FALSE);
|
|||
|
|
return status;
|
|||
|
|
}
|