mirror of
https://github.com/MeexReay/bRAC.git
synced 2025-05-06 13:38:04 +03:00
init commit
This commit is contained in:
parent
95b22b4004
commit
9bc1cdcba7
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
@ -0,0 +1 @@
|
||||
/target
|
7
Cargo.lock
generated
Normal file
7
Cargo.lock
generated
Normal file
@ -0,0 +1,7 @@
|
||||
# This file is automatically @generated by Cargo.
|
||||
# It is not intended for manual editing.
|
||||
version = 4
|
||||
|
||||
[[package]]
|
||||
name = "bRAC"
|
||||
version = "0.1.0"
|
6
Cargo.toml
Normal file
6
Cargo.toml
Normal file
@ -0,0 +1,6 @@
|
||||
[package]
|
||||
name = "bRAC"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
[dependencies]
|
109
src/main.rs
Normal file
109
src/main.rs
Normal file
@ -0,0 +1,109 @@
|
||||
use std::{error::Error, io::{stdin, stdout, BufRead, Read, Write}, net::TcpStream, thread};
|
||||
|
||||
const MAX_MESSAGES: usize = 100;
|
||||
|
||||
type E = Box<dyn Error>;
|
||||
|
||||
fn send_message(host: &str, message: &str) -> Result<(), E> {
|
||||
let mut stream = TcpStream::connect(host)?;
|
||||
stream.write_all(&[0x01])?;
|
||||
stream.write_all(message.as_bytes())?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn sanitize_string(s: &str, sanitize_newlines: bool) -> String {
|
||||
let mut sanitized = s.replace(&['\x08', '\x0D', '\x1B'][..], "");
|
||||
|
||||
if sanitize_newlines {
|
||||
sanitized = sanitized.replace("\n", "\\\\n");
|
||||
if !s.ends_with('\n') {
|
||||
sanitized.push('\n');
|
||||
}
|
||||
}
|
||||
|
||||
sanitized
|
||||
}
|
||||
|
||||
|
||||
/// max messages count: 100
|
||||
fn read_messages(host: &str) -> Result<Vec<String>, E> {
|
||||
let mut stream = TcpStream::connect(host)?;
|
||||
stream.write_all(&[0x00])?;
|
||||
let packet_size = {
|
||||
let mut buf= vec![0; 10];
|
||||
stream.read(&mut buf)?;
|
||||
String::from_utf8(buf)?.trim_matches(char::from(0)).parse()?
|
||||
};
|
||||
stream.write_all(&[0x01])?;
|
||||
let packet_data = {
|
||||
let mut buf = vec![0; packet_size];
|
||||
stream.read_exact(&mut buf)?;
|
||||
let buf_str = String::from_utf8_lossy(&buf).to_string();
|
||||
let start_null = buf_str.len() - buf_str.trim_start_matches(char::from(0)).len();
|
||||
let mut buf = vec![0; start_null];
|
||||
stream.read_exact(&mut buf)?;
|
||||
format!("{}{}", &buf_str, String::from_utf8_lossy(&buf).to_string())
|
||||
};
|
||||
let packet_data = sanitize_string(&packet_data, false);
|
||||
let mut lines: Vec<String> = packet_data.split("\n").map(|o| o.to_string()).collect();
|
||||
lines.reverse();
|
||||
lines.truncate(MAX_MESSAGES);
|
||||
lines.reverse();
|
||||
Ok(lines)
|
||||
}
|
||||
|
||||
fn print_console(messages: Vec<String>) -> Result<(), E> {
|
||||
let mut out = stdout().lock();
|
||||
let text = format!("{}\n> ", messages.join("\n"));
|
||||
out.write_all(text.as_bytes())?;
|
||||
out.flush()?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn recv_loop(host: &str) -> Result<(), E> {
|
||||
let mut cache = Vec::new();
|
||||
while let Ok(messages) = read_messages(host) {
|
||||
if cache == messages { continue }
|
||||
print_console(messages.clone())?;
|
||||
cache = messages;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn read_host() -> Option<String> {
|
||||
let mut out = stdout().lock();
|
||||
out.write_all("Host (default: meex.lol:11234) > ".as_bytes()).ok()?;
|
||||
out.flush().ok()?;
|
||||
stdin().lock().lines().next()
|
||||
.map(|o| o.ok())
|
||||
.flatten()
|
||||
.map(|o| o.trim().to_string())
|
||||
}
|
||||
|
||||
fn main() {
|
||||
let host = read_host();
|
||||
|
||||
let host = if let Some(host) = &host {
|
||||
if host.is_empty() {
|
||||
"meex.lol:11234"
|
||||
} else {
|
||||
host.as_str()
|
||||
}
|
||||
} else {
|
||||
"meex.lol:11234"
|
||||
}.to_string();
|
||||
|
||||
thread::spawn({
|
||||
let host = host.clone();
|
||||
|
||||
move || {
|
||||
let _ = recv_loop(&host);
|
||||
println!("Connection closed");
|
||||
}
|
||||
});
|
||||
|
||||
let mut lines = stdin().lock().lines();
|
||||
while let Some(Ok(message)) = lines.next() {
|
||||
send_message(&host, &message).expect("Error sending message");
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user