diff options
Diffstat (limited to 'include')
| -rw-r--r-- | include/base.h | 59 | ||||
| -rw-r--r-- | include/extensible.h | 2 | ||||
| -rw-r--r-- | include/inspsocket.h | 4 | ||||
| -rw-r--r-- | include/modules.h | 4 |
4 files changed, 28 insertions, 41 deletions
diff --git a/include/base.h b/include/base.h index b39c7d077..bc096b85b 100644 --- a/include/base.h +++ b/include/base.h @@ -81,35 +81,46 @@ class CoreExport refcountbase refcountbase(); virtual ~refcountbase(); inline unsigned int GetReferenceCount() const { return refcount; } - friend class reference_base; void* operator new(size_t); void operator delete(void*); + inline void refcount_inc() const { refcount++; } + inline bool refcount_dec() const { refcount--; return !refcount; } private: // uncopyable refcountbase(const refcountbase&); void operator=(const refcountbase&); }; -class CoreExport reference_base +/** Base class for use count tracking. Uses reference<>, but does not + * cause object deletion when the last user is removed. + */ +class CoreExport usecountbase { - protected: - template<typename T> static inline unsigned int inc(T* v) { return ++(v->refcount); } - template<typename T> static inline unsigned int dec(T* v) { return --(v->refcount); } - + mutable unsigned int usecount; + public: + usecountbase() : usecount(0) { } + ~usecountbase(); + inline unsigned int GetUseCount() const { return usecount; } + inline void refcount_inc() const { usecount++; } + inline bool refcount_dec() const { usecount--; return false; } + private: + // uncopyable + usecountbase(const usecountbase&); + void operator=(const usecountbase&); }; template <typename T> -class reference : public reference_base +class reference { T* value; public: reference() : value(0) { } - reference(T* v) : value(v) { if (value) inc(value); } - reference(const reference<T>& v) : value(v.value) { if (value) inc(value); } + reference(T* v) : value(v) { if (value) value->refcount_inc(); } + reference(const reference<T>& v) : value(v.value) { if (value) value->refcount_inc(); } reference<T>& operator=(const reference<T>& other) { if (other.value) - inc(other.value); + other.value->refcount_inc(); this->reference::~reference(); value = other.value; return *this; @@ -117,12 +128,8 @@ class reference : public reference_base ~reference() { - if (value) - { - int rc = dec(value); - if (rc == 0) - delete value; - } + if (value && value->refcount_dec()) + delete value; } inline operator bool() const { return value; } inline operator T*() const { return value; } @@ -130,8 +137,6 @@ class reference : public reference_base inline T& operator*() const { return *value; } inline bool operator<(const reference<T>& other) const { return value < other.value; } inline bool operator>(const reference<T>& other) const { return value > other.value; } - inline bool operator==(const reference<T>& other) const { return value == other.value; } - inline bool operator!=(const reference<T>& other) const { return value != other.value; } private: void* operator new(size_t); void operator delete(void*); @@ -190,22 +195,6 @@ class CoreExport ModuleException : public CoreException ModuleException(const std::string &message, Module* me = NULL); }; -/** Module reference, similar to reference<Module> - */ -class CoreExport ModuleRef : public reference_base -{ - Module* const value; - public: - ModuleRef(Module* v); - ~ModuleRef(); - inline operator Module*() const { return value; } - inline Module* operator->() const { return value; } - inline Module& operator*() const { return *value; } - private: - ModuleRef(const ModuleRef&); - void operator=(const ModuleRef&); - void* operator new(size_t); - void operator delete(void*); -}; +typedef const reference<Module> ModuleRef; #endif diff --git a/include/extensible.h b/include/extensible.h index 81859b1c8..487b67408 100644 --- a/include/extensible.h +++ b/include/extensible.h @@ -12,7 +12,7 @@ enum SerializeFormat /** Class represnting an extension of some object */ -class CoreExport ExtensionItem : public refcountbase +class CoreExport ExtensionItem : public usecountbase { public: const std::string key; diff --git a/include/inspsocket.h b/include/inspsocket.h index 3e5c75235..8172338ac 100644 --- a/include/inspsocket.h +++ b/include/inspsocket.h @@ -94,7 +94,7 @@ class CoreExport SocketTimeout : public Timer class CoreExport StreamSocket : public EventHandler { /** Module that handles raw I/O for this socket, or NULL */ - Module *IOHook; + reference<Module> IOHook; /** Private send queue. Note that individual strings may be shared */ std::deque<std::string> sendq; @@ -105,7 +105,7 @@ class CoreExport StreamSocket : public EventHandler protected: std::string recvq; public: - StreamSocket() : IOHook(NULL), sendq_len(0) {} + StreamSocket() : sendq_len(0) {} inline Module* GetIOHook() { return IOHook; } inline void AddIOHook(Module* m) { IOHook = m; } inline void DelIOHook() { IOHook = NULL; } diff --git a/include/modules.h b/include/modules.h index 8bda9a725..26c4ef466 100644 --- a/include/modules.h +++ b/include/modules.h @@ -298,10 +298,8 @@ enum Implementation * its methods will be called when irc server events occur. class inherited from module must be * instantiated by the ModuleFactory class (see relevent section) for the module to be initialised. */ -class CoreExport Module : public classbase +class CoreExport Module : public classbase, public usecountbase { - unsigned int refcount; - friend class reference_base; public: /** File that this module was loaded from */ |