YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-04-01 10:15:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

http_private: use CanonicalSocketAddr when calling DB

Browse source
This commit is contained in:
Joakim Frostegård 2022-04-03 00:39:06 +02:00
parent 96e128bb90
commit 5d4f30bb7b
2 changed files with 11 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

11
aquatic_http_private/src/workers/socket/db.rs
View file

@ -1,5 +1,6 @@
use std::net::{IpAddr, SocketAddr};
use std::net::IpAddr;
use aquatic_common::CanonicalSocketAddr;
use aquatic_http_protocol::{
common::AnnounceEvent, request::AnnounceRequest, response::FailureResponse,
};
@ -30,14 +31,14 @@ struct AnnounceProcedureParameters {
impl AnnounceProcedureParameters {
fn new(
source_addr: SocketAddr,
source_addr: CanonicalSocketAddr,
user_agent: Option<String>,
user_token: String, // FIXME: length
request: &AnnounceRequest,
) -> Self {
Self {
source_ip: source_addr.ip(),
source_port: source_addr.port(),
source_ip: source_addr.get().ip(),
source_port: source_addr.get().port(),
user_agent,
user_token,
info_hash: hex::encode(request.info_hash.0),
@ -59,7 +60,7 @@ struct AnnounceProcedureResults {
pub async fn validate_announce_request(
pool: &Pool<MySql>,
source_addr: SocketAddr,
source_addr: CanonicalSocketAddr,
user_agent: Option<String>,
user_token: String,
request: AnnounceRequest,