feat: add client mismatch exit

skifli · Sep 30, 2024 · 7b21224 · 7b21224
1 parent 773a749
commit 7b21224
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 4 deletions.
diff --git a/bruty_client/Cargo.toml b/bruty_client/Cargo.toml
@@ -1,7 +1,7 @@
 [package]
 name = "bruty_client"
 authors = ["skifli"]
-version = "0.2.2"
+version = "0.2.3"
 edition = "2021"
 
 [dependencies]

diff --git a/bruty_client/src/main.rs b/bruty_client/src/main.rs
@@ -119,6 +119,12 @@ async fn handle_msg(
                 error_code.explanation
             );
 
+            if error_code.code == bruty_share::ErrorCode::UnsupportedClientVersion {
+                log::error!("Unsupported client version, please update");
+
+                std::process::exit(1);
+            }
+
             return false;
         }
         bruty_share::OperationCode::TestRequestData => {

diff --git a/bruty_server/src/payload_handlers.rs b/bruty_server/src/payload_handlers.rs
@@ -1,7 +1,7 @@
 use crate::{SplitSinkExt, WebSocketSender};
 use futures_util::SinkExt;
 
-const ALLOWED_CLIENT_VERSIONS: &[&str] = &["0.2.2"];
+const ALLOWED_CLIENT_VERSIONS: &[&str] = &["0.2.3"];
 
 /// Checks if the connection is authenticated.
 /// If not, it sends an InvalidSession OP code and closes the connection.

diff --git a/bruty_share/src/lib.rs b/bruty_share/src/lib.rs
@@ -11,7 +11,7 @@ pub const VALID_CHARS: &[char] = &[
 ];
 
 /// Error codes sent with an InvalidSession OP code
-#[derive(serde::Serialize, serde::Deserialize, Debug)]
+#[derive(serde::Serialize, serde::Deserialize, PartialEq, Debug)]
 pub enum ErrorCode {
     UnknownError,
     UnexpectedOP,
@@ -122,7 +122,7 @@ pub struct TestingResultData {
 /// Data sent with an InvalidSession OP code
 #[derive(serde::Serialize, serde::Deserialize, Debug)]
 pub struct InvalidSessionData {
-    code: ErrorCode,
+    pub code: ErrorCode,
     pub description: String,
     pub explanation: String,
 }