Debut d'un contrôle point
This commit is contained in:
253
pmoupnp/src/ssdp/client.rs
Normal file
253
pmoupnp/src/ssdp/client.rs
Normal file
@@ -0,0 +1,253 @@
|
||||
//! Client SSDP pour la découverte des devices UPnP
|
||||
|
||||
use super::{MAX_AGE, SSDP_MULTICAST_ADDR, SSDP_PORT};
|
||||
use socket2::{Domain, Protocol, Socket, Type};
|
||||
use std::collections::HashMap;
|
||||
use std::net::{SocketAddr, UdpSocket};
|
||||
use std::sync::Arc;
|
||||
use std::time::Duration;
|
||||
use tracing::{debug, info, warn};
|
||||
|
||||
/// Événements SSDP intéressants pour un control point
|
||||
#[derive(Debug, Clone)]
|
||||
pub enum SsdpEvent {
|
||||
Alive {
|
||||
usn: String,
|
||||
nt: String,
|
||||
location: String,
|
||||
server: String,
|
||||
max_age: u32,
|
||||
from: SocketAddr,
|
||||
},
|
||||
ByeBye {
|
||||
usn: String,
|
||||
nt: String,
|
||||
from: SocketAddr,
|
||||
},
|
||||
SearchResponse {
|
||||
usn: String,
|
||||
st: String,
|
||||
location: String,
|
||||
server: String,
|
||||
max_age: u32,
|
||||
from: SocketAddr,
|
||||
},
|
||||
}
|
||||
|
||||
/// Client SSDP pour envoyer des M-SEARCH et écouter les annonces
|
||||
pub struct SsdpClient {
|
||||
socket: Arc<UdpSocket>,
|
||||
}
|
||||
|
||||
impl SsdpClient {
|
||||
/// Crée un nouveau client SSDP
|
||||
pub fn new() -> std::io::Result<Self> {
|
||||
let addr = format!("{}:{}", SSDP_MULTICAST_ADDR, SSDP_PORT);
|
||||
|
||||
let socket2 = Socket::new(Domain::IPV4, Type::DGRAM, Some(Protocol::UDP))?;
|
||||
socket2.set_reuse_address(true)?;
|
||||
|
||||
#[cfg(unix)]
|
||||
{
|
||||
use std::os::unix::io::AsRawFd;
|
||||
let fd = socket2.as_raw_fd();
|
||||
let optval: libc::c_int = 1;
|
||||
unsafe {
|
||||
let result = libc::setsockopt(
|
||||
fd,
|
||||
libc::SOL_SOCKET,
|
||||
libc::SO_REUSEPORT,
|
||||
&optval as *const _ as *const libc::c_void,
|
||||
std::mem::size_of_val(&optval) as libc::socklen_t,
|
||||
);
|
||||
if result != 0 {
|
||||
return Err(std::io::Error::last_os_error());
|
||||
}
|
||||
}
|
||||
debug!("✅ SsdpClient SO_REUSEPORT enabled (Unix)");
|
||||
}
|
||||
|
||||
#[cfg(windows)]
|
||||
{
|
||||
debug!("✅ SsdpClient SO_REUSEADDR enabled (Windows - SO_REUSEPORT not needed)");
|
||||
}
|
||||
|
||||
let bind_addr: SocketAddr = format!("0.0.0.0:{}", SSDP_PORT).parse().unwrap();
|
||||
socket2.bind(&bind_addr.into())?;
|
||||
|
||||
let socket: UdpSocket = socket2.into();
|
||||
socket.join_multicast_v4(
|
||||
&SSDP_MULTICAST_ADDR.parse().unwrap(),
|
||||
&"0.0.0.0".parse().unwrap(),
|
||||
)?;
|
||||
socket.set_read_timeout(Some(Duration::from_secs(1)))?;
|
||||
socket.set_multicast_loop_v4(false)?;
|
||||
|
||||
info!("✅ SSDP client ready on {}", addr);
|
||||
|
||||
Ok(Self {
|
||||
socket: Arc::new(socket),
|
||||
})
|
||||
}
|
||||
|
||||
/// Envoie un M-SEARCH pour un type donné
|
||||
pub fn send_msearch(&self, st: &str, mx: u32) -> std::io::Result<()> {
|
||||
let mx = mx.max(1); // MX doit être >= 1
|
||||
let msg = format!(
|
||||
"M-SEARCH * HTTP/1.1\r\n\
|
||||
HOST: {}:{}\r\n\
|
||||
MAN: \"ssdp:discover\"\r\n\
|
||||
MX: {}\r\n\
|
||||
ST: {}\r\n\
|
||||
USER-AGENT: PMOMusic SSDP Client\r\n\
|
||||
\r\n",
|
||||
SSDP_MULTICAST_ADDR, SSDP_PORT, mx, st
|
||||
);
|
||||
|
||||
let addr: SocketAddr = format!("{}:{}", SSDP_MULTICAST_ADDR, SSDP_PORT)
|
||||
.parse()
|
||||
.unwrap();
|
||||
|
||||
match self.socket.send_to(msg.as_bytes(), addr) {
|
||||
Ok(_) => {
|
||||
info!("📤 M-SEARCH sent (ST={}, MX={})", st, mx);
|
||||
debug!(
|
||||
"📨 M-SEARCH payload\n<details>\n\n```\n{}\n```\n</details>\n",
|
||||
msg
|
||||
);
|
||||
Ok(())
|
||||
}
|
||||
Err(e) => {
|
||||
warn!("❌ Failed to send M-SEARCH: {}", e);
|
||||
Err(e)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Boucle de réception bloquante pour traiter les événements SSDP
|
||||
pub fn run_event_loop<F>(&self, mut on_event: F) -> !
|
||||
where
|
||||
F: FnMut(SsdpEvent) + Send + 'static,
|
||||
{
|
||||
let socket = Arc::clone(&self.socket);
|
||||
let mut buf = [0u8; 8192];
|
||||
loop {
|
||||
match socket.recv_from(&mut buf) {
|
||||
Ok((n, from)) => {
|
||||
let data = String::from_utf8_lossy(&buf[..n]);
|
||||
if let Some(event) = parse_message(&data, from) {
|
||||
debug!(
|
||||
"📥 SSDP datagram from {}\n<details>\n\n```\n{}\n```\n</details>\n",
|
||||
from, data
|
||||
);
|
||||
on_event(event);
|
||||
}
|
||||
}
|
||||
Err(e) if e.kind() == std::io::ErrorKind::WouldBlock => {
|
||||
// Timeout, recommencer
|
||||
continue;
|
||||
}
|
||||
Err(e) => {
|
||||
warn!("❌ SSDP client read error: {}", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn parse_message(data: &str, from: SocketAddr) -> Option<SsdpEvent> {
|
||||
let mut lines = data.lines();
|
||||
let first_line = lines.next()?.trim();
|
||||
let headers = parse_headers(lines);
|
||||
|
||||
if first_line.to_ascii_uppercase().starts_with("NOTIFY") {
|
||||
handle_notify(&headers, from)
|
||||
} else if first_line.to_ascii_uppercase().starts_with("HTTP/1.1 200") {
|
||||
handle_search_response(&headers, from)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
}
|
||||
|
||||
fn handle_notify(headers: &HashMap<String, String>, from: SocketAddr) -> Option<SsdpEvent> {
|
||||
let nts = headers.get("NTS")?.to_ascii_lowercase();
|
||||
let nt = headers.get("NT")?.to_string();
|
||||
let usn = headers.get("USN")?.to_string();
|
||||
|
||||
if nts == "ssdp:alive" {
|
||||
let location = headers.get("LOCATION")?.to_string();
|
||||
let server = headers.get("SERVER")?.to_string();
|
||||
let max_age = parse_max_age(headers.get("CACHE-CONTROL"));
|
||||
Some(SsdpEvent::Alive {
|
||||
usn,
|
||||
nt,
|
||||
location,
|
||||
server,
|
||||
max_age,
|
||||
from,
|
||||
})
|
||||
} else if nts == "ssdp:byebye" {
|
||||
Some(SsdpEvent::ByeBye { usn, nt, from })
|
||||
} else {
|
||||
None
|
||||
}
|
||||
}
|
||||
|
||||
fn handle_search_response(
|
||||
headers: &HashMap<String, String>,
|
||||
from: SocketAddr,
|
||||
) -> Option<SsdpEvent> {
|
||||
let st = headers.get("ST")?.to_string();
|
||||
let usn = headers.get("USN")?.to_string();
|
||||
let location = headers.get("LOCATION")?.to_string();
|
||||
let server = headers.get("SERVER")?.to_string();
|
||||
let max_age = parse_max_age(headers.get("CACHE-CONTROL"));
|
||||
|
||||
Some(SsdpEvent::SearchResponse {
|
||||
usn,
|
||||
st,
|
||||
location,
|
||||
server,
|
||||
max_age,
|
||||
from,
|
||||
})
|
||||
}
|
||||
|
||||
fn parse_headers<'a, I>(lines: I) -> HashMap<String, String>
|
||||
where
|
||||
I: Iterator<Item = &'a str>,
|
||||
{
|
||||
let mut headers = HashMap::new();
|
||||
for line in lines {
|
||||
let line = line.trim();
|
||||
if line.is_empty() {
|
||||
break;
|
||||
}
|
||||
if let Some((name, value)) = line.split_once(':') {
|
||||
headers.insert(name.trim().to_ascii_uppercase(), value.trim().to_string());
|
||||
}
|
||||
}
|
||||
headers
|
||||
}
|
||||
|
||||
fn parse_max_age(value: Option<&String>) -> u32 {
|
||||
if let Some(v) = value {
|
||||
for part in v.split(',') {
|
||||
let part = part.trim();
|
||||
if let Some(rest) = part.strip_prefix("max-age=") {
|
||||
if let Ok(age) = rest.trim().parse::<u32>() {
|
||||
return age;
|
||||
}
|
||||
} else {
|
||||
let lower = part.to_ascii_lowercase();
|
||||
if let Some(idx) = lower.find("max-age=") {
|
||||
let raw = &part[idx + 8..];
|
||||
if let Ok(age) = raw.trim().parse::<u32>() {
|
||||
return age;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
MAX_AGE
|
||||
}
|
||||
@@ -22,9 +22,11 @@
|
||||
//! - **Max-Age**: 1800 secondes (30 minutes)
|
||||
//! - **Announcement Period**: 900 secondes (15 minutes, Max-Age/2)
|
||||
|
||||
mod client;
|
||||
mod device;
|
||||
mod server;
|
||||
|
||||
pub use client::{SsdpClient, SsdpEvent};
|
||||
pub use device::SsdpDevice;
|
||||
pub use server::SsdpServer;
|
||||
|
||||
|
||||
@@ -290,7 +290,7 @@ impl UpnpServerExt for Server {
|
||||
limit: usize,
|
||||
) -> Result<Arc<CoverCache>, anyhow::Error> {
|
||||
// Délègue à l'implémentation pmocovers (qui enregistre WebP + JPEG + API)
|
||||
use pmocovers::CoverCacheExt;
|
||||
|
||||
let cache = pmocovers::CoverCacheExt::init_cover_cache(self, cache_dir, limit).await?;
|
||||
Ok(cache)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user