-
Notifications
You must be signed in to change notification settings - Fork 53
/
rvbuffer.cpp
80 lines (66 loc) · 2.33 KB
/
rvbuffer.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#include <catch2/catch_test_macros.hpp>
#include <catch2/matchers/catch_matchers_string.hpp>
#include <libriscv/machine.hpp>
extern std::vector<uint8_t> build_and_load(const std::string& code,
const std::string& args = "-O2 -static", bool cpp = false);
static const uint64_t MAX_MEMORY = 8ul << 20; /* 8MB */
static const uint64_t MAX_INSTRUCTIONS = 10'000'000ul;
using namespace riscv;
TEST_CASE("Sequential buffer", "[Buffer]")
{
const auto binary = build_and_load(R"M(
int main() {
return 666;
})M");
riscv::Machine<RISCV64> machine { binary, { .memory_max = MAX_MEMORY } };
// We need to install Linux system calls for maximum gucciness
machine.setup_linux_syscalls();
// We need to create a Linux environment for runtimes to work well
machine.setup_linux(
{"vmcall"},
{"LC_TYPE=C", "LC_ALL=C", "USER=root"});
auto origin = machine.memory.mmap_allocate(1);
static const char hello[] = "hello world!";
for (auto addr = origin; addr < origin + 4096 - 12; addr ++)
{
machine.memory.memcpy(addr, hello, sizeof(hello));
auto buf = machine.memory.rvbuffer(addr, 12);
REQUIRE(buf.is_sequential());
REQUIRE(buf.size() == 12);
REQUIRE(buf.strview() == "hello world!");
REQUIRE(buf.to_string() == "hello world!");
}
// maxlen works
REQUIRE_THROWS_WITH([&] {
machine.memory.rvbuffer(origin, 128, 127);
}(), Catch::Matchers::ContainsSubstring("Protection fault"));
}
TEST_CASE("Boundary buffer", "[Buffer]")
{
const auto binary = build_and_load(R"M(
int main() {
return 666;
})M");
riscv::Machine<RISCV64> machine { binary, { .memory_max = MAX_MEMORY } };
// We need to install Linux system calls for maximum gucciness
machine.setup_linux_syscalls();
// We need to create a Linux environment for runtimes to work well
machine.setup_linux(
{"vmcall"},
{"LC_TYPE=C", "LC_ALL=C", "USER=root"});
auto origin = machine.memory.mmap_allocate(1);
static const char hello[] = "hello world!";
char buffer[13];
for (auto addr = origin + 4096 - 11; addr < origin + 4095; addr++)
{
machine.memory.memcpy(addr, hello, sizeof(hello));
auto buf = machine.memory.rvbuffer(addr, 12);
REQUIRE(buf.is_sequential() == false);
REQUIRE(buf.size() == 12);
REQUIRE(buf.to_string() == "hello world!");
// String view is no longer possible
buf.copy_to(buffer, 12);
buffer[12] = 0;
REQUIRE(std::string(buffer) == "hello world!");
}
}