From 7fd6a7322a4d084e43f858286e75b47de3070712 Mon Sep 17 00:00:00 2001
From: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
Date: Wed, 4 May 2022 17:46:49 +0200
Subject: [PATCH] Update local energy (size_max)

---
 org/qmckl_local_energy.org | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/org/qmckl_local_energy.org b/org/qmckl_local_energy.org
index c5018c0..aaad9f2 100644
--- a/org/qmckl_local_energy.org
+++ b/org/qmckl_local_energy.org
@@ -1058,11 +1058,11 @@ E_L = KE + PE
 *** Get
 
     #+begin_src c :comments org :tangle (eval h_func) :noweb yes
-qmckl_exit_code qmckl_get_local_energy(qmckl_context context, double* const local_energy);
+qmckl_exit_code qmckl_get_local_energy(qmckl_context context, double* const local_energy, const int64_t size_max);
     #+end_src
 
     #+begin_src c :comments org :tangle (eval c) :noweb yes  :exports none
-qmckl_exit_code qmckl_get_local_energy(qmckl_context context, double * const local_energy) {
+qmckl_exit_code qmckl_get_local_energy(qmckl_context context, double * const local_energy, const int64_t size_max) {
 
   if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
     return QMCKL_NULL_CONTEXT;
@@ -1086,8 +1086,14 @@ qmckl_exit_code qmckl_get_local_energy(qmckl_context context, double * const loc
   qmckl_context_struct* const ctx = (qmckl_context_struct* const) context;
   assert (ctx != NULL);
 
-  size_t sze = ctx->electron.walk_num * sizeof(double);
-  memcpy(local_energy, ctx->local_energy.e_local, sze);
+  const int64_t sze = ctx->electron.walk_num;
+  if (size_max < sze) {
+    return qmckl_failwith( context,                                                                                          
+                           QMCKL_INVALID_ARG_3,                                                                              
+                           "qmckl_get_local_energy",                                                                      
+                           "input array too small");     
+  }
+  memcpy(local_energy, ctx->local_energy.e_local, sze * sizeof(double));
 
   return QMCKL_SUCCESS;
 }
@@ -1292,7 +1298,7 @@ end function qmckl_compute_local_energy_f
 
 double local_energy[walk_num];
 
-rc = qmckl_get_local_energy(context, &(local_energy[0]));
+rc = qmckl_get_local_energy(context, &(local_energy[0]), walk_num);
 assert (rc == QMCKL_SUCCESS);
 
      #+end_src