gctf2023/pwn/flipper/dist/common/include/ustl/memlink.h
2023-11-24 13:11:34 -05:00

99 lines
4.2 KiB
C++

// This file is part of the uSTL library, an STL implementation.
//
// Copyright (c) 2005 by Mike Sharov <msharov@users.sourceforge.net>
// This file is free software, distributed under the MIT License.
#pragma once
#include "cmemlink.h"
#include "ualgo.h"
namespace ustl {
/// \class memlink memlink.h ustl.h
/// \ingroup MemoryManagement
///
/// \brief Wrapper for pointer to block with size.
///
/// Use this class the way you would a pointer to an allocated unstructured block.
/// The pointer and block size are available through member functions and cast operator.
///
/// Example usage:
/// \code
/// void* p = malloc (46721);
/// memlink a, b;
/// a.link (p, 46721);
/// assert (a.size() == 46721));
/// b = a;
/// assert (b.size() == 46721));
/// assert (b.begin() + 34 == a.begin + 34);
/// assert (0 == memcmp (a, b, 12));
/// a.fill (673, b, 42, 67);
/// b.erase (87, 12);
/// \endcode
///
class memlink : public cmemlink {
public:
typedef value_type* pointer;
typedef cmemlink::pointer const_pointer;
typedef cmemlink::const_iterator const_iterator;
typedef pointer iterator;
typedef const memlink& rcself_t;
public:
inline memlink (void) : cmemlink() {}
inline memlink (void* p, size_type n) : cmemlink (p, n) {}
inline memlink (const void* p, size_type n) : cmemlink (p, n) {}
inline memlink (rcself_t l) : cmemlink (l) {}
inline explicit memlink (const cmemlink& l) : cmemlink (l) {}
inline pointer data (void) { return const_cast<pointer>(cmemlink::data()); }
inline const_pointer data (void) const { return cmemlink::data(); }
inline iterator begin (void) { return iterator (data()); }
inline iterator iat (size_type i) { assert (i <= size()); return begin() + i; }
inline iterator end (void) { return iat (size()); }
inline const_iterator begin (void) const { return cmemlink::begin(); }
inline const_iterator end (void) const { return cmemlink::end(); }
inline const_iterator iat (size_type i) const { return cmemlink::iat (i); }
size_type writable_size (void) const { return size(); }
inline rcself_t operator= (const cmemlink& l) { cmemlink::operator= (l); return *this; }
inline rcself_t operator= (rcself_t l) { cmemlink::operator= (l); return *this; }
inline void link (const void* p, size_type n) { cmemlink::link (p, n); }
inline void link (void* p, size_type n) { cmemlink::link (p, n); }
inline void link (const cmemlink& l) { cmemlink::link (l); }
inline void link (memlink& l) { cmemlink::link (l); }
inline void link (const void* first, const void* last) { link (first, distance (first, last)); }
inline void link (void* first, void* last) { link (first, distance (first, last)); }
inline void relink (const void* p, size_type n) { cmemlink::relink (p, n); }
inline void relink (void* p, size_type n) { cmemlink::relink (p, n); }
inline void swap (memlink& l) { cmemlink::swap (l); }
void fill (const_iterator start, const void* p, size_type elsize, size_type elCount = 1) noexcept;
inline void insert (const_iterator start, size_type size);
inline void erase (const_iterator start, size_type size);
void read (istream& is);
};
/// Shifts the data in the linked block from \p start to \p start + \p n.
/// The contents of the uncovered bytes is undefined.
inline void memlink::insert (const_iterator cstart, size_type n)
{
assert (data() || !n);
assert (cmemlink::begin() || !n);
assert (cstart >= begin() && cstart + n <= end());
iterator start = const_cast<iterator>(cstart);
rotate (start, end() - n, end());
}
/// Shifts the data in the linked block from \p start + \p n to \p start.
/// The contents of the uncovered bytes is undefined.
inline void memlink::erase (const_iterator cstart, size_type n)
{
assert (data() || !n);
assert (cmemlink::begin() || !n);
assert (cstart >= begin() && cstart + n <= end());
iterator start = const_cast<iterator>(cstart);
rotate (start, start + n, end());
}
/// Use with memlink-derived classes to allocate and link to stack space.
#define alloca_link(m,n) (m).link (alloca (n), (n))
} // namespace ustl