| // Copyright 2018 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef RETRY_ALLOCATOR_H |
| #define RETRY_ALLOCATOR_H |
| |
| #include <map> |
| |
| #include <lib/fit/function.h> |
| |
| #include "macros.h" |
| |
| namespace magma { |
| |
| // This is an allocator that calls a user-specified allocation function on the proposed address |
| // range, and if that fails it tries new address ranges until it succeeds or it runs out of address |
| // space. |
| class RetryAllocator final { |
| public: |
| using AllocationFunction = fit::function<bool(uint64_t)>; |
| |
| static std::unique_ptr<RetryAllocator> Create(uint64_t base, uint64_t size); |
| |
| bool Alloc(size_t size, uint8_t align_pow2, AllocationFunction map_function, |
| uint64_t* addr_out); |
| bool Free(uint64_t addr, uint64_t size); |
| |
| uint64_t base() const { return base_; } |
| uint64_t size() const { return size_; } |
| |
| private: |
| RetryAllocator(uint64_t base, size_t size); |
| |
| // Retrys from the address to the length of the region. |
| std::map<uint64_t, uint64_t> free_regions_; |
| |
| uint64_t base_; |
| uint64_t size_; |
| |
| DISALLOW_COPY_AND_ASSIGN(RetryAllocator); |
| }; |
| |
| } // namespace magma |
| |
| #endif // RETRY_ALLOCATOR_H |