From f23f751e80e93d8203303d9f34ff9a230985b833 Mon Sep 17 00:00:00 2001
From: Ryan Huffman <ryanhuffman@gmail.com>
Date: Mon, 26 Feb 2018 15:27:19 -0800
Subject: [PATCH] Add check to make sure entity data gets written to zip

---
 domain-server/src/EntitiesBackupHandler.cpp | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/domain-server/src/EntitiesBackupHandler.cpp b/domain-server/src/EntitiesBackupHandler.cpp
index c9488df51f..76b32094a6 100644
--- a/domain-server/src/EntitiesBackupHandler.cpp
+++ b/domain-server/src/EntitiesBackupHandler.cpp
@@ -35,7 +35,12 @@ void EntitiesBackupHandler::createBackup(const QString& backupName, QuaZip& zip)
             qCritical().nospace() << "Failed to open " << ENTITIES_BACKUP_FILENAME << " for writing in zip";
             return;
         }
-        zipFile.write(entitiesFile.readAll());
+        auto entityData = entitiesFile.readAll();
+        if (zipFile.write(entityData) != entityData.size()) {
+            qCritical() << "Failed to write entities file to backup";
+            zipFile.close();
+            return;
+        }
         zipFile.close();
         if (zipFile.getZipError() != UNZ_OK) {
             qCritical().nospace() << "Failed to zip " << ENTITIES_BACKUP_FILENAME << ": " << zipFile.getZipError();