fix: buffer overflow in NGramModel.prob

cmusphinx · Dec 28, 2023 · 563b6a8 · 563b6a8
1 parent 29a6654
commit 563b6a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cython/_pocketsphinx.pyx b/cython/_pocketsphinx.pyx
@@ -610,7 +610,7 @@ cdef class NGramModel:
         cdef const char **cwords
         cdef int prob
         bwords = [w.encode("utf-8") for w in words]
-        cwords = <const char **>malloc(len(bwords))
+        cwords = <const char **>malloc(len(bwords) * sizeof(char *))
         for i, w in enumerate(bwords):
             cwords[i] = w
         prob = ngram_prob(self.lm, cwords, len(words))