From aa3703e9f5c5c9d3649ccdb0a21e5543f5435352 Mon Sep 17 00:00:00 2001 From: Changhua Date: Mon, 19 Feb 2024 20:08:22 +0800 Subject: [PATCH] Return error if path not exists when sending image and file. --- WeChatFerry/spy/rpc_server.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/WeChatFerry/spy/rpc_server.cpp b/WeChatFerry/spy/rpc_server.cpp index 1e3b847..72eb6fb 100644 --- a/WeChatFerry/spy/rpc_server.cpp +++ b/WeChatFerry/spy/rpc_server.cpp @@ -2,6 +2,7 @@ #include #include +#include #include #include #include @@ -37,6 +38,8 @@ #define BASE_URL "tcp://0.0.0.0" #define G_BUF_SIZE (16 * 1024 * 1024) +namespace fs = std::filesystem; + extern int IsLogin(void); // Defined in spy.cpp bool gIsListening = false; @@ -251,6 +254,9 @@ bool func_send_img(char *path, char *receiver, uint8_t *out, size_t *len) if ((path == NULL) || (receiver == NULL)) { LOG_ERROR("Empty path or receiver."); rsp.msg.status = -1; + } else if (!fs::exists(path)) { + LOG_ERROR("Path does not exists: {}", path); + rsp.msg.status = -2; } else { SendImageMessage(receiver, path); rsp.msg.status = 0; @@ -275,6 +281,9 @@ bool func_send_file(char *path, char *receiver, uint8_t *out, size_t *len) if ((path == NULL) || (receiver == NULL)) { LOG_ERROR("Empty path or receiver."); rsp.msg.status = -1; + } else if (!fs::exists(path)) { + LOG_ERROR("Path does not exists: {}", path); + rsp.msg.status = -2; } else { SendImageMessage(receiver, path); rsp.msg.status = 0;