Files
swift-mirror/stdlib/toolchain/CompatibilityDynamicReplacements/DynamicReplaceable.cpp
Michael Gottesman 3e64d03a52 [exclusivity] Split out of Exclusivity.cpp FunctionReplacement and the Global TLSContext.
Some notes:

1. Even though I refactored out AccessSet/Access from Exclusivity.cpp ->
ExclusivityPrivate.h, I left the actual implementations of insert/remove in
Exclusivity.cpp to allow for the most aggressive optimization for use in
Exclusivity.cpp without exposing a bunch of internal details to other parts of
the runtime. Smaller routines like getHead() and manipulating the linked list
directly I left as methods that can be used by other parts of the runtime. I am
going to use these methods to enable backwards deployment of exclusivity support
for concurrency.

2. I moved function replacements out of the Exclusivity header/cpp files since
it has nothing to do with Exclusivity beyond taking advantage of the TLS context
that we are already using.
2021-07-26 12:26:08 -07:00

72 lines
2.4 KiB
C++

//===--- ProtocolConformance.cpp - Swift protocol conformance checking ----===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2019 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
//
// Runtime support for dynamic replaceable functions.
//
// This implementation is intended to be backward-deployed into Swift 5.0
// runtimes.
//
//===----------------------------------------------------------------------===//
#include "swift/Runtime/Exclusivity.h"
#include "swift/Runtime/FunctionReplacement.h"
#include "swift/Runtime/Once.h"
#include "swift/Runtime/ThreadLocalStorage.h"
using namespace swift;
__attribute__((visibility("hidden"), weak))
extern "C" char *swift_getFunctionReplacement50(char **ReplFnPtr, char *CurrFn) {
// Call the current implementation if it is available.
if (swift_getFunctionReplacement)
return swift_getFunctionReplacement(ReplFnPtr, CurrFn);
char *ReplFn = *ReplFnPtr;
char *RawReplFn = ReplFn;
#if SWIFT_PTRAUTH
RawReplFn = ptrauth_strip(RawReplFn, ptrauth_key_function_pointer);
#endif
if (RawReplFn == CurrFn)
return nullptr;
auto origKey =
(uintptr_t)SWIFT_THREAD_GETSPECIFIC(SWIFT_COMPATIBILITY_50_TLS_KEY);
if ((origKey & 0x1) != 0) {
auto mask = ((uintptr_t)-1) < 1;
auto resetKey = origKey & mask;
SWIFT_THREAD_SETSPECIFIC(SWIFT_COMPATIBILITY_50_TLS_KEY, (void *)resetKey);
return nullptr;
}
return ReplFn;
}
__attribute__((visibility("hidden"), weak))
extern "C" char *swift_getOrigOfReplaceable50(char **OrigFnPtr) {
// Call the current implementation if it is available.
if (swift_getOrigOfReplaceable)
return swift_getOrigOfReplaceable(OrigFnPtr);
char *OrigFn = *OrigFnPtr;
auto origKey =
(uintptr_t)SWIFT_THREAD_GETSPECIFIC(SWIFT_COMPATIBILITY_50_TLS_KEY);
auto newKey = origKey | 0x1;
SWIFT_THREAD_SETSPECIFIC(SWIFT_COMPATIBILITY_50_TLS_KEY, (void *)newKey);
return OrigFn;
}
// Allow this library to get force-loaded by autolinking
__attribute__((weak, visibility("hidden")))
extern "C"
char _swift_FORCE_LOAD_$_swiftCompatibilityDynamicReplacements = 0;