サンプルコードで身につける: リクエストの形 — 行末は CRLF、Host は必須、Connection: close で終わりが分かる

fn main() {
    // HTTP/1.1 のリクエストは「行」の集まり。行末は CRLF("\r\n")で、空行 1 つがヘッダーの終わり
    let request = "GET /index.html HTTP/1.1\r\nHost: example.com\r\nConnection: close\r\n\r\n";

    println!("--- 送るバイト列({} バイト)。\\r\\n を見えるように表示 ---", request.len());
    println!("{}", request.escape_debug());

    println!("\n--- 行ごとに見る ---");
    for (i, line) in request.split_terminator("\r\n").enumerate() {
        let what = if i == 0 {
            "request-line = method SP request-target SP HTTP-version"
        } else if line.is_empty() {
            "空行 = ヘッダーの終わり。ここから先が本文(GET では無し)"
        } else {
            "field-line = name \":\" value"
        };
        println!("{i}: {line:<26} <- {what}");
    }
}

const HTTP_DEFAULT_PORT: u16 = 80;

struct Url {
    host: String,
    port: u16,
    path: String,
    query: Option<String>,
}

fn parse(input: &str) -> Option<Url> {
    let (_, rest) = input.split_once("://")?;
    let authority_end = rest.find(['/', '?']).unwrap_or(rest.len());
    let (authority, tail) = (&rest[..authority_end], &rest[authority_end..]);
    let (host, port) = match authority.rsplit_once(':') {
        Some((h, p)) => (h, p.parse().ok()?),
        None => (authority, HTTP_DEFAULT_PORT),
    };
    let (path, query) = match tail.split_once('?') {
        Some((p, q)) => (p, Some(q.to_string())),
        None => (tail, None),
    };
    Some(Url {
        host: host.to_string(),
        port,
        path: if path.is_empty() { "/".to_string() } else { path.to_string() },
        query,
    })
}

impl Url {
    fn request_target(&self) -> String {
        match &self.query {
            Some(q) => format!("{}?{}", self.path, q),
            None => self.path.clone(),
        }
    }
    fn host_header(&self) -> String {
        if self.port == HTTP_DEFAULT_PORT {
            self.host.clone()
        } else {
            format!("{}:{}", self.host, self.port)
        }
    }
}

/// 送信するリクエストの文字列を組み立てる。
/// Host は HTTP/1.1 で必須(RFC 9112 3.2)。Connection: close は「この 1 往復で接続を閉じる」宣言で、
/// これがあると「サーバーが閉じるまで読む」だけで本文の終わりが分かる
fn build_request(url: &Url) -> String {
    format!(
        "GET {target} HTTP/1.1\r\n\
         Host: {host}\r\n\
         User-Agent: tinybrowser/0.1\r\n\
         Accept: text/html\r\n\
         Connection: close\r\n\
         \r\n",
        target = url.request_target(),
        host = url.host_header(),
    )
}

fn main() {
    for input in ["http://example.com/index.html?a=1", "http://localhost:8000/"].iter() {
        let url = parse(input).expect("URL が読めない");
        let req = build_request(&url);
        println!("{input}");
        println!("  {:?}", req); // {:?} なら \r\n が見える
        println!("  request-line は {:?}", req.lines().next().unwrap());
        println!("  末尾が空行で終わる: {}\n", req.ends_with("\r\n\r\n"));
    }
}

/// 受け取ったリクエストのヘッダー部を検査する。
/// モックサーバーや疎通確認ツールを書くとき、サーバー側はこの順番で判断している
fn validate(raw: &str) -> Result<String, String> {
    // 1. ヘッダー部の終わりは空行(CRLF CRLF)。見つからなければ、そもそも区切りが分からない
    let head = match raw.find("\r\n\r\n") {
        Some(i) => &raw[..i],
        None => return Err("空行(CRLF CRLF)が無く、ヘッダーの終わりが分からない".to_string()),
    };
    let mut lines = head.split("\r\n");

    // 2. request-line = method SP request-target SP HTTP-version の 3 要素
    let request_line = lines.next().unwrap_or("");
    let parts: Vec<&str> = request_line.split(' ').collect();
    if parts.len() != 3 || !parts[2].starts_with("HTTP/") {
        return Err(format!("request-line が 3 要素でない: {request_line:?}"));
    }
    if parts[0] != "GET" {
        return Err(format!("対応していないメソッド: {}", parts[0]));
    }

    // 3. field-line を読む。HTTP/1.1 では Host が必須(RFC 9112 3.2)
    let mut host = None;
    for line in lines {
        let (name, value) = line
            .split_once(':')
            .ok_or_else(|| format!("field-line に ':' が無い: {line:?}"))?;
        if name.eq_ignore_ascii_case("host") {
            host = Some(value.trim().to_string());
        }
    }
    let host = host.ok_or("Host ヘッダーが無い(HTTP/1.1 では必須。400 Bad Request を返す)")?;
    Ok(format!("{} {} を {} に", parts[0], parts[1], host))
}

fn main() {
    let samples = [
        ("正しい", "GET /index.html HTTP/1.1\r\nHost: example.com\r\nConnection: close\r\n\r\n"),
        ("Host が無い", "GET / HTTP/1.1\r\nConnection: close\r\n\r\n"),
        ("行末が LF だけ", "GET / HTTP/1.1\nHost: example.com\n\n"),
        ("request-line が壊れている", "GET /index.html\r\nHost: example.com\r\n\r\n"),
        ("POST", "POST /api HTTP/1.1\r\nHost: example.com\r\n\r\n"),
    ];
    for (label, raw) in samples.iter() {
        match validate(raw) {
            Ok(summary) => println!("OK  [{label}] {summary}"),
            Err(why) => println!("NG  [{label}] {why}"),
        }
    }
}