aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorQuentin Dufour <quentin@deuxfleurs.fr>2024-01-24 18:57:50 +0100
committerQuentin Dufour <quentin@deuxfleurs.fr>2024-01-24 18:57:50 +0100
commitbbb050e3990125e51ae434654ae6fdea4f621650 (patch)
tree3a861ac53674cfe1d62b167dcc512d924382a833 /src
parent0adb92e8ff34c1f1671e7afccd27874372d68bbd (diff)
downloadaerogramme-bbb050e3990125e51ae434654ae6fdea4f621650.tar.gz
aerogramme-bbb050e3990125e51ae434654ae6fdea4f621650.zip
Basic response encoding
Diffstat (limited to 'src')
-rw-r--r--src/auth.rs45
1 files changed, 44 insertions, 1 deletions
diff --git a/src/auth.rs b/src/auth.rs
index b2d0fae..05c88ce 100644
--- a/src/auth.rs
+++ b/src/auth.rs
@@ -4,7 +4,7 @@ use std::sync::Arc;
use anyhow::{Result, anyhow};
use futures::stream::{FuturesUnordered, StreamExt};
use tokio::io::BufStream;
-use tokio::io::AsyncBufReadExt;
+use tokio::io::{AsyncBufReadExt, AsyncWriteExt};
use tokio::net::{TcpListener, TcpStream};
use tokio::sync::watch;
@@ -117,6 +117,7 @@ impl NetLoop {
}
async fn run(mut self) -> Result<()> {
+ let mut resp_buff = BytesMut::new();
let mut buff: Vec<u8> = Vec::new();
loop {
buff.clear();
@@ -129,6 +130,12 @@ impl NetLoop {
}
let (input, cmd) = client_command(&buff).map_err(|_| anyhow!("Unable to parse command"))?;
println!("input: {:?}, cmd: {:?}", input, cmd);
+ ServerCommand::Version {
+ major: 1,
+ minor: 2,
+ }.encode(&mut resp_buff)?;
+ self.stream.write_all(&resp_buff).await?;
+ self.stream.flush().await?;
},
_ = self.stop.changed() => {
tracing::debug!("Server is stopping, quitting this runner");
@@ -508,3 +515,39 @@ fn server_command(buf: &u8) -> IResult<&u8, ServerCommand> {
// DOVECOT AUTH ENCODING
//
// ------------------------------------------------------------------
+use tokio_util::bytes::{BufMut, BytesMut};
+trait Encode {
+ fn encode(&self, out: &mut BytesMut) -> Result<()>;
+}
+
+fn tab_enc(out: &mut BytesMut) {
+ out.put(&[0x09][..])
+}
+
+fn lf_enc(out: &mut BytesMut) {
+ out.put(&[0x0A][..])
+}
+
+impl Encode for ServerCommand {
+ fn encode(&self, out: &mut BytesMut) -> Result<()> {
+ match self {
+ Self::Version { major, minor } => {
+ out.put(&b"VERSION"[..]);
+ tab_enc(out);
+ out.put(major.to_string().as_bytes());
+ tab_enc(out);
+ out.put(minor.to_string().as_bytes());
+ lf_enc(out);
+ },
+ Self::Spid(v) => unimplemented!(),
+ Self::Cuid(v) => unimplemented!(),
+ Self::Mech { kind, parameters } => unimplemented!(),
+ Self::Cookie(v) => unimplemented!(),
+ Self::Done => unimplemented!(),
+ Self::Fail {id, user_id, code } => unimplemented!(),
+ Self::Cont { id, data } => unimplemented!(),
+ Self::Ok { id, user_id, parameters } => unimplemented!(),
+ }
+ Ok(())
+ }
+}