custom_buffer.h

Go to the documentation of this file.

#pragma once
#include <windows.h>
 
namespace util {
 
    struct AutoBuffer
    {
        AutoBuffer() : buf(nullptr), max_size(0), buf_size(0) { }
 
        ~AutoBuffer() {
            if (buf) {
                ::free(buf);
                buf = nullptr;
            }
            max_size = 0;
            buf_size = 0;
        }
 
        BYTE* alloc(size_t _buf_size)
        {
            if (_buf_size > max_size) {
                BYTE* allocated = (BYTE*)::realloc((void*)buf, _buf_size);
                if (!allocated) {
                    return nullptr;
                }
                buf = allocated;
                max_size = _buf_size;
            }
            buf_size = _buf_size;
            ::memset(buf, 0, max_size);
            return buf;
        }
 
        BYTE* buf;
        size_t max_size;
        size_t buf_size;
    };
 
}; //namespace util