implement progress_bar workaound

Prevent progress bar overrun if flash_size is divisible by 254

flasher/memory.py

@@ -101,24 +101,48 @@ def write_memory(ecu, payload, flash_start, flash_size, progress_callback=False)
 	packets_to_write = int(flash_size/254)
 	packets_written = 0
 
-	while packets_to_write >= packets_written:
-		if (progress_callback):
-			progress_callback.title('Packet {}/{}'.format(packets_written, packets_to_write))
-
-		payload_packet_start = packets_written*254
-		payload_packet_end = payload_packet_start+254
-		payload_packet = payload[payload_packet_start:payload_packet_end]
-
-		while True:
-			try:
-				ecu.bus.execute(TransferData(list(payload_packet)))
-				break
-			except (GKBusTimeoutException):
-				print('Timed Out! Trying again...')
-				continue
-		packets_written += 1
-
-		if (progress_callback):
-			progress_callback(len(payload_packet))
+#Workaround to prevent progress bar overrun if flash_size is divisible by 254
+	if (flash_size%254==0) :
+			while packets_to_write > packets_written:
+				if (progress_callback):
+					progress_callback.title('Packet {}/{}'.format(packets_written+1, packets_to_write))
+
+				payload_packet_start = packets_written*254
+				payload_packet_end = payload_packet_start+254
+				payload_packet = payload[payload_packet_start:payload_packet_end]
+
+				while True:
+					try:
+						ecu.bus.execute(TransferData(list(payload_packet)))
+						break
+					except (GKBusTimeoutException):
+						print('Timed Out! Trying again...')
+						continue
+
+				packets_written += 1
+
+				if (progress_callback):
+					progress_callback(len(payload_packet))
+	else:
+			while packets_to_write >= packets_written:
+				if (progress_callback):
+					progress_callback.title('Packet {}/{}'.format(packets_written, packets_to_write))
+
+				payload_packet_start = packets_written*254
+				payload_packet_end = payload_packet_start+254
+				payload_packet = payload[payload_packet_start:payload_packet_end]
+
+				while True:
+					try:
+						ecu.bus.execute(TransferData(list(payload_packet)))
+						break
+					except (GKBusTimeoutException):
+						print('Timed Out! Trying again...')
+						continue
+
+				packets_written += 1
+
+				if (progress_callback):
+					progress_callback(len(payload_packet))
 
 	ecu.bus.execute(RequestTransferExit())