606368f6f6
Track the cpu on which the reference count referred to by shared_ptr was created, and prevent modifying it on the wrong cpu.
67 lines
1.7 KiB
C++
67 lines
1.7 KiB
C++
/*
|
|
* This file is open source software, licensed to you under the terms
|
|
* of the Apache License, Version 2.0 (the "License"). See the NOTICE file
|
|
* distributed with this work for additional information regarding copyright
|
|
* ownership. You may not use this file except in compliance with the License.
|
|
*
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing,
|
|
* software distributed under the License is distributed on an
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
* KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations
|
|
* under the License.
|
|
*/
|
|
/*
|
|
* Copyright (C) 2015 Cloudius Systems, Ltd.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#ifdef DEBUG_SHARED_PTR
|
|
|
|
#include <thread>
|
|
#include <cassert>
|
|
|
|
// A counter that is only comfortable being incremented on the cpu
|
|
// it was created on. Useful for verifying that a shared_ptr
|
|
// or lw_shared_ptr isn't misued across cores.
|
|
class debug_shared_ptr_counter_type {
|
|
long _counter = 0;
|
|
std::thread::id _cpu = std::this_thread::get_id();
|
|
public:
|
|
debug_shared_ptr_counter_type(long x) : _counter(x) {}
|
|
operator long() const {
|
|
check();
|
|
return _counter;
|
|
}
|
|
debug_shared_ptr_counter_type& operator++() {
|
|
check();
|
|
++_counter;
|
|
return *this;
|
|
}
|
|
long operator++(int) {
|
|
check();
|
|
return _counter++;
|
|
}
|
|
debug_shared_ptr_counter_type& operator--() {
|
|
check();
|
|
--_counter;
|
|
return *this;
|
|
}
|
|
long operator--(int) {
|
|
check();
|
|
return _counter--;
|
|
}
|
|
private:
|
|
void check() const {
|
|
assert(_cpu == std::this_thread::get_id());
|
|
}
|
|
};
|
|
|
|
#endif
|
|
|