src/doc/book/listings/ch20-web-server/listing-20-07/src/main.rs

   1 use std::fs;
   2 use std::io::prelude::*;
   3 use std::net::TcpListener;
   4 use std::net::TcpStream;
   5
   6 fn main() {
   7     let listener = TcpListener::bind("127.0.0.1:7878").unwrap();
   8
   9     for stream in listener.incoming() {
  10         let stream = stream.unwrap();
  11
  12         handle_connection(stream);
  13     }
  14 }
  15
  16 fn handle_connection(mut stream: TcpStream) {
  17     let mut buffer = [0; 1024];
  18     stream.read(&mut buffer).unwrap();
  19
  20     let get = b"GET / HTTP/1.1\r\n";
  21
  22     if buffer.starts_with(get) {
  23         let contents = fs::read_to_string("hello.html").unwrap();
  24
  25         let response = format!(
  26             "HTTP/1.1 200 OK\r\nContent-Length: {}\r\n\r\n{}",
  27             contents.len(),
  28             contents
  29         );
  30
  31         stream.write(response.as_bytes()).unwrap();
  32         stream.flush().unwrap();
  33     // ANCHOR: here
  34     // --snip--
  35     } else {
  36         let status_line = "HTTP/1.1 404 NOT FOUND";
  37         let contents = fs::read_to_string("404.html").unwrap();
  38
  39         let response = format!(
  40             "{}\r\nContent-Length: {}\r\n\r\n{}",
  41             status_line,
  42             contents.len(),
  43             contents
  44         );
  45
  46         stream.write(response.as_bytes()).unwrap();
  47         stream.flush().unwrap();
  48     }
  49     // ANCHOR_END: here
  50 }