flowistry_pdg/
rustc_impls.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
use std::fmt;

use super::rustc_proxies::*;
use crate::{
    pdg::GlobalLocation,
    rustc::{def_id, hir, middle::ty::tls, mir},
};

pub fn bbref_to_u32(r: &mir::BasicBlock) -> u32 {
    r.as_u32()
}

impl From<BasicBlock> for mir::BasicBlock {
    fn from(bb: BasicBlock) -> mir::BasicBlock {
        mir::BasicBlock::from_u32(bb.private)
    }
}

impl From<Location> for mir::Location {
    fn from(
        Location {
            block,
            statement_index,
        }: Location,
    ) -> mir::Location {
        mir::Location {
            block,
            statement_index,
        }
    }
}

impl From<mir::Location> for Location {
    fn from(
        mir::Location {
            block,
            statement_index,
        }: mir::Location,
    ) -> Location {
        Location {
            block,
            statement_index,
        }
    }
}

pub fn item_local_id_as_u32(i: &hir::ItemLocalId) -> u32 {
    i.as_u32()
}

impl From<ItemLocalId> for hir::ItemLocalId {
    fn from(proxy: ItemLocalId) -> hir::ItemLocalId {
        hir::ItemLocalId::from_u32(proxy.private)
    }
}

pub fn def_index_as_u32(i: &def_id::DefIndex) -> u32 {
    i.as_u32()
}

pub fn crate_num_as_u32(num: &hir::def_id::CrateNum) -> u32 {
    (*num).into()
}

impl From<CrateNum> for hir::def_id::CrateNum {
    fn from(value: CrateNum) -> Self {
        hir::def_id::CrateNum::from_u32(value.private)
    }
}

impl From<DefIndex> for def_id::DefIndex {
    fn from(proxy: DefIndex) -> def_id::DefIndex {
        def_id::DefIndex::from_u32(proxy.private)
    }
}

impl fmt::Display for GlobalLocation {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        tls::with_opt(|opt_tcx| match opt_tcx {
            Some(tcx) => match tcx.opt_item_name(self.function) {
                Some(name) => name.fmt(f),
                None => write!(f, "<closure>"),
            },
            None => write!(f, "{:?}", self.function),
        })?;
        write!(f, "::{}", self.location)
    }
}