فهرست منبع

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

zhouwg 6 ماه پیش
والد
کامیت
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.
     // Open file and dump.
     char fname[512];
     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");
     FILE * f = fopen(fname, "w");
     if (!f) {
     if (!f) {
         printf("Failed to open %s\n", fname);
         printf("Failed to open %s\n", fname);