Introduce size_ptr to store pointer and size of the object, and VLATO_prt to store pointers to Variable Length Array-Terminated objects

nataraj-hates-MS-for-stealing-github · nataraj-hates-MS-for-stealing-github · commit ce000e5bc014 · 2021-08-02T12:51:13.000+03:00
diff --git a/blobstamper/helpers.h b/blobstamper/helpers.h
@@ -76,4 +76,46 @@ max precision
 					   "%.999g", __val);
   }
 
+
+template<class T> class sized_ptr
+{
+  private:
+    T* _ptr{nullptr};
+    size_t _size;
+  public:
+
+    operator T*() const {return _ptr;};
+    size_t size() {return _size;};
+    sized_ptr(T* ptr, size_t size): _ptr{ptr}, _size{size} {};
+    ~sized_ptr() {if (_ptr) free(_ptr);};
+};
+
+template<class T, class ArrayT> class VLATO_ptr
+{
+  private:
+    T* _ptr{nullptr};
+    size_t _length;
+  public:
+
+    operator T*() const {return _ptr;};
+    operator sized_ptr<T>() {sized_ptr<T> res(_ptr,size()); _ptr=NULL; return res;};
+
+    size_t length() {return _length;};
+    size_t size()   {return sizeof(T) + _length * sizeof(ArrayT);};
+    VLATO_ptr(size_t length);
+    VLATO_ptr(T* ptr, size_t length): _ptr{ptr}, _length{length} {};
+    ~VLATO_ptr() {if (_ptr) free(_ptr);}
+};
+
+
+template<class T, class ArrayT>
+VLATO_ptr<T,ArrayT>::VLATO_ptr(size_t length)
+{
+  _ptr = (T*) malloc(sizeof(T) + sizeof(ArrayT) * length);
+  _length = length;
+}
+
+
+
+
 #endif  /*HELPERS_H*/
