]> git.lizzy.rs Git - rust.git/blob - src/test/ui/structs-enums/tag-align-dyn-variants.rs
Rollup merge of #93112 - pietroalbini:pa-cve-2022-21658-nightly, r=pietroalbini
[rust.git] / src / test / ui / structs-enums / tag-align-dyn-variants.rs
1 // run-pass
2 #![allow(dead_code)]
3 #![allow(deprecated)]
4 #![allow(non_snake_case)]
5
6 use std::mem;
7
8 enum Tag<A,B> {
9     VarA(A),
10     VarB(B),
11 }
12
13 struct Rec<A,B> {
14     chA: u8,
15     tA: Tag<A,B>,
16     chB: u8,
17     tB: Tag<A,B>,
18 }
19
20 fn mk_rec<A,B>(a: A, b: B) -> Rec<A,B> {
21     Rec { chA:0, tA:Tag::VarA(a), chB:1, tB:Tag::VarB(b) }
22 }
23
24 fn is_aligned<A>(amnt: usize, u: &A) -> bool {
25     let p: usize = unsafe { mem::transmute(u) };
26     return (p & (amnt-1)) == 0;
27 }
28
29 fn variant_data_is_aligned<A,B>(amnt: usize, u: &Tag<A,B>) -> bool {
30     match u {
31       &Tag::VarA(ref a) => is_aligned(amnt, a),
32       &Tag::VarB(ref b) => is_aligned(amnt, b)
33     }
34 }
35
36 pub fn main() {
37     let u64_align = std::mem::min_align_of::<u64>();
38     let x = mk_rec(22u64, 23u64);
39     assert!(is_aligned(u64_align, &x.tA));
40     assert!(variant_data_is_aligned(u64_align, &x.tA));
41     assert!(is_aligned(u64_align, &x.tB));
42     assert!(variant_data_is_aligned(u64_align, &x.tB));
43
44     let x = mk_rec(22u64, 23u32);
45     assert!(is_aligned(u64_align, &x.tA));
46     assert!(variant_data_is_aligned(u64_align, &x.tA));
47     assert!(is_aligned(u64_align, &x.tB));
48     assert!(variant_data_is_aligned(4, &x.tB));
49
50     let x = mk_rec(22u32, 23u64);
51     assert!(is_aligned(u64_align, &x.tA));
52     assert!(variant_data_is_aligned(4, &x.tA));
53     assert!(is_aligned(u64_align, &x.tB));
54     assert!(variant_data_is_aligned(u64_align, &x.tB));
55
56     let x = mk_rec(22u32, 23u32);
57     assert!(is_aligned(4, &x.tA));
58     assert!(variant_data_is_aligned(4, &x.tA));
59     assert!(is_aligned(4, &x.tB));
60     assert!(variant_data_is_aligned(4, &x.tB));
61
62     let x = mk_rec(22f64, 23f64);
63     assert!(is_aligned(u64_align, &x.tA));
64     assert!(variant_data_is_aligned(u64_align, &x.tA));
65     assert!(is_aligned(u64_align, &x.tB));
66     assert!(variant_data_is_aligned(u64_align, &x.tB));
67 }