sad/g1r
1
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

fix threading in invade

Browse Source
This commit is contained in:
legitnull 2023-02-18 14:01:55 -07:00
parent 6acbbf1413
commit 9f1a92d1b3
2 changed files with 15 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main.rs
View File

@ -74,7 +74,7 @@ fn main() {
// RESPOND TO PINGS
if message.starts_with("PING") {
println!("[%] PONG");
println!("[%] PONG {}", config.nick);
ssl_stream.write_all("PONG ircd.chat\r\n".as_bytes()).unwrap();
continue; // skip processing the PING message further
}

34
src/modules/invade.rs
View File

@ -1,7 +1,7 @@
use crate::modules::Command;
use std::cell::RefCell;
use std::io::{BufRead, BufReader, Write};
use std::io::{Write};
use std::net::TcpStream;
use std::rc::Rc;
@ -9,8 +9,6 @@ use openssl::ssl::{SslConnector, SslMethod};
use serde::Deserialize;
use toml::Value;
#[derive(Clone, Deserialize)]
struct Config {
invaders: Vec<String>,
@ -39,34 +37,30 @@ impl Command for InvadeCommand {
let thread_invader = invader.to_string();
let config_clone = config.clone();
let screaming = scream.to_string();
std::thread::spawn(move || {
let stream = TcpStream::connect(format!("{}:{}", config_clone.server, config_clone.port)).unwrap();
let stream = TcpStream::connect((config_clone.server.as_str(), config_clone.port)).unwrap();
let connector = SslConnector::builder(SslMethod::tls()).unwrap().build();
let ssl_stream = connector.connect(&config_clone.server, stream).unwrap();
let ssl_stream = Rc::new(RefCell::new(ssl_stream));
let mut ssl_stream = connector.connect(config_clone.server.as_str(), stream).unwrap();
let nick_command = format!("NICK {}\r\n", thread_invader);
let user_command = format!("USER {} 0 * :{}\r\n", thread_invader, thread_invader);
ssl_stream.borrow_mut().write_all(nick_command.as_bytes()).unwrap();
ssl_stream.borrow_mut().write_all(user_command.as_bytes()).unwrap();
ssl_stream.write_all(nick_command.as_bytes()).unwrap();
ssl_stream.write_all(user_command.as_bytes()).unwrap();
let join_command = format!("JOIN {} \r\n", thread_channel);
ssl_stream.borrow_mut().write_all(join_command.as_bytes()).unwrap();
ssl_stream.write_all(join_command.as_bytes()).unwrap();
let msg = format!("PRIVMSG {} :{}\r\n", thread_channel, screaming);
ssl_stream.borrow_mut().write_all(msg.as_bytes()).unwrap();
ssl_stream.write_all(msg.as_bytes()).unwrap();
loop {
let mut ssl_stream_ref = ssl_stream.borrow_mut();
let mut reader = BufReader::new(&mut *ssl_stream_ref);
let mut message = String::new();
match reader.read_line(&mut message) {
let mut buffer = [0; 512];
match ssl_stream.ssl_read(&mut buffer) {
Ok(0) => break,
Ok(_) => {
Ok(n) => {
let message = String::from_utf8_lossy(&buffer[..n]);
if message.starts_with("PING") {
let response = message.replace("PING", "PONG");
ssl_stream.borrow_mut().write_all(response.as_bytes()).unwrap();
println!("[%] PONG {}", thread_invader);
ssl_stream.write_all(response.as_bytes()).unwrap();
}
}
Err(e) => {
@ -83,4 +77,4 @@ impl Command for InvadeCommand {
response
}
}
}