blob: 5af4876867337d275ca177c335ba295bb21e3fda (
plain)
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
81
82
83
84
85
86
87
88
89
90
91
92
93
|
#pragma once
// IWYU pragma: private, include "rlbox.hpp"
// IWYU pragma: friend "rlbox_.*\.hpp"
#include <map>
#ifndef RLBOX_USE_CUSTOM_SHARED_LOCK
# include <shared_mutex>
#endif
#include <type_traits>
#include "rlbox_helpers.hpp"
#include "rlbox_type_traits.hpp"
#include "rlbox_types.hpp"
namespace rlbox {
template<typename T_PointerType>
class app_pointer_map
{
private:
using T_PointerTypeUnsigned = detail::unsigned_int_of_size_t<T_PointerType>;
std::map<T_PointerTypeUnsigned, void*> pointer_map;
T_PointerTypeUnsigned counter = 1;
#ifndef RLBOX_SINGLE_THREADED_INVOCATIONS
RLBOX_SHARED_LOCK(map_mutex);
#endif
T_PointerType get_unused_index(T_PointerType max_ptr_val)
{
const auto max_val = (T_PointerTypeUnsigned)max_ptr_val;
for (T_PointerTypeUnsigned i = counter; i <= max_val; i++) {
if (pointer_map.find(i) == pointer_map.end()) {
counter = i + 1;
return (T_PointerType)i;
}
}
for (T_PointerTypeUnsigned i = 1; i < counter; i++) {
if (pointer_map.find(i) == pointer_map.end()) {
counter = i + 1;
return (T_PointerType)i;
}
}
detail::dynamic_check(false, "Could not find free app pointer slot");
return 0;
}
public:
app_pointer_map()
{
// ensure we can't use app pointer 0 as this is sometimes confused as null
// by the sandbox
pointer_map[0] = nullptr;
}
T_PointerType get_app_pointer_idx(void* ptr, T_PointerType max_ptr_val)
{
#ifndef RLBOX_SINGLE_THREADED_INVOCATIONS
RLBOX_ACQUIRE_UNIQUE_GUARD(lock, map_mutex);
#endif
T_PointerType idx = get_unused_index(max_ptr_val);
T_PointerTypeUnsigned idx_int = (T_PointerTypeUnsigned)idx;
pointer_map[idx_int] = ptr;
return idx;
}
void remove_app_ptr(T_PointerType idx)
{
#ifndef RLBOX_SINGLE_THREADED_INVOCATIONS
RLBOX_ACQUIRE_UNIQUE_GUARD(lock, map_mutex);
#endif
T_PointerTypeUnsigned idx_int = (T_PointerTypeUnsigned)idx;
auto it = pointer_map.find(idx_int);
detail::dynamic_check(it != pointer_map.end(),
"Error: removing a non-existing app pointer");
pointer_map.erase(it);
}
void* lookup_index(T_PointerType idx)
{
#ifndef RLBOX_SINGLE_THREADED_INVOCATIONS
RLBOX_ACQUIRE_SHARED_GUARD(lock, map_mutex);
#endif
T_PointerTypeUnsigned idx_int = (T_PointerTypeUnsigned)idx;
auto it = pointer_map.find(idx_int);
detail::dynamic_check(it != pointer_map.end(),
"Error: looking up a non-existing app pointer");
return it->second;
}
};
}
|
