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

http_protocol: allow announce/scrape url suffix; use in load test

Browse source
This commit is contained in:
Joakim Frostegård 2022-04-03 19:17:57 +02:00
parent 05faa10f4b
commit b2a3c034b0
3 changed files with 15 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

18
aquatic_http_protocol/src/request.rs
View file

@ -22,8 +22,10 @@ pub struct AnnounceRequest {
}
impl AnnounceRequest {
fn write<W: Write>(&self, output: &mut W) -> ::std::io::Result<()> {
output.write_all(b"GET /announce?info_hash=")?;
fn write<W: Write>(&self, output: &mut W, url_suffix: &[u8]) -> ::std::io::Result<()> {
output.write_all(b"GET /announce")?;
output.write_all(url_suffix)?;
output.write_all(b"?info_hash=")?;
urlencode_20_bytes(self.info_hash.0, output)?;
output.write_all(b"&peer_id=")?;
@ -167,8 +169,10 @@ pub struct ScrapeRequest {
}
impl ScrapeRequest {
fn write<W: Write>(&self, output: &mut W) -> ::std::io::Result<()> {
fn write<W: Write>(&self, output: &mut W, url_suffix: &[u8]) -> ::std::io::Result<()> {
output.write_all(b"GET /scrape?")?;
output.write_all(url_suffix)?;
output.write_all(b"?")?;
let mut first = true;
@ -307,10 +311,10 @@ impl Request {
}
}
pub fn write<W: Write>(&self, output: &mut W) -> ::std::io::Result<()> {
pub fn write<W: Write>(&self, output: &mut W, url_suffix: &[u8]) -> ::std::io::Result<()> {
match self {
Self::Announce(r) => r.write(output),
Self::Scrape(r) => r.write(output),
Self::Announce(r) => r.write(output, url_suffix),
Self::Scrape(r) => r.write(output, url_suffix),
}
}
}
@ -436,7 +440,7 @@ mod tests {
let mut bytes = Vec::new();
request.write(&mut bytes).unwrap();
request.write(&mut bytes, &[]).unwrap();
let parsed_request = Request::from_bytes(&bytes[..]).unwrap();