70 lines
2.5 KiB
Rust
70 lines
2.5 KiB
Rust
use greetd_ipc::{codec::TokioCodec, AuthMessageType, ErrorType, Request, Response};
|
|
use std::{env, fs, io, thread};
|
|
use tokio::net::UnixListener;
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let greetd_sock = env::current_dir().unwrap().join("socket");
|
|
if greetd_sock.exists() {
|
|
fs::remove_file(&greetd_sock).unwrap();
|
|
}
|
|
|
|
let listener = UnixListener::bind(&greetd_sock).unwrap();
|
|
println!("listening at {:?}", greetd_sock);
|
|
unsafe { env::set_var("GREETD_SOCK", &greetd_sock) };
|
|
thread::spawn(|| {
|
|
cosmic_greeter::greeter::main().unwrap();
|
|
});
|
|
|
|
loop {
|
|
let (socket, _addr) = listener.accept().await.unwrap();
|
|
println!("new connection");
|
|
|
|
loop {
|
|
let request = {
|
|
socket.readable().await.unwrap();
|
|
|
|
let mut bytes = Vec::with_capacity(4096);
|
|
match socket.try_read_buf(&mut bytes) {
|
|
Ok(0) => break,
|
|
Ok(count) => {
|
|
println!("read {} bytes", count);
|
|
}
|
|
Err(err) => match err.kind() {
|
|
io::ErrorKind::WouldBlock => continue,
|
|
_ => {
|
|
println!("failed to read socket: {:?}", err);
|
|
break;
|
|
}
|
|
},
|
|
}
|
|
|
|
let mut cursor = io::Cursor::new(bytes);
|
|
Request::read_from(&mut cursor).await.unwrap()
|
|
};
|
|
println!("{:?}", request);
|
|
|
|
let response = match request {
|
|
Request::CreateSession { .. } => Response::AuthMessage {
|
|
auth_message_type: AuthMessageType::Secret,
|
|
auth_message: "MOCKING:".to_string(),
|
|
},
|
|
Request::PostAuthMessageResponse { response } => {
|
|
match response.as_ref().map(|x| x.as_str()) {
|
|
Some("password") => Response::Success,
|
|
_ => Response::Error {
|
|
error_type: ErrorType::AuthError,
|
|
description: "pam_authenticate: AUTH_ERR".to_string(),
|
|
},
|
|
}
|
|
}
|
|
Request::StartSession { .. } => Response::Success,
|
|
Request::CancelSession => Response::Success,
|
|
};
|
|
|
|
let mut bytes = Vec::with_capacity(4096);
|
|
response.write_to(&mut bytes).await.unwrap();
|
|
socket.try_write(&bytes).unwrap();
|
|
}
|
|
}
|
|
}
|