mirror of
https://github.com/upx/upx
synced 2025-10-05 19:20:23 +08:00
89 lines
3.0 KiB
C++
89 lines
3.0 KiB
C++
/* mem.h --
|
|
|
|
This file is part of the UPX executable compressor.
|
|
|
|
Copyright (C) 1996-2022 Markus Franz Xaver Johannes Oberhumer
|
|
Copyright (C) 1996-2022 Laszlo Molnar
|
|
All Rights Reserved.
|
|
|
|
UPX and the UCL library are free software; you can redistribute them
|
|
and/or modify them under the terms of the GNU General Public License as
|
|
published by the Free Software Foundation; either version 2 of
|
|
the License, or (at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; see the file COPYING.
|
|
If not, write to the Free Software Foundation, Inc.,
|
|
59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
|
|
Markus F.X.J. Oberhumer Laszlo Molnar
|
|
<markus@oberhumer.com> <ezerotven+github@gmail.com>
|
|
*/
|
|
|
|
|
|
#ifndef __UPX_MEM_H
|
|
#define __UPX_MEM_H 1
|
|
|
|
|
|
/*************************************************************************
|
|
//
|
|
**************************************************************************/
|
|
|
|
class MemBuffer
|
|
{
|
|
public:
|
|
MemBuffer() : b(nullptr), b_size(0) { }
|
|
explicit MemBuffer(upx_uint64_t size);
|
|
~MemBuffer();
|
|
|
|
static unsigned getSizeForCompression(unsigned uncompressed_size, unsigned extra=0);
|
|
static unsigned getSizeForUncompression(unsigned uncompressed_size, unsigned extra=0);
|
|
|
|
void alloc(upx_uint64_t size);
|
|
void allocForCompression(unsigned uncompressed_size, unsigned extra=0);
|
|
void allocForUncompression(unsigned uncompressed_size, unsigned extra=0);
|
|
|
|
void dealloc();
|
|
|
|
void checkState() const;
|
|
|
|
unsigned getSize() const { return b_size; }
|
|
|
|
operator unsigned char * () { return b; }
|
|
operator const unsigned char * () const { return b; }
|
|
|
|
void *getVoidPtr() { return (void *) b; }
|
|
const void *getVoidPtr() const { return (const void *) b; }
|
|
|
|
void fill(unsigned off, unsigned len, int value);
|
|
void clear(unsigned off, unsigned len) { fill(off, len, 0); }
|
|
void clear() { fill(0, b_size, 0); }
|
|
|
|
// If the entire range [skip, take+skip) is inside the buffer,
|
|
// then return &b[skip]; else throwCantPack(sprintf(errfmt, skip, take)).
|
|
// This is similar to BoundedPtr, except only checks once.
|
|
unsigned char *subref(char const *errfmt, unsigned skip, unsigned take);
|
|
|
|
private:
|
|
unsigned char *b;
|
|
unsigned b_size;
|
|
|
|
static unsigned global_alloc_counter;
|
|
|
|
// disable copy, assignment and move assignment
|
|
MemBuffer(const MemBuffer &) = delete;
|
|
MemBuffer& operator= (const MemBuffer &) = delete;
|
|
MemBuffer& operator= (MemBuffer &&) = delete;
|
|
// disable dynamic allocation
|
|
ACC_CXX_DISABLE_NEW_DELETE
|
|
};
|
|
|
|
#endif /* already included */
|
|
|
|
/* vim:set ts=4 sw=4 et: */
|