如何使用带有hyper的multipart / form-data发布图像?

时间:2018-07-18 09:04:08

标签: post rust hyper

我正尝试像cURL一样使用hyper发布图像文件:

curl -F smfile=@11.jpg https://httpbin.org/post --trace-ascii -

结果是:

{
  "args": {},
  "data": "",
  "files": {
    "smfile": "data:image/jpeg;base64,..."
  },
  "form": {},
  "headers": {
    "Accept": "/",
    "Connection": "close",
    "Content-Length": "1709",
    "Content-Type": "multipart/form-data; boundary=------------------------58370e136081470e",
    "Expect": "100-continue",
    "Host": "httpbin.org",
    "User-Agent": "curl/7.59.0"
  },
  "json": null,
  "origin": "myip",
  "url": "https://httpbin.org/post"
}

我了解到Content-Type应该设置为带有边界标记的multipart/form-data。这是我的代码:

extern crate futures;
extern crate hyper;
extern crate hyper_tls;
extern crate tokio;

use futures::{future, Future};
use hyper::header::CONTENT_TYPE;
use hyper::rt::Stream;
use hyper::{Body, Client, Method, Request};
use hyper_tls::HttpsConnector;
use std::fs::File;
use std::io::prelude::*;
use std::io::{self, Write};

const BOUNDARY: &'static str = "------------------------ea3bbcf87c101592";

fn main() {
    tokio::run(future::lazy(|| {
        let https = HttpsConnector::new(4).unwrap();
        let client = Client::builder().build::<_, hyper::Body>(https);

        let mut req = Request::new(Body::from(image_data()));

        req.headers_mut().insert(
            CONTENT_TYPE,
            format!("multipart/form-data; boundary={}", BOUNDARY)
                .parse()
                .unwrap(),
        );
        *req.method_mut() = Method::POST;
        *req.uri_mut() = "https://httpbin.org/post".parse().unwrap();

        client
            .request(req)
            .and_then(|res| {
                println!("status: {}", res.status());

                res.into_body().for_each(|chunk| {
                    io::stdout()
                        .write_all(&chunk)
                        .map_err(|e| panic!("stdout error: {}", e))
                })
            })
            .map_err(|e| println!("request error: {}", e))
    }));
}

fn image_data() -> Vec<u8> {
    let mut result: Vec<u8> = Vec::new();
    result.extend_from_slice(format!("--{}\r\n", BOUNDARY).as_bytes());
    result
        .extend_from_slice(format!("Content-Disposition: form-data; name=\"text\"\r\n").as_bytes());
    result.extend_from_slice("title\r\n".as_bytes());
    result.extend_from_slice(format!("--{}\r\n", BOUNDARY).as_bytes());
    result.extend_from_slice(
        format!("Content-Disposition: form-data; name=\"smfile\"; filename=\"11.jpg\"\r\n")
            .as_bytes(),
    );
    result.extend_from_slice("Content-Type: image/jpeg\r\n\r\n".as_bytes());

    let mut f = File::open("11.jpg").unwrap();
    let mut file_data = Vec::new();
    f.read_to_end(&mut file_data).unwrap();

    result.append(&mut file_data);

    result.extend_from_slice(format!("--{}--\r\n", BOUNDARY).as_bytes());
    result
}

complete code

请注意,运行此代码需要一个名为11.jpg的JPEG文件。这可以是任何JPEG文件。

httpbin显示我什么都没张贴:

{
  "args": {},
  "data": "",
  "files": {},
  "form": {},
  "headers": {
    "Connection": "close",
    "Content-Length": "1803",
    "Content-Type": "multipart/form-data; boundary=------------------------ea3bbcf87c101592",
    "Host": "httpbin.org"
  },
  "json": null,
  "origin": "myip",
  "url": "https://httpbin.org/post"
}

我不知道该如何解决。

1 个答案:

答案 0 :(得分:2)

您没有在最终边界之前正确放置换行符/回车对。

这是我编写您的身体生成代码的方式,需要更少的分配:

fn image_data() -> io::Result<Vec<u8>> {
    let mut data = Vec::new();
    write!(&mut data, "--{}\r\n", BOUNDARY)?;
    write!(&mut data, "Content-Disposition: form-data; name=\"smfile\"; filename=\"11.jpg\"\r\n")?;
    write!(&mut data, "Content-Type: image/jpeg\r\n")?;
    write!(&mut data, "\r\n")?;

    let mut f = File::open("11.jpg")?;
    f.read_to_end(&mut data)?;

    write!(&mut data, "\r\n")?; // The key thing you are missing
    write!(&mut data, "--{}--\r\n", BOUNDARY)?;

    Ok(data)
}

调用此代码也可以简化:

fn main() {
    let https = HttpsConnector::new(4).unwrap();
    let client = Client::builder().build::<_, hyper::Body>(https);

    let data = image_data().unwrap();
    let req = Request::post("https://httpbin.org/post")
        .header(CONTENT_TYPE, &*format!("multipart/form-data; boundary={}", BOUNDARY))
        .body(data.into())
        .unwrap();

    tokio::run(future::lazy(move || {
        client
            .request(req)
            .and_then(|res| {
                println!("status: {}", res.status());

                res.into_body().for_each(|chunk| {
                    io::stdout()
                        .write_all(&chunk)
                        .map_err(|e| panic!("stdout error: {}", e))
                })
            })
            .map_err(|e| println!("request error: {}", e))
    }));
}