YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-03-31 17:55:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

udp: invalid connection id integration test: test scrape request too

Browse source
This commit is contained in:
Joakim Frostegård 2023-08-28 23:30:46 +02:00
parent 15051ec8f1
commit 227bf6163c
1 changed files with 38 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

56
aquatic_udp/tests/invalid_connection_id.rs
View file

@ -12,11 +12,11 @@ use anyhow::Context;
use aquatic_udp::{common::BUFFER_SIZE, config::Config}; use aquatic_udp::{common::BUFFER_SIZE, config::Config};
use aquatic_udp_protocol::{ use aquatic_udp_protocol::{
common::PeerId, AnnounceEvent, AnnounceRequest, ConnectionId, InfoHash, NumberOfBytes, common::PeerId, AnnounceEvent, AnnounceRequest, ConnectionId, InfoHash, NumberOfBytes,
NumberOfPeers, PeerKey, Port, Request, TransactionId, NumberOfPeers, PeerKey, Port, Request, ScrapeRequest, TransactionId,
}; };
#[test] #[test]
fn test_announce_with_invalid_connection_id() -> anyhow::Result<()> { fn test_invalid_connection_id() -> anyhow::Result<()> {
const TRACKER_PORT: u16 = 40_112; const TRACKER_PORT: u16 = 40_112;
let mut config = Config::default(); let mut config = Config::default();
@ -32,29 +32,49 @@ fn test_announce_with_invalid_connection_id() -> anyhow::Result<()> {
socket.set_read_timeout(Some(Duration::from_secs(1)))?; socket.set_read_timeout(Some(Duration::from_secs(1)))?;
// Make sure that the tracker in fact responds to requests // Send connect request to make sure that the tracker in fact responds to
// valid requests
let connection_id = connect(&socket, tracker_addr).with_context(|| "connect")?; let connection_id = connect(&socket, tracker_addr).with_context(|| "connect")?;
let invalid_connection_id = ConnectionId(!connection_id.0);
let announce_request = Request::Announce(AnnounceRequest {
connection_id: invalid_connection_id,
transaction_id: TransactionId(0),
info_hash: InfoHash([0; 20]),
peer_id: PeerId([0; 20]),
bytes_downloaded: NumberOfBytes(0),
bytes_uploaded: NumberOfBytes(0),
bytes_left: NumberOfBytes(0),
event: AnnounceEvent::Started,
ip_address: None,
key: PeerKey(0),
peers_wanted: NumberOfPeers(10),
port: Port(1),
});
let scrape_request = Request::Scrape(ScrapeRequest {
connection_id: invalid_connection_id,
transaction_id: TransactionId(0),
info_hashes: vec![InfoHash([0; 20])],
});
no_response(&socket, tracker_addr, announce_request).with_context(|| "announce")?;
no_response(&socket, tracker_addr, scrape_request).with_context(|| "scrape")?;
Ok(())
}
fn no_response(
socket: &UdpSocket,
tracker_addr: SocketAddr,
request: Request,
) -> anyhow::Result<()> {
let mut buffer = [0u8; BUFFER_SIZE]; let mut buffer = [0u8; BUFFER_SIZE];
{ {
let mut buffer = Cursor::new(&mut buffer[..]); let mut buffer = Cursor::new(&mut buffer[..]);
let request = Request::Announce(AnnounceRequest {
connection_id: ConnectionId(!connection_id.0),
transaction_id: TransactionId(0),
info_hash: InfoHash([0; 20]),
peer_id: PeerId([0; 20]),
bytes_downloaded: NumberOfBytes(0),
bytes_uploaded: NumberOfBytes(0),
bytes_left: NumberOfBytes(0),
event: AnnounceEvent::Started,
ip_address: None,
key: PeerKey(0),
peers_wanted: NumberOfPeers(-1),
port: Port(1),
});
request request
.write(&mut buffer) .write(&mut buffer)
.with_context(|| "write request")?; .with_context(|| "write request")?;