Giving up on accepting trait arg

h8r/.gitignore

@@ -0,0 +1 @@
+/target/

h8r/Cargo.lock

@@ -0,0 +1,7 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+version = 4
+
+[[package]]
+name = "h8r"
+version = "0.1.0"

h8r/Cargo.toml

@@ -0,0 +1,6 @@
+[package]
+name = "h8r"
+version = "0.1.0"
+edition = "2024"
+
+[dependencies]

h8r/README.md

@@ -0,0 +1,5 @@
+# h8r gonna h8
+
+h + len(ttpserve) + r
+
+A li'l HTTP server for learning stuff.

h8r/TODO.md

@@ -0,0 +1,14 @@
+- [x] server on fixed port responding "oh no"
+- [x] configurable addr
+- [x] serve "index.txt" file if exists, else 404
+- [ ] serve relative paths with mime type text/plain
+- [ ] guess mime type
+- [ ] configurable working directory
+- [ ] string responses
+- [ ] automatic directory index
+- [ ] content ranges
+- [ ] error page paths
+- [ ] liquid template rendering
+- [ ] upstream proxy via prefix
+- [ ] upstream proxy path match
+- [ ] upstream proxy header middleware rules

h8r/src/main.rs

@@ -0,0 +1,75 @@
+use std::env;
+use std::fs;
+use std::io::{self, Write};
+use std::net;
+
+fn main() -> io::Result<()> {
+    let addr = env::var("H8R_ADDR").unwrap_or("127.0.0.1:17321".to_string());
+
+    run_server(addr)?;
+
+    Ok(())
+}
+
+fn run_server(addr: String) -> Result<(), io::Error> {
+    let listener = net::TcpListener::bind(&addr).unwrap();
+
+    eprintln!("h8r: listening at {}", addr);
+
+    for stream in listener.incoming() {
+        let mut stream = stream.unwrap();
+        handle_conn(&mut stream)?;
+        stream.shutdown(net::Shutdown::Both)?;
+    }
+
+    Ok(())
+}
+
+fn handle_conn(outstream: &mut net::TcpStream) -> io::Result<()> {
+    eprintln!("h8r: attempting to respond");
+
+    let relpath = "index.txt";
+
+    let index_res = fs::read_to_string(relpath);
+    if index_res.is_ok() {
+        let head = concat!["HTTP/1.1 200 OK\r\n", "content-type: text/plain\r\n"];
+        let index = index_res.unwrap();
+        let index_len = index.len();
+        outstream.write_all(head.as_bytes()).unwrap();
+        outstream
+            .write_all(format!("content-length: {}\r\n", index_len).as_bytes())
+            .unwrap();
+        outstream.write_all("\r\n".as_bytes()).unwrap();
+        outstream.write_all(index.as_bytes()).unwrap();
+    } else {
+        let response = concat![
+            "HTTP/1.1 404 Not Found\r\n",
+            "content-type: text/plain\r\n",
+            "\r\n",
+            "oh no\n",
+        ];
+
+        outstream.write_all(response.as_bytes()).unwrap();
+    }
+
+    outstream.flush()?;
+    Ok(())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn handle_conn_writes_response() {
+        let test_addr = "127.0.0.1:27321".to_string();
+
+        let _l = net::TcpListener::bind(&test_addr);
+        let mut stream = match net::TcpStream::connect(&("127.0.0.1", 27321)) {
+            Ok(s) => s,
+            Err(e) => panic!("cannot connect to test server: {}", e),
+        };
+
+        let _ = handle_conn(&mut stream);
+    }
+}

tcp-challenge/server/cmd/server/main.go

@@ -0,0 +1,121 @@
+package main
+
+import (
+	"bufio"
+	"context"
+	"errors"
+	"fmt"
+	"log"
+	"net"
+	"os"
+	"os/signal"
+	"strconv"
+	"strings"
+	"syscall"
+	"time"
+)
+
+func main() {
+	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer stop()
+
+	addr := ":7765"
+	if len(os.Args) > 1 {
+		addr = os.Args[1]
+	}
+
+	poolSize := 10
+	if len(os.Args) > 2 {
+		if v, err := strconv.ParseInt(os.Args[2], 10, 64); err != nil {
+			log.Fatal(err)
+		} else {
+			poolSize = int(v)
+		}
+	}
+
+	log.Printf("listening at %v (workers=%v)", addr, poolSize)
+
+	listener, err := net.Listen("tcp", addr)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	go func() {
+		<-ctx.Done()
+		if err := listener.Close(); err != nil {
+			log.Printf("failed to close listener: %v", err)
+		}
+	}()
+
+	workers := []*worker{}
+	requests := make(chan net.Conn)
+
+	for i := 0; i < poolSize; i++ {
+		w := &worker{id: i, r: requests}
+		go w.Start(ctx)
+		workers = append(workers, w)
+	}
+
+	for {
+		select {
+		case <-ctx.Done():
+			return
+		case <-time.After(time.Microsecond):
+			continue
+		default:
+			log.Printf("entering accept condition")
+
+			conn, err := listener.Accept()
+			if err != nil {
+				if errors.Is(err, net.ErrClosed) {
+					return
+				}
+
+				log.Printf("failed to accept connection: %v", err)
+				continue
+			}
+
+			conn.SetDeadline(time.Now().Add(5 * time.Second))
+
+			log.Printf("accepted connection %v", conn.RemoteAddr())
+			requests <- conn
+		}
+	}
+}
+
+type worker struct {
+	id int
+	r  chan net.Conn
+}
+
+func (w *worker) Start(ctx context.Context) {
+	log.Printf("[worker %v] starting loop", w.id)
+
+	for {
+		select {
+		case <-ctx.Done():
+			log.Printf("[worker %v] exiting", w.id)
+			return
+		case conn := <-w.r:
+			log.Printf("[worker %v] handling connection %v", w.id, conn.RemoteAddr())
+
+			fmt.Fprintf(conn, "ITS ECHO TIME\n")
+
+			scanner := bufio.NewScanner(conn)
+			for scanner.Scan() {
+				req := strings.TrimSpace(scanner.Text())
+
+				if req == "BYE" {
+					fmt.Fprintf(conn, "BYE BYE NOW\n")
+					break
+				}
+
+				fmt.Fprintf(conn, "YOU SAID: %v\n", req)
+			}
+
+			if err := conn.Close(); err != nil {
+				log.Printf("[worker %v] failed to echo: %v", w.id, err)
+			}
+		}
+	}
+}

tcp-challenge/server/go.mod

@@ -0,0 +1,3 @@
+module git.meatballhat.com/x/box-o-sand/tcp-challenge/server
+
+go 1.21.4