Implement kill_client

isototest/src/connection.rs

@@ -1,3 +1,9 @@
+// SPDX-FileCopyrightTest: Christopher Hock <[email protected]>
+// SPDX-License-Identifier: GPL-2.0-or-later
+
+//! # Connection Module
+//!
+//! This module handles the VncClient and its connection to the VncServer.
 use tokio::{self, net::TcpStream};
 use vnc::{PixelFormat, VncClient, VncConnector, VncError};
 
@@ -52,3 +58,23 @@ pub async fn create_vnc_client(
 
     Ok(vnc)
 }
+
+/// Stop VNC engine, release all resources
+///
+/// # Parameters
+///
+/// * client: `VncClient` - The client to kill.
+///
+/// # Returns
+///
+/// * `Ok(())` - In case the client terminates correctly.
+/// * `Err(VncError)` - Escalates the `VncError` upwards, if the `.close()` function of `vnc-rs`
+/// returns an error.
+pub async fn kill_client(client: VncClient) -> Result<(), VncError> {
+    match client.close().await {
+        Ok(_) => {},
+        Err(e) => return Err(e)
+    };
+    drop(client);
+    Ok(())
+}