spikard 0.4.0-arm64-darwin-23
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +7 -0
- data/LICENSE +1 -0
- data/README.md +659 -0
- data/ext/spikard_rb/Cargo.toml +17 -0
- data/ext/spikard_rb/extconf.rb +10 -0
- data/ext/spikard_rb/src/lib.rs +6 -0
- data/lib/spikard/app.rb +405 -0
- data/lib/spikard/background.rb +27 -0
- data/lib/spikard/config.rb +396 -0
- data/lib/spikard/converters.rb +13 -0
- data/lib/spikard/handler_wrapper.rb +113 -0
- data/lib/spikard/provide.rb +214 -0
- data/lib/spikard/response.rb +173 -0
- data/lib/spikard/schema.rb +243 -0
- data/lib/spikard/sse.rb +111 -0
- data/lib/spikard/streaming_response.rb +44 -0
- data/lib/spikard/testing.rb +221 -0
- data/lib/spikard/upload_file.rb +131 -0
- data/lib/spikard/version.rb +5 -0
- data/lib/spikard/websocket.rb +59 -0
- data/lib/spikard.rb +43 -0
- data/sig/spikard.rbs +366 -0
- data/vendor/bundle/ruby/3.4.0/gems/diff-lcs-1.6.2/mise.toml +5 -0
- data/vendor/bundle/ruby/3.4.0/gems/rake-compiler-dock-1.10.0/build/buildkitd.toml +2 -0
- data/vendor/crates/spikard-bindings-shared/Cargo.toml +63 -0
- data/vendor/crates/spikard-bindings-shared/examples/config_extraction.rs +139 -0
- data/vendor/crates/spikard-bindings-shared/src/config_extractor.rs +561 -0
- data/vendor/crates/spikard-bindings-shared/src/conversion_traits.rs +194 -0
- data/vendor/crates/spikard-bindings-shared/src/di_traits.rs +246 -0
- data/vendor/crates/spikard-bindings-shared/src/error_response.rs +403 -0
- data/vendor/crates/spikard-bindings-shared/src/handler_base.rs +274 -0
- data/vendor/crates/spikard-bindings-shared/src/lib.rs +25 -0
- data/vendor/crates/spikard-bindings-shared/src/lifecycle_base.rs +298 -0
- data/vendor/crates/spikard-bindings-shared/src/lifecycle_executor.rs +637 -0
- data/vendor/crates/spikard-bindings-shared/src/response_builder.rs +309 -0
- data/vendor/crates/spikard-bindings-shared/src/test_client_base.rs +248 -0
- data/vendor/crates/spikard-bindings-shared/src/validation_helpers.rs +355 -0
- data/vendor/crates/spikard-bindings-shared/tests/comprehensive_coverage.rs +502 -0
- data/vendor/crates/spikard-bindings-shared/tests/error_response_edge_cases.rs +389 -0
- data/vendor/crates/spikard-bindings-shared/tests/handler_base_integration.rs +413 -0
- data/vendor/crates/spikard-core/Cargo.toml +40 -0
- data/vendor/crates/spikard-core/src/bindings/mod.rs +3 -0
- data/vendor/crates/spikard-core/src/bindings/response.rs +133 -0
- data/vendor/crates/spikard-core/src/debug.rs +63 -0
- data/vendor/crates/spikard-core/src/di/container.rs +726 -0
- data/vendor/crates/spikard-core/src/di/dependency.rs +273 -0
- data/vendor/crates/spikard-core/src/di/error.rs +118 -0
- data/vendor/crates/spikard-core/src/di/factory.rs +538 -0
- data/vendor/crates/spikard-core/src/di/graph.rs +545 -0
- data/vendor/crates/spikard-core/src/di/mod.rs +192 -0
- data/vendor/crates/spikard-core/src/di/resolved.rs +411 -0
- data/vendor/crates/spikard-core/src/di/value.rs +283 -0
- data/vendor/crates/spikard-core/src/errors.rs +39 -0
- data/vendor/crates/spikard-core/src/http.rs +153 -0
- data/vendor/crates/spikard-core/src/lib.rs +29 -0
- data/vendor/crates/spikard-core/src/lifecycle.rs +422 -0
- data/vendor/crates/spikard-core/src/metadata.rs +397 -0
- data/vendor/crates/spikard-core/src/parameters.rs +723 -0
- data/vendor/crates/spikard-core/src/problem.rs +310 -0
- data/vendor/crates/spikard-core/src/request_data.rs +189 -0
- data/vendor/crates/spikard-core/src/router.rs +249 -0
- data/vendor/crates/spikard-core/src/schema_registry.rs +183 -0
- data/vendor/crates/spikard-core/src/type_hints.rs +304 -0
- data/vendor/crates/spikard-core/src/validation/error_mapper.rs +689 -0
- data/vendor/crates/spikard-core/src/validation/mod.rs +459 -0
- data/vendor/crates/spikard-http/Cargo.toml +58 -0
- data/vendor/crates/spikard-http/examples/sse-notifications.rs +147 -0
- data/vendor/crates/spikard-http/examples/websocket-chat.rs +91 -0
- data/vendor/crates/spikard-http/src/auth.rs +247 -0
- data/vendor/crates/spikard-http/src/background.rs +1562 -0
- data/vendor/crates/spikard-http/src/bindings/mod.rs +3 -0
- data/vendor/crates/spikard-http/src/bindings/response.rs +1 -0
- data/vendor/crates/spikard-http/src/body_metadata.rs +8 -0
- data/vendor/crates/spikard-http/src/cors.rs +490 -0
- data/vendor/crates/spikard-http/src/debug.rs +63 -0
- data/vendor/crates/spikard-http/src/di_handler.rs +1878 -0
- data/vendor/crates/spikard-http/src/handler_response.rs +532 -0
- data/vendor/crates/spikard-http/src/handler_trait.rs +861 -0
- data/vendor/crates/spikard-http/src/handler_trait_tests.rs +284 -0
- data/vendor/crates/spikard-http/src/lib.rs +524 -0
- data/vendor/crates/spikard-http/src/lifecycle/adapter.rs +149 -0
- data/vendor/crates/spikard-http/src/lifecycle.rs +428 -0
- data/vendor/crates/spikard-http/src/middleware/mod.rs +285 -0
- data/vendor/crates/spikard-http/src/middleware/multipart.rs +930 -0
- data/vendor/crates/spikard-http/src/middleware/urlencoded.rs +541 -0
- data/vendor/crates/spikard-http/src/middleware/validation.rs +287 -0
- data/vendor/crates/spikard-http/src/openapi/mod.rs +309 -0
- data/vendor/crates/spikard-http/src/openapi/parameter_extraction.rs +535 -0
- data/vendor/crates/spikard-http/src/openapi/schema_conversion.rs +867 -0
- data/vendor/crates/spikard-http/src/openapi/spec_generation.rs +678 -0
- data/vendor/crates/spikard-http/src/query_parser.rs +369 -0
- data/vendor/crates/spikard-http/src/response.rs +399 -0
- data/vendor/crates/spikard-http/src/server/handler.rs +1557 -0
- data/vendor/crates/spikard-http/src/server/lifecycle_execution.rs +98 -0
- data/vendor/crates/spikard-http/src/server/mod.rs +806 -0
- data/vendor/crates/spikard-http/src/server/request_extraction.rs +630 -0
- data/vendor/crates/spikard-http/src/server/routing_factory.rs +497 -0
- data/vendor/crates/spikard-http/src/sse.rs +961 -0
- data/vendor/crates/spikard-http/src/testing/form.rs +14 -0
- data/vendor/crates/spikard-http/src/testing/multipart.rs +60 -0
- data/vendor/crates/spikard-http/src/testing/test_client.rs +285 -0
- data/vendor/crates/spikard-http/src/testing.rs +377 -0
- data/vendor/crates/spikard-http/src/websocket.rs +831 -0
- data/vendor/crates/spikard-http/tests/background_behavior.rs +918 -0
- data/vendor/crates/spikard-http/tests/common/handlers.rs +308 -0
- data/vendor/crates/spikard-http/tests/common/mod.rs +21 -0
- data/vendor/crates/spikard-http/tests/di_integration.rs +202 -0
- data/vendor/crates/spikard-http/tests/doc_snippets.rs +4 -0
- data/vendor/crates/spikard-http/tests/lifecycle_execution.rs +1135 -0
- data/vendor/crates/spikard-http/tests/multipart_behavior.rs +688 -0
- data/vendor/crates/spikard-http/tests/server_config_builder.rs +324 -0
- data/vendor/crates/spikard-http/tests/sse_behavior.rs +728 -0
- data/vendor/crates/spikard-http/tests/websocket_behavior.rs +724 -0
- data/vendor/crates/spikard-rb/Cargo.toml +43 -0
- data/vendor/crates/spikard-rb/build.rs +199 -0
- data/vendor/crates/spikard-rb/src/background.rs +63 -0
- data/vendor/crates/spikard-rb/src/config/mod.rs +5 -0
- data/vendor/crates/spikard-rb/src/config/server_config.rs +283 -0
- data/vendor/crates/spikard-rb/src/conversion.rs +459 -0
- data/vendor/crates/spikard-rb/src/di/builder.rs +105 -0
- data/vendor/crates/spikard-rb/src/di/mod.rs +413 -0
- data/vendor/crates/spikard-rb/src/handler.rs +612 -0
- data/vendor/crates/spikard-rb/src/integration/mod.rs +3 -0
- data/vendor/crates/spikard-rb/src/lib.rs +1857 -0
- data/vendor/crates/spikard-rb/src/lifecycle.rs +275 -0
- data/vendor/crates/spikard-rb/src/metadata/mod.rs +5 -0
- data/vendor/crates/spikard-rb/src/metadata/route_extraction.rs +427 -0
- data/vendor/crates/spikard-rb/src/runtime/mod.rs +5 -0
- data/vendor/crates/spikard-rb/src/runtime/server_runner.rs +326 -0
- data/vendor/crates/spikard-rb/src/server.rs +283 -0
- data/vendor/crates/spikard-rb/src/sse.rs +231 -0
- data/vendor/crates/spikard-rb/src/testing/client.rs +404 -0
- data/vendor/crates/spikard-rb/src/testing/mod.rs +7 -0
- data/vendor/crates/spikard-rb/src/testing/sse.rs +143 -0
- data/vendor/crates/spikard-rb/src/testing/websocket.rs +221 -0
- data/vendor/crates/spikard-rb/src/websocket.rs +233 -0
- data/vendor/crates/spikard-rb/tests/magnus_ffi_tests.rs +14 -0
- metadata +213 -0
|
@@ -0,0 +1,930 @@
|
|
|
1
|
+
//! Multipart form-data parsing
|
|
2
|
+
|
|
3
|
+
use axum::extract::Multipart;
|
|
4
|
+
use serde_json::json;
|
|
5
|
+
|
|
6
|
+
/// Size threshold for streaming vs buffering multipart fields
|
|
7
|
+
/// Fields larger than this will be streamed chunk-by-chunk
|
|
8
|
+
const MULTIPART_STREAMING_THRESHOLD: usize = 1024 * 1024;
|
|
9
|
+
|
|
10
|
+
/// Parse multipart/form-data to JSON
|
|
11
|
+
///
|
|
12
|
+
/// This handles:
|
|
13
|
+
/// - File uploads → {"filename": "...", "size": N, "content": "...", "content_type": "..."}
|
|
14
|
+
/// - Form fields → plain string values
|
|
15
|
+
/// - Mixed files and data → combined in single JSON object
|
|
16
|
+
/// - Large files → streamed chunk-by-chunk (async)
|
|
17
|
+
/// - Small files → buffered in memory
|
|
18
|
+
/// - Multiple values with same field name → aggregated into arrays
|
|
19
|
+
///
|
|
20
|
+
/// Streaming strategy:
|
|
21
|
+
/// - Files > 1MB: Use field.chunk().await for async streaming
|
|
22
|
+
/// - Files <= 1MB: Use field.bytes().await for buffered loading
|
|
23
|
+
pub async fn parse_multipart_to_json(
|
|
24
|
+
mut multipart: Multipart,
|
|
25
|
+
) -> Result<serde_json::Value, Box<dyn std::error::Error + Send + Sync>> {
|
|
26
|
+
use rustc_hash::FxHashMap;
|
|
27
|
+
|
|
28
|
+
let mut field_values: FxHashMap<String, Vec<serde_json::Value>> = FxHashMap::default();
|
|
29
|
+
|
|
30
|
+
while let Some(field) = multipart.next_field().await? {
|
|
31
|
+
let name = field.name().ok_or("Field missing name")?.to_string();
|
|
32
|
+
|
|
33
|
+
let field_value = if let Some(filename) = field.file_name() {
|
|
34
|
+
let filename = filename.to_string();
|
|
35
|
+
let content_type = field
|
|
36
|
+
.content_type()
|
|
37
|
+
.map(|ct| ct.to_string())
|
|
38
|
+
.unwrap_or_else(|| "application/octet-stream".to_string());
|
|
39
|
+
|
|
40
|
+
let bytes = field.bytes().await?;
|
|
41
|
+
let size = bytes.len();
|
|
42
|
+
|
|
43
|
+
let is_text_like = content_type.starts_with("text/") || content_type == "application/json";
|
|
44
|
+
let content = if is_text_like || size <= MULTIPART_STREAMING_THRESHOLD {
|
|
45
|
+
String::from_utf8_lossy(&bytes).to_string()
|
|
46
|
+
} else {
|
|
47
|
+
format!("<binary data, {} bytes>", size)
|
|
48
|
+
};
|
|
49
|
+
|
|
50
|
+
json!({
|
|
51
|
+
"filename": filename,
|
|
52
|
+
"size": size,
|
|
53
|
+
"content": content,
|
|
54
|
+
"content_type": content_type
|
|
55
|
+
})
|
|
56
|
+
} else {
|
|
57
|
+
let value = field.text().await?;
|
|
58
|
+
|
|
59
|
+
if (value.starts_with('[') && value.ends_with(']')) || (value.starts_with('{') && value.ends_with('}')) {
|
|
60
|
+
if let Ok(parsed_json) = serde_json::from_str::<serde_json::Value>(&value) {
|
|
61
|
+
parsed_json
|
|
62
|
+
} else {
|
|
63
|
+
json!(value)
|
|
64
|
+
}
|
|
65
|
+
} else {
|
|
66
|
+
json!(value)
|
|
67
|
+
}
|
|
68
|
+
};
|
|
69
|
+
|
|
70
|
+
field_values.entry(name).or_default().push(field_value);
|
|
71
|
+
}
|
|
72
|
+
|
|
73
|
+
let result: serde_json::Map<String, serde_json::Value> = field_values
|
|
74
|
+
.into_iter()
|
|
75
|
+
.map(|(key, mut values)| {
|
|
76
|
+
if values.len() == 1 {
|
|
77
|
+
let val = values.pop().unwrap_or(serde_json::Value::Null);
|
|
78
|
+
(key, val)
|
|
79
|
+
} else {
|
|
80
|
+
(key, serde_json::Value::Array(values))
|
|
81
|
+
}
|
|
82
|
+
})
|
|
83
|
+
.collect();
|
|
84
|
+
|
|
85
|
+
Ok(json!(result))
|
|
86
|
+
}
|
|
87
|
+
|
|
88
|
+
#[cfg(test)]
|
|
89
|
+
mod tests {
|
|
90
|
+
use super::*;
|
|
91
|
+
use axum::extract::FromRequest;
|
|
92
|
+
use axum::extract::Multipart;
|
|
93
|
+
|
|
94
|
+
/// Helper function to create a multipart boundary string
|
|
95
|
+
fn create_multipart_body(boundary: &str, parts: Vec<String>) -> Vec<u8> {
|
|
96
|
+
let mut body = Vec::new();
|
|
97
|
+
for part in parts {
|
|
98
|
+
body.extend_from_slice(format!("--{}\r\n", boundary).as_bytes());
|
|
99
|
+
body.extend_from_slice(part.as_bytes());
|
|
100
|
+
body.extend_from_slice(b"\r\n");
|
|
101
|
+
}
|
|
102
|
+
body.extend_from_slice(format!("--{}--\r\n", boundary).as_bytes());
|
|
103
|
+
body
|
|
104
|
+
}
|
|
105
|
+
|
|
106
|
+
#[tokio::test]
|
|
107
|
+
async fn test_single_file_upload() {
|
|
108
|
+
// Create a mock multipart request with a single file
|
|
109
|
+
let boundary = "boundary123";
|
|
110
|
+
let file_content = "Hello, World!";
|
|
111
|
+
let parts = vec![
|
|
112
|
+
"Content-Disposition: form-data; name=\"file\"; filename=\"test.txt\"\r\nContent-Type: text/plain\r\n\r\nHello, World!".to_string(),
|
|
113
|
+
];
|
|
114
|
+
let body = create_multipart_body(boundary, parts);
|
|
115
|
+
|
|
116
|
+
// Create a test request
|
|
117
|
+
let request = axum::http::Request::builder()
|
|
118
|
+
.method("POST")
|
|
119
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
120
|
+
.body(axum::body::Body::from(body))
|
|
121
|
+
.unwrap();
|
|
122
|
+
|
|
123
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
124
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
125
|
+
|
|
126
|
+
assert!(result.is_object());
|
|
127
|
+
let obj = result.as_object().unwrap();
|
|
128
|
+
assert!(obj.contains_key("file"));
|
|
129
|
+
|
|
130
|
+
let file_obj = &obj["file"];
|
|
131
|
+
assert_eq!(file_obj["filename"], "test.txt");
|
|
132
|
+
assert_eq!(file_obj["size"], 13);
|
|
133
|
+
assert_eq!(file_obj["content"], file_content);
|
|
134
|
+
assert_eq!(file_obj["content_type"], "text/plain");
|
|
135
|
+
}
|
|
136
|
+
|
|
137
|
+
#[tokio::test]
|
|
138
|
+
async fn test_multiple_files() {
|
|
139
|
+
let boundary = "boundary123";
|
|
140
|
+
let parts = vec![
|
|
141
|
+
"Content-Disposition: form-data; name=\"file1\"; filename=\"file1.txt\"\r\nContent-Type: text/plain\r\n\r\nContent 1".to_string(),
|
|
142
|
+
"Content-Disposition: form-data; name=\"file2\"; filename=\"file2.txt\"\r\nContent-Type: text/plain\r\n\r\nContent 2".to_string(),
|
|
143
|
+
];
|
|
144
|
+
let body = create_multipart_body(boundary, parts);
|
|
145
|
+
|
|
146
|
+
let request = axum::http::Request::builder()
|
|
147
|
+
.method("POST")
|
|
148
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
149
|
+
.body(axum::body::Body::from(body))
|
|
150
|
+
.unwrap();
|
|
151
|
+
|
|
152
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
153
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
154
|
+
|
|
155
|
+
let obj = result.as_object().unwrap();
|
|
156
|
+
assert!(obj.contains_key("file1"));
|
|
157
|
+
assert!(obj.contains_key("file2"));
|
|
158
|
+
assert_eq!(obj["file1"]["filename"], "file1.txt");
|
|
159
|
+
assert_eq!(obj["file2"]["filename"], "file2.txt");
|
|
160
|
+
}
|
|
161
|
+
|
|
162
|
+
#[tokio::test]
|
|
163
|
+
async fn test_mixed_form_data_and_files() {
|
|
164
|
+
let boundary = "boundary123";
|
|
165
|
+
let parts = vec![
|
|
166
|
+
"Content-Disposition: form-data; name=\"username\"\r\n\r\njohn_doe".to_string(),
|
|
167
|
+
"Content-Disposition: form-data; name=\"avatar\"; filename=\"avatar.png\"\r\nContent-Type: image/png\r\n\r\nPNG_BINARY_DATA".to_string(),
|
|
168
|
+
];
|
|
169
|
+
let body = create_multipart_body(boundary, parts);
|
|
170
|
+
|
|
171
|
+
let request = axum::http::Request::builder()
|
|
172
|
+
.method("POST")
|
|
173
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
174
|
+
.body(axum::body::Body::from(body))
|
|
175
|
+
.unwrap();
|
|
176
|
+
|
|
177
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
178
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
179
|
+
|
|
180
|
+
let obj = result.as_object().unwrap();
|
|
181
|
+
assert_eq!(obj["username"], "john_doe");
|
|
182
|
+
assert_eq!(obj["avatar"]["filename"], "avatar.png");
|
|
183
|
+
assert_eq!(obj["avatar"]["content_type"], "image/png");
|
|
184
|
+
}
|
|
185
|
+
|
|
186
|
+
#[tokio::test]
|
|
187
|
+
async fn test_content_type_validation() {
|
|
188
|
+
let boundary = "boundary123";
|
|
189
|
+
let parts = vec![
|
|
190
|
+
"Content-Disposition: form-data; name=\"data\"; filename=\"data.json\"\r\nContent-Type: application/json\r\n\r\n{}".to_string(),
|
|
191
|
+
];
|
|
192
|
+
let body = create_multipart_body(boundary, parts);
|
|
193
|
+
|
|
194
|
+
let request = axum::http::Request::builder()
|
|
195
|
+
.method("POST")
|
|
196
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
197
|
+
.body(axum::body::Body::from(body))
|
|
198
|
+
.unwrap();
|
|
199
|
+
|
|
200
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
201
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
202
|
+
|
|
203
|
+
let obj = result.as_object().unwrap();
|
|
204
|
+
assert_eq!(obj["data"]["content_type"], "application/json");
|
|
205
|
+
}
|
|
206
|
+
|
|
207
|
+
#[tokio::test]
|
|
208
|
+
async fn test_text_file_handling() {
|
|
209
|
+
let boundary = "boundary123";
|
|
210
|
+
let text_content = "This is a plain text file\nwith multiple lines";
|
|
211
|
+
let parts = vec![
|
|
212
|
+
"Content-Disposition: form-data; name=\"document\"; filename=\"document.txt\"\r\nContent-Type: text/plain\r\n\r\nThis is a plain text file\nwith multiple lines".to_string(),
|
|
213
|
+
];
|
|
214
|
+
let body = create_multipart_body(boundary, parts);
|
|
215
|
+
|
|
216
|
+
let request = axum::http::Request::builder()
|
|
217
|
+
.method("POST")
|
|
218
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
219
|
+
.body(axum::body::Body::from(body))
|
|
220
|
+
.unwrap();
|
|
221
|
+
|
|
222
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
223
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
224
|
+
|
|
225
|
+
let obj = result.as_object().unwrap();
|
|
226
|
+
assert_eq!(obj["document"]["content"], text_content);
|
|
227
|
+
assert!(obj["document"]["content"].is_string());
|
|
228
|
+
}
|
|
229
|
+
|
|
230
|
+
#[tokio::test]
|
|
231
|
+
async fn test_binary_file_handling() {
|
|
232
|
+
let boundary = "boundary123";
|
|
233
|
+
// Create binary data (1MB + 1 byte to exceed threshold)
|
|
234
|
+
let binary_data: Vec<u8> = vec![0xFF; 1024 * 1024 + 1];
|
|
235
|
+
let mut body = Vec::new();
|
|
236
|
+
body.extend_from_slice(format!("--{}\r\n", boundary).as_bytes());
|
|
237
|
+
body.extend_from_slice(b"Content-Disposition: form-data; name=\"binary\"; filename=\"binary.bin\"\r\nContent-Type: application/octet-stream\r\n\r\n");
|
|
238
|
+
body.extend_from_slice(&binary_data);
|
|
239
|
+
body.extend_from_slice(format!("\r\n--{}--\r\n", boundary).as_bytes());
|
|
240
|
+
|
|
241
|
+
let request = axum::http::Request::builder()
|
|
242
|
+
.method("POST")
|
|
243
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
244
|
+
.body(axum::body::Body::from(body))
|
|
245
|
+
.unwrap();
|
|
246
|
+
|
|
247
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
248
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
249
|
+
|
|
250
|
+
let obj = result.as_object().unwrap();
|
|
251
|
+
let content = obj["binary"]["content"].as_str().unwrap();
|
|
252
|
+
assert!(content.contains("<binary data"));
|
|
253
|
+
assert!(content.contains("1048577 bytes"));
|
|
254
|
+
}
|
|
255
|
+
|
|
256
|
+
#[tokio::test]
|
|
257
|
+
async fn test_json_field_parsing() {
|
|
258
|
+
let boundary = "boundary123";
|
|
259
|
+
let json_value = r#"{"key":"value","nested":{"inner":"data"}}"#;
|
|
260
|
+
let parts = vec![format!(
|
|
261
|
+
"Content-Disposition: form-data; name=\"metadata\"\r\n\r\n{}",
|
|
262
|
+
json_value
|
|
263
|
+
)];
|
|
264
|
+
let body = create_multipart_body(boundary, parts);
|
|
265
|
+
|
|
266
|
+
let request = axum::http::Request::builder()
|
|
267
|
+
.method("POST")
|
|
268
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
269
|
+
.body(axum::body::Body::from(body))
|
|
270
|
+
.unwrap();
|
|
271
|
+
|
|
272
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
273
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
274
|
+
|
|
275
|
+
let obj = result.as_object().unwrap();
|
|
276
|
+
let metadata = &obj["metadata"];
|
|
277
|
+
assert!(metadata.is_object());
|
|
278
|
+
assert_eq!(metadata["key"], "value");
|
|
279
|
+
assert_eq!(metadata["nested"]["inner"], "data");
|
|
280
|
+
}
|
|
281
|
+
|
|
282
|
+
#[tokio::test]
|
|
283
|
+
async fn test_multiple_values_same_name() {
|
|
284
|
+
let boundary = "boundary123";
|
|
285
|
+
let parts = vec![
|
|
286
|
+
"Content-Disposition: form-data; name=\"tags\"\r\n\r\nrust".to_string(),
|
|
287
|
+
"Content-Disposition: form-data; name=\"tags\"\r\n\r\nweb".to_string(),
|
|
288
|
+
"Content-Disposition: form-data; name=\"tags\"\r\n\r\nserver".to_string(),
|
|
289
|
+
];
|
|
290
|
+
let body = create_multipart_body(boundary, parts);
|
|
291
|
+
|
|
292
|
+
let request = axum::http::Request::builder()
|
|
293
|
+
.method("POST")
|
|
294
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
295
|
+
.body(axum::body::Body::from(body))
|
|
296
|
+
.unwrap();
|
|
297
|
+
|
|
298
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
299
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
300
|
+
|
|
301
|
+
let obj = result.as_object().unwrap();
|
|
302
|
+
let tags = &obj["tags"];
|
|
303
|
+
assert!(tags.is_array());
|
|
304
|
+
let tags_array = tags.as_array().unwrap();
|
|
305
|
+
assert_eq!(tags_array.len(), 3);
|
|
306
|
+
assert_eq!(tags_array[0], "rust");
|
|
307
|
+
assert_eq!(tags_array[1], "web");
|
|
308
|
+
assert_eq!(tags_array[2], "server");
|
|
309
|
+
}
|
|
310
|
+
|
|
311
|
+
#[tokio::test]
|
|
312
|
+
async fn test_empty_field() {
|
|
313
|
+
let boundary = "boundary123";
|
|
314
|
+
let parts = vec!["Content-Disposition: form-data; name=\"empty_field\"\r\n\r\n".to_string()];
|
|
315
|
+
let body = create_multipart_body(boundary, parts);
|
|
316
|
+
|
|
317
|
+
let request = axum::http::Request::builder()
|
|
318
|
+
.method("POST")
|
|
319
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
320
|
+
.body(axum::body::Body::from(body))
|
|
321
|
+
.unwrap();
|
|
322
|
+
|
|
323
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
324
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
325
|
+
|
|
326
|
+
let obj = result.as_object().unwrap();
|
|
327
|
+
assert_eq!(obj["empty_field"], "");
|
|
328
|
+
}
|
|
329
|
+
|
|
330
|
+
#[tokio::test]
|
|
331
|
+
async fn test_default_content_type_when_missing() {
|
|
332
|
+
let boundary = "boundary123";
|
|
333
|
+
let parts = vec![
|
|
334
|
+
"Content-Disposition: form-data; name=\"file\"; filename=\"no_type.bin\"\r\n\r\nbinary content".to_string(),
|
|
335
|
+
];
|
|
336
|
+
let body = create_multipart_body(boundary, parts);
|
|
337
|
+
|
|
338
|
+
let request = axum::http::Request::builder()
|
|
339
|
+
.method("POST")
|
|
340
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
341
|
+
.body(axum::body::Body::from(body))
|
|
342
|
+
.unwrap();
|
|
343
|
+
|
|
344
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
345
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
346
|
+
|
|
347
|
+
let obj = result.as_object().unwrap();
|
|
348
|
+
assert_eq!(obj["file"]["content_type"], "application/octet-stream");
|
|
349
|
+
}
|
|
350
|
+
|
|
351
|
+
#[tokio::test]
|
|
352
|
+
async fn test_json_array_field_parsing() {
|
|
353
|
+
let boundary = "boundary123";
|
|
354
|
+
let json_array = r#"[1,2,3,4,5]"#;
|
|
355
|
+
let parts = vec![format!(
|
|
356
|
+
"Content-Disposition: form-data; name=\"numbers\"\r\n\r\n{}",
|
|
357
|
+
json_array
|
|
358
|
+
)];
|
|
359
|
+
let body = create_multipart_body(boundary, parts);
|
|
360
|
+
|
|
361
|
+
let request = axum::http::Request::builder()
|
|
362
|
+
.method("POST")
|
|
363
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
364
|
+
.body(axum::body::Body::from(body))
|
|
365
|
+
.unwrap();
|
|
366
|
+
|
|
367
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
368
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
369
|
+
|
|
370
|
+
let obj = result.as_object().unwrap();
|
|
371
|
+
let numbers = &obj["numbers"];
|
|
372
|
+
assert!(numbers.is_array());
|
|
373
|
+
assert_eq!(numbers[0], 1);
|
|
374
|
+
assert_eq!(numbers[4], 5);
|
|
375
|
+
}
|
|
376
|
+
|
|
377
|
+
#[tokio::test]
|
|
378
|
+
async fn test_mixed_json_and_text_files() {
|
|
379
|
+
let boundary = "boundary123";
|
|
380
|
+
let parts = vec![
|
|
381
|
+
"Content-Disposition: form-data; name=\"config\"; filename=\"config.json\"\r\nContent-Type: application/json\r\n\r\n{\"enabled\":true}".to_string(),
|
|
382
|
+
"Content-Disposition: form-data; name=\"readme\"; filename=\"README.txt\"\r\nContent-Type: text/plain\r\n\r\nThis is a readme".to_string(),
|
|
383
|
+
];
|
|
384
|
+
let body = create_multipart_body(boundary, parts);
|
|
385
|
+
|
|
386
|
+
let request = axum::http::Request::builder()
|
|
387
|
+
.method("POST")
|
|
388
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
389
|
+
.body(axum::body::Body::from(body))
|
|
390
|
+
.unwrap();
|
|
391
|
+
|
|
392
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
393
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
394
|
+
|
|
395
|
+
let obj = result.as_object().unwrap();
|
|
396
|
+
assert_eq!(obj["config"]["content_type"], "application/json");
|
|
397
|
+
assert_eq!(obj["readme"]["content_type"], "text/plain");
|
|
398
|
+
assert!(obj["config"]["content"].is_string());
|
|
399
|
+
}
|
|
400
|
+
|
|
401
|
+
#[tokio::test]
|
|
402
|
+
async fn test_file_size_calculation() {
|
|
403
|
+
let boundary = "boundary123";
|
|
404
|
+
let file_content = "Exact size test";
|
|
405
|
+
let expected_size = file_content.len();
|
|
406
|
+
let parts = vec![format!(
|
|
407
|
+
"Content-Disposition: form-data; name=\"file\"; filename=\"size.txt\"\r\nContent-Type: text/plain\r\n\r\n{}",
|
|
408
|
+
file_content
|
|
409
|
+
)];
|
|
410
|
+
let body = create_multipart_body(boundary, parts);
|
|
411
|
+
|
|
412
|
+
let request = axum::http::Request::builder()
|
|
413
|
+
.method("POST")
|
|
414
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
415
|
+
.body(axum::body::Body::from(body))
|
|
416
|
+
.unwrap();
|
|
417
|
+
|
|
418
|
+
let multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
419
|
+
let result = parse_multipart_to_json(multipart).await.unwrap();
|
|
420
|
+
|
|
421
|
+
let obj = result.as_object().unwrap();
|
|
422
|
+
assert_eq!(obj["file"]["size"], expected_size);
|
|
423
|
+
}
|
|
424
|
+
|
|
425
|
+
// --- NEW COMPREHENSIVE TESTS FOR BOUNDARY PARSING & FILE UPLOADS ---
|
|
426
|
+
|
|
427
|
+
#[tokio::test]
|
|
428
|
+
async fn test_boundary_with_special_characters() {
|
|
429
|
+
let boundary: &str = "boundary-with-dashes_and_underscores.123";
|
|
430
|
+
let parts: Vec<String> = vec!["Content-Disposition: form-data; name=\"field\"\r\n\r\nvalue".to_string()];
|
|
431
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
432
|
+
|
|
433
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
434
|
+
.method("POST")
|
|
435
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
436
|
+
.body(axum::body::Body::from(body))
|
|
437
|
+
.unwrap();
|
|
438
|
+
|
|
439
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
440
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
441
|
+
|
|
442
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
443
|
+
assert_eq!(obj["field"], "value");
|
|
444
|
+
}
|
|
445
|
+
|
|
446
|
+
#[tokio::test]
|
|
447
|
+
async fn test_boundary_parsing_with_crlf() {
|
|
448
|
+
// RFC 7578 requires CRLF line endings for multipart bodies.
|
|
449
|
+
// This test verifies proper boundary parsing with spec-compliant CRLF.
|
|
450
|
+
let boundary: &str = "boundary123";
|
|
451
|
+
let mut body: Vec<u8> = Vec::new();
|
|
452
|
+
|
|
453
|
+
body.extend_from_slice(format!("--{}\r\n", boundary).as_bytes());
|
|
454
|
+
body.extend_from_slice(b"Content-Disposition: form-data; name=\"field\"\r\n");
|
|
455
|
+
body.extend_from_slice(b"\r\n");
|
|
456
|
+
body.extend_from_slice(b"value");
|
|
457
|
+
body.extend_from_slice(b"\r\n");
|
|
458
|
+
body.extend_from_slice(format!("--{}--\r\n", boundary).as_bytes());
|
|
459
|
+
|
|
460
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
461
|
+
.method("POST")
|
|
462
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
463
|
+
.body(axum::body::Body::from(body))
|
|
464
|
+
.unwrap();
|
|
465
|
+
|
|
466
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
467
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
468
|
+
|
|
469
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
470
|
+
assert!(obj.contains_key("field"));
|
|
471
|
+
assert_eq!(obj["field"], "value");
|
|
472
|
+
}
|
|
473
|
+
|
|
474
|
+
#[tokio::test]
|
|
475
|
+
async fn test_file_upload_with_no_content_type_header() {
|
|
476
|
+
let boundary: &str = "boundary123";
|
|
477
|
+
let parts: Vec<String> = vec![
|
|
478
|
+
"Content-Disposition: form-data; name=\"file\"; filename=\"nocontenttype.bin\"\r\n\r\nbinary_content_here"
|
|
479
|
+
.to_string(),
|
|
480
|
+
];
|
|
481
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
482
|
+
|
|
483
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
484
|
+
.method("POST")
|
|
485
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
486
|
+
.body(axum::body::Body::from(body))
|
|
487
|
+
.unwrap();
|
|
488
|
+
|
|
489
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
490
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
491
|
+
|
|
492
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
493
|
+
let file_obj: &serde_json::Value = &obj["file"];
|
|
494
|
+
|
|
495
|
+
assert_eq!(file_obj["content_type"], "application/octet-stream");
|
|
496
|
+
assert_eq!(file_obj["filename"], "nocontenttype.bin");
|
|
497
|
+
}
|
|
498
|
+
|
|
499
|
+
#[tokio::test]
|
|
500
|
+
async fn test_empty_file_upload() {
|
|
501
|
+
let boundary: &str = "boundary123";
|
|
502
|
+
let parts: Vec<String> = vec![
|
|
503
|
+
"Content-Disposition: form-data; name=\"emptyfile\"; filename=\"empty.txt\"\r\nContent-Type: text/plain\r\n\r\n".to_string(),
|
|
504
|
+
];
|
|
505
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
506
|
+
|
|
507
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
508
|
+
.method("POST")
|
|
509
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
510
|
+
.body(axum::body::Body::from(body))
|
|
511
|
+
.unwrap();
|
|
512
|
+
|
|
513
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
514
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
515
|
+
|
|
516
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
517
|
+
let file_obj: &serde_json::Value = &obj["emptyfile"];
|
|
518
|
+
|
|
519
|
+
assert_eq!(file_obj["size"], 0);
|
|
520
|
+
assert_eq!(file_obj["content"], "");
|
|
521
|
+
}
|
|
522
|
+
|
|
523
|
+
#[tokio::test]
|
|
524
|
+
async fn test_very_large_field_name() {
|
|
525
|
+
let boundary: &str = "boundary123";
|
|
526
|
+
let long_field_name: String = "field_".repeat(100);
|
|
527
|
+
let parts: Vec<String> = vec![format!(
|
|
528
|
+
"Content-Disposition: form-data; name=\"{}\"\r\n\r\nvalue",
|
|
529
|
+
long_field_name
|
|
530
|
+
)];
|
|
531
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
532
|
+
|
|
533
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
534
|
+
.method("POST")
|
|
535
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
536
|
+
.body(axum::body::Body::from(body))
|
|
537
|
+
.unwrap();
|
|
538
|
+
|
|
539
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
540
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
541
|
+
|
|
542
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
543
|
+
assert!(obj.contains_key(&long_field_name));
|
|
544
|
+
assert_eq!(obj[&long_field_name], "value");
|
|
545
|
+
}
|
|
546
|
+
|
|
547
|
+
#[tokio::test]
|
|
548
|
+
async fn test_very_large_field_value() {
|
|
549
|
+
let boundary: &str = "boundary123";
|
|
550
|
+
let large_value: String = "x".repeat(10_000);
|
|
551
|
+
let parts: Vec<String> = vec![format!(
|
|
552
|
+
"Content-Disposition: form-data; name=\"large\"\r\n\r\n{}",
|
|
553
|
+
large_value
|
|
554
|
+
)];
|
|
555
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
556
|
+
|
|
557
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
558
|
+
.method("POST")
|
|
559
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
560
|
+
.body(axum::body::Body::from(body))
|
|
561
|
+
.unwrap();
|
|
562
|
+
|
|
563
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
564
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
565
|
+
|
|
566
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
567
|
+
let retrieved_value: &str = obj["large"].as_str().unwrap();
|
|
568
|
+
assert_eq!(retrieved_value.len(), 10_000);
|
|
569
|
+
}
|
|
570
|
+
|
|
571
|
+
#[tokio::test]
|
|
572
|
+
async fn test_multiple_files_same_name() {
|
|
573
|
+
let boundary: &str = "boundary123";
|
|
574
|
+
let parts: Vec<String> = vec![
|
|
575
|
+
"Content-Disposition: form-data; name=\"document\"; filename=\"doc1.txt\"\r\nContent-Type: text/plain\r\n\r\nContent 1".to_string(),
|
|
576
|
+
"Content-Disposition: form-data; name=\"document\"; filename=\"doc2.txt\"\r\nContent-Type: text/plain\r\n\r\nContent 2".to_string(),
|
|
577
|
+
];
|
|
578
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
579
|
+
|
|
580
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
581
|
+
.method("POST")
|
|
582
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
583
|
+
.body(axum::body::Body::from(body))
|
|
584
|
+
.unwrap();
|
|
585
|
+
|
|
586
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
587
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
588
|
+
|
|
589
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
590
|
+
let docs: &serde_json::Value = &obj["document"];
|
|
591
|
+
assert!(docs.is_array());
|
|
592
|
+
|
|
593
|
+
let docs_array: &Vec<serde_json::Value> = docs.as_array().unwrap();
|
|
594
|
+
assert_eq!(docs_array.len(), 2);
|
|
595
|
+
assert_eq!(docs_array[0]["filename"], "doc1.txt");
|
|
596
|
+
assert_eq!(docs_array[1]["filename"], "doc2.txt");
|
|
597
|
+
}
|
|
598
|
+
|
|
599
|
+
#[tokio::test]
|
|
600
|
+
async fn test_unicode_in_filename() {
|
|
601
|
+
let boundary: &str = "boundary123";
|
|
602
|
+
let parts: Vec<String> = vec![
|
|
603
|
+
"Content-Disposition: form-data; name=\"file\"; filename=\"файл_文件_🎯.txt\"\r\nContent-Type: text/plain\r\n\r\nUnicode content".to_string(),
|
|
604
|
+
];
|
|
605
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
606
|
+
|
|
607
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
608
|
+
.method("POST")
|
|
609
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
610
|
+
.body(axum::body::Body::from(body))
|
|
611
|
+
.unwrap();
|
|
612
|
+
|
|
613
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
614
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
615
|
+
|
|
616
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
617
|
+
let filename: &str = obj["file"]["filename"].as_str().unwrap();
|
|
618
|
+
assert!(filename.contains("файл"));
|
|
619
|
+
assert!(filename.contains("文件"));
|
|
620
|
+
}
|
|
621
|
+
|
|
622
|
+
#[tokio::test]
|
|
623
|
+
async fn test_file_with_crlf_in_content() {
|
|
624
|
+
let boundary: &str = "boundary123";
|
|
625
|
+
let content_with_crlf: &str = "Line 1\r\nLine 2\r\nLine 3";
|
|
626
|
+
let parts: Vec<String> = vec![format!(
|
|
627
|
+
"Content-Disposition: form-data; name=\"file\"; filename=\"crlf.txt\"\r\nContent-Type: text/plain\r\n\r\n{}",
|
|
628
|
+
content_with_crlf
|
|
629
|
+
)];
|
|
630
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
631
|
+
|
|
632
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
633
|
+
.method("POST")
|
|
634
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
635
|
+
.body(axum::body::Body::from(body))
|
|
636
|
+
.unwrap();
|
|
637
|
+
|
|
638
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
639
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
640
|
+
|
|
641
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
642
|
+
let retrieved_content: &str = obj["file"]["content"].as_str().unwrap();
|
|
643
|
+
assert!(retrieved_content.contains("Line 1"));
|
|
644
|
+
assert!(retrieved_content.contains("Line 2"));
|
|
645
|
+
}
|
|
646
|
+
|
|
647
|
+
#[tokio::test]
|
|
648
|
+
async fn test_boundary_on_line_boundary() {
|
|
649
|
+
let boundary: &str = "boundary";
|
|
650
|
+
let mut body: Vec<u8> = Vec::new();
|
|
651
|
+
|
|
652
|
+
body.extend_from_slice(format!("--{}\r\n", boundary).as_bytes());
|
|
653
|
+
body.extend_from_slice(b"Content-Disposition: form-data; name=\"field1\"\r\n\r\nvalue1\r\n");
|
|
654
|
+
body.extend_from_slice(format!("--{}\r\n", boundary).as_bytes());
|
|
655
|
+
body.extend_from_slice(b"Content-Disposition: form-data; name=\"field2\"\r\n\r\nvalue2\r\n");
|
|
656
|
+
body.extend_from_slice(format!("--{}--\r\n", boundary).as_bytes());
|
|
657
|
+
|
|
658
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
659
|
+
.method("POST")
|
|
660
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
661
|
+
.body(axum::body::Body::from(body))
|
|
662
|
+
.unwrap();
|
|
663
|
+
|
|
664
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
665
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
666
|
+
|
|
667
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
668
|
+
assert_eq!(obj["field1"], "value1");
|
|
669
|
+
assert_eq!(obj["field2"], "value2");
|
|
670
|
+
}
|
|
671
|
+
|
|
672
|
+
#[tokio::test]
|
|
673
|
+
async fn test_content_type_with_charset_parameter() {
|
|
674
|
+
let boundary: &str = "boundary123";
|
|
675
|
+
let parts: Vec<String> = vec![
|
|
676
|
+
"Content-Disposition: form-data; name=\"file\"; filename=\"utf8.txt\"\r\nContent-Type: text/plain; charset=utf-8\r\n\r\nUTF-8 content".to_string(),
|
|
677
|
+
];
|
|
678
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
679
|
+
|
|
680
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
681
|
+
.method("POST")
|
|
682
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
683
|
+
.body(axum::body::Body::from(body))
|
|
684
|
+
.unwrap();
|
|
685
|
+
|
|
686
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
687
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
688
|
+
|
|
689
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
690
|
+
let content_type: &str = obj["file"]["content_type"].as_str().unwrap();
|
|
691
|
+
assert!(content_type.contains("text/plain"));
|
|
692
|
+
}
|
|
693
|
+
|
|
694
|
+
#[tokio::test]
|
|
695
|
+
async fn test_null_bytes_in_binary_file() {
|
|
696
|
+
let boundary: &str = "boundary123";
|
|
697
|
+
let mut body: Vec<u8> = Vec::new();
|
|
698
|
+
|
|
699
|
+
body.extend_from_slice(format!("--{}\r\n", boundary).as_bytes());
|
|
700
|
+
body.extend_from_slice(b"Content-Disposition: form-data; name=\"binary\"; filename=\"nullbytes.bin\"\r\nContent-Type: application/octet-stream\r\n\r\n");
|
|
701
|
+
body.extend_from_slice(&[0x00, 0x01, 0x02, 0xFF, 0xFE, 0xFD]);
|
|
702
|
+
body.extend_from_slice(b"\r\n");
|
|
703
|
+
body.extend_from_slice(format!("--{}--\r\n", boundary).as_bytes());
|
|
704
|
+
|
|
705
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
706
|
+
.method("POST")
|
|
707
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
708
|
+
.body(axum::body::Body::from(body))
|
|
709
|
+
.unwrap();
|
|
710
|
+
|
|
711
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
712
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
713
|
+
|
|
714
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
715
|
+
let file_obj: &serde_json::Value = &obj["binary"];
|
|
716
|
+
assert_eq!(file_obj["size"], 6);
|
|
717
|
+
}
|
|
718
|
+
|
|
719
|
+
#[tokio::test]
|
|
720
|
+
async fn test_form_field_with_malformed_json_stays_string() {
|
|
721
|
+
let boundary: &str = "boundary123";
|
|
722
|
+
let parts: Vec<String> =
|
|
723
|
+
vec!["Content-Disposition: form-data; name=\"malformed\"\r\n\r\n{invalid json}".to_string()];
|
|
724
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
725
|
+
|
|
726
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
727
|
+
.method("POST")
|
|
728
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
729
|
+
.body(axum::body::Body::from(body))
|
|
730
|
+
.unwrap();
|
|
731
|
+
|
|
732
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
733
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
734
|
+
|
|
735
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
736
|
+
assert!(obj["malformed"].is_string());
|
|
737
|
+
assert_eq!(obj["malformed"], "{invalid json}");
|
|
738
|
+
}
|
|
739
|
+
|
|
740
|
+
#[tokio::test]
|
|
741
|
+
async fn test_multiple_files_different_sizes_around_threshold() {
|
|
742
|
+
// Test files around the 1MB streaming threshold
|
|
743
|
+
// - Small file: below threshold (buffered)
|
|
744
|
+
// - Large file: exceeds threshold (marked as binary)
|
|
745
|
+
let boundary: &str = "boundary123";
|
|
746
|
+
let small_content: Vec<u8> = "small".repeat(100).into_bytes();
|
|
747
|
+
// Large file just over 1MB threshold to trigger binary handling
|
|
748
|
+
let large_content: Vec<u8> = vec![0x42; 1024 * 1024 + 1]; // 'B' repeated
|
|
749
|
+
|
|
750
|
+
let mut body: Vec<u8> = Vec::new();
|
|
751
|
+
|
|
752
|
+
// Small file
|
|
753
|
+
body.extend_from_slice(format!("--{}\r\n", boundary).as_bytes());
|
|
754
|
+
body.extend_from_slice(b"Content-Disposition: form-data; name=\"small\"; filename=\"small.txt\"\r\nContent-Type: text/plain\r\n\r\n");
|
|
755
|
+
body.extend_from_slice(&small_content);
|
|
756
|
+
body.extend_from_slice(b"\r\n");
|
|
757
|
+
|
|
758
|
+
// Large file (exceeds 1MB threshold)
|
|
759
|
+
body.extend_from_slice(format!("--{}\r\n", boundary).as_bytes());
|
|
760
|
+
body.extend_from_slice(b"Content-Disposition: form-data; name=\"large\"; filename=\"large.bin\"\r\nContent-Type: application/octet-stream\r\n\r\n");
|
|
761
|
+
body.extend_from_slice(&large_content);
|
|
762
|
+
body.extend_from_slice(b"\r\n");
|
|
763
|
+
|
|
764
|
+
// Final boundary
|
|
765
|
+
body.extend_from_slice(format!("--{}--\r\n", boundary).as_bytes());
|
|
766
|
+
|
|
767
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
768
|
+
.method("POST")
|
|
769
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
770
|
+
.body(axum::body::Body::from(body))
|
|
771
|
+
.unwrap();
|
|
772
|
+
|
|
773
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
774
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
775
|
+
|
|
776
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
777
|
+
|
|
778
|
+
// Verify small file is included in full
|
|
779
|
+
assert!(obj.contains_key("small"));
|
|
780
|
+
assert!(obj["small"]["content"].is_string());
|
|
781
|
+
let small_str = obj["small"]["content"].as_str().unwrap();
|
|
782
|
+
assert_eq!(small_str.len(), 500); // "small" (5) * 100
|
|
783
|
+
|
|
784
|
+
// Verify large file exceeds threshold and is marked as binary
|
|
785
|
+
assert!(obj.contains_key("large"));
|
|
786
|
+
let large_content_str = obj["large"]["content"].as_str().unwrap();
|
|
787
|
+
assert!(large_content_str.contains("<binary data"));
|
|
788
|
+
assert!(large_content_str.contains("1048577 bytes")); // 1MB + 1 byte
|
|
789
|
+
}
|
|
790
|
+
|
|
791
|
+
#[tokio::test]
|
|
792
|
+
async fn test_header_case_insensitivity() {
|
|
793
|
+
let boundary: &str = "boundary123";
|
|
794
|
+
let parts: Vec<String> =
|
|
795
|
+
vec!["content-disposition: form-data; name=\"field\"\r\nContent-TYPE: text/plain\r\n\r\nvalue".to_string()];
|
|
796
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
797
|
+
|
|
798
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
799
|
+
.method("POST")
|
|
800
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
801
|
+
.body(axum::body::Body::from(body))
|
|
802
|
+
.unwrap();
|
|
803
|
+
|
|
804
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
805
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
806
|
+
|
|
807
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
808
|
+
assert_eq!(obj["field"], "value");
|
|
809
|
+
}
|
|
810
|
+
|
|
811
|
+
#[tokio::test]
|
|
812
|
+
async fn test_field_value_containing_boundary_string() {
|
|
813
|
+
let boundary: &str = "boundary123";
|
|
814
|
+
let value_with_boundary: &str = "This contains --boundary123 but is not a real boundary";
|
|
815
|
+
let parts: Vec<String> = vec![format!(
|
|
816
|
+
"Content-Disposition: form-data; name=\"field\"\r\n\r\n{}",
|
|
817
|
+
value_with_boundary
|
|
818
|
+
)];
|
|
819
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
820
|
+
|
|
821
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
822
|
+
.method("POST")
|
|
823
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
824
|
+
.body(axum::body::Body::from(body))
|
|
825
|
+
.unwrap();
|
|
826
|
+
|
|
827
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
828
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
829
|
+
|
|
830
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
831
|
+
assert_eq!(obj["field"], value_with_boundary);
|
|
832
|
+
}
|
|
833
|
+
|
|
834
|
+
#[tokio::test]
|
|
835
|
+
async fn test_json_object_and_array_aggregation_with_files() {
|
|
836
|
+
let boundary: &str = "boundary123";
|
|
837
|
+
let parts: Vec<String> = vec![
|
|
838
|
+
"Content-Disposition: form-data; name=\"config\"\r\n\r\n{\"debug\":true}".to_string(),
|
|
839
|
+
"Content-Disposition: form-data; name=\"file\"; filename=\"data.txt\"\r\nContent-Type: text/plain\r\n\r\nfile content".to_string(),
|
|
840
|
+
];
|
|
841
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
842
|
+
|
|
843
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
844
|
+
.method("POST")
|
|
845
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
846
|
+
.body(axum::body::Body::from(body))
|
|
847
|
+
.unwrap();
|
|
848
|
+
|
|
849
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
850
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
851
|
+
|
|
852
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
853
|
+
assert!(obj["config"].is_object());
|
|
854
|
+
assert!(obj["file"].is_object());
|
|
855
|
+
assert_eq!(obj["file"]["filename"], "data.txt");
|
|
856
|
+
}
|
|
857
|
+
|
|
858
|
+
#[tokio::test]
|
|
859
|
+
async fn test_consecutive_empty_fields() {
|
|
860
|
+
let boundary: &str = "boundary123";
|
|
861
|
+
let parts: Vec<String> = vec![
|
|
862
|
+
"Content-Disposition: form-data; name=\"field1\"\r\n\r\n".to_string(),
|
|
863
|
+
"Content-Disposition: form-data; name=\"field2\"\r\n\r\n".to_string(),
|
|
864
|
+
"Content-Disposition: form-data; name=\"field3\"\r\n\r\nactual_value".to_string(),
|
|
865
|
+
];
|
|
866
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
867
|
+
|
|
868
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
869
|
+
.method("POST")
|
|
870
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
871
|
+
.body(axum::body::Body::from(body))
|
|
872
|
+
.unwrap();
|
|
873
|
+
|
|
874
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
875
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
876
|
+
|
|
877
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
878
|
+
assert_eq!(obj["field1"], "");
|
|
879
|
+
assert_eq!(obj["field2"], "");
|
|
880
|
+
assert_eq!(obj["field3"], "actual_value");
|
|
881
|
+
}
|
|
882
|
+
|
|
883
|
+
#[tokio::test]
|
|
884
|
+
async fn test_filename_with_quotes_and_special_chars() {
|
|
885
|
+
let boundary: &str = "boundary123";
|
|
886
|
+
let parts: Vec<String> = vec![
|
|
887
|
+
"Content-Disposition: form-data; name=\"file\"; filename=\"file[2024].txt\"\r\nContent-Type: text/plain\r\n\r\nContent".to_string(),
|
|
888
|
+
];
|
|
889
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
890
|
+
|
|
891
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
892
|
+
.method("POST")
|
|
893
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
894
|
+
.body(axum::body::Body::from(body))
|
|
895
|
+
.unwrap();
|
|
896
|
+
|
|
897
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
898
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
899
|
+
|
|
900
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
901
|
+
let filename: &str = obj["file"]["filename"].as_str().unwrap();
|
|
902
|
+
assert_eq!(filename, "file[2024].txt");
|
|
903
|
+
}
|
|
904
|
+
|
|
905
|
+
#[tokio::test]
|
|
906
|
+
async fn test_text_like_content_type_variant_detection() {
|
|
907
|
+
let boundary: &str = "boundary123";
|
|
908
|
+
let parts: Vec<String> = vec![
|
|
909
|
+
"Content-Disposition: form-data; name=\"html\"; filename=\"page.html\"\r\nContent-Type: text/html\r\n\r\n<html></html>".to_string(),
|
|
910
|
+
"Content-Disposition: form-data; name=\"xml\"; filename=\"data.xml\"\r\nContent-Type: application/xml\r\n\r\n<data></data>".to_string(),
|
|
911
|
+
];
|
|
912
|
+
let body: Vec<u8> = create_multipart_body(boundary, parts);
|
|
913
|
+
|
|
914
|
+
let request: axum::http::Request<axum::body::Body> = axum::http::Request::builder()
|
|
915
|
+
.method("POST")
|
|
916
|
+
.header("content-type", format!("multipart/form-data; boundary={}", boundary))
|
|
917
|
+
.body(axum::body::Body::from(body))
|
|
918
|
+
.unwrap();
|
|
919
|
+
|
|
920
|
+
let multipart: Multipart = Multipart::from_request(request, &()).await.unwrap();
|
|
921
|
+
let result: serde_json::Value = parse_multipart_to_json(multipart).await.unwrap();
|
|
922
|
+
|
|
923
|
+
let obj: &serde_json::Map<String, serde_json::Value> = result.as_object().unwrap();
|
|
924
|
+
let html_content: &str = obj["html"]["content"].as_str().unwrap();
|
|
925
|
+
let xml_content: &str = obj["xml"]["content"].as_str().unwrap();
|
|
926
|
+
|
|
927
|
+
assert_eq!(html_content, "<html></html>");
|
|
928
|
+
assert_eq!(xml_content, "<data></data>");
|
|
929
|
+
}
|
|
930
|
+
}
|