midefos/martillo-maldito
midefos/martillo-maldito
1
0
Fork 0
Code

better commands

Browse Source
This commit is contained in:
midefos 2024-05-20 23:54:15 +02:00
parent 1919ec183e
commit f7c2f313f7
1 changed files with 35 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
src/tpc_command_server.rs
View File

@ -19,9 +19,7 @@ pub fn start_tcp_command_server() {
}
fn handle_client(mut stream: &TcpStream) {
let reader = BufReader::new(stream);
for line in reader.lines() {
for line in BufReader::new(stream).lines() {
let buffer = match line {
Ok(data) => data,
Err(_) => return,
@ -44,27 +42,48 @@ fn handle_client(mut stream: &TcpStream) {
fn handle_command(command: &str, arguments: Vec<&str>) -> String {
match command {
"ban" => {
if let (Some(ip), Some(port)) = (arguments.get(0), arguments.get(1)) {
"banport" => {
if let Some(port) = arguments.get(0) {
let iptables = iptables::new(false).unwrap();
let _ = iptables.append_unique(
"filter",
"INPUT",
&format!("-s {} -p tcp --dport {} -j ACCEPT", ip, port),
);
let _ = iptables.append_unique(
"filter",
"INPUT",
&format!("-p tcp --dport {} -j DROP", port),
);
format!("banned port {}, only {} allowed", port, ip)
format!("banned port {} for all ips", port)
} else {
"missing args for ban: ip and port".to_string()
"missing args for banport: port".to_string()
}
}
"unban" => {
"unbanport" => {
if let Some(port) = arguments.get(0) {
let iptables = iptables::new(false).unwrap();
let _ = iptables.delete(
"filter",
"INPUT",
&format!("-p tcp --dport {} -j DROP", port),
);
format!("unbanned port {}", port)
} else {
"missing args for unbanport: port".to_string()
}
}
"allowipport" => {
if let (Some(ip), Some(port)) = (arguments.get(0), arguments.get(1)) {
let iptables = iptables::new(false).unwrap();
let _ = iptables.append_unique(
"filter",
"INPUT",
&format!("-s {} -p tcp --dport {} -j ACCEPT", ip, port),
);
format!("allowed {} to access {}", ip, port)
} else {
"missing args for allowipport: ip and port".to_string()
}
}
"removeipport" => {
if let (Some(ip), Some(port)) = (arguments.get(0), arguments.get(1)) {
let iptables = iptables::new(false).unwrap();
let _ = iptables.delete(
@ -72,16 +91,9 @@ fn handle_command(command: &str, arguments: Vec<&str>) -> String {
"INPUT",
&format!("-s {} -p tcp --dport {} -j ACCEPT", ip, port),
);
let _ = iptables.delete(
"filter",
"INPUT",
&format!("-p tcp --dport {} -j DROP", port),
);
format!("unbanned port {}, used for {}", port, ip)
format!("rm {} access to {}", ip, port)
} else {
"missing args for unban: ip and port".to_string()
"missing args for rmipport: ip and port".to_string()
}
}
_ => {