X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=test%2Flib%2Fcommon.hpp;h=b5ef3525d2f35503d38a5ca54f38f3c4b8bf3246;hb=a93908cc9c40eef179d33a89fc011e2e3c54c438;hp=ecfa07d413535b8aab9ed83650729320d7d3670b;hpb=e9b1c46e3490e2940ec36cfa53ffcee5945a27d2;p=awesomized%2Flibmemcached

diff --git a/test/lib/common.hpp b/test/lib/common.hpp
index ecfa07d4..b5ef3525 100644
--- a/test/lib/common.hpp
+++ b/test/lib/common.hpp
@@ -105,7 +105,7 @@ public:
   }
 };
 
-template<class T, void (*F)(void*) = free>
+template<class T>
 class Malloced {
   T *ptr;
 public:
@@ -113,9 +113,15 @@ public:
   Malloced(T *ptr_)
   : ptr{ptr_}
   {}
+  Malloced &operator=(T *ptr_) {
+    if (ptr)
+      free(ptr);
+    ptr = ptr_;
+    return *this;
+  }
   ~Malloced() {
     if(ptr)
-      F(ptr);
+      free(ptr);
   }
   auto operator *() {
     return ptr;