protover.rs 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947
  1. // Copyright (c) 2016-2018, The Tor Project, Inc. */
  2. // See LICENSE for licensing information */
  3. use std::collections::hash_map;
  4. use std::collections::HashMap;
  5. use std::ffi::CStr;
  6. use std::fmt;
  7. use std::str;
  8. use std::str::FromStr;
  9. use std::string::String;
  10. use external::c_tor_version_as_new_as;
  11. use errors::ProtoverError;
  12. use protoset::ProtoSet;
  13. use protoset::Version;
  14. /// The first version of Tor that included "proto" entries in its descriptors.
  15. /// Authorities should use this to decide whether to guess proto lines.
  16. ///
  17. /// C_RUST_COUPLED:
  18. /// protover.h `FIRST_TOR_VERSION_TO_ADVERTISE_PROTOCOLS`
  19. const FIRST_TOR_VERSION_TO_ADVERTISE_PROTOCOLS: &'static str = "0.2.9.3-alpha";
  20. /// The maximum number of subprotocol version numbers we will attempt to expand
  21. /// before concluding that someone is trying to DoS us
  22. ///
  23. /// C_RUST_COUPLED: protover.c `MAX_PROTOCOLS_TO_EXPAND`
  24. const MAX_PROTOCOLS_TO_EXPAND: usize = (1 << 16);
  25. /// The maximum size an `UnknownProtocol`'s name may be.
  26. pub(crate) const MAX_PROTOCOL_NAME_LENGTH: usize = 100;
  27. /// Known subprotocols in Tor. Indicates which subprotocol a relay supports.
  28. ///
  29. /// C_RUST_COUPLED: protover.h `protocol_type_t`
  30. #[derive(Clone, Hash, Eq, PartialEq, Debug)]
  31. pub enum Protocol {
  32. Cons,
  33. Desc,
  34. DirCache,
  35. HSDir,
  36. HSIntro,
  37. HSRend,
  38. Link,
  39. LinkAuth,
  40. Microdesc,
  41. Relay,
  42. }
  43. impl fmt::Display for Protocol {
  44. fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
  45. write!(f, "{:?}", self)
  46. }
  47. }
  48. /// Translates a string representation of a protocol into a Proto type.
  49. /// Error if the string is an unrecognized protocol name.
  50. ///
  51. /// C_RUST_COUPLED: protover.c `PROTOCOL_NAMES`
  52. impl FromStr for Protocol {
  53. type Err = ProtoverError;
  54. fn from_str(s: &str) -> Result<Self, Self::Err> {
  55. match s {
  56. "Cons" => Ok(Protocol::Cons),
  57. "Desc" => Ok(Protocol::Desc),
  58. "DirCache" => Ok(Protocol::DirCache),
  59. "HSDir" => Ok(Protocol::HSDir),
  60. "HSIntro" => Ok(Protocol::HSIntro),
  61. "HSRend" => Ok(Protocol::HSRend),
  62. "Link" => Ok(Protocol::Link),
  63. "LinkAuth" => Ok(Protocol::LinkAuth),
  64. "Microdesc" => Ok(Protocol::Microdesc),
  65. "Relay" => Ok(Protocol::Relay),
  66. _ => Err(ProtoverError::UnknownProtocol),
  67. }
  68. }
  69. }
  70. /// A protocol string which is not one of the `Protocols` we currently know
  71. /// about.
  72. #[derive(Clone, Debug, Hash, Eq, PartialEq)]
  73. pub struct UnknownProtocol(String);
  74. impl fmt::Display for UnknownProtocol {
  75. fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
  76. write!(f, "{}", self.0)
  77. }
  78. }
  79. fn is_valid_proto(s: &str) -> bool {
  80. s.chars().all(|c| c.is_ascii_alphanumeric() || c == '-')
  81. }
  82. impl FromStr for UnknownProtocol {
  83. type Err = ProtoverError;
  84. fn from_str(s: &str) -> Result<Self, Self::Err> {
  85. if !is_valid_proto(s) {
  86. Err(ProtoverError::InvalidProtocol)
  87. } else if s.len() <= MAX_PROTOCOL_NAME_LENGTH {
  88. Ok(UnknownProtocol(s.to_string()))
  89. } else {
  90. Err(ProtoverError::ExceedsNameLimit)
  91. }
  92. }
  93. }
  94. impl UnknownProtocol {
  95. /// Create an `UnknownProtocol`, ignoring whether or not it
  96. /// exceeds MAX_PROTOCOL_NAME_LENGTH.
  97. fn from_str_any_len(s: &str) -> Result<Self, ProtoverError> {
  98. if !is_valid_proto(s) {
  99. return Err(ProtoverError::InvalidProtocol);
  100. }
  101. Ok(UnknownProtocol(s.to_string()))
  102. }
  103. }
  104. impl From<Protocol> for UnknownProtocol {
  105. fn from(p: Protocol) -> UnknownProtocol {
  106. UnknownProtocol(p.to_string())
  107. }
  108. }
  109. /// Get a CStr representation of current supported protocols, for
  110. /// passing to C, or for converting to a `&str` for Rust.
  111. ///
  112. /// # Returns
  113. ///
  114. /// An `&'static CStr` whose value is the existing protocols supported by tor.
  115. /// Returned data is in the format as follows:
  116. ///
  117. /// "HSDir=1-1 LinkAuth=1"
  118. ///
  119. /// # Note
  120. ///
  121. /// Rust code can use the `&'static CStr` as a normal `&'a str` by
  122. /// calling `protover::get_supported_protocols`.
  123. ///
  124. // C_RUST_COUPLED: protover.c `protover_get_supported_protocols`
  125. pub(crate) fn get_supported_protocols_cstr() -> &'static CStr {
  126. cstr!(
  127. "Cons=1-2 \
  128. Desc=1-2 \
  129. DirCache=1-2 \
  130. HSDir=1-2 \
  131. HSIntro=3-4 \
  132. HSRend=1-2 \
  133. Link=1-5 \
  134. LinkAuth=1,3 \
  135. Microdesc=1-2 \
  136. Relay=1-2"
  137. )
  138. }
  139. /// A map of protocol names to the versions of them which are supported.
  140. #[derive(Clone, Debug, PartialEq, Eq)]
  141. pub struct ProtoEntry(HashMap<Protocol, ProtoSet>);
  142. impl Default for ProtoEntry {
  143. fn default() -> ProtoEntry {
  144. ProtoEntry(HashMap::new())
  145. }
  146. }
  147. impl ProtoEntry {
  148. /// Get an iterator over the `Protocol`s and their `ProtoSet`s in this `ProtoEntry`.
  149. pub fn iter(&self) -> hash_map::Iter<Protocol, ProtoSet> {
  150. self.0.iter()
  151. }
  152. /// Translate the supported tor versions from a string into a
  153. /// ProtoEntry, which is useful when looking up a specific
  154. /// subprotocol.
  155. pub fn supported() -> Result<Self, ProtoverError> {
  156. let supported_cstr: &'static CStr = get_supported_protocols_cstr();
  157. let supported: &str = supported_cstr.to_str().unwrap_or("");
  158. supported.parse()
  159. }
  160. pub fn len(&self) -> usize {
  161. self.0.len()
  162. }
  163. pub fn get(&self, protocol: &Protocol) -> Option<&ProtoSet> {
  164. self.0.get(protocol)
  165. }
  166. pub fn insert(&mut self, key: Protocol, value: ProtoSet) {
  167. self.0.insert(key, value);
  168. }
  169. pub fn remove(&mut self, key: &Protocol) -> Option<ProtoSet> {
  170. self.0.remove(key)
  171. }
  172. pub fn is_empty(&self) -> bool {
  173. self.0.is_empty()
  174. }
  175. }
  176. impl FromStr for ProtoEntry {
  177. type Err = ProtoverError;
  178. /// Parse a string of subprotocol types and their version numbers.
  179. ///
  180. /// # Inputs
  181. ///
  182. /// * A `protocol_entry` string, comprised of a keywords, an "=" sign, and
  183. /// one or more version numbers, each separated by a space. For example,
  184. /// `"Cons=3-4 HSDir=1"`.
  185. ///
  186. /// # Returns
  187. ///
  188. /// A `Result` whose `Ok` value is a `ProtoEntry`.
  189. /// Otherwise, the `Err` value of this `Result` is a `ProtoverError`.
  190. fn from_str(protocol_entry: &str) -> Result<ProtoEntry, ProtoverError> {
  191. let mut proto_entry: ProtoEntry = ProtoEntry::default();
  192. let entries = protocol_entry.split(' ');
  193. for entry in entries {
  194. let mut parts = entry.splitn(2, '=');
  195. let proto = match parts.next() {
  196. Some(n) => n,
  197. None => return Err(ProtoverError::Unparseable),
  198. };
  199. let vers = match parts.next() {
  200. Some(n) => n,
  201. None => return Err(ProtoverError::Unparseable),
  202. };
  203. let versions: ProtoSet = vers.parse()?;
  204. let proto_name: Protocol = proto.parse()?;
  205. proto_entry.insert(proto_name, versions);
  206. if proto_entry.len() > MAX_PROTOCOLS_TO_EXPAND {
  207. return Err(ProtoverError::ExceedsMax);
  208. }
  209. }
  210. Ok(proto_entry)
  211. }
  212. }
  213. /// Generate an implementation of `ToString` for either a `ProtoEntry` or an
  214. /// `UnvalidatedProtoEntry`.
  215. macro_rules! impl_to_string_for_proto_entry {
  216. ($t:ty) => {
  217. impl ToString for $t {
  218. fn to_string(&self) -> String {
  219. let mut parts: Vec<String> = Vec::new();
  220. for (protocol, versions) in self.iter() {
  221. parts.push(format!("{}={}", protocol.to_string(), versions.to_string()));
  222. }
  223. parts.sort_unstable();
  224. parts.join(" ")
  225. }
  226. }
  227. };
  228. }
  229. impl_to_string_for_proto_entry!(ProtoEntry);
  230. impl_to_string_for_proto_entry!(UnvalidatedProtoEntry);
  231. /// A `ProtoEntry`, but whose `Protocols` can be any `UnknownProtocol`, not just
  232. /// the supported ones enumerated in `Protocols`. The protocol versions are
  233. /// validated, however.
  234. #[derive(Clone, Debug, PartialEq, Eq)]
  235. pub struct UnvalidatedProtoEntry(HashMap<UnknownProtocol, ProtoSet>);
  236. impl Default for UnvalidatedProtoEntry {
  237. fn default() -> UnvalidatedProtoEntry {
  238. UnvalidatedProtoEntry(HashMap::new())
  239. }
  240. }
  241. impl UnvalidatedProtoEntry {
  242. /// Get an iterator over the `Protocol`s and their `ProtoSet`s in this `ProtoEntry`.
  243. pub fn iter(&self) -> hash_map::Iter<UnknownProtocol, ProtoSet> {
  244. self.0.iter()
  245. }
  246. pub fn get(&self, protocol: &UnknownProtocol) -> Option<&ProtoSet> {
  247. self.0.get(protocol)
  248. }
  249. pub fn insert(&mut self, key: UnknownProtocol, value: ProtoSet) {
  250. self.0.insert(key, value);
  251. }
  252. pub fn remove(&mut self, key: &UnknownProtocol) -> Option<ProtoSet> {
  253. self.0.remove(key)
  254. }
  255. pub fn is_empty(&self) -> bool {
  256. self.0.is_empty()
  257. }
  258. pub fn len(&self) -> usize {
  259. let mut total: usize = 0;
  260. for (_, versions) in self.iter() {
  261. total += versions.len();
  262. }
  263. total
  264. }
  265. /// Determine if we support every protocol a client supports, and if not,
  266. /// determine which protocols we do not have support for.
  267. ///
  268. /// # Returns
  269. ///
  270. /// Optionally, return parameters which the client supports but which we do not.
  271. ///
  272. /// # Examples
  273. /// ```
  274. /// use protover::UnvalidatedProtoEntry;
  275. ///
  276. /// let protocols: UnvalidatedProtoEntry = "LinkAuth=1 Microdesc=1-2 Relay=2".parse().unwrap();
  277. /// let unsupported: Option<UnvalidatedProtoEntry> = protocols.all_supported();
  278. /// assert_eq!(true, unsupported.is_none());
  279. ///
  280. /// let protocols: UnvalidatedProtoEntry = "Link=1-2 Wombat=9".parse().unwrap();
  281. /// let unsupported: Option<UnvalidatedProtoEntry> = protocols.all_supported();
  282. /// assert_eq!(true, unsupported.is_some());
  283. /// assert_eq!("Wombat=9", &unsupported.unwrap().to_string());
  284. /// ```
  285. pub fn all_supported(&self) -> Option<UnvalidatedProtoEntry> {
  286. let mut unsupported: UnvalidatedProtoEntry = UnvalidatedProtoEntry::default();
  287. let supported: ProtoEntry = match ProtoEntry::supported() {
  288. Ok(x) => x,
  289. Err(_) => return None,
  290. };
  291. for (protocol, versions) in self.iter() {
  292. let is_supported: Result<Protocol, ProtoverError> = protocol.0.parse();
  293. let supported_protocol: Protocol;
  294. // If the protocol wasn't even in the enum, then we definitely don't
  295. // know about it and don't support any of its versions.
  296. if is_supported.is_err() {
  297. if !versions.is_empty() {
  298. unsupported.insert(protocol.clone(), versions.clone());
  299. }
  300. continue;
  301. } else {
  302. supported_protocol = is_supported.unwrap();
  303. }
  304. let maybe_supported_versions: Option<&ProtoSet> = supported.get(&supported_protocol);
  305. let supported_versions: &ProtoSet;
  306. let mut unsupported_versions: ProtoSet;
  307. // If the protocol wasn't in the map, then we don't know about it
  308. // and don't support any of its versions. Add its versions to the
  309. // map (if it has versions).
  310. if maybe_supported_versions.is_none() {
  311. if !versions.is_empty() {
  312. unsupported.insert(protocol.clone(), versions.clone());
  313. }
  314. continue;
  315. } else {
  316. supported_versions = maybe_supported_versions.unwrap();
  317. }
  318. unsupported_versions = versions.clone();
  319. unsupported_versions.retain(|x| !supported_versions.contains(x));
  320. if !unsupported_versions.is_empty() {
  321. unsupported.insert(protocol.clone(), unsupported_versions);
  322. }
  323. }
  324. if unsupported.is_empty() {
  325. return None;
  326. }
  327. Some(unsupported)
  328. }
  329. /// Determine if we have support for some protocol and version.
  330. ///
  331. /// # Inputs
  332. ///
  333. /// * `proto`, an `UnknownProtocol` to test support for
  334. /// * `vers`, a `Version` which we will go on to determine whether the
  335. /// specified protocol supports.
  336. ///
  337. /// # Return
  338. ///
  339. /// Returns `true` iff this `UnvalidatedProtoEntry` includes support for the
  340. /// indicated protocol and version, and `false` otherwise.
  341. ///
  342. /// # Examples
  343. ///
  344. /// ```
  345. /// # use std::str::FromStr;
  346. /// use protover::*;
  347. /// # use protover::errors::ProtoverError;
  348. ///
  349. /// # fn do_test () -> Result<UnvalidatedProtoEntry, ProtoverError> {
  350. /// let proto: UnvalidatedProtoEntry = "Link=3-4 Cons=1 Doggo=3-5".parse()?;
  351. /// assert_eq!(true, proto.supports_protocol(&Protocol::Cons.into(), &1));
  352. /// assert_eq!(false, proto.supports_protocol(&Protocol::Cons.into(), &5));
  353. /// assert_eq!(true, proto.supports_protocol(&UnknownProtocol::from_str("Doggo")?, &4));
  354. /// # Ok(proto)
  355. /// # } fn main () { do_test(); }
  356. /// ```
  357. pub fn supports_protocol(&self, proto: &UnknownProtocol, vers: &Version) -> bool {
  358. let supported_versions: &ProtoSet = match self.get(proto) {
  359. Some(n) => n,
  360. None => return false,
  361. };
  362. supported_versions.contains(&vers)
  363. }
  364. /// As `UnvalidatedProtoEntry::supports_protocol()`, but also returns `true`
  365. /// if any later version of the protocol is supported.
  366. ///
  367. /// # Examples
  368. /// ```
  369. /// use protover::*;
  370. /// # use protover::errors::ProtoverError;
  371. ///
  372. /// # fn do_test () -> Result<UnvalidatedProtoEntry, ProtoverError> {
  373. /// let proto: UnvalidatedProtoEntry = "Link=3-4 Cons=5".parse()?;
  374. ///
  375. /// assert_eq!(true, proto.supports_protocol_or_later(&Protocol::Cons.into(), &5));
  376. /// assert_eq!(true, proto.supports_protocol_or_later(&Protocol::Cons.into(), &4));
  377. /// assert_eq!(false, proto.supports_protocol_or_later(&Protocol::Cons.into(), &6));
  378. /// # Ok(proto)
  379. /// # } fn main () { do_test(); }
  380. /// ```
  381. pub fn supports_protocol_or_later(&self, proto: &UnknownProtocol, vers: &Version) -> bool {
  382. let supported_versions: &ProtoSet = match self.get(&proto) {
  383. Some(n) => n,
  384. None => return false,
  385. };
  386. supported_versions.iter().any(|v| v.1 >= *vers)
  387. }
  388. /// Split a string containing (potentially) several protocols and their
  389. /// versions into a `Vec` of tuples of string in `(protocol, versions)`
  390. /// form.
  391. ///
  392. /// # Inputs
  393. ///
  394. /// A &str in the form `"Link=3-4 Cons=5"`.
  395. ///
  396. /// # Returns
  397. ///
  398. /// A `Result` whose `Ok` variant is a `Vec<(&str, &str)>` of `(protocol,
  399. /// versions)`, or whose `Err` variant is a `ProtoverError`.
  400. ///
  401. /// # Errors
  402. ///
  403. /// This will error with a `ProtoverError::Unparseable` if any of the
  404. /// following are true:
  405. ///
  406. /// * If a protocol name is an empty string, e.g. `"Cons=1,3 =3-5"`.
  407. /// * If an entry has no equals sign, e.g. `"Cons=1,3 Desc"`.
  408. /// * If there is leading or trailing whitespace, e.g. `" Cons=1,3 Link=3"`.
  409. /// * If there is any other extra whitespice, e.g. `"Cons=1,3 Link=3"`.
  410. fn parse_protocol_and_version_str<'a>(
  411. protocol_string: &'a str,
  412. ) -> Result<Vec<(&'a str, &'a str)>, ProtoverError> {
  413. let mut protovers: Vec<(&str, &str)> = Vec::new();
  414. for subproto in protocol_string.split(' ') {
  415. let mut parts = subproto.splitn(2, '=');
  416. let name = match parts.next() {
  417. Some("") => return Err(ProtoverError::Unparseable),
  418. Some(n) => n,
  419. None => return Err(ProtoverError::Unparseable),
  420. };
  421. let vers = match parts.next() {
  422. Some(n) => n,
  423. None => return Err(ProtoverError::Unparseable),
  424. };
  425. protovers.push((name, vers));
  426. }
  427. Ok(protovers)
  428. }
  429. }
  430. impl FromStr for UnvalidatedProtoEntry {
  431. type Err = ProtoverError;
  432. /// Parses a protocol list without validating the protocol names.
  433. ///
  434. /// # Inputs
  435. ///
  436. /// * `protocol_string`, a string comprised of keys and values, both which are
  437. /// strings. The keys are the protocol names while values are a string
  438. /// representation of the supported versions.
  439. ///
  440. /// The input is _not_ expected to be a subset of the Protocol types
  441. ///
  442. /// # Returns
  443. ///
  444. /// A `Result` whose `Ok` value is an `UnvalidatedProtoEntry`.
  445. ///
  446. /// The returned `Result`'s `Err` value is an `ProtoverError`.
  447. ///
  448. /// # Errors
  449. ///
  450. /// This function will error if:
  451. ///
  452. /// * The protocol string does not follow the "protocol_name=version_list"
  453. /// expected format, or
  454. /// * If the version string is malformed. See `impl FromStr for ProtoSet`.
  455. fn from_str(protocol_string: &str) -> Result<UnvalidatedProtoEntry, ProtoverError> {
  456. let mut parsed: UnvalidatedProtoEntry = UnvalidatedProtoEntry::default();
  457. let parts: Vec<(&str, &str)> =
  458. UnvalidatedProtoEntry::parse_protocol_and_version_str(protocol_string)?;
  459. for &(name, vers) in parts.iter() {
  460. let versions = ProtoSet::from_str(vers)?;
  461. let protocol = UnknownProtocol::from_str(name)?;
  462. parsed.insert(protocol, versions);
  463. }
  464. Ok(parsed)
  465. }
  466. }
  467. impl UnvalidatedProtoEntry {
  468. /// Create an `UnknownProtocol`, ignoring whether or not it
  469. /// exceeds MAX_PROTOCOL_NAME_LENGTH.
  470. pub(crate) fn from_str_any_len(
  471. protocol_string: &str,
  472. ) -> Result<UnvalidatedProtoEntry, ProtoverError> {
  473. let mut parsed: UnvalidatedProtoEntry = UnvalidatedProtoEntry::default();
  474. let parts: Vec<(&str, &str)> =
  475. UnvalidatedProtoEntry::parse_protocol_and_version_str(protocol_string)?;
  476. for &(name, vers) in parts.iter() {
  477. let versions = ProtoSet::from_str(vers)?;
  478. let protocol = UnknownProtocol::from_str_any_len(name)?;
  479. parsed.insert(protocol, versions);
  480. }
  481. Ok(parsed)
  482. }
  483. }
  484. /// Pretend a `ProtoEntry` is actually an `UnvalidatedProtoEntry`.
  485. impl From<ProtoEntry> for UnvalidatedProtoEntry {
  486. fn from(proto_entry: ProtoEntry) -> UnvalidatedProtoEntry {
  487. let mut unvalidated: UnvalidatedProtoEntry = UnvalidatedProtoEntry::default();
  488. for (protocol, versions) in proto_entry.iter() {
  489. unvalidated.insert(UnknownProtocol::from(protocol.clone()), versions.clone());
  490. }
  491. unvalidated
  492. }
  493. }
  494. /// A mapping of protocols to a count of how many times each of their `Version`s
  495. /// were voted for or supported.
  496. ///
  497. /// # Warning
  498. ///
  499. /// The "protocols" are *not* guaranteed to be known/supported `Protocol`s, in
  500. /// order to allow new subprotocols to be introduced even if Directory
  501. /// Authorities don't yet know of them.
  502. pub struct ProtoverVote(HashMap<UnknownProtocol, HashMap<Version, usize>>);
  503. impl Default for ProtoverVote {
  504. fn default() -> ProtoverVote {
  505. ProtoverVote(HashMap::new())
  506. }
  507. }
  508. impl IntoIterator for ProtoverVote {
  509. type Item = (UnknownProtocol, HashMap<Version, usize>);
  510. type IntoIter = hash_map::IntoIter<UnknownProtocol, HashMap<Version, usize>>;
  511. fn into_iter(self) -> Self::IntoIter {
  512. self.0.into_iter()
  513. }
  514. }
  515. impl ProtoverVote {
  516. pub fn entry(
  517. &mut self,
  518. key: UnknownProtocol,
  519. ) -> hash_map::Entry<UnknownProtocol, HashMap<Version, usize>> {
  520. self.0.entry(key)
  521. }
  522. /// Protocol voting implementation.
  523. ///
  524. /// Given a slice of `UnvalidatedProtoEntry`s and a vote `threshold`, return
  525. /// a new `UnvalidatedProtoEntry` encoding all of the protocols that are
  526. /// listed by at least `threshold` of the inputs.
  527. ///
  528. /// # Examples
  529. ///
  530. /// ```
  531. /// use protover::ProtoverVote;
  532. /// use protover::UnvalidatedProtoEntry;
  533. ///
  534. /// let protos: &[UnvalidatedProtoEntry] = &["Link=3-4".parse().unwrap(),
  535. /// "Link=3".parse().unwrap()];
  536. /// let vote = ProtoverVote::compute(protos, &2);
  537. /// assert_eq!("Link=3", vote.to_string());
  538. /// ```
  539. // C_RUST_COUPLED: protover.c protover_compute_vote
  540. pub fn compute(
  541. proto_entries: &[UnvalidatedProtoEntry],
  542. threshold: &usize,
  543. ) -> UnvalidatedProtoEntry {
  544. let mut all_count: ProtoverVote = ProtoverVote::default();
  545. let mut final_output: UnvalidatedProtoEntry = UnvalidatedProtoEntry::default();
  546. if proto_entries.is_empty() {
  547. return final_output;
  548. }
  549. // parse and collect all of the protos and their versions and collect them
  550. for vote in proto_entries {
  551. // C_RUST_DIFFERS: This doesn't actually differ, bu this check on
  552. // the total is here to make it match. Because the C version calls
  553. // expand_protocol_list() which checks if there would be too many
  554. // subprotocols *or* individual version numbers, i.e. more than
  555. // MAX_PROTOCOLS_TO_EXPAND, and does this *per vote*, we need to
  556. // match it's behaviour and ensure we're not allowing more than it
  557. // would.
  558. if vote.len() > MAX_PROTOCOLS_TO_EXPAND {
  559. continue;
  560. }
  561. for (protocol, versions) in vote.iter() {
  562. let supported_vers: &mut HashMap<Version, usize> =
  563. all_count.entry(protocol.clone()).or_insert(HashMap::new());
  564. for version in versions.clone().expand() {
  565. let counter: &mut usize = supported_vers.entry(version).or_insert(0);
  566. *counter += 1;
  567. }
  568. }
  569. }
  570. for (protocol, mut versions) in all_count {
  571. // Go through and remove versions that are less than the threshold
  572. versions.retain(|_, count| *count as usize >= *threshold);
  573. if versions.len() > 0 {
  574. let voted_versions: Vec<Version> = versions.keys().cloned().collect();
  575. let voted_protoset: ProtoSet = ProtoSet::from(voted_versions);
  576. final_output.insert(protocol, voted_protoset);
  577. }
  578. }
  579. final_output
  580. }
  581. }
  582. /// Returns a boolean indicating whether the given protocol and version is
  583. /// supported in any of the existing Tor protocols
  584. ///
  585. /// # Examples
  586. /// ```
  587. /// use protover::is_supported_here;
  588. /// use protover::Protocol;
  589. ///
  590. /// let is_supported = is_supported_here(&Protocol::Link, &10);
  591. /// assert_eq!(false, is_supported);
  592. ///
  593. /// let is_supported = is_supported_here(&Protocol::Link, &1);
  594. /// assert_eq!(true, is_supported);
  595. /// ```
  596. pub fn is_supported_here(proto: &Protocol, vers: &Version) -> bool {
  597. let currently_supported: ProtoEntry = match ProtoEntry::supported() {
  598. Ok(result) => result,
  599. Err(_) => return false,
  600. };
  601. let supported_versions = match currently_supported.get(proto) {
  602. Some(n) => n,
  603. None => return false,
  604. };
  605. supported_versions.contains(vers)
  606. }
  607. /// Since older versions of Tor cannot infer their own subprotocols,
  608. /// determine which subprotocols are supported by older Tor versions.
  609. ///
  610. /// # Inputs
  611. ///
  612. /// * `version`, a string comprised of "[0-9a-z.-]"
  613. ///
  614. /// # Returns
  615. ///
  616. /// A `&'static CStr` encoding a list of protocol names and supported
  617. /// versions. The string takes the following format:
  618. ///
  619. /// "HSDir=1-1 LinkAuth=1"
  620. ///
  621. /// This function returns the protocols that are supported by the version input,
  622. /// only for tor versions older than `FIRST_TOR_VERSION_TO_ADVERTISE_PROTOCOLS`
  623. /// (but not older than 0.2.4.19). For newer tors (or older than 0.2.4.19), it
  624. /// returns an empty string.
  625. ///
  626. /// # Note
  627. ///
  628. /// This function is meant to be called for/within FFI code. If you'd
  629. /// like to use this code in Rust, please see `compute_for_old_tor()`.
  630. //
  631. // C_RUST_COUPLED: src/rust/protover.c `compute_for_old_tor`
  632. pub(crate) fn compute_for_old_tor_cstr(version: &str) -> &'static CStr {
  633. let empty: &'static CStr = cstr!("");
  634. if c_tor_version_as_new_as(version, FIRST_TOR_VERSION_TO_ADVERTISE_PROTOCOLS) {
  635. return empty;
  636. }
  637. if c_tor_version_as_new_as(version, "0.2.9.1-alpha") {
  638. return cstr!(
  639. "Cons=1-2 Desc=1-2 DirCache=1 HSDir=1 HSIntro=3 HSRend=1-2 \
  640. Link=1-4 LinkAuth=1 Microdesc=1-2 Relay=1-2"
  641. );
  642. }
  643. if c_tor_version_as_new_as(version, "0.2.7.5") {
  644. return cstr!(
  645. "Cons=1-2 Desc=1-2 DirCache=1 HSDir=1 HSIntro=3 HSRend=1 \
  646. Link=1-4 LinkAuth=1 Microdesc=1-2 Relay=1-2"
  647. );
  648. }
  649. if c_tor_version_as_new_as(version, "0.2.4.19") {
  650. return cstr!(
  651. "Cons=1 Desc=1 DirCache=1 HSDir=1 HSIntro=3 HSRend=1 \
  652. Link=1-4 LinkAuth=1 Microdesc=1 Relay=1-2"
  653. );
  654. }
  655. empty
  656. }
  657. /// Since older versions of Tor cannot infer their own subprotocols,
  658. /// determine which subprotocols are supported by older Tor versions.
  659. ///
  660. /// # Inputs
  661. ///
  662. /// * `version`, a string comprised of "[0-9a-z.-]"
  663. ///
  664. /// # Returns
  665. ///
  666. /// A `Result` whose `Ok` value is an `&'static str` encoding a list of protocol
  667. /// names and supported versions. The string takes the following format:
  668. ///
  669. /// "HSDir=1-1 LinkAuth=1"
  670. ///
  671. /// This function returns the protocols that are supported by the version input,
  672. /// only for tor versions older than `FIRST_TOR_VERSION_TO_ADVERTISE_PROTOCOLS`.
  673. /// (but not older than 0.2.4.19). For newer tors (or older than 0.2.4.19), its
  674. /// `Ok` `Result` contains an empty string.
  675. ///
  676. /// Otherwise, its `Err` contains a `ProtoverError::Unparseable` if the
  677. /// `version` string was invalid utf-8.
  678. ///
  679. /// # Note
  680. ///
  681. /// This function is meant to be called for/within non-FFI Rust code.
  682. //
  683. // C_RUST_COUPLED: src/rust/protover.c `compute_for_old_tor`
  684. pub fn compute_for_old_tor(version: &str) -> Result<&'static str, ProtoverError> {
  685. // .to_str() fails with a Utf8Error if it couldn't validate the
  686. // utf-8, so convert that here into an Unparseable ProtoverError.
  687. compute_for_old_tor_cstr(version)
  688. .to_str()
  689. .or(Err(ProtoverError::Unparseable))
  690. }
  691. #[cfg(test)]
  692. mod test {
  693. use std::str::FromStr;
  694. use std::string::ToString;
  695. use super::*;
  696. macro_rules! parse_proto {
  697. ($e:expr) => {{
  698. let proto: Result<UnknownProtocol, _> = $e.parse();
  699. let proto2 = UnknownProtocol::from_str_any_len($e);
  700. assert_eq!(proto, proto2);
  701. proto
  702. }};
  703. }
  704. #[test]
  705. fn test_protocol_from_str() {
  706. assert!(parse_proto!("Cons").is_ok());
  707. assert!(parse_proto!("123").is_ok());
  708. assert!(parse_proto!("1-2-3").is_ok());
  709. let err = Err(ProtoverError::InvalidProtocol);
  710. assert_eq!(err, parse_proto!("a_b_c"));
  711. assert_eq!(err, parse_proto!("a b"));
  712. assert_eq!(err, parse_proto!("a,"));
  713. assert_eq!(err, parse_proto!("b."));
  714. assert_eq!(err, parse_proto!("é"));
  715. }
  716. macro_rules! assert_protoentry_is_parseable {
  717. ($e:expr) => {
  718. let protoentry: Result<ProtoEntry, ProtoverError> = $e.parse();
  719. assert!(protoentry.is_ok(), format!("{:?}", protoentry.err()));
  720. };
  721. }
  722. macro_rules! assert_protoentry_is_unparseable {
  723. ($e:expr) => {
  724. let protoentry: Result<ProtoEntry, ProtoverError> = $e.parse();
  725. assert!(protoentry.is_err());
  726. };
  727. }
  728. #[test]
  729. fn test_protoentry_from_str_multiple_protocols_multiple_versions() {
  730. assert_protoentry_is_parseable!("Cons=3-4 Link=1,3-5");
  731. }
  732. #[test]
  733. fn test_protoentry_from_str_empty() {
  734. assert_protoentry_is_unparseable!("");
  735. }
  736. #[test]
  737. fn test_protoentry_from_str_single_protocol_single_version() {
  738. assert_protoentry_is_parseable!("HSDir=1");
  739. }
  740. #[test]
  741. fn test_protoentry_from_str_unknown_protocol() {
  742. assert_protoentry_is_unparseable!("Ducks=5-7,8");
  743. }
  744. #[test]
  745. fn test_protoentry_from_str_allowed_number_of_versions() {
  746. assert_protoentry_is_parseable!("Desc=1-4294967294");
  747. }
  748. #[test]
  749. fn test_protoentry_from_str_too_many_versions() {
  750. assert_protoentry_is_unparseable!("Desc=1-4294967295");
  751. }
  752. #[test]
  753. fn test_protoentry_from_str_() {
  754. assert_protoentry_is_unparseable!("");
  755. }
  756. #[test]
  757. fn test_protoentry_all_supported_single_protocol_single_version() {
  758. let protocol: UnvalidatedProtoEntry = "Cons=1".parse().unwrap();
  759. let unsupported: Option<UnvalidatedProtoEntry> = protocol.all_supported();
  760. assert_eq!(true, unsupported.is_none());
  761. }
  762. #[test]
  763. fn test_protoentry_all_supported_multiple_protocol_multiple_versions() {
  764. let protocols: UnvalidatedProtoEntry = "Link=3-4 Desc=2".parse().unwrap();
  765. let unsupported: Option<UnvalidatedProtoEntry> = protocols.all_supported();
  766. assert_eq!(true, unsupported.is_none());
  767. }
  768. #[test]
  769. fn test_protoentry_all_supported_three_values() {
  770. let protocols: UnvalidatedProtoEntry = "LinkAuth=1 Microdesc=1-2 Relay=2".parse().unwrap();
  771. let unsupported: Option<UnvalidatedProtoEntry> = protocols.all_supported();
  772. assert_eq!(true, unsupported.is_none());
  773. }
  774. #[test]
  775. fn test_protoentry_all_supported_unknown_protocol() {
  776. let protocols: UnvalidatedProtoEntry = "Wombat=9".parse().unwrap();
  777. let unsupported: Option<UnvalidatedProtoEntry> = protocols.all_supported();
  778. assert_eq!(true, unsupported.is_some());
  779. assert_eq!("Wombat=9", &unsupported.unwrap().to_string());
  780. }
  781. #[test]
  782. fn test_protoentry_all_supported_unsupported_high_version() {
  783. let protocols: UnvalidatedProtoEntry = "HSDir=12-100".parse().unwrap();
  784. let unsupported: Option<UnvalidatedProtoEntry> = protocols.all_supported();
  785. assert_eq!(true, unsupported.is_some());
  786. assert_eq!("HSDir=12-100", &unsupported.unwrap().to_string());
  787. }
  788. #[test]
  789. fn test_protoentry_all_supported_unsupported_low_version() {
  790. let protocols: UnvalidatedProtoEntry = "HSIntro=2-3".parse().unwrap();
  791. let unsupported: Option<UnvalidatedProtoEntry> = protocols.all_supported();
  792. assert_eq!(true, unsupported.is_some());
  793. assert_eq!("HSIntro=2", &unsupported.unwrap().to_string());
  794. }
  795. #[test]
  796. fn test_contract_protocol_list() {
  797. let mut versions = "";
  798. assert_eq!(
  799. String::from(versions),
  800. ProtoSet::from_str(&versions).unwrap().to_string()
  801. );
  802. versions = "1";
  803. assert_eq!(
  804. String::from(versions),
  805. ProtoSet::from_str(&versions).unwrap().to_string()
  806. );
  807. versions = "1-2";
  808. assert_eq!(
  809. String::from(versions),
  810. ProtoSet::from_str(&versions).unwrap().to_string()
  811. );
  812. versions = "1,3";
  813. assert_eq!(
  814. String::from(versions),
  815. ProtoSet::from_str(&versions).unwrap().to_string()
  816. );
  817. versions = "1-4";
  818. assert_eq!(
  819. String::from(versions),
  820. ProtoSet::from_str(&versions).unwrap().to_string()
  821. );
  822. versions = "1,3,5-7";
  823. assert_eq!(
  824. String::from(versions),
  825. ProtoSet::from_str(&versions).unwrap().to_string()
  826. );
  827. versions = "1-3,500";
  828. assert_eq!(
  829. String::from(versions),
  830. ProtoSet::from_str(&versions).unwrap().to_string()
  831. );
  832. }
  833. }