6502/llvm-6502
1
0
Fork 0
mirror of https://github.com/c64scene-ar/llvm-6502.git synced 2024-12-13 20:32:21 +00:00
Code Issues Projects Releases Wiki Activity

Protect the GC table in Function.cpp

Browse Source 
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@73647 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Owen Anderson 2009-06-17 22:23:31 +00:00
parent bf823e7590
commit b983d67465
2 changed files with 39 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
include/llvm/System/RWMutex.h
View File

@ -79,6 +79,20 @@ namespace llvm
/// @}
};
/// ScopedReader - RAII acquisition of a writer lock
struct ScopedReader {
RWMutex* mutex;
explicit ScopedReader(RWMutex* m) {
mutex = m;
mutex->reader_acquire();
}
~ScopedReader() {
mutex->reader_release();
}
};
/// ScopedWriter - RAII acquisition of a writer lock
struct ScopedWriter {
RWMutex* mutex;

44
lib/VMCore/Function.cpp
View File

@ -16,7 +16,9 @@
#include "llvm/IntrinsicInst.h"
#include "llvm/CodeGen/ValueTypes.h"
#include "llvm/Support/LeakDetector.h"
#include "llvm/Support/ManagedStatic.h"
#include "llvm/Support/StringPool.h"
#include "llvm/System/RWMutex.h"
#include "SymbolTableListTraitsImpl.h"
#include "llvm/ADT/DenseMap.h"
#include "llvm/ADT/StringExtras.h"
@ -228,38 +230,42 @@ void Function::removeAttribute(unsigned i, Attributes attr) {
// allocating an additional word in Function for programs which do not use GC
// (i.e., most programs) at the cost of increased overhead for clients which do
// use GC.
static DenseMap<const Function*,PooledStringPtr> *GCNames;
static StringPool *GCNamePool;
static ManagedStatic<DenseMap<const Function*,PooledStringPtr> > GCNames;
static ManagedStatic<StringPool> GCNamePool;
static ManagedStatic<sys::RWMutex> GCLock;
bool Function::hasGC() const {
return GCNames && GCNames->count(this);
if (llvm_is_multithreaded()) {
sys::ScopedReader Reader(&*GCLock);
return GCNames->count(this);
} else
return GCNames->count(this);
}
const char *Function::getGC() const {
assert(hasGC() && "Function has no collector");
return *(*GCNames)[this];
if (llvm_is_multithreaded()) {
sys::ScopedReader Reader(&*GCLock);
return *(*GCNames)[this];
} else
return *(*GCNames)[this];
}
void Function::setGC(const char *Str) {
if (!GCNamePool)
GCNamePool = new StringPool();
if (!GCNames)
GCNames = new DenseMap<const Function*,PooledStringPtr>();
(*GCNames)[this] = GCNamePool->intern(Str);
if (llvm_is_multithreaded()) {
sys::ScopedWriter Writer(&*GCLock);
(*GCNames)[this] = GCNamePool->intern(Str);
} else
(*GCNames)[this] = GCNamePool->intern(Str);
}
void Function::clearGC() {
if (GCNames) {
if (llvm_is_multithreaded()) {
sys::ScopedWriter Writer(&*GCLock);
GCNames->erase(this);
} else
GCNames->erase(this);
if (GCNames->empty()) {
delete GCNames;
GCNames = 0;
if (GCNamePool->empty()) {
delete GCNamePool;
GCNamePool = 0;
}
}
}
}
/// copyAttributesFrom - copy all additional attributes (those not needed to