protocol feat.

joelittyab-git · joelittyab-git · commit 35dbcba373fd · 2025-05-16T22:45:16.000+05:30
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/net/src/cryptography/mod.rs b/net/src/cryptography/mod.rs
diff --git a/net/src/lib.rs b/net/src/lib.rs
@@ -13,4 +13,7 @@
 pub mod socket;
 
 
-pub mod protocol;
+pub mod protocol;
+
+
+pub mod cryptography;
diff --git a/net/src/protocol/data.rs b/net/src/protocol/data.rs
diff --git a/net/src/protocol/interface.rs b/net/src/protocol/interface.rs
@@ -406,8 +406,10 @@ pub enum MTPHeaderUnit {
           to: MessagePublish,
      },
 
+     //TODO
      QueueCreation{
-          
+          name:String,
+          access:QueueAccess
      }
 }
 
diff --git a/net/src/socket/client/data.rs b/net/src/socket/client/data.rs
@@ -0,0 +1,5 @@
+pub struct ClientSocketData{
+
+}
+
+impl ClientSocketData{}
diff --git a/net/src/socket/data.rs b/net/src/socket/data.rs
@@ -1,3 +1,5 @@
+use crate::protocol::{error::ProtocolError, interface::{MessageTransferProtocol, MessageTransferProtocolPayload}};
+
 /// Represents different types of data with their associated values.
 ///
 /// This enum is used to represent data in various formats, including
@@ -192,4 +194,9 @@ impl Data{
 
           String::from_utf16_lossy(&utf16_encoded)
      }
+}
+
+pub trait ProtocolParser {
+     fn from_raw(&self, bytes:Vec<u8>)->Result<Self, ProtocolError> where Self: Sized;
+     fn to_bytes(&self)->Result<Vec<u8>, ProtocolError>;
 }
diff --git a/server/Cargo.toml b/server/Cargo.toml
@@ -5,3 +5,4 @@ edition = "2021"
 
 [dependencies]
 tokio = { version = "1", features = ["full"] }
+net = { path = "../net" }
diff --git a/server/src/core/mod.rs b/server/src/core/mod.rs
@@ -0,0 +1,20 @@
+use crate::preface::prologue::{error::PrologueError, interface::Prologue};
+
+struct CoreServer<P:Prologue>{
+     prologue:P,
+}
+
+impl <P:Prologue>CoreServer<P> {
+     fn new(prologue:P)->Self
+     where P:Prologue{
+          Self{
+               prologue
+          }
+     }
+
+     fn start(&self){
+          self.prologue.open_socket().inspect_err(|err|{
+               println!("Something went wrong {err}");
+          });
+     }
+}
diff --git a/server/src/main.rs b/server/src/main.rs
@@ -1,3 +1,7 @@
+mod preface;
+mod security;
+mod core;
+
 use tokio::runtime::Runtime;
 
 fn main() {
diff --git a/server/src/preface/mod.rs b/server/src/preface/mod.rs
@@ -0,0 +1,3 @@
+pub mod prologue;
+
+pub mod preface_middleware;
diff --git a/server/src/preface/preface_middleware/error.rs b/server/src/preface/preface_middleware/error.rs
@@ -0,0 +1,21 @@
+use net::protocol::error::ProtocolError;
+
+pub enum MiddlewareError {
+     InvalidData{
+          info:String
+     },
+
+     ProtocolError{
+          source:ProtocolError
+     },
+
+     CError{
+          info:String
+     }
+}
+
+impl MiddlewareError {
+     pub fn new(info:String)->Self{
+          Self::CError { info }
+     }
+}
diff --git a/server/src/preface/preface_middleware/mod.rs b/server/src/preface/preface_middleware/mod.rs
@@ -0,0 +1,9 @@
+pub mod error;
+
+use error::MiddlewareError;
+use net::socket::server::data::SocketData;
+
+pub trait Middleware {
+     fn parse(&self, socket_data:SocketData)->Result<SocketData, MiddlewareError>;
+}
+
diff --git a/server/src/preface/prologue/error.rs b/server/src/preface/prologue/error.rs
@@ -0,0 +1,20 @@
+use std::fmt::Display;
+
+use net::socket::server::error::ServerSocketError;
+
+pub enum PrologueError {
+     SocketError{
+          source:ServerSocketError,
+          info:String
+     },
+
+     MiddlewareError{
+
+     }
+}
+
+impl Display for PrologueError{
+     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        todo!()
+     }
+}
diff --git a/server/src/preface/prologue/interface.rs b/server/src/preface/prologue/interface.rs
@@ -0,0 +1,8 @@
+use net::socket::server::{data::SocketData, error::ServerSocketError};
+
+use super::error::PrologueError;
+
+pub trait Prologue{
+     fn open_socket(&self)->Result<(), PrologueError>;
+     async fn receive_data(&self)->Result<SocketData, PrologueError>;
+}
diff --git a/server/src/preface/prologue/mod.rs b/server/src/preface/prologue/mod.rs
@@ -0,0 +1,59 @@
+
+pub mod error;
+pub mod interface;
+
+use net::socket::server::data::SocketData;
+use crate::security::manager::SecurityManager;
+
+use super::preface_middleware::Middleware;
+
+pub struct ServerPrologue<S>
+where S:SecurityManager{
+     security:S,
+     middlewares:Vec<Box<dyn Middleware>>
+}
+
+impl <T:SecurityManager>ServerPrologue<T> {
+     pub fn new(security:T)->Self
+     where T:SecurityManager{
+          //middlewares
+          let middlewares = Vec::new();
+
+          Self{
+               security,
+               middlewares
+          }
+     }
+
+     pub fn config_middleware(&mut self, middleware:T)->&mut Self
+     where T:Middleware + 'static{
+          self.middlewares.push(Box::new(middleware));
+          return self;
+     }
+
+     pub fn add_middleware(&mut self, middleware:T)
+     where T:Middleware + 'static{
+          self.middlewares.push(Box::new(middleware));
+     }
+
+     async fn run_middlewares(&self, socket_data:SocketData)->SocketData{
+          let mut socket_data_transfered = socket_data;
+
+          // Runs each middleware configured in for this ServerPrologue
+          for middleware in &self.middlewares{
+               socket_data_transfered =  middleware.parse(socket_data_transfered);
+          }
+
+          socket_data_transfered
+     }
+
+     async fn run_prologue(&mut self, socket_data:SocketData){
+
+     }
+
+     async fn secure(&mut self, socket_data:SocketData){
+
+     }
+
+
+}
diff --git a/server/src/security/manager.rs b/server/src/security/manager.rs
@@ -0,0 +1,7 @@
+
+
+
+/// Security manager for the server prologue
+pub trait SecurityManager {
+    
+}
diff --git a/server/src/security/mod.rs b/server/src/security/mod.rs
@@ -0,0 +1 @@
+pub mod manager;

Original file line number	Diff line number	Diff line change
`@@ -406,8 +406,10 @@ pub enum MTPHeaderUnit {`
`406`	`406`	`to: MessagePublish,`
`407`	`407`	`},`
`408`	`408`
	`409`	`+ //TODO`
`409`	`410`	`QueueCreation{`
`410`		`-`
	`411`	`+ name:String,`
	`412`	`+ access:QueueAccess`
`411`	`413`	`}`
`412`	`414`	`}`
`413`	`415`
-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +pub struct ClientSocketData{
++
 +}
++
 +impl ClientSocketData{}
Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,5 @@`
	`1`	`+use crate::protocol::{error::ProtocolError, interface::{MessageTransferProtocol, MessageTransferProtocolPayload}};`
	`2`	`+`
`1`	`3`	`/// Represents different types of data with their associated values.`
`2`	`4`	`///`
`3`	`5`	`/// This enum is used to represent data in various formats, including`
`@@ -192,4 +194,9 @@ impl Data{`
`192`	`194`
`193`	`195`	`String::from_utf16_lossy(&utf16_encoded)`
`194`	`196`	`}`
	`197`	`+}`
	`198`	`+`
	`199`	`+pub trait ProtocolParser {`
	`200`	`+ fn from_raw(&self, bytes:Vec<u8>)->Result<Self, ProtocolError> where Self: Sized;`
	`201`	`+ fn to_bytes(&self)->Result<Vec<u8>, ProtocolError>;`
`195`	`202`	`}`
Original file line number	Diff line number	Diff line change
`@@ -5,3 +5,4 @@ edition = "2021"`
`5`	`5`
`6`	`6`	`[dependencies]`
`7`	`7`	`tokio = { version = "1", features = ["full"] }`
	`8`	`+net = { path = "../net" }`