src/tools/miri/tests/pass/enums.rs

   1 enum MyEnum {
   2     MyEmptyVariant,
   3     MyNewtypeVariant(i32),
   4     MyTupleVariant(i32, i32),
   5     MyStructVariant { my_first_field: i32, my_second_field: i32 },
   6 }
   7
   8 fn test(me: MyEnum) {
   9     match me {
  10         MyEnum::MyEmptyVariant => {}
  11         MyEnum::MyNewtypeVariant(ref val) => assert_eq!(val, &42),
  12         MyEnum::MyTupleVariant(ref a, ref b) => {
  13             assert_eq!(a, &43);
  14             assert_eq!(b, &44);
  15         }
  16         MyEnum::MyStructVariant { ref my_first_field, ref my_second_field } => {
  17             assert_eq!(my_first_field, &45);
  18             assert_eq!(my_second_field, &46);
  19         }
  20     }
  21 }
  22
  23 fn discriminant_overflow() {
  24     // Tests for https://github.com/rust-lang/rust/issues/62138.
  25     #[repr(u8)]
  26     #[allow(dead_code)]
  27     enum WithWraparoundInvalidValues {
  28         X = 1,
  29         Y = 254,
  30     }
  31
  32     #[allow(dead_code)]
  33     enum Foo {
  34         A,
  35         B,
  36         C(WithWraparoundInvalidValues),
  37     }
  38
  39     let x = Foo::B;
  40     match x {
  41         Foo::B => {}
  42         _ => panic!(),
  43     }
  44 }
  45
  46 fn more_discriminant_overflow() {
  47     pub enum Infallible {}
  48
  49     // The check that the `bool` field of `V1` is encoding a "niche variant"
  50     // (i.e. not `V1`, so `V3` or `V4`) used to be mathematically incorrect,
  51     // causing valid `V1` values to be interpreted as other variants.
  52     #[allow(dead_code)]
  53     pub enum E1 {
  54         V1 { f: bool },
  55         V2 { f: Infallible },
  56         V3,
  57         V4,
  58     }
  59
  60     // Computing the discriminant used to be done using the niche type (here `u8`,
  61     // from the `bool` field of `V1`), overflowing for variants with large enough
  62     // indices (`V3` and `V4`), causing them to be interpreted as other variants.
  63     #[allow(dead_code)]
  64     #[rustfmt::skip] // rustfmt prefers every variant on its own line
  65     pub enum E2<X> {
  66         V1 { f: bool },
  67
  68         /*_00*/ _01(X), _02(X), _03(X), _04(X), _05(X), _06(X), _07(X),
  69         _08(X), _09(X), _0A(X), _0B(X), _0C(X), _0D(X), _0E(X), _0F(X),
  70         _10(X), _11(X), _12(X), _13(X), _14(X), _15(X), _16(X), _17(X),
  71         _18(X), _19(X), _1A(X), _1B(X), _1C(X), _1D(X), _1E(X), _1F(X),
  72         _20(X), _21(X), _22(X), _23(X), _24(X), _25(X), _26(X), _27(X),
  73         _28(X), _29(X), _2A(X), _2B(X), _2C(X), _2D(X), _2E(X), _2F(X),
  74         _30(X), _31(X), _32(X), _33(X), _34(X), _35(X), _36(X), _37(X),
  75         _38(X), _39(X), _3A(X), _3B(X), _3C(X), _3D(X), _3E(X), _3F(X),
  76         _40(X), _41(X), _42(X), _43(X), _44(X), _45(X), _46(X), _47(X),
  77         _48(X), _49(X), _4A(X), _4B(X), _4C(X), _4D(X), _4E(X), _4F(X),
  78         _50(X), _51(X), _52(X), _53(X), _54(X), _55(X), _56(X), _57(X),
  79         _58(X), _59(X), _5A(X), _5B(X), _5C(X), _5D(X), _5E(X), _5F(X),
  80         _60(X), _61(X), _62(X), _63(X), _64(X), _65(X), _66(X), _67(X),
  81         _68(X), _69(X), _6A(X), _6B(X), _6C(X), _6D(X), _6E(X), _6F(X),
  82         _70(X), _71(X), _72(X), _73(X), _74(X), _75(X), _76(X), _77(X),
  83         _78(X), _79(X), _7A(X), _7B(X), _7C(X), _7D(X), _7E(X), _7F(X),
  84         _80(X), _81(X), _82(X), _83(X), _84(X), _85(X), _86(X), _87(X),
  85         _88(X), _89(X), _8A(X), _8B(X), _8C(X), _8D(X), _8E(X), _8F(X),
  86         _90(X), _91(X), _92(X), _93(X), _94(X), _95(X), _96(X), _97(X),
  87         _98(X), _99(X), _9A(X), _9B(X), _9C(X), _9D(X), _9E(X), _9F(X),
  88         _A0(X), _A1(X), _A2(X), _A3(X), _A4(X), _A5(X), _A6(X), _A7(X),
  89         _A8(X), _A9(X), _AA(X), _AB(X), _AC(X), _AD(X), _AE(X), _AF(X),
  90         _B0(X), _B1(X), _B2(X), _B3(X), _B4(X), _B5(X), _B6(X), _B7(X),
  91         _B8(X), _B9(X), _BA(X), _BB(X), _BC(X), _BD(X), _BE(X), _BF(X),
  92         _C0(X), _C1(X), _C2(X), _C3(X), _C4(X), _C5(X), _C6(X), _C7(X),
  93         _C8(X), _C9(X), _CA(X), _CB(X), _CC(X), _CD(X), _CE(X), _CF(X),
  94         _D0(X), _D1(X), _D2(X), _D3(X), _D4(X), _D5(X), _D6(X), _D7(X),
  95         _D8(X), _D9(X), _DA(X), _DB(X), _DC(X), _DD(X), _DE(X), _DF(X),
  96         _E0(X), _E1(X), _E2(X), _E3(X), _E4(X), _E5(X), _E6(X), _E7(X),
  97         _E8(X), _E9(X), _EA(X), _EB(X), _EC(X), _ED(X), _EE(X), _EF(X),
  98         _F0(X), _F1(X), _F2(X), _F3(X), _F4(X), _F5(X), _F6(X), _F7(X),
  99         _F8(X), _F9(X), _FA(X), _FB(X), _FC(X), _FD(X), _FE(X), _FF(X),
 100
 101         V3,
 102         V4,
 103     }
 104
 105     if let E1::V2 { .. } = (E1::V1 { f: true }) {
 106         unreachable!()
 107     }
 108     if let E1::V1 { .. } = (E1::V1 { f: true }) {
 109     } else {
 110         unreachable!()
 111     }
 112
 113     if let E2::V1 { .. } = E2::V3::<Infallible> {
 114         unreachable!()
 115     }
 116     if let E2::V3 { .. } = E2::V3::<Infallible> {
 117     } else {
 118         unreachable!()
 119     }
 120 }
 121
 122 fn overaligned_casts() {
 123     #[allow(dead_code)]
 124     #[repr(align(8))]
 125     enum Aligned {
 126         Zero = 0,
 127         One = 1,
 128     }
 129
 130     let aligned = Aligned::Zero;
 131     assert_eq!(aligned as u8, 0);
 132 }
 133
 134 fn main() {
 135     test(MyEnum::MyEmptyVariant);
 136     test(MyEnum::MyNewtypeVariant(42));
 137     test(MyEnum::MyTupleVariant(43, 44));
 138     test(MyEnum::MyStructVariant { my_first_field: 45, my_second_field: 46 });
 139
 140     discriminant_overflow();
 141     more_discriminant_overflow();
 142     overaligned_casts();
 143 }