Fix esp-mbedtls support after esp-rs/esp-mbedtls@7de5d0a

src/client.rs

@@ -27,7 +27,7 @@ where
 
 /// Type for TLS configuration of HTTP client.
 #[cfg(feature = "esp-mbedtls")]
-pub struct TlsConfig<'a> {
+pub struct TlsConfig<'a, const RX_SIZE: usize = 4096, const TX_SIZE: usize = 4096> {
  /// Minimum TLS version for the connection
  version: crate::TlsVersion,
 
@@ -36,6 +36,12 @@ pub struct TlsConfig<'a> {
 
  /// Will use hardware acceleration on the ESP32 if it contains the RSA peripheral.
  rsa: Option<&'a mut esp_mbedtls::hal::peripherals::RSA>,
+
+ /// Buffer for the reading side of the TLS connection
+ read_buffer: &'a mut [u8; RX_SIZE],
+
+ /// Buffer for the writing side of the TLS connection
+ write_buffer: &'a mut [u8; TX_SIZE],
 }
 
 /// Type for TLS configuration of HTTP client.
@@ -69,8 +75,10 @@ impl<'a> TlsConfig<'a> {
 }
 
 #[cfg(feature = "esp-mbedtls")]
-impl<'a> TlsConfig<'a> {
+impl<'a, const RX_SIZE: usize, const TX_SIZE: usize> TlsConfig<'a, RX_SIZE, TX_SIZE> {
  pub fn new(
+ read_buffer: &'a mut [u8; RX_SIZE],
+ write_buffer: &'a mut [u8; TX_SIZE],
  version: crate::TlsVersion,
  certificates: crate::Certificates<'a>,
  rsa: Option<&'a mut esp_mbedtls::hal::peripherals::RSA>,
@@ -79,6 +87,8 @@ impl<'a> TlsConfig<'a> {
  version,
  certificates,
  rsa,
+ read_buffer,
+ write_buffer,
  }
  }
 }
@@ -130,18 +140,21 @@ where
  if url.scheme() == UrlScheme::HTTPS {
  #[cfg(feature = "esp-mbedtls")]
  if let Some(tls) = self.tls.as_mut() {
- let session = esp_mbedtls::asynch::Session::new(
+ let mut session = esp_mbedtls::asynch::Session::new(
  conn,
  host,
  esp_mbedtls::Mode::Client,
  tls.version,
  tls.certificates,
- // Create a inner Some(&mut Rsa) because Rsa doesn't implement Copy
- tls.rsa.as_mut().map(|p| p as &mut esp_mbedtls::hal::peripherals::RSA),
- )?
- .connect()
- .await?;
- Ok(HttpConnection::Tls(session))
+ tls.read_buffer,
+ tls.write_buffer,
+ )?;
+
+ if let Some(rsa) = tls.rsa.as_mut() {
+ session = session.with_hardware_rsa(rsa as &mut esp_mbedtls::hal::peripherals::RSA);
+ }
+
+ Ok(HttpConnection::Tls(session.connect().await?))
  } else {
  Ok(HttpConnection::Plain(conn))
  }
@@ -220,7 +233,7 @@ where
  Plain(C),
  PlainBuffered(BufferedWrite<'conn, C>),
  #[cfg(feature = "esp-mbedtls")]
- Tls(esp_mbedtls::asynch::AsyncConnectedSession<C, 4096>),
+ Tls(esp_mbedtls::asynch::AsyncConnectedSession<'conn, C, 4096, 4096>),
  #[cfg(feature = "embedded-tls")]
  Tls(embedded_tls::TlsConnection<'conn, C, embedded_tls::Aes128GcmSha256>),
  #[cfg(all(not(feature = "embedded-tls"), not(feature = "esp-mbedtls")))]