fix(smir): accept core drop glue names

Author: Stevengre

kmir/src/kmir/smir.py

@@ -19,6 +19,7 @@
 
 _LOGGER: Final = logging.getLogger(__name__)
 _LOG_FORMAT: Final = '%(levelname)s %(asctime)s %(name)s - %(message)s'
+_DROP_IN_PLACE_PREFIXES: Final = ('std::ptr::drop_in_place::<', 'core::ptr::drop_in_place::<')
 
 
 AdtDef = NewType('AdtDef', int)
@@ -185,7 +186,7 @@ def drop_function_tys(self) -> dict[Ty, Ty]:
             if mono_item is None:
                 continue
 
-            if not mono_item['name'].startswith('std::ptr::drop_in_place::<'):
+            if not mono_item['name'].startswith(_DROP_IN_PLACE_PREFIXES):
                 continue
 
             body = mono_item.get('body')

kmir/src/tests/unit/test_smir.py

@@ -214,3 +214,47 @@ def test_call_edges_preserve_drop_glue_for_index_projection() -> None:
     )
 
     assert smir_info.call_edges == {10: {12}, 12: set()}
+
+
+def test_drop_function_tys_accept_core_drop_in_place_names() -> None:
+    smir_info = SMIRInfo(
+        {
+            'name': 'core-drop-glue',
+            'allocs': [],
+            'types': [
+                [
+                    2,
+                    {
+                        'StructType': {
+                            'name': 'Inner',
+                            'adt_def': 2,
+                            'fields': [],
+                            'layout': None,
+                        }
+                    },
+                ],
+                [3, {'PtrType': {'pointee_type': 2}}],
+            ],
+            'functions': [
+                [12, {'NormalSym': 'drop_inner'}],
+            ],
+            'items': [
+                {
+                    'symbol_name': 'drop_inner',
+                    'mono_item_kind': {
+                        'MonoItemFn': {
+                            'name': 'core::ptr::drop_in_place::<Inner>',
+                            'body': {
+                                'arg_count': 1,
+                                'locals': [{'ty': 0}, {'ty': 3}],
+                                'blocks': [],
+                            },
+                        }
+                    },
+                },
+            ],
+            'spans': [],
+        }
+    )
+
+    assert smir_info.drop_function_tys == {2: 12}