Check configdrive size before writing to partition

Avoids writing a configdrive out to disk that is larger than
the intended partition.

Change-Id: I4e067ccb23ba528d96e4faad39219f67b4178e82

1 files changed, 11 insertions, 0 deletions

diff --git a/ironic_python_agent/errors.py b/ironic_python_agent/errors.py
index 4223ab9f..151d8cea 100644
--- a/ironic_python_agent/errors.py
+++ b/ironic_python_agent/errors.py
@@ -151,6 +151,17 @@ class ImageWriteError(RESTError):
         self.details = self.details.format(device, exit_code)
 
 
+class ConfigDriveTooLargeError(RESTError):
+    """Error raised when a configdrive is larger than the partition."""
+    message = 'Configdrive is too large for intended partition.'
+
+    def __init__(self, filename, filesize):
+        details = ('Configdrive at {0} has size {1}, which is larger than '
+                   'the intended partition.').format(filename, filesize)
+        super(ConfigDriveTooLargeError, self).__init__(details)
+        self.details = details
+
+
 class ConfigDriveWriteError(RESTError):
     """Error raised when a configdrive directory cannot be written to a
     device.