Validates FORMAT field sizes before writing.

statgen · Jun 30, 2023 · 78f94e2 · 78f94e2
1 parent 9ba1ab1
commit 78f94e2
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/include/savvy/writer.hpp b/include/savvy/writer.hpp
@@ -354,6 +354,14 @@ namespace savvy
           pbwt_format_pointers.back() = &sort_context_.format_contexts[it->first][it->second.size()];
         }
 
+        if (it->second.size() == 0 || (it->second.size() % n_samples_) != 0)
+        {
+          std::cerr << "Error: FMT/" << it->first << " is either empty or not divisible by the number of samples" << std::endl;
+          ofs_.setstate(ofs_.rdstate() | std::ios::failbit);
+          return *this;
+        }
+
+
         if (file_format_ == format::sav2 || it->first != "PH")
           ++n_fmt;
       }