X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=src%2Flzms_common.c;h=9d9d7b2ae88bb5d4c116dd82981349c5a51f5733;hp=b23a2785fff907dad62f92a749cda041957b7b9c;hb=723d5dbc1705200082f640453f19233a386bc655;hpb=f9a2ea6477380519ee90f2094e36464a437ee6cc

diff --git a/src/lzms_common.c b/src/lzms_common.c
index b23a2785..9d9d7b2a 100644
--- a/src/lzms_common.c
+++ b/src/lzms_common.c
@@ -351,9 +351,13 @@ lzms_get_num_offset_slots(size_t uncompressed_size)
 }
 
 void
-lzms_init_probability_entries(struct lzms_probability_entry *entries, size_t count)
+lzms_init_probabilities(struct lzms_probabilites *probs)
 {
-	for (size_t i = 0; i < count; i++) {
+	struct lzms_probability_entry *entries =
+		(struct lzms_probability_entry *)probs;
+	size_t num_entries = sizeof(struct lzms_probabilites) /
+			     sizeof(struct lzms_probability_entry);
+	for (size_t i = 0; i < num_entries; i++) {
 		entries[i].num_recent_zero_bits = LZMS_INITIAL_PROBABILITY;
 		entries[i].recent_bits = LZMS_INITIAL_RECENT_BITS;
 	}