netbird/IronRDP
0
0
Fork 0
mirror of https://github.com/netbirdio/IronRDP.git synced 2026-05-22 18:43:12 -07:00
Code Issues Packages Projects Releases Wiki Activity

feat(ironrdp-tokio): add async ReqwestNetworkClient::send method (#859)

Browse Source
This commit is contained in:
Pavlo Myroniuk
2025-07-08 05:57:43 -04:00
committed by GitHub
parent 4dc5945019
commit 7e23a8bb97
1 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/ironrdp-tokio/src/reqwest.rs
+11 -9
View File
@@ -20,15 +20,7 @@ impl AsyncNetworkClient for ReqwestNetworkClient {
&'a mut self,
request: &'a sspi::generator::NetworkRequest,
) -> Pin<Box<dyn Future<Output = ConnectorResult<Vec<u8>>> + 'a>> {
Box::pin(async move {
match &request.protocol {
sspi::network_client::NetworkProtocol::Tcp => self.send_tcp(&request.url, &request.data).await,
sspi::network_client::NetworkProtocol::Udp => self.send_udp(&request.url, &request.data).await,
sspi::network_client::NetworkProtocol::Http | sspi::network_client::NetworkProtocol::Https => {
self.send_http(&request.url, &request.data).await
}
}
})
Box::pin(ReqwestNetworkClient::send(self, request))
}
}
@@ -45,6 +37,16 @@ impl Default for ReqwestNetworkClient {
}
impl ReqwestNetworkClient {
pub async fn send<'a>(&'a mut self, request: &'a sspi::generator::NetworkRequest) -> ConnectorResult<Vec<u8>> {
match &request.protocol {
sspi::network_client::NetworkProtocol::Tcp => self.send_tcp(&request.url, &request.data).await,
sspi::network_client::NetworkProtocol::Udp => self.send_udp(&request.url, &request.data).await,
sspi::network_client::NetworkProtocol::Http | sspi::network_client::NetworkProtocol::Https => {
self.send_http(&request.url, &request.data).await
}
}
}
async fn send_tcp(&self, url: &Url, data: &[u8]) -> ConnectorResult<Vec<u8>> {
let addr = format!("{}:{}", url.host_str().unwrap_or_default(), url.port().unwrap_or(88));