wasmer/examples/http-server/src/main.rs

#![feature(wasi_ext)]

use kwasm_net::{Epoll, Tcp4Listener, TcpStream, schedule};
use std::sync::Arc;

fn serve(stream: Arc<TcpStream>, mut all: Vec<u8>) {
    let stream2 = stream.clone();
    stream.read_async(
        Vec::with_capacity(512),
        move |result| {
            match result {
                Ok(buf) => {
                    if buf.len() == 0 {
                        return;
                    }
                    all.extend_from_slice(&buf);
                    if all.len() > 4096 {
                        println!("header too large");
                        return;
                    }
                    let s = match ::std::str::from_utf8(&all) {
                        Ok(x) => x,
                        Err(e) => {
                            println!("not utf8: {:?}", e);
                            return;
                        }
                    };
                    if let Some(pos) = s.find("\r\n\r\n") {
                        stream2.write_all_async(
                            format!(
                                "HTTP/1.0 200 OK\r\nContent-Type: text/plain\r\n\r\nYour headers: \n\n{}\n",
                                ::std::str::from_utf8(&all[..pos]).unwrap()
                            ).into_bytes(),
                            |result| {}
                        );
                    } else {
                        schedule(|| serve(stream2, all));
                    }
                }
                Err(code) => {
                    println!("failed to read; code = {}", code);
                }
            }
        }
    );
}

fn main() {
    let epoll = Arc::new(Epoll::new());
    let listener = Arc::new(Tcp4Listener::new("0.0.0.0", 2011, 128).unwrap());

    listener.accept_async(epoll.clone(), |stream| {
        match stream {
            Ok(stream) => {
                serve(stream, vec![]);
                Ok(())
            },
            Err(code) => {
                println!("failed to accept; code = {}", code);
                Err(())
            }
        }
    });
    println!("start epoll");
    unsafe {
        epoll.run();
    }
}