331 lines
10 KiB
C++
331 lines
10 KiB
C++
#pragma once
|
|
|
|
#include <hex.hpp>
|
|
|
|
#include <hex/helpers/concepts.hpp>
|
|
#include <hex/helpers/fs.hpp>
|
|
|
|
#include <array>
|
|
#include <bit>
|
|
#include <cstring>
|
|
#include <cctype>
|
|
#include <concepts>
|
|
#include <functional>
|
|
#include <limits>
|
|
#include <memory>
|
|
#include <optional>
|
|
#include <string>
|
|
#include <type_traits>
|
|
#include <variant>
|
|
#include <vector>
|
|
|
|
#if defined(OS_MACOS)
|
|
#include <hex/helpers/utils_macos.hpp>
|
|
#elif defined(OS_LINUX)
|
|
#include <hex/helpers/utils_linux.hpp>
|
|
#endif
|
|
|
|
struct ImVec2;
|
|
|
|
namespace hex {
|
|
|
|
namespace prv {
|
|
class Provider;
|
|
}
|
|
|
|
template<typename T>
|
|
[[nodiscard]] std::vector<T> sampleData(const std::vector<T> &data, size_t count) {
|
|
size_t stride = std::max(1.0, double(data.size()) / count);
|
|
|
|
std::vector<T> result;
|
|
result.reserve(count);
|
|
|
|
for (size_t i = 0; i < data.size(); i += stride) {
|
|
result.push_back(data[i]);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
[[nodiscard]] float operator""_scaled(long double value);
|
|
[[nodiscard]] float operator""_scaled(unsigned long long value);
|
|
[[nodiscard]] ImVec2 scaled(const ImVec2 &vector);
|
|
|
|
template<typename T>
|
|
[[nodiscard]] std::vector<T> operator|(const std::vector<T> &lhs, const std::vector<T> &rhs) {
|
|
std::vector<T> result;
|
|
|
|
std::copy(lhs.begin(), lhs.end(), std::back_inserter(result));
|
|
std::copy(rhs.begin(), rhs.end(), std::back_inserter(result));
|
|
|
|
return result;
|
|
}
|
|
|
|
[[nodiscard]] std::string to_string(u128 value);
|
|
[[nodiscard]] std::string to_string(i128 value);
|
|
|
|
[[nodiscard]] std::string toLower(std::string string);
|
|
[[nodiscard]] std::string toUpper(std::string string);
|
|
|
|
[[nodiscard]] std::vector<u8> parseHexString(std::string string);
|
|
[[nodiscard]] std::optional<u8> parseBinaryString(const std::string &string);
|
|
[[nodiscard]] std::string toByteString(u64 bytes);
|
|
[[nodiscard]] std::string makePrintable(u8 c);
|
|
|
|
void startProgram(const std::string &command);
|
|
int executeCommand(const std::string &command);
|
|
void openWebpage(std::string url);
|
|
|
|
[[nodiscard]] std::string encodeByteString(const std::vector<u8> &bytes);
|
|
[[nodiscard]] std::vector<u8> decodeByteString(const std::string &string);
|
|
|
|
[[nodiscard]] std::wstring utf8ToUtf16(const std::string& utf8);
|
|
[[nodiscard]] std::string utf16ToUtf8(const std::wstring& utf16);
|
|
|
|
[[nodiscard]] constexpr u64 extract(u8 from, u8 to, const std::unsigned_integral auto &value) {
|
|
if (from < to) std::swap(from, to);
|
|
|
|
using ValueType = std::remove_cvref_t<decltype(value)>;
|
|
ValueType mask = (std::numeric_limits<ValueType>::max() >> (((sizeof(value) * 8) - 1) - (from - to))) << to;
|
|
|
|
return (value & mask) >> to;
|
|
}
|
|
|
|
[[nodiscard]] inline u64 extract(u32 from, u32 to, const std::vector<u8> &bytes) {
|
|
u8 index = 0;
|
|
while (from > 32 && to > 32) {
|
|
from -= 8;
|
|
to -= 8;
|
|
index++;
|
|
}
|
|
|
|
u64 value = 0;
|
|
std::memcpy(&value, &bytes[index], std::min(sizeof(value), bytes.size() - index));
|
|
u64 mask = (std::numeric_limits<u64>::max() >> (64 - (from + 1)));
|
|
|
|
return (value & mask) >> to;
|
|
}
|
|
|
|
[[nodiscard]] constexpr i128 signExtend(size_t numBits, i128 value) {
|
|
i128 mask = 1ULL << (numBits - 1);
|
|
return (value ^ mask) - mask;
|
|
}
|
|
|
|
template<std::integral T>
|
|
[[nodiscard]] constexpr T swapBitOrder(size_t numBits, T value) {
|
|
T result = 0x00;
|
|
|
|
for (size_t bit = 0; bit < numBits; bit++) {
|
|
result <<= 1;
|
|
result |= (value & (1 << bit)) != 0;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
[[nodiscard]] constexpr size_t strnlen(const char *s, size_t n) {
|
|
size_t i = 0;
|
|
while (i < n && s[i] != '\x00') i++;
|
|
|
|
return i;
|
|
}
|
|
|
|
template<size_t>
|
|
struct SizeTypeImpl { };
|
|
|
|
template<>
|
|
struct SizeTypeImpl<1> { using Type = u8; };
|
|
template<>
|
|
struct SizeTypeImpl<2> { using Type = u16; };
|
|
template<>
|
|
struct SizeTypeImpl<4> { using Type = u32; };
|
|
template<>
|
|
struct SizeTypeImpl<8> { using Type = u64; };
|
|
template<>
|
|
struct SizeTypeImpl<16> { using Type = u128; };
|
|
|
|
template<size_t Size>
|
|
using SizeType = typename SizeTypeImpl<Size>::Type;
|
|
|
|
template<typename T>
|
|
[[nodiscard]] constexpr T changeEndianess(const T &value, size_t size, std::endian endian) {
|
|
if (endian == std::endian::native)
|
|
return value;
|
|
|
|
size = std::min(size, sizeof(T));
|
|
|
|
std::array<uint8_t, sizeof(T)> data = { 0 };
|
|
std::memcpy(&data[0], &value, size);
|
|
|
|
for (uint32_t i = 0; i < size / 2; i++) {
|
|
std::swap(data[i], data[size - 1 - i]);
|
|
}
|
|
|
|
T result = { };
|
|
std::memcpy(&result, &data[0], size);
|
|
|
|
return result;
|
|
}
|
|
|
|
template<typename T>
|
|
[[nodiscard]] constexpr T changeEndianess(const T &value, std::endian endian) {
|
|
return changeEndianess(value, sizeof(value), endian);
|
|
}
|
|
|
|
[[nodiscard]] constexpr u128 bitmask(u8 bits) {
|
|
return u128(-1) >> (128 - bits);
|
|
}
|
|
|
|
template<class T>
|
|
[[nodiscard]] constexpr T bit_width(T x) noexcept {
|
|
return std::numeric_limits<T>::digits - std::countl_zero(x);
|
|
}
|
|
|
|
template<typename T>
|
|
[[nodiscard]] constexpr T bit_ceil(T x) noexcept {
|
|
if (x <= 1u)
|
|
return T(1);
|
|
|
|
return T(1) << bit_width(T(x - 1));
|
|
}
|
|
|
|
template<std::integral T, std::integral U>
|
|
[[nodiscard]] auto powi(T base, U exp) {
|
|
using ResultType = decltype(T{} * U{});
|
|
|
|
if (exp < 0)
|
|
return ResultType(0);
|
|
|
|
ResultType result = 1;
|
|
|
|
while (exp != 0) {
|
|
if ((exp & 0b1) == 0b1)
|
|
result *= base;
|
|
exp >>= 1;
|
|
base *= base;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
template<typename T, typename... Args>
|
|
void moveToVector(std::vector<T> &buffer, T &&first, Args &&...rest) {
|
|
buffer.push_back(std::move(first));
|
|
|
|
if constexpr (sizeof...(rest) > 0)
|
|
moveToVector(buffer, std::move(rest)...);
|
|
}
|
|
|
|
template<typename T, typename... Args>
|
|
[[nodiscard]] std::vector<T> moveToVector(T &&first, Args &&...rest) {
|
|
std::vector<T> result;
|
|
moveToVector(result, T(std::move(first)), std::move(rest)...);
|
|
|
|
return result;
|
|
}
|
|
|
|
[[nodiscard]] std::vector<std::string> splitString(const std::string &string, const std::string &delimiter);
|
|
[[nodiscard]] std::string combineStrings(const std::vector<std::string> &strings, const std::string &delimiter = "");
|
|
[[nodiscard]] std::string replaceStrings(std::string string, const std::string &search, const std::string &replace);
|
|
|
|
[[nodiscard]] std::string toEngineeringString(double value);
|
|
|
|
[[nodiscard]] inline std::vector<u8> parseByteString(const std::string &string) {
|
|
auto byteString = std::string(string);
|
|
std::erase(byteString, ' ');
|
|
|
|
if ((byteString.length() % 2) != 0) return {};
|
|
|
|
std::vector<u8> result;
|
|
for (u32 i = 0; i < byteString.length(); i += 2) {
|
|
if (!std::isxdigit(byteString[i]) || !std::isxdigit(byteString[i + 1]))
|
|
return {};
|
|
|
|
result.push_back(std::strtoul(byteString.substr(i, 2).c_str(), nullptr, 16));
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
[[nodiscard]] std::string toBinaryString(std::unsigned_integral auto number) {
|
|
if (number == 0) return "0";
|
|
|
|
std::string result;
|
|
for (i16 bit = hex::bit_width(number) - 1; bit >= 0; bit -= 1)
|
|
result += (number & (0b1 << bit)) == 0 ? '0' : '1';
|
|
|
|
return result;
|
|
}
|
|
|
|
[[nodiscard]] float float16ToFloat32(u16 float16);
|
|
|
|
[[nodiscard]] inline bool equalsIgnoreCase(const std::string &left, const std::string &right) {
|
|
return std::equal(left.begin(), left.end(), right.begin(), right.end(), [](char a, char b) {
|
|
return tolower(a) == tolower(b);
|
|
});
|
|
}
|
|
|
|
[[nodiscard]] inline bool containsIgnoreCase(const std::string &a, const std::string &b) {
|
|
auto iter = std::search(a.begin(), a.end(), b.begin(), b.end(), [](char ch1, char ch2) {
|
|
return std::toupper(ch1) == std::toupper(ch2);
|
|
});
|
|
|
|
return iter != a.end();
|
|
}
|
|
|
|
template<typename T, typename... VariantTypes>
|
|
[[nodiscard]] T get_or(const std::variant<VariantTypes...> &variant, T alt) {
|
|
const T *value = std::get_if<T>(&variant);
|
|
if (value == nullptr)
|
|
return alt;
|
|
else
|
|
return *value;
|
|
}
|
|
|
|
template<std::integral T>
|
|
[[nodiscard]] T alignTo(T value, T alignment) {
|
|
T remainder = value % alignment;
|
|
|
|
return remainder != 0 ? value + (alignment - remainder) : value;
|
|
}
|
|
|
|
[[nodiscard]] std::optional<u8> hexCharToValue(char c);
|
|
|
|
[[nodiscard]] bool isProcessElevated();
|
|
|
|
[[nodiscard]] std::optional<std::string> getEnvironmentVariable(const std::string &env);
|
|
|
|
[[nodiscard]] inline std::string limitStringLength(const std::string &string, size_t maxLength) {
|
|
// If the string is shorter than the max length, return it as is
|
|
if (string.size() < maxLength)
|
|
return string;
|
|
|
|
// If the string is longer than the max length, find the last space before the max length
|
|
auto it = string.begin() + maxLength;
|
|
while (it != string.begin() && !std::isspace(*it)) --it;
|
|
|
|
// If there's no space before the max length, just cut the string
|
|
if (it == string.begin()) {
|
|
it = string.begin() + maxLength;
|
|
|
|
// Try to find a UTF-8 character boundary
|
|
while (it != string.begin() && (*it & 0x80) != 0x00) --it;
|
|
++it;
|
|
}
|
|
|
|
// If we still didn't find a valid boundary, just return the string as is
|
|
if (it == string.begin())
|
|
return string;
|
|
|
|
// Append
|
|
return std::string(string.begin(), it) + "…";
|
|
}
|
|
|
|
[[nodiscard]] std::optional<std::fs::path> getInitialFilePath();
|
|
|
|
[[nodiscard]] std::string generateHexView(u64 offset, u64 size, prv::Provider *provider);
|
|
[[nodiscard]] std::string generateHexView(u64 offset, const std::vector<u8> &data);
|
|
|
|
[[nodiscard]] std::string formatSystemError(i32 error);
|
|
|
|
}
|