libdd-common 1.0.0

Shared utilities for Datadog libraries including HTTP/HTTPS connectors, container entity detection, tag validation, rate limiting, and Unix/Windows platform helpers
Documentation 
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
51
52
53
54

// Copyright 2021-Present Datadog, Inc. https://www.datadoghq.com/
// SPDX-License-Identifier: Apache-2.0

use std::ffi::OsString;
use std::os::unix::ffi::{OsStrExt, OsStringExt};
use std::path::{Path, PathBuf};

/// Creates a new Uri, with the `unix` scheme, and the path to the socket
/// encoded as a hex string, to prevent special characters in the url authority
pub fn socket_path_to_uri(path: &Path) -> Result<hyper::Uri, hyper::http::Error> {
    let path = hex::encode(path.as_os_str().as_bytes());
    hyper::Uri::builder()
        .scheme("unix")
        .authority(path)
        .path_and_query("")
        .build()
}

pub fn socket_path_from_uri(uri: &hyper::Uri) -> anyhow::Result<PathBuf> {
    if uri.scheme_str() != Some("unix") {
        return Err(super::errors::Error::InvalidUrl.into());
    }

    let path = hex::decode(
        uri.authority()
            .ok_or(super::errors::Error::InvalidUrl)?
            .as_str(),
    )
    .map_err(|_| super::errors::Error::InvalidUrl)?;
    Ok(PathBuf::from(OsString::from_vec(path)))
}

#[test]
fn test_encode_unix_socket_path_absolute() {
    let expected_path = "/path/to/a/socket.sock".as_ref();
    let uri = socket_path_to_uri(expected_path).unwrap();
    assert_eq!(uri.scheme_str(), Some("unix"));

    let actual_path = socket_path_from_uri(&uri).unwrap();
    assert_eq!(actual_path.as_path(), Path::new(expected_path))
}

#[test]
fn test_encode_unix_socket_relative_path() {
    let expected_path = "relative/path/to/a/socket.sock".as_ref();
    let uri = socket_path_to_uri(expected_path).unwrap();
    let actual_path = socket_path_from_uri(&uri).unwrap();
    assert_eq!(actual_path.as_path(), Path::new(expected_path));

    let expected_path = "./relative/path/to/a/socket.sock".as_ref();
    let uri = socket_path_to_uri(expected_path).unwrap();
    let actual_path = socket_path_from_uri(&uri).unwrap();
    assert_eq!(actual_path.as_path(), Path::new(expected_path));
}