Skip to content

Commit 9397189

Browse files
jdonszelmannjdonszelmann
committed
unsafe impl
1 parent 60b645d commit 9397189

File tree

13 files changed

+175
-53
lines changed

13 files changed

+175
-53
lines changed

‎compiler/rustc_ast/src/ast.rs

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1952,7 +1952,13 @@ pub struct MacroDef {
19521952
/// `true` if macro was defined with `macro_rules`.
19531953
pub macro_rules: bool,
19541954

1955-
pub eii_macro_for: Option<Path>,
1955+
pub eii_macro_for: Option<EiiMacroFor>,
1956+
}
1957+
1958+
#[derive(Clone, Encodable, Decodable, Debug, HashStable_Generic)]
1959+
pub struct EiiMacroFor {
1960+
pub extern_item_path: Path,
1961+
pub impl_unsafe: bool,
19561962
}
19571963

19581964
#[derive(Clone, Encodable, Decodable, Debug, Copy, Hash, Eq, PartialEq)]
@@ -3582,7 +3588,16 @@ pub struct Fn {
35823588
pub body: Option<P<Block>>,
35833589

35843590
/// This fn implements some EII, pointed to by the `path`
3585-
pub eii_impl: ThinVec<(NodeId, Path)>,
3591+
pub eii_impl: ThinVec<EIIImpl>,
3592+
}
3593+
3594+
#[derive(Clone, Encodable, Decodable, Debug)]
3595+
pub struct EIIImpl {
3596+
pub node_id: NodeId,
3597+
pub eii_macro_path: Path,
3598+
pub impl_safety: Safety,
3599+
pub span: Span,
3600+
pub inner_span: Span,
35863601
}
35873602

35883603
#[derive(Clone, Encodable, Decodable, Debug)]

‎compiler/rustc_ast/src/mut_visit.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -735,8 +735,8 @@ fn walk_mac<T: MutVisitor>(vis: &mut T, mac: &mut MacCall) {
735735

736736
fn walk_macro_def<T: MutVisitor>(vis: &mut T, macro_def: &mut MacroDef) {
737737
let MacroDef { body, macro_rules: _, eii_macro_for } = macro_def;
738-
if let Some(path) = eii_macro_for {
739-
vis.visit_path(path);
738+
if let Some(EiiMacroFor { extern_item_path, impl_unsafe: _ }) = eii_macro_for {
739+
vis.visit_path(extern_item_path);
740740
}
741741
visit_delim_args(vis, body);
742742
}
@@ -846,9 +846,9 @@ fn walk_fn<T: MutVisitor>(vis: &mut T, kind: FnKind<'_>) {
846846
visit_defaultness(vis, defaultness);
847847
vis.visit_ident(ident);
848848

849-
for (node_id, path) in eii_impl {
849+
for EIIImpl { node_id, eii_macro_path, .. } in eii_impl {
850850
vis.visit_id(node_id);
851-
vis.visit_path(path);
851+
vis.visit_path(eii_macro_path);
852852
}
853853
vis.visit_fn_header(header);
854854
vis.visit_generics(generics);

‎compiler/rustc_ast/src/visit.rs

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -479,8 +479,8 @@ impl WalkItemKind for ItemKind {
479479
ItemKind::MacroDef(ident, ts) => {
480480
try_visit!(visitor.visit_ident(ident));
481481
try_visit!(visitor.visit_mac_def(ts, id));
482-
if let Some(i) = &ts.eii_macro_for {
483-
try_visit!(visitor.visit_path(i, id));
482+
if let Some(EiiMacroFor { extern_item_path, impl_unsafe: _ }) = &ts.eii_macro_for {
483+
try_visit!(visitor.visit_path(extern_item_path, id));
484484
}
485485
}
486486
ItemKind::Delegation(box Delegation {
@@ -961,9 +961,8 @@ pub fn walk_fn<'a, V: Visitor<'a>>(visitor: &mut V, kind: FnKind<'a>) -> V::Resu
961961
// Visibility is visited as a part of the item.
962962
try_visit!(visitor.visit_ident(ident));
963963

964-
// Identifier and visibility are visited as a part of the item.
965-
for (node_id, path) in eii_impl {
966-
try_visit!(visitor.visit_path(path, *node_id));
964+
for EIIImpl { node_id, eii_macro_path, .. } in eii_impl {
965+
try_visit!(visitor.visit_path(eii_macro_path, *node_id));
967966
}
968967

969968
try_visit!(visitor.visit_fn_header(header));

‎compiler/rustc_ast_lowering/src/item.rs

Lines changed: 15 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -155,18 +155,25 @@ impl<'hir> LoweringContext<'_, 'hir> {
155155
) -> Vec<hir::Attribute> {
156156
match i {
157157
ItemKind::Fn(box Fn { eii_impl, .. }) => {
158-
let mut res = Vec::new();
159-
160-
for (id, path) in eii_impl {
161-
let did = self.lower_path_simple_eii(*id, path);
162-
res.push(hir::Attribute::Parsed(AttributeKind::EiiImpl { eii_macro: did }));
158+
let mut eii_impls = ThinVec::new();
159+
for EIIImpl { node_id, eii_macro_path, impl_safety, span, inner_span } in eii_impl {
160+
let did = self.lower_path_simple_eii(*node_id, eii_macro_path);
161+
eii_impls.push(rustc_attr_parsing::EIIImpl {
162+
eii_macro: did,
163+
span: self.lower_span(*span),
164+
inner_span: self.lower_span(*inner_span),
165+
impl_marked_unsafe: self
166+
.lower_safety(*impl_safety, hir::Safety::Safe)
167+
.is_unsafe(),
168+
})
163169
}
164170

165-
res
171+
vec![hir::Attribute::Parsed(AttributeKind::EiiImpl(eii_impls))]
166172
}
167-
ItemKind::MacroDef(_, MacroDef { eii_macro_for: Some(path), .. }) => {
173+
ItemKind::MacroDef(_, MacroDef { eii_macro_for: Some(EiiMacroFor { extern_item_path, impl_unsafe }), .. }) => {
168174
vec![hir::Attribute::Parsed(AttributeKind::EiiMacroFor {
169-
eii_extern_item: self.lower_path_simple_eii(id, path),
175+
eii_extern_item: self.lower_path_simple_eii(id, extern_item_path),
176+
impl_unsafe: *impl_unsafe,
170177
})]
171178
}
172179
ItemKind::ExternCrate(..)

‎compiler/rustc_ast_passes/src/ast_validation.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -954,8 +954,8 @@ impl<'a> Visitor<'a> for AstValidator<'a> {
954954
self.visit_attrs_vis_ident(&item.attrs, &item.vis, ident);
955955
self.check_defaultness(item.span, *defaultness);
956956

957-
for (id, path) in eii_impl {
958-
self.visit_path(path, *id);
957+
for EIIImpl { node_id, eii_macro_path, .. } in eii_impl {
958+
self.visit_path(eii_macro_path, *node_id);
959959
}
960960

961961
let is_intrinsic = item.attrs.iter().any(|a| a.has_name(sym::rustc_intrinsic));

‎compiler/rustc_ast_pretty/src/pprust/state/item.rs

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
use ast::StaticItem;
22
use itertools::{Itertools, Position};
3-
use rustc_ast as ast;
4-
use rustc_ast::ModKind;
53
use rustc_ast::ptr::P;
4+
use rustc_ast::{self as ast, EIIImpl, ModKind, Safety};
65
use rustc_span::Ident;
76

87
use crate::pp::BoxMarker;
@@ -677,9 +676,16 @@ impl<'a> State<'a> {
677676

678677
let body_cb_ib = body.as_ref().map(|body| (body, self.head("")));
679678

680-
for (_, path) in eii_impl {
679+
for EIIImpl { eii_macro_path, impl_safety, .. } in eii_impl {
681680
self.word("#[");
682-
self.print_path(path, false, 0);
681+
if let Safety::Unsafe(..) = impl_safety {
682+
self.word("unsafe");
683+
self.popen();
684+
}
685+
self.print_path(eii_macro_path, false, 0);
686+
if let Safety::Unsafe(..) = impl_safety {
687+
self.pclose();
688+
}
683689
self.word("]");
684690
self.hardbreak();
685691
}

‎compiler/rustc_attr_data_structures/src/attributes.rs

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,14 @@ impl Deprecation {
139139
}
140140
}
141141

142+
#[derive(Clone, Debug, HashStable_Generic, Encodable, Decodable, PrintAttribute)]
143+
pub struct EIIImpl {
144+
pub eii_macro: DefId,
145+
pub impl_marked_unsafe: bool,
146+
pub span: Span,
147+
pub inner_span: Span,
148+
}
149+
142150
/// Represent parsed, *built in*, inert attributes.
143151
///
144152
/// That means attributes that are not actually ever expanded.
@@ -190,12 +198,11 @@ pub enum AttributeKind {
190198
span: Span,
191199
comment: Symbol,
192200
},
193-
Eii(Span),
194-
EiiImpl {
195-
eii_macro: DefId,
196-
},
201+
EiiImpl(ThinVec<EIIImpl>),
197202
EiiMacroFor {
198203
eii_extern_item: DefId,
204+
/// whether or not it is unsafe to implement this EII
205+
impl_unsafe: bool,
199206
},
200207
MacroTransparency(Transparency),
201208
Repr(ThinVec<(ReprAttr, Span)>),

‎compiler/rustc_builtin_macros/src/eii.rs

Lines changed: 26 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
use rustc_ast::{DUMMY_NODE_ID, ItemKind, ast};
1+
use rustc_ast::{DUMMY_NODE_ID, EIIImpl, EiiMacroFor, ItemKind, ast};
22
use rustc_expand::base::{Annotatable, ExtCtxt};
3-
use rustc_span::Span;
3+
use rustc_span::{Span, kw};
44

55
pub(crate) fn eii_macro_for(
66
ecx: &mut ExtCtxt<'_>,
@@ -13,7 +13,22 @@ pub(crate) fn eii_macro_for(
1313

1414
let Some(list) = meta_item.meta_item_list() else { panic!("expected list") };
1515

16-
d.eii_macro_for = Some(list[0].meta_item().unwrap().path.clone());
16+
let Some(extern_item_path) = list.get(0).and_then(|i| i.meta_item()).map(|i| i.path.clone())
17+
else {
18+
panic!("expected a path to an `extern` item");
19+
};
20+
21+
let impl_unsafe = if let Some(i) = list.get(1) {
22+
if i.lit().and_then(|i| i.kind.str()).is_some_and(|i| i == kw::Unsafe) {
23+
true
24+
} else {
25+
panic!("expected the string `\"unsafe\"` here or no other arguments");
26+
}
27+
} else {
28+
false
29+
};
30+
31+
d.eii_macro_for = Some(EiiMacroFor { extern_item_path, impl_unsafe });
1732

1833
// Return the original item and the new methods.
1934
vec![item]
@@ -31,7 +46,14 @@ pub(crate) fn eii_macro(
3146

3247
assert!(meta_item.is_word());
3348

34-
f.eii_impl.push((DUMMY_NODE_ID, meta_item.path.clone()));
49+
f.eii_impl.push(EIIImpl {
50+
node_id: DUMMY_NODE_ID,
51+
eii_macro_path: meta_item.path.clone(),
52+
impl_safety: meta_item.unsafety,
53+
span,
54+
inner_span: meta_item.span,
55+
is_default: false,
56+
});
3557

3658
vec![item]
3759
}

‎compiler/rustc_hir_analysis/src/check/wfcheck.rs

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ use std::ops::{ControlFlow, Deref};
33

44
use hir::intravisit::{self, Visitor};
55
use rustc_abi::ExternAbi;
6-
use rustc_attr_parsing::{AttributeKind, find_attr};
6+
use rustc_attr_parsing::{AttributeKind, EIIImpl, find_attr};
77
use rustc_data_structures::fx::{FxHashSet, FxIndexMap, FxIndexSet};
88
use rustc_errors::codes::*;
99
use rustc_errors::{Applicability, ErrorGuaranteed, pluralize, struct_span_code_err};
@@ -40,7 +40,7 @@ use rustc_trait_selection::traits::{
4040
use tracing::{debug, instrument};
4141
use {rustc_ast as ast, rustc_hir as hir};
4242

43-
use super::compare_eii::compare_eii_predicate_entailment;
43+
use super::compare_eii::{compare_eii_function_types, compare_eii_predicate_entailment};
4444
use crate::autoderef::Autoderef;
4545
use crate::collect::CollectItemTypesVisitor;
4646
use crate::constrained_generic_params::{Parameter, identify_constrained_generic_params};
@@ -1324,14 +1324,16 @@ fn check_item_fn(
13241324
enter_wf_checking_ctxt(tcx, span, def_id, |wfcx| {
13251325
// does the function have an EiiImpl attribute? that contains the defid of a *macro*
13261326
// that was used to mark the implementation. This is a two step process.
1327-
if let Some(eii_macro) =
1328-
find_attr!(tcx.get_all_attrs(def_id), AttributeKind::EiiImpl {eii_macro} => *eii_macro)
1327+
for EIIImpl { eii_macro, .. } in
1328+
find_attr!(tcx.get_all_attrs(def_id), AttributeKind::EiiImpl(impls) => impls)
1329+
.into_iter()
1330+
.flatten()
13291331
{
13301332
// we expect this macro to have the `EiiMacroFor` attribute, that points to a function
13311333
// signature that we'd like to compare the function we're currently checking with
1332-
if let Some(eii_extern_item) = find_attr!(tcx.get_all_attrs(eii_macro), AttributeKind::EiiMacroFor {eii_extern_item} => *eii_extern_item)
1334+
if let Some(eii_extern_item) = find_attr!(tcx.get_all_attrs(*eii_macro), AttributeKind::EiiMacroFor {eii_extern_item, ..} => *eii_extern_item)
13331335
{
1334-
let _ = compare_eii_predicate_entailment(tcx, def_id, eii_extern_item);
1336+
let _ = compare_eii_function_types(tcx, def_id, eii_extern_item);
13351337
} else {
13361338
panic!(
13371339
"EII impl macro {eii_macro:?} did not have an eii macro for attribute pointing to a function"

‎compiler/rustc_passes/messages.ftl

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -311,6 +311,12 @@ passes_duplicate_lang_item_crate_depends =
311311
.first_definition_path = first definition in `{$orig_crate_name}` loaded from {$orig_path}
312312
.second_definition_path = second definition in `{$crate_name}` loaded from {$path}
313313
314+
passes_eii_impl_not_function =
315+
`eii_macro_for` is only valid on functions
316+
317+
passes_eii_impl_requires_unsafe =
318+
`#[{$name}]` is unsafe to implement
319+
passes_eii_impl_requires_unsafe_suggestion = wrap the attribute in `unsafe(...)`
314320
passes_export_name =
315321
attribute should be applied to a free function, impl method or static
316322
.label = not a free function, impl method or static

0 commit comments

Comments
 (0)