std: make sure HashMap from_iter uses random initialization by default
It turns out that HashMap's from_iter implementation was being initialized without the sip keys being randomized. This adds a custom default hasher that should avoid this potential vulnerability.
This commit is contained in:
parent
287dcb77b3
commit
8284ef63a5
@ -16,15 +16,13 @@ use collections::{Collection, Mutable, Set, MutableSet, Map, MutableMap};
|
||||
use default::Default;
|
||||
use fmt::Show;
|
||||
use fmt;
|
||||
use hash::{Hash, Hasher, sip};
|
||||
use hash::{Hash, Hasher, RandomSipHasher};
|
||||
use iter::{Iterator, FilterMap, Chain, Repeat, Zip, Extendable};
|
||||
use iter::{range, range_inclusive, FromIterator};
|
||||
use iter;
|
||||
use mem::replace;
|
||||
use num;
|
||||
use option::{Some, None, Option};
|
||||
use rand::Rng;
|
||||
use rand;
|
||||
use result::{Ok, Err};
|
||||
|
||||
mod table {
|
||||
@ -733,7 +731,7 @@ impl DefaultResizePolicy {
|
||||
/// }
|
||||
/// ```
|
||||
#[deriving(Clone)]
|
||||
pub struct HashMap<K, V, H = sip::SipHasher> {
|
||||
pub struct HashMap<K, V, H = RandomSipHasher> {
|
||||
// All hashes are keyed on these values, to prevent hash collision attacks.
|
||||
hasher: H,
|
||||
|
||||
@ -1033,18 +1031,15 @@ impl<K: Eq + Hash<S>, V, S, H: Hasher<S>> MutableMap<K, V> for HashMap<K, V, H>
|
||||
|
||||
}
|
||||
|
||||
impl<K: Hash + Eq, V> HashMap<K, V, sip::SipHasher> {
|
||||
impl<K: Hash + Eq, V> HashMap<K, V, RandomSipHasher> {
|
||||
/// Create an empty HashMap.
|
||||
pub fn new() -> HashMap<K, V, sip::SipHasher> {
|
||||
pub fn new() -> HashMap<K, V, RandomSipHasher> {
|
||||
HashMap::with_capacity(INITIAL_CAPACITY)
|
||||
}
|
||||
|
||||
/// Creates an empty hash map with the given initial capacity.
|
||||
pub fn with_capacity(capacity: uint) -> HashMap<K, V, sip::SipHasher> {
|
||||
let mut r = rand::task_rng();
|
||||
let r0 = r.gen();
|
||||
let r1 = r.gen();
|
||||
let hasher = sip::SipHasher::new_with_keys(r0, r1);
|
||||
pub fn with_capacity(capacity: uint) -> HashMap<K, V, RandomSipHasher> {
|
||||
let hasher = RandomSipHasher::new();
|
||||
HashMap::with_capacity_and_hasher(capacity, hasher)
|
||||
}
|
||||
}
|
||||
@ -1489,7 +1484,7 @@ pub type SetMoveItems<K> =
|
||||
/// HashMap where the value is (). As with the `HashMap` type, a `HashSet`
|
||||
/// requires that the elements implement the `Eq` and `Hash` traits.
|
||||
#[deriving(Clone)]
|
||||
pub struct HashSet<T, H = sip::SipHasher> {
|
||||
pub struct HashSet<T, H = RandomSipHasher> {
|
||||
map: HashMap<T, (), H>
|
||||
}
|
||||
|
||||
@ -1529,15 +1524,15 @@ impl<T: Eq + Hash<S>, S, H: Hasher<S>> MutableSet<T> for HashSet<T, H> {
|
||||
fn remove(&mut self, value: &T) -> bool { self.map.remove(value) }
|
||||
}
|
||||
|
||||
impl<T: Hash + Eq> HashSet<T, sip::SipHasher> {
|
||||
impl<T: Hash + Eq> HashSet<T, RandomSipHasher> {
|
||||
/// Create an empty HashSet
|
||||
pub fn new() -> HashSet<T, sip::SipHasher> {
|
||||
pub fn new() -> HashSet<T, RandomSipHasher> {
|
||||
HashSet::with_capacity(INITIAL_CAPACITY)
|
||||
}
|
||||
|
||||
/// Create an empty HashSet with space for at least `n` elements in
|
||||
/// the hash table.
|
||||
pub fn with_capacity(capacity: uint) -> HashSet<T, sip::SipHasher> {
|
||||
pub fn with_capacity(capacity: uint) -> HashSet<T, RandomSipHasher> {
|
||||
HashSet { map: HashMap::with_capacity(capacity) }
|
||||
}
|
||||
}
|
||||
|
51
src/libstd/hash.rs
Normal file
51
src/libstd/hash.rs
Normal file
@ -0,0 +1,51 @@
|
||||
// Copyright 2014 The Rust Project Developers. See the COPYRIGHT
|
||||
// file at the top-level directory of this distribution and at
|
||||
// http://rust-lang.org/COPYRIGHT.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
||||
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
||||
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
||||
// option. This file may not be copied, modified, or distributed
|
||||
// except according to those terms.
|
||||
|
||||
//! Generic hashing support.
|
||||
|
||||
pub use core_collections::hash::{Hash, Hasher, Writer, hash, sip};
|
||||
|
||||
use default::Default;
|
||||
use rand::Rng;
|
||||
use rand;
|
||||
|
||||
/// `RandomSipHasher` computes the SipHash algorithm from a stream of bytes
|
||||
/// initialized with random keys.
|
||||
#[deriving(Clone)]
|
||||
pub struct RandomSipHasher {
|
||||
hasher: sip::SipHasher,
|
||||
}
|
||||
|
||||
impl RandomSipHasher {
|
||||
/// Construct a new `RandomSipHasher` that is initialized with random keys.
|
||||
#[inline]
|
||||
pub fn new() -> RandomSipHasher {
|
||||
let mut r = rand::task_rng();
|
||||
let r0 = r.gen();
|
||||
let r1 = r.gen();
|
||||
RandomSipHasher {
|
||||
hasher: sip::SipHasher::new_with_keys(r0, r1),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Hasher<sip::SipState> for RandomSipHasher {
|
||||
#[inline]
|
||||
fn hash<T: Hash<sip::SipState>>(&self, value: &T) -> u64 {
|
||||
self.hasher.hash(value)
|
||||
}
|
||||
}
|
||||
|
||||
impl Default for RandomSipHasher {
|
||||
#[inline]
|
||||
fn default() -> RandomSipHasher {
|
||||
RandomSipHasher::new()
|
||||
}
|
||||
}
|
@ -166,7 +166,6 @@ pub use core::option;
|
||||
pub use alloc::owned;
|
||||
pub use alloc::rc;
|
||||
|
||||
pub use core_collections::hash;
|
||||
pub use core_collections::slice;
|
||||
pub use core_collections::str;
|
||||
pub use core_collections::string;
|
||||
@ -236,6 +235,7 @@ pub mod to_str;
|
||||
/* Common data structures */
|
||||
|
||||
pub mod collections;
|
||||
pub mod hash;
|
||||
|
||||
/* Tasks and communication */
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user