6 use rustc_span::{FileName, SourceFileHash, SourceFileHashAlgorithm};
8 use cranelift_codegen::ir::{StackSlots, ValueLabel, ValueLoc};
9 use cranelift_codegen::isa::TargetIsa;
10 use cranelift_codegen::ValueLocRange;
13 self, Address, AttributeValue, DwarfUnit, Expression, LineProgram, LineString, Location,
14 LocationList, Range, RangeList, UnitEntryId, Writer, FileInfo,
16 use gimli::{Encoding, Format, LineEncoding, RunTimeEndian, X86_64};
18 pub(crate) use emit::{DebugReloc, DebugRelocName};
20 fn target_endian(tcx: TyCtxt<'_>) -> RunTimeEndian {
21 use rustc_target::abi::Endian;
23 match tcx.data_layout.endian {
24 Endian::Big => RunTimeEndian::Big,
25 Endian::Little => RunTimeEndian::Little,
29 const MD5_LEN: usize = 16;
31 pub(crate) struct DebugContext<'tcx> {
34 endian: RunTimeEndian,
35 symbols: indexmap::IndexMap<FuncId, String>,
38 unit_range_list: RangeList,
40 types: FxHashMap<Ty<'tcx>, UnitEntryId>,
43 impl<'tcx> DebugContext<'tcx> {
44 pub(crate) fn new(tcx: TyCtxt<'tcx>, address_size: u8) -> Self {
45 let encoding = Encoding {
46 format: Format::Dwarf32,
47 // TODO: this should be configurable
48 // macOS doesn't seem to support DWARF > 3
49 // 5 version is required for md5 file hash
50 version: if tcx.sess.target.target.options.is_like_osx {
58 let mut dwarf = DwarfUnit::new(encoding);
60 // FIXME: how to get version when building out of tree?
61 // Normally this would use option_env!("CFG_VERSION").
62 let producer = format!("cg_clif (rustc {})", "unknown version");
63 let comp_dir = tcx.sess.working_dir.0.to_string_lossy().into_owned();
64 let (name, md5) = match tcx.sess.local_crate_source_file.clone() {
66 let name = path.to_string_lossy().into_owned();
69 .get_source_file(&FileName::Real(path))
71 .filter(|h| matches!(h, SourceFileHash { kind: SourceFileHashAlgorithm::Md5, .. }))
73 let mut buf = [0u8; MD5_LEN];
74 buf.copy_from_slice(h.hash_bytes());
79 None => (tcx.crate_name(LOCAL_CRATE).to_string(), None),
82 let mut line_program = LineProgram::new(
84 LineEncoding::default(),
85 LineString::new(comp_dir.as_bytes(), encoding, &mut dwarf.line_strings),
86 LineString::new(name.as_bytes(), encoding, &mut dwarf.line_strings),
90 md5: md5.unwrap_or_default(),
93 line_program.file_has_md5 = md5.is_some();
95 dwarf.unit.line_program = line_program;
98 let name = dwarf.strings.add(name);
99 let comp_dir = dwarf.strings.add(comp_dir);
101 let root = dwarf.unit.root();
102 let root = dwarf.unit.get_mut(root);
104 gimli::DW_AT_producer,
105 AttributeValue::StringRef(dwarf.strings.add(producer)),
108 gimli::DW_AT_language,
109 AttributeValue::Language(gimli::DW_LANG_Rust),
111 root.set(gimli::DW_AT_name, AttributeValue::StringRef(name));
112 root.set(gimli::DW_AT_comp_dir, AttributeValue::StringRef(comp_dir));
115 AttributeValue::Address(Address::Constant(0)),
122 endian: target_endian(tcx),
123 symbols: indexmap::IndexMap::new(),
126 unit_range_list: RangeList(Vec::new()),
128 types: FxHashMap::default(),
132 fn dwarf_ty(&mut self, ty: Ty<'tcx>) -> UnitEntryId {
133 if let Some(type_id) = self.types.get(ty) {
137 let new_entry = |dwarf: &mut DwarfUnit, tag| dwarf.unit.add(dwarf.unit.root(), tag);
139 let primitive = |dwarf: &mut DwarfUnit, ate| {
140 let type_id = new_entry(dwarf, gimli::DW_TAG_base_type);
141 let type_entry = dwarf.unit.get_mut(type_id);
142 type_entry.set(gimli::DW_AT_encoding, AttributeValue::Encoding(ate));
146 let name = format!("{}", ty);
147 let layout = self.tcx.layout_of(ParamEnv::reveal_all().and(ty)).unwrap();
149 let type_id = match ty.kind {
150 ty::Bool => primitive(&mut self.dwarf, gimli::DW_ATE_boolean),
151 ty::Char => primitive(&mut self.dwarf, gimli::DW_ATE_UTF),
152 ty::Uint(_) => primitive(&mut self.dwarf, gimli::DW_ATE_unsigned),
153 ty::Int(_) => primitive(&mut self.dwarf, gimli::DW_ATE_signed),
154 ty::Float(_) => primitive(&mut self.dwarf, gimli::DW_ATE_float),
155 ty::Ref(_, pointee_ty, _mutbl)
156 | ty::RawPtr(ty::TypeAndMut {
160 let type_id = new_entry(&mut self.dwarf, gimli::DW_TAG_pointer_type);
162 // Ensure that type is inserted before recursing to avoid duplicates
163 self.types.insert(ty, type_id);
165 let pointee = self.dwarf_ty(pointee_ty);
167 let type_entry = self.dwarf.unit.get_mut(type_id);
169 //type_entry.set(gimli::DW_AT_mutable, AttributeValue::Flag(mutbl == rustc_hir::Mutability::Mut));
170 type_entry.set(gimli::DW_AT_type, AttributeValue::ThisUnitEntryRef(pointee));
174 ty::Adt(adt_def, _substs) if adt_def.is_struct() && !layout.is_unsized() => {
175 let type_id = new_entry(&mut self.dwarf, gimli::DW_TAG_structure_type);
177 // Ensure that type is inserted before recursing to avoid duplicates
178 self.types.insert(ty, type_id);
180 let variant = adt_def.non_enum_variant();
182 for (field_idx, field_def) in variant.fields.iter().enumerate() {
183 let field_offset = layout.fields.offset(field_idx);
184 let field_layout = layout.field(&layout::LayoutCx {
186 param_env: ParamEnv::reveal_all(),
187 }, field_idx).unwrap();
189 let field_type = self.dwarf_ty(field_layout.ty);
191 let field_id = self.dwarf.unit.add(type_id, gimli::DW_TAG_member);
192 let field_entry = self.dwarf.unit.get_mut(field_id);
194 field_entry.set(gimli::DW_AT_name, AttributeValue::String(field_def.ident.as_str().to_string().into_bytes()));
195 field_entry.set(gimli::DW_AT_data_member_location, AttributeValue::Udata(field_offset.bytes()));
196 field_entry.set(gimli::DW_AT_type, AttributeValue::ThisUnitEntryRef(field_type));
201 _ => new_entry(&mut self.dwarf, gimli::DW_TAG_structure_type),
204 let type_entry = self.dwarf.unit.get_mut(type_id);
206 type_entry.set(gimli::DW_AT_name, AttributeValue::String(name.into_bytes()));
208 gimli::DW_AT_byte_size,
209 AttributeValue::Udata(layout.size.bytes()),
212 self.types.insert(ty, type_id);
218 pub(crate) struct FunctionDebugContext<'a, 'tcx> {
219 debug_context: &'a mut DebugContext<'tcx>,
220 entry_id: UnitEntryId,
222 instance: Instance<'tcx>,
223 mir: &'tcx mir::Body<'tcx>,
226 impl<'a, 'tcx> FunctionDebugContext<'a, 'tcx> {
228 debug_context: &'a mut DebugContext<'tcx>,
229 instance: Instance<'tcx>,
233 let mir = *debug_context.tcx.instance_mir(instance.def);
235 let (symbol, _) = debug_context.symbols.insert_full(func_id, name.to_string());
237 // FIXME: add to appropriate scope intead of root
238 let scope = debug_context.dwarf.unit.root();
240 let entry_id = debug_context
243 .add(scope, gimli::DW_TAG_subprogram);
244 let entry = debug_context.dwarf.unit.get_mut(entry_id);
245 let name_id = debug_context.dwarf.strings.add(name);
247 gimli::DW_AT_linkage_name,
248 AttributeValue::StringRef(name_id),
251 FunctionDebugContext {
260 fn define_local(&mut self, name: String, ty: Ty<'tcx>) -> UnitEntryId {
261 let ty = self.debug_context.tcx.subst_and_normalize_erasing_regions(
262 self.instance.substs,
263 ty::ParamEnv::reveal_all(),
266 let dw_ty = self.debug_context.dwarf_ty(ty);
272 .add(self.entry_id, gimli::DW_TAG_variable);
273 let var_entry = self.debug_context.dwarf.unit.get_mut(var_id);
275 var_entry.set(gimli::DW_AT_name, AttributeValue::String(name.into_bytes()));
276 var_entry.set(gimli::DW_AT_type, AttributeValue::ThisUnitEntryRef(dw_ty));
281 pub(crate) fn define(
285 source_info_set: &indexmap::IndexSet<SourceInfo>,
286 local_map: FxHashMap<mir::Local, CPlace<'tcx>>,
288 let end = self.create_debug_lines(context, isa, source_info_set);
293 .push(Range::StartLength {
294 begin: Address::Symbol {
301 // FIXME make it more reliable and implement scopes before re-enabling this.
303 let value_labels_ranges = context.build_value_labels_ranges(isa).unwrap();
305 for (local, _local_decl) in self.mir.local_decls.iter_enumerated() {
306 let var_id = self.define_local(format!("{:?}", local), &self.mir.local_decls[local].ty);
308 let location = place_location(
313 &value_labels_ranges,
316 projection: ty::List::empty(),
320 let var_entry = self.debug_context.dwarf.unit.get_mut(var_id);
321 var_entry.set(gimli::DW_AT_location, location);
325 // FIXME create locals for all entries in mir.var_debug_info
329 fn place_location<'a, 'tcx>(
330 func_debug_ctx: &mut FunctionDebugContext<'a, 'tcx>,
333 local_map: &FxHashMap<mir::Local, CPlace<'tcx>>,
334 #[allow(rustc::default_hash_types)]
335 value_labels_ranges: &std::collections::HashMap<ValueLabel, Vec<ValueLocRange>>,
337 ) -> AttributeValue {
338 assert!(place.projection.is_empty()); // FIXME implement them
340 match local_map[&place.local].inner() {
341 CPlaceInner::Var(local) => {
342 let value_label = cranelift_codegen::ir::ValueLabel::from_u32(local.as_u32());
343 if let Some(value_loc_ranges) = value_labels_ranges.get(&value_label) {
344 let loc_list = LocationList(
347 .map(|value_loc_range| Location::StartEnd {
348 begin: Address::Symbol {
349 symbol: func_debug_ctx.symbol,
350 addend: i64::from(value_loc_range.start),
352 end: Address::Symbol {
353 symbol: func_debug_ctx.symbol,
354 addend: i64::from(value_loc_range.end),
357 translate_loc(isa, value_loc_range.loc, &context.func.stack_slots).unwrap(),
362 let loc_list_id = func_debug_ctx.debug_context.dwarf.unit.locations.add(loc_list);
364 AttributeValue::LocationListRef(loc_list_id)
366 // FIXME set value labels for unused locals
368 AttributeValue::Exprloc(Expression(vec![]))
371 CPlaceInner::Addr(_, _) => {
372 // FIXME implement this (used by arguments and returns)
374 AttributeValue::Exprloc(Expression(vec![]))
376 // For PointerBase::Stack:
377 //AttributeValue::Exprloc(Expression(translate_loc(ValueLoc::Stack(*stack_slot), &context.func.stack_slots).unwrap()))
382 // Adapted from https://github.com/CraneStation/wasmtime/blob/5a1845b4caf7a5dba8eda1fef05213a532ed4259/crates/debug/src/transform/expression.rs#L59-L137
383 fn translate_loc(isa: &dyn TargetIsa, loc: ValueLoc, stack_slots: &StackSlots) -> Option<Vec<u8>> {
385 ValueLoc::Reg(reg) => {
386 let machine_reg = isa.map_dwarf_register(reg).unwrap();
387 assert!(machine_reg <= 32); // FIXME
388 Some(vec![gimli::constants::DW_OP_reg0.0 + machine_reg as u8])
390 ValueLoc::Stack(ss) => {
391 if let Some(ss_offset) = stack_slots[ss].offset {
392 let endian = gimli::RunTimeEndian::Little;
393 let mut writer = write::EndianVec::new(endian);
395 .write_u8(gimli::constants::DW_OP_breg0.0 + X86_64::RBP.0 as u8)
397 writer.write_sleb128(ss_offset as i64 + 16).expect("ss wr");
398 let buf = writer.into_vec();