blob: a01f8c78edccb6777ef09a8cc3ada00bbe508b92 (
plain)
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
|
// -*- coding: utf-8 -*-
//
// Simple CMS
//
// Copyright (C) 2011-2024 Michael Büsch <m@bues.ch>
//
// Licensed under the Apache License version 2.0
// or the MIT license, at your option.
// SPDX-License-Identifier: Apache-2.0 OR MIT
use anyhow::{self as ah, Context as _};
use multer::{parse_boundary, Constraints, Multipart, SizeLimit};
use std::collections::HashMap;
const LIMIT_WHOLE_STREAM: u64 = 1024 * 1024;
const LIMIT_PER_FIELD: u64 = 1024 * 128;
pub struct FormFields {
items: HashMap<String, Vec<u8>>,
}
impl FormFields {
pub async fn new(body: &[u8], body_mime: &str) -> ah::Result<Self> {
let boundary = parse_boundary(body_mime).context("Parse form-data boundary")?;
let sizelim = SizeLimit::new()
.whole_stream(LIMIT_WHOLE_STREAM)
.per_field(LIMIT_PER_FIELD);
let constr = Constraints::new().size_limit(sizelim);
let mut multipart = Multipart::with_reader_with_constraints(body, boundary, constr);
let mut items = HashMap::new();
while let Some(field) = multipart.next_field().await.context("Multipart field")? {
let Some(name) = field.name() else {
continue;
};
let name = name.to_string();
let Ok(data) = field.bytes().await else {
continue;
};
let data = data.to_vec();
items.insert(name, data);
}
Ok(Self { items })
}
pub fn into_items(self) -> HashMap<String, Vec<u8>> {
self.items
}
}
// vim: ts=4 sw=4 expandtab
|
