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
55
56
57
58
59
60
61
62

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

use std::path::{Path, PathBuf};

/// Windows Named Pipe
/// https://docs.microsoft.com/en-us/windows/win32/ipc/named-pipes
///
/// The form a windows named pipe path is either local to the computer:
/// \\.\pipe\pipename
/// or targeting a remote server
/// \\ServerName\pipe\pipename
///
/// Build a URI from a Path representing a named pipe
/// `path` - named pipe path. ex: \\.\pipe\pipename
pub fn named_pipe_path_to_uri(path: &Path) -> Result<hyper::Uri, hyper::http::Error> {
    #[allow(clippy::unwrap_used)]
    let path = hex::encode(path.as_os_str().to_str().unwrap());
    hyper::Uri::builder()
        .scheme("windows")
        .authority(path)
        .path_and_query("")
        .build()
}

pub fn named_pipe_path_from_uri(uri: &hyper::Uri) -> anyhow::Result<PathBuf> {
    if uri.scheme_str() != Some("windows") {
        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)?;

    match String::from_utf8(path) {
        Ok(s) => Ok(PathBuf::from(s.as_str())),
        _ => Err(super::errors::Error::InvalidUrl.into()),
    }
}

#[test]
fn test_encode_named_pipe_for_local_server() {
    let expected_path = r"\\.\pipe\pipename".as_ref();
    let uri = named_pipe_path_to_uri(expected_path).unwrap();
    assert_eq!(uri.scheme_str(), Some("windows"));

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

#[test]
fn test_encode_named_pipe_for_remote_server() {
    let expected_path = r"\\servername\pipe\pipename".as_ref();
    let uri = named_pipe_path_to_uri(expected_path).unwrap();
    assert_eq!(uri.scheme_str(), Some("windows"));

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