Skip to content
Snippets Groups Projects
Unverified Commit 2e89719d authored by bunnei's avatar bunnei Committed by GitHub
Browse files

Merge pull request #1048 from lioncash/atomic

kernel/object: Tighten object against data races
parents baaafbd5 3476f5b4
No related branches found
No related tags found
No related merge requests found
...@@ -11,7 +11,7 @@ ...@@ -11,7 +11,7 @@
namespace Kernel { namespace Kernel {
unsigned int Object::next_object_id; std::atomic<u32> Object::next_object_id{0};
/// Initialize the kernel /// Initialize the kernel
void Init() { void Init() {
......
...@@ -4,6 +4,7 @@ ...@@ -4,6 +4,7 @@
#pragma once #pragma once
#include <atomic>
#include <string> #include <string>
#include <utility> #include <utility>
...@@ -42,8 +43,8 @@ public: ...@@ -42,8 +43,8 @@ public:
virtual ~Object(); virtual ~Object();
/// Returns a unique identifier for the object. For debugging purposes only. /// Returns a unique identifier for the object. For debugging purposes only.
unsigned int GetObjectId() const { u32 GetObjectId() const {
return object_id; return object_id.load(std::memory_order_relaxed);
} }
virtual std::string GetTypeName() const { virtual std::string GetTypeName() const {
...@@ -61,23 +62,23 @@ public: ...@@ -61,23 +62,23 @@ public:
bool IsWaitable() const; bool IsWaitable() const;
public: public:
static unsigned int next_object_id; static std::atomic<u32> next_object_id;
private: private:
friend void intrusive_ptr_add_ref(Object*); friend void intrusive_ptr_add_ref(Object*);
friend void intrusive_ptr_release(Object*); friend void intrusive_ptr_release(Object*);
unsigned int ref_count = 0; std::atomic<u32> ref_count{0};
unsigned int object_id = next_object_id++; std::atomic<u32> object_id{next_object_id++};
}; };
// Special functions used by boost::instrusive_ptr to do automatic ref-counting // Special functions used by boost::instrusive_ptr to do automatic ref-counting
inline void intrusive_ptr_add_ref(Object* object) { inline void intrusive_ptr_add_ref(Object* object) {
++object->ref_count; object->ref_count.fetch_add(1, std::memory_order_relaxed);
} }
inline void intrusive_ptr_release(Object* object) { inline void intrusive_ptr_release(Object* object) {
if (--object->ref_count == 0) { if (object->ref_count.fetch_sub(1, std::memory_order_acq_rel) == 1) {
delete object; delete object;
} }
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment