mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
3e64d03a52
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.
42 lines
1.2 KiB
C++
42 lines
1.2 KiB
C++
//===--- FunctionReplacement.cpp ------------------------------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2021 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "swift/Runtime/FunctionReplacement.h"
|
|
#include "SwiftTLSContext.h"
|
|
|
|
using namespace swift;
|
|
using namespace swift::runtime;
|
|
|
|
char *swift::swift_getFunctionReplacement(char **ReplFnPtr, char *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 &ctx = SwiftTLSContext::get();
|
|
if (ctx.CallOriginalOfReplacedFunction) {
|
|
ctx.CallOriginalOfReplacedFunction = false;
|
|
return nullptr;
|
|
}
|
|
return ReplFn;
|
|
}
|
|
|
|
char *swift::swift_getOrigOfReplaceable(char **OrigFnPtr) {
|
|
char *OrigFn = *OrigFnPtr;
|
|
SwiftTLSContext::get().CallOriginalOfReplacedFunction = true;
|
|
return OrigFn;
|
|
}
|