cyonjan/WeChatFerry
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add another socket

Browse Source
This commit is contained in:
Changhua 2023-02-20 00:58:12 +08:00
parent d6f47d7d5a
commit afc90aad10
1 changed files with 29 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
spy/rpc_server.cpp
View File

@ -27,6 +27,8 @@
#include "util.h" #include "util.h"
#define G_BUF_SIZE (16 * 1024 * 1024) #define G_BUF_SIZE (16 * 1024 * 1024)
#define CMD_URL "tcp://0.0.0.0:10086"
#define MSG_URL "tcp://0.0.0.0:10087"
extern int IsLogin(void); // Defined in spy.cpp extern int IsLogin(void); // Defined in spy.cpp
extern string GetSelfWxid(); // Defined in spy.cpp extern string GetSelfWxid(); // Defined in spy.cpp
@ -206,14 +208,32 @@ bool func_send_img(char *path, char *receiver, uint8_t *out, size_t *len)
static void PushMessage() static void PushMessage()
{ {
static uint8_t buffer[1024] = { 0 }; static nng_socket msg_sock;
static uint8_t buffer[G_BUF_SIZE] = { 0 };
int rv; int rv;
Response rsp = Response_init_default; Response rsp = Response_init_default;
rsp.func = Functions_FUNC_ENABLE_RECV_TXT; rsp.func = Functions_FUNC_ENABLE_RECV_TXT;
rsp.which_msg = Response_wxmsg_tag; rsp.which_msg = Response_wxmsg_tag;
pb_ostream_t stream = pb_ostream_from_buffer(buffer, 1024); pb_ostream_t stream = pb_ostream_from_buffer(buffer, G_BUF_SIZE);
char *url = (char *)MSG_URL;
if ((rv = nng_pair1_open(&msg_sock)) != 0) {
LOG_ERROR("nng_pair0_open error {}", rv);
return;
}
if ((rv = nng_listen(msg_sock, url, NULL, 0)) != 0) {
LOG_ERROR("nng_listen error {}", rv);
return;
}
LOG_INFO("Server listening on {}", url);
if ((rv = nng_setopt_ms(msg_sock, NNG_OPT_SENDTIMEO, 2000)) != 0) {
LOG_ERROR("nng_setopt_ms: {}", rv);
return;
}
while (gIsListening) { while (gIsListening) {
// 中断式，兼顾及时性和CPU使用率 // 中断式，兼顾及时性和CPU使用率
@ -226,7 +246,7 @@ static void PushMessage()
} }
while (!gMsgQueue.empty()) { while (!gMsgQueue.empty()) {
auto wxmgs = gMsgQueue.front(); auto wxmgs = gMsgQueue.front();
rsp.msg.wxmsg.is_self = wxmgs.is_self; rsp.msg.wxmsg.is_self = wxmgs.is_self;
rsp.msg.wxmsg.is_group = wxmgs.is_group; rsp.msg.wxmsg.is_group = wxmgs.is_group;
rsp.msg.wxmsg.type = wxmgs.type; rsp.msg.wxmsg.type = wxmgs.type;
@ -241,13 +261,15 @@ static void PushMessage()
LOG_ERROR("Encoding failed: {}", PB_GET_ERROR(&stream)); LOG_ERROR("Encoding failed: {}", PB_GET_ERROR(&stream));
continue; continue;
} }
rv = nng_send(sock, buffer, stream.bytes_written, 0);
rv = nng_send(msg_sock, buffer, stream.bytes_written, 0);
if (rv != 0) { if (rv != 0) {
LOG_ERROR("nng_send: {}", rv); LOG_ERROR("nng_send: {}", rv);
} }
} }
ResetEvent(g_hEvent); ResetEvent(g_hEvent);
} }
nng_close(msg_sock);
} }
bool func_enable_recv_txt(uint8_t *out, size_t *len) bool func_enable_recv_txt(uint8_t *out, size_t *len)
@ -263,7 +285,7 @@ bool func_enable_recv_txt(uint8_t *out, size_t *len)
NULL // unnamed NULL // unnamed
); );
if (g_hEvent == NULL) { if (g_hEvent == NULL) {
LOG_ERROR("CreateMutex error: {}", GetLastError()); LOG_ERROR("CreateEvent error: {}", GetLastError());
return false; return false;
} }
@ -403,7 +425,7 @@ static bool dispatcher(uint8_t *in, size_t in_len, uint8_t *out, size_t *out_len
static int RunServer() static int RunServer()
{ {
int rv = 0; int rv = 0;
char *url = (char *)"tcp://0.0.0.0:10086"; char *url = (char *)CMD_URL;
if ((rv = nng_pair1_open(&sock)) != 0) { if ((rv = nng_pair1_open(&sock)) != 0) {
LOG_ERROR("nng_pair0_open error {}", rv); LOG_ERROR("nng_pair0_open error {}", rv);
return rv; return rv;
@ -416,7 +438,7 @@ static int RunServer()
LOG_INFO("Server listening on {}", url); LOG_INFO("Server listening on {}", url);
if ((rv = nng_setopt_ms(sock, NNG_OPT_SENDTIMEO, 1000)) != 0) { if ((rv = nng_setopt_ms(sock, NNG_OPT_SENDTIMEO, 1000)) != 0) {
LOG_ERROR("nng_recv: {}", rv); LOG_ERROR("nng_setopt_ms: {}", rv);
return rv; return rv;
} }