Fix switch screen and support dragging files to remote

server/2015Remote/2015RemoteDlg.cpp

@@ -1142,7 +1142,7 @@ BOOL CMy2015RemoteDlg::OnInitDialog()
     THIS_CFG.SetStr("settings", "MasterHash", GetMasterHash());
 
     UPDATE_SPLASH(20, "正在初始化文件上传模块...");
-    int ret = InitFileUpload(GetHMAC());
+    int ret = InitFileUpload(GetHMAC(), 64, 50, Logf);
     g_hKeyboardHook = SetWindowsHookEx(WH_KEYBOARD_LL, LowLevelKeyboardProc, AfxGetInstanceHandle(), 0);
 
     UPDATE_SPLASH(25, "正在初始化视频墙...");
@@ -2345,6 +2345,27 @@ BOOL CMy2015RemoteDlg::AuthorizeClient(const std::string& sn, const std::string&
     return VerifyMessage(pwd, (BYTE*)passcode.c_str(), passcode.length(), hmac);
 }
 
+// 从char数组解析出多个路径
+std::vector<std::string> ParseMultiStringPath(const char* buffer, size_t size)
+{
+	std::vector<std::string> paths;
+
+	const char* p = buffer;
+	const char* end = buffer + size;
+
+	while (p < end)
+	{
+		size_t len = strlen(p);
+		if (len > 0)
+		{
+			paths.emplace_back(p, len);
+		}
+		p += len + 1;
+	}
+
+	return paths;
+}
+
 VOID CMy2015RemoteDlg::MessageHandle(CONTEXT_OBJECT* ContextObject)
 {
     if (isClosed) {
@@ -2401,14 +2422,17 @@ VOID CMy2015RemoteDlg::MessageHandle(CONTEXT_OBJECT* ContextObject)
     }
     case COMMAND_GET_FILE: {
         // 发送文件
-        int result;
-        auto files = GetClipboardFiles(result);
+		std::string dir = (char*)(szBuffer + 1);
+		char* ptr = (char*)szBuffer + 1 + dir.length() + 1;
+        auto specified = *ptr ? ParseMultiStringPath(ptr, len - 2 - dir.length()) : std::vector<std::string>{};
+        int result = 0;
+        auto files = specified.empty() ? GetClipboardFiles(result): specified;
         if (!files.empty()) {
-            std::string dir = (char*)(szBuffer + 1);
             std::string hash = GetPwdHash(), hmac = GetHMAC(100);
             std::thread(FileBatchTransferWorker, files, dir, ContextObject, SendData, FinishSend,
                         hash, hmac).detach();
         } else {
+            ContextObject->CancelIO();
             Mprintf("GetClipboardFiles failed: %d\n", result);
         }
         break;

server/2015Remote/ScreenSpyDlg.cpp

@@ -167,6 +167,7 @@ BEGIN_MESSAGE_MAP(CScreenSpyDlg, CDialog)
     ON_WM_TIMER()
     ON_COMMAND(ID_EXIT_FULLSCREEN, &CScreenSpyDlg::OnExitFullscreen)
     ON_MESSAGE(WM_DISCONNECT, &CScreenSpyDlg::OnDisconnect)
+    ON_WM_DROPFILES()
 END_MESSAGE_MAP()
 
 
@@ -220,6 +221,9 @@ BOOL CScreenSpyDlg::OnInitDialog()
 {
     CDialog::OnInitDialog();
     SetIcon(m_hIcon,FALSE);
+    DragAcceptFiles(TRUE);
+	ChangeWindowMessageFilter(WM_DROPFILES, MSGFLT_ADD);
+	ChangeWindowMessageFilter(0x0049, MSGFLT_ADD);
 
     PrepareDrawing(m_BitmapInfor_Full);
 
@@ -1101,3 +1105,50 @@ void CScreenSpyDlg::UpdateCtrlStatus(BOOL ctrl) {
 	m_bIsCtrl = ctrl;
 	SetClassLongPtr(m_hWnd, GCLP_HCURSOR, m_bIsCtrl ? (LONG_PTR)m_hRemoteCursor : (LONG_PTR)LoadCursor(NULL, IDC_NO));
 }
+
+// 将多个路径组合成单\0分隔的char数组
+// 格式: "path1\0path2\0path3\0"
+std::vector<char> BuildMultiStringPath(const std::vector<std::string>& paths)
+{
+	std::vector<char> result;
+
+	for (const auto& path : paths)
+	{
+		result.insert(result.end(), path.begin(), path.end());
+		result.push_back('\0');
+	}
+
+	return result;
+}
+
+void CScreenSpyDlg::OnDropFiles(HDROP hDropInfo)
+{
+    if (m_bIsCtrl && m_bConnected) {
+        UINT nFiles = DragQueryFile(hDropInfo, 0xFFFFFFFF, NULL, 0);
+        std::vector<std::string> list;
+        for (UINT i = 0; i < nFiles; i++)
+        {
+            TCHAR szPath[MAX_PATH];
+            DragQueryFile(hDropInfo, i, szPath, MAX_PATH);
+            list.push_back(szPath);
+        }
+        std::string GetPwdHash();
+        std::string GetHMAC(int offset);
+        std::vector<std::string> PreprocessFilesSimple(const std::vector<std::string>&inputFiles);
+        auto files = PreprocessFilesSimple(list);
+        auto str = BuildMultiStringPath(files);
+        BYTE* szBuffer = new BYTE[1 + 80 + str.size()];
+        szBuffer[0] = { COMMAND_GET_FOLDER };
+        std::string masterId = GetPwdHash(), hmac = GetHMAC(100);
+        memcpy((char*)szBuffer + 1, masterId.c_str(), masterId.length());
+        memcpy((char*)szBuffer + 1 + masterId.length(), hmac.c_str(), hmac.length());
+        memcpy(szBuffer + 1 + 80, str.data(), str.size());
+        m_ContextObject->Send2Client(szBuffer, 81 + str.size());
+        Mprintf("【Ctrl+V】 从本地拖拽文件到远程 \n");
+        SAFE_DELETE_ARRAY(szBuffer);
+    }
+
+	DragFinish(hDropInfo);
+
+    CDialogBase::OnDropFiles(hDropInfo);
+}

server/2015Remote/ScreenSpyDlg.h

@@ -112,6 +112,7 @@ public:
     void EnterFullScreen();
     bool LeaveFullScreen();
     void UpdateCtrlStatus(BOOL ctrl);
+    void OnDropFiles(HDROP hDropInfo);
 
     afx_msg void OnHScroll(UINT nSBCode, UINT nPos, CScrollBar* pScrollBar);
     afx_msg void OnVScroll(UINT nSBCode, UINT nPos, CScrollBar* pScrollBar);