leyoda/cosmic-greeter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix just mock server

Browse source
This commit is contained in:
Frederic Laing 2025-11-12 16:24:05 +01:00 • committed by Jeremy Soller
parent cf8559e07e
commit a29c1a9162
2 changed files with 94 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

117
examples/server.rs
View file

@ -2,67 +2,74 @@ use greetd_ipc::{AuthMessageType, ErrorType, Request, Response, codec::TokioCode
use std::{env, fs, io, thread};
use tokio::net::UnixListener;
#[tokio::main]
async fn main() {
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);
// Set up the socket environment variable before spawning greeter
unsafe { env::set_var("GREETD_SOCK", &greetd_sock) };
thread::spawn(|| {
cosmic_greeter::greeter::main().unwrap();
// Spawn the mock greetd server in a background thread
thread::spawn(move || {
let rt = tokio::runtime::Runtime::new().unwrap();
rt.block_on(async {
let listener = UnixListener::bind(&greetd_sock).unwrap();
println!("listening at {:?}", greetd_sock);
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_deref() {
Some("password") => Response::Success,
_ => Response::Error {
error_type: ErrorType::AuthError,
description: "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();
}
}
});
});
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_deref() {
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();
}
}
// Run greeter on main thread (required for winit event loop)
cosmic_greeter::greeter::main().unwrap();
}