mirror of
https://github.com/MeexReay/sRAC.git
synced 2025-05-06 05:08:02 +03:00
wrac implementation
This commit is contained in:
parent
2fdc5e6b8d
commit
1a9f7f4e63
312
src/main.rs
312
src/main.rs
@ -7,6 +7,7 @@ use rand::{distr::Alphanumeric, Rng};
|
|||||||
|
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use rustls::{pki_types::{pem::PemObject, CertificateDer, PrivateKeyDer}, ServerConfig, ServerConnection, StreamOwned};
|
use rustls::{pki_types::{pem::PemObject, CertificateDer, PrivateKeyDer}, ServerConfig, ServerConnection, StreamOwned};
|
||||||
|
use tungstenite::{accept, Bytes, Message, WebSocket};
|
||||||
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
@ -186,123 +187,244 @@ fn accept_stream(
|
|||||||
messages: Arc<RwLock<Vec<u8>>>,
|
messages: Arc<RwLock<Vec<u8>>>,
|
||||||
accounts: Arc<RwLock<Vec<Account>>>
|
accounts: Arc<RwLock<Vec<Account>>>
|
||||||
) -> Result<(), Box<dyn Error>> {
|
) -> Result<(), Box<dyn Error>> {
|
||||||
let mut buf = vec![0];
|
if args.enable_wrac {
|
||||||
stream.read_exact(&mut buf)?;
|
let mut websocket = accept(stream).unwrap();
|
||||||
|
|
||||||
if buf[0] == 0x00 {
|
while let Ok(msg) = websocket.read() {
|
||||||
let mut messages = messages.read().unwrap().clone();
|
if let Some(data) = if msg.is_binary() {
|
||||||
|
Some(msg.into_data().to_vec())
|
||||||
|
} else if msg.is_text() {
|
||||||
|
msg.into_text().ok().map(|o| o.as_bytes().to_vec())
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
} {
|
||||||
|
let mut data = data;
|
||||||
|
let Some(id) = data.drain(..1).next() else { return Ok(()) };
|
||||||
|
|
||||||
if let Some(splash) = &args.splash {
|
if id == 0x00 {
|
||||||
stream.write_all((splash.len() + messages.len()).to_string().as_bytes())?;
|
let mut messages = messages.read().unwrap().clone();
|
||||||
|
|
||||||
let mut id = vec![0];
|
if data.is_empty() {
|
||||||
stream.read_exact(&mut id)?;
|
if let Some(splash) = &args.splash {
|
||||||
|
websocket.write(Message::Binary(Bytes::from((messages.len() + splash.len()).to_string().as_bytes().to_vec())))?;
|
||||||
if id[0] == 0x01 {
|
} else {
|
||||||
messages.append(&mut splash.clone().as_bytes().to_vec());
|
websocket.write(Message::Binary(Bytes::from(messages.len().to_string().as_bytes().to_vec())))?;
|
||||||
stream.write_all(&messages)?;
|
}
|
||||||
} else if id[0] == 0x02 {
|
} else {
|
||||||
let mut buf = vec![0; 10];
|
let Some(id) = data.drain(..1).next() else { return Ok(()) };
|
||||||
let size = stream.read(&mut buf)?;
|
|
||||||
buf.truncate(size);
|
|
||||||
|
|
||||||
let len: usize = String::from_utf8(buf)?.parse()?;
|
|
||||||
stream.write_all(&messages[(len - splash.len())..])?;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
stream.write_all(messages.len().to_string().as_bytes())?;
|
|
||||||
|
|
||||||
let mut id = vec![0];
|
if id == 0x01 {
|
||||||
stream.read_exact(&mut id)?;
|
if let Some(splash) = &args.splash {
|
||||||
|
messages.append(&mut splash.clone().as_bytes().to_vec());
|
||||||
|
}
|
||||||
|
websocket.write(Message::Binary(Bytes::from(messages)))?;
|
||||||
|
} else if id == 0x02 {
|
||||||
|
let last_size: usize = String::from_utf8(data)?.parse()?;
|
||||||
|
if let Some(splash) = &args.splash {
|
||||||
|
websocket.write(Message::Binary(Bytes::from(messages[(last_size - splash.len())..].to_vec())))?;
|
||||||
|
} else {
|
||||||
|
websocket.write(Message::Binary(Bytes::from(messages[last_size..].to_vec())))?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if id == 0x01 {
|
||||||
|
if !args.auth_only {
|
||||||
|
add_message(&mut data, messages.clone(), Some(addr.ip()), args.sanitize, args.messages_file.clone())?;
|
||||||
|
}
|
||||||
|
} else if id == 0x02 {
|
||||||
|
let msg = String::from_utf8_lossy(&data).to_string();
|
||||||
|
|
||||||
|
let mut segments = msg.split("\n");
|
||||||
|
|
||||||
|
let Some(name) = segments.next() else { return Ok(()) };
|
||||||
|
let Some(password) = segments.next() else { return Ok(()) };
|
||||||
|
let Some(text) = segments.next() else { return Ok(()) };
|
||||||
|
|
||||||
|
let mut sent = false;
|
||||||
|
|
||||||
if id[0] == 0x01 {
|
for user in accounts.read().unwrap().iter() {
|
||||||
stream.write_all(&messages)?;
|
if user.name() == name {
|
||||||
} else if id[0] == 0x02 {
|
if user.check_password(password) {
|
||||||
let mut buf = vec![0; 10];
|
add_message(&mut text.as_bytes().to_vec(), messages.clone(), None, args.sanitize, args.messages_file.clone())?;
|
||||||
let size = stream.read(&mut buf)?;
|
} else {
|
||||||
buf.truncate(size);
|
websocket.write(Message::Binary(Bytes::from(vec![0x02])))?;
|
||||||
|
}
|
||||||
|
sent = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !sent {
|
||||||
|
websocket.write(Message::Binary(Bytes::from(vec![0x01])))?;
|
||||||
|
}
|
||||||
|
} else if id == 0x03 {
|
||||||
|
let msg = String::from_utf8_lossy(&data).to_string();
|
||||||
|
|
||||||
|
let mut segments = msg.split("\n");
|
||||||
|
|
||||||
|
let Some(name) = segments.next() else { return Ok(()) };
|
||||||
|
let Some(password) = segments.next() else { return Ok(()) };
|
||||||
|
|
||||||
|
let addr = addr.ip().to_string();
|
||||||
|
|
||||||
|
let now: i64 = Local::now().timestamp_millis();
|
||||||
|
|
||||||
let len: usize = String::from_utf8(buf)?.parse()?;
|
let mut continue_send = false;
|
||||||
stream.write_all(&messages[len..])?;
|
|
||||||
|
for user in accounts.read().unwrap().iter() {
|
||||||
|
if user.name() == name {
|
||||||
|
websocket.write(Message::Binary(Bytes::from(vec![0x01])))?;
|
||||||
|
continue_send = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if user.addr() == addr && ((now - user.date()) as usize) < 1000 * args.register_timeout {
|
||||||
|
websocket.write(Message::Binary(Bytes::from(vec![0x01])))?;
|
||||||
|
continue_send = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if continue_send {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
let account = Account::new(name.to_string(), password.to_string(), addr, now);
|
||||||
|
|
||||||
|
if let Some(accounts_file) = args.accounts_file.clone() {
|
||||||
|
let mut file = OpenOptions::new()
|
||||||
|
.write(true)
|
||||||
|
.append(true)
|
||||||
|
.create(true)
|
||||||
|
.open(accounts_file)?;
|
||||||
|
|
||||||
|
file.write_all(&account.to_bytes())?;
|
||||||
|
file.write_all(b"\n")?;
|
||||||
|
file.flush()?;
|
||||||
|
}
|
||||||
|
|
||||||
|
accounts.write().unwrap().push(account);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else if buf[0] == 0x01 {
|
} else {
|
||||||
if !args.auth_only {
|
let mut buf = vec![0];
|
||||||
|
stream.read_exact(&mut buf)?;
|
||||||
|
|
||||||
|
if buf[0] == 0x00 {
|
||||||
|
let mut messages = messages.read().unwrap().clone();
|
||||||
|
|
||||||
|
if let Some(splash) = &args.splash {
|
||||||
|
stream.write_all((splash.len() + messages.len()).to_string().as_bytes())?;
|
||||||
|
|
||||||
|
let mut id = vec![0];
|
||||||
|
stream.read_exact(&mut id)?;
|
||||||
|
|
||||||
|
if id[0] == 0x01 {
|
||||||
|
messages.append(&mut splash.clone().as_bytes().to_vec());
|
||||||
|
stream.write_all(&messages)?;
|
||||||
|
} else if id[0] == 0x02 {
|
||||||
|
let mut buf = vec![0; 10];
|
||||||
|
let size = stream.read(&mut buf)?;
|
||||||
|
buf.truncate(size);
|
||||||
|
|
||||||
|
let len: usize = String::from_utf8(buf)?.parse()?;
|
||||||
|
stream.write_all(&messages[(len - splash.len())..])?;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
stream.write_all(messages.len().to_string().as_bytes())?;
|
||||||
|
|
||||||
|
let mut id = vec![0];
|
||||||
|
stream.read_exact(&mut id)?;
|
||||||
|
|
||||||
|
if id[0] == 0x01 {
|
||||||
|
stream.write_all(&messages)?;
|
||||||
|
} else if id[0] == 0x02 {
|
||||||
|
let mut buf = vec![0; 10];
|
||||||
|
let size = stream.read(&mut buf)?;
|
||||||
|
buf.truncate(size);
|
||||||
|
|
||||||
|
let len: usize = String::from_utf8(buf)?.parse()?;
|
||||||
|
stream.write_all(&messages[len..])?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if buf[0] == 0x01 {
|
||||||
|
if !args.auth_only {
|
||||||
|
let mut buf = vec![0; 1024];
|
||||||
|
let size = stream.read(&mut buf)?;
|
||||||
|
buf.truncate(size);
|
||||||
|
|
||||||
|
add_message(&mut buf, messages.clone(), Some(addr.ip()), args.sanitize, args.messages_file.clone())?;
|
||||||
|
}
|
||||||
|
} else if buf[0] == 0x02 {
|
||||||
|
let mut buf = vec![0; 8192];
|
||||||
|
let size = stream.read(&mut buf)?;
|
||||||
|
buf.truncate(size);
|
||||||
|
|
||||||
|
let msg = String::from_utf8_lossy(&buf).to_string();
|
||||||
|
|
||||||
|
let mut segments = msg.split("\n");
|
||||||
|
|
||||||
|
let Some(name) = segments.next() else { return Ok(()) };
|
||||||
|
let Some(password) = segments.next() else { return Ok(()) };
|
||||||
|
let Some(text) = segments.next() else { return Ok(()) };
|
||||||
|
|
||||||
|
for user in accounts.read().unwrap().iter() {
|
||||||
|
if user.name() == name {
|
||||||
|
if user.check_password(password) {
|
||||||
|
add_message(&mut text.as_bytes().to_vec(), messages.clone(), None, args.sanitize, args.messages_file.clone())?;
|
||||||
|
} else {
|
||||||
|
stream.write_all(&[0x02])?;
|
||||||
|
}
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
stream.write_all(&[0x01])?;
|
||||||
|
} else if buf[0] == 0x03 {
|
||||||
let mut buf = vec![0; 1024];
|
let mut buf = vec![0; 1024];
|
||||||
let size = stream.read(&mut buf)?;
|
let size = stream.read(&mut buf)?;
|
||||||
buf.truncate(size);
|
buf.truncate(size);
|
||||||
|
|
||||||
add_message(&mut buf, messages.clone(), Some(addr.ip()), args.sanitize, args.messages_file.clone())?;
|
|
||||||
}
|
|
||||||
} else if buf[0] == 0x02 {
|
|
||||||
let mut buf = vec![0; 8192];
|
|
||||||
let size = stream.read(&mut buf)?;
|
|
||||||
buf.truncate(size);
|
|
||||||
|
|
||||||
let msg = String::from_utf8_lossy(&buf).to_string();
|
let msg = String::from_utf8_lossy(&buf).to_string();
|
||||||
|
|
||||||
let mut segments = msg.split("\n");
|
let mut segments = msg.split("\n");
|
||||||
|
|
||||||
let Some(name) = segments.next() else { return Ok(()) };
|
let Some(name) = segments.next() else { return Ok(()) };
|
||||||
let Some(password) = segments.next() else { return Ok(()) };
|
let Some(password) = segments.next() else { return Ok(()) };
|
||||||
let Some(text) = segments.next() else { return Ok(()) };
|
|
||||||
|
|
||||||
for user in accounts.read().unwrap().iter() {
|
let addr = addr.ip().to_string();
|
||||||
if user.name() == name {
|
|
||||||
if user.check_password(password) {
|
let now: i64 = Local::now().timestamp_millis();
|
||||||
add_message(&mut text.as_bytes().to_vec(), messages.clone(), None, args.sanitize, args.messages_file.clone())?;
|
|
||||||
} else {
|
for user in accounts.read().unwrap().iter() {
|
||||||
stream.write_all(&[0x02])?;
|
if user.name() == name {
|
||||||
|
stream.write_all(&[0x01])?;
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
if user.addr() == addr && ((now - user.date()) as usize) < 1000 * args.register_timeout {
|
||||||
|
stream.write_all(&[0x01])?;
|
||||||
|
return Ok(());
|
||||||
}
|
}
|
||||||
return Ok(());
|
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
stream.write_all(&[0x01])?;
|
let account = Account::new(name.to_string(), password.to_string(), addr, now);
|
||||||
} else if buf[0] == 0x03 {
|
|
||||||
let mut buf = vec![0; 1024];
|
|
||||||
let size = stream.read(&mut buf)?;
|
|
||||||
buf.truncate(size);
|
|
||||||
|
|
||||||
let msg = String::from_utf8_lossy(&buf).to_string();
|
if let Some(accounts_file) = args.accounts_file.clone() {
|
||||||
|
let mut file = OpenOptions::new()
|
||||||
|
.write(true)
|
||||||
|
.append(true)
|
||||||
|
.create(true)
|
||||||
|
.open(accounts_file)?;
|
||||||
|
|
||||||
let mut segments = msg.split("\n");
|
file.write_all(&account.to_bytes())?;
|
||||||
|
file.write_all(b"\n")?;
|
||||||
let Some(name) = segments.next() else { return Ok(()) };
|
file.flush()?;
|
||||||
let Some(password) = segments.next() else { return Ok(()) };
|
|
||||||
|
|
||||||
let addr = addr.ip().to_string();
|
|
||||||
|
|
||||||
let now: i64 = Local::now().timestamp_millis();
|
|
||||||
|
|
||||||
for user in accounts.read().unwrap().iter() {
|
|
||||||
if user.name() == name {
|
|
||||||
stream.write_all(&[0x01])?;
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
if user.addr() == addr && ((now - user.date()) as usize) < 1000 * args.register_timeout {
|
|
||||||
stream.write_all(&[0x01])?;
|
|
||||||
return Ok(());
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
println!("user registered: {name}");
|
||||||
|
|
||||||
|
accounts.write().unwrap().push(account);
|
||||||
}
|
}
|
||||||
|
|
||||||
let account = Account::new(name.to_string(), password.to_string(), addr, now);
|
|
||||||
|
|
||||||
if let Some(accounts_file) = args.accounts_file.clone() {
|
|
||||||
let mut file = OpenOptions::new()
|
|
||||||
.write(true)
|
|
||||||
.append(true)
|
|
||||||
.create(true)
|
|
||||||
.open(accounts_file)?;
|
|
||||||
|
|
||||||
file.write_all(&account.to_bytes())?;
|
|
||||||
file.write_all(b"\n")?;
|
|
||||||
file.flush()?;
|
|
||||||
}
|
|
||||||
|
|
||||||
println!("user registered: {name}");
|
|
||||||
|
|
||||||
accounts.write().unwrap().push(account);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user