backup/barrier
1
0
Fork 0
mirror of https://github.com/debauchee/barrier.git synced 2026-07-17 11:31:35 +08:00
Code Issues Projects Releases Wiki Activity

created a new thread to write file to drop directory on client side

Browse Source
This commit is contained in:
jerry
2013-09-25 14:49:04 +00:00
parent 8301d50ab7
commit 6e50945bf9
5 changed files with 55 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
src/lib/client/CClient.cpp
View File

@@ -68,7 +68,8 @@ CClient::CClient(IEventQueue* events,
m_events(events),
m_cryptoStream(NULL),
m_crypto(crypto),
m_sendFileThread(NULL)
m_sendFileThread(NULL),
m_writeToDropDirThread(NULL)
{
assert(m_socketFactory != NULL);
assert(m_screen != NULL);
@@ -722,26 +723,40 @@ void
CClient::onFileRecieveCompleted()
{
if (isReceivedFileSizeValid()) {
m_fileTransferDes = m_screen->getDropTarget();
if (!m_fileTransferDes.empty()) {
std::fstream file;
#ifdef SYSAPI_WIN32
m_fileTransferDes.append("\\");
#else
m_fileTransferDes.append("/");
#endif
m_fileTransferDes.append(m_dragFileList.at(0));
file.open(m_fileTransferDes.c_str(), std::ios::out | std::ios::binary);
if (!file.is_open()) {
// TODO: file open failed
}
m_writeToDropDirThread = new CThread(
new TMethodJob<CClient>(
this, &CClient::writeToDropDirThread));
}
}
file.write(m_receivedFileData.c_str(), m_receivedFileData.size());
file.close();
}
else {
LOG((CLOG_ERR "drop file failed: drop target is empty"));
void
CClient::writeToDropDirThread(void*)
{
while (m_screen->getFakeDraggingStarted()) {
ARCH->sleep(.1f);
}
m_fileTransferDes = m_screen->getDropTarget();
if (!m_fileTransferDes.empty() && m_dragFileList.size() > 0) {
std::fstream file;
CString dropTarget = m_fileTransferDes;
#ifdef SYSAPI_WIN32
dropTarget.append("\\");
#else
dropTarget.append("/");
#endif
dropTarget.append(m_dragFileList.at(0));
file.open(dropTarget.c_str(), std::ios::out | std::ios::binary);
if (!file.is_open()) {
// TODO: file open failed
}
file.write(m_receivedFileData.c_str(), m_receivedFileData.size());
file.close();
}
else {
LOG((CLOG_ERR "drop file failed: drop target is empty"));
}
}

2
src/lib/client/CClient.h
View File

@@ -182,6 +182,7 @@ private:
void sendConnectionFailedEvent(const char* msg);
void sendFileChunk(const void* data);
void sendFileThread(void*);
void writeToDropDirThread(void*);
void setupConnecting();
void setupConnection();
void setupScreen();
@@ -234,6 +235,7 @@ private:
CDragFileList m_dragFileList;
CString m_dragFileExt;
CThread* m_sendFileThread;
CThread* m_writeToDropDirThread;
};
#endif