Răsfoiți Sursa

rpc : avoid uninitialized memory in serialize_tensor (#13210)

Zero out the name and padding buffers.
Justin Santa Barbara 8 luni în urmă
părinte
comite
8efbdadc61
1 a modificat fișierele cu 5 adăugiri și 0 ștergeri
  1. 5 0
      ggml/src/ggml-rpc/ggml-rpc.cpp

+ 5 - 0
ggml/src/ggml-rpc/ggml-rpc.cpp

@@ -518,6 +518,11 @@ static rpc_tensor serialize_tensor(const ggml_tensor * tensor) {
     result.view_src = reinterpret_cast<uint64_t>(tensor->view_src);
     result.view_src = reinterpret_cast<uint64_t>(tensor->view_src);
     result.view_offs = tensor->view_offs;
     result.view_offs = tensor->view_offs;
     result.data = reinterpret_cast<uint64_t>(tensor->data);
     result.data = reinterpret_cast<uint64_t>(tensor->data);
+
+    // Avoid sending uninitialized data over the wire
+    memset(result.name, 0, sizeof(result.name));
+    memset(result.padding, 0, sizeof(result.padding));
+
     snprintf(result.name, GGML_MAX_NAME, "%s", tensor->name);
     snprintf(result.name, GGML_MAX_NAME, "%s", tensor->name);
     return result;
     return result;
 }
 }