From 764c7afee781c2d2d1966df6214073d467446767 Mon Sep 17 00:00:00 2001 From: ngxson Date: Fri, 22 Mar 2024 15:10:52 +0100 Subject: [PATCH] fix llama_split_prefix --- llama.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/llama.cpp b/llama.cpp index 7da11d6be..0af78c6a3 100644 --- a/llama.cpp +++ b/llama.cpp @@ -14829,13 +14829,13 @@ LLAMA_API int llama_split_path(char * split_path, size_t maxlen, const char * pa int llama_split_prefix(char * dest, size_t maxlen, const char * split_path, int split_no, int split_count) { std::string str_split_path(split_path); char postfix[32]; - sprintf(postfix, "-%05d-of-%05d.gguf", split_no + 1, split_count); + snprintf(postfix, 32, "-%05d-of-%05d.gguf", split_no + 1, split_count); std::string str_postfix(postfix); // check if dest ends with postfix - auto size_prefix = str_split_path.size() - str_postfix.size(); + int size_prefix = str_split_path.size() - str_postfix.size(); if (size_prefix > 0 && str_split_path.find(str_postfix, size_prefix) != std::string::npos) { - strncpy(dest, split_path, std::min(size_prefix, maxlen)); + snprintf(dest, std::min((size_t) size_prefix, maxlen), "%s", split_path); return size_prefix; }