averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

[Runtime] Refactor swift_generic_initWithTake to use shared abstraction (#66379)

Browse Source 
We introduced shared abstractions for handling the different tags for the other runtime functions, so we should use them here as well.
This commit is contained in:
Dario Rexin
2023-06-06 14:27:56 -07:00
committed by GitHub
parent 4325c0ae0b
commit 1425e32c86
1 changed files with 44 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

112
stdlib/public/runtime/BytecodeLayouts.cpp
View File

@@ -384,10 +384,52 @@ swift_generic_initWithCopy(swift::OpaqueValue *dest, swift::OpaqueValue *src,
return dest;
}
struct TakeHandler {
static inline void handleMetatype(const Metadata *type, uintptr_t addrOffset,
uint8_t *dest, uint8_t *src) {
if (SWIFT_UNLIKELY(!type->getValueWitnesses()->isBitwiseTakable())) {
type->vw_initializeWithTake(
(OpaqueValue*)((uintptr_t)dest + addrOffset),
(OpaqueValue*)((uintptr_t)src + addrOffset));
}
}
static inline void handleSinglePayloadEnumSimple(const uint8_t *typeLayout,
size_t &offset,
uintptr_t &addrOffset,
uint8_t *dest,
uint8_t *src) {
::handleSinglePayloadEnumSimple(typeLayout, offset, src, addrOffset);
}
static inline void handleSinglePayloadEnumFN(const uint8_t *typeLayout,
size_t &offset, bool resolved,
uintptr_t &addrOffset,
uint8_t *dest, uint8_t *src) {
::handleSinglePayloadEnumFN(typeLayout, offset, resolved, src, addrOffset);
}
static inline void handleReference(RefCountingKind tag, uintptr_t addrOffset,
uint8_t *dest, uint8_t *src) {
if (tag == RefCountingKind::UnknownWeak) {
swift_unknownObjectWeakTakeInit(
(WeakReference*)((uintptr_t)dest + addrOffset),
(WeakReference*)((uintptr_t)src + addrOffset));
} else if (tag == RefCountingKind::Existential) {
auto *type = getExistentialTypeMetadata(
(OpaqueValue*)((uintptr_t)src + addrOffset));
if (SWIFT_UNLIKELY(!type->getValueWitnesses()->isBitwiseTakable())) {
type->vw_initializeWithTake(
(OpaqueValue*)((uintptr_t)dest + addrOffset),
(OpaqueValue*)((uintptr_t)src + addrOffset));
}
}
}
};
extern "C" swift::OpaqueValue *
swift_generic_initWithTake(swift::OpaqueValue *dest, swift::OpaqueValue *src,
const Metadata *metadata) {
const uint8_t *typeLayout = metadata->getLayoutString();
size_t size = metadata->vw_size();
memcpy(dest, src, size);
@@ -396,73 +438,7 @@ swift_generic_initWithTake(swift::OpaqueValue *dest, swift::OpaqueValue *src,
return dest;
}
auto offset = layoutStringHeaderSize;
uintptr_t addrOffset = 0;
while (true) {
uint64_t skip = readBytes<uint64_t>(typeLayout, offset);
auto tag = static_cast<RefCountingKind>(skip >> 56);
skip &= ~(0xffULL << 56);
addrOffset += skip;
switch (tag) {
case RefCountingKind::UnknownWeak:
swift_unknownObjectWeakTakeInit(
(WeakReference*)((uintptr_t)dest + addrOffset),
(WeakReference*)((uintptr_t)src + addrOffset));
break;
case RefCountingKind::Metatype: {
auto *type = readBytes<const Metadata*>(typeLayout, offset);
if (SWIFT_UNLIKELY(!type->getValueWitnesses()->isBitwiseTakable())) {
type->vw_initializeWithTake(
(OpaqueValue*)((uintptr_t)dest + addrOffset),
(OpaqueValue*)((uintptr_t)src + addrOffset));
}
break;
}
case RefCountingKind::Existential: {
auto *type = getExistentialTypeMetadata(
(OpaqueValue*)((uintptr_t)src + addrOffset));
if (SWIFT_UNLIKELY(!type->getValueWitnesses()->isBitwiseTakable())) {
type->vw_initializeWithTake(
(OpaqueValue*)((uintptr_t)dest + addrOffset),
(OpaqueValue*)((uintptr_t)src + addrOffset));
}
break;
}
case RefCountingKind::Resilient: {
auto *type = getResilientTypeMetadata(metadata, typeLayout, offset);
if (SWIFT_UNLIKELY(!type->getValueWitnesses()->isBitwiseTakable())) {
type->vw_initializeWithTake((OpaqueValue*)((uintptr_t)dest + addrOffset),
(OpaqueValue*)((uintptr_t)src + addrOffset));
}
break;
}
case RefCountingKind::SinglePayloadEnumSimple: {
handleSinglePayloadEnumSimple(typeLayout, offset, (uint8_t *)src,
addrOffset);
break;
}
case RefCountingKind::SinglePayloadEnumFN: {
handleSinglePayloadEnumFN(typeLayout, offset, false, (uint8_t *)src,
addrOffset);
break;
}
case RefCountingKind::SinglePayloadEnumFNResolved: {
handleSinglePayloadEnumFN(typeLayout, offset, true, (uint8_t *)src,
addrOffset);
break;
}
case RefCountingKind::End:
return dest;
default:
break;
}
}
handleRefCounts<0, TakeHandler>(metadata, (uint8_t *)dest, (uint8_t *)src);
return dest;
}