Loading...
--- xnu/xnu-792.17.14/libkern/c++/OSSymbol.cpp
+++ xnu/xnu-1228/libkern/c++/OSSymbol.cpp
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2000 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 2000-2007 Apple Inc. All rights reserved.
*
* @APPLE_OSREFERENCE_LICENSE_HEADER_START@
*
@@ -51,6 +51,24 @@
#define ACCUMSIZE(s)
#endif
+#define INITIAL_POOL_SIZE (exp2ml(1 + log2(kInitBucketCount)))
+
+#define GROW_FACTOR (1)
+#define SHRINK_FACTOR (3)
+
+#define GROW_POOL() do \
+ if (count * GROW_FACTOR > nBuckets) { \
+ reconstructSymbols(true); \
+ } \
+while (0)
+
+#define SHRINK_POOL() do \
+ if (count * SHRINK_FACTOR < nBuckets && \
+ nBuckets > INITIAL_POOL_SIZE) { \
+ reconstructSymbols(false); \
+ } \
+while (0)
+
class OSSymbolPool
{
private:
@@ -84,7 +102,8 @@
static unsigned long log2(unsigned int x);
static unsigned long exp2ml(unsigned int x);
- void reconstructSymbols();
+ void reconstructSymbols(void);
+ void reconstructSymbols(bool grow);
public:
static void *operator new(size_t size);
@@ -119,14 +138,14 @@
void OSSymbolPool::operator delete(void *mem, size_t size)
{
- kfree((vm_offset_t)mem, size);
+ kfree(mem, size);
ACCUMSIZE(-size);
}
bool OSSymbolPool::init()
{
count = 0;
- nBuckets = exp2ml(1 + log2(kInitBucketCount));
+ nBuckets = INITIAL_POOL_SIZE;
buckets = (Bucket *) kalloc(nBuckets * sizeof(Bucket));
ACCUMSIZE(nBuckets * sizeof(Bucket));
if (!buckets)
@@ -151,12 +170,12 @@
OSSymbolPool::~OSSymbolPool()
{
if (buckets) {
- kfree((vm_offset_t)buckets, nBuckets * sizeof(Bucket));
+ kfree(buckets, nBuckets * sizeof(Bucket));
ACCUMSIZE(-(nBuckets * sizeof(Bucket)));
}
if (poolGate)
- kfree((vm_offset_t) poolGate, 36 * 4);
+ kfree(poolGate, 36 * 4);
}
unsigned long OSSymbolPool::log2(unsigned int x)
@@ -198,14 +217,35 @@
return thisBucket->symbolP[stateP->j];
}
-void OSSymbolPool::reconstructSymbols()
-{
- OSSymbolPool old(this);
+void OSSymbolPool::reconstructSymbols(void)
+{
+ this->reconstructSymbols(true);
+}
+
+void OSSymbolPool::reconstructSymbols(bool grow)
+{
+ unsigned int new_nBuckets = nBuckets;
OSSymbol *insert;
OSSymbolPoolState state;
- nBuckets += nBuckets + 1;
+ if (grow) {
+ new_nBuckets += new_nBuckets + 1;
+ } else {
+ /* Don't shrink the pool below the default initial size.
+ */
+ if (nBuckets <= INITIAL_POOL_SIZE) {
+ return;
+ }
+ new_nBuckets = (new_nBuckets - 1) / 2;
+ }
+
+ /* Create old pool to iterate after doing above check, cause it
+ * gets finalized at return.
+ */
+ OSSymbolPool old(this);
+
count = 0;
+ nBuckets = new_nBuckets;
buckets = (Bucket *) kalloc(nBuckets * sizeof(Bucket));
ACCUMSIZE(nBuckets * sizeof(Bucket));
/* @@@ gvdl: Zero test and panic if can't set up pool */
@@ -281,8 +321,7 @@
thisBucket->symbolP = list;
thisBucket->count++;
count++;
- if (count > nBuckets)
- reconstructSymbols();
+ GROW_POOL();
return sym;
}
@@ -301,11 +340,10 @@
/* @@@ gvdl: Zero test and panic if can't set up pool */
list[0] = sym;
bcopy(thisBucket->symbolP, list + 1, j * sizeof(OSSymbol *));
- kfree((vm_offset_t)thisBucket->symbolP, j * sizeof(OSSymbol *));
+ kfree(thisBucket->symbolP, j * sizeof(OSSymbol *));
ACCUMSIZE(-(j * sizeof(OSSymbol *)));
thisBucket->symbolP = list;
- if (count > nBuckets)
- reconstructSymbols();
+ GROW_POOL();
return sym;
}
@@ -331,6 +369,7 @@
thisBucket->symbolP = 0;
count--;
thisBucket->count--;
+ SHRINK_POOL();
return;
}
return;
@@ -340,20 +379,22 @@
probeSymbol = list[0];
if (probeSymbol == sym) {
thisBucket->symbolP = (OSSymbol **) list[1];
- kfree((vm_offset_t)list, 2 * sizeof(OSSymbol *));
+ kfree(list, 2 * sizeof(OSSymbol *));
ACCUMSIZE(-(2 * sizeof(OSSymbol *)));
count--;
thisBucket->count--;
+ SHRINK_POOL();
return;
}
probeSymbol = list[1];
if (probeSymbol == sym) {
thisBucket->symbolP = (OSSymbol **) list[0];
- kfree((vm_offset_t)list, 2 * sizeof(OSSymbol *));
+ kfree(list, 2 * sizeof(OSSymbol *));
ACCUMSIZE(-(2 * sizeof(OSSymbol *)));
count--;
thisBucket->count--;
+ SHRINK_POOL();
return;
}
return;
@@ -373,7 +414,7 @@
bcopy(thisBucket->symbolP + thisBucket->count-j,
list + thisBucket->count-1-j,
j * sizeof(OSSymbol *));
- kfree((vm_offset_t)thisBucket->symbolP, thisBucket->count * sizeof(OSSymbol *));
+ kfree(thisBucket->symbolP, thisBucket->count * sizeof(OSSymbol *));
ACCUMSIZE(-(thisBucket->count * sizeof(OSSymbol *)));
thisBucket->symbolP = list;
count--;