test

Dependencies:   mbed Watchdog

Dependents:   STM32-MC_node

Committer:
ommpy
Date:
Mon Jul 06 17:18:59 2020 +0530
Revision:
0:d383e2dee0f7
first commit

Who changed what in which revision?

UserRevisionLine numberNew contents of line
ommpy 0:d383e2dee0f7 1 /* mbed Microcontroller Library
ommpy 0:d383e2dee0f7 2 * Copyright (c) 2006-2018 ARM Limited
ommpy 0:d383e2dee0f7 3 * SPDX-License-Identifier: Apache-2.0
ommpy 0:d383e2dee0f7 4 *
ommpy 0:d383e2dee0f7 5 * Licensed under the Apache License, Version 2.0 (the "License");
ommpy 0:d383e2dee0f7 6 * you may not use this file except in compliance with the License.
ommpy 0:d383e2dee0f7 7 * You may obtain a copy of the License at
ommpy 0:d383e2dee0f7 8 *
ommpy 0:d383e2dee0f7 9 * http://www.apache.org/licenses/LICENSE-2.0
ommpy 0:d383e2dee0f7 10 *
ommpy 0:d383e2dee0f7 11 * Unless required by applicable law or agreed to in writing, software
ommpy 0:d383e2dee0f7 12 * distributed under the License is distributed on an "AS IS" BASIS,
ommpy 0:d383e2dee0f7 13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
ommpy 0:d383e2dee0f7 14 * See the License for the specific language governing permissions and
ommpy 0:d383e2dee0f7 15 * limitations under the License.
ommpy 0:d383e2dee0f7 16 */
ommpy 0:d383e2dee0f7 17
ommpy 0:d383e2dee0f7 18 #ifndef __SHAREDPTR_H__
ommpy 0:d383e2dee0f7 19 #define __SHAREDPTR_H__
ommpy 0:d383e2dee0f7 20
ommpy 0:d383e2dee0f7 21 #include <stdlib.h>
ommpy 0:d383e2dee0f7 22
ommpy 0:d383e2dee0f7 23 #include <stdint.h>
ommpy 0:d383e2dee0f7 24 #include <stddef.h>
ommpy 0:d383e2dee0f7 25
ommpy 0:d383e2dee0f7 26 #include "platform/mbed_critical.h"
ommpy 0:d383e2dee0f7 27
ommpy 0:d383e2dee0f7 28 namespace mbed {
ommpy 0:d383e2dee0f7 29
ommpy 0:d383e2dee0f7 30 /** Shared pointer class.
ommpy 0:d383e2dee0f7 31 *
ommpy 0:d383e2dee0f7 32 * A shared pointer is a "smart" pointer that retains ownership of an object using
ommpy 0:d383e2dee0f7 33 * reference counting across all smart pointers referencing that object.
ommpy 0:d383e2dee0f7 34 *
ommpy 0:d383e2dee0f7 35 * @code
ommpy 0:d383e2dee0f7 36 * #include "platform/SharedPtr.h"
ommpy 0:d383e2dee0f7 37 *
ommpy 0:d383e2dee0f7 38 * void test() {
ommpy 0:d383e2dee0f7 39 * struct MyStruct { int a; };
ommpy 0:d383e2dee0f7 40 *
ommpy 0:d383e2dee0f7 41 * // Create shared pointer
ommpy 0:d383e2dee0f7 42 * SharedPtr<MyStruct> ptr( new MyStruct );
ommpy 0:d383e2dee0f7 43 *
ommpy 0:d383e2dee0f7 44 * // Increase reference count
ommpy 0:d383e2dee0f7 45 * SharedPtr<MyStruct> ptr2( ptr );
ommpy 0:d383e2dee0f7 46 *
ommpy 0:d383e2dee0f7 47 * ptr = NULL; // Reference to the struct instance is still held by ptr2
ommpy 0:d383e2dee0f7 48 *
ommpy 0:d383e2dee0f7 49 * ptr2 = NULL; // The raw pointer is freed
ommpy 0:d383e2dee0f7 50 * }
ommpy 0:d383e2dee0f7 51 * @endcode
ommpy 0:d383e2dee0f7 52 *
ommpy 0:d383e2dee0f7 53 *
ommpy 0:d383e2dee0f7 54 * It is similar to the std::shared_ptr class introduced in C++11;
ommpy 0:d383e2dee0f7 55 * however, this is not a compatible implementation (no weak pointer, no make_shared, no custom deleters and so on.)
ommpy 0:d383e2dee0f7 56 *
ommpy 0:d383e2dee0f7 57 * Usage: SharedPtr<Class> ptr(new Class())
ommpy 0:d383e2dee0f7 58 *
ommpy 0:d383e2dee0f7 59 * When ptr is passed around by value, the copy constructor and
ommpy 0:d383e2dee0f7 60 * destructor manages the reference count of the raw pointer.
ommpy 0:d383e2dee0f7 61 * If the counter reaches zero, delete is called on the raw pointer.
ommpy 0:d383e2dee0f7 62 *
ommpy 0:d383e2dee0f7 63 * To avoid loops, use "weak" references by calling the original
ommpy 0:d383e2dee0f7 64 * pointer directly through ptr.get().
ommpy 0:d383e2dee0f7 65 */
ommpy 0:d383e2dee0f7 66
ommpy 0:d383e2dee0f7 67 template <class T>
ommpy 0:d383e2dee0f7 68 class SharedPtr {
ommpy 0:d383e2dee0f7 69 public:
ommpy 0:d383e2dee0f7 70 /**
ommpy 0:d383e2dee0f7 71 * @brief Create empty SharedPtr not pointing to anything.
ommpy 0:d383e2dee0f7 72 * @details Used for variable declaration.
ommpy 0:d383e2dee0f7 73 */
ommpy 0:d383e2dee0f7 74 SharedPtr(): _ptr(NULL), _counter(NULL)
ommpy 0:d383e2dee0f7 75 {
ommpy 0:d383e2dee0f7 76 }
ommpy 0:d383e2dee0f7 77
ommpy 0:d383e2dee0f7 78 /**
ommpy 0:d383e2dee0f7 79 * @brief Create new SharedPtr
ommpy 0:d383e2dee0f7 80 * @param ptr Pointer to take control over
ommpy 0:d383e2dee0f7 81 */
ommpy 0:d383e2dee0f7 82 SharedPtr(T *ptr): _ptr(ptr), _counter(NULL)
ommpy 0:d383e2dee0f7 83 {
ommpy 0:d383e2dee0f7 84 // Allocate counter on the heap, so it can be shared
ommpy 0:d383e2dee0f7 85 if (_ptr != NULL) {
ommpy 0:d383e2dee0f7 86 _counter = new uint32_t;
ommpy 0:d383e2dee0f7 87 *_counter = 1;
ommpy 0:d383e2dee0f7 88 }
ommpy 0:d383e2dee0f7 89 }
ommpy 0:d383e2dee0f7 90
ommpy 0:d383e2dee0f7 91 /**
ommpy 0:d383e2dee0f7 92 * @brief Destructor.
ommpy 0:d383e2dee0f7 93 * @details Decrement reference counter, and delete object if no longer pointed to.
ommpy 0:d383e2dee0f7 94 */
ommpy 0:d383e2dee0f7 95 ~SharedPtr()
ommpy 0:d383e2dee0f7 96 {
ommpy 0:d383e2dee0f7 97 decrement_counter();
ommpy 0:d383e2dee0f7 98 }
ommpy 0:d383e2dee0f7 99
ommpy 0:d383e2dee0f7 100 /**
ommpy 0:d383e2dee0f7 101 * @brief Copy constructor.
ommpy 0:d383e2dee0f7 102 * @details Create new SharedPtr from other SharedPtr by
ommpy 0:d383e2dee0f7 103 * copying pointer to original object and pointer to counter.
ommpy 0:d383e2dee0f7 104 * @param source Object being copied from.
ommpy 0:d383e2dee0f7 105 */
ommpy 0:d383e2dee0f7 106 SharedPtr(const SharedPtr &source): _ptr(source._ptr), _counter(source._counter)
ommpy 0:d383e2dee0f7 107 {
ommpy 0:d383e2dee0f7 108 // Increment reference counter
ommpy 0:d383e2dee0f7 109 if (_ptr != NULL) {
ommpy 0:d383e2dee0f7 110 core_util_atomic_incr_u32(_counter, 1);
ommpy 0:d383e2dee0f7 111 }
ommpy 0:d383e2dee0f7 112 }
ommpy 0:d383e2dee0f7 113
ommpy 0:d383e2dee0f7 114 /**
ommpy 0:d383e2dee0f7 115 * @brief Assignment operator.
ommpy 0:d383e2dee0f7 116 * @details Cleanup previous reference and assign new pointer and counter.
ommpy 0:d383e2dee0f7 117 * @param source Object being assigned from.
ommpy 0:d383e2dee0f7 118 * @return Object being assigned.
ommpy 0:d383e2dee0f7 119 */
ommpy 0:d383e2dee0f7 120 SharedPtr operator=(const SharedPtr &source)
ommpy 0:d383e2dee0f7 121 {
ommpy 0:d383e2dee0f7 122 if (this != &source) {
ommpy 0:d383e2dee0f7 123 // Clean up by decrementing counter
ommpy 0:d383e2dee0f7 124 decrement_counter();
ommpy 0:d383e2dee0f7 125
ommpy 0:d383e2dee0f7 126 // Assign new values
ommpy 0:d383e2dee0f7 127 _ptr = source.get();
ommpy 0:d383e2dee0f7 128 _counter = source.get_counter();
ommpy 0:d383e2dee0f7 129
ommpy 0:d383e2dee0f7 130 // Increment new counter
ommpy 0:d383e2dee0f7 131 if (_ptr != NULL) {
ommpy 0:d383e2dee0f7 132 core_util_atomic_incr_u32(_counter, 1);
ommpy 0:d383e2dee0f7 133 }
ommpy 0:d383e2dee0f7 134 }
ommpy 0:d383e2dee0f7 135
ommpy 0:d383e2dee0f7 136 return *this;
ommpy 0:d383e2dee0f7 137 }
ommpy 0:d383e2dee0f7 138
ommpy 0:d383e2dee0f7 139 /**
ommpy 0:d383e2dee0f7 140 * @brief Replaces the managed pointer with a new unmanaged pointer.
ommpy 0:d383e2dee0f7 141 * @param[in] ptr the new raw pointer to manage.
ommpy 0:d383e2dee0f7 142 */
ommpy 0:d383e2dee0f7 143 void reset(T *ptr)
ommpy 0:d383e2dee0f7 144 {
ommpy 0:d383e2dee0f7 145 // Clean up by decrementing counter
ommpy 0:d383e2dee0f7 146 decrement_counter();
ommpy 0:d383e2dee0f7 147
ommpy 0:d383e2dee0f7 148 _ptr = ptr;
ommpy 0:d383e2dee0f7 149 if (ptr != NULL) {
ommpy 0:d383e2dee0f7 150 // Allocate counter on the heap, so it can be shared
ommpy 0:d383e2dee0f7 151 _counter = new uint32_t;
ommpy 0:d383e2dee0f7 152 *_counter = 1;
ommpy 0:d383e2dee0f7 153 } else {
ommpy 0:d383e2dee0f7 154 _counter = NULL;
ommpy 0:d383e2dee0f7 155 }
ommpy 0:d383e2dee0f7 156 }
ommpy 0:d383e2dee0f7 157
ommpy 0:d383e2dee0f7 158 /**
ommpy 0:d383e2dee0f7 159 * @brief Replace the managed pointer with a NULL pointer.
ommpy 0:d383e2dee0f7 160 */
ommpy 0:d383e2dee0f7 161 void reset()
ommpy 0:d383e2dee0f7 162 {
ommpy 0:d383e2dee0f7 163 reset(NULL);
ommpy 0:d383e2dee0f7 164 }
ommpy 0:d383e2dee0f7 165
ommpy 0:d383e2dee0f7 166 /**
ommpy 0:d383e2dee0f7 167 * @brief Raw pointer accessor.
ommpy 0:d383e2dee0f7 168 * @details Get raw pointer to object pointed to.
ommpy 0:d383e2dee0f7 169 * @return Pointer.
ommpy 0:d383e2dee0f7 170 */
ommpy 0:d383e2dee0f7 171 T *get() const
ommpy 0:d383e2dee0f7 172 {
ommpy 0:d383e2dee0f7 173 return _ptr;
ommpy 0:d383e2dee0f7 174 }
ommpy 0:d383e2dee0f7 175
ommpy 0:d383e2dee0f7 176 /**
ommpy 0:d383e2dee0f7 177 * @brief Reference count accessor.
ommpy 0:d383e2dee0f7 178 * @return Reference count.
ommpy 0:d383e2dee0f7 179 */
ommpy 0:d383e2dee0f7 180 uint32_t use_count() const
ommpy 0:d383e2dee0f7 181 {
ommpy 0:d383e2dee0f7 182 if (_ptr != NULL) {
ommpy 0:d383e2dee0f7 183 core_util_critical_section_enter();
ommpy 0:d383e2dee0f7 184 uint32_t current_counter = *_counter;
ommpy 0:d383e2dee0f7 185 core_util_critical_section_exit();
ommpy 0:d383e2dee0f7 186 return current_counter;
ommpy 0:d383e2dee0f7 187 } else {
ommpy 0:d383e2dee0f7 188 return 0;
ommpy 0:d383e2dee0f7 189 }
ommpy 0:d383e2dee0f7 190 }
ommpy 0:d383e2dee0f7 191
ommpy 0:d383e2dee0f7 192 /**
ommpy 0:d383e2dee0f7 193 * @brief Dereference object operator.
ommpy 0:d383e2dee0f7 194 * @details Override to return the object pointed to.
ommpy 0:d383e2dee0f7 195 */
ommpy 0:d383e2dee0f7 196 T &operator*() const
ommpy 0:d383e2dee0f7 197 {
ommpy 0:d383e2dee0f7 198 return *_ptr;
ommpy 0:d383e2dee0f7 199 }
ommpy 0:d383e2dee0f7 200
ommpy 0:d383e2dee0f7 201 /**
ommpy 0:d383e2dee0f7 202 * @brief Dereference object member operator.
ommpy 0:d383e2dee0f7 203 * @details Override to return return member in object pointed to.
ommpy 0:d383e2dee0f7 204 */
ommpy 0:d383e2dee0f7 205 T *operator->() const
ommpy 0:d383e2dee0f7 206 {
ommpy 0:d383e2dee0f7 207 return _ptr;
ommpy 0:d383e2dee0f7 208 }
ommpy 0:d383e2dee0f7 209
ommpy 0:d383e2dee0f7 210 /**
ommpy 0:d383e2dee0f7 211 * @brief Boolean conversion operator.
ommpy 0:d383e2dee0f7 212 * @return Whether or not the pointer is NULL.
ommpy 0:d383e2dee0f7 213 */
ommpy 0:d383e2dee0f7 214 operator bool() const
ommpy 0:d383e2dee0f7 215 {
ommpy 0:d383e2dee0f7 216 return (_ptr != NULL);
ommpy 0:d383e2dee0f7 217 }
ommpy 0:d383e2dee0f7 218
ommpy 0:d383e2dee0f7 219 private:
ommpy 0:d383e2dee0f7 220 /**
ommpy 0:d383e2dee0f7 221 * @brief Get pointer to reference counter.
ommpy 0:d383e2dee0f7 222 * @return Pointer to reference counter.
ommpy 0:d383e2dee0f7 223 */
ommpy 0:d383e2dee0f7 224 uint32_t *get_counter() const
ommpy 0:d383e2dee0f7 225 {
ommpy 0:d383e2dee0f7 226 return _counter;
ommpy 0:d383e2dee0f7 227 }
ommpy 0:d383e2dee0f7 228
ommpy 0:d383e2dee0f7 229 /**
ommpy 0:d383e2dee0f7 230 * @brief Decrement reference counter.
ommpy 0:d383e2dee0f7 231 * @details If count reaches zero, free counter and delete object pointed to.
ommpy 0:d383e2dee0f7 232 * Does not modify our own pointers - assumption is they will be overwritten
ommpy 0:d383e2dee0f7 233 * or destroyed immediately afterwards.
ommpy 0:d383e2dee0f7 234 */
ommpy 0:d383e2dee0f7 235 void decrement_counter()
ommpy 0:d383e2dee0f7 236 {
ommpy 0:d383e2dee0f7 237 if (_ptr != NULL) {
ommpy 0:d383e2dee0f7 238 uint32_t new_value = core_util_atomic_decr_u32(_counter, 1);
ommpy 0:d383e2dee0f7 239 if (new_value == 0) {
ommpy 0:d383e2dee0f7 240 delete _counter;
ommpy 0:d383e2dee0f7 241 delete _ptr;
ommpy 0:d383e2dee0f7 242 }
ommpy 0:d383e2dee0f7 243 }
ommpy 0:d383e2dee0f7 244 }
ommpy 0:d383e2dee0f7 245
ommpy 0:d383e2dee0f7 246 private:
ommpy 0:d383e2dee0f7 247 // Pointer to shared object
ommpy 0:d383e2dee0f7 248 T *_ptr;
ommpy 0:d383e2dee0f7 249
ommpy 0:d383e2dee0f7 250 // Pointer to shared reference counter
ommpy 0:d383e2dee0f7 251 uint32_t *_counter;
ommpy 0:d383e2dee0f7 252 };
ommpy 0:d383e2dee0f7 253
ommpy 0:d383e2dee0f7 254 /** Non-member relational operators.
ommpy 0:d383e2dee0f7 255 */
ommpy 0:d383e2dee0f7 256 template <class T, class U>
ommpy 0:d383e2dee0f7 257 bool operator== (const SharedPtr<T> &lhs, const SharedPtr<U> &rhs)
ommpy 0:d383e2dee0f7 258 {
ommpy 0:d383e2dee0f7 259 return (lhs.get() == rhs.get());
ommpy 0:d383e2dee0f7 260 }
ommpy 0:d383e2dee0f7 261
ommpy 0:d383e2dee0f7 262 template <class T, typename U>
ommpy 0:d383e2dee0f7 263 bool operator== (const SharedPtr<T> &lhs, U rhs)
ommpy 0:d383e2dee0f7 264 {
ommpy 0:d383e2dee0f7 265 return (lhs.get() == (T *) rhs);
ommpy 0:d383e2dee0f7 266 }
ommpy 0:d383e2dee0f7 267
ommpy 0:d383e2dee0f7 268 template <class T, typename U>
ommpy 0:d383e2dee0f7 269 bool operator== (U lhs, const SharedPtr<T> &rhs)
ommpy 0:d383e2dee0f7 270 {
ommpy 0:d383e2dee0f7 271 return ((T *) lhs == rhs.get());
ommpy 0:d383e2dee0f7 272 }
ommpy 0:d383e2dee0f7 273
ommpy 0:d383e2dee0f7 274 /** Non-member relational operators.
ommpy 0:d383e2dee0f7 275 */
ommpy 0:d383e2dee0f7 276 template <class T, class U>
ommpy 0:d383e2dee0f7 277 bool operator!= (const SharedPtr<T> &lhs, const SharedPtr<U> &rhs)
ommpy 0:d383e2dee0f7 278 {
ommpy 0:d383e2dee0f7 279 return (lhs.get() != rhs.get());
ommpy 0:d383e2dee0f7 280 }
ommpy 0:d383e2dee0f7 281
ommpy 0:d383e2dee0f7 282 template <class T, typename U>
ommpy 0:d383e2dee0f7 283 bool operator!= (const SharedPtr<T> &lhs, U rhs)
ommpy 0:d383e2dee0f7 284 {
ommpy 0:d383e2dee0f7 285 return (lhs.get() != (T *) rhs);
ommpy 0:d383e2dee0f7 286 }
ommpy 0:d383e2dee0f7 287
ommpy 0:d383e2dee0f7 288 template <class T, typename U>
ommpy 0:d383e2dee0f7 289 bool operator!= (U lhs, const SharedPtr<T> &rhs)
ommpy 0:d383e2dee0f7 290 {
ommpy 0:d383e2dee0f7 291 return ((T *) lhs != rhs.get());
ommpy 0:d383e2dee0f7 292 }
ommpy 0:d383e2dee0f7 293
ommpy 0:d383e2dee0f7 294 } /* namespace mbed */
ommpy 0:d383e2dee0f7 295
ommpy 0:d383e2dee0f7 296 #ifndef MBED_NO_GLOBAL_USING_DIRECTIVE
ommpy 0:d383e2dee0f7 297 using mbed::SharedPtr;
ommpy 0:d383e2dee0f7 298 #endif
ommpy 0:d383e2dee0f7 299
ommpy 0:d383e2dee0f7 300 #endif // __SHAREDPTR_H__