1 //===-------------------- InterpBuiltinBitCast.cpp --------------*- C++ -*-===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 #include "InterpBuiltinBitCast.h" 9 #include "BitcastBuffer.h" 10 #include "Boolean.h" 11 #include "Context.h" 12 #include "Floating.h" 13 #include "Integral.h" 14 #include "InterpState.h" 15 #include "MemberPointer.h" 16 #include "Pointer.h" 17 #include "Record.h" 18 #include "clang/AST/ASTContext.h" 19 #include "clang/AST/RecordLayout.h" 20 #include "clang/Basic/TargetInfo.h" 21 22 using namespace clang; 23 using namespace clang::interp; 24 25 /// Implement __builtin_bit_cast and related operations. 26 /// Since our internal representation for data is more complex than 27 /// something we can simply memcpy or memcmp, we first bitcast all the data 28 /// into a buffer, which we then later use to copy the data into the target. 29 30 // TODO: 31 // - Try to minimize heap allocations. 32 // - Optimize the common case of only pushing and pulling full 33 // bytes to/from the buffer. 34 35 /// Used to iterate over pointer fields. 36 using DataFunc = llvm::function_ref<bool(const Pointer &P, PrimType Ty, 37 Bits BitOffset, bool PackedBools)>; 38 39 #define BITCAST_TYPE_SWITCH(Expr, B) \ 40 do { \ 41 switch (Expr) { \ 42 TYPE_SWITCH_CASE(PT_Sint8, B) \ 43 TYPE_SWITCH_CASE(PT_Uint8, B) \ 44 TYPE_SWITCH_CASE(PT_Sint16, B) \ 45 TYPE_SWITCH_CASE(PT_Uint16, B) \ 46 TYPE_SWITCH_CASE(PT_Sint32, B) \ 47 TYPE_SWITCH_CASE(PT_Uint32, B) \ 48 TYPE_SWITCH_CASE(PT_Sint64, B) \ 49 TYPE_SWITCH_CASE(PT_Uint64, B) \ 50 TYPE_SWITCH_CASE(PT_IntAP, B) \ 51 TYPE_SWITCH_CASE(PT_IntAPS, B) \ 52 TYPE_SWITCH_CASE(PT_Bool, B) \ 53 default: \ 54 llvm_unreachable("Unhandled bitcast type"); \ 55 } \ 56 } while (0) 57 58 #define BITCAST_TYPE_SWITCH_FIXED_SIZE(Expr, B) \ 59 do { \ 60 switch (Expr) { \ 61 TYPE_SWITCH_CASE(PT_Sint8, B) \ 62 TYPE_SWITCH_CASE(PT_Uint8, B) \ 63 TYPE_SWITCH_CASE(PT_Sint16, B) \ 64 TYPE_SWITCH_CASE(PT_Uint16, B) \ 65 TYPE_SWITCH_CASE(PT_Sint32, B) \ 66 TYPE_SWITCH_CASE(PT_Uint32, B) \ 67 TYPE_SWITCH_CASE(PT_Sint64, B) \ 68 TYPE_SWITCH_CASE(PT_Uint64, B) \ 69 TYPE_SWITCH_CASE(PT_Bool, B) \ 70 default: \ 71 llvm_unreachable("Unhandled bitcast type"); \ 72 } \ 73 } while (0) 74 75 static void swapBytes(std::byte *M, size_t N) { 76 for (size_t I = 0; I != (N / 2); ++I) 77 std::swap(M[I], M[N - 1 - I]); 78 } 79 80 /// We use this to recursively iterate over all fields and elements of a pointer 81 /// and extract relevant data for a bitcast. 82 static bool enumerateData(const Pointer &P, const Context &Ctx, Bits Offset, 83 Bits BitsToRead, DataFunc F) { 84 const Descriptor *FieldDesc = P.getFieldDesc(); 85 assert(FieldDesc); 86 87 // Primitives. 88 if (FieldDesc->isPrimitive()) 89 return F(P, FieldDesc->getPrimType(), Offset, /*PackedBools=*/false); 90 91 // Primitive arrays. 92 if (FieldDesc->isPrimitiveArray()) { 93 QualType ElemType = FieldDesc->getElemQualType(); 94 size_t ElemSizeInBits = Ctx.getASTContext().getTypeSize(ElemType); 95 PrimType ElemT = *Ctx.classify(ElemType); 96 // Special case, since the bools here are packed. 97 bool PackedBools = FieldDesc->getType()->isExtVectorBoolType(); 98 unsigned NumElems = FieldDesc->getNumElems(); 99 bool Ok = true; 100 for (unsigned I = P.getIndex(); I != NumElems; ++I) { 101 Ok = Ok && F(P.atIndex(I), ElemT, Offset, PackedBools); 102 Offset += PackedBools ? 1 : ElemSizeInBits; 103 if (Offset >= BitsToRead) 104 break; 105 } 106 return Ok; 107 } 108 109 // Composite arrays. 110 if (FieldDesc->isCompositeArray()) { 111 QualType ElemType = FieldDesc->getElemQualType(); 112 size_t ElemSizeInBits = Ctx.getASTContext().getTypeSize(ElemType); 113 for (unsigned I = 0; I != FieldDesc->getNumElems(); ++I) { 114 enumerateData(P.atIndex(I).narrow(), Ctx, Offset, BitsToRead, F); 115 Offset += ElemSizeInBits; 116 if (Offset >= BitsToRead) 117 break; 118 } 119 return true; 120 } 121 122 // Records. 123 if (FieldDesc->isRecord()) { 124 const Record *R = FieldDesc->ElemRecord; 125 const ASTRecordLayout &Layout = 126 Ctx.getASTContext().getASTRecordLayout(R->getDecl()); 127 bool Ok = true; 128 129 for (const Record::Field &Fi : R->fields()) { 130 if (Fi.isUnnamedBitField()) 131 continue; 132 Pointer Elem = P.atField(Fi.Offset); 133 Bits BitOffset = 134 Offset + Bits(Layout.getFieldOffset(Fi.Decl->getFieldIndex())); 135 Ok = Ok && enumerateData(Elem, Ctx, BitOffset, BitsToRead, F); 136 } 137 for (const Record::Base &B : R->bases()) { 138 Pointer Elem = P.atField(B.Offset); 139 CharUnits ByteOffset = 140 Layout.getBaseClassOffset(cast<CXXRecordDecl>(B.Decl)); 141 Bits BitOffset = Offset + Bits(Ctx.getASTContext().toBits(ByteOffset)); 142 Ok = Ok && enumerateData(Elem, Ctx, BitOffset, BitsToRead, F); 143 // FIXME: We should only (need to) do this when bitcasting OUT of the 144 // buffer, not when copying data into it. 145 if (Ok) 146 Elem.initialize(); 147 } 148 149 return Ok; 150 } 151 152 llvm_unreachable("Unhandled data type"); 153 } 154 155 static bool enumeratePointerFields(const Pointer &P, const Context &Ctx, 156 Bits BitsToRead, DataFunc F) { 157 return enumerateData(P, Ctx, Bits::zero(), BitsToRead, F); 158 } 159 160 // This function is constexpr if and only if To, From, and the types of 161 // all subobjects of To and From are types T such that... 162 // (3.1) - is_union_v<T> is false; 163 // (3.2) - is_pointer_v<T> is false; 164 // (3.3) - is_member_pointer_v<T> is false; 165 // (3.4) - is_volatile_v<T> is false; and 166 // (3.5) - T has no non-static data members of reference type 167 // 168 // NOTE: This is a version of checkBitCastConstexprEligibilityType() in 169 // ExprConstant.cpp. 170 static bool CheckBitcastType(InterpState &S, CodePtr OpPC, QualType T, 171 bool IsToType) { 172 enum { 173 E_Union = 0, 174 E_Pointer, 175 E_MemberPointer, 176 E_Volatile, 177 E_Reference, 178 }; 179 enum { C_Member, C_Base }; 180 181 auto diag = [&](int Reason) -> bool { 182 const Expr *E = S.Current->getExpr(OpPC); 183 S.FFDiag(E, diag::note_constexpr_bit_cast_invalid_type) 184 << static_cast<int>(IsToType) << (Reason == E_Reference) << Reason 185 << E->getSourceRange(); 186 return false; 187 }; 188 auto note = [&](int Construct, QualType NoteType, SourceRange NoteRange) { 189 S.Note(NoteRange.getBegin(), diag::note_constexpr_bit_cast_invalid_subtype) 190 << NoteType << Construct << T.getUnqualifiedType() << NoteRange; 191 return false; 192 }; 193 194 T = T.getCanonicalType(); 195 196 if (T->isUnionType()) 197 return diag(E_Union); 198 if (T->isPointerType()) 199 return diag(E_Pointer); 200 if (T->isMemberPointerType()) 201 return diag(E_MemberPointer); 202 if (T.isVolatileQualified()) 203 return diag(E_Volatile); 204 205 if (const RecordDecl *RD = T->getAsRecordDecl()) { 206 if (const auto *CXXRD = dyn_cast<CXXRecordDecl>(RD)) { 207 for (const CXXBaseSpecifier &BS : CXXRD->bases()) { 208 if (!CheckBitcastType(S, OpPC, BS.getType(), IsToType)) 209 return note(C_Base, BS.getType(), BS.getBeginLoc()); 210 } 211 } 212 for (const FieldDecl *FD : RD->fields()) { 213 if (FD->getType()->isReferenceType()) 214 return diag(E_Reference); 215 if (!CheckBitcastType(S, OpPC, FD->getType(), IsToType)) 216 return note(C_Member, FD->getType(), FD->getSourceRange()); 217 } 218 } 219 220 if (T->isArrayType() && 221 !CheckBitcastType(S, OpPC, S.getASTContext().getBaseElementType(T), 222 IsToType)) 223 return false; 224 225 return true; 226 } 227 228 static bool readPointerToBuffer(const Context &Ctx, const Pointer &FromPtr, 229 BitcastBuffer &Buffer, bool ReturnOnUninit) { 230 const ASTContext &ASTCtx = Ctx.getASTContext(); 231 Endian TargetEndianness = 232 ASTCtx.getTargetInfo().isLittleEndian() ? Endian::Little : Endian::Big; 233 234 return enumeratePointerFields( 235 FromPtr, Ctx, Buffer.size(), 236 [&](const Pointer &P, PrimType T, Bits BitOffset, 237 bool PackedBools) -> bool { 238 CharUnits ObjectReprChars = ASTCtx.getTypeSizeInChars(P.getType()); 239 Bits BitWidth = Bits(ASTCtx.toBits(ObjectReprChars)); 240 Bits FullBitWidth = BitWidth; 241 242 if (const FieldDecl *FD = P.getField(); FD && FD->isBitField()) { 243 BitWidth = Bits(std::min(FD->getBitWidthValue(ASTCtx), 244 (unsigned)FullBitWidth.getQuantity())); 245 } else if (T == PT_Bool && PackedBools) 246 BitWidth = Bits(1); 247 248 if (BitWidth.isZero()) 249 return true; 250 251 if (!P.isInitialized()) { 252 assert(false && "Implement uninitialized value tracking"); 253 return ReturnOnUninit; 254 } 255 256 assert(P.isInitialized()); 257 // nullptr_t is a PT_Ptr for us, but it's still not std::is_pointer_v. 258 if (T == PT_Ptr) 259 assert(false && "Implement casting to pointer types"); 260 261 auto Buff = 262 std::make_unique<std::byte[]>(ObjectReprChars.getQuantity()); 263 // Work around floating point types that contain unused padding bytes. 264 // This is really just `long double` on x86, which is the only 265 // fundamental type with padding bytes. 266 if (T == PT_Float) { 267 const Floating &F = P.deref<Floating>(); 268 Bits NumBits = Bits( 269 llvm::APFloatBase::getSizeInBits(F.getAPFloat().getSemantics())); 270 assert(NumBits.isFullByte()); 271 assert(NumBits.getQuantity() <= FullBitWidth.getQuantity()); 272 F.bitcastToMemory(Buff.get()); 273 // Now, only (maybe) swap the actual size of the float, excluding the 274 // padding bits. 275 if (llvm::sys::IsBigEndianHost) 276 swapBytes(Buff.get(), NumBits.roundToBytes()); 277 278 } else { 279 BITCAST_TYPE_SWITCH(T, { P.deref<T>().bitcastToMemory(Buff.get()); }); 280 281 if (llvm::sys::IsBigEndianHost) 282 swapBytes(Buff.get(), FullBitWidth.roundToBytes()); 283 } 284 285 Buffer.pushData(Buff.get(), BitOffset, BitWidth, TargetEndianness); 286 return true; 287 }); 288 } 289 290 bool clang::interp::DoBitCast(InterpState &S, CodePtr OpPC, const Pointer &Ptr, 291 std::byte *Buff, size_t BuffSize, 292 bool &HasIndeterminateBits) { 293 assert(Ptr.isLive()); 294 assert(Ptr.isBlockPointer()); 295 assert(Buff); 296 297 Bits BitSize = Bytes(BuffSize).toBits(); 298 BitcastBuffer Buffer(BitSize); 299 if (!CheckBitcastType(S, OpPC, Ptr.getType(), /*IsToType=*/false)) 300 return false; 301 302 bool Success = readPointerToBuffer(S.getContext(), Ptr, Buffer, 303 /*ReturnOnUninit=*/false); 304 HasIndeterminateBits = !Buffer.allInitialized(); 305 306 const ASTContext &ASTCtx = S.getASTContext(); 307 Endian TargetEndianness = 308 ASTCtx.getTargetInfo().isLittleEndian() ? Endian::Little : Endian::Big; 309 auto B = Buffer.copyBits(Bits::zero(), BitSize, BitSize, TargetEndianness); 310 311 std::memcpy(Buff, B.get(), BuffSize); 312 313 if (llvm::sys::IsBigEndianHost) 314 swapBytes(Buff, BuffSize); 315 316 return Success; 317 } 318 bool clang::interp::DoBitCastPtr(InterpState &S, CodePtr OpPC, 319 const Pointer &FromPtr, Pointer &ToPtr) { 320 const ASTContext &ASTCtx = S.getASTContext(); 321 CharUnits ObjectReprChars = ASTCtx.getTypeSizeInChars(ToPtr.getType()); 322 323 return DoBitCastPtr(S, OpPC, FromPtr, ToPtr, ObjectReprChars.getQuantity()); 324 } 325 326 bool clang::interp::DoBitCastPtr(InterpState &S, CodePtr OpPC, 327 const Pointer &FromPtr, Pointer &ToPtr, 328 size_t Size) { 329 assert(FromPtr.isLive()); 330 assert(FromPtr.isBlockPointer()); 331 assert(ToPtr.isBlockPointer()); 332 333 QualType FromType = FromPtr.getType(); 334 QualType ToType = ToPtr.getType(); 335 336 if (!CheckBitcastType(S, OpPC, ToType, /*IsToType=*/true)) 337 return false; 338 if (!CheckBitcastType(S, OpPC, FromType, /*IsToType=*/false)) 339 return false; 340 341 const ASTContext &ASTCtx = S.getASTContext(); 342 BitcastBuffer Buffer(Bytes(Size).toBits()); 343 readPointerToBuffer(S.getContext(), FromPtr, Buffer, 344 /*ReturnOnUninit=*/false); 345 346 // Now read the values out of the buffer again and into ToPtr. 347 Endian TargetEndianness = 348 ASTCtx.getTargetInfo().isLittleEndian() ? Endian::Little : Endian::Big; 349 bool Success = enumeratePointerFields( 350 ToPtr, S.getContext(), Buffer.size(), 351 [&](const Pointer &P, PrimType T, Bits BitOffset, 352 bool PackedBools) -> bool { 353 CharUnits ObjectReprChars = ASTCtx.getTypeSizeInChars(P.getType()); 354 Bits FullBitWidth = Bits(ASTCtx.toBits(ObjectReprChars)); 355 if (T == PT_Float) { 356 const auto &Semantics = ASTCtx.getFloatTypeSemantics(P.getType()); 357 Bits NumBits = Bits(llvm::APFloatBase::getSizeInBits(Semantics)); 358 assert(NumBits.isFullByte()); 359 assert(NumBits.getQuantity() <= FullBitWidth.getQuantity()); 360 auto M = Buffer.copyBits(BitOffset, NumBits, FullBitWidth, 361 TargetEndianness); 362 363 if (llvm::sys::IsBigEndianHost) 364 swapBytes(M.get(), NumBits.roundToBytes()); 365 366 P.deref<Floating>() = Floating::bitcastFromMemory(M.get(), Semantics); 367 P.initialize(); 368 return true; 369 } 370 371 Bits BitWidth; 372 if (const FieldDecl *FD = P.getField(); FD && FD->isBitField()) 373 BitWidth = Bits(std::min(FD->getBitWidthValue(ASTCtx), 374 (unsigned)FullBitWidth.getQuantity())); 375 else if (T == PT_Bool && PackedBools) 376 BitWidth = Bits(1); 377 else 378 BitWidth = FullBitWidth; 379 380 auto Memory = Buffer.copyBits(BitOffset, BitWidth, FullBitWidth, 381 TargetEndianness); 382 if (llvm::sys::IsBigEndianHost) 383 swapBytes(Memory.get(), FullBitWidth.roundToBytes()); 384 385 BITCAST_TYPE_SWITCH_FIXED_SIZE(T, { 386 if (BitWidth.nonZero()) 387 P.deref<T>() = T::bitcastFromMemory(Memory.get(), T::bitWidth()) 388 .truncate(BitWidth.getQuantity()); 389 else 390 P.deref<T>() = T::zero(); 391 }); 392 P.initialize(); 393 return true; 394 }); 395 396 return Success; 397 } 398