1 files changed, 156 insertions, 0 deletions
diff --git a/src/lib/barrier/FileChunk.cpp b/src/lib/barrier/FileChunk.cpp
new file mode 100644
index 0000000..3a98568
--- /dev/null
+++ b/src/lib/barrier/FileChunk.cpp
@@ -0,0 +1,156 @@
+/*
+ * barrier -- mouse and keyboard sharing utility
+ * Copyright (C) 2015-2016 Symless Ltd.
+ * 
+ * This package is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * found in the file LICENSE that should have accompanied this file.
+ * 
+ * This package is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "barrier/FileChunk.h"
+
+#include "barrier/ProtocolUtil.h"
+#include "barrier/protocol_types.h"
+#include "io/IStream.h"
+#include "base/Stopwatch.h"
+#include "base/Log.h"
+
+static const UInt16 kIntervalThreshold = 1;
+
+FileChunk::FileChunk(size_t size) :
+    Chunk(size)
+{
+        m_dataSize = size - FILE_CHUNK_META_SIZE;
+}
+
+FileChunk*
+FileChunk::start(const String& size)
+{
+    size_t sizeLength = size.size();
+    FileChunk* start = new FileChunk(sizeLength + FILE_CHUNK_META_SIZE);
+    char* chunk = start->m_chunk;
+    chunk[0] = kDataStart;
+    memcpy(&chunk[1], size.c_str(), sizeLength);
+    chunk[sizeLength + 1] = '\0';
+
+    return start;
+}
+
+FileChunk*
+FileChunk::data(UInt8* data, size_t dataSize)
+{
+    FileChunk* chunk = new FileChunk(dataSize + FILE_CHUNK_META_SIZE);
+    char* chunkData = chunk->m_chunk;
+    chunkData[0] = kDataChunk;
+    memcpy(&chunkData[1], data, dataSize);
+    chunkData[dataSize + 1] = '\0';
+
+    return chunk;
+}
+
+FileChunk*
+FileChunk::end()
+{
+    FileChunk* end = new FileChunk(FILE_CHUNK_META_SIZE);
+    char* chunk = end->m_chunk;
+    chunk[0] = kDataEnd;
+    chunk[1] = '\0';
+
+    return end;
+}
+
+int
+FileChunk::assemble(barrier::IStream* stream, String& dataReceived, size_t& expectedSize)
+{
+    // parse
+    UInt8 mark = 0;
+    String content;
+    static size_t receivedDataSize;
+    static double elapsedTime;
+    static Stopwatch stopwatch;
+
+    if (!ProtocolUtil::readf(stream, kMsgDFileTransfer + 4, &mark, &content)) {
+        return kError;
+    }
+
+    switch (mark) {
+    case kDataStart:
+        dataReceived.clear();
+        expectedSize = barrier::string::stringToSizeType(content);
+        receivedDataSize = 0;
+        elapsedTime = 0;
+        stopwatch.reset();
+
+        if (CLOG->getFilter() >= kDEBUG2) {
+            LOG((CLOG_DEBUG2 "recv file size=%s", content.c_str()));
+            stopwatch.start();
+        }
+        return kStart;
+
+    case kDataChunk:
+        dataReceived.append(content);
+        if (CLOG->getFilter() >= kDEBUG2) {
+                LOG((CLOG_DEBUG2 "recv file chunck size=%i", content.size()));
+                double interval = stopwatch.getTime();
+                receivedDataSize += content.size();
+                LOG((CLOG_DEBUG2 "recv file interval=%f s", interval));
+                if (interval >= kIntervalThreshold) {
+                    double averageSpeed = receivedDataSize / interval / 1000;
+                    LOG((CLOG_DEBUG2 "recv file average speed=%f kb/s", averageSpeed));
+
+                    receivedDataSize = 0;
+                    elapsedTime += interval;
+                    stopwatch.reset();
+                }
+            }
+        return kNotFinish;
+
+    case kDataEnd:
+        if (expectedSize != dataReceived.size()) {
+            LOG((CLOG_ERR "corrupted clipboard data, expected size=%d actual size=%d", expectedSize, dataReceived.size()));
+            return kError;
+        }
+
+        if (CLOG->getFilter() >= kDEBUG2) {
+            LOG((CLOG_DEBUG2 "file transfer finished"));
+            elapsedTime += stopwatch.getTime();
+            double averageSpeed = expectedSize / elapsedTime / 1000;
+            LOG((CLOG_DEBUG2 "file transfer finished: total time consumed=%f s", elapsedTime));
+            LOG((CLOG_DEBUG2 "file transfer finished: total data received=%i kb", expectedSize / 1000));
+            LOG((CLOG_DEBUG2 "file transfer finished: total average speed=%f kb/s", averageSpeed));
+        }
+        return kFinish;
+    }
+
+    return kError;
+}
+
+void
+FileChunk::send(barrier::IStream* stream, UInt8 mark, char* data, size_t dataSize)
+{
+    String chunk(data, dataSize);
+
+    switch (mark) {
+    case kDataStart:
+        LOG((CLOG_DEBUG2 "sending file chunk start: size=%s", data));
+        break;
+
+    case kDataChunk:
+        LOG((CLOG_DEBUG2 "sending file chunk: size=%i", chunk.size()));
+        break;
+
+    case kDataEnd:
+        LOG((CLOG_DEBUG2 "sending file finished"));
+        break;
+    }
+
+    ProtocolUtil::writef(stream, kMsgDFileTransfer, mark, &chunk);
+}