From: Dominik Csapak <d.csapak@proxmox.com>
To: Dietmar Maurer <dietmar@proxmox.com>, pbs-devel@lists.proxmox.com
Subject: Re: [pbs-devel] [PATCH proxmox-backup v3] client/http_client: add necessary brackets
Date: Wed, 5 May 2021 11:07:19 +0200 [thread overview]
Message-ID: <1371080a-1d81-f6b3-663a-07f81900119d@proxmox.com> (raw)
In-Reply-To: <20210505083654.16303-1-dietmar@proxmox.com>
yes, looks way better than my approach
On 5/5/21 10:36, Dietmar Maurer wrote:
> if we are given a 'naked' ipv6 without square brackets around it,
> we need to add them ourselves, since the address is ambigious otherwise
> when we add the port.
>
> e.g. giving 'fe80::1' as address we arrive at the url (with the default port)
> 'https://fe80::1:8007/'
>
> Signed-off-by: Dominik Csapak <d.csapak@proxmox.com>
> Signed-off-by: Dietmar Maurer <dietmar@proxmox.com>
> ---
>
> changes from v2:
> * add build_uri() method which returns an Uri
> * try to avoid unnecessary allocation (result in some code duplication)
>
> changes from v1:
> * move the actual mapping to the request building functions
>
> src/client/http_client.rs | 106 +++++++++++++++++++++-----------------
> 1 file changed, 58 insertions(+), 48 deletions(-)
>
> diff --git a/src/client/http_client.rs b/src/client/http_client.rs
> index 76ab0391..f2fada23 100644
> --- a/src/client/http_client.rs
> +++ b/src/client/http_client.rs
> @@ -273,6 +273,26 @@ fn load_ticket_info(prefix: &str, server: &str, userid: &Userid) -> Option<(Stri
> }
> }
>
> +fn build_uri(server: &str, port: u16, path: &str, query: Option<String>) -> Result<Uri, Error> {
> + let path = path.trim_matches('/');
> + let bytes = server.as_bytes();
> + let len = bytes.len();
> + let uri = if len > 3 && bytes.contains(&b':') && bytes[0] != b'[' && bytes[len-1] != b']' {
> + if let Some(query) = query {
> + format!("https://[{}]:{}/{}?{}", server, port, path, query)
> + } else {
> + format!("https://[{}]:{}/{}", server, port, path)
> + }
> + } else {
> + if let Some(query) = query {
> + format!("https://{}:{}/{}?{}", server, port, path, query)
> + } else {
> + format!("https://{}:{}/{}", server, port, path)
> + }
> + };
> + Ok(uri.parse()?)
> +}
> +
> impl HttpClient {
> pub fn new(
> server: &str,
> @@ -614,16 +634,11 @@ impl HttpClient {
> data: Option<Value>,
> ) -> Result<Value, Error> {
>
> - let path = path.trim_matches('/');
> - let mut url = format!("https://{}:{}/{}", &self.server, self.port, path);
> -
> - if let Some(data) = data {
> - let query = tools::json_object_to_query(data).unwrap();
> - url.push('?');
> - url.push_str(&query);
> - }
> -
> - let url: Uri = url.parse().unwrap();
> + let query = match data {
> + Some(data) => Some(tools::json_object_to_query(data)?),
> + None => None,
> + };
> + let url = build_uri(&self.server, self.port, path, query)?;
>
> let req = Request::builder()
> .method("POST")
> @@ -757,39 +772,38 @@ impl HttpClient {
> }
>
> pub fn request_builder(server: &str, port: u16, method: &str, path: &str, data: Option<Value>) -> Result<Request<Body>, Error> {
> - let path = path.trim_matches('/');
> - let url: Uri = format!("https://{}:{}/{}", server, port, path).parse()?;
> -
> if let Some(data) = data {
> if method == "POST" {
> + let url = build_uri(server, port, path, None)?;
> let request = Request::builder()
> .method(method)
> .uri(url)
> .header("User-Agent", "proxmox-backup-client/1.0")
> .header(hyper::header::CONTENT_TYPE, "application/json")
> .body(Body::from(data.to_string()))?;
> - return Ok(request);
> + Ok(request)
> } else {
> let query = tools::json_object_to_query(data)?;
> - let url: Uri = format!("https://{}:{}/{}?{}", server, port, path, query).parse()?;
> + let url = build_uri(server, port, path, Some(query))?;
> let request = Request::builder()
> .method(method)
> .uri(url)
> .header("User-Agent", "proxmox-backup-client/1.0")
> .header(hyper::header::CONTENT_TYPE, "application/x-www-form-urlencoded")
> .body(Body::empty())?;
> - return Ok(request);
> + Ok(request)
> }
> - }
> -
> - let request = Request::builder()
> - .method(method)
> - .uri(url)
> - .header("User-Agent", "proxmox-backup-client/1.0")
> - .header(hyper::header::CONTENT_TYPE, "application/x-www-form-urlencoded")
> - .body(Body::empty())?;
> + } else {
> + let url = build_uri(server, port, path, None)?;
> + let request = Request::builder()
> + .method(method)
> + .uri(url)
> + .header("User-Agent", "proxmox-backup-client/1.0")
> + .header(hyper::header::CONTENT_TYPE, "application/x-www-form-urlencoded")
> + .body(Body::empty())?;
>
> - Ok(request)
> + Ok(request)
> + }
> }
> }
>
> @@ -970,29 +984,25 @@ impl H2Client {
> let path = path.trim_matches('/');
>
> let content_type = content_type.unwrap_or("application/x-www-form-urlencoded");
> + let query = match param {
> + Some(param) => {
> + let query = tools::json_object_to_query(param)?;
> + // We detected problem with hyper around 6000 characters - so 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());
> + }
> + Some(query)
> + }
> + None => None,
> + };
>
> - if let Some(param) = param {
> - let query = tools::json_object_to_query(param)?;
> - // 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 request = Request::builder()
> - .method(method)
> - .uri(url)
> - .header("User-Agent", "proxmox-backup-client/1.0")
> - .header(hyper::header::CONTENT_TYPE, content_type)
> - .body(())?;
> - Ok(request)
> - } else {
> - let url: Uri = format!("https://{}:8007/{}", server, path).parse()?;
> - let request = Request::builder()
> - .method(method)
> - .uri(url)
> - .header("User-Agent", "proxmox-backup-client/1.0")
> - .header(hyper::header::CONTENT_TYPE, content_type)
> - .body(())?;
> -
> - Ok(request)
> - }
> + let url = build_uri(server, 8007, path, query)?;
> + let request = Request::builder()
> + .method(method)
> + .uri(url)
> + .header("User-Agent", "proxmox-backup-client/1.0")
> + .header(hyper::header::CONTENT_TYPE, content_type)
> + .body(())?;
> + Ok(request)
> }
> }
>
next prev parent reply other threads:[~2021-05-05 9:07 UTC|newest]
Thread overview: 3+ messages / expand[flat|nested] mbox.gz Atom feed top
2021-05-05 8:36 Dietmar Maurer
2021-05-05 9:07 ` Dominik Csapak [this message]
2021-05-05 9:09 ` [pbs-devel] applied: " Dietmar Maurer
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-all from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
--in-reply-to=1371080a-1d81-f6b3-663a-07f81900119d@proxmox.com \
--to=d.csapak@proxmox.com \
--cc=dietmar@proxmox.com \
--cc=pbs-devel@lists.proxmox.com \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox