tyler/proxmox-backup
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

src/client/http_client.rs: encode parameters inside body

Browse Source 
To avoid problems with long uris.
This commit is contained in:
Dietmar Maurer 2019-05-24 08:32:55 +02:00
parent 9bb675ec23
commit eb2bdd1bcc
1 changed files with 19 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/client/http_client.rs
View File

@ -571,7 +571,7 @@ impl BackupClient {
let upload_queue = upload_queue.clone(); let upload_queue = upload_queue.clone();
let upload_data; let upload_data;
let request; let mut request;
match merged_chunk_info { match merged_chunk_info {
MergedChunkInfo::New(chunk_info) => { MergedChunkInfo::New(chunk_info) => {
@ -588,8 +588,10 @@ impl BackupClient {
} }
println!("append existing chunks ({})", digest_list.len()); println!("append existing chunks ({})", digest_list.len());
let param = json!({ "wid": wid, "digest-list": digest_list }); let param = json!({ "wid": wid, "digest-list": digest_list });
request = H2Client::request_builder("localhost", "PUT", "dynamic_index", Some(param)).unwrap(); request = H2Client::request_builder("localhost", "PUT", "dynamic_index", None).unwrap();
upload_data = None; request.headers_mut().insert(hyper::header::CONTENT_TYPE, HeaderValue::from_static("application/json"));
let param_data = bytes::Bytes::from(param.to_string().as_bytes());
upload_data = Some(param_data);
} }
} }
@ -800,12 +802,14 @@ impl H2Client {
}) })
} }
// Note: We always encode parameters with the url
pub fn request_builder(server: &str, method: &str, path: &str, data: Option<Value>) -> Result<Request<()>, Error> { pub fn request_builder(server: &str, method: &str, path: &str, data: Option<Value>) -> Result<Request<()>, Error> {
let path = path.trim_matches('/'); let path = path.trim_matches('/');
let url: Uri = format!("https://{}:8007/{}", server, path).parse()?;
if let Some(data) = data { if let Some(data) = data {
let query = tools::json_object_to_query(data)?; let query = tools::json_object_to_query(data)?;
// We detected problem with hyper around 6000 characters - seo we try to keep on the safe side
if query.len() > 4096 { bail!("h2 query data too large ({} bytes) - please encode data inside body", query.len()); }
let url: Uri = format!("https://{}:8007/{}?{}", server, path, query).parse()?; let url: Uri = format!("https://{}:8007/{}?{}", server, path, query).parse()?;
let request = Request::builder() let request = Request::builder()
.method(method) .method(method)
@ -814,8 +818,8 @@ impl H2Client {
.header(hyper::header::CONTENT_TYPE, "application/x-www-form-urlencoded") .header(hyper::header::CONTENT_TYPE, "application/x-www-form-urlencoded")
.body(())?; .body(())?;
return Ok(request); return Ok(request);
} } else {
let url: Uri = format!("https://{}:8007/{}", server, path).parse()?;
let request = Request::builder() let request = Request::builder()
.method(method) .method(method)
.uri(url) .uri(url)
@ -826,3 +830,4 @@ impl H2Client {
Ok(request) Ok(request)
} }
} }
}