如何读取 FastCGI 请求的所有输出？

我正在尝试使用 Rust 向 PHP 文件发出请求FastCGI协议。感谢以下答案，我得到了这段代码我之前的问题:

use std::os::unix::net::{UnixStream};
use std::io::{Read, Write};
use std::str;

fn main() {
    const FCGI_VERSION_1: u8    = 1;

    const FCGI_BEGIN_REQUEST:u8 = 1;
    const FCGI_END_REQUEST: u8  = 3;
    const FCGI_STDIN: u8        = 5;
    const FCGI_STDOUT: u8       = 6;

    const FCGI_RESPONDER: u16  = 1;

    const FCGI_PARAMS: u8 = 4;

    const FCGI_GET_VALUES: u8 = 9;

    let socket_path = "/run/php-fpm/php-fpm.sock";

    let mut socket = match UnixStream::connect(socket_path) {
        Ok(sock) => sock,
        Err(e) => {
            println!("Couldn't connect: {e:?}");
            return
        }
    };

    let requestId: u16 = 1;
    let role: u16 = FCGI_RESPONDER;
    let beginRequest = vec![
       // FCGI_Header
       FCGI_VERSION_1, FCGI_BEGIN_REQUEST,
       (requestId >> 8) as u8, (requestId & 0xFF) as u8,
       0x00, 0x08, // This is the size of `FCGI_BeginRequestBody`
       0, 0,
       // FCGI_BeginRequestBody
       (role >> 8) as u8, (role & 0xFF) as u8,
       0, // Flags
       0, 0, 0, 0, 0, // Reserved
    ];
    socket.write_all(&beginRequest).unwrap();

    let param_name = "SCRIPT_FILENAME".as_bytes();
    let param_value = "/var/www/html/index.php".as_bytes();
    let lengths = [ param_name.len() as u8, param_value.len() as u8 ];
    socket.write_all (&lengths).unwrap();
    socket.write_all (param_name).unwrap();
    socket.write_all (param_value).unwrap();

    let requestHeader = vec![
       FCGI_VERSION_1, FCGI_STDIN,
       (requestId >> 8) as u8, (requestId & 0xFF) as u8,
       0, 0,
       0, 0,
    ];
    socket.write_all(&requestHeader).unwrap();

    let mut responseHeader = [0u8; 8];
    socket.read_exact (&mut responseHeader).unwrap();
    assert!(responseHeader[1] == FCGI_STDOUT);  // TODO: proper handling of message type
    let responseLength = ((responseHeader[4] as usize) << 8) | (responseHeader[5] as usize);
    let mut responseBody = Vec::new();
    responseBody.resize (responseLength, 0);
    socket.read_exact (&mut responseBody).unwrap();
    println!("{:?}", std::str::from_utf8(responseBody));
}

该代码启动一个请求，然后写入SCRIPT_FILENAME参数，然后读取 PHP 文件的输出。这是 PHP 文件：

<?php

echo "First file";

?>

当我运行 Rust 文件时，控制台显示如下：

Ok("X-Powered-By: PHP/8.1.11\r\nContent-type: text/html; charset=UTF-8\r\n\r\n")

控制台没有向我显示程序的输出，所以我想问题可能是什么Jmb在他给我的回答中说:

响应可能会分为多个 FCGI_STDOUT 消息，因此您需要一个循环来重新组装它们。

现在我看到了FastCGI协议规范示例3我意识到在该示例中程序的输出仍然是分开的（或者我认为是这样）。那么我怎样才能读取程序的所有输出呢？我知道 Jmb 说我需要一个循环来做到这一点，那么我应该如何实现循环呢？或者在哪里？我想澄清一下，我对 FastCGI 还很陌生，而且对二进制协议没有任何经验，所以如果我遗漏了一些看起来很明显的东西，我真的很抱歉。

您的代码存在一些问题。

Issue #1

最后一个字节FCGI_Record被命名padding。它表示在该有效负载之后应跳过多少字节。您将其设置为 0，因此那里并不重要，但是从服务器读取时，您必须跳过这么多字节：

    let mut pad = vec![0; responseHeader[7] as usize];
    socket.read_exact (&mut pad).unwrap();

这解释了如果您尝试执行读取循环，您将得到垃圾而不是FCGI_END_REQUEST frame.

Issue #2

The params部分需要标头，但您正在发送原始数据。 您必须计算序列化参数的完整长度并在此之前发送标头：

    let params_len: u16 = (param_name.len() + param_value.len() + lengths.len()) as u16;
    let paramsRequest = vec![
       FCGI_VERSION_1, FCGI_PARAMS,
       (requestId >> 8) as u8, (requestId & 0xFF) as u8,
       (params_len >> 8) as u8, (params_len & 0xFF) as u8,
       0, 0,
    ];
    socket.write_all (&paramsRequest).unwrap();

Issue #3

我没有在任何地方找到所需的最小参数php-fpm, but 浏览源代码，你至少需要添加REQUEST_METHOD:

    let param1_name = "SCRIPT_FILENAME".as_bytes();
    let param1_value = "/var/www/html/index.php".as_bytes();
    let lengths1 = [ param1_name.len() as u8, param1_value.len() as u8 ];
    let params1_len: u16 = (param1_name.len() + param1_value.len() + lengths1.len()) as u16;

    let param2_name = b"REQUEST_METHOD";
    let param2_value = b"GET";
    let lengths2 = [ param2_name.len() as u8, param2_value.len() as u8 ];
    let params2_len: u16 = (param2_name.len() + param2_value.len() + lengths2.len()) as u16;

    let params_len = params1_len + params2_len;
    let paramsRequest = vec![
       FCGI_VERSION_1, FCGI_PARAMS,
       (requestId >> 8) as u8, (requestId & 0xFF) as u8,
       (params_len >> 8) as u8, (params_len & 0xFF) as u8,
       0, 0,
    ];
    socket.write_all (&paramsRequest).unwrap();
    socket.write_all (&lengths1).unwrap();
    socket.write_all (param1_name).unwrap();
    socket.write_all (param1_value).unwrap();
    socket.write_all (&lengths2).unwrap();
    socket.write_all (param2_name).unwrap();
    socket.write_all (param2_value).unwrap();

在这一点上，我认为您应该考虑为这些东西编写正确的类型和序列化原语，而不是原始字节数组......

通过这些更改，我们已经获得了完整的输出：

Ok("X-Powered-By: PHP/8.1.11\r\nContent-type: text/html; charset=UTF-8\r\n\r\nFirst file")

Issue #4

您应该循环读取响应代码，连接任何FCGI_STDOUT数据，记录或忽略其他消息，可能记录FCGI_STDERR消息，直到您收到FCGI_END_REQUEST.

在这个特定的示例中，输出非常短，以至于它适合一条消息，但是如果不执行循环，您就看不到FCGI_END_REQUEST.

之后FCGI_END_REQUEST，取决于您是否使用过FCGI_KEEP_CONN您可以关闭套接字或保持其打开状态以发送新请求。