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

rpc : avoid uninitialized memory in serialize_tensor (#13210)

Zero out the name and padding buffers.
Justin Santa Barbara 8 месяцев назад
Родитель
Сommit
8efbdadc61
1 измененных файлов с 5 добавлено и 0 удалено
  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_offs = tensor->view_offs;
     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);
     return result;
 }