| OLD | NEW |
| 1 // Copyright (c) 2008, Google Inc. | 1 // Copyright (c) 2008, Google Inc. |
| 2 // All rights reserved. | 2 // All rights reserved. |
| 3 // | 3 // |
| 4 // Redistribution and use in source and binary forms, with or without | 4 // Redistribution and use in source and binary forms, with or without |
| 5 // modification, are permitted provided that the following conditions are | 5 // modification, are permitted provided that the following conditions are |
| 6 // met: | 6 // met: |
| 7 // | 7 // |
| 8 // * Redistributions of source code must retain the above copyright | 8 // * Redistributions of source code must retain the above copyright |
| 9 // notice, this list of conditions and the following disclaimer. | 9 // notice, this list of conditions and the following disclaimer. |
| 10 // * Redistributions in binary form must reproduce the above | 10 // * Redistributions in binary form must reproduce the above |
| (...skipping 19 matching lines...) Expand all Loading... |
| 30 // --- | 30 // --- |
| 31 // Author: Sanjay Ghemawat <opensource@google.com> | 31 // Author: Sanjay Ghemawat <opensource@google.com> |
| 32 | 32 |
| 33 #ifndef TCMALLOC_PAGE_HEAP_ALLOCATOR_H_ | 33 #ifndef TCMALLOC_PAGE_HEAP_ALLOCATOR_H_ |
| 34 #define TCMALLOC_PAGE_HEAP_ALLOCATOR_H_ | 34 #define TCMALLOC_PAGE_HEAP_ALLOCATOR_H_ |
| 35 | 35 |
| 36 #include <stddef.h> // for NULL, size_t | 36 #include <stddef.h> // for NULL, size_t |
| 37 | 37 |
| 38 #include "common.h" // for MetaDataAlloc | 38 #include "common.h" // for MetaDataAlloc |
| 39 #include "free_list.h" // for FL_Push/FL_Pop | 39 #include "free_list.h" // for FL_Push/FL_Pop |
| 40 #include "internal_logging.h" // for ASSERT, CRASH | 40 #include "internal_logging.h" // for ASSERT |
| 41 #include "system-alloc.h" // for TCMalloc_SystemAddGuard | 41 #include "system-alloc.h" // for TCMalloc_SystemAddGuard |
| 42 | 42 |
| 43 namespace tcmalloc { | 43 namespace tcmalloc { |
| 44 | 44 |
| 45 // Simple allocator for objects of a specified type. External locking | 45 // Simple allocator for objects of a specified type. External locking |
| 46 // is required before accessing one of these objects. | 46 // is required before accessing one of these objects. |
| 47 template <class T> | 47 template <class T> |
| 48 class PageHeapAllocator { | 48 class PageHeapAllocator { |
| 49 public: | 49 public: |
| 50 // We use an explicit Init function because these variables are statically | 50 // We use an explicit Init function because these variables are statically |
| (...skipping 13 matching lines...) Expand all Loading... |
| 64 // Consult free list | 64 // Consult free list |
| 65 void* result; | 65 void* result; |
| 66 if (free_list_ != NULL) { | 66 if (free_list_ != NULL) { |
| 67 result = FL_Pop(&free_list_); | 67 result = FL_Pop(&free_list_); |
| 68 } else { | 68 } else { |
| 69 if (free_avail_ < sizeof(T)) { | 69 if (free_avail_ < sizeof(T)) { |
| 70 // Need more room. We assume that MetaDataAlloc returns | 70 // Need more room. We assume that MetaDataAlloc returns |
| 71 // suitably aligned memory. | 71 // suitably aligned memory. |
| 72 free_area_ = reinterpret_cast<char*>(MetaDataAlloc(kAllocIncrement)); | 72 free_area_ = reinterpret_cast<char*>(MetaDataAlloc(kAllocIncrement)); |
| 73 if (free_area_ == NULL) { | 73 if (free_area_ == NULL) { |
| 74 CRASH("FATAL ERROR: Out of memory trying to allocate internal " | 74 Log(kCrash, __FILE__, __LINE__, |
| 75 "tcmalloc data (%d bytes, object-size %d)\n", | 75 "FATAL ERROR: Out of memory trying to allocate internal " |
| 76 kAllocIncrement, static_cast<int>(sizeof(T))); | 76 "tcmalloc data (bytes, object-size)", |
| 77 kAllocIncrement, sizeof(T)); |
| 77 } | 78 } |
| 78 | 79 |
| 79 // This guard page protects the metadata from being corrupted by a | 80 // This guard page protects the metadata from being corrupted by a |
| 80 // buffer overrun. We currently have no mechanism for freeing it, since | 81 // buffer overrun. We currently have no mechanism for freeing it, since |
| 81 // we never release the metadata buffer. If that changes we'll need to | 82 // we never release the metadata buffer. If that changes we'll need to |
| 82 // add something like TCMalloc_SystemRemoveGuard. | 83 // add something like TCMalloc_SystemRemoveGuard. |
| 83 size_t guard_size = TCMalloc_SystemAddGuard(free_area_, | 84 size_t guard_size = TCMalloc_SystemAddGuard(free_area_, |
| 84 kAllocIncrement); | 85 kAllocIncrement); |
| 85 free_area_ += guard_size; | 86 free_area_ += guard_size; |
| 86 free_avail_ = kAllocIncrement - guard_size; | 87 free_avail_ = kAllocIncrement - guard_size; |
| 87 if (free_avail_ < sizeof(T)) { | 88 if (free_avail_ < sizeof(T)) { |
| 88 CRASH("FATAL ERROR: Insufficient memory to guard internal tcmalloc " | 89 Log(kCrash, __FILE__, __LINE__, |
| 89 "data (%d bytes, object-size %d, guard-size %d)\n", | 90 "FATAL ERROR: Insufficient memory to guard internal tcmalloc " |
| 90 kAllocIncrement, static_cast<int>(sizeof(T)), guard_size); | 91 "data (%d bytes, object-size %d, guard-size %d)\n", |
| 92 kAllocIncrement, static_cast<int>(sizeof(T)), guard_size); |
| 91 } | 93 } |
| 92 } | 94 } |
| 93 result = free_area_; | 95 result = free_area_; |
| 94 free_area_ += sizeof(T); | 96 free_area_ += sizeof(T); |
| 95 free_avail_ -= sizeof(T); | 97 free_avail_ -= sizeof(T); |
| 96 } | 98 } |
| 97 inuse_++; | 99 inuse_++; |
| 98 return reinterpret_cast<T*>(result); | 100 return reinterpret_cast<T*>(result); |
| 99 } | 101 } |
| 100 | 102 |
| (...skipping 15 matching lines...) Expand all Loading... |
| 116 // Free list of already carved objects | 118 // Free list of already carved objects |
| 117 void* free_list_; | 119 void* free_list_; |
| 118 | 120 |
| 119 // Number of allocated but unfreed objects | 121 // Number of allocated but unfreed objects |
| 120 int inuse_; | 122 int inuse_; |
| 121 }; | 123 }; |
| 122 | 124 |
| 123 } // namespace tcmalloc | 125 } // namespace tcmalloc |
| 124 | 126 |
| 125 #endif // TCMALLOC_PAGE_HEAP_ALLOCATOR_H_ | 127 #endif // TCMALLOC_PAGE_HEAP_ALLOCATOR_H_ |
| OLD | NEW |