Added tests for RSA functions

Author: ccw808

MTA10_Server/core/Server.cpp

@@ -14,9 +14,6 @@
 #include "CServerImpl.h"
 #define ALLOC_STATS_MODULE_NAME "core"
 #include "SharedUtil.hpp"
-#if defined(_DEBUG) 
-    #include "SharedUtil.Tests.hpp"
-#endif
 #ifdef WIN_x86
     // TODO - 64 bit file hooks
     #include "SharedUtil.Win32Utf8FileHooks.hpp"
@@ -37,10 +34,6 @@ MTAEXPORT int Run ( int iArgumentCount, char* szArguments [] )
         }
     }
 
-    #if defined(_DEBUG) 
-        SharedUtil_Tests ();
-    #endif
-
     #ifdef WIN32
         // Disable critical error message boxes
         SetErrorMode ( SEM_FAILCRITICALERRORS );

MTA10_Server/mods/deathmatch/CServer.cpp

@@ -16,6 +16,9 @@
 #include "SharedUtil.Thread.h"
 #include "SharedUtil.IntervalCounter.h"
 #include "SharedUtil.IntervalCounter.hpp"
+#if defined(_DEBUG) 
+    #include "SharedUtil.Tests.hpp"
+#endif
 
 CServerInterface* g_pServerInterface = NULL;
 CNetServer* g_pNetServer = NULL;
@@ -41,6 +44,9 @@ void CServer::ServerInitialize ( CServerInterface* pServerInterface )
     g_pServerInterface = pServerInterface;
     g_pNetServer = pServerInterface->GetNetwork ();
     g_pRealNetServer = g_pNetServer;
+    #if defined(_DEBUG) 
+        SharedUtil_Tests();
+    #endif
 }
 
 

MTA10_Server/mods/deathmatch/StdInc.h

@@ -297,3 +297,6 @@ struct SAclRequest;
 #include "Config.h"
 #define SHOW_SELF_COMPILE_WARNING
 #include "../../version.h"
+
+extern CNetServer* g_pRealNetServer;
+extern CGame* g_pGame;

Shared/sdk/SharedUtil.Tests.hpp

@@ -21,6 +21,9 @@ void 	SharedUtil_WildcardMatch_Tests  ( void );
 void    SharedUtil_Collection_Tests     ( void );
 void    SharedUtil_String_Tests         ( void );
 void    SharedUtil_Hash_Tests           ( void );
+void    SharedUtil_RSA_Tests            ( void );
+void    RSATestCrypt                    ( const SString& input, const SString& keyEncrypt, const SString& keyDecrypt );
+
 
 ///////////////////////////////////////////////////////////////
 //
@@ -38,6 +41,9 @@ void SharedUtil_Tests ( void )
     SharedUtil_Collection_Tests ();
     SharedUtil_String_Tests ();
     SharedUtil_Hash_Tests ();
+#ifndef MTA_CLIENT
+    SharedUtil_RSA_Tests ();
+#endif
 }
 
 
@@ -899,3 +905,51 @@ void SharedUtil_Hash_Tests ( void )
 
     FileDelete( szTempFilename );
 }
+
+
+#ifndef MTA_CLIENT
+///////////////////////////////////////////////////////////////
+//
+// SharedUtil_RSA_Tests
+//
+// Test behaviour of Raknet RSA code
+//
+///////////////////////////////////////////////////////////////
+void SharedUtil_RSA_Tests( void )
+{
+    // Generate keys
+    char privateKeyTemp[64];
+    char publicKeyTemp[68];
+    g_pRealNetServer->RSAGenerateKeys( privateKeyTemp, sizeof( privateKeyTemp ), publicKeyTemp, sizeof( publicKeyTemp ) );
+    SStringX privateKey( privateKeyTemp, sizeof( privateKeyTemp ) );
+    SStringX publicKey( publicKeyTemp, sizeof( publicKeyTemp ) );
+
+    // Short data length
+    SString input = "hello";
+    RSATestCrypt( input, publicKey, privateKey );
+    RSATestCrypt( input, privateKey, publicKey );
+
+    // Long data length
+    while( input.size() < 3000 )
+        input += input + "goodbye";
+    RSATestCrypt( input, publicKey, privateKey );
+    RSATestCrypt( input, privateKey, publicKey );
+}
+
+void RSATestCrypt( const SString& input, const SString& keyEncrypt, const SString& keyDecrypt )
+{
+    // Encrypt
+    std::vector < char > encryptedBuffer;
+    encryptedBuffer.resize( input.size() + 100 );
+    uint uiEncryptedSize = g_pRealNetServer->RSAEncryptData( &input[0], input.size(), &keyEncrypt[0], keyEncrypt.size(), &encryptedBuffer[0], encryptedBuffer.size() );
+
+    // Decrypt
+    std::vector < char > decryptedBuffer;
+    decryptedBuffer.resize( uiEncryptedSize );
+    uint uiDecryptedSize = g_pRealNetServer->RSADecryptData( &encryptedBuffer[0], uiEncryptedSize, &keyDecrypt[0], keyDecrypt.size(), &decryptedBuffer[0], decryptedBuffer.size() );
+
+    // Check decrypted is the same as original
+    assert( input.size() == uiDecryptedSize );
+    assert( memcmp( &input[0], &decryptedBuffer[0], uiDecryptedSize ) == 0 );
+}
+#endif