mirror of
https://github.com/Xahau/xahaud.git
synced 2025-12-06 17:27:52 +00:00
fix for current codebase
This commit is contained in:
93
include/xrpl/hook/guard_checker.cpp
Normal file
93
include/xrpl/hook/guard_checker.cpp
Normal file
@@ -0,0 +1,93 @@
|
||||
#include "Guard.h"
|
||||
#include <fcntl.h>
|
||||
#include <iostream>
|
||||
#include <optional>
|
||||
#include <ostream>
|
||||
#include <stdint.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <string_view>
|
||||
#include <sys/ioctl.h>
|
||||
#include <sys/stat.h>
|
||||
#include <unistd.h>
|
||||
#include <vector>
|
||||
|
||||
int
|
||||
main(int argc, char** argv)
|
||||
{
|
||||
const char* fin = 0;
|
||||
|
||||
if (argc > 2)
|
||||
return fprintf(
|
||||
stderr, "Guard Checker\n\tUsage: %s somefile.wasm\n", argv[0]);
|
||||
else if (argc == 1)
|
||||
fin = "-";
|
||||
else
|
||||
fin = argv[1];
|
||||
|
||||
int fd = 0;
|
||||
if (strcmp(fin, "-") != 0)
|
||||
fd = open(fin, O_RDONLY);
|
||||
|
||||
if (fd < 0)
|
||||
return fprintf(stderr, "Could not open file for reading:`%s`\n", fin);
|
||||
|
||||
off_t len = fd == 0 ? 0 : lseek(fd, 0, SEEK_END);
|
||||
|
||||
if (fd != 0)
|
||||
lseek(fd, 0, SEEK_SET);
|
||||
|
||||
int length_known = len > 0;
|
||||
|
||||
uint8_t hook_data[0x100000U];
|
||||
|
||||
uint8_t* ptr = hook_data;
|
||||
|
||||
size_t upto = 0;
|
||||
|
||||
fcntl(fd, F_SETFL, fcntl(0, F_GETFL) & ~O_NONBLOCK);
|
||||
|
||||
while (ptr - hook_data < sizeof(hook_data))
|
||||
{
|
||||
if (length_known)
|
||||
{
|
||||
if (upto >= len)
|
||||
break;
|
||||
}
|
||||
else
|
||||
len = upto + 1;
|
||||
|
||||
size_t bytes_read = read(fd, ptr + upto, len - upto);
|
||||
|
||||
if (!length_known && bytes_read == 0)
|
||||
break;
|
||||
|
||||
if (bytes_read < 0)
|
||||
return fprintf(
|
||||
stderr,
|
||||
"Error reading file `%s`, only %ld bytes could be read\n",
|
||||
fin,
|
||||
upto);
|
||||
|
||||
upto += bytes_read;
|
||||
}
|
||||
|
||||
std::vector<uint8_t> hook(upto);
|
||||
memcpy(hook.data(), hook_data, upto);
|
||||
|
||||
printf("Read %ld bytes from `%s` successfully...\n", upto, fin);
|
||||
|
||||
close(fd);
|
||||
|
||||
auto result = validateGuards(hook, std::cout, "", 3);
|
||||
|
||||
if (!result)
|
||||
{
|
||||
printf("Hook validation failed.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
printf("\nHook validation successful!\n");
|
||||
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user