Skip to content

Commit

Permalink
Fix stable hashing of AllocIds
Browse files Browse the repository at this point in the history
  • Loading branch information
oli-obk committed Mar 28, 2018
1 parent a0b0f5f commit fa60b72
Show file tree
Hide file tree
Showing 3 changed files with 50 additions and 12 deletions.
4 changes: 4 additions & 0 deletions src/librustc/ich/hcx.rs
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ use hir::map::definitions::Definitions;
use ich::{self, CachingCodemapView, Fingerprint};
use middle::cstore::CrateStore;
use ty::{TyCtxt, fast_reject};
use mir::interpret::AllocId;
use session::Session;

use std::cmp::Ord;
Expand Down Expand Up @@ -59,6 +60,8 @@ pub struct StableHashingContext<'a> {
// CachingCodemapView, so we initialize it lazily.
raw_codemap: &'a CodeMap,
caching_codemap: Option<CachingCodemapView<'a>>,

pub(super) alloc_id_recursion_tracker: FxHashSet<AllocId>,
}

#[derive(PartialEq, Eq, Clone, Copy)]
Expand Down Expand Up @@ -102,6 +105,7 @@ impl<'a> StableHashingContext<'a> {
hash_spans: hash_spans_initial,
hash_bodies: true,
node_id_hashing_mode: NodeIdHashingMode::HashDefPath,
alloc_id_recursion_tracker: Default::default(),
}
}

Expand Down
30 changes: 18 additions & 12 deletions src/librustc/ich/impls_ty.rs
Original file line number Diff line number Diff line change
Expand Up @@ -379,13 +379,13 @@ impl_stable_hash_for!(struct mir::interpret::MemoryPointer {
});

enum AllocDiscriminant {
Static,
Constant,
Alloc,
ExternStatic,
Function,
}
impl_stable_hash_for!(enum self::AllocDiscriminant {
Static,
Constant,
Alloc,
ExternStatic,
Function
});

Expand All @@ -397,17 +397,23 @@ impl<'a> HashStable<StableHashingContext<'a>> for mir::interpret::AllocId {
) {
ty::tls::with_opt(|tcx| {
let tcx = tcx.expect("can't hash AllocIds during hir lowering");
if let Some(def_id) = tcx.interpret_interner.get_corresponding_static_def_id(*self) {
AllocDiscriminant::Static.hash_stable(hcx, hasher);
// statics are unique via their DefId
def_id.hash_stable(hcx, hasher);
} else if let Some(alloc) = tcx.interpret_interner.get_alloc(*self) {
// not a static, can't be recursive, hash the allocation
AllocDiscriminant::Constant.hash_stable(hcx, hasher);
alloc.hash_stable(hcx, hasher);
if let Some(alloc) = tcx.interpret_interner.get_alloc(*self) {
AllocDiscriminant::Alloc.hash_stable(hcx, hasher);
if !hcx.alloc_id_recursion_tracker.insert(*self) {
tcx
.interpret_interner
.get_corresponding_static_def_id(*self)
.hash_stable(hcx, hasher);
alloc.hash_stable(hcx, hasher);
assert!(hcx.alloc_id_recursion_tracker.remove(self));
}
} else if let Some(inst) = tcx.interpret_interner.get_fn(*self) {
AllocDiscriminant::Function.hash_stable(hcx, hasher);
inst.hash_stable(hcx, hasher);
} else if let Some(def_id) = tcx.interpret_interner
.get_corresponding_static_def_id(*self) {
AllocDiscriminant::ExternStatic.hash_stable(hcx, hasher);
def_id.hash_stable(hcx, hasher);
} else {
bug!("no allocation for {}", self);
}
Expand Down
28 changes: 28 additions & 0 deletions src/test/incremental/static_stable_hash/issue-49301.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
// Copyright 2018 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.

// https://github.com/rust-lang/rust/issues/49081

// revisions:rpass1 rpass2

#[cfg(rpass1)]
pub static A: &str = "hello";
#[cfg(rpass2)]
pub static A: &str = "xxxxx";

#[cfg(rpass1)]
fn main() {
assert_eq!(A, "hello");
}

#[cfg(rpass2)]
fn main() {
assert_eq!(A, "xxxxx");
}

0 comments on commit fa60b72

Please sign in to comment.