xref: /aosp_15_r20/external/pigweed/pw_allocator/allocator.cc (revision 61c4878ac05f98d0ceed94b57d316916de578985)
1*61c4878aSAndroid Build Coastguard Worker // Copyright 2023 The Pigweed Authors
2*61c4878aSAndroid Build Coastguard Worker //
3*61c4878aSAndroid Build Coastguard Worker // Licensed under the Apache License, Version 2.0 (the "License"); you may not
4*61c4878aSAndroid Build Coastguard Worker // use this file except in compliance with the License. You may obtain a copy of
5*61c4878aSAndroid Build Coastguard Worker // the License at
6*61c4878aSAndroid Build Coastguard Worker //
7*61c4878aSAndroid Build Coastguard Worker //     https://www.apache.org/licenses/LICENSE-2.0
8*61c4878aSAndroid Build Coastguard Worker //
9*61c4878aSAndroid Build Coastguard Worker // Unless required by applicable law or agreed to in writing, software
10*61c4878aSAndroid Build Coastguard Worker // distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
11*61c4878aSAndroid Build Coastguard Worker // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
12*61c4878aSAndroid Build Coastguard Worker // License for the specific language governing permissions and limitations under
13*61c4878aSAndroid Build Coastguard Worker // the License.
14*61c4878aSAndroid Build Coastguard Worker 
15*61c4878aSAndroid Build Coastguard Worker #include "pw_allocator/allocator.h"
16*61c4878aSAndroid Build Coastguard Worker 
17*61c4878aSAndroid Build Coastguard Worker #include <algorithm>
18*61c4878aSAndroid Build Coastguard Worker #include <cstring>
19*61c4878aSAndroid Build Coastguard Worker 
20*61c4878aSAndroid Build Coastguard Worker namespace pw {
21*61c4878aSAndroid Build Coastguard Worker 
22*61c4878aSAndroid Build Coastguard Worker using ::pw::allocator::Layout;
23*61c4878aSAndroid Build Coastguard Worker 
DoReallocate(void * ptr,Layout new_layout)24*61c4878aSAndroid Build Coastguard Worker void* Allocator::DoReallocate(void* ptr, Layout new_layout) {
25*61c4878aSAndroid Build Coastguard Worker   if (Resize(ptr, new_layout.size())) {
26*61c4878aSAndroid Build Coastguard Worker     return ptr;
27*61c4878aSAndroid Build Coastguard Worker   }
28*61c4878aSAndroid Build Coastguard Worker   Result<Layout> old_layout = GetUsableLayout(ptr);
29*61c4878aSAndroid Build Coastguard Worker   if (!old_layout.ok()) {
30*61c4878aSAndroid Build Coastguard Worker     return nullptr;
31*61c4878aSAndroid Build Coastguard Worker   }
32*61c4878aSAndroid Build Coastguard Worker   void* new_ptr = Allocate(new_layout);
33*61c4878aSAndroid Build Coastguard Worker   if (new_ptr == nullptr) {
34*61c4878aSAndroid Build Coastguard Worker     return nullptr;
35*61c4878aSAndroid Build Coastguard Worker   }
36*61c4878aSAndroid Build Coastguard Worker   if (ptr != nullptr) {
37*61c4878aSAndroid Build Coastguard Worker     std::memcpy(new_ptr, ptr, std::min(new_layout.size(), old_layout->size()));
38*61c4878aSAndroid Build Coastguard Worker     Deallocate(ptr);
39*61c4878aSAndroid Build Coastguard Worker   }
40*61c4878aSAndroid Build Coastguard Worker   return new_ptr;
41*61c4878aSAndroid Build Coastguard Worker }
42*61c4878aSAndroid Build Coastguard Worker 
DoReallocate(void * ptr,Layout old_layout,size_t new_size)43*61c4878aSAndroid Build Coastguard Worker void* Allocator::DoReallocate(void* ptr, Layout old_layout, size_t new_size) {
44*61c4878aSAndroid Build Coastguard Worker   return Reallocate(ptr, Layout(new_size, old_layout.alignment()));
45*61c4878aSAndroid Build Coastguard Worker }
46*61c4878aSAndroid Build Coastguard Worker 
47*61c4878aSAndroid Build Coastguard Worker }  // namespace pw
48