LunaHook-mirror/LunaHook/util/textunion.h
恍兮惚兮 34771acb29 elf
2024-07-10 21:13:08 +08:00

55 lines
1.3 KiB
C++

#pragma once
inline size_t str_len(const char *s){return strlen(s);}
inline size_t str_len(const wchar_t *s){return wcslen(s);}
template<class CharT>
struct TextUnion
{
enum { ShortTextCapacity = 0x10/sizeof(CharT) };
union {
const CharT *text; // 0x0
CharT chars[ShortTextCapacity];
};
int size, // 0x10
capacity;
bool isValid() const
{
if (size <= 0 || size > capacity)
return false;
const CharT *t = getText();
return Engine::isAddressWritable(t, size) && str_len(t) == size;
}
const CharT *getText() const
{ return capacity < ShortTextCapacity ? chars : text; }
void setText(const CharT *_text, int _size)
{
if (_size < ShortTextCapacity)
::memcpy(chars, _text, (_size + 1) * sizeof(CharT));
else
text = _text;
capacity = size = _size;
}
void setLongText(const CharT *_text, int _size)
{
text = _text;
size = _size;
capacity = max(ShortTextCapacity, _size);
}
void setText(const std::basic_string<CharT> &text)
{ setText((const CharT *)text.c_str(), text.size()); }
void setLongText(const std::basic_string<CharT> &text)
{ setLongText((const CharT *)text.c_str(), text.size()); }
};
using TextUnionA=TextUnion<char>;
using TextUnionW=TextUnion<wchar_t>;
// EOF