Просмотр исходного кода

opencl : fix possible buffer overflow in dump_tensor (#14490)

zhouwg 6 месяцев назад
Родитель
Сommit
307e79d33d
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      ggml/src/ggml-opencl/ggml-opencl.cpp

+ 1 - 1
ggml/src/ggml-opencl/ggml-opencl.cpp

@@ -3199,7 +3199,7 @@ static void dump_tensor(ggml_backend_t backend, const struct ggml_tensor * tenso
 
     // Open file and dump.
     char fname[512];
-    sprintf(fname, "./tensor-dumps/%s.txt", tensor->name);
+    snprintf(fname, sizeof(fname), "./tensor-dumps/%s.txt", tensor->name);
     FILE * f = fopen(fname, "w");
     if (!f) {
         printf("Failed to open %s\n", fname);