76 lines
3.7 KiB
Rust
76 lines
3.7 KiB
Rust
use echo_api::{NetView, Priv, Sender, ServiceCtx, Store};
|
|
|
|
// SESSION LIST <min> | SESSION VIEW <ip>: inspect live per-IP session counts.
|
|
// EXCEPTION ADD <ip-mask> <limit> [reason] | DEL <ip-mask> | LIST: manage the
|
|
// per-IP-mask allowances that override the default limit. All admin-only.
|
|
pub fn handle_session(me: &str, from: &Sender, args: &[&str], ctx: &mut ServiceCtx, net: &dyn NetView) {
|
|
if !from.privs.has(Priv::Oper) {
|
|
ctx.notice(me, from.uid, "Access denied — SESSION needs the \x02operator\x02 privilege.");
|
|
return;
|
|
}
|
|
match args.get(1).map(|s| s.to_ascii_uppercase()).as_deref() {
|
|
Some("LIST") => {
|
|
let min = args.get(2).and_then(|n| n.parse::<u32>().ok()).unwrap_or(2);
|
|
let over = net.sessions_over(min);
|
|
if over.is_empty() {
|
|
ctx.notice(me, from.uid, format!("No IP has {min} or more sessions."));
|
|
return;
|
|
}
|
|
for (ip, n) in &over {
|
|
ctx.notice(me, from.uid, format!(" \x02{n}\x02 sessions from \x02{ip}\x02"));
|
|
}
|
|
ctx.notice(me, from.uid, format!("End of session list ({} IP(s)).", over.len()));
|
|
}
|
|
Some("VIEW") => {
|
|
let Some(&ip) = args.get(2) else {
|
|
ctx.notice(me, from.uid, "Syntax: SESSION VIEW <ip>");
|
|
return;
|
|
};
|
|
ctx.notice(me, from.uid, format!("\x02{ip}\x02 has \x02{}\x02 live session(s).", net.session_count(ip)));
|
|
}
|
|
_ => ctx.notice(me, from.uid, "Syntax: SESSION LIST <min> | SESSION VIEW <ip>"),
|
|
}
|
|
}
|
|
|
|
pub fn handle_exception(me: &str, from: &Sender, args: &[&str], ctx: &mut ServiceCtx, db: &mut dyn Store) {
|
|
if !from.privs.has(Priv::Oper) {
|
|
ctx.notice(me, from.uid, "Access denied — EXCEPTION needs the \x02admin\x02 privilege.");
|
|
return;
|
|
}
|
|
match args.get(1).map(|s| s.to_ascii_uppercase()).as_deref() {
|
|
Some("ADD") => {
|
|
let (Some(&mask), Some(limit)) = (args.get(2), args.get(3).and_then(|n| n.parse::<u32>().ok())) else {
|
|
ctx.notice(me, from.uid, "Syntax: EXCEPTION ADD <ip-mask> <limit> [reason]");
|
|
return;
|
|
};
|
|
let reason = if args.len() > 4 { args[4..].join(" ") } else { "No reason given".to_string() };
|
|
db.session_except_add(mask, limit, &reason);
|
|
let note = if limit == 0 { " (unlimited)".to_string() } else { format!(" (limit {limit})") };
|
|
ctx.notice(me, from.uid, format!("Session exception for \x02{mask}\x02 set{note}."));
|
|
}
|
|
Some("DEL") | Some("REMOVE") => {
|
|
let Some(&mask) = args.get(2) else {
|
|
ctx.notice(me, from.uid, "Syntax: EXCEPTION DEL <ip-mask>");
|
|
return;
|
|
};
|
|
if db.session_except_del(mask) {
|
|
ctx.notice(me, from.uid, format!("Session exception for \x02{mask}\x02 removed."));
|
|
} else {
|
|
ctx.notice(me, from.uid, format!("No session exception matches \x02{mask}\x02."));
|
|
}
|
|
}
|
|
Some("LIST") => {
|
|
let list = db.session_exceptions();
|
|
if list.is_empty() {
|
|
ctx.notice(me, from.uid, "No session exceptions.");
|
|
return;
|
|
}
|
|
for (mask, limit, reason) in &list {
|
|
let lim = if *limit == 0 { "unlimited".to_string() } else { limit.to_string() };
|
|
ctx.notice(me, from.uid, format!(" \x02{mask}\x02 → {lim} — {reason}"));
|
|
}
|
|
ctx.notice(me, from.uid, format!("End of exception list ({} shown).", list.len()));
|
|
}
|
|
_ => ctx.notice(me, from.uid, "Syntax: EXCEPTION ADD <ip-mask> <limit> [reason] | EXCEPTION DEL <ip-mask> | EXCEPTION LIST"),
|
|
}
|
|
}
|