From 4b612f825f22de838a262015e3d4134e276a1891 Mon Sep 17 00:00:00 2001
From: Gabriel Studer <gabriel.studer@unibas.ch>
Date: Wed, 17 Nov 2021 09:00:33 +0100
Subject: [PATCH] check for healthy stream after reading

---
 modules/io/src/mol/omf.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/modules/io/src/mol/omf.cc b/modules/io/src/mol/omf.cc
index 8068041f4..3d3c144de 100644
--- a/modules/io/src/mol/omf.cc
+++ b/modules/io/src/mol/omf.cc
@@ -1155,6 +1155,10 @@ void OMF::FromStream(std::istream& stream) {
   Load(stream, bond_chain_names_);
   Load(stream, bond_atoms_);
   Load(stream, bond_orders_);
+
+  if(!stream.good()) {
+    throw ost::Error("Cannot read corrupted OMF stream");
+  }
 }
 
 void OMF::FillChain(ost::mol::ChainHandle& chain, ost::mol::XCSEditor& ed,
-- 
GitLab