Skip to content

Commit

Permalink
fix: call update_connection_history for proxified connections
Browse files Browse the repository at this point in the history
  • Loading branch information
link2xt committed Oct 20, 2024
1 parent 1b82470 commit 7ac39ba
Show file tree
Hide file tree
Showing 3 changed files with 10 additions and 0 deletions.
1 change: 1 addition & 0 deletions src/imap/client.rs
Original file line number Diff line number Diff line change
Expand Up @@ -180,6 +180,7 @@ impl Client {
Client::connect_insecure_proxy(context, host, port, proxy_config).await?
}
};
update_connection_history(context, "imap", host, port, host, time()).await?;
Ok(client)
} else {
let load_cache = match security {
Expand Down
8 changes: 8 additions & 0 deletions src/net.rs
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,14 @@ pub(crate) async fn prune_connection_history(context: &Context) -> Result<()> {
Ok(())
}

/// Update the timestamp of the last successfull connection
/// to the given `host` and `port`
/// with the given application protocol `alpn`.
///
/// `addr` is the string representation of IP address.
/// If connection is made over a proxy which does
/// its own DNS resolution,
/// `addr` should be the same as `host`.
pub(crate) async fn update_connection_history(
context: &Context,
alpn: &str,
Expand Down
1 change: 1 addition & 0 deletions src/smtp/connect.rs
Original file line number Diff line number Diff line change
Expand Up @@ -168,6 +168,7 @@ async fn connect_stream(
connect_insecure_proxy(context, host, port, proxy_config.clone()).await?
}
};
update_connection_history(context, "smtp", host, port, host, time()).await?;
Ok(stream)
} else {
let load_cache = match security {
Expand Down

0 comments on commit 7ac39ba

Please sign in to comment.