From 6232d952564854c7582269fdbe323ca6f6276f1b Mon Sep 17 00:00:00 2001
From: Eduard-Mihai Burtescu <eddyb@lyken.rs>
Date: Fri, 26 Nov 2021 12:51:56 +0200
Subject: [PATCH] abi: demonstrate overriding the behavior of the `layout_of`
 query. (#808)

---
 crates/rustc_codegen_spirv/src/abi.rs | 78 ++++++++++++++++++++++++++-
 1 file changed, 77 insertions(+), 1 deletion(-)

diff --git a/crates/rustc_codegen_spirv/src/abi.rs b/crates/rustc_codegen_spirv/src/abi.rs
index 49c50de53b..227d3cd1ff 100644
--- a/crates/rustc_codegen_spirv/src/abi.rs
+++ b/crates/rustc_codegen_spirv/src/abi.rs
@@ -20,7 +20,9 @@ use rustc_span::def_id::DefId;
 use rustc_span::Span;
 use rustc_span::DUMMY_SP;
 use rustc_target::abi::call::{ArgAbi, ArgAttributes, FnAbi, PassMode};
-use rustc_target::abi::{Abi, Align, FieldsShape, Primitive, Scalar, Size, VariantIdx, Variants};
+use rustc_target::abi::{
+    Abi, Align, FieldsShape, Layout, Primitive, Scalar, Size, TagEncoding, VariantIdx, Variants,
+};
 use rustc_target::spec::abi::Abi as SpecAbi;
 use std::cell::RefCell;
 use std::collections::hash_map::Entry;
@@ -89,6 +91,80 @@ pub(crate) fn provide(providers: &mut Providers) {
         let result = (rustc_interface::DEFAULT_QUERY_PROVIDERS.fn_abi_of_instance)(tcx, key);
         Ok(readjust_fn_abi(tcx, result?))
     };
+
+    // FIXME(eddyb) remove this by deriving `Clone` for `Layout` upstream.
+    fn clone_layout(layout: &Layout) -> Layout {
+        let Layout {
+            ref fields,
+            ref variants,
+            abi,
+            largest_niche,
+            align,
+            size,
+        } = *layout;
+        Layout {
+            fields: match *fields {
+                FieldsShape::Primitive => FieldsShape::Primitive,
+                FieldsShape::Union(count) => FieldsShape::Union(count),
+                FieldsShape::Array { stride, count } => FieldsShape::Array { stride, count },
+                FieldsShape::Arbitrary {
+                    ref offsets,
+                    ref memory_index,
+                } => FieldsShape::Arbitrary {
+                    offsets: offsets.clone(),
+                    memory_index: memory_index.clone(),
+                },
+            },
+            variants: match *variants {
+                Variants::Single { index } => Variants::Single { index },
+                Variants::Multiple {
+                    tag,
+                    ref tag_encoding,
+                    tag_field,
+                    ref variants,
+                } => Variants::Multiple {
+                    tag,
+                    tag_encoding: match *tag_encoding {
+                        TagEncoding::Direct => TagEncoding::Direct,
+                        TagEncoding::Niche {
+                            dataful_variant,
+                            ref niche_variants,
+                            niche_start,
+                        } => TagEncoding::Niche {
+                            dataful_variant,
+                            niche_variants: niche_variants.clone(),
+                            niche_start,
+                        },
+                    },
+                    tag_field,
+                    variants: variants.iter().map(clone_layout).collect(),
+                },
+            },
+            abi,
+            largest_niche,
+            align,
+            size,
+        }
+    }
+    providers.layout_of = |tcx, key| {
+        let TyAndLayout { ty, mut layout } =
+            (rustc_interface::DEFAULT_QUERY_PROVIDERS.layout_of)(tcx, key)?;
+
+        // FIXME(eddyb) make use of this - at this point, it's just a placeholder.
+        #[allow(clippy::match_single_binding)]
+        let hide_niche = match ty.kind() {
+            _ => false,
+        };
+
+        if hide_niche {
+            layout = tcx.arena.alloc(Layout {
+                largest_niche: None,
+                ..clone_layout(layout)
+            });
+        }
+
+        Ok(TyAndLayout { ty, layout })
+    };
 }
 
 pub(crate) fn provide_extern(providers: &mut Providers) {