[clang] [clang][bytecode] Handle bitcasts involving bitfields (PR #116843)
Timm Baeder via cfe-commits
cfe-commits at lists.llvm.org
Tue Nov 19 09:19:32 PST 2024
https://github.com/tbaederr created https://github.com/llvm/llvm-project/pull/116843
Let's see what the CI has to say about this.
>From a1b1c2a040df2a4510806571a01eae4c546e9caa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Timm=20B=C3=A4der?= <tbaeder at redhat.com>
Date: Fri, 8 Nov 2024 14:37:56 +0100
Subject: [PATCH] [clang][bytecode] Handle bitcasts involving bitfields
---
clang/lib/AST/ByteCode/BitcastBuffer.h | 125 ++++++++
clang/lib/AST/ByteCode/Boolean.h | 2 +-
clang/lib/AST/ByteCode/Compiler.cpp | 2 +
clang/lib/AST/ByteCode/Integral.h | 1 +
.../lib/AST/ByteCode/InterpBuiltinBitCast.cpp | 196 ++++--------
.../ByteCode/builtin-bit-cast-bitfields.cpp | 300 ++++++++++++++++++
clang/test/AST/ByteCode/builtin-bit-cast.cpp | 87 -----
.../unittests/AST/ByteCode/BitcastBuffer.cpp | 81 +++++
clang/unittests/AST/ByteCode/CMakeLists.txt | 1 +
9 files changed, 581 insertions(+), 214 deletions(-)
create mode 100644 clang/lib/AST/ByteCode/BitcastBuffer.h
create mode 100644 clang/test/AST/ByteCode/builtin-bit-cast-bitfields.cpp
create mode 100644 clang/unittests/AST/ByteCode/BitcastBuffer.cpp
diff --git a/clang/lib/AST/ByteCode/BitcastBuffer.h b/clang/lib/AST/ByteCode/BitcastBuffer.h
new file mode 100644
index 00000000000000..1b2dcfdcd32a41
--- /dev/null
+++ b/clang/lib/AST/ByteCode/BitcastBuffer.h
@@ -0,0 +1,125 @@
+
+
+#ifndef LLVM_CLANG_AST_INTERP_BITCAST_BUFFER_H
+#define LLVM_CLANG_AST_INTERP_BITCAST_BUFFER_H
+
+#include "llvm/Support/raw_ostream.h"
+#include <bitset>
+#include <cassert>
+#include <cstring>
+#include <memory>
+#include <sstream>
+
+enum class Endian { Little, Big };
+
+static inline bool bitof(std::byte B, unsigned BitIndex) {
+ assert(BitIndex < 8);
+ return (B & (std::byte{1} << BitIndex)) != std::byte{0};
+}
+
+static inline bool fullByte(unsigned N) { return N % 8 == 0; }
+
+/// Track what bits have been initialized to known values and which ones
+/// have indeterminate value.
+/// All offsets are in bits.
+struct BitcastBuffer {
+ size_t FinalBitSize = 0;
+ std::unique_ptr<std::byte[]> Data;
+
+ BitcastBuffer(size_t FinalBitSize) : FinalBitSize(FinalBitSize) {
+ assert(fullByte(FinalBitSize));
+ unsigned ByteSize = FinalBitSize / 8;
+ Data = std::make_unique<std::byte[]>(ByteSize);
+ }
+
+ size_t size() const { return FinalBitSize; }
+
+ bool allInitialized() const {
+ // FIXME: Implement.
+ return true;
+ }
+
+ /// \p Data must be in the given endianness.
+ void pushData(const std::byte *data, size_t BitOffset, size_t BitWidth,
+ size_t FullSize, Endian DataEndianness) {
+ assert(fullByte(FullSize));
+
+ for (unsigned It = 0; It != BitWidth; ++It) {
+ bool BitValue;
+ BitValue = bitof(data[It / 8], It % 8);
+ if (!BitValue)
+ continue;
+
+ unsigned DstBit;
+ if (DataEndianness == Endian::Big)
+ DstBit = size() - BitOffset - BitWidth + It;
+ else
+ DstBit = BitOffset + It;
+ unsigned DstByte = (DstBit / 8);
+
+ Data[DstByte] |= std::byte{1} << (DstBit % 8);
+ }
+ }
+
+ std::unique_ptr<std::byte[]> copyBits(unsigned BitOffset, unsigned BitWidth,
+ unsigned FullBitWidth,
+ Endian DataEndianness) const {
+ assert(BitWidth <= FullBitWidth);
+ assert(fullByte(FullBitWidth));
+ std::unique_ptr<std::byte[]> Out =
+ std::make_unique<std::byte[]>(FullBitWidth / 8);
+
+ for (unsigned It = 0; It != BitWidth; ++It) {
+ unsigned BitIndex;
+ if (DataEndianness == Endian::Little)
+ BitIndex = BitOffset + It;
+ else
+ BitIndex = size() - BitWidth - BitOffset + It;
+
+ bool BitValue = bitof(Data[BitIndex / 8], BitIndex % 8);
+ if (!BitValue)
+ continue;
+ unsigned DstBit = It;
+ unsigned DstByte(DstBit / 8);
+ Out[DstByte] |= std::byte{1} << (DstBit % 8);
+ }
+
+ return Out;
+ }
+
+#if 0
+ template<typename T>
+ static std::string hex(T t) {
+ std::stringstream stream;
+ stream << std::hex << (int)t;
+ return std::string(stream.str());
+ }
+
+
+ void dump(bool AsHex = true) const {
+ llvm::errs() << "LSB\n ";
+ unsigned LineLength = 0;
+ for (unsigned I = 0; I != (FinalBitSize / 8); ++I) {
+ std::byte B = Data[I];
+ if (AsHex) {
+ std::stringstream stream;
+ stream << std::hex << (int)B;
+ llvm::errs() << stream.str();
+ LineLength += stream.str().size() + 1;
+ } else {
+ llvm::errs() << std::bitset<8>((int)B).to_string();
+ LineLength += 8 + 1;
+ // llvm::errs() << (int)B;
+ }
+ llvm::errs() << ' ';
+ }
+ llvm::errs() << '\n';
+
+ for (unsigned I = 0; I != LineLength; ++I)
+ llvm::errs() << ' ';
+ llvm::errs() << "MSB\n";
+ }
+#endif
+};
+
+#endif
diff --git a/clang/lib/AST/ByteCode/Boolean.h b/clang/lib/AST/ByteCode/Boolean.h
index 78d75e75c7531a..bd46523f330609 100644
--- a/clang/lib/AST/ByteCode/Boolean.h
+++ b/clang/lib/AST/ByteCode/Boolean.h
@@ -84,7 +84,7 @@ class Boolean final {
static Boolean bitcastFromMemory(const std::byte *Buff, unsigned BitWidth) {
// Boolean width is currently always 8 for all supported targets. If this
// changes we need to get the bool width from the target info.
- assert(BitWidth == 8);
+ // assert(BitWidth == 8);
bool Val = static_cast<bool>(*Buff);
return Boolean(Val);
}
diff --git a/clang/lib/AST/ByteCode/Compiler.cpp b/clang/lib/AST/ByteCode/Compiler.cpp
index 7cf2519d6a71fb..e0423dfe2091af 100644
--- a/clang/lib/AST/ByteCode/Compiler.cpp
+++ b/clang/lib/AST/ByteCode/Compiler.cpp
@@ -2710,6 +2710,8 @@ bool Compiler<Emitter>::VisitMaterializeTemporaryExpr(
return false;
if (!this->visitInitializer(SubExpr))
return false;
+ if (!this->emitFinishInit(SubExpr))
+ return false;
if (IsStatic)
return this->emitInitGlobalTempComp(TempDecl, E);
return true;
diff --git a/clang/lib/AST/ByteCode/Integral.h b/clang/lib/AST/ByteCode/Integral.h
index ca3674263aef4f..bb1688a8a7622c 100644
--- a/clang/lib/AST/ByteCode/Integral.h
+++ b/clang/lib/AST/ByteCode/Integral.h
@@ -181,6 +181,7 @@ template <unsigned Bits, bool Signed> class Integral final {
}
Integral truncate(unsigned TruncBits) const {
+ assert(TruncBits >= 1);
if (TruncBits >= Bits)
return *this;
const ReprT BitMask = (ReprT(1) << ReprT(TruncBits)) - 1;
diff --git a/clang/lib/AST/ByteCode/InterpBuiltinBitCast.cpp b/clang/lib/AST/ByteCode/InterpBuiltinBitCast.cpp
index 7e8853d3469317..b089a9b7469cf4 100644
--- a/clang/lib/AST/ByteCode/InterpBuiltinBitCast.cpp
+++ b/clang/lib/AST/ByteCode/InterpBuiltinBitCast.cpp
@@ -6,6 +6,7 @@
//
//===----------------------------------------------------------------------===//
#include "InterpBuiltinBitCast.h"
+#include "BitcastBuffer.h"
#include "Boolean.h"
#include "Context.h"
#include "Floating.h"
@@ -17,6 +18,8 @@
#include "clang/AST/ASTContext.h"
#include "clang/AST/RecordLayout.h"
#include "clang/Basic/TargetInfo.h"
+#include <bitset>
+#include <cmath>
using namespace clang;
using namespace clang::interp;
@@ -61,80 +64,11 @@ using DataFunc = llvm::function_ref<bool(const Pointer &P, PrimType Ty,
} \
} while (0)
-static bool bitof(std::byte B, unsigned BitIndex) {
- return (B & (std::byte{1} << BitIndex)) != std::byte{0};
-}
-
static void swapBytes(std::byte *M, size_t N) {
for (size_t I = 0; I != (N / 2); ++I)
std::swap(M[I], M[N - 1 - I]);
}
-/// Track what bits have been initialized to known values and which ones
-/// have indeterminate value.
-/// All offsets are in bits.
-struct BitcastBuffer {
- size_t SizeInBits = 0;
- llvm::SmallVector<std::byte> Data;
-
- BitcastBuffer() = default;
-
- size_t size() const { return SizeInBits; }
-
- const std::byte *data() const { return Data.data(); }
-
- std::byte *getBytes(unsigned BitOffset) const {
- assert(BitOffset % 8 == 0);
- assert(BitOffset < SizeInBits);
- return const_cast<std::byte *>(data() + (BitOffset / 8));
- }
-
- bool allInitialized() const {
- // FIXME: Implement.
- return true;
- }
-
- bool atByteBoundary() const { return (Data.size() * 8) == SizeInBits; }
-
- void pushBit(bool Value) {
- if (atByteBoundary())
- Data.push_back(std::byte{0});
-
- if (Value)
- Data.back() |= (std::byte{1} << (SizeInBits % 8));
- ++SizeInBits;
- }
-
- void pushData(const std::byte *data, size_t BitWidth, bool BigEndianTarget) {
- bool OnlyFullBytes = BitWidth % 8 == 0;
- unsigned NBytes = BitWidth / 8;
-
- size_t BitsHandled = 0;
- // Read all full bytes first
- for (size_t I = 0; I != NBytes; ++I) {
- std::byte B =
- BigEndianTarget ? data[NBytes - OnlyFullBytes - I] : data[I];
- for (unsigned X = 0; X != 8; ++X) {
- pushBit(bitof(B, X));
- ++BitsHandled;
- }
- }
-
- if (BitsHandled == BitWidth)
- return;
-
- // Rest of the bits.
- assert((BitWidth - BitsHandled) < 8);
- std::byte B = BigEndianTarget ? data[0] : data[NBytes];
- for (size_t I = 0, E = (BitWidth - BitsHandled); I != E; ++I) {
- pushBit(bitof(B, I));
- ++BitsHandled;
- }
-
- assert(BitsHandled == BitWidth);
- }
-};
-
/// We use this to recursively iterate over all fields and elemends of a pointer
/// and extract relevant data for a bitcast.
static bool enumerateData(const Pointer &P, const Context &Ctx, size_t Offset,
@@ -148,28 +82,28 @@ static bool enumerateData(const Pointer &P, const Context &Ctx, size_t Offset,
// Primitive arrays.
if (FieldDesc->isPrimitiveArray()) {
- bool BigEndianTarget = Ctx.getASTContext().getTargetInfo().isBigEndian();
QualType ElemType = FieldDesc->getElemQualType();
size_t ElemSizeInBits = Ctx.getASTContext().getTypeSize(ElemType);
PrimType ElemT = *Ctx.classify(ElemType);
// Special case, since the bools here are packed.
bool PackedBools = FieldDesc->getType()->isExtVectorBoolType();
+ unsigned NumElems = FieldDesc->getNumElems();
bool Ok = true;
- for (unsigned I = 0; I != FieldDesc->getNumElems(); ++I) {
- unsigned Index = BigEndianTarget ? (FieldDesc->getNumElems() - 1 - I) : I;
+ for (unsigned I = 0; I != NumElems; ++I) {
+ unsigned Index = I;
Ok = Ok && F(P.atIndex(Index), ElemT, Offset, PackedBools);
- Offset += ElemSizeInBits;
+ Offset += PackedBools ? 1 : ElemSizeInBits;
}
return Ok;
}
// Composite arrays.
if (FieldDesc->isCompositeArray()) {
- bool BigEndianTarget = Ctx.getASTContext().getTargetInfo().isBigEndian();
+ // bool BigEndianTarget = Ctx.getASTContext().getTargetInfo().isBigEndian();
QualType ElemType = FieldDesc->getElemQualType();
size_t ElemSizeInBits = Ctx.getASTContext().getTypeSize(ElemType);
for (unsigned I = 0; I != FieldDesc->getNumElems(); ++I) {
- unsigned Index = BigEndianTarget ? (FieldDesc->getNumElems() - 1 - I) : I;
+ unsigned Index = I;
enumerateData(P.atIndex(Index).narrow(), Ctx, Offset, F);
Offset += ElemSizeInBits;
}
@@ -178,7 +112,6 @@ static bool enumerateData(const Pointer &P, const Context &Ctx, size_t Offset,
// Records.
if (FieldDesc->isRecord()) {
- bool BigEndianTarget = Ctx.getASTContext().getTargetInfo().isBigEndian();
const Record *R = FieldDesc->ElemRecord;
const ASTRecordLayout &Layout =
Ctx.getASTContext().getASTRecordLayout(R->getDecl());
@@ -186,8 +119,7 @@ static bool enumerateData(const Pointer &P, const Context &Ctx, size_t Offset,
auto enumerateFields = [&]() -> void {
for (unsigned I = 0, N = R->getNumFields(); I != N; ++I) {
- const Record::Field *Fi =
- R->getField(BigEndianTarget ? (N - 1 - I) : I);
+ const Record::Field *Fi = R->getField(I);
Pointer Elem = P.atField(Fi->Offset);
size_t BitOffset =
Offset + Layout.getFieldOffset(Fi->Decl->getFieldIndex());
@@ -196,7 +128,7 @@ static bool enumerateData(const Pointer &P, const Context &Ctx, size_t Offset,
};
auto enumerateBases = [&]() -> void {
for (unsigned I = 0, N = R->getNumBases(); I != N; ++I) {
- const Record::Base *B = R->getBase(BigEndianTarget ? (N - 1 - I) : I);
+ const Record::Base *B = R->getBase(I);
Pointer Elem = P.atField(B->Offset);
CharUnits ByteOffset =
Layout.getBaseClassOffset(cast<CXXRecordDecl>(B->Decl));
@@ -204,14 +136,8 @@ static bool enumerateData(const Pointer &P, const Context &Ctx, size_t Offset,
Ok = Ok && enumerateData(Elem, Ctx, BitOffset, F);
}
};
-
- if (BigEndianTarget) {
- enumerateFields();
- enumerateBases();
- } else {
- enumerateBases();
- enumerateFields();
- }
+ enumerateBases();
+ enumerateFields();
return Ok;
}
@@ -295,26 +221,26 @@ static bool CheckBitcastType(InterpState &S, CodePtr OpPC, QualType T,
static bool readPointerToBuffer(const Context &Ctx, const Pointer &FromPtr,
BitcastBuffer &Buffer, bool ReturnOnUninit) {
const ASTContext &ASTCtx = Ctx.getASTContext();
- bool SwapData = (ASTCtx.getTargetInfo().isLittleEndian() !=
- llvm::sys::IsLittleEndianHost);
- bool BigEndianTarget = ASTCtx.getTargetInfo().isBigEndian();
+ Endian TargetEndianness =
+ ASTCtx.getTargetInfo().isLittleEndian() ? Endian::Little : Endian::Big;
return enumeratePointerFields(
FromPtr, Ctx,
[&](const Pointer &P, PrimType T, size_t BitOffset,
bool PackedBools) -> bool {
- if (!P.isInitialized()) {
- assert(false && "Implement uninitialized value tracking");
- return ReturnOnUninit;
- }
+ // if (!P.isInitialized()) {
+ // assert(false && "Implement uninitialized value tracking");
+ // return ReturnOnUninit;
+ // }
- assert(P.isInitialized());
+ // assert(P.isInitialized());
// nullptr_t is a PT_Ptr for us, but it's still not std::is_pointer_v.
if (T == PT_Ptr)
assert(false && "Implement casting to pointer types");
CharUnits ObjectReprChars = ASTCtx.getTypeSizeInChars(P.getType());
unsigned BitWidth = ASTCtx.toBits(ObjectReprChars);
+ unsigned FullSize = BitWidth;
llvm::SmallVector<std::byte> Buff(ObjectReprChars.getQuantity());
// Work around floating point types that contain unused padding bytes.
// This is really just `long double` on x86, which is the only
@@ -328,7 +254,7 @@ static bool readPointerToBuffer(const Context &Ctx, const Pointer &FromPtr,
F.bitcastToMemory(Buff.data());
// Now, only (maybe) swap the actual size of the float, excluding the
// padding bits.
- if (SwapData)
+ if (llvm::sys::IsBigEndianHost)
swapBytes(Buff.data(), NumBits / 8);
} else {
@@ -337,20 +263,15 @@ static bool readPointerToBuffer(const Context &Ctx, const Pointer &FromPtr,
else if (T == PT_Bool && PackedBools)
BitWidth = 1;
- BITCAST_TYPE_SWITCH(T, {
- T Val = P.deref<T>();
- Val.bitcastToMemory(Buff.data());
- });
- if (SwapData)
- swapBytes(Buff.data(), ObjectReprChars.getQuantity());
- }
+ BITCAST_TYPE_SWITCH(T,
+ { P.deref<T>().bitcastToMemory(Buff.data()); });
- if (BitWidth != (Buff.size() * 8) && BigEndianTarget) {
- Buffer.pushData(Buff.data() + (Buff.size() - 1 - (BitWidth / 8)),
- BitWidth, BigEndianTarget);
- } else {
- Buffer.pushData(Buff.data(), BitWidth, BigEndianTarget);
+ if (llvm::sys::IsBigEndianHost)
+ swapBytes(Buff.data(), FullSize / 8);
}
+
+ Buffer.pushData(Buff.data(), BitOffset, BitWidth, FullSize,
+ TargetEndianness);
return true;
});
}
@@ -362,7 +283,7 @@ bool clang::interp::DoBitCast(InterpState &S, CodePtr OpPC, const Pointer &Ptr,
assert(Ptr.isBlockPointer());
assert(Buff);
- BitcastBuffer Buffer;
+ BitcastBuffer Buffer(BuffSize * 8);
if (!CheckBitcastType(S, OpPC, Ptr.getType(), /*IsToType=*/false))
return false;
@@ -371,13 +292,20 @@ bool clang::interp::DoBitCast(InterpState &S, CodePtr OpPC, const Pointer &Ptr,
assert(Buffer.size() == BuffSize * 8);
HasIndeterminateBits = !Buffer.allInitialized();
- std::memcpy(Buff, Buffer.data(), BuffSize);
+
+ const ASTContext &ASTCtx = S.getASTContext();
+ Endian TargetEndianness =
+ ASTCtx.getTargetInfo().isLittleEndian() ? Endian::Little : Endian::Big;
+ auto B = Buffer.copyBits(0, BuffSize * 8, BuffSize * 8, TargetEndianness);
+
+ std::memcpy(Buff, B.get(), BuffSize);
if (llvm::sys::IsBigEndianHost)
swapBytes(Buff, BuffSize);
return Success;
}
+/// ---------------------------------------------------------------------------------------------------------------------
bool clang::interp::DoBitCastPtr(InterpState &S, CodePtr OpPC,
const Pointer &FromPtr, Pointer &ToPtr) {
@@ -394,43 +322,59 @@ bool clang::interp::DoBitCastPtr(InterpState &S, CodePtr OpPC,
if (!CheckBitcastType(S, OpPC, ToType, /*IsToType=*/true))
return false;
- BitcastBuffer Buffer;
+ const ASTContext &ASTCtx = S.getASTContext();
+
+ CharUnits ObjectReprChars = ASTCtx.getTypeSizeInChars(ToType);
+ BitcastBuffer Buffer(ASTCtx.toBits(ObjectReprChars));
readPointerToBuffer(S.getContext(), FromPtr, Buffer,
/*ReturnOnUninit=*/false);
// Now read the values out of the buffer again and into ToPtr.
- const ASTContext &ASTCtx = S.getASTContext();
- size_t BitOffset = 0;
+ Endian TargetEndianness =
+ ASTCtx.getTargetInfo().isLittleEndian() ? Endian::Little : Endian::Big;
bool Success = enumeratePointerFields(
ToPtr, S.getContext(),
- [&](const Pointer &P, PrimType T, size_t _, bool PackedBools) -> bool {
+ [&](const Pointer &P, PrimType T, size_t BitOffset,
+ bool PackedBools) -> bool {
+ CharUnits ObjectReprChars = ASTCtx.getTypeSizeInChars(P.getType());
+ unsigned FullBitWidth = ASTCtx.toBits(ObjectReprChars);
if (T == PT_Float) {
- CharUnits ObjectReprChars = ASTCtx.getTypeSizeInChars(P.getType());
const auto &Semantics = ASTCtx.getFloatTypeSemantics(P.getType());
unsigned NumBits = llvm::APFloatBase::getSizeInBits(Semantics);
assert(NumBits % 8 == 0);
assert(NumBits <= ASTCtx.toBits(ObjectReprChars));
- std::byte *M = Buffer.getBytes(BitOffset);
+ auto M = Buffer.copyBits(BitOffset, NumBits, FullBitWidth,
+ TargetEndianness);
if (llvm::sys::IsBigEndianHost)
- swapBytes(M, NumBits / 8);
+ swapBytes(M.get(), NumBits / 8);
- P.deref<Floating>() = Floating::bitcastFromMemory(M, Semantics);
+ P.deref<Floating>() = Floating::bitcastFromMemory(M.get(), Semantics);
P.initialize();
- BitOffset += ASTCtx.toBits(ObjectReprChars);
return true;
}
- BITCAST_TYPE_SWITCH_FIXED_SIZE(T, {
- std::byte *M = Buffer.getBytes(BitOffset);
+ unsigned BitWidth;
+ if (const FieldDecl *FD = P.getField(); FD && FD->isBitField())
+ BitWidth = FD->getBitWidthValue(ASTCtx);
+ else if (T == PT_Bool && PackedBools)
+ BitWidth = 1;
+ else
+ BitWidth = ASTCtx.toBits(ObjectReprChars);
- if (llvm::sys::IsBigEndianHost)
- swapBytes(M, T::bitWidth() / 8);
+ auto Memory = Buffer.copyBits(BitOffset, BitWidth, FullBitWidth,
+ TargetEndianness);
+ if (llvm::sys::IsBigEndianHost)
+ swapBytes(Memory.get(), FullBitWidth / 8);
- P.deref<T>() = T::bitcastFromMemory(M, T::bitWidth());
- P.initialize();
- BitOffset += T::bitWidth();
+ BITCAST_TYPE_SWITCH_FIXED_SIZE(T, {
+ if (BitWidth > 0)
+ P.deref<T>() = T::bitcastFromMemory(Memory.get(), T::bitWidth())
+ .truncate(BitWidth);
+ else
+ P.deref<T>() = T::zero();
});
+ P.initialize();
return true;
});
diff --git a/clang/test/AST/ByteCode/builtin-bit-cast-bitfields.cpp b/clang/test/AST/ByteCode/builtin-bit-cast-bitfields.cpp
new file mode 100644
index 00000000000000..084ec6e75a4472
--- /dev/null
+++ b/clang/test/AST/ByteCode/builtin-bit-cast-bitfields.cpp
@@ -0,0 +1,300 @@
+// RUN: %clang_cc1 -verify=expected,both -std=c++2a -fsyntax-only -fexperimental-new-constant-interpreter %s
+// RUN: %clang_cc1 -verify=expected,both -std=c++2a -fsyntax-only -triple aarch64_be-linux-gnu -fexperimental-new-constant-interpreter %s
+// RUN: %clang_cc1 -verify=expected,both -std=c++2a -fsyntax-only -fexperimental-new-constant-interpreter -triple powerpc64le-unknown-unknown -mabi=ieeelongdouble %s
+// RUN: %clang_cc1 -verify=expected,both -std=c++2a -fsyntax-only -fexperimental-new-constant-interpreter -triple powerpc64-unknown-unknown -mabi=ieeelongdouble %s
+
+#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
+# define LITTLE_END 1
+#elif __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
+# define LITTLE_END 0
+#else
+# error "huh?"
+#endif
+
+typedef decltype(nullptr) nullptr_t;
+typedef __INTPTR_TYPE__ intptr_t;
+
+static_assert(sizeof(int) == 4);
+static_assert(sizeof(long long) == 8);
+
+template <class To, class From>
+constexpr To bit_cast(const From &from) {
+ static_assert(sizeof(To) == sizeof(From));
+ return __builtin_bit_cast(To, from);
+}
+
+template <class Intermediate, class Init>
+constexpr bool check_round_trip(const Init &init) {
+ return bit_cast<Init>(bit_cast<Intermediate>(init)) == init;
+}
+
+template <class Intermediate, class Init>
+constexpr Init round_trip(const Init &init) {
+ return bit_cast<Init>(bit_cast<Intermediate>(init));
+}
+
+namespace std {
+enum byte : unsigned char {};
+} // namespace std
+
+template <int N, typename T = unsigned char, int Pad = 0>
+struct bits {
+ T : Pad;
+ T bits : N;
+
+ constexpr bool operator==(const T& rhs) const {
+ return bits == rhs;
+ }
+};
+
+template <int N, typename T, int P>
+constexpr bool operator==(const struct bits<N, T, P>& lhs, const struct bits<N, T, P>& rhs) {
+ return lhs.bits == rhs.bits;
+}
+
+template<int N>
+struct bytes {
+ using size_t = unsigned int;
+ unsigned char d[N];
+
+ constexpr unsigned char operator[](size_t index) {
+ if (index < N)
+ return d[index];
+ return -1;
+ }
+};
+
+namespace Sanity {
+ /// This is just one byte, and we extract 2 bits from it.
+ ///
+ /// 3 is 0000'0011.
+ /// For both LE and BE, the buffer will contain exactly that
+ /// byte, unaltered and not reordered in any way. It contains all 8 bits.
+ static_assert(__builtin_bit_cast(bits<2>, (unsigned char)3) == (LITTLE_END ? 3 : 0));
+
+ /// Similarly, we have one full byte of data, with the two most-significant
+ /// bits set:
+ /// 192 is 1100'0000
+ static_assert(__builtin_bit_cast(bits<2>, (unsigned char)192) == (LITTLE_END ? 0 : 3));
+
+
+ /// Here we are instead bitcasting two 1-bits into a destination of 8 bits.
+ /// On LE, we should pick the two least-significant bits. On BE, the opposite.
+ /// NOTE: Can't verify this with gcc.
+ constexpr auto B1 = bits<2>{3};
+ static_assert(__builtin_bit_cast(unsigned char, B1) == (LITTLE_END ? 3 : 192));
+
+ /// This should be 0000'0110.
+ /// On LE, this should result in 6.
+ /// On BE, 1100'0000 = 192.
+ constexpr auto B2 = bits<3>{6};
+ static_assert(__builtin_bit_cast(unsigned char, B2) == (LITTLE_END ? 6 : 192));
+
+ constexpr auto B3 = bits<4>{6};
+ static_assert(__builtin_bit_cast(unsigned char, B3) == (LITTLE_END ? 6 : 96));
+
+ struct B {
+ std::byte b0 : 4;
+ std::byte b1 : 4;
+ };
+
+ /// We can properly decompose one byte (8 bit) int two 4-bit bitfields.
+ constexpr struct { unsigned char b0; } T = {0xee};
+ constexpr B MB = __builtin_bit_cast(B, T);
+ static_assert(MB.b0 == 0xe);
+ static_assert(MB.b1 == 0xe);
+}
+
+namespace BitFields {
+ struct BitFields {
+ unsigned a : 2;
+ unsigned b : 30;
+ };
+
+ constexpr unsigned A = __builtin_bit_cast(unsigned, BitFields{3, 16});
+ static_assert(A == (LITTLE_END ? 67 : 3221225488));
+
+ struct S {
+ unsigned a : 2;
+ unsigned b : 28;
+ unsigned c : 2;
+ };
+
+ constexpr S s = __builtin_bit_cast(S, 0xFFFFFFFF);
+ static_assert(s.a == 3);
+ static_assert(s.b == 268435455);
+ static_assert(s.c == 3);
+
+ void bitfield_indeterminate() {
+ struct BF { unsigned char z : 2; };
+ enum byte : unsigned char {};
+
+ constexpr BF bf = {0x3};
+ /// Requires bitcasts to composite types.
+ static_assert(bit_cast<bits<2>>(bf).bits == bf.z);
+ static_assert(bit_cast<unsigned char>(bf));
+
+ static_assert(__builtin_bit_cast(byte, bf));
+
+ struct M {
+ // ref-note at +1 {{subobject declared here}}
+ unsigned char mem[sizeof(BF)];
+ };
+ // ref-error at +2 {{initialized by a constant expression}}
+ // ref-note at +1 {{not initialized}}
+ constexpr M m = bit_cast<M>(bf);
+
+ constexpr auto f = []() constexpr {
+ // bits<24, unsigned int, LITTLE_END ? 0 : 8> B = {0xc0ffee};
+ constexpr struct { unsigned short b1; unsigned char b0; } B = {0xc0ff, 0xee};
+ return bit_cast<bytes<4>>(B);
+ };
+
+ static_assert(f()[0] + f()[1] + f()[2] == 0xc0 + 0xff + 0xee);
+ {
+ // ref-error at +2 {{initialized by a constant expression}}
+ // ref-note at +1 {{read of uninitialized object is not allowed in a constant expression}}
+ constexpr auto _bad = f()[3];
+ }
+
+ struct B {
+ unsigned short s0 : 8;
+ unsigned short s1 : 8;
+ std::byte b0 : 4;
+ std::byte b1 : 4;
+ std::byte b2 : 4;
+ };
+ constexpr auto g = [f]() constexpr {
+ return bit_cast<B>(f());
+ };
+ static_assert(g().s0 + g().s1 + g().b0 + g().b1 == 0xc0 + 0xff + 0xe + 0xe);
+ {
+ // ref-error at +2 {{initialized by a constant expression}}
+ // ref-note at +1 {{read of uninitialized object is not allowed in a constant expression}}
+ constexpr auto _bad = g().b2;
+ }
+ }
+}
+
+typedef bool bool32 __attribute__((ext_vector_type(32)));
+struct pad {
+ unsigned short s;
+ unsigned char c;
+};
+
+constexpr auto v = bit_cast<bool32>(0xa1c0ffee);
+#if LITTLE_END
+static_assert(!v[0]);
+static_assert(v[1]);
+static_assert(v[2]);
+static_assert(v[3]);
+static_assert(!v[4]);
+static_assert(v[5]);
+static_assert(v[6]);
+static_assert(v[7]);
+
+static_assert(v[8]);
+static_assert(v[9]);
+static_assert(v[10]);
+static_assert(v[11]);
+static_assert(v[12]);
+static_assert(v[13]);
+static_assert(v[14]);
+static_assert(v[15]);
+
+static_assert(!v[16]);
+static_assert(!v[17]);
+static_assert(!v[18]);
+static_assert(!v[19]);
+static_assert(!v[20]);
+static_assert(!v[21]);
+static_assert(v[22]);
+static_assert(v[23]);
+
+static_assert(v[24]);
+static_assert(!v[25]);
+static_assert(!v[26]);
+static_assert(!v[27]);
+static_assert(!v[28]);
+static_assert(v[29]);
+static_assert(!v[30]);
+static_assert(v[31]);
+
+#else
+static_assert(v[0]);
+static_assert(!v[1]);
+static_assert(v[2]);
+static_assert(!v[3]);
+static_assert(!v[4]);
+static_assert(!v[5]);
+static_assert(!v[6]);
+static_assert(v[7]);
+
+static_assert(v[8]);
+static_assert(v[9]);
+static_assert(!v[10]);
+static_assert(!v[11]);
+static_assert(!v[12]);
+static_assert(!v[13]);
+static_assert(!v[14]);
+static_assert(!v[15]);
+
+static_assert(v[16]);
+static_assert(v[17]);
+static_assert(v[18]);
+static_assert(v[19]);
+static_assert(v[20]);
+static_assert(v[21]);
+static_assert(v[22]);
+static_assert(v[23]);
+
+static_assert(v[24]);
+static_assert(v[25]);
+static_assert(v[26]);
+static_assert(!v[27]);
+static_assert(v[28]);
+static_assert(v[29]);
+static_assert(v[30]);
+static_assert(!v[31]);
+#endif
+
+constexpr auto p = bit_cast<pad>(v);
+static_assert(p.s == (LITTLE_END ? 0xffee : 0xa1c0));
+static_assert(p.c == (LITTLE_END ? 0xc0 : 0xff));
+
+namespace TwoShorts {
+ struct B {
+ unsigned short s0 : 8;
+ unsigned short s1 : 8;
+ };
+ constexpr struct { unsigned short b1;} T = {0xc0ff};
+ constexpr B MB = __builtin_bit_cast(B, T);
+#if LITTLE_END
+ static_assert(MB.s0 == 0xff);
+ static_assert(MB.s1 == 0xc0);
+#else
+ static_assert(MB.s0 == 0xc0);
+ static_assert(MB.s1 == 0xff);
+
+#endif
+}
+
+
+typedef bool bool8 __attribute__((ext_vector_type(8)));
+typedef bool bool9 __attribute__((ext_vector_type(9)));
+typedef bool bool16 __attribute__((ext_vector_type(16)));
+typedef bool bool17 __attribute__((ext_vector_type(17)));
+typedef bool bool32 __attribute__((ext_vector_type(32)));
+typedef bool bool128 __attribute__((ext_vector_type(128)));
+
+static_assert(bit_cast<unsigned char>(bool8{1,0,1,0,1,0,1,0}) == (LITTLE_END ? 0x55 : 0xAA), "");
+constexpr bool8 b8 = __builtin_bit_cast(bool8, 0x55); // both-error {{'__builtin_bit_cast' source type 'int' does not match destination type 'bool8' (vector of 8 'bool' values) (4 vs 1 bytes)}}
+static_assert(check_round_trip<bool8>(static_cast<unsigned char>(0)), "");
+static_assert(check_round_trip<bool8>(static_cast<unsigned char>(1)), "");
+static_assert(check_round_trip<bool8>(static_cast<unsigned char>(0x55)), "");
+
+static_assert(bit_cast<unsigned short>(bool16{1,1,1,1,1,0,0,0, 1,1,1,1,0,1,0,0}) == (LITTLE_END ? 0x2F1F : 0xF8F4), "");
+
+static_assert(check_round_trip<bool16>(static_cast<short>(0xCAFE)), "");
+static_assert(check_round_trip<bool32>(static_cast<int>(0xCAFEBABE)), "");
+static_assert(check_round_trip<bool128>(static_cast<__int128_t>(0xCAFEBABE0C05FEFEULL)), "");
diff --git a/clang/test/AST/ByteCode/builtin-bit-cast.cpp b/clang/test/AST/ByteCode/builtin-bit-cast.cpp
index 60e8c3a615c5e6..b4e677de0e1446 100644
--- a/clang/test/AST/ByteCode/builtin-bit-cast.cpp
+++ b/clang/test/AST/ByteCode/builtin-bit-cast.cpp
@@ -164,72 +164,6 @@ namespace bitint {
// ref-note {{initializer of 'IB' is not a constant expression}}
}
-namespace BitFields {
- struct BitFields {
- unsigned a : 2;
- unsigned b : 30;
- };
-
- constexpr unsigned A = __builtin_bit_cast(unsigned, BitFields{3, 16}); // ref-error {{must be initialized by a constant expression}} \
- // ref-note {{not yet supported}} \
- // ref-note {{declared here}}
- static_assert(A == (LITTLE_END ? 67 : 3221225488)); // ref-error {{not an integral constant expression}} \
- // ref-note {{initializer of 'A'}}
-
-
- void bitfield_indeterminate() {
- struct BF { unsigned char z : 2; };
- enum byte : unsigned char {};
-
- constexpr BF bf = {0x3};
- /// Requires bitcasts to composite types.
- // static_assert(bit_cast<bits<2>>(bf).bits == bf.z);
- // static_assert(bit_cast<unsigned char>(bf));
-
-#if 0
- // static_assert(__builtin_bit_cast(byte, bf));
-
- struct M {
- // expected-note at +1 {{subobject declared here}}
- unsigned char mem[sizeof(BF)];
- };
- // expected-error at +2 {{initialized by a constant expression}}
- // expected-note at +1 {{not initialized}}
- constexpr M m = bit_cast<M>(bf);
-
- constexpr auto f = []() constexpr {
- // bits<24, unsigned int, LITTLE_END ? 0 : 8> B = {0xc0ffee};
- constexpr struct { unsigned short b1; unsigned char b0; } B = {0xc0ff, 0xee};
- return bit_cast<bytes<4>>(B);
- };
-
- static_assert(f()[0] + f()[1] + f()[2] == 0xc0 + 0xff + 0xee);
- {
- // expected-error at +2 {{initialized by a constant expression}}
- // expected-note at +1 {{read of uninitialized object is not allowed in a constant expression}}
- constexpr auto _bad = f()[3];
- }
-
- struct B {
- unsigned short s0 : 8;
- unsigned short s1 : 8;
- std::byte b0 : 4;
- std::byte b1 : 4;
- std::byte b2 : 4;
- };
- constexpr auto g = [f]() constexpr {
- return bit_cast<B>(f());
- };
- static_assert(g().s0 + g().s1 + g().b0 + g().b1 == 0xc0 + 0xff + 0xe + 0xe);
- {
- // expected-error at +2 {{initialized by a constant expression}}
- // expected-note at +1 {{read of uninitialized object is not allowed in a constant expression}}
- constexpr auto _bad = g().b2;
- }
-#endif
- }
-}
-
namespace Classes {
class A {
public:
@@ -488,27 +422,6 @@ static_assert(bit_cast<unsigned long long>(test_vector) == (LITTLE_END
static_assert(check_round_trip<uint2>(0xCAFEBABE0C05FEFEULL), "");
static_assert(check_round_trip<byte8>(0xCAFEBABE0C05FEFEULL), "");
-typedef bool bool8 __attribute__((ext_vector_type(8)));
-typedef bool bool9 __attribute__((ext_vector_type(9)));
-typedef bool bool16 __attribute__((ext_vector_type(16)));
-typedef bool bool17 __attribute__((ext_vector_type(17)));
-typedef bool bool32 __attribute__((ext_vector_type(32)));
-typedef bool bool128 __attribute__((ext_vector_type(128)));
-
-static_assert(bit_cast<unsigned char>(bool8{1,0,1,0,1,0,1,0}) == (LITTLE_END ? 0x55 : 0xAA), "");
-constexpr bool8 b8 = __builtin_bit_cast(bool8, 0x55); // both-error {{'__builtin_bit_cast' source type 'int' does not match destination type 'bool8' (vector of 8 'bool' values) (4 vs 1 bytes)}}
-#if 0
-static_assert(check_round_trip<bool8>(static_cast<unsigned char>(0)), "");
-static_assert(check_round_trip<bool8>(static_cast<unsigned char>(1)), "");
-static_assert(check_round_trip<bool8>(static_cast<unsigned char>(0x55)), "");
-
-static_assert(bit_cast<unsigned short>(bool16{1,1,1,1,1,0,0,0, 1,1,1,1,0,1,0,0}) == (LITTLE_END ? 0x2F1F : 0xF8F4), "");
-
-static_assert(check_round_trip<bool16>(static_cast<short>(0xCAFE)), "");
-static_assert(check_round_trip<bool32>(static_cast<int>(0xCAFEBABE)), "");
-static_assert(check_round_trip<bool128>(static_cast<__int128_t>(0xCAFEBABE0C05FEFEULL)), "");
-#endif
-
#if 0
// expected-error at +2 {{constexpr variable 'bad_bool9_to_short' must be initialized by a constant expression}}
// expected-note at +1 {{bit_cast involving type 'bool __attribute__((ext_vector_type(9)))' (vector of 9 'bool' values) is not allowed in a constant expression; element size 1 * element count 9 is not a multiple of the byte size 8}}
diff --git a/clang/unittests/AST/ByteCode/BitcastBuffer.cpp b/clang/unittests/AST/ByteCode/BitcastBuffer.cpp
new file mode 100644
index 00000000000000..04da56447542be
--- /dev/null
+++ b/clang/unittests/AST/ByteCode/BitcastBuffer.cpp
@@ -0,0 +1,81 @@
+#include "../../../lib/AST/ByteCode/BitcastBuffer.h"
+#include "clang/AST/ASTContext.h"
+#include "gtest/gtest.h"
+#include <bitset>
+#include <cassert>
+#include <cmath>
+#include <memory>
+#include <string>
+
+TEST(BitcastBuffer, PushData) {
+ BitcastBuffer Buff1(sizeof(int) * 8);
+
+ const unsigned V = 0xCAFEBABE;
+ std::byte Data[sizeof(V)];
+ std::memcpy(Data, &V, sizeof(V));
+
+ Endian HostEndianness =
+ llvm::sys::IsLittleEndianHost ? Endian::Little : Endian::Big;
+
+ Buff1.pushData(Data, 0, sizeof(V) * 8, sizeof(V) * 8, HostEndianness);
+
+ // The buffer is in host-endianness.
+ if (llvm::sys::IsLittleEndianHost) {
+ ASSERT_EQ(Buff1.Data[0], std::byte{0xbe});
+ ASSERT_EQ(Buff1.Data[1], std::byte{0xba});
+ ASSERT_EQ(Buff1.Data[2], std::byte{0xfe});
+ ASSERT_EQ(Buff1.Data[3], std::byte{0xca});
+ } else {
+ ASSERT_EQ(Buff1.Data[0], std::byte{0xca});
+ ASSERT_EQ(Buff1.Data[1], std::byte{0xfe});
+ ASSERT_EQ(Buff1.Data[2], std::byte{0xba});
+ ASSERT_EQ(Buff1.Data[3], std::byte{0xbe});
+ }
+
+ {
+ unsigned V2;
+ auto D = Buff1.copyBits(0, sizeof(V) * 8, sizeof(V) * 8, Endian::Little);
+ std::memcpy(&V2, D.get(), sizeof(V));
+ ASSERT_EQ(V, V2);
+
+ D = Buff1.copyBits(0, sizeof(V) * 8, sizeof(V) * 8, Endian::Big);
+ std::memcpy(&V2, D.get(), sizeof(V));
+ ASSERT_EQ(V, V2);
+ }
+
+ BitcastBuffer Buff2(sizeof(int) * 8);
+ {
+ short s1 = 0xCAFE;
+ short s2 = 0xBABE;
+ std::byte sdata[2];
+
+ std::memcpy(sdata, &s1, sizeof(s1));
+ Buff2.pushData(sdata, 0, sizeof(s1) * 8, sizeof(s1) * 8, HostEndianness);
+ std::memcpy(sdata, &s2, sizeof(s2));
+ Buff2.pushData(sdata, sizeof(s1) * 8, sizeof(s2) * 8, sizeof(s2) * 8,
+ HostEndianness);
+ }
+
+ if (llvm::sys::IsLittleEndianHost) {
+ ASSERT_EQ(Buff2.Data[0], std::byte{0xfe});
+ ASSERT_EQ(Buff2.Data[1], std::byte{0xca});
+ ASSERT_EQ(Buff2.Data[2], std::byte{0xbe});
+ ASSERT_EQ(Buff2.Data[3], std::byte{0xba});
+ } else {
+ ASSERT_EQ(Buff2.Data[0], std::byte{0xba});
+ ASSERT_EQ(Buff2.Data[1], std::byte{0xbe});
+ ASSERT_EQ(Buff2.Data[2], std::byte{0xca});
+ ASSERT_EQ(Buff2.Data[3], std::byte{0xfe});
+ }
+
+ {
+ unsigned V;
+ auto D = Buff2.copyBits(0, sizeof(V) * 8, sizeof(V) * 8, Endian::Little);
+ std::memcpy(&V, D.get(), sizeof(V));
+ ASSERT_EQ(V, 0xBABECAFE);
+
+ D = Buff2.copyBits(0, sizeof(V) * 8, sizeof(V) * 8, Endian::Big);
+ std::memcpy(&V, D.get(), sizeof(V));
+ ASSERT_EQ(V, 0xBABECAFE);
+ }
+}
diff --git a/clang/unittests/AST/ByteCode/CMakeLists.txt b/clang/unittests/AST/ByteCode/CMakeLists.txt
index ea727cdd4412be..b862fb4834fbdc 100644
--- a/clang/unittests/AST/ByteCode/CMakeLists.txt
+++ b/clang/unittests/AST/ByteCode/CMakeLists.txt
@@ -1,4 +1,5 @@
add_clang_unittest(InterpTests
+ BitcastBuffer.cpp
Descriptor.cpp
toAPValue.cpp
)
More information about the cfe-commits
mailing list