bblanchon/ArduinoJson
1
0
Fork 1
mirror of https://github.com/bblanchon/ArduinoJson.git synced 2025-07-29 18:27:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

Tests: add TimebombAllocator

Browse Source
This commit is contained in:
Benoit Blanchon
2023-04-07 18:28:41 +02:00
parent 22e4f216c3
commit d95a3bd19a
1 changed files with 35 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
extras/tests/Helpers/Allocators.hpp
View File

@ -177,3 +177,38 @@ class ControllableAllocator : public ArduinoJson::Allocator {
bool _enabled;
Allocator* _upstream;
};
class TimebombAllocator : public ArduinoJson::Allocator {
public:
TimebombAllocator(
size_t initialCountdown,
Allocator* upstream = ArduinoJson::detail::DefaultAllocator::instance())
: _countdown(initialCountdown), _upstream(upstream) {}
virtual ~TimebombAllocator() {}
void* allocate(size_t n) override {
if (!_countdown)
return nullptr;
_countdown--;
return _upstream->allocate(n);
}
void deallocate(void* p) override {
_upstream->deallocate(p);
}
void* reallocate(void* ptr, size_t n) override {
if (!_countdown)
return nullptr;
_countdown--;
return _upstream->reallocate(ptr, n);
}
void setCountdown(size_t value) {
_countdown = value;
}
private:
size_t _countdown = 0;
Allocator* _upstream;
};