#include "../headers/BTreesPrivate.h"
static OSStatus AddNewRootNode (BTreeControlBlockPtr btreePtr,
NodeDescPtr leftNode,
NodeDescPtr rightNode );
static OSStatus CollapseTree (BTreeControlBlockPtr btreePtr,
BlockDescriptor *blockPtr );
static OSStatus RotateLeft (BTreeControlBlockPtr btreePtr,
NodeDescPtr leftNode,
NodeDescPtr rightNode,
UInt16 rightInsertIndex,
KeyPtr keyPtr,
UInt8 * recPtr,
UInt16 recSize,
UInt16 *insertIndex,
UInt32 *insertNodeNum,
Boolean *recordFit,
UInt16 *recsRotated );
static Boolean RotateRecordLeft (BTreeControlBlockPtr btreePtr,
NodeDescPtr leftNode,
NodeDescPtr rightNode );
static OSStatus SplitLeft (BTreeControlBlockPtr btreePtr,
BlockDescriptor *leftNode,
BlockDescriptor *rightNode,
UInt32 rightNodeNum,
UInt16 index,
KeyPtr keyPtr,
UInt8 * recPtr,
UInt16 recSize,
UInt16 *insertIndex,
UInt32 *insertNodeNum,
UInt16 *recsRotated );
static OSStatus InsertLevel (BTreeControlBlockPtr btreePtr,
TreePathTable treePathTable,
InsertKey *primaryKey,
InsertKey *secondaryKey,
BlockDescriptor *targetNode,
UInt16 index,
UInt16 level,
UInt32 *insertNode );
static OSErr InsertNode (BTreeControlBlockPtr btreePtr,
InsertKey *key,
BlockDescriptor *rightNode,
UInt32 node,
UInt16 index,
UInt32 *newNode,
UInt16 *newIndex,
BlockDescriptor *leftNode,
Boolean *updateParent,
Boolean *insertParent,
Boolean *rootSplit );
static UInt16 GetKeyLength (const BTreeControlBlock *btreePtr,
const BTreeKey *key,
Boolean forLeafNode );
OSStatus SearchTree (BTreeControlBlockPtr btreePtr,
BTreeKeyPtr searchKey,
TreePathTable treePathTable,
UInt32 *nodeNum,
BlockDescriptor *nodePtr,
UInt16 *returnIndex )
{
OSStatus err;
SInt16 level; UInt32 curNodeNum; NodeRec nodeRec;
UInt16 index;
Boolean keyFound;
SInt8 nodeKind; KeyPtr keyPtr;
UInt8 * dataPtr;
UInt16 dataSize;
curNodeNum = btreePtr->rootNode;
level = btreePtr->treeDepth;
if (level == 0) {
err = fsBTEmptyErr;
goto ErrorExit;
}
treePathTable [0].node = 0;
treePathTable [0].index = 0;
while (true)
{
if (curNodeNum == 0)
{
err = btBadNode;
goto ErrorExit;
}
err = GetNode (btreePtr, curNodeNum, &nodeRec);
if (err != noErr)
{
goto ErrorExit;
}
if (((BTNodeDescriptor*)nodeRec.buffer)->height != level)
{
err = btBadNode;
goto ReleaseAndExit;
}
nodeKind = ((BTNodeDescriptor*)nodeRec.buffer)->kind;
if (level == 1)
{
if (nodeKind != kBTLeafNode)
{
err = btBadNode;
goto ReleaseAndExit;
}
}
else
{
if (nodeKind != kBTIndexNode)
{
err = btBadNode;
goto ReleaseAndExit;
}
}
keyFound = SearchNode (btreePtr, nodeRec.buffer, searchKey, &index);
treePathTable [level].node = curNodeNum;
if (nodeKind == kBTLeafNode)
{
treePathTable [level].index = index;
break; }
if ( (keyFound != true) && (index != 0))
--index;
treePathTable [level].index = index;
err = GetRecordByIndex (btreePtr, nodeRec.buffer, index, &keyPtr, &dataPtr, &dataSize);
if (err != noErr)
{
(void) TrashNode(btreePtr, &nodeRec);
goto ErrorExit;
}
curNodeNum = *(UInt32 *)dataPtr;
err = ReleaseNode (btreePtr, &nodeRec);
if (err != noErr)
{
goto ErrorExit;
}
--level;
}
*nodeNum = curNodeNum;
*nodePtr = nodeRec;
*returnIndex = index;
if (keyFound)
return noErr; else
return fsBTRecordNotFoundErr;
ReleaseAndExit:
(void) ReleaseNode(btreePtr, &nodeRec);
ErrorExit:
*nodeNum = 0;
nodePtr->buffer = nil;
nodePtr->blockHeader = nil;
*returnIndex = 0;
return err;
}
OSStatus InsertTree ( BTreeControlBlockPtr btreePtr,
TreePathTable treePathTable,
KeyPtr keyPtr,
UInt8 * recPtr,
UInt16 recSize,
BlockDescriptor *targetNode,
UInt16 index,
UInt16 level,
Boolean replacingKey,
UInt32 *insertNode )
{
InsertKey primaryKey;
OSStatus err;
primaryKey.keyPtr = keyPtr;
primaryKey.keyLength = GetKeyLength(btreePtr, primaryKey.keyPtr, (level == 1));
primaryKey.recPtr = recPtr;
primaryKey.recSize = recSize;
primaryKey.replacingKey = replacingKey;
primaryKey.skipRotate = false;
err = InsertLevel (btreePtr, treePathTable, &primaryKey, nil,
targetNode, index, level, insertNode );
return err;
}
OSStatus InsertLevel (BTreeControlBlockPtr btreePtr,
TreePathTable treePathTable,
InsertKey *primaryKey,
InsertKey *secondaryKey,
BlockDescriptor *targetNode,
UInt16 index,
UInt16 level,
UInt32 *insertNode )
{
OSStatus err;
BlockDescriptor leftNode;
UInt32 targetNodeNum;
UInt32 newNodeNum;
UInt16 newIndex;
Boolean insertParent;
Boolean updateParent;
Boolean newRoot;
InsertKey insertKey;
#if defined(applec) && !defined(__SC__)
PanicIf ((level == 1) && (((NodeDescPtr)targetNode->buffer)->kind != kBTLeafNode), "\P InsertLevel: non-leaf at level 1! ");
#endif
leftNode.buffer = nil;
leftNode.blockHeader = nil;
targetNodeNum = treePathTable [level].node;
insertParent = false;
updateParent = false;
ModifyBlockStart(btreePtr->fileRefNum, targetNode);
err = InsertNode (btreePtr, primaryKey, targetNode, targetNodeNum, index,
&newNodeNum, &newIndex, &leftNode, &updateParent, &insertParent, &newRoot );
M_ExitOnError (err);
if ( newRoot )
{
treePathTable [level + 1].node = btreePtr->rootNode;
treePathTable [level + 1].index = 1; }
if ( level == 1 )
*insertNode = newNodeNum;
if ( secondaryKey != nil )
{
Boolean temp;
err = InsertNode (btreePtr, secondaryKey, targetNode, newNodeNum, newIndex,
&newNodeNum, &newIndex, &leftNode, &updateParent, &insertParent, &temp);
M_ExitOnError (err);
if ( DEBUG_BUILD && updateParent && newRoot )
DebugStr("\p InsertLevel: New root from primary key, update from secondary key...");
}
if ( insertParent || updateParent )
{
BlockDescriptor parentNode;
UInt32 parentNodeNum;
KeyPtr keyPtr;
UInt8 * recPtr;
UInt16 recSize;
parentNode.buffer = nil;
parentNode.blockHeader = nil;
secondaryKey = nil;
PanicIf ( (level == btreePtr->treeDepth), "\p InsertLevel: unfinished insert!?");
++level;
index = treePathTable [level].index;
parentNodeNum = treePathTable [level].node;
PanicIf ( parentNodeNum == 0, "\p InsertLevel: parent node is zero!?");
err = GetNode (btreePtr, parentNodeNum, &parentNode); M_ExitOnError (err);
#if defined(applec) && !defined(__SC__)
if (DEBUG_BUILD && level > 1)
PanicIf ( ((NodeDescPtr)parentNode.buffer)->kind != kBTIndexNode, "\P InsertLevel: parent node not an index node! ");
#endif
if ( updateParent )
{
ModifyBlockStart(btreePtr->fileRefNum, &parentNode);
GetRecordByIndex (btreePtr, parentNode.buffer, index, &keyPtr, &recPtr, &recSize);
PanicIf( (*(UInt32 *) recPtr) != targetNodeNum, "\p InsertLevel: parent ptr doesn't match target node!");
DeleteRecord (btreePtr, parentNode.buffer, index);
primaryKey->keyPtr = (KeyPtr) GetRecordAddress( btreePtr, targetNode->buffer, 0 );
primaryKey->keyLength = GetKeyLength(btreePtr, primaryKey->keyPtr, false);
primaryKey->recPtr = (UInt8 *) &targetNodeNum;
primaryKey->recSize = sizeof(targetNodeNum);
primaryKey->replacingKey = kReplaceRecord;
primaryKey->skipRotate = insertParent; }
if ( insertParent )
{
InsertKey *insertKeyPtr;
if ( updateParent )
{
insertKeyPtr = &insertKey;
secondaryKey = &insertKey;
}
else
{
insertKeyPtr = primaryKey;
}
insertKeyPtr->keyPtr = (KeyPtr) GetRecordAddress (btreePtr, leftNode.buffer, 0);
insertKeyPtr->keyLength = GetKeyLength(btreePtr, insertKeyPtr->keyPtr, false);
insertKeyPtr->recPtr = (UInt8 *) &((NodeDescPtr)targetNode->buffer)->bLink;
insertKeyPtr->recSize = sizeof(UInt32);
insertKeyPtr->replacingKey = kInsertRecord;
insertKeyPtr->skipRotate = false; }
err = InsertLevel (btreePtr, treePathTable, primaryKey, secondaryKey,
&parentNode, index, level, insertNode );
M_ExitOnError (err);
}
err = UpdateNode (btreePtr, targetNode, 0, kLockTransaction); M_ExitOnError (err);
err = UpdateNode (btreePtr, &leftNode, 0, kLockTransaction); M_ExitOnError (err);
return noErr;
ErrorExit:
(void) ReleaseNode (btreePtr, targetNode);
(void) ReleaseNode (btreePtr, &leftNode);
Panic ("\p InsertLevel: an error occurred!");
return err;
}
static OSErr InsertNode (BTreeControlBlockPtr btreePtr,
InsertKey *key,
BlockDescriptor *rightNode,
UInt32 node,
UInt16 index,
UInt32 *newNode,
UInt16 *newIndex,
BlockDescriptor *leftNode,
Boolean *updateParent,
Boolean *insertParent,
Boolean *rootSplit )
{
BlockDescriptor *targetNode;
UInt32 leftNodeNum;
UInt16 recsRotated;
OSErr err;
Boolean recordFit;
*rootSplit = false;
PanicIf ( rightNode->buffer == leftNode->buffer, "\p InsertNode: rightNode == leftNode, huh?");
leftNodeNum = ((NodeDescPtr) rightNode->buffer)->bLink;
if ( node == leftNodeNum )
targetNode = leftNode;
else
targetNode = rightNode;
recordFit = InsertKeyRecord (btreePtr, targetNode->buffer, index, key->keyPtr, key->keyLength, key->recPtr, key->recSize);
if ( recordFit )
{
*newNode = node;
*newIndex = index;
if ( (index == 0) && (((NodeDescPtr) targetNode->buffer)->height != btreePtr->treeDepth) )
*updateParent = true; }
if ( !recordFit && leftNodeNum > 0 )
{
PanicIf ( leftNode->buffer != nil, "\p InsertNode: leftNode already aquired!");
if ( leftNode->buffer == nil )
{
err = GetNode (btreePtr, leftNodeNum, leftNode); M_ExitOnError (err);
ModifyBlockStart(btreePtr->fileRefNum, leftNode);
}
PanicIf ( ((NodeDescPtr) leftNode->buffer)->fLink != node, "\p InsertNode, RotateLeft: invalid sibling link!" );
if ( !key->skipRotate ) {
err = RotateLeft (btreePtr, leftNode->buffer, rightNode->buffer, index, key->keyPtr, key->recPtr,
key->recSize, newIndex, newNode, &recordFit, &recsRotated );
M_ExitOnError (err);
if ( recordFit )
{
if ( key->replacingKey || (recsRotated > 1) || (index > 0) )
*updateParent = true;
}
}
}
if ( !recordFit )
{
err = SplitLeft (btreePtr, leftNode, rightNode, node, index, key->keyPtr,
key->recPtr, key->recSize, newIndex, newNode, &recsRotated);
M_ExitOnError (err);
if ( ((NodeDescPtr) rightNode->buffer)->height == btreePtr->treeDepth )
{
err = AddNewRootNode (btreePtr, leftNode->buffer, rightNode->buffer); M_ExitOnError (err);
*rootSplit = true;
}
else
{
*insertParent = true;
if ( key->replacingKey || (recsRotated > 1) || (index > 0) )
*updateParent = true;
}
}
return noErr;
ErrorExit:
(void) ReleaseNode (btreePtr, leftNode);
return err;
}
OSStatus DeleteTree (BTreeControlBlockPtr btreePtr,
TreePathTable treePathTable,
BlockDescriptor *targetNode,
UInt16 index,
UInt16 level )
{
OSStatus err;
BlockDescriptor parentNode;
BTNodeDescriptor *targetNodePtr;
UInt32 targetNodeNum;
Boolean deleteRequired;
Boolean updateRequired;
parentNode.buffer = nil;
parentNode.blockHeader = nil;
deleteRequired = false;
updateRequired = false;
targetNodeNum = treePathTable[level].node;
targetNodePtr = targetNode->buffer;
PanicIf (targetNodePtr == nil, "\pDeleteTree: targetNode has nil buffer!");
ModifyBlockStart(btreePtr->fileRefNum, targetNode);
DeleteRecord (btreePtr, targetNodePtr, index);
if ( targetNodePtr->numRecords == 0 ) {
BlockDescriptor siblingNode;
UInt32 siblingNodeNum;
deleteRequired = true;
siblingNode.buffer = nil;
siblingNode.blockHeader = nil;
siblingNodeNum = targetNodePtr->bLink; if ( siblingNodeNum != 0 )
{
err = GetNode (btreePtr, siblingNodeNum, &siblingNode);
M_ExitOnError (err);
ModifyBlockStart(btreePtr->fileRefNum, &siblingNode);
((NodeDescPtr)siblingNode.buffer)->fLink = targetNodePtr->fLink;
err = UpdateNode (btreePtr, &siblingNode, 0, kLockTransaction);
M_ExitOnError (err);
}
else if ( targetNodePtr->kind == kBTLeafNode ) {
btreePtr->firstLeafNode = targetNodePtr->fLink;
}
siblingNodeNum = targetNodePtr->fLink; if ( siblingNodeNum != 0 )
{
err = GetNode (btreePtr, siblingNodeNum, &siblingNode);
M_ExitOnError (err);
ModifyBlockStart(btreePtr->fileRefNum, &siblingNode);
((NodeDescPtr)siblingNode.buffer)->bLink = targetNodePtr->bLink;
err = UpdateNode (btreePtr, &siblingNode, 0, kLockTransaction);
M_ExitOnError (err);
}
else if ( targetNodePtr->kind == kBTLeafNode ) {
btreePtr->lastLeafNode = targetNodePtr->bLink;
}
ClearNode (btreePtr, targetNodePtr);
err = UpdateNode (btreePtr, targetNode, 0, kLockTransaction);
M_ExitOnError (err);
err = FreeNode (btreePtr, targetNodeNum);
M_ExitOnError (err);
}
else if ( index == 0 ) {
updateRequired = true; }
if ( level == btreePtr->treeDepth ) {
deleteRequired = false;
updateRequired = false;
if ( targetNode->buffer == nil ) {
btreePtr->rootNode = 0;
btreePtr->treeDepth = 0;
}
else if ( ((NodeDescPtr)targetNode->buffer)->numRecords == 1 )
{
err = CollapseTree (btreePtr, targetNode);
M_ExitOnError (err);
}
}
if ( updateRequired || deleteRequired )
{
++level;
index = treePathTable [level].index;
err = GetNode (btreePtr, treePathTable[level].node, &parentNode);
M_ExitOnError (err);
if ( updateRequired )
{
KeyPtr keyPtr;
UInt8 * recPtr;
UInt16 recSize;
UInt32 insertNode;
ModifyBlockStart(btreePtr->fileRefNum, &parentNode);
GetRecordByIndex (btreePtr, parentNode.buffer, index, &keyPtr, &recPtr, &recSize);
PanicIf( (*(UInt32 *) recPtr) != targetNodeNum, "\p DeleteTree: parent ptr doesn't match targetNodeNum!!");
DeleteRecord (btreePtr, parentNode.buffer, index);
keyPtr = (KeyPtr) GetRecordAddress( btreePtr, targetNode->buffer, 0 );
recPtr = (UInt8 *) &targetNodeNum;
recSize = sizeof(targetNodeNum);
err = InsertTree (btreePtr, treePathTable, keyPtr, recPtr, recSize,
&parentNode, index, level, kReplaceRecord, &insertNode);
M_ExitOnError (err);
}
else {
err = DeleteTree (btreePtr, treePathTable, &parentNode, index, level);
M_ExitOnError (err);
}
}
err = UpdateNode (btreePtr, targetNode, 0, kLockTransaction);
M_ExitOnError (err);
return noErr;
ErrorExit:
(void) ReleaseNode (btreePtr, targetNode);
(void) ReleaseNode (btreePtr, &parentNode);
return err;
}
static OSStatus CollapseTree (BTreeControlBlockPtr btreePtr,
BlockDescriptor *blockPtr )
{
OSStatus err;
UInt32 originalRoot;
UInt32 nodeNum;
originalRoot = btreePtr->rootNode;
ModifyBlockStart(btreePtr->fileRefNum, blockPtr);
while (true)
{
if ( ((NodeDescPtr)blockPtr->buffer)->numRecords > 1)
break;
if ( ((NodeDescPtr)blockPtr->buffer)->kind == kBTLeafNode)
break;
nodeNum = btreePtr->rootNode;
btreePtr->rootNode = GetChildNodeNum (btreePtr, blockPtr->buffer, 0);
--btreePtr->treeDepth;
ClearNode (btreePtr, blockPtr->buffer);
err = UpdateNode (btreePtr, blockPtr, 0, kLockTransaction);
M_ExitOnError (err);
err = FreeNode (btreePtr, nodeNum);
M_ExitOnError (err);
err = GetNode (btreePtr, btreePtr->rootNode, blockPtr);
M_ExitOnError (err);
ModifyBlockStart(btreePtr->fileRefNum, blockPtr);
}
if (btreePtr->rootNode != originalRoot)
M_BTreeHeaderDirty (btreePtr);
err = UpdateNode (btreePtr, blockPtr, 0, kLockTransaction); M_ExitOnError (err);
return noErr;
ErrorExit:
(void) ReleaseNode (btreePtr, blockPtr);
return err;
}
static OSStatus RotateLeft (BTreeControlBlockPtr btreePtr,
NodeDescPtr leftNode,
NodeDescPtr rightNode,
UInt16 rightInsertIndex,
KeyPtr keyPtr,
UInt8 * recPtr,
UInt16 recSize,
UInt16 *insertIndex,
UInt32 *insertNodeNum,
Boolean *recordFit,
UInt16 *recsRotated )
{
OSStatus err;
SInt32 insertSize;
SInt32 nodeSize;
SInt32 leftSize, rightSize;
SInt32 moveSize = 0;
UInt16 keyLength;
UInt16 lengthFieldSize;
UInt16 index, moveIndex;
Boolean didItFit;
keyLength = GetKeyLength(btreePtr, keyPtr, (rightNode->kind == kBTLeafNode));
if ( btreePtr->attributes & kBTBigKeysMask )
lengthFieldSize = sizeof(UInt16);
else
lengthFieldSize = sizeof(UInt8);
insertSize = keyLength + lengthFieldSize + recSize + sizeof(UInt16);
if ( M_IsOdd (insertSize) )
++insertSize;
nodeSize = btreePtr->nodeSize;
rightSize = nodeSize - GetNodeFreeSize (btreePtr, rightNode) + insertSize;
leftSize = nodeSize - GetNodeFreeSize (btreePtr, leftNode);
moveIndex = 0;
while ( leftSize < rightSize )
{
if ( moveIndex < rightInsertIndex )
{
moveSize = GetRecordSize (btreePtr, rightNode, moveIndex) + 2;
}
else if ( moveIndex == rightInsertIndex )
{
moveSize = insertSize;
}
else {
moveSize = GetRecordSize (btreePtr, rightNode, moveIndex - 1) + 2;
}
leftSize += moveSize;
rightSize -= moveSize;
++moveIndex;
}
if ( leftSize > nodeSize ) {
leftSize -= moveSize;
rightSize += moveSize;
--moveIndex;
}
if ( rightSize > nodeSize ) {
*insertIndex = 0;
*insertNodeNum = 0;
*recordFit = false;
*recsRotated = 0;
return noErr;
}
*recsRotated = moveIndex;
*recordFit = true;
index = 0;
while ( index < moveIndex )
{
if ( index == rightInsertIndex ) {
UInt16 leftInsertIndex;
leftInsertIndex = leftNode->numRecords;
didItFit = InsertKeyRecord (btreePtr, leftNode, leftInsertIndex,
keyPtr, keyLength, recPtr, recSize);
if ( !didItFit )
{
Panic ("\pRotateLeft: InsertKeyRecord (left) returned false!");
err = fsBTBadRotateErr;
goto ErrorExit;
}
*insertIndex = leftInsertIndex;
*insertNodeNum = rightNode->bLink;
}
else
{
didItFit = RotateRecordLeft (btreePtr, leftNode, rightNode);
if ( !didItFit )
{
Panic ("\pRotateLeft: RotateRecordLeft returned false!");
err = fsBTBadRotateErr;
goto ErrorExit;
}
}
++index;
}
if ( moveIndex <= rightInsertIndex ) {
rightInsertIndex -= index;
didItFit = InsertKeyRecord (btreePtr, rightNode, rightInsertIndex,
keyPtr, keyLength, recPtr, recSize);
if ( !didItFit )
{
Panic ("\pRotateLeft: InsertKeyRecord (right) returned false!");
err = fsBTBadRotateErr;
goto ErrorExit;
}
*insertIndex = rightInsertIndex;
*insertNodeNum = leftNode->fLink;
}
return noErr;
ErrorExit:
*insertIndex = 0;
*insertNodeNum = 0;
*recordFit = false;
*recsRotated = 0;
return err;
}
static OSStatus SplitLeft (BTreeControlBlockPtr btreePtr,
BlockDescriptor *leftNode,
BlockDescriptor *rightNode,
UInt32 rightNodeNum,
UInt16 index,
KeyPtr keyPtr,
UInt8 * recPtr,
UInt16 recSize,
UInt16 *insertIndex,
UInt32 *insertNodeNum,
UInt16 *recsRotated )
{
OSStatus err;
NodeDescPtr left, right;
UInt32 newNodeNum;
Boolean recordFit;
right = rightNode->buffer;
left = leftNode->buffer;
PanicIf ( right->bLink != 0 && left == 0, "\p SplitLeft: left sibling missing!?" );
if ( (right->height == 1) && (right->kind != kBTLeafNode) )
return fsBTInvalidNodeErr;
if ( left != nil )
{
if ( left->fLink != rightNodeNum )
return fsBTInvalidNodeErr;
if ( left->height != right->height )
return fsBTInvalidNodeErr;
if ( left->kind != right->kind )
return fsBTInvalidNodeErr; }
err = AllocateNode (btreePtr, &newNodeNum);
M_ExitOnError (err);
if ( left != nil )
{
ModifyBlockStart(btreePtr->fileRefNum, leftNode);
left->fLink = newNodeNum;
err = UpdateNode (btreePtr, leftNode, 0, kLockTransaction);
M_ExitOnError (err);
}
err = GetNewNode (btreePtr, newNodeNum, leftNode);
M_ExitOnError (err);
ModifyBlockStart(btreePtr->fileRefNum, leftNode);
left = leftNode->buffer;
left->fLink = rightNodeNum;
left->bLink = right->bLink;
left->kind = right->kind;
left->height = right->height;
right->bLink = newNodeNum;
if ( (left->kind == kBTLeafNode) && (left->bLink == 0) )
{
btreePtr->firstLeafNode = newNodeNum;
M_BTreeHeaderDirty (btreePtr); }
err = RotateLeft (btreePtr, left, right, index, keyPtr, recPtr, recSize,
insertIndex, insertNodeNum, &recordFit, recsRotated);
M_ExitOnError (err);
return noErr;
ErrorExit:
(void) ReleaseNode (btreePtr, leftNode);
(void) ReleaseNode (btreePtr, rightNode);
*insertIndex = 0;
*insertNodeNum = 0;
*recsRotated = 0;
return err;
}
static Boolean RotateRecordLeft (BTreeControlBlockPtr btreePtr,
NodeDescPtr leftNode,
NodeDescPtr rightNode )
{
UInt16 size;
UInt8 * recPtr;
Boolean recordFit;
size = GetRecordSize (btreePtr, rightNode, 0);
recPtr = GetRecordAddress (btreePtr, rightNode, 0);
recordFit = InsertRecord (btreePtr, leftNode, leftNode->numRecords, recPtr, size);
if ( !recordFit )
return false;
DeleteRecord (btreePtr, rightNode, 0);
return true;
}
static OSStatus AddNewRootNode (BTreeControlBlockPtr btreePtr,
NodeDescPtr leftNode,
NodeDescPtr rightNode )
{
OSStatus err;
BlockDescriptor rootNode;
UInt32 rootNum;
KeyPtr keyPtr;
Boolean didItFit;
UInt16 keyLength;
rootNode.buffer = nil;
rootNode.blockHeader = nil;
PanicIf (leftNode == nil, "\pAddNewRootNode: leftNode == nil");
PanicIf (rightNode == nil, "\pAddNewRootNode: rightNode == nil");
err = AllocateNode (btreePtr, &rootNum);
M_ExitOnError (err);
err = GetNewNode (btreePtr, rootNum, &rootNode);
M_ExitOnError (err);
ModifyBlockStart(btreePtr->fileRefNum, &rootNode);
((NodeDescPtr)rootNode.buffer)->kind = kBTIndexNode;
((NodeDescPtr)rootNode.buffer)->height = ++btreePtr->treeDepth;
keyPtr = (KeyPtr) GetRecordAddress (btreePtr, leftNode, 0);
keyLength = GetKeyLength(btreePtr, keyPtr, false);
didItFit = InsertKeyRecord ( btreePtr, rootNode.buffer, 0, keyPtr, keyLength,
(UInt8 *) &rightNode->bLink, 4 );
PanicIf ( !didItFit, "\pAddNewRootNode:InsertKeyRecord failed for left index record");
keyPtr = (KeyPtr) GetRecordAddress (btreePtr, rightNode, 0);
keyLength = GetKeyLength(btreePtr, keyPtr, false);
didItFit = InsertKeyRecord ( btreePtr, rootNode.buffer, 1, keyPtr, keyLength,
(UInt8 *) &leftNode->fLink, 4 );
PanicIf ( !didItFit, "\pAddNewRootNode:InsertKeyRecord failed for right index record");
err = UpdateNode (btreePtr, &rootNode, 0, kLockTransaction);
M_ExitOnError (err);
btreePtr->rootNode = rootNum;
btreePtr->flags |= kBTHeaderDirty;
return noErr;
ErrorExit:
return err;
}
static UInt16 GetKeyLength ( const BTreeControlBlock *btreePtr, const BTreeKey *key, Boolean forLeafNode )
{
UInt16 length;
if ( forLeafNode || btreePtr->attributes & kBTVariableIndexKeysMask )
length = KeyLength (btreePtr, key); else
length = btreePtr->maxKeyLength;
return length;
}