Spaces:
Runtime error
Runtime error
File size: 2,396 Bytes
9552aa0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 | pub fn construct_fastcgi_record(record_type: u8, request_id: u16, content: &[u8]) -> Vec<u8> {
let mut record = Vec::new();
// FastCGI version: FCGI_VERSION_1
record.push(1);
// Record type
record.extend_from_slice(&(record_type.to_be_bytes()));
// Request ID
record.extend_from_slice(&(request_id.to_be_bytes()));
// Content length
let content_length = content.len() as u16;
record.extend_from_slice(&(content_length.to_be_bytes()));
// Padding length
let content_length_modulo = (content_length % 8) as u8;
let padding_length = match content_length_modulo {
0 => 0,
_ => 8 - content_length_modulo,
};
record.extend_from_slice(&(padding_length.to_be_bytes()));
// Reserved
record.push(0);
// Content
record.extend_from_slice(content);
// Padding
record.append(&mut vec![0u8; padding_length as usize]);
record
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_construct_fastcgi_record() {
// Test case 1: Empty content
let record_type = 1;
let request_id = 1234;
let content: &[u8] = &[];
let expected_record = vec![
1, // FastCGI version
1, // Record type
4, 210, // Request ID
0, 0, // Content length
0, // Padding length
0, // Reserved
];
assert_eq!(
construct_fastcgi_record(record_type, request_id, content),
expected_record
);
// Test case 2: Content with length 5
let record_type = 2;
let request_id = 5678;
let content = b"Hello";
let expected_record = vec![
1, // FastCGI version
2, // Record type
22, 46, // Request ID
0, 5, // Content length
3, // Padding length
0, // Reserved
72, 101, 108, 108, 111, // Content
0, 0, 0, // Padding
];
assert_eq!(
construct_fastcgi_record(record_type, request_id, content),
expected_record
);
// Test case 3: Content with length 8 (no padding needed)
let record_type = 3;
let request_id = 9012;
let content = b"12345678";
let expected_record = vec![
1, // FastCGI version
3, // Record type
35, 52, // Request ID
0, 8, // Content length
0, // Padding length
0, // Reserved
49, 50, 51, 52, 53, 54, 55, 56, // Content
];
assert_eq!(
construct_fastcgi_record(record_type, request_id, content),
expected_record
);
}
}
|