mirror of
https://github.com/scylladb/scylladb.git
synced 2026-06-03 05:26:58 +00:00
fcb8d040e8
Instead of lengthy blurbs, switch to single-line, machine-readable standardized (https://spdx.dev) license identifiers. The Linux kernel switched long ago, so there is strong precedent. Three cases are handled: AGPL-only, Apache-only, and dual licensed. For the latter case, I chose (AGPL-3.0-or-later and Apache-2.0), reasoning that our changes are extensive enough to apply our license. The changes we applied mechanically with a script, except to licenses/README.md. Closes #9937
54 lines
1.4 KiB
C++
54 lines
1.4 KiB
C++
/*
|
|
* Copyright (C) 2017-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "utils/allocation_strategy.hh"
|
|
|
|
class failure_injecting_allocation_strategy : public allocation_strategy {
|
|
allocation_strategy& _delegate;
|
|
uint64_t _alloc_count = 0;
|
|
uint64_t _fail_at = std::numeric_limits<uint64_t>::max();
|
|
public:
|
|
failure_injecting_allocation_strategy(allocation_strategy& delegate) : _delegate(delegate) {}
|
|
|
|
virtual void* alloc(migrate_fn mf, size_t size, size_t alignment) override {
|
|
if (_alloc_count >= _fail_at) {
|
|
stop_failing();
|
|
throw std::bad_alloc();
|
|
}
|
|
++_alloc_count;
|
|
return _delegate.alloc(mf, size, alignment);
|
|
}
|
|
|
|
virtual void free(void* ptr, size_t size) override {
|
|
_delegate.free(ptr, size);
|
|
}
|
|
|
|
virtual void free(void* ptr) override {
|
|
_delegate.free(ptr);
|
|
}
|
|
|
|
virtual size_t object_memory_size_in_allocator(const void* obj) const noexcept override {
|
|
return _delegate.object_memory_size_in_allocator(obj);
|
|
}
|
|
|
|
// Counts allocation attempts which are not failed due to fail_at().
|
|
uint64_t alloc_count() const {
|
|
return _alloc_count;
|
|
}
|
|
|
|
void fail_after(uint64_t count) {
|
|
_fail_at = _alloc_count + count;
|
|
}
|
|
|
|
void stop_failing() {
|
|
_fail_at = std::numeric_limits<uint64_t>::max();
|
|
}
|
|
};
|