#include <architecture/byte_order.h>
#include "hfs_endian.h"
#include "hfs_dbg.h"
#undef ENDIAN_DEBUG
int hfs_swap_HFSPlusBTInternalNode (BlockDescriptor *src, HFSCatalogNodeID fileID, int unswap);
int hfs_swap_HFSBTInternalNode (BlockDescriptor *src, HFSCatalogNodeID fileID, int unswap);
void
hfs_swap_HFSPlusForkData (
HFSPlusForkData *src
)
{
int i;
src->logicalSize = SWAP_BE64 (src->logicalSize);
src->clumpSize = SWAP_BE32 (src->clumpSize);
src->totalBlocks = SWAP_BE32 (src->totalBlocks);
for (i = 0; i < kHFSPlusExtentDensity; i++) {
src->extents[i].startBlock = SWAP_BE32 (src->extents[i].startBlock);
src->extents[i].blockCount = SWAP_BE32 (src->extents[i].blockCount);
}
}
int
hfs_swap_BTNode (
BlockDescriptor *src,
int isHFSPlus,
HFSCatalogNodeID fileID,
int unswap
)
{
BTNodeDescriptor *srcDesc = src->buffer;
UInt16 *srcOffs = NULL;
UInt32 i;
int error = 0;
#ifdef ENDIAN_DEBUG
if (unswap == 0) {
printf ("BE -> LE Swap\n");
} else if (unswap == 1) {
printf ("LE -> BE Swap\n");
} else if (unswap == 3) {
printf ("Not swapping descriptors\n");
} else {
panic ("%s This is impossible", "hfs_swap_BTNode:");
}
#endif
if (unswap == 0) {
srcDesc->fLink = SWAP_BE32 (srcDesc->fLink);
srcDesc->bLink = SWAP_BE32 (srcDesc->bLink);
srcDesc->numRecords = SWAP_BE16 (srcDesc->numRecords);
srcOffs = (UInt16 *)((char *)src->buffer + (src->blockSize - ((srcDesc->numRecords + 1) * sizeof (UInt16))));
if ((char *)srcOffs > ((char *)src->buffer + src->blockSize)) {
panic ("%s Too many records in the B-Tree node", "hfs_swap_BTNode:");
}
for (i = 0; i < srcDesc->numRecords + 1; i++) {
srcOffs[i] = SWAP_BE16 (srcOffs[i]);
if (srcOffs[i] >= src->blockSize) {
panic ("%s B-Tree node offset out of range", "hfs_swap_BTNode:");
}
}
}
if ((srcDesc->kind == kBTIndexNode) ||
(srcDesc-> kind == kBTLeafNode)) {
if (isHFSPlus) {
error = hfs_swap_HFSPlusBTInternalNode (src, fileID, unswap);
} else {
error = hfs_swap_HFSBTInternalNode (src, fileID, unswap);
}
} else if (srcDesc-> kind == kBTMapNode) {
} else if (srcDesc-> kind == kBTHeaderNode) {
BTHeaderRec *srcHead = (BTHeaderRec *)((char *)src->buffer + 14);
srcHead->treeDepth = SWAP_BE16 (srcHead->treeDepth);
srcHead->rootNode = SWAP_BE32 (srcHead->rootNode);
srcHead->leafRecords = SWAP_BE32 (srcHead->leafRecords);
srcHead->firstLeafNode = SWAP_BE32 (srcHead->firstLeafNode);
srcHead->lastLeafNode = SWAP_BE32 (srcHead->lastLeafNode);
srcHead->nodeSize = SWAP_BE16 (srcHead->nodeSize);
srcHead->maxKeyLength = SWAP_BE16 (srcHead->maxKeyLength);
srcHead->totalNodes = SWAP_BE32 (srcHead->totalNodes);
srcHead->freeNodes = SWAP_BE32 (srcHead->freeNodes);
srcHead->clumpSize = SWAP_BE32 (srcHead->clumpSize);
srcHead->attributes = SWAP_BE32 (srcHead->attributes);
}
if (unswap == 1) {
srcDesc->fLink = SWAP_BE32 (srcDesc->fLink);
srcDesc->bLink = SWAP_BE32 (srcDesc->bLink);
srcOffs = (UInt16 *)((char *)src->buffer + (src->blockSize - ((srcDesc->numRecords + 1) * sizeof (UInt16))));
if ((char *)srcOffs > ((char *)src->buffer + src->blockSize)) {
panic ("%s Too many records in the B-Tree node", "hfs_swap_BTNode:");
}
for (i = 0; i < srcDesc->numRecords + 1; i++) {
if (srcOffs[i] >= src->blockSize) {
panic ("%s B-Tree node offset out of range", "hfs_swap_BTNode:");
}
srcOffs[i] = SWAP_BE16 (srcOffs[i]);
}
srcDesc->numRecords = SWAP_BE16 (srcDesc->numRecords);
}
return (error);
}
int
hfs_swap_HFSPlusBTInternalNode (
BlockDescriptor *src,
HFSCatalogNodeID fileID,
int unswap
)
{
BTNodeDescriptor *srcDesc = src->buffer;
UInt16 *srcOffs = (UInt16 *)((char *)src->buffer + (src->blockSize - (srcDesc->numRecords * sizeof (UInt16))));
UInt32 i;
UInt32 j;
if (fileID == kHFSExtentsFileID) {
HFSPlusExtentKey *srcKey;
HFSPlusExtentDescriptor *srcRec;
for (i = 0; i < srcDesc->numRecords; i++) {
srcKey = (HFSPlusExtentKey *)((char *)src->buffer + srcOffs[i]);
if (!unswap) srcKey->keyLength = SWAP_BE16 (srcKey->keyLength);
srcRec = (HFSPlusExtentDescriptor *)((char *)srcKey + srcKey->keyLength + 2);
if (unswap) srcKey->keyLength = SWAP_BE16 (srcKey->keyLength);
srcKey->fileID = SWAP_BE32 (srcKey->fileID);
srcKey->startBlock = SWAP_BE32 (srcKey->startBlock);
if (srcDesc->kind == kBTIndexNode) {
*((UInt32 *)srcRec) = SWAP_BE32 (*((UInt32 *)srcRec));
continue;
}
for (j = 0; j < kHFSPlusExtentDensity; j++) {
srcRec[j].startBlock = SWAP_BE32 (srcRec[j].startBlock);
srcRec[j].blockCount = SWAP_BE32 (srcRec[j].blockCount);
}
}
} else if (fileID == kHFSCatalogFileID) {
HFSPlusCatalogKey *srcKey;
SInt16 *srcPtr;
for (i = 0; i < srcDesc->numRecords; i++) {
srcKey = (HFSPlusCatalogKey *)((char *)src->buffer + srcOffs[i]);
if (!unswap) srcKey->keyLength = SWAP_BE16 (srcKey->keyLength);
srcPtr = (SInt16 *)((char *)srcKey + srcKey->keyLength + 2);
if (unswap) srcKey->keyLength = SWAP_BE16 (srcKey->keyLength);
srcKey->parentID = SWAP_BE32 (srcKey->parentID);
if (!unswap) srcKey->nodeName.length = SWAP_BE16 (srcKey->nodeName.length);
for (j = 0; j < srcKey->nodeName.length; j++) {
srcKey->nodeName.unicode[j] = SWAP_BE16 (srcKey->nodeName.unicode[j]);
}
if (unswap) srcKey->nodeName.length = SWAP_BE16 (srcKey->nodeName.length);
if (srcDesc->kind == kBTIndexNode) {
*((UInt32 *)srcPtr) = SWAP_BE32 (*((UInt32 *)srcPtr));
continue;
}
if (!unswap) srcPtr[0] = SWAP_BE16 (srcPtr[0]);
if (srcPtr[0] == kHFSPlusFolderRecord) {
HFSPlusCatalogFolder *srcRec = (HFSPlusCatalogFolder *)srcPtr;
srcRec->flags = SWAP_BE16 (srcRec->flags);
srcRec->valence = SWAP_BE32 (srcRec->valence);
srcRec->folderID = SWAP_BE32 (srcRec->folderID);
srcRec->createDate = SWAP_BE32 (srcRec->createDate);
srcRec->contentModDate = SWAP_BE32 (srcRec->contentModDate);
srcRec->attributeModDate = SWAP_BE32 (srcRec->attributeModDate);
srcRec->accessDate = SWAP_BE32 (srcRec->accessDate);
srcRec->backupDate = SWAP_BE32 (srcRec->backupDate);
srcRec->bsdInfo.ownerID = SWAP_BE32 (srcRec->bsdInfo.ownerID);
srcRec->bsdInfo.groupID = SWAP_BE32 (srcRec->bsdInfo.groupID);
srcRec->bsdInfo.fileMode = SWAP_BE16 (srcRec->bsdInfo.fileMode);
srcRec->bsdInfo.special.iNodeNum = SWAP_BE32 (srcRec->bsdInfo.special.iNodeNum);
srcRec->textEncoding = SWAP_BE32 (srcRec->textEncoding);
} else if (srcPtr[0] == kHFSPlusFileRecord) {
HFSPlusCatalogFile *srcRec = (HFSPlusCatalogFile *)srcPtr;
srcRec->flags = SWAP_BE16 (srcRec->flags);
srcRec->fileID = SWAP_BE32 (srcRec->fileID);
srcRec->createDate = SWAP_BE32 (srcRec->createDate);
srcRec->contentModDate = SWAP_BE32 (srcRec->contentModDate);
srcRec->attributeModDate = SWAP_BE32 (srcRec->attributeModDate);
srcRec->accessDate = SWAP_BE32 (srcRec->accessDate);
srcRec->backupDate = SWAP_BE32 (srcRec->backupDate);
srcRec->bsdInfo.ownerID = SWAP_BE32 (srcRec->bsdInfo.ownerID);
srcRec->bsdInfo.groupID = SWAP_BE32 (srcRec->bsdInfo.groupID);
srcRec->bsdInfo.fileMode = SWAP_BE16 (srcRec->bsdInfo.fileMode);
srcRec->bsdInfo.special.iNodeNum = SWAP_BE32 (srcRec->bsdInfo.special.iNodeNum);
srcRec->textEncoding = SWAP_BE32 (srcRec->textEncoding);
hfs_swap_HFSPlusForkData (&srcRec->dataFork);
hfs_swap_HFSPlusForkData (&srcRec->resourceFork);
} else if ((srcPtr[0] == kHFSPlusFolderThreadRecord) ||
(srcPtr[0] == kHFSPlusFileThreadRecord)) {
HFSPlusCatalogThread *srcRec = (HFSPlusCatalogThread *)srcPtr;
srcRec->parentID = SWAP_BE32 (srcRec->parentID);
if (!unswap) srcRec->nodeName.length = SWAP_BE16 (srcRec->nodeName.length);
for (j = 0; j < srcRec->nodeName.length; j++) {
srcRec->nodeName.unicode[j] = SWAP_BE16 (srcRec->nodeName.unicode[j]);
}
if (unswap) srcRec->nodeName.length = SWAP_BE16 (srcRec->nodeName.length);
} else {
panic ("%s unrecognized catalog record type", "hfs_swap_BTNode:");
}
if (unswap) srcPtr[0] = SWAP_BE16 (srcPtr[0]);
}
} else {
panic ("%s unrecognized B-Tree type", "hfs_swap_BTNode:");
}
return (0);
}
int
hfs_swap_HFSBTInternalNode (
BlockDescriptor *src,
HFSCatalogNodeID fileID,
int unswap
)
{
BTNodeDescriptor *srcDesc = src->buffer;
UInt16 *srcOffs = (UInt16 *)((char *)src->buffer + (src->blockSize - (srcDesc->numRecords * sizeof (UInt16))));
UInt32 i;
UInt32 j;
if (fileID == kHFSExtentsFileID) {
HFSExtentKey *srcKey;
HFSExtentDescriptor *srcRec;
for (i = 0; i < srcDesc->numRecords; i++) {
srcKey = (HFSExtentKey *)((char *)src->buffer + srcOffs[i]);
srcKey->fileID = SWAP_BE32 (srcKey->fileID);
srcKey->startBlock = SWAP_BE16 (srcKey->startBlock);
srcRec = (HFSExtentDescriptor *)((char *)srcKey + ((srcKey->keyLength + 2) & ~1));
if (srcDesc->kind == kBTIndexNode) {
*((UInt32 *)srcRec) = SWAP_BE32 (*((UInt32 *)srcRec));
continue;
}
for (j = 0; j < kHFSExtentDensity; j++) {
srcRec[j].startBlock = SWAP_BE16 (srcRec[j].startBlock);
srcRec[j].blockCount = SWAP_BE16 (srcRec[j].blockCount);
}
}
} else if (fileID == kHFSCatalogFileID) {
HFSCatalogKey *srcKey;
SInt16 *srcPtr;
for (i = 0; i < srcDesc->numRecords; i++) {
srcKey = (HFSCatalogKey *)((char *)src->buffer + srcOffs[i]);
srcKey->parentID = SWAP_BE32 (srcKey->parentID);
srcPtr = (SInt16 *)((char *)srcKey + ((srcKey->keyLength + 2) & ~1));
if (srcDesc->kind == kBTIndexNode) {
*((UInt32 *)srcPtr) = SWAP_BE32 (*((UInt32 *)srcPtr));
continue;
}
if (!unswap) srcPtr[0] = SWAP_BE16 (srcPtr[0]);
if (srcPtr[0] == kHFSFolderRecord) {
HFSCatalogFolder *srcRec = (HFSCatalogFolder *)srcPtr;
srcRec->flags = SWAP_BE16 (srcRec->flags);
srcRec->valence = SWAP_BE16 (srcRec->valence);
srcRec->folderID = SWAP_BE32 (srcRec->folderID);
srcRec->createDate = SWAP_BE32 (srcRec->createDate);
srcRec->modifyDate = SWAP_BE32 (srcRec->modifyDate);
srcRec->backupDate = SWAP_BE32 (srcRec->backupDate);
} else if (srcPtr[0] == kHFSFileRecord) {
HFSCatalogFile *srcRec = (HFSCatalogFile *)srcPtr;
srcRec->flags = srcRec->flags;
srcRec->fileType = srcRec->fileType;
srcRec->fileID = SWAP_BE32 (srcRec->fileID);
srcRec->dataStartBlock = SWAP_BE16 (srcRec->dataStartBlock);
srcRec->dataLogicalSize = SWAP_BE32 (srcRec->dataLogicalSize);
srcRec->dataPhysicalSize = SWAP_BE32 (srcRec->dataPhysicalSize);
srcRec->rsrcStartBlock = SWAP_BE16 (srcRec->rsrcStartBlock);
srcRec->rsrcLogicalSize = SWAP_BE32 (srcRec->rsrcLogicalSize);
srcRec->rsrcPhysicalSize = SWAP_BE32 (srcRec->rsrcPhysicalSize);
srcRec->createDate = SWAP_BE32 (srcRec->createDate);
srcRec->modifyDate = SWAP_BE32 (srcRec->modifyDate);
srcRec->backupDate = SWAP_BE32 (srcRec->backupDate);
srcRec->clumpSize = SWAP_BE16 (srcRec->clumpSize);
for (j = 0; j < kHFSExtentDensity * 2; j++) {
srcRec->dataExtents[j].startBlock = SWAP_BE16 (srcRec->dataExtents[j].startBlock);
srcRec->dataExtents[j].blockCount = SWAP_BE16 (srcRec->dataExtents[j].blockCount);
}
} else if ((srcPtr[0] == kHFSFolderThreadRecord) ||
(srcPtr[0] == kHFSFileThreadRecord)) {
HFSCatalogThread *srcRec = (HFSCatalogThread *)srcPtr;
srcRec->parentID = SWAP_BE32 (srcRec->parentID);
} else {
panic ("%s unrecognized catalog record type", "hfs_swap_BTNode:");
}
if (unswap) srcPtr[0] = SWAP_BE16 (srcPtr[0]);
}
} else {
panic ("%s unrecognized B-Tree type", "hfs_swap_BTNode:");
}
return (0);
}