Add ResourceGuard

navicat-patcher/ResourceGuard.hpp

@@ -0,0 +1,112 @@
+#pragma once
+
+template<typename __ResourceTraits>
+class ResourceGuard {
+private:
+    using HandleType = typename __ResourceTraits::HandleType;
+    HandleType _Handle;
+    bool _DismissFlag;
+public:
+
+    ResourceGuard() noexcept :
+        _Handle(__ResourceTraits::InvalidValue),
+        _DismissFlag(false) {}
+
+    explicit ResourceGuard(HandleType Handle) :
+        _Handle(Handle),
+        _DismissFlag(false) {}
+
+    ResourceGuard(const ResourceGuard<__ResourceTraits>& Other) noexcept :
+        _Handle(Other._Handle),
+        _DismissFlag(false) {}
+
+    ResourceGuard(ResourceGuard<__ResourceTraits>&& Other) noexcept :
+        _Handle(Other._Handle),
+        _DismissFlag(false) { Other._Handle = __ResourceTraits::InvalidValue; }
+
+    ResourceGuard<__ResourceTraits>&
+    operator=(const ResourceGuard<__ResourceTraits>& Other) noexcept {
+        _Handle = Other._Handle;
+        return *this;
+    }
+
+    ResourceGuard<__ResourceTraits>&
+    operator=(ResourceGuard<__ResourceTraits>&& Other) noexcept {
+        _Handle = Other._Handle;
+        Other._Handle = __ResourceTraits::InvalidValue;
+        return *this;
+    }
+
+    operator HandleType() const noexcept {
+        return _Handle;
+    }
+
+    // Check if handle is a valid handle
+    bool IsValid() const noexcept {
+        return _Handle != __ResourceTraits::InvalidValue;
+    }
+
+    HandleType GetHandle() const noexcept {
+        return _Handle;
+    }
+
+    // Abandon possession of the previous handle without any conditions
+    // You must make sure that the previous handle has been released
+    //   OR has been possessed by others
+    void TakeHoldOf(HandleType Handle) noexcept {
+        _Handle = Handle;
+    }
+
+    // If dismiss, the handle won't be released when ResourceGuard is destructed.
+    void Dismiss() noexcept {
+        _DismissFlag = true;
+    }
+
+    // Cancel Dismiss() operation
+    void DismissCancel() noexcept {
+        _DismissFlag = false;
+    }
+
+    // Force release
+    void Release() {
+        if (_Handle != __ResourceTraits::InvalidValue) {
+            __ResourceTraits::Releasor(_Handle);
+            _Handle = __ResourceTraits::InvalidValue;
+        }
+    }
+
+    ~ResourceGuard() {
+        if (!_DismissFlag && _Handle != __ResourceTraits::InvalidValue) {
+            __ResourceTraits::Releasor(_Handle);
+            _Handle = __ResourceTraits::InvalidValue;
+        }
+    }
+};
+
+template<typename __ClassType>
+struct CppObjectTraits {
+    using HandleType = __ClassType*;
+    static const HandleType InvalidValue;
+    static inline void Releasor(HandleType pObject) {
+        delete pObject;
+    }
+};
+
+template<typename __ClassType>
+inline const typename CppObjectTraits<__ClassType>::HandleType
+    CppObjectTraits<__ClassType>::InvalidValue = nullptr;
+
+template<typename __ClassType>
+struct CppDynamicArrayTraits {
+    using HandleType = __ClassType*;
+    static const HandleType InvalidValue;
+    static inline void Releasor(HandleType pArray) {
+        delete[] pArray;
+    }
+};
+
+template<typename __ClassType>
+inline const typename CppDynamicArrayTraits<__ClassType>::HandleType
+    CppDynamicArrayTraits<__ClassType>::InvalidValue = nullptr;
+
+

navicat-patcher/ResourceGuardOpenssl.hpp

@@ -0,0 +1,41 @@
+#pragma once
+#include <openssl/bio.h>
+#include <openssl/bn.h>
+#include <openssl/rsa.h>
+
+struct OpensslBIOTraits {
+    using HandleType = BIO*;
+    static const HandleType InvalidValue;
+    static constexpr auto& Releasor = BIO_free;
+};
+
+inline const OpensslBIOTraits::HandleType
+    OpensslBIOTraits::InvalidValue = nullptr;
+
+struct OpensslBIOChainTraits {
+    using HandleType = BIO*;
+    static const HandleType InvalidValue;
+    static constexpr auto& Releasor = BIO_free_all;
+};
+
+inline const OpensslBIOChainTraits::HandleType
+    OpensslBIOChainTraits::InvalidValue = nullptr;
+
+struct OpensslBNTraits {
+    using HandleType = BIGNUM*;
+    static const HandleType InvalidValue;
+    static constexpr auto& Releasor = BN_free;
+};
+
+inline const OpensslBNTraits::HandleType
+    OpensslBNTraits::InvalidValue = nullptr;
+
+struct OpensslRSATraits {
+    using HandleType = RSA*;
+    static const HandleType InvalidValue;
+    static constexpr auto& Releasor = RSA_free;
+};
+
+inline const OpensslRSATraits::HandleType
+    OpensslRSATraits::InvalidValue = nullptr;
+

navicat-patcher/ResourceGuardWin32.hpp

@@ -0,0 +1,31 @@
+#pragma once
+#include "ResourceGuard.hpp"
+#include <windows.h>
+
+struct GenericHandleTraits {
+    using HandleType = HANDLE;
+    static const HandleType InvalidValue;
+    static constexpr auto& Releasor = CloseHandle;
+};
+
+inline const GenericHandleTraits::HandleType 
+    GenericHandleTraits::InvalidValue = NULL;
+
+struct FileHandleTraits {
+    using HandleType = HANDLE;
+    static const HandleType InvalidValue;
+    static constexpr auto& Releasor = CloseHandle;
+};
+
+inline const FileHandleTraits::HandleType
+    FileHandleTraits::InvalidValue = INVALID_HANDLE_VALUE;
+
+struct MapViewTraits {
+    using HandleType = PVOID;
+    static const HandleType InvalidValue;
+    static constexpr auto& Releasor = UnmapViewOfFile;
+};
+
+inline const MapViewTraits::HandleType
+    MapViewTraits::InvalidValue = NULL;
+