From 943697af6520f727c107f38f8182daa2ff342e6e Mon Sep 17 00:00:00 2001 From: Noah Swerhun Date: Mon, 3 Jul 2023 16:44:57 -0500 Subject: [PATCH] initial commit --- .gitignore | 1 + Cargo.lock | 7 ++++ Cargo.toml | 14 +++++++ README.md | 12 +++++- html/404.html | 11 ++++++ html/index.html | 11 ++++++ html/slow.html | 11 ++++++ src/lib.rs | 103 ++++++++++++++++++++++++++++++++++++++++++++++++ src/main.rs | 62 +++++++++++++++++++++++++++++ 9 files changed, 231 insertions(+), 1 deletion(-) create mode 100644 .gitignore create mode 100644 Cargo.lock create mode 100644 Cargo.toml create mode 100644 html/404.html create mode 100644 html/index.html create mode 100644 html/slow.html create mode 100644 src/lib.rs create mode 100644 src/main.rs diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ea8c4bf --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +/target diff --git a/Cargo.lock b/Cargo.lock new file mode 100644 index 0000000..5ae25eb --- /dev/null +++ b/Cargo.lock @@ -0,0 +1,7 @@ +# This file is automatically @generated by Cargo. +# It is not intended for manual editing. +version = 3 + +[[package]] +name = "rust-webserver" +version = "0.1.0" diff --git a/Cargo.toml b/Cargo.toml new file mode 100644 index 0000000..cbd6b9d --- /dev/null +++ b/Cargo.toml @@ -0,0 +1,14 @@ +[package] +name = "rust-webserver" +version = "0.1.0" +edition = "2021" + +# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html + +[dependencies] + +[profile.release] +lto = true +strip = true +panic = "abort" +codegen-units = 1 diff --git a/README.md b/README.md index 822ec69..4585bf1 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,13 @@ # rust-webserver -Very basic multithreaded TCP server in Rust \ No newline at end of file +Very basic multithreaded TCP server in Rust. + +See `src/main.rs`. Server will listen for a certain number of connections on a +given TCP port, and then gracefully exit. + +This project is taken from The Rust Book. + +**Run:** +``` +cargo run --release +``` diff --git a/html/404.html b/html/404.html new file mode 100644 index 0000000..f0a0add --- /dev/null +++ b/html/404.html @@ -0,0 +1,11 @@ + + + + + Error! + + +

Oops!

+

Sorry, I don't know what you're asking for.

+ + diff --git a/html/index.html b/html/index.html new file mode 100644 index 0000000..801f275 --- /dev/null +++ b/html/index.html @@ -0,0 +1,11 @@ + + + + + Welcome! + + +

Hello!

+

Hi from Rust

+ + diff --git a/html/slow.html b/html/slow.html new file mode 100644 index 0000000..fa20a91 --- /dev/null +++ b/html/slow.html @@ -0,0 +1,11 @@ + + + + + That took a while + + +

Hi

+

You just waited 5secs to see this page.

+ + diff --git a/src/lib.rs b/src/lib.rs new file mode 100644 index 0000000..459f740 --- /dev/null +++ b/src/lib.rs @@ -0,0 +1,103 @@ +use std::{ + fmt::Display, + thread, + error::Error, + sync::{ + mpsc, + Arc, + Mutex, + }, +}; + +pub struct ThreadPool { + workers: Vec, + sender: Option>, +} + +type Job = Box; + +struct Worker { + id: usize, + thread: Option>, +} + +impl Worker { + fn new(id: usize, receiver: Arc>>) -> Worker { + Worker { id, thread: Some(thread::spawn(move || { + loop { + match (*receiver).lock().unwrap().recv() { + Ok(job) => { + println!("Worker {} got a job... running.", id); + job(); + }, + Err(_) => { + println!("Worker {} stopping.", id); + break; + }, + } + } + }))} + } +} + +#[derive(Debug)] +pub enum ThreadPoolError { + ZeroSize +} + +impl Error for ThreadPoolError { + fn description(&self) -> &str { + match *self { + ThreadPoolError::ZeroSize => "ZeroPoolSize", + } + } +} + +impl Display for ThreadPoolError { + fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { + match *self { + ThreadPoolError::ZeroSize => + f.write_str("Thread pool size is zero"), + } + } +} + +impl ThreadPool { + pub fn new(size: usize) -> Result { + if size == 0 { + return Err(ThreadPoolError::ZeroSize); + } + + let mut workers: Vec = Vec::with_capacity(size); + let (sender, receiver) = mpsc::channel(); + + let receiver = Arc::new(Mutex::new(receiver)); + + for i in 0..size { + workers.push(Worker::new(i, Arc::clone(&receiver))); + } + + Ok(ThreadPool { workers, sender: Some(sender) }) + } + + pub fn execute(&self, f: F) + where + F: FnOnce() + Send + 'static + { + let job = Box::new(f); + (*self).sender.as_ref().unwrap().send(job).unwrap(); + } +} + +impl Drop for ThreadPool { + fn drop(&mut self) { + drop((*self).sender.take()); + + for worker in &mut (*self).workers { + println!("Shutting down worker {}.", worker.id); + if let Some(thread) = worker.thread.take() { + thread.join().unwrap(); + } + } + } +} diff --git a/src/main.rs b/src/main.rs new file mode 100644 index 0000000..ed5a5e4 --- /dev/null +++ b/src/main.rs @@ -0,0 +1,62 @@ +use std::{ + net::{TcpListener, TcpStream}, + io::{prelude::*, BufReader}, + fs, + thread, + time::Duration, +}; + +use rust_webserver::ThreadPool; + +fn parse_request(stream: &TcpStream) -> Vec{ + let buf_reader = BufReader::new(stream); + + buf_reader + .lines() + .map(|result| result.unwrap()) + .take_while(|line| !line.is_empty()) + .collect() +} + +fn generate_reposne(request: &Vec) -> String { + let request_header = (*(request.first().unwrap())).clone(); + + let (status, file) = + match &request_header[..] { + "GET / HTTP/1.1" => ("HTTP/1.1 200 OK", "html/index.html"), + "GET /sleep HTTP/1.1" => { + thread::sleep(Duration::from_secs(5)); + ("HTTP/1.1 200 OK", "html/slow.html") + }, + _ => ("HTTP/1.1 404 NOT FOUND", "html/404.html") + }; + + let contents = fs::read_to_string(file).unwrap(); + let len = contents.len(); + format!("{status}\r\nContent-Length: {len}\r\n\r\n{contents}") +} + +fn handle_connection(mut stream: TcpStream) { + let request = parse_request(&stream); + let response = generate_reposne(&request); + + stream.write_all(response.as_bytes()).unwrap(); +} + +fn main() { + let port = 7878; + let connections = 3; + let listener = TcpListener::bind(format!("127.0.0.1:{}", port)).unwrap(); + + let pool = ThreadPool::new(10).unwrap(); + + println!("Listening for {} connnections on port {}", connections, port); + for stream in listener.incoming().take(connections) { + let stream = stream.unwrap(); + + pool.execute(|| { + handle_connection(stream); + }); + + } +}