feat(gear): add flush after data wrote to stream

src/exceptions.rs

@@ -1,10 +1,10 @@
 //! # Oblivion exception
 //! All exceptions to the Oblivion function return `OblivionException`.
+#[cfg(feature = "python")]
+use pyo3::prelude::*;
 use ring::error::Unspecified;
 use scrypt::errors::InvalidOutputLen;
 use thiserror::Error;
-#[cfg(feature = "python")]
-use pyo3::prelude::*;
 
 /// ## Oblivion exception iterator
 /// Use an iterator as the type of exception returned by a function.
@@ -55,6 +55,8 @@ pub enum OblivionException {
     EncryptError { error: Unspecified },
     #[error("Exception while decrypting: {error:?}")]
     DecryptError { error: Unspecified },
+    #[error("Failed to send some data: {message}")]
+    TCPWriteFailed { message: String },
 }
 
 #[cfg(feature = "python")]

src/utils/gear.rs

@@ -35,7 +35,7 @@ impl AbsoluteNonceSequence {
 ///
 /// Used to abstract Oblivion's handling of transmitted data, wrapping all data type conversions.
 pub struct Socket {
-    tcp: TcpStream,
+    pub tcp: TcpStream,
 }
 
 impl Socket {
@@ -101,9 +101,16 @@ impl Socket {
     }
 
     pub async fn send(&mut self, data: &[u8]) -> Result<(), OblivionException> {
-        match self.tcp.write(data).await {
-            Ok(_) => Ok(()),
-            Err(_) => Err(OblivionException::UnexpectedDisconnection),
+        match self.tcp.write_all(&data).await {
+            Ok(_) => match self.tcp.flush().await {
+                Ok(_) => Ok(()),
+                Err(e) => Err(OblivionException::TCPWriteFailed {
+                    message: e.to_string(),
+                }),
+            },
+            Err(e) => Err(OblivionException::TCPWriteFailed {
+                message: e.to_string(),
+            }),
         }
     }