initial commit
This commit is contained in:
commit
b32f00499a
2
.gitignore
vendored
Normal file
2
.gitignore
vendored
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
.vscode/
|
||||||
|
target/
|
||||||
558
Cargo.lock
generated
Normal file
558
Cargo.lock
generated
Normal file
@ -0,0 +1,558 @@
|
|||||||
|
# This file is automatically @generated by Cargo.
|
||||||
|
# It is not intended for manual editing.
|
||||||
|
version = 4
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "anyhow"
|
||||||
|
version = "1.0.98"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "e16d2d3311acee920a9eb8d33b8cbc1787ce4a264e85f964c2404b969bdcd487"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "autocfg"
|
||||||
|
version = "1.5.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "bitflags"
|
||||||
|
version = "2.9.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1b8e56985ec62d17e9c1001dc89c88ecd7dc08e47eba5ec7c29c7b5eeecde967"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "bumpalo"
|
||||||
|
version = "3.19.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "46c5e41b57b8bba42a04676d81cb89e9ee8e859a1a66f80a5a72e1cb76b34d43"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "byteorder"
|
||||||
|
version = "1.5.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1fd0f2584146f6f2ef48085050886acf353beff7305ebd1ae69500e27c67f64b"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "cfg-if"
|
||||||
|
version = "1.0.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "9555578bc9e57714c812a1f84e4fc5b4d21fcb063490c624de019f7464c91268"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "deranged"
|
||||||
|
version = "0.4.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "9c9e6a11ca8224451684bc0d7d5a7adbf8f2fd6887261a1cfc3c0432f9d4068e"
|
||||||
|
dependencies = [
|
||||||
|
"powerfmt",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "flume"
|
||||||
|
version = "0.11.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "da0e4dd2a88388a1f4ccc7c9ce104604dab68d9f408dc34cd45823d5a9069095"
|
||||||
|
dependencies = [
|
||||||
|
"futures-core",
|
||||||
|
"futures-sink",
|
||||||
|
"nanorand",
|
||||||
|
"spin",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "futures-core"
|
||||||
|
version = "0.3.31"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "05f29059c0c2090612e8d742178b0580d2dc940c837851ad723096f87af6663e"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "futures-sink"
|
||||||
|
version = "0.3.31"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "e575fab7d1e0dcb8d0c7bcf9a63ee213816ab51902e6d244a95819acacf1d4f7"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "getrandom"
|
||||||
|
version = "0.2.16"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "335ff9f135e4384c8150d6f27c6daed433577f86b4750418338c01a1a2528592"
|
||||||
|
dependencies = [
|
||||||
|
"cfg-if",
|
||||||
|
"js-sys",
|
||||||
|
"libc",
|
||||||
|
"wasi 0.11.1+wasi-snapshot-preview1",
|
||||||
|
"wasm-bindgen",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "getrandom"
|
||||||
|
version = "0.3.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "26145e563e54f2cadc477553f1ec5ee650b00862f0a58bcd12cbdc5f0ea2d2f4"
|
||||||
|
dependencies = [
|
||||||
|
"cfg-if",
|
||||||
|
"libc",
|
||||||
|
"r-efi",
|
||||||
|
"wasi 0.14.2+wasi-0.2.4",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "hash32"
|
||||||
|
version = "0.3.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "47d60b12902ba28e2730cd37e95b8c9223af2808df9e902d4df49588d1470606"
|
||||||
|
dependencies = [
|
||||||
|
"byteorder",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "heapless"
|
||||||
|
version = "0.8.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "0bfb9eb618601c89945a70e254898da93b13be0388091d42117462b265bb3fad"
|
||||||
|
dependencies = [
|
||||||
|
"hash32",
|
||||||
|
"stable_deref_trait",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "itoa"
|
||||||
|
version = "1.0.15"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "4a5f13b858c8d314ee3e8f639011f7ccefe71f97f96e50151fb991f267928e2c"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "js-sys"
|
||||||
|
version = "0.3.77"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1cfaf33c695fc6e08064efbc1f72ec937429614f25eef83af942d0e227c3a28f"
|
||||||
|
dependencies = [
|
||||||
|
"once_cell",
|
||||||
|
"wasm-bindgen",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "libc"
|
||||||
|
version = "0.2.174"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1171693293099992e19cddea4e8b849964e9846f4acee11b3948bcc337be8776"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "lock_api"
|
||||||
|
version = "0.4.13"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "96936507f153605bddfcda068dd804796c84324ed2510809e5b2a624c81da765"
|
||||||
|
dependencies = [
|
||||||
|
"autocfg",
|
||||||
|
"scopeguard",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "log"
|
||||||
|
version = "0.4.27"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "13dc2df351e3202783a1fe0d44375f7295ffb4049267b0f3018346dc122a1d94"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "meteoro"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"anyhow",
|
||||||
|
"flume",
|
||||||
|
"heapless",
|
||||||
|
"log",
|
||||||
|
"rand",
|
||||||
|
"simplelog",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "nanorand"
|
||||||
|
version = "0.7.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "6a51313c5820b0b02bd422f4b44776fbf47961755c74ce64afc73bfad10226c3"
|
||||||
|
dependencies = [
|
||||||
|
"getrandom 0.2.16",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "num-conv"
|
||||||
|
version = "0.1.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "51d515d32fb182ee37cda2ccdcb92950d6a3c2893aa280e540671c2cd0f3b1d9"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "num_threads"
|
||||||
|
version = "0.1.7"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "5c7398b9c8b70908f6371f47ed36737907c87c52af34c268fed0bf0ceb92ead9"
|
||||||
|
dependencies = [
|
||||||
|
"libc",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "once_cell"
|
||||||
|
version = "1.21.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "42f5e15c9953c5e4ccceeb2e7382a716482c34515315f7b03532b8b4e8393d2d"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "powerfmt"
|
||||||
|
version = "0.2.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "439ee305def115ba05938db6eb1644ff94165c5ab5e9420d1c1bcedbba909391"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "ppv-lite86"
|
||||||
|
version = "0.2.21"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "85eae3c4ed2f50dcfe72643da4befc30deadb458a9b590d720cde2f2b1e97da9"
|
||||||
|
dependencies = [
|
||||||
|
"zerocopy",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "proc-macro2"
|
||||||
|
version = "1.0.95"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "02b3e5e68a3a1a02aad3ec490a98007cbc13c37cbe84a3cd7b8e406d76e7f778"
|
||||||
|
dependencies = [
|
||||||
|
"unicode-ident",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "quote"
|
||||||
|
version = "1.0.40"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1885c039570dc00dcb4ff087a89e185fd56bae234ddc7f056a945bf36467248d"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "r-efi"
|
||||||
|
version = "5.3.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "69cdb34c158ceb288df11e18b4bd39de994f6657d83847bdffdbd7f346754b0f"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rand"
|
||||||
|
version = "0.9.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "6db2770f06117d490610c7488547d543617b21bfa07796d7a12f6f1bd53850d1"
|
||||||
|
dependencies = [
|
||||||
|
"rand_chacha",
|
||||||
|
"rand_core",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rand_chacha"
|
||||||
|
version = "0.9.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "d3022b5f1df60f26e1ffddd6c66e8aa15de382ae63b3a0c1bfc0e4d3e3f325cb"
|
||||||
|
dependencies = [
|
||||||
|
"ppv-lite86",
|
||||||
|
"rand_core",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rand_core"
|
||||||
|
version = "0.9.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "99d9a13982dcf210057a8a78572b2217b667c3beacbf3a0d8b454f6f82837d38"
|
||||||
|
dependencies = [
|
||||||
|
"getrandom 0.3.3",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "scopeguard"
|
||||||
|
version = "1.2.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "94143f37725109f92c262ed2cf5e59bce7498c01bcc1502d7b9afe439a4e9f49"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "serde"
|
||||||
|
version = "1.0.219"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "5f0e2c6ed6606019b4e29e69dbaba95b11854410e5347d525002456dbbb786b6"
|
||||||
|
dependencies = [
|
||||||
|
"serde_derive",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "serde_derive"
|
||||||
|
version = "1.0.219"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "5b0276cf7f2c73365f7157c8123c21cd9a50fbbd844757af28ca1f5925fc2a00"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "simplelog"
|
||||||
|
version = "0.12.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "16257adbfaef1ee58b1363bdc0664c9b8e1e30aed86049635fb5f147d065a9c0"
|
||||||
|
dependencies = [
|
||||||
|
"log",
|
||||||
|
"termcolor",
|
||||||
|
"time",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "spin"
|
||||||
|
version = "0.9.8"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "6980e8d7511241f8acf4aebddbb1ff938df5eebe98691418c4468d0b72a96a67"
|
||||||
|
dependencies = [
|
||||||
|
"lock_api",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "stable_deref_trait"
|
||||||
|
version = "1.2.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "a8f112729512f8e442d81f95a8a7ddf2b7c6b8a1a6f509a95864142b30cab2d3"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "syn"
|
||||||
|
version = "2.0.104"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "17b6f705963418cdb9927482fa304bc562ece2fdd4f616084c50b7023b435a40"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"unicode-ident",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "termcolor"
|
||||||
|
version = "1.4.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "06794f8f6c5c898b3275aebefa6b8a1cb24cd2c6c79397ab15774837a0bc5755"
|
||||||
|
dependencies = [
|
||||||
|
"winapi-util",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "time"
|
||||||
|
version = "0.3.41"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8a7619e19bc266e0f9c5e6686659d394bc57973859340060a69221e57dbc0c40"
|
||||||
|
dependencies = [
|
||||||
|
"deranged",
|
||||||
|
"itoa",
|
||||||
|
"libc",
|
||||||
|
"num-conv",
|
||||||
|
"num_threads",
|
||||||
|
"powerfmt",
|
||||||
|
"serde",
|
||||||
|
"time-core",
|
||||||
|
"time-macros",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "time-core"
|
||||||
|
version = "0.1.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "c9e9a38711f559d9e3ce1cdb06dd7c5b8ea546bc90052da6d06bb76da74bb07c"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "time-macros"
|
||||||
|
version = "0.2.22"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "3526739392ec93fd8b359c8e98514cb3e8e021beb4e5f597b00a0221f8ed8a49"
|
||||||
|
dependencies = [
|
||||||
|
"num-conv",
|
||||||
|
"time-core",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "unicode-ident"
|
||||||
|
version = "1.0.18"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "5a5f39404a5da50712a4c1eecf25e90dd62b613502b7e925fd4e4d19b5c96512"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasi"
|
||||||
|
version = "0.11.1+wasi-snapshot-preview1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "ccf3ec651a847eb01de73ccad15eb7d99f80485de043efb2f370cd654f4ea44b"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasi"
|
||||||
|
version = "0.14.2+wasi-0.2.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "9683f9a5a998d873c0d21fcbe3c083009670149a8fab228644b8bd36b2c48cb3"
|
||||||
|
dependencies = [
|
||||||
|
"wit-bindgen-rt",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasm-bindgen"
|
||||||
|
version = "0.2.100"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1edc8929d7499fc4e8f0be2262a241556cfc54a0bea223790e71446f2aab1ef5"
|
||||||
|
dependencies = [
|
||||||
|
"cfg-if",
|
||||||
|
"once_cell",
|
||||||
|
"wasm-bindgen-macro",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasm-bindgen-backend"
|
||||||
|
version = "0.2.100"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "2f0a0651a5c2bc21487bde11ee802ccaf4c51935d0d3d42a6101f98161700bc6"
|
||||||
|
dependencies = [
|
||||||
|
"bumpalo",
|
||||||
|
"log",
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
"wasm-bindgen-shared",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasm-bindgen-macro"
|
||||||
|
version = "0.2.100"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "7fe63fc6d09ed3792bd0897b314f53de8e16568c2b3f7982f468c0bf9bd0b407"
|
||||||
|
dependencies = [
|
||||||
|
"quote",
|
||||||
|
"wasm-bindgen-macro-support",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasm-bindgen-macro-support"
|
||||||
|
version = "0.2.100"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8ae87ea40c9f689fc23f209965b6fb8a99ad69aeeb0231408be24920604395de"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
"wasm-bindgen-backend",
|
||||||
|
"wasm-bindgen-shared",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasm-bindgen-shared"
|
||||||
|
version = "0.2.100"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1a05d73b933a847d6cccdda8f838a22ff101ad9bf93e33684f39c1f5f0eece3d"
|
||||||
|
dependencies = [
|
||||||
|
"unicode-ident",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "winapi-util"
|
||||||
|
version = "0.1.9"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "cf221c93e13a30d793f7645a0e7762c55d169dbb0a49671918a2319d289b10bb"
|
||||||
|
dependencies = [
|
||||||
|
"windows-sys",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows-sys"
|
||||||
|
version = "0.59.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1e38bc4d79ed67fd075bcc251a1c39b32a1776bbe92e5bef1f0bf1f8c531853b"
|
||||||
|
dependencies = [
|
||||||
|
"windows-targets",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows-targets"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "9b724f72796e036ab90c1021d4780d4d3d648aca59e491e6b98e725b84e99973"
|
||||||
|
dependencies = [
|
||||||
|
"windows_aarch64_gnullvm",
|
||||||
|
"windows_aarch64_msvc",
|
||||||
|
"windows_i686_gnu",
|
||||||
|
"windows_i686_gnullvm",
|
||||||
|
"windows_i686_msvc",
|
||||||
|
"windows_x86_64_gnu",
|
||||||
|
"windows_x86_64_gnullvm",
|
||||||
|
"windows_x86_64_msvc",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_aarch64_gnullvm"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "32a4622180e7a0ec044bb555404c800bc9fd9ec262ec147edd5989ccd0c02cd3"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_aarch64_msvc"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "09ec2a7bb152e2252b53fa7803150007879548bc709c039df7627cabbd05d469"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_i686_gnu"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8e9b5ad5ab802e97eb8e295ac6720e509ee4c243f69d781394014ebfe8bbfa0b"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_i686_gnullvm"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "0eee52d38c090b3caa76c563b86c3a4bd71ef1a819287c19d586d7334ae8ed66"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_i686_msvc"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "240948bc05c5e7c6dabba28bf89d89ffce3e303022809e73deaefe4f6ec56c66"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_x86_64_gnu"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "147a5c80aabfbf0c7d901cb5895d1de30ef2907eb21fbbab29ca94c5b08b1a78"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_x86_64_gnullvm"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "24d5b23dc417412679681396f2b49f3de8c1473deb516bd34410872eff51ed0d"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "windows_x86_64_msvc"
|
||||||
|
version = "0.52.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wit-bindgen-rt"
|
||||||
|
version = "0.39.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "6f42320e61fe2cfd34354ecb597f86f413484a798ba44a8ca1165c58d42da6c1"
|
||||||
|
dependencies = [
|
||||||
|
"bitflags",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "zerocopy"
|
||||||
|
version = "0.8.26"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1039dd0d3c310cf05de012d8a39ff557cb0d23087fd44cad61df08fc31907a2f"
|
||||||
|
dependencies = [
|
||||||
|
"zerocopy-derive",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "zerocopy-derive"
|
||||||
|
version = "0.8.26"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "9ecf5b4cc5364572d7f4c329661bcc82724222973f2cab6f050a4e5c22f75181"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
12
Cargo.toml
Normal file
12
Cargo.toml
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
[package]
|
||||||
|
name = "meteoro"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
anyhow = "1.0.98"
|
||||||
|
flume = { version = "0.11.1", features = ["eventual-fairness", "select"] }
|
||||||
|
heapless = "0.8.0"
|
||||||
|
log = { version = "0.4.27", features = ["release_max_level_info"] }
|
||||||
|
rand = "0.9.2"
|
||||||
|
simplelog = "0.12.2"
|
||||||
12
src/constants.rs
Normal file
12
src/constants.rs
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
pub const METEORO_HELP_TEXT: &'static str = "Usage:\tmeteoro [OPTIONS]
|
||||||
|
|
||||||
|
OPTIONS:
|
||||||
|
|
||||||
|
-a IPV4_ADDRESS Provide an IPv4 address to bind to
|
||||||
|
(example: -a 192.168.1.2:6969)
|
||||||
|
|
||||||
|
-A IPV6_ADDRESS Provide an IPv6 address to bind to
|
||||||
|
(example: -A [::1]:6970)
|
||||||
|
|
||||||
|
-h | --help Show this help text
|
||||||
|
-v | --version Print the program's build & version information";
|
||||||
89
src/main.rs
Normal file
89
src/main.rs
Normal file
@ -0,0 +1,89 @@
|
|||||||
|
use std::{
|
||||||
|
env,
|
||||||
|
net::{Ipv4Addr, Ipv6Addr, SocketAddr, SocketAddrV4, SocketAddrV6},
|
||||||
|
process::ExitCode,
|
||||||
|
};
|
||||||
|
|
||||||
|
use log::LevelFilter;
|
||||||
|
use simplelog::{ColorChoice, ConfigBuilder, TermLogger, TerminalMode};
|
||||||
|
|
||||||
|
mod constants;
|
||||||
|
mod tracker;
|
||||||
|
use tracker::Tracker;
|
||||||
|
|
||||||
|
fn main() -> ExitCode {
|
||||||
|
#[cfg(debug_assertions)]
|
||||||
|
let log_level: LevelFilter = LevelFilter::Trace;
|
||||||
|
#[cfg(not(debug_assertions))]
|
||||||
|
let log_level: LevelFilter = LevelFilter::Info;
|
||||||
|
|
||||||
|
let logger_config = ConfigBuilder::new()
|
||||||
|
.set_target_level(LevelFilter::Off)
|
||||||
|
.set_location_level(LevelFilter::Off)
|
||||||
|
.set_thread_level(LevelFilter::Off)
|
||||||
|
.set_time_level(LevelFilter::Info)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
TermLogger::init(
|
||||||
|
log_level,
|
||||||
|
logger_config,
|
||||||
|
TerminalMode::Mixed,
|
||||||
|
ColorChoice::Auto,
|
||||||
|
)
|
||||||
|
.expect("failed to initialise terminal logging");
|
||||||
|
|
||||||
|
let args: Vec<String> = env::args().collect();
|
||||||
|
|
||||||
|
if args.contains(&String::from("--help")) || args.contains(&String::from("-h")) {
|
||||||
|
println!("{}", constants::METEORO_HELP_TEXT);
|
||||||
|
return ExitCode::SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
if args.contains(&String::from("--version")) || args.contains(&String::from("-v")) {
|
||||||
|
println!("{} {}", env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION"));
|
||||||
|
return ExitCode::SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
let ipv4_flag_idxs: Vec<usize> = args
|
||||||
|
.iter()
|
||||||
|
.enumerate()
|
||||||
|
.filter_map(|(idx, s)| if s == "-a" { Some(idx) } else { None })
|
||||||
|
.collect();
|
||||||
|
let ipv6_flag_idxs: Vec<usize> = args
|
||||||
|
.iter()
|
||||||
|
.enumerate()
|
||||||
|
.filter_map(|(idx, s)| if s == "-A" { Some(idx) } else { None })
|
||||||
|
.collect();
|
||||||
|
let mut ip_addrs: Vec<SocketAddr> = ipv4_flag_idxs
|
||||||
|
.iter()
|
||||||
|
.filter_map(|idx| {
|
||||||
|
Some(SocketAddr::V4(
|
||||||
|
args.get(*idx + 1)?.parse::<SocketAddrV4>().ok()?,
|
||||||
|
))
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
ip_addrs.extend(ipv6_flag_idxs.iter().filter_map(|idx| {
|
||||||
|
Some(SocketAddr::V6(
|
||||||
|
args.get(*idx + 1)?.parse::<SocketAddrV6>().ok()?,
|
||||||
|
))
|
||||||
|
}));
|
||||||
|
|
||||||
|
if ip_addrs.is_empty() {
|
||||||
|
ip_addrs.push(SocketAddr::V4(SocketAddrV4::new(Ipv4Addr::LOCALHOST, 6969)));
|
||||||
|
ip_addrs.push(SocketAddr::V6(SocketAddrV6::new(
|
||||||
|
Ipv6Addr::LOCALHOST,
|
||||||
|
6969,
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut tracker = Tracker::new();
|
||||||
|
return match tracker.start_on(&ip_addrs) {
|
||||||
|
Ok(_) => ExitCode::SUCCESS,
|
||||||
|
Err(e) => {
|
||||||
|
log::error!("Fatal error: {}", e);
|
||||||
|
ExitCode::FAILURE
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
22
src/tracker/bittorrent/info_hash.rs
Normal file
22
src/tracker/bittorrent/info_hash.rs
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
use std::fmt;
|
||||||
|
use std::fmt::Write;
|
||||||
|
|
||||||
|
pub const INFO_HASH_SIZE: usize = 20;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Copy, Hash, Eq, PartialEq)]
|
||||||
|
pub struct InfoHash {
|
||||||
|
pub bytes: [u8; INFO_HASH_SIZE],
|
||||||
|
}
|
||||||
|
|
||||||
|
impl fmt::Display for InfoHash {
|
||||||
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||||
|
const INFO_HASH_STR_LEN: usize = INFO_HASH_SIZE * 2;
|
||||||
|
let mut hex_string: heapless::String<INFO_HASH_STR_LEN> = heapless::String::new();
|
||||||
|
|
||||||
|
for b in self.bytes {
|
||||||
|
let _ = write!(hex_string, "{:02x}", b);
|
||||||
|
}
|
||||||
|
|
||||||
|
write!(f, "{}", hex_string)
|
||||||
|
}
|
||||||
|
}
|
||||||
4
src/tracker/bittorrent/mod.rs
Normal file
4
src/tracker/bittorrent/mod.rs
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
pub mod info_hash;
|
||||||
|
pub mod peer;
|
||||||
|
pub mod protocol;
|
||||||
|
pub mod udp;
|
||||||
20
src/tracker/bittorrent/peer.rs
Normal file
20
src/tracker/bittorrent/peer.rs
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
use std::fmt;
|
||||||
|
|
||||||
|
const MAX_PEER_ID_LENGTH: usize = 20;
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct PeerId {
|
||||||
|
pub bytes: [u8; MAX_PEER_ID_LENGTH],
|
||||||
|
}
|
||||||
|
|
||||||
|
impl PeerId {
|
||||||
|
pub fn to_string_lossy(&self) -> String {
|
||||||
|
String::from_utf8_lossy(&self.bytes).to_string()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl fmt::Display for PeerId {
|
||||||
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||||
|
write!(f, "{}", self.to_string_lossy())
|
||||||
|
}
|
||||||
|
}
|
||||||
62
src/tracker/bittorrent/protocol.rs
Normal file
62
src/tracker/bittorrent/protocol.rs
Normal file
@ -0,0 +1,62 @@
|
|||||||
|
/// BitTorrent's UDP magic `connection_id` constant
|
||||||
|
pub const UDP_MAGIC: i64 = 0x41727101980;
|
||||||
|
|
||||||
|
#[repr(i32)]
|
||||||
|
#[derive(Debug, Clone, Copy, Eq, PartialEq)]
|
||||||
|
pub enum Action {
|
||||||
|
Connect = 0,
|
||||||
|
Announce = 1,
|
||||||
|
Scrape = 2,
|
||||||
|
Error = 3,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Action {
|
||||||
|
pub fn from_i32(val: i32) -> Option<Action> {
|
||||||
|
match val {
|
||||||
|
0 => Some(Action::Connect),
|
||||||
|
1 => Some(Action::Announce),
|
||||||
|
2 => Some(Action::Scrape),
|
||||||
|
3 => Some(Action::Error),
|
||||||
|
_ => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn to_i32(&self) -> i32 {
|
||||||
|
match self {
|
||||||
|
Action::Connect => 0,
|
||||||
|
Action::Announce => 1,
|
||||||
|
Action::Scrape => 2,
|
||||||
|
Action::Error => 3,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[repr(i32)]
|
||||||
|
#[derive(Debug, Clone, Copy, Eq, PartialEq)]
|
||||||
|
pub enum Event {
|
||||||
|
None = 0,
|
||||||
|
Completed = 1,
|
||||||
|
Started = 2,
|
||||||
|
Stopped = 3,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Event {
|
||||||
|
pub fn from_i32(val: i32) -> Option<Event> {
|
||||||
|
match val {
|
||||||
|
0 => Some(Event::None),
|
||||||
|
1 => Some(Event::Completed),
|
||||||
|
2 => Some(Event::Started),
|
||||||
|
3 => Some(Event::Stopped),
|
||||||
|
_ => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn to_i32(&self) -> i32 {
|
||||||
|
match self {
|
||||||
|
Event::None => 0,
|
||||||
|
Event::Completed => 1,
|
||||||
|
Event::Started => 2,
|
||||||
|
Event::Stopped => 3,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
213
src/tracker/bittorrent/udp.rs
Normal file
213
src/tracker/bittorrent/udp.rs
Normal file
@ -0,0 +1,213 @@
|
|||||||
|
use std::{
|
||||||
|
fmt,
|
||||||
|
io::{Cursor, Write},
|
||||||
|
net::{Ipv4Addr, Ipv6Addr, SocketAddrV4, SocketAddrV6},
|
||||||
|
};
|
||||||
|
|
||||||
|
use crate::tracker::bittorrent::{
|
||||||
|
info_hash::InfoHash,
|
||||||
|
peer::PeerId,
|
||||||
|
protocol::{Action, Event},
|
||||||
|
};
|
||||||
|
|
||||||
|
pub const MIN_ANNOUNCE_REQUEST_SIZE: usize = 98;
|
||||||
|
pub const MIN_SCRAPE_REQUEST_SIZE: usize = 36;
|
||||||
|
pub const MIN_CONNECTION_RESPONSE_SIZE: usize = 16;
|
||||||
|
pub const MIN_ANNOUNCE_RESPONSE_SIZE: usize = 20;
|
||||||
|
pub const MIN_SCRAPE_RESPONSE_SIZE: usize = 8;
|
||||||
|
pub const SCRAPE_RESPONSE_ENTRY_SIZE: usize = 12;
|
||||||
|
|
||||||
|
pub const IPV4_SIZE: usize = Ipv4Addr::BITS as usize / 8;
|
||||||
|
pub const IPV6_SIZE: usize = Ipv6Addr::BITS as usize / 8;
|
||||||
|
pub const PORT_SIZE: usize = size_of::<u16>();
|
||||||
|
pub const IPV4_ADDR_PAIR_SIZE: usize = IPV4_SIZE + PORT_SIZE;
|
||||||
|
pub const IPV6_ADDR_PAIR_SIZE: usize = IPV6_SIZE + PORT_SIZE;
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct ConnectRequest {
|
||||||
|
pub protocol_id: i64,
|
||||||
|
pub transaction_id: i32,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct ConnectResponse {
|
||||||
|
pub transaction_id: i32,
|
||||||
|
pub connection_id: i64,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ConnectResponse {
|
||||||
|
pub fn write_to_buf(&self, buf: &mut [u8]) -> usize {
|
||||||
|
let mut c = Cursor::new(buf);
|
||||||
|
let mut written: usize = 0;
|
||||||
|
|
||||||
|
const ACTION: i32 = Action::Connect as i32;
|
||||||
|
|
||||||
|
written += c.write(&ACTION.to_be_bytes()).unwrap();
|
||||||
|
written += c.write(&self.transaction_id.to_be_bytes()).unwrap();
|
||||||
|
written += c.write(&self.connection_id.to_be_bytes()).unwrap();
|
||||||
|
|
||||||
|
return written;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct AnnounceRequest {
|
||||||
|
pub connection_id: i64,
|
||||||
|
pub transaction_id: i32,
|
||||||
|
/// Info hash of the torrent
|
||||||
|
pub info_hash: InfoHash,
|
||||||
|
/// Peer's ID
|
||||||
|
pub peer_id: PeerId,
|
||||||
|
/// Number of bytes peer has downloaded this session
|
||||||
|
pub downloaded: i64,
|
||||||
|
/// Number of bytes peer has remaining to complete the download
|
||||||
|
pub left: i64,
|
||||||
|
/// Number of bytes peer has uploaded this session
|
||||||
|
pub uploaded: i64,
|
||||||
|
/// One of 4 announce events. `Event::None` is used to update presence but not change peer's state.
|
||||||
|
pub event: Event,
|
||||||
|
/// Peer's desired IPv4 address to be sent the announce response on. `None` implies that the packet's source address should be used.
|
||||||
|
pub ipv4_address: Option<Ipv4Addr>,
|
||||||
|
/// Client's unique key for tracking a single client over multiple connections (sockets)
|
||||||
|
pub key: u32,
|
||||||
|
/// Maximum number of info hashes the client is requesting. `None` implies a tracker-defined default value should be used.
|
||||||
|
pub num_want: Option<u32>,
|
||||||
|
/// Peer's desired port to be the announce response on. If `ipv4_address` is `Some` then this should be used.
|
||||||
|
pub port: u16,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub enum AnnounceResponse {
|
||||||
|
V4(AnnounceResponseV4),
|
||||||
|
V6(AnnounceResponseV6),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct AnnounceResponseCommon {
|
||||||
|
pub transaction_id: i32,
|
||||||
|
/// Number of seconds peer should wait before re-announcing
|
||||||
|
pub interval: i32,
|
||||||
|
/// Number of peers still downloading
|
||||||
|
pub leechers: i32,
|
||||||
|
/// Number of peers who have completed the download and are seeding
|
||||||
|
pub seeders: i32,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct AnnounceResponseV4 {
|
||||||
|
pub inner: AnnounceResponseCommon,
|
||||||
|
pub peers: Vec<SocketAddrV4>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct AnnounceResponseV6 {
|
||||||
|
pub inner: AnnounceResponseCommon,
|
||||||
|
pub peers: Vec<SocketAddrV6>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AnnounceResponseCommon {
|
||||||
|
pub fn write_to_buf(&self, buf: &mut [u8]) -> usize {
|
||||||
|
const ACTION: i32 = Action::Announce as i32;
|
||||||
|
buf[0..4].copy_from_slice(&ACTION.to_be_bytes());
|
||||||
|
buf[4..8].copy_from_slice(&self.transaction_id.to_be_bytes());
|
||||||
|
buf[8..12].copy_from_slice(&self.interval.to_be_bytes());
|
||||||
|
buf[12..16].copy_from_slice(&self.leechers.to_be_bytes());
|
||||||
|
buf[16..20].copy_from_slice(&self.seeders.to_be_bytes());
|
||||||
|
|
||||||
|
return MIN_ANNOUNCE_RESPONSE_SIZE;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AnnounceResponseV4 {
|
||||||
|
pub fn write_to_buf(&self, buf: &mut [u8]) -> usize {
|
||||||
|
let written = self.inner.write_to_buf(buf);
|
||||||
|
|
||||||
|
for (offset, addr) in (written..buf.len())
|
||||||
|
.step_by(IPV4_ADDR_PAIR_SIZE)
|
||||||
|
.zip(&self.peers)
|
||||||
|
{
|
||||||
|
buf[offset..offset + IPV4_SIZE].copy_from_slice(&addr.ip().octets());
|
||||||
|
buf[offset + IPV4_SIZE..offset + IPV4_ADDR_PAIR_SIZE]
|
||||||
|
.copy_from_slice(&addr.port().to_be_bytes());
|
||||||
|
}
|
||||||
|
|
||||||
|
return written + self.peers.len() * IPV4_ADDR_PAIR_SIZE;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AnnounceResponseV6 {
|
||||||
|
pub fn write_to_buf(&self, buf: &mut [u8]) -> usize {
|
||||||
|
let written = self.inner.write_to_buf(buf);
|
||||||
|
|
||||||
|
for (offset, addr) in (written..buf.len())
|
||||||
|
.step_by(IPV6_ADDR_PAIR_SIZE)
|
||||||
|
.zip(&self.peers)
|
||||||
|
{
|
||||||
|
buf[offset..offset + IPV6_SIZE].copy_from_slice(&addr.ip().octets());
|
||||||
|
buf[offset + IPV6_SIZE..offset + IPV6_ADDR_PAIR_SIZE]
|
||||||
|
.copy_from_slice(&addr.port().to_be_bytes());
|
||||||
|
}
|
||||||
|
|
||||||
|
return written + self.peers.len() * IPV6_ADDR_PAIR_SIZE;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct ScrapeRequest {
|
||||||
|
pub connection_id: i64,
|
||||||
|
pub transaction_id: i32,
|
||||||
|
pub info_hashes: Vec<InfoHash>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Default)]
|
||||||
|
pub struct ScrapeStats {
|
||||||
|
/// Number of connected peers who have completed the download and are seeding
|
||||||
|
pub seeders: i32,
|
||||||
|
/// Number of peers who have ever completed the download
|
||||||
|
pub completed: i32,
|
||||||
|
/// Number of connected peers who have not completed the download and are seeding
|
||||||
|
pub leechers: i32,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct ScrapeResponse {
|
||||||
|
pub transaction_id: i32,
|
||||||
|
pub stats: Vec<ScrapeStats>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ScrapeResponse {
|
||||||
|
pub fn write_to_buf(&self, buf: &mut [u8]) -> usize {
|
||||||
|
let mut c = Cursor::new(buf);
|
||||||
|
let mut written: usize = 0;
|
||||||
|
|
||||||
|
const ACTION: i32 = Action::Scrape as i32;
|
||||||
|
written += c.write(&ACTION.to_be_bytes()).unwrap();
|
||||||
|
written += c.write(&self.transaction_id.to_be_bytes()).unwrap();
|
||||||
|
|
||||||
|
for s in &self.stats {
|
||||||
|
written += c.write(&s.seeders.to_be_bytes()).unwrap();
|
||||||
|
written += c.write(&s.completed.to_be_bytes()).unwrap();
|
||||||
|
written += c.write(&s.leechers.to_be_bytes()).unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
return written;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct ErrorResponse {
|
||||||
|
pub transaction_id: i32,
|
||||||
|
pub message: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub enum UdpRequest {
|
||||||
|
Connect(ConnectRequest),
|
||||||
|
Announce(AnnounceRequest),
|
||||||
|
Scrape(ScrapeRequest),
|
||||||
|
}
|
||||||
|
|
||||||
|
pub enum UdpResponse {
|
||||||
|
Connect(ConnectResponse),
|
||||||
|
Announce(AnnounceResponse),
|
||||||
|
Scrape(ScrapeResponse),
|
||||||
|
Error(ErrorResponse),
|
||||||
|
}
|
||||||
5
src/tracker/mod.rs
Normal file
5
src/tracker/mod.rs
Normal file
@ -0,0 +1,5 @@
|
|||||||
|
mod bittorrent;
|
||||||
|
mod tracker;
|
||||||
|
mod udp_server;
|
||||||
|
|
||||||
|
pub use tracker::Tracker;
|
||||||
427
src/tracker/tracker.rs
Normal file
427
src/tracker/tracker.rs
Normal file
@ -0,0 +1,427 @@
|
|||||||
|
use std::collections::HashMap;
|
||||||
|
use std::net::{SocketAddr, SocketAddrV4, SocketAddrV6};
|
||||||
|
use std::sync::atomic::{AtomicBool, Ordering};
|
||||||
|
use std::sync::{Arc, RwLock};
|
||||||
|
use std::thread;
|
||||||
|
use std::time::{Duration, Instant};
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use flume::Sender;
|
||||||
|
use rand::Rng;
|
||||||
|
|
||||||
|
use crate::tracker::bittorrent::info_hash::InfoHash;
|
||||||
|
use crate::tracker::bittorrent::protocol::Event;
|
||||||
|
use crate::tracker::bittorrent::udp::{
|
||||||
|
AnnounceResponse, AnnounceResponseCommon, AnnounceResponseV4, AnnounceResponseV6,
|
||||||
|
ConnectResponse, ScrapeResponse, ScrapeStats, UdpRequest, UdpResponse,
|
||||||
|
};
|
||||||
|
use crate::tracker::udp_server::UdpServer;
|
||||||
|
|
||||||
|
pub struct RequestMessage {
|
||||||
|
pub(crate) request: UdpRequest,
|
||||||
|
pub(crate) src_addr: SocketAddr,
|
||||||
|
pub(crate) server_id: u32,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct ResponseMessage {
|
||||||
|
pub(crate) response: UdpResponse,
|
||||||
|
pub(crate) dst_addr: SocketAddr,
|
||||||
|
}
|
||||||
|
|
||||||
|
struct PeerMetadata {
|
||||||
|
socket_addr: SocketAddr,
|
||||||
|
last_active: Instant,
|
||||||
|
}
|
||||||
|
|
||||||
|
struct PeerStatus {
|
||||||
|
connection_id: i64,
|
||||||
|
last_event: Event,
|
||||||
|
last_active: Instant,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub const GARBAGE_COLLECTION_INTERVAL: Duration = Duration::from_secs(20);
|
||||||
|
|
||||||
|
pub const CONNECTION_EXPIRE_TIME: Duration = Duration::from_secs(90);
|
||||||
|
pub const DEFAULT_ANNOUNCE_INTERVAL: Duration = Duration::from_secs(60);
|
||||||
|
pub const DEFAULT_ANNOUNCE_WANT: u32 = 80;
|
||||||
|
|
||||||
|
type ConnectionIdMap = HashMap<i64, PeerMetadata>;
|
||||||
|
type InfoHashMap = HashMap<InfoHash, Vec<PeerStatus>>;
|
||||||
|
|
||||||
|
pub static SHOULD_STOP_SIGNAL: AtomicBool = AtomicBool::new(false);
|
||||||
|
|
||||||
|
pub struct Tracker {
|
||||||
|
server_threads: Vec<thread::JoinHandle<()>>,
|
||||||
|
response_channels: Vec<Sender<ResponseMessage>>,
|
||||||
|
/// Map of IPs & ports to peer connection IDs
|
||||||
|
peers: Arc<RwLock<ConnectionIdMap>>,
|
||||||
|
info_hashes: Arc<RwLock<InfoHashMap>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Tracker {
|
||||||
|
pub fn new() -> Self {
|
||||||
|
Tracker {
|
||||||
|
server_threads: Vec::new(),
|
||||||
|
response_channels: Vec::new(),
|
||||||
|
peers: Arc::new(RwLock::new(HashMap::new())),
|
||||||
|
info_hashes: Arc::new(RwLock::new(HashMap::new())),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn start_on(&mut self, addrs: &[SocketAddr]) -> Result<()> {
|
||||||
|
if addrs.is_empty() {
|
||||||
|
log::error!("No addresses provided for tracker to listen on");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Only one request channel set is necessary as all the server threads send on the same one and the tracker is the only reader
|
||||||
|
let (req_send, req_recv) = flume::unbounded::<RequestMessage>();
|
||||||
|
|
||||||
|
let mut server_id: u32 = 0;
|
||||||
|
|
||||||
|
for a in addrs {
|
||||||
|
// Each UDP server gets its own response channels
|
||||||
|
let (resp_send, resp_recv) = flume::unbounded::<ResponseMessage>();
|
||||||
|
let req_chan = req_send.clone();
|
||||||
|
let a = a.clone();
|
||||||
|
|
||||||
|
self.server_threads.push(
|
||||||
|
thread::Builder::new()
|
||||||
|
.name(format!(
|
||||||
|
"{} server thread (ID {})",
|
||||||
|
env!("CARGO_PKG_NAME"),
|
||||||
|
server_id
|
||||||
|
))
|
||||||
|
.spawn(move || {
|
||||||
|
let server = UdpServer::new(a, server_id);
|
||||||
|
loop {
|
||||||
|
match server.start(req_chan.clone(), resp_recv.clone()) {
|
||||||
|
Err(e) => {
|
||||||
|
log::error!("Failed to start server {}: {}", server_id, e)
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
})?,
|
||||||
|
);
|
||||||
|
|
||||||
|
self.response_channels.push(resp_send);
|
||||||
|
|
||||||
|
server_id += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
let gc_peers = self.peers.clone();
|
||||||
|
let gc_info_hashes = self.info_hashes.clone();
|
||||||
|
let _gc_thread = thread::spawn(move || {
|
||||||
|
let peers = gc_peers;
|
||||||
|
let info_hashes = gc_info_hashes;
|
||||||
|
|
||||||
|
loop {
|
||||||
|
thread::sleep(GARBAGE_COLLECTION_INTERVAL);
|
||||||
|
|
||||||
|
let start = Instant::now();
|
||||||
|
|
||||||
|
let expired = Tracker::purge_expired_connections(&mut peers.write().unwrap());
|
||||||
|
Tracker::purge_expired_swarm_peers(&mut info_hashes.write().unwrap(), &expired);
|
||||||
|
|
||||||
|
log::trace!("Garbage collection took {} ns", start.elapsed().as_nanos())
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
let mut rng = rand::rng();
|
||||||
|
|
||||||
|
loop {
|
||||||
|
let request = match req_recv.recv() {
|
||||||
|
Ok(r) => r,
|
||||||
|
_ => {
|
||||||
|
log::error!("Internal tracker error: server request channel closed");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
match request.request {
|
||||||
|
UdpRequest::Connect(connect) => {
|
||||||
|
log::trace!(
|
||||||
|
"(Server {}) Connect request: {:?}",
|
||||||
|
request.server_id,
|
||||||
|
connect
|
||||||
|
);
|
||||||
|
|
||||||
|
let new_id: i64 = rng.random();
|
||||||
|
|
||||||
|
let mut peers = self.peers.write().unwrap();
|
||||||
|
|
||||||
|
peers.insert(
|
||||||
|
new_id,
|
||||||
|
PeerMetadata {
|
||||||
|
socket_addr: request.src_addr,
|
||||||
|
last_active: Instant::now(),
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
match self.response_channels[request.server_id as usize].send(ResponseMessage {
|
||||||
|
response: UdpResponse::Connect(ConnectResponse {
|
||||||
|
transaction_id: connect.transaction_id,
|
||||||
|
connection_id: new_id,
|
||||||
|
}),
|
||||||
|
dst_addr: request.src_addr,
|
||||||
|
}) {
|
||||||
|
Ok(r) => r,
|
||||||
|
_ => {
|
||||||
|
log::error!("Internal tracker error: server response channel closed");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
UdpRequest::Announce(announce) => {
|
||||||
|
log::trace!(
|
||||||
|
"(Server {}) Announce request: {:?}",
|
||||||
|
request.server_id,
|
||||||
|
announce
|
||||||
|
);
|
||||||
|
|
||||||
|
if !self.connection_valid(announce.connection_id, request.src_addr) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ensure we honour the desired number of peers, within our bounaries
|
||||||
|
|
||||||
|
let n_announce_want: u32 = if let Some(n) = announce.num_want {
|
||||||
|
if n < DEFAULT_ANNOUNCE_WANT {
|
||||||
|
n
|
||||||
|
} else {
|
||||||
|
DEFAULT_ANNOUNCE_WANT
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
DEFAULT_ANNOUNCE_WANT
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut n_announce_entries: u32 = 0;
|
||||||
|
let mut n_leechers: i32 = 0;
|
||||||
|
let mut n_seeders: i32 = 0;
|
||||||
|
let mut v4_peers: Vec<SocketAddrV4> = Vec::new();
|
||||||
|
let mut v6_peers: Vec<SocketAddrV6> = Vec::new();
|
||||||
|
|
||||||
|
let mut info_hashes = self.info_hashes.write().unwrap();
|
||||||
|
|
||||||
|
match info_hashes.get_mut(&announce.info_hash) {
|
||||||
|
Some(swarm) => {
|
||||||
|
if announce.event == Event::Stopped {
|
||||||
|
swarm.retain(|status| {
|
||||||
|
status.connection_id != announce.connection_id
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
let peers = self.peers.read().unwrap();
|
||||||
|
|
||||||
|
for peer_status in swarm {
|
||||||
|
if peer_status.last_event == Event::Stopped {
|
||||||
|
// Skip any peers who have stopped their download
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// State update for ourselves
|
||||||
|
|
||||||
|
if peer_status.connection_id == announce.connection_id {
|
||||||
|
if peer_status.last_event != announce.event
|
||||||
|
&& announce.event != Event::None
|
||||||
|
{
|
||||||
|
// Update last event if the event has changed and is not just a `None` for notifying presence
|
||||||
|
peer_status.last_event = announce.event
|
||||||
|
} else {
|
||||||
|
// Peer status has not changed
|
||||||
|
peer_status.last_active = Instant::now();
|
||||||
|
}
|
||||||
|
|
||||||
|
continue; // don't give the peer their own connection ID's address
|
||||||
|
}
|
||||||
|
|
||||||
|
if n_announce_entries >= n_announce_want {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Collect and add the relevant peer's address to the appropriate vector
|
||||||
|
|
||||||
|
if let Some(a) = peers.get(&peer_status.connection_id) {
|
||||||
|
if a.socket_addr == request.src_addr {
|
||||||
|
continue; // don't give the peer an expired ID with the same address
|
||||||
|
}
|
||||||
|
|
||||||
|
match &a.socket_addr {
|
||||||
|
SocketAddr::V4(v4) => {
|
||||||
|
if request.src_addr.is_ipv4() {
|
||||||
|
v4_peers.push(*v4);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
SocketAddr::V6(v6) => {
|
||||||
|
if request.src_addr.is_ipv6() {
|
||||||
|
v6_peers.push(*v6);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Keep track of the seeders, leechers & announce entries gathered
|
||||||
|
|
||||||
|
if peer_status.last_event == Event::Completed {
|
||||||
|
n_seeders += 1;
|
||||||
|
} else {
|
||||||
|
n_leechers += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
n_announce_entries += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None => {
|
||||||
|
// Info hash isn't currently tracked
|
||||||
|
// No relevant peers in the swarm
|
||||||
|
|
||||||
|
info_hashes.insert(
|
||||||
|
announce.info_hash,
|
||||||
|
vec![PeerStatus {
|
||||||
|
connection_id: announce.connection_id,
|
||||||
|
last_event: announce.event,
|
||||||
|
last_active: Instant::now(),
|
||||||
|
}],
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let resp_common = AnnounceResponseCommon {
|
||||||
|
transaction_id: announce.transaction_id,
|
||||||
|
interval: DEFAULT_ANNOUNCE_INTERVAL.as_secs() as i32,
|
||||||
|
leechers: n_leechers,
|
||||||
|
seeders: n_seeders,
|
||||||
|
};
|
||||||
|
let response = if request.src_addr.is_ipv4() {
|
||||||
|
// IPv4
|
||||||
|
AnnounceResponse::V4(AnnounceResponseV4 {
|
||||||
|
inner: resp_common,
|
||||||
|
peers: v4_peers,
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
// IPv6
|
||||||
|
AnnounceResponse::V6(AnnounceResponseV6 {
|
||||||
|
inner: resp_common,
|
||||||
|
peers: v6_peers,
|
||||||
|
})
|
||||||
|
};
|
||||||
|
|
||||||
|
match self.response_channels[request.server_id as usize].send(ResponseMessage {
|
||||||
|
response: UdpResponse::Announce(response),
|
||||||
|
dst_addr: request.src_addr,
|
||||||
|
}) {
|
||||||
|
Ok(r) => r,
|
||||||
|
_ => {
|
||||||
|
log::error!("Internal tracker error: server response channel closed");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
UdpRequest::Scrape(scrape) => {
|
||||||
|
log::trace!(
|
||||||
|
"(Server {}) Scrape request: {:?}",
|
||||||
|
request.server_id,
|
||||||
|
scrape
|
||||||
|
);
|
||||||
|
|
||||||
|
if !self.connection_valid(scrape.connection_id, request.src_addr) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
let info_hashes = self.info_hashes.read().unwrap();
|
||||||
|
|
||||||
|
let mut scrape_stats: Vec<ScrapeStats> = Vec::new();
|
||||||
|
for info_hash in &scrape.info_hashes {
|
||||||
|
match info_hashes.get(info_hash) {
|
||||||
|
Some(peers) => {
|
||||||
|
scrape_stats.push(ScrapeStats {
|
||||||
|
seeders: peers
|
||||||
|
.iter()
|
||||||
|
.filter(|&peer_status| {
|
||||||
|
peer_status.last_event == Event::Completed
|
||||||
|
})
|
||||||
|
.count()
|
||||||
|
as i32,
|
||||||
|
completed: 0, // TODO: keep track of this
|
||||||
|
leechers: peers
|
||||||
|
.iter()
|
||||||
|
.filter(|&peer_status| {
|
||||||
|
peer_status.last_event == Event::Started
|
||||||
|
})
|
||||||
|
.count()
|
||||||
|
as i32,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
None => scrape_stats.push(ScrapeStats::default()),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
match self.response_channels[request.server_id as usize].send(ResponseMessage {
|
||||||
|
response: UdpResponse::Scrape(ScrapeResponse {
|
||||||
|
transaction_id: scrape.transaction_id,
|
||||||
|
stats: scrape_stats,
|
||||||
|
}),
|
||||||
|
dst_addr: request.src_addr,
|
||||||
|
}) {
|
||||||
|
Ok(r) => r,
|
||||||
|
_ => {
|
||||||
|
log::error!("Internal tracker error: server response channel closed");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
self.server_threads.drain(..).for_each(|t| {
|
||||||
|
let _ = t.join();
|
||||||
|
});
|
||||||
|
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
|
fn connection_valid(&self, connection_id: i64, src_addr: SocketAddr) -> bool {
|
||||||
|
let peers = self.peers.read().unwrap();
|
||||||
|
|
||||||
|
return match peers.get(&connection_id) {
|
||||||
|
Some(metadata) => metadata.socket_addr == src_addr,
|
||||||
|
None => false,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
fn purge_expired_connections(peers: &mut ConnectionIdMap) -> Vec<i64> {
|
||||||
|
let mut purged: Vec<i64> = Vec::new();
|
||||||
|
|
||||||
|
for (id, metadata) in peers.iter() {
|
||||||
|
if metadata.last_active.elapsed() > CONNECTION_EXPIRE_TIME {
|
||||||
|
purged.push(*id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for id in &purged {
|
||||||
|
peers.remove(id);
|
||||||
|
}
|
||||||
|
|
||||||
|
return purged;
|
||||||
|
}
|
||||||
|
|
||||||
|
fn purge_expired_swarm_peers(info_hashes: &mut InfoHashMap, expired_connections: &[i64]) {
|
||||||
|
info_hashes.retain(|_, swarm| {
|
||||||
|
swarm.is_empty()
|
||||||
|
|| swarm
|
||||||
|
.iter()
|
||||||
|
.find(|status| {
|
||||||
|
expired_connections.contains(&status.connection_id)
|
||||||
|
|| status.last_active.elapsed() > DEFAULT_ANNOUNCE_INTERVAL * 2
|
||||||
|
|| status.last_event == Event::Stopped
|
||||||
|
})
|
||||||
|
.is_some()
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn should_stop() -> bool {
|
||||||
|
SHOULD_STOP_SIGNAL.load(Ordering::Relaxed)
|
||||||
|
}
|
||||||
|
}
|
||||||
257
src/tracker/udp_server.rs
Normal file
257
src/tracker/udp_server.rs
Normal file
@ -0,0 +1,257 @@
|
|||||||
|
use std::net::{Ipv4Addr, SocketAddr, UdpSocket};
|
||||||
|
use std::thread;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use flume::{Receiver, Sender};
|
||||||
|
use log;
|
||||||
|
|
||||||
|
use crate::tracker::Tracker;
|
||||||
|
use crate::tracker::bittorrent::info_hash::InfoHash;
|
||||||
|
use crate::tracker::bittorrent::peer::PeerId;
|
||||||
|
use crate::tracker::bittorrent::protocol::Event;
|
||||||
|
use crate::tracker::bittorrent::udp::{
|
||||||
|
AnnounceRequest, AnnounceResponse, MIN_ANNOUNCE_REQUEST_SIZE, MIN_SCRAPE_REQUEST_SIZE,
|
||||||
|
ScrapeRequest, UdpResponse,
|
||||||
|
};
|
||||||
|
use crate::tracker::{
|
||||||
|
bittorrent::{
|
||||||
|
protocol::{Action, UDP_MAGIC},
|
||||||
|
udp::{ConnectRequest, UdpRequest},
|
||||||
|
},
|
||||||
|
tracker::{RequestMessage, ResponseMessage},
|
||||||
|
};
|
||||||
|
|
||||||
|
pub struct UdpServer {
|
||||||
|
addr: SocketAddr,
|
||||||
|
server_id: u32,
|
||||||
|
}
|
||||||
|
|
||||||
|
const UDP_RECV_BUF_SIZE: usize = u16::MAX as usize;
|
||||||
|
const UDP_SEND_BUF_SIZE: usize = u16::MAX as usize; // Could also be 1500 for typical Ethernet MTU (?)
|
||||||
|
|
||||||
|
impl UdpServer {
|
||||||
|
pub fn new(addr: SocketAddr, id: u32) -> Self {
|
||||||
|
UdpServer {
|
||||||
|
addr: addr,
|
||||||
|
server_id: id,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn start(
|
||||||
|
&self,
|
||||||
|
req_chan: Sender<RequestMessage>,
|
||||||
|
resp_chan: Receiver<ResponseMessage>,
|
||||||
|
) -> Result<()> {
|
||||||
|
let recv_socket = UdpSocket::bind(self.addr)?;
|
||||||
|
let send_socket = recv_socket.try_clone()?;
|
||||||
|
let local_addr = recv_socket.local_addr()?;
|
||||||
|
let id = self.server_id;
|
||||||
|
|
||||||
|
log::info!("Starting UDP server on: {:?}", local_addr);
|
||||||
|
|
||||||
|
let recv_thread = thread::spawn(move || {
|
||||||
|
let mut buf: [u8; UDP_RECV_BUF_SIZE] = [0; UDP_RECV_BUF_SIZE];
|
||||||
|
|
||||||
|
loop {
|
||||||
|
if Tracker::should_stop() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
match recv_socket.recv_from(&mut buf) {
|
||||||
|
Ok((n_bytes, src)) => {
|
||||||
|
log::trace!("(Server {}) Received {} bytes from {}", id, n_bytes, src);
|
||||||
|
|
||||||
|
match try_parse_packet(&buf[..n_bytes]) {
|
||||||
|
Some(req) => {
|
||||||
|
if req_chan
|
||||||
|
.send(RequestMessage {
|
||||||
|
request: req,
|
||||||
|
src_addr: src,
|
||||||
|
server_id: id,
|
||||||
|
})
|
||||||
|
.is_err()
|
||||||
|
{
|
||||||
|
log::error!(
|
||||||
|
"Internal tracker error: server {} request channel closed",
|
||||||
|
id
|
||||||
|
);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None => continue,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
Err(error) => {
|
||||||
|
log::error!("Failed to receive on socket: {}", error);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
log::debug!("Server {} receive thread shutting exiting ...", id)
|
||||||
|
});
|
||||||
|
|
||||||
|
let send_thread = thread::spawn(move || {
|
||||||
|
let mut buf: [u8; UDP_SEND_BUF_SIZE] = [0; UDP_SEND_BUF_SIZE];
|
||||||
|
|
||||||
|
loop {
|
||||||
|
if Tracker::should_stop() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
let response = match resp_chan.recv() {
|
||||||
|
Ok(r) => r,
|
||||||
|
_ => {
|
||||||
|
log::error!(
|
||||||
|
"Internal tracker error: server {} response channel closed",
|
||||||
|
id
|
||||||
|
);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
log::trace!("(Server {}) Tracker response", id);
|
||||||
|
|
||||||
|
let n_bytes = match response.response {
|
||||||
|
UdpResponse::Connect(connect) => connect.write_to_buf(&mut buf),
|
||||||
|
UdpResponse::Announce(announce) => match announce {
|
||||||
|
AnnounceResponse::V4(v4) => v4.write_to_buf(&mut buf),
|
||||||
|
AnnounceResponse::V6(v6) => v6.write_to_buf(&mut buf),
|
||||||
|
},
|
||||||
|
UdpResponse::Scrape(scrape) => scrape.write_to_buf(&mut buf),
|
||||||
|
UdpResponse::Error(_) => 0,
|
||||||
|
};
|
||||||
|
|
||||||
|
let n_sent = match send_socket.send_to(&buf[..n_bytes], response.dst_addr) {
|
||||||
|
Ok(n) => n,
|
||||||
|
Err(e) => {
|
||||||
|
log::error!(
|
||||||
|
"Failed to send {} bytes to {}: {}",
|
||||||
|
n_bytes,
|
||||||
|
response.dst_addr,
|
||||||
|
e
|
||||||
|
);
|
||||||
|
0
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
log::trace!(
|
||||||
|
"(Server {}) Sent {} bytes to {}: {:x?}",
|
||||||
|
id,
|
||||||
|
n_sent,
|
||||||
|
response.dst_addr,
|
||||||
|
&buf[..n_bytes]
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
log::debug!("Server {} send thread shutting exiting ...", id)
|
||||||
|
});
|
||||||
|
|
||||||
|
let _ = recv_thread.join();
|
||||||
|
let _ = send_thread.join();
|
||||||
|
|
||||||
|
log::info!("Stopped UDP server on: {:?}", local_addr);
|
||||||
|
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_parse_packet(buf: &[u8]) -> Option<UdpRequest> {
|
||||||
|
if buf.len() < 16 {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
let action: Action = Action::from_i32(i32::from_be_bytes(buf[8..12].try_into().ok()?))?;
|
||||||
|
|
||||||
|
let request = match action {
|
||||||
|
Action::Connect => UdpRequest::Connect(try_parse_connect(&buf)?),
|
||||||
|
Action::Announce => UdpRequest::Announce(try_parse_announce(&buf)?),
|
||||||
|
Action::Scrape => UdpRequest::Scrape(try_parse_scrape(&buf)?),
|
||||||
|
_ => return None,
|
||||||
|
};
|
||||||
|
|
||||||
|
return Some(request);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_parse_connect(buf: &[u8]) -> Option<ConnectRequest> {
|
||||||
|
// Buffer length is checked to be at least 16 in `try_parse_packet`
|
||||||
|
|
||||||
|
let protocol_id: i64 = i64::from_be_bytes(buf[0..8].try_into().ok()?);
|
||||||
|
|
||||||
|
if protocol_id != UDP_MAGIC {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
Some(ConnectRequest {
|
||||||
|
protocol_id: protocol_id,
|
||||||
|
transaction_id: i32::from_be_bytes(buf[12..16].try_into().ok()?),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_parse_announce(buf: &[u8]) -> Option<AnnounceRequest> {
|
||||||
|
if buf.len() < MIN_ANNOUNCE_REQUEST_SIZE {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
let connection_id: i64 = i64::from_be_bytes(buf[0..8].try_into().ok()?);
|
||||||
|
let transaction_id: i32 = i32::from_be_bytes(buf[12..16].try_into().ok()?);
|
||||||
|
let info_hash: InfoHash = InfoHash {
|
||||||
|
bytes: buf[16..36].try_into().ok()?,
|
||||||
|
};
|
||||||
|
let peer_id: PeerId = PeerId {
|
||||||
|
bytes: buf[36..56].try_into().unwrap_or_default(),
|
||||||
|
};
|
||||||
|
let downloaded: i64 = i64::from_be_bytes(buf[56..64].try_into().ok()?);
|
||||||
|
let left: i64 = i64::from_be_bytes(buf[64..72].try_into().ok()?);
|
||||||
|
let uploaded: i64 = i64::from_be_bytes(buf[72..80].try_into().ok()?);
|
||||||
|
let event: Event = Event::from_i32(i32::from_be_bytes(buf[80..84].try_into().ok()?))?;
|
||||||
|
let ip_addr: Option<Ipv4Addr> = if buf[84..88].iter().all(|b| *b == 0x00) {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(Ipv4Addr::from_bits(u32::from_be_bytes(
|
||||||
|
buf[84..88].try_into().ok()?,
|
||||||
|
)))
|
||||||
|
};
|
||||||
|
let key: u32 = u32::from_be_bytes(buf[88..92].try_into().ok()?);
|
||||||
|
let num_want: Option<u32> = {
|
||||||
|
let want = i32::from_be_bytes(buf[92..96].try_into().ok()?);
|
||||||
|
if want < 0 { None } else { Some(want as u32) }
|
||||||
|
};
|
||||||
|
let port: u16 = u16::from_be_bytes(buf[96..98].try_into().ok()?);
|
||||||
|
|
||||||
|
Some(AnnounceRequest {
|
||||||
|
connection_id: connection_id,
|
||||||
|
transaction_id: transaction_id,
|
||||||
|
info_hash: info_hash,
|
||||||
|
peer_id: peer_id,
|
||||||
|
downloaded: downloaded,
|
||||||
|
left: left,
|
||||||
|
uploaded: uploaded,
|
||||||
|
event: event,
|
||||||
|
ipv4_address: ip_addr,
|
||||||
|
key: key,
|
||||||
|
num_want: num_want,
|
||||||
|
port: port,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_parse_scrape(buf: &[u8]) -> Option<ScrapeRequest> {
|
||||||
|
if buf.len() < MIN_SCRAPE_REQUEST_SIZE {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
let connection_id: i64 = i64::from_be_bytes(buf[0..8].try_into().ok()?);
|
||||||
|
let transaction_id: i32 = i32::from_be_bytes(buf[12..16].try_into().ok()?);
|
||||||
|
let info_hashes: Vec<InfoHash> = buf[16..]
|
||||||
|
.chunks_exact(20)
|
||||||
|
.map(|b| InfoHash {
|
||||||
|
bytes: b.try_into().unwrap(), // unwrap: `chunks_exact` guarantees the size
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
Some(ScrapeRequest {
|
||||||
|
connection_id: connection_id,
|
||||||
|
transaction_id: transaction_id,
|
||||||
|
info_hashes: info_hashes,
|
||||||
|
})
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user