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