Change some assertions to exceptions

TigerVNC · Nov 22, 2024 · 213c12c · 213c12c
1 parent 895b751
commit 213c12c
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 5 deletions.
diff --git a/common/rdr/AESInStream.cxx b/common/rdr/AESInStream.cxx
@@ -22,6 +22,8 @@
 
 #include <assert.h>
 
+#include <stdexcept>
+
 #include <rdr/AESInStream.h>
 
 #ifdef HAVE_NETTLE
@@ -36,7 +38,7 @@ AESInStream::AESInStream(InStream* _in, const uint8_t* key,
   else if (keySize == 256)
     EAX_SET_KEY(&eaxCtx256, aes256_set_encrypt_key, aes256_encrypt, key);
   else
-    assert(!"incorrect key size");
+    throw std::out_of_range("Incorrect key size");
 }
 
 AESInStream::~AESInStream() {}

diff --git a/common/rdr/AESOutStream.cxx b/common/rdr/AESOutStream.cxx
@@ -21,7 +21,9 @@
 #endif
 
 #include <assert.h>
-#include <rdr/Exception.h>
+
+#include <stdexcept>
+
 #include <rdr/AESOutStream.h>
 
 #ifdef HAVE_NETTLE
@@ -39,7 +41,7 @@ AESOutStream::AESOutStream(OutStream* _out, const uint8_t* key,
   else if (keySize == 256)
     EAX_SET_KEY(&eaxCtx256, aes256_set_encrypt_key, aes256_encrypt, key);
   else
-    assert(!"incorrect key size");
+    throw std::out_of_range("Incorrect key size");
 }
 
 AESOutStream::~AESOutStream()

diff --git a/common/rfb/CSecurityRSAAES.cxx b/common/rfb/CSecurityRSAAES.cxx
@@ -126,7 +126,9 @@ bool CSecurityRSAAES::processMsg()
       writeCredentials();
       return true;
   }
-  assert(!"unreachable");
+
+  throw std::logic_error("Invalid state");
+
   return false;
 }
 

diff --git a/common/rfb/SSecurityRSAAES.cxx b/common/rfb/SSecurityRSAAES.cxx
@@ -278,7 +278,9 @@ bool SSecurityRSAAES::processMsg()
         verifyPass();
       return true;
   }
-  assert(!"unreachable");
+
+  throw std::logic_error("Invalid state");
+
   return false;
 }