From 40565418ba236d4383b33d6d863375d881406803 Mon Sep 17 00:00:00 2001 From: Peng Wu Date: Fri, 24 Jun 2011 10:56:46 +0800 Subject: add a check --- utils/training/gen_k_mixture_model.cpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'utils') diff --git a/utils/training/gen_k_mixture_model.cpp b/utils/training/gen_k_mixture_model.cpp index e09abe2..19dd102 100644 --- a/utils/training/gen_k_mixture_model.cpp +++ b/utils/training/gen_k_mixture_model.cpp @@ -298,6 +298,10 @@ static bool post_processing_unigram(KMixtureModelBigram * bigram, KMixtureModelMagicHeader magic_header; assert(bigram->get_magic_header(magic_header)); + if ( magic_header.m_total_freq + delta < magic_header.m_total_freq ){ + fprintf(stderr, "the m_total_freq in magic header overflows.\n"); + return false; + } magic_header.m_total_freq += total_freq; assert(bigram->set_magic_header(magic_header)); -- cgit