ICU-22764 Fix gendict memory safety in toml uchars mode

unicode-org · May 3, 2024 · 23bf38f · 23bf38f
1 parent 464531e
commit 23bf38f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/icu4c/source/tools/gendict/gendict.cpp b/icu4c/source/tools/gendict/gendict.cpp
@@ -438,7 +438,9 @@ int  main(int argc, char **argv) {
         fprintf(f, "transform_type = \"%s\"\n", isOffset ? "offset" : "none");
         fprintf(f, "transform_offset = %d\n", offset);
 
-        usrc_writeArray(f, "trie_data = [\n  ",  outData, isBytesTrie ? 8 : 16, outDataSize, "  ", "\n]\n");
+        int32_t outDataWidth = isBytesTrie ? 8 : 16;
+        int32_t outDataLength = isBytesTrie ? outDataSize : outDataSize / U_SIZEOF_UCHAR;
+        usrc_writeArray(f, "trie_data = [\n  ",  outData, outDataWidth, outDataLength, "  ", "\n]\n");
 
 
         fclose(f);