embed.rs: Added error check when ISO file incomplete

src/live/embed.rs

@@ -346,6 +346,25 @@ impl KargEmbedAreas {
             );
         }
 
+        let all_paths: Vec<String> = iso.walk()
+                                        .unwrap()
+                                        .map(|v| v.unwrap().0)
+                                        .collect::<Vec<String>>();
+        let mut all_files: Vec<iso9660::File> = Vec::new();
+        for path in all_paths {
+            if path.contains('.'){
+                let file = iso.get_path(&path)
+                                 .with_context(|| format!("looking up '{:?}'", path))?
+                                 .try_into_file()?;
+                all_files.push(file);
+            }
+        }
+        all_files.sort_by_key(|d| d.address.as_offset());
+        if all_files.last().unwrap().address.as_offset() + u64::from(all_files.last().unwrap().length) > iso.as_file()?.metadata()?.len()   {
+            bail!("ISO Image is wrong and can't be modified");
+        }
+
+
         // writable regions
         let mut regions = Vec::new();
         for loc in info.files {