Migrated the host module to C++ exceptions as well.

Also enabled the latest C++ revision in the project. Replaced the Random library with the one provided by C++11. Implemented a simple AES256 encryption class. Various other fixes and improvements.
2025-08-06 08:01:47 +02:00 · 2016-03-10 05:57:13 +02:00
parent 3162221e7f
commit 70e5f0ba21
124 changed files with 14873 additions and 14062 deletions
--- a/modules/mmdb/Database.cpp
+++ b/modules/mmdb/Database.cpp
@@ -0,0 +1,116 @@
+// ------------------------------------------------------------------------------------------------
+#include "Database.hpp"
+#include "SockAddr.hpp"
+#include "LookupResult.hpp"
+#include "Module.hpp"
+
+// ------------------------------------------------------------------------------------------------
+#include <sqrat.h>
+
+// ------------------------------------------------------------------------------------------------
+namespace SqMod {
+
+// ------------------------------------------------------------------------------------------------
+SQInteger Database::Typename(HSQUIRRELVM vm)
+{
+    static const SQChar name[] = _SC("SqMMDBDatabase");
+    sq_pushstring(vm, name, sizeof(name));
+    return 1;
+}
+
+// ------------------------------------------------------------------------------------------------
+void Database::Validate() const
+{
+    // Is the document handle valid?
+    if (!m_Db)
+        SqThrowF("Invalid Maxmind database reference");
+}
+
+// ------------------------------------------------------------------------------------------------
+Int32 Database::Cmp(const Database & o) const
+{
+    if (m_Db == o.m_Db)
+        return 0;
+    else if (m_Db.m_Ptr > o.m_Db.m_Ptr)
+        return 1;
+    else
+        return -1;
+}
+
+// ------------------------------------------------------------------------------------------------
+void Database::Open(CSStr filepath)
+{
+    Open(filepath, 0);
+}
+
+// ------------------------------------------------------------------------------------------------
+void Database::Open(CSStr filepath, Uint32 flags)
+{
+    // Is there a database handle available?
+    if (!m_Db)
+        m_Db = DbRef(true); // Create a database handle
+    // Check if the database handle could be allocated one more time
+    if (!m_Db)
+        SqThrowF("Unable to create a Maxmind database reference");
+    // Are there any other references?
+    else if (m_Db.Count() > 1)
+        // To load new values now, would mean to cause undefined behavior in existing references
+        SqThrowF("Loading is disabled while database is referenced");
+    // Validate the specified file path
+    else if (!filepath || strlen(filepath) <= 0)
+        SqThrowF("Invalid database file path");
+    // Let's attempt to open the specified database
+    const Int32 status = MMDB_open(filepath, flags, m_Db.m_Ptr);
+    // Validate the result of the operation
+    if (status != MMDB_SUCCESS)
+    {
+        // Release the database reference
+        m_Db.Drop();
+        // Now it's safe to throw the error
+        SqThrowF("Unable to open the specified database [%s]", MMDB_strerror(status));
+    }
+}
+
+// ------------------------------------------------------------------------------------------------
+LookupResult Database::LookupString(CSStr addr)
+{
+    // Validate the database handle
+    Validate();
+    // Validate the specified string
+    if (!addr || strlen(addr) <= 0)
+        SqThrowF("Invalid address string");
+    // Dummy variables to obtain the status codes
+    int gai_error, mmdb_error;
+    // Attempt to perform the actual lookup
+    MMDB_lookup_result_s result = MMDB_lookup_string(m_Db, addr, &gai_error, &mmdb_error);
+    // Validate the result of the getaddrinfo() function call
+    if (gai_error != 0)
+        SqThrowF("Unable to resolve address (%s) because [%s]", addr, gai_strerror(gai_error));
+    // Validate the lookup status code
+    else if (mmdb_error != MMDB_SUCCESS)
+        SqThrowF("Unable to lookup address (%s) because [%s]", addr, MMDB_strerror(mmdb_error));
+    // Now it's safe to return the lookup result
+    return LookupResult(m_Db, result);
+}
+
+// ------------------------------------------------------------------------------------------------
+LookupResult Database::LookupSockAddr(SockAddr & addr)
+{
+    // Validate the database handle
+    Validate();
+    // Validate the specified socket address
+    if (!addr.IsValid())
+        SqThrowF("Invalid address instance");
+    // Dummy variable to obtain the status codes
+    int mmdb_error;
+    // Attempt to perform the actual lookup
+    MMDB_lookup_result_s result = MMDB_lookup_sockaddr(m_Db, addr.GetHandle()->ai_addr, &mmdb_error);
+    // Validate the lookup status code
+    if (mmdb_error != MMDB_SUCCESS)
+        SqThrowF("Unable to lookup address (%s) because [%s]",
+                    addr.GetAddress(), MMDB_strerror(mmdb_error));
+    // Now it's safe to return the lookup result
+    return LookupResult(m_Db, result);
+}
+
+} // Namespace:: SqMod