You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
125 lines
3.1 KiB
C++
125 lines
3.1 KiB
C++
#include "memflow.hpp"
|
|
#include <cstring>
|
|
#include <stdio.h>
|
|
#include <mutex>
|
|
#include <memory>
|
|
|
|
#define INRANGE(x, a, b) (x >= a && x <= b)
|
|
#define getBits(x) (INRANGE(x, '0', '9') ? (x - '0') : ((x & (~0x20)) - 'A' + 0xa))
|
|
#define getByte(x) (getBits(x[0]) << 4 | getBits(x[1]))
|
|
|
|
typedef uint8_t *PBYTE;
|
|
typedef uint8_t BYTE;
|
|
typedef unsigned long DWORD;
|
|
typedef unsigned short WORD;
|
|
typedef WORD *PWORD;
|
|
|
|
static std::unique_ptr<ConnectorInstance<>> conn = nullptr;
|
|
static std::unique_ptr<OsInstance<>> kernel = nullptr;
|
|
|
|
// set MAX_PHYADDR to a reasonable value, larger values will take more time to traverse.
|
|
constexpr uint64_t MAX_PHYADDR = 0xFFFFFFFFF;
|
|
|
|
inline uint64_t GetFurtherDistance(uint64_t A, uint64_t Min, uint64_t Max)
|
|
{
|
|
uint64_t distanceToMin = (A > Min) ? (A - Min) : (Min - A);
|
|
uint64_t distanceToMax = (A > Max) ? (A - Max) : (Max - A);
|
|
|
|
return (distanceToMin > distanceToMax) ? distanceToMin : distanceToMax;
|
|
}
|
|
|
|
inline bool isMatch(const PBYTE addr, const PBYTE pat, const PBYTE msk)
|
|
{
|
|
size_t n = 0;
|
|
while (addr[n] == pat[n] || msk[n] == (BYTE)'?')
|
|
{
|
|
if (!msk[++n])
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
size_t findPattern(const PBYTE rangeStart, size_t len, const char *pattern);
|
|
|
|
typedef struct Process
|
|
{
|
|
IntoProcessInstance<> hProcess;
|
|
uint64_t baseaddr = 0;
|
|
} Process;
|
|
|
|
enum class process_status : BYTE
|
|
{
|
|
NOT_FOUND,
|
|
FOUND_NO_ACCESS,
|
|
FOUND_READY
|
|
};
|
|
|
|
class Memory
|
|
{
|
|
private:
|
|
Process proc;
|
|
process_status status = process_status::NOT_FOUND;
|
|
std::mutex m;
|
|
uint64_t lastCorrectDtbPhysicalAddress = 0x0;
|
|
|
|
public:
|
|
~Memory() = default;
|
|
|
|
uint64_t get_proc_baseaddr();
|
|
|
|
process_status get_proc_status();
|
|
|
|
void check_proc();
|
|
|
|
void open_proc(const char *name);
|
|
|
|
void close_proc();
|
|
|
|
template <typename T>
|
|
bool Read(uint64_t address, T &out);
|
|
|
|
template <typename T>
|
|
bool ReadArray(uint64_t address, T out[], size_t len);
|
|
|
|
template <typename T>
|
|
bool Write(uint64_t address, const T &value);
|
|
|
|
template <typename T>
|
|
bool WriteArray(uint64_t address, const T value[], size_t len);
|
|
|
|
uint64_t ScanPointer(uint64_t ptr_address, const uint32_t offsets[], int level);
|
|
|
|
bool bruteforceDtb(uint64_t dtbStartPhysicalAddr, const uint64_t stepPage);
|
|
|
|
bool testDtbValue(const uint64_t &dtb_val);
|
|
};
|
|
|
|
template <typename T>
|
|
inline bool Memory::Read(uint64_t address, T &out)
|
|
{
|
|
std::lock_guard<std::mutex> l(m);
|
|
return proc.baseaddr && proc.hProcess.read_raw_into(address, CSliceMut<uint8_t>((char *)&out, sizeof(T))) == 0;
|
|
}
|
|
|
|
template <typename T>
|
|
inline bool Memory::ReadArray(uint64_t address, T out[], size_t len)
|
|
{
|
|
std::lock_guard<std::mutex> l(m);
|
|
return proc.baseaddr && proc.hProcess.read_raw_into(address, CSliceMut<uint8_t>((char *)out, sizeof(T) * len)) == 0;
|
|
}
|
|
|
|
template <typename T>
|
|
inline bool Memory::Write(uint64_t address, const T &value)
|
|
{
|
|
std::lock_guard<std::mutex> l(m);
|
|
return proc.baseaddr && proc.hProcess.write_raw(address, CSliceRef<uint8_t>((char *)&value, sizeof(T))) == 0;
|
|
}
|
|
|
|
template <typename T>
|
|
inline bool Memory::WriteArray(uint64_t address, const T value[], size_t len)
|
|
{
|
|
std::lock_guard<std::mutex> l(m);
|
|
return proc.baseaddr && proc.hProcess.write_raw(address, CSliceRef<uint8_t>((char *)value, sizeof(T) * len)) == 0;
|
|
} |