1 /*
2  * Copyright (C) 2015 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #ifndef _MEMORY_REPLAY_POINTERS_H
18 #define _MEMORY_REPLAY_POINTERS_H
19 
20 #include <stdatomic.h>
21 #include <stdint.h>
22 
23 struct pointer_data {
24   std::atomic_uintptr_t key_pointer;
25   void* pointer;
26 };
27 
28 class Pointers {
29  public:
30   Pointers(size_t max_allocs);
31   virtual ~Pointers();
32 
33   void Add(uintptr_t key_pointer, void* pointer);
34 
35   void* Remove(uintptr_t key_pointer);
36 
max_pointers()37   size_t max_pointers() { return max_pointers_; }
38 
39   void FreeAll();
40 
41  private:
42   pointer_data* FindEmpty(uintptr_t key_pointer);
43   pointer_data* Find(uintptr_t key_pointer);
44   size_t GetHash(uintptr_t key_pointer);
45 
46   pointer_data* pointers_ = nullptr;
47   size_t pointers_size_ = 0;
48   size_t max_pointers_ = 0;
49 };
50 
51 #endif // _MEMORY_REPLAY_POINTERS_H
52