diff --git a/WeChatFerry/spy/rpc_server.cpp b/WeChatFerry/spy/rpc_server.cpp index 679d38f..1a8b842 100644 --- a/WeChatFerry/spy/rpc_server.cpp +++ b/WeChatFerry/spy/rpc_server.cpp @@ -33,6 +33,10 @@ #include "user_info.h" #include "util.h" +#ifdef _WIN32 +#include +#endif + #define URL_SIZE 20 #define BASE_URL "tcp://0.0.0.0" #define G_BUF_SIZE (16 * 1024 * 1024) @@ -252,12 +256,35 @@ 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; + bool fileExist = false; +#ifdef _WIN32 + // 将 char* 路径转换为 wide-char 格式 + std::wstring wpath = String2Wstring(path); + + // 使用 GetFileAttributesW() 检查文件是否存在 + DWORD attributes = GetFileAttributesW(wpath.c_str()); + if (attributes == INVALID_FILE_ATTRIBUTES) { + LOG_ERROR("Path does not exist: {}", path); + rsp.msg.status = -2; + } else { + fileExist = true; + } +#else + // On Linux and other systems + std::string spath(path); + if (!fs::exists(spath)) { + LOG_ERROR("Path does not exist: {}", path); + rsp.msg.status = -2; + return false; + } else { + fileExist = true; + } +#endif + if(fileExist) { + SendImageMessage(receiver, path); + rsp.msg.status = 0; + } } pb_ostream_t stream = pb_ostream_from_buffer(out, *len); @@ -279,12 +306,38 @@ 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; + } + else { + bool fileExist = false; +#ifdef _WIN32 + // 将 char* 路径转换为 wide-char 格式 + std::wstring wpath = String2Wstring(path); + + // 使用 GetFileAttributesW() 检查文件是否存在 + DWORD attributes = GetFileAttributesW(wpath.c_str()); + if (attributes == INVALID_FILE_ATTRIBUTES) { + LOG_ERROR("Path does not exist: {}", path); + rsp.msg.status = -2; + } + else { + fileExist = true; + } +#else + // On Linux and other systems + std::string spath(path); + if (!fs::exists(spath)) { + LOG_ERROR("Path does not exist: {}", path); + rsp.msg.status = -2; + return false; + } + else { + fileExist = true; + } +#endif + if (fileExist) { + SendImageMessage(receiver, path); + rsp.msg.status = 0; + } } pb_ostream_t stream = pb_ostream_from_buffer(out, *len);