fixed #79 all data is zero when file transfer from target to windows PC

Signed-off-by: Frank Li <Frank.Li@nxp.com>
nxp-imx · Jan 25, 2019 · c83c28e · c83c28e
1 parent 15806b2
commit c83c28e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/libuuu/fastboot.cpp b/libuuu/fastboot.cpp
@@ -63,8 +63,6 @@ int FastBoot::Transport(string cmd, void *p, size_t size, vector<uint8_t> *input
 		{
 			size_t sz;
 			sz = strtoul(buff+4, NULL, 16);
-			if (sz > size)
-				sz = size;
 
 			if (input)
 			{
@@ -76,6 +74,9 @@ int FastBoot::Transport(string cmd, void *p, size_t size, vector<uint8_t> *input
 			}
 			else
 			{
+				if (sz > size)
+					sz = size;
+
 				if (m_pTrans->write(p, sz))
 					return -1;
 			}