From e0903983ad6342e11b19ffae82e4ba59ed7aa5e7 Mon Sep 17 00:00:00 2001
From: "Jerry (Xinyu Hou)" <xinyu@synergy-project.org>
Date: Thu, 4 Jun 2015 10:19:36 -0700
Subject: [PATCH] Validated file expected size #4584

---
 src/lib/synergy/FileChunk.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/lib/synergy/FileChunk.cpp b/src/lib/synergy/FileChunk.cpp
index 9675ce0f..403b32a4 100644
--- a/src/lib/synergy/FileChunk.cpp
+++ b/src/lib/synergy/FileChunk.cpp
@@ -114,6 +114,12 @@ FileChunk::assemble(synergy::IStream* stream, String& dataReceived, size_t& expe
 		return kNotFinish;
 
 	case kDataEnd:
+		if (expectedSize != dataReceived.size()) {
+			LOG((CLOG_ERR "corrupted clipboard data, expected size=%d actual size=%d", expectedSize, dataReceived.size()));
+			LOG((CLOG_NOTIFY "File Transmission Failed: Corrupted file data."));
+			return kError;
+		}
+
 		if (CLOG->getFilter() >= kDEBUG2) {
 			LOG((CLOG_DEBUG2 "file data transfer finished"));
 			elapsedTime += stopwatch.getTime();