Mozilla/mozilla/xpinstall/src/nsAppleSingleDecoder.cpp

758 lines
18 KiB
C++

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* The contents of this file are subject to the Netscape Public License
* Version 1.1 (the "License"); you may not use this file except in
* compliance with the License. You may obtain a copy of the License at
* http://www.mozilla.org/NPL/
*
* Software distributed under the License is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
* for the specific language governing rights and limitations under the
* License.
*
* The Original Code is Mozilla Communicator client code, released March
* 31, 1998.
*
* The Initial Developer of the Original Code is Netscape Communications
* Corporation. Portions created by Netscape are Copyright (C) 1999
* Netscape Communications Corporation. All Rights Reserved.
*
* Contributors:
* Samir Gehani <sgehani@netscape.com>
*/
#ifndef _NS_APPLESINGLEDECODER_H_
#include "nsAppleSingleDecoder.h"
#endif
#include "MoreFilesExtras.h"
#include "MoreDesktopMgr.h"
#include "IterateDirectory.h"
#ifdef MOZILLA_CLIENT
#include "nsISupportsUtils.h"
#endif
/*----------------------------------------------------------------------*
* Constructors/Destructor
*----------------------------------------------------------------------*/
#ifdef MOZILLA_CLIENT
MOZ_DECL_CTOR_COUNTER(nsAppleSingleDecoder)
#endif
nsAppleSingleDecoder::nsAppleSingleDecoder(FSSpec *inSpec, FSSpec *outSpec)
: mInSpec(NULL),
mOutSpec(NULL),
mInRefNum(0),
mRenameReqd(false)
{
#ifdef MOZILLA_CLIENT
MOZ_COUNT_CTOR(nsAppleSingleDecoder);
#endif
if (inSpec && outSpec)
{
/* merely point to FSSpecs, not own 'em */
mInSpec = inSpec;
mOutSpec = outSpec;
}
}
nsAppleSingleDecoder::nsAppleSingleDecoder()
: mInSpec(NULL),
mOutSpec(NULL),
mInRefNum(0),
mRenameReqd(false)
{
#ifdef MOZILLA_CLIENT
MOZ_COUNT_CTOR(nsAppleSingleDecoder);
#endif
}
nsAppleSingleDecoder::~nsAppleSingleDecoder()
{
/* not freeing FSSpecs since we don't own 'em */
#ifdef MOZILLA_CLIENT
MOZ_COUNT_DTOR(nsAppleSingleDecoder);
#endif
}
#pragma mark -
/*----------------------------------------------------------------------*
* Public methods
*----------------------------------------------------------------------*/
OSErr
nsAppleSingleDecoder::Decode()
{
OSErr err = noErr;
ASHeader header;
long bytesRead = sizeof(header);
// param check
if (!mInSpec || !mOutSpec)
return paramErr;
// check for existence
FSSpec tmp;
err = FSMakeFSSpec(mInSpec->vRefNum, mInSpec->parID, mInSpec->name, &tmp);
if (err == fnfErr)
return err;
MAC_ERR_CHECK(FSpOpenDF( mInSpec, fsRdPerm, &mInRefNum ));
MAC_ERR_CHECK(FSRead( mInRefNum, &bytesRead, &header ));
if ( (bytesRead != sizeof(header)) ||
(header.magicNum != 0x00051600) ||
(header.versionNum != 0x00020000) ||
(header.numEntries == 0) ) // empty file?
return -1;
// create the outSpec which we'll rename correctly later
err = FSMakeFSSpec( mInSpec->vRefNum, mInSpec->parID, "\pdecode", mOutSpec );
if (err!=noErr && err!=fnfErr)
return err;
MAC_ERR_CHECK(FSMakeUnique( mOutSpec ));
MAC_ERR_CHECK(FSpCreate( mOutSpec, 'MOZZ', '????', 0 ));
/* Loop through the entries, processing each.
** Set the time/date stamps last, because otherwise they'll
** be destroyed when we write.
*/
{
Boolean hasDateEntry = false;
ASEntry dateEntry;
long offset;
ASEntry entry;
for ( int i=0; i < header.numEntries; i++ )
{
offset = sizeof( ASHeader ) + sizeof( ASEntry ) * i;
MAC_ERR_CHECK(SetFPos( mInRefNum, fsFromStart, offset ));
bytesRead = sizeof(entry);
MAC_ERR_CHECK(FSRead( mInRefNum, &bytesRead, &entry ));
if (bytesRead != sizeof(entry))
return -1;
if ( entry.entryID == AS_FILEDATES )
{
hasDateEntry = true;
dateEntry = entry;
}
else
MAC_ERR_CHECK(ProcessASEntry( entry ));
}
if ( hasDateEntry )
MAC_ERR_CHECK(ProcessASEntry( dateEntry ));
}
// close the inSpec
FSClose( mInRefNum );
// rename if need be
if (mRenameReqd)
{
FSSpec old; // delete old version of target file
FSMakeFSSpec(mInSpec->vRefNum, mInSpec->parID, mInSpec->name, &old);
MAC_ERR_CHECK(FSpDelete(&old));
MAC_ERR_CHECK(FSpRename(mOutSpec, mInSpec->name));
// reflect change in outSpec
nsAppleSingleDecoder::PLstrncpy( mOutSpec->name, mInSpec->name, mInSpec->name[0] );
mOutSpec->name[0] = mInSpec->name[0];
mRenameReqd = false; // XXX redundant reinit?
}
if (mType == 'APPL')
{
// need to register in desktop database
DTSetAPPL( NULL,
mOutSpec->vRefNum,
mCreator,
mOutSpec->parID,
mOutSpec->name );
}
return err;
}
OSErr
nsAppleSingleDecoder::Decode(FSSpec *inSpec, FSSpec *outSpec)
{
OSErr err = noErr;
// param check
if (inSpec && outSpec)
{
mInSpec = inSpec; // reinit
mOutSpec = outSpec;
mRenameReqd = false;
}
else
return paramErr;
err = Decode();
return err;
}
pascal void
DecodeDirIterateFilter(const CInfoPBRec * const cpbPtr, Boolean *quitFlag, void *yourDataPtr)
{
OSErr err = noErr;
FSSpec currFSp, outFSp;
nsAppleSingleDecoder* thisObj = NULL;
Boolean isDir = false;
long dummy;
// param check
if (!yourDataPtr || !cpbPtr || !quitFlag)
return;
*quitFlag = false;
// extract 'this' -- an nsAppleSingleDecoder instance
thisObj = (nsAppleSingleDecoder*) yourDataPtr;
// make an FSSpec from the CInfoPBRec*
err = FSMakeFSSpec(cpbPtr->hFileInfo.ioVRefNum, cpbPtr->hFileInfo.ioFlParID,
cpbPtr->hFileInfo.ioNamePtr, &currFSp);
if (err == noErr)
{
FSpGetDirectoryID(&currFSp, &dummy, &isDir);
// if current FSSpec is file
if (!isDir)
{
// if file is in AppleSingle format
if (nsAppleSingleDecoder::IsAppleSingleFile(&currFSp))
{
// decode file
thisObj->Decode(&currFSp, &outFSp);
}
}
else
{
// else if current FSSpec is folder ignore
// XXX never reached?
return;
}
}
}
OSErr
nsAppleSingleDecoder::DecodeFolder(FSSpec *aFolder)
{
OSErr err = noErr;
long dummy;
Boolean isDir = false;
// check that FSSpec is folder
if (aFolder)
{
FSpGetDirectoryID(aFolder, &dummy, &isDir);
if (!isDir)
return dirNFErr;
}
// recursively enumerate contents of folder (maxLevels=0 means recurse all)
FSpIterateDirectory(aFolder, 0, DecodeDirIterateFilter, (void*)this);
return err;
}
Boolean
nsAppleSingleDecoder::IsAppleSingleFile(FSSpec *inSpec)
{
OSErr err;
Boolean bAppleSingle = false;
short inRefNum;
UInt32 magic;
long bytesRead = sizeof(magic);
// param checks
if (!inSpec)
return false;
// check for existence
FSSpec tmp;
err = FSMakeFSSpec(inSpec->vRefNum, inSpec->parID, inSpec->name, &tmp);
if (err!=noErr)
return false;
// open and read the magic number len bytes
err = FSpOpenDF( inSpec, fsRdPerm, &inRefNum );
if (err!=noErr)
return false;
err = FSRead( inRefNum, &bytesRead, &magic );
if (err!=noErr)
return false;
FSClose(inRefNum);
if (bytesRead != sizeof(magic))
return false;
// check if bytes read match magic number
bAppleSingle = (magic == 0x00051600);
return bAppleSingle;
}
#pragma mark -
/*----------------------------------------------------------------------*
* Private methods
*----------------------------------------------------------------------*/
OSErr
nsAppleSingleDecoder::ProcessASEntry(ASEntry inEntry)
{
switch (inEntry.entryID)
{
case AS_DATA:
return ProcessDataFork( inEntry );
break;
case AS_RESOURCE:
return ProcessResourceFork( inEntry );
break;
case AS_REALNAME:
ProcessRealName( inEntry );
break;
// return 0; // Ignore these errors in ASD <--- XXX remove
case AS_COMMENT:
// return ProcessComment( inEntry );
break;
case AS_ICONBW:
// return ProcessIconBW( inEntry );
break;
case AS_ICONCOLOR:
// return ProcessIconColor( inEntry );
break;
case AS_FILEDATES:
return ProcessFileDates( inEntry );
break;
case AS_FINDERINFO:
return ProcessFinderInfo( inEntry );
break;
case AS_MACINFO:
return ProcessMacInfo( inEntry );
break;
case AS_PRODOSINFO:
case AS_MSDOSINFO:
case AS_AFPNAME:
case AS_AFPINFO:
case AS_AFPDIRID:
default:
return 0;
}
return 0;
}
OSErr
nsAppleSingleDecoder::ProcessDataFork(ASEntry inEntry)
{
OSErr err = noErr;
SInt16 refNum;
/* Setup the files */
err = FSpOpenDF (mOutSpec, fsWrPerm, &refNum);
if ( err == noErr )
err = EntryToMacFile( inEntry, refNum );
FSClose( refNum );
return err;
}
OSErr
nsAppleSingleDecoder::ProcessResourceFork(ASEntry inEntry)
{
OSErr err = noErr;
SInt16 refNum;
err = FSpOpenRF(mOutSpec, fsWrPerm, &refNum);
if ( err == noErr )
err = EntryToMacFile( inEntry, refNum );
FSClose( refNum );
return err;
}
OSErr
nsAppleSingleDecoder::ProcessRealName(ASEntry inEntry)
{
OSErr err = noErr;
Str255 newName;
long bytesRead;
if ( inEntry.entryLength > 32 ) /* Max file name length for the Mac */
return -1;
MAC_ERR_CHECK(SetFPos(mInRefNum, fsFromStart, inEntry.entryOffset));
bytesRead = inEntry.entryLength;
MAC_ERR_CHECK(FSRead(mInRefNum, &bytesRead, &newName[1]));
if (bytesRead != inEntry.entryLength)
return -1;
newName[0] = inEntry.entryLength;
err = FSpRename(mOutSpec, newName);
if (err == dupFNErr)
{
// if we are trying to rename temp decode file to src name, rename later
if (nsAppleSingleDecoder::PLstrcmp(newName, mInSpec->name))
{
mRenameReqd = true;
return noErr;
}
FSSpec old; // delete old version of target file
FSMakeFSSpec(mOutSpec->vRefNum, mOutSpec->parID, newName, &old);
MAC_ERR_CHECK(FSpDelete(&old));
MAC_ERR_CHECK(FSpRename(mOutSpec, newName));
}
nsAppleSingleDecoder::PLstrncpy( mOutSpec->name, newName, inEntry.entryLength );
mOutSpec->name[0] = inEntry.entryLength;
return err;
}
OSErr
nsAppleSingleDecoder::ProcessFileDates(ASEntry inEntry)
{
OSErr err = noErr;
ASFileDates dates;
long bytesRead;
if ( inEntry.entryLength != sizeof(dates) ) /* Max file name length for the Mac */
return -1;
MAC_ERR_CHECK(SetFPos(mInRefNum, fsFromStart, inEntry.entryOffset));
bytesRead = inEntry.entryLength;
MAC_ERR_CHECK(FSRead(mInRefNum, &bytesRead, &dates));
if (bytesRead != inEntry.entryLength)
return -1;
Str31 name;
nsAppleSingleDecoder::PLstrncpy(name, mOutSpec->name, mOutSpec->name[0]);
name[0] = mOutSpec->name[0];
CInfoPBRec pb;
pb.hFileInfo.ioNamePtr = &name[0];
pb.hFileInfo.ioVRefNum = mOutSpec->vRefNum;
pb.hFileInfo.ioDirID = mOutSpec->parID;
pb.hFileInfo.ioFDirIndex = 0; /* use ioNamePtr and ioDirID */
err = PBGetCatInfoSync(&pb);
if ( err != noErr )
return -1;
#define YR_2000_SECONDS 3029529600
pb.hFileInfo.ioFlCrDat = dates.create + YR_2000_SECONDS;
pb.hFileInfo.ioFlMdDat = dates.modify + YR_2000_SECONDS;
pb.hFileInfo.ioFlBkDat = dates.backup + YR_2000_SECONDS;
/* Not sure if mac has the last access time */
nsAppleSingleDecoder::PLstrncpy(name, mOutSpec->name, mOutSpec->name[0]);
name[0] = mOutSpec->name[0];
pb.hFileInfo.ioNamePtr = name;
pb.hFileInfo.ioVRefNum = mOutSpec->vRefNum;
pb.hFileInfo.ioDirID = mOutSpec->parID;
pb.hFileInfo.ioFDirIndex = 0; /* use ioNamePtr and ioDirID */
err = PBSetCatInfo(&pb, false);
return err;
}
OSErr
nsAppleSingleDecoder::ProcessFinderInfo(ASEntry inEntry)
{
OSErr err = noErr;
ASFinderInfo info;
long bytesRead;
if (inEntry.entryLength != sizeof( ASFinderInfo ))
return -1;
MAC_ERR_CHECK(SetFPos(mInRefNum, fsFromStart, inEntry.entryOffset));
bytesRead = sizeof(info);
MAC_ERR_CHECK(FSRead(mInRefNum, &bytesRead, &info));
if (bytesRead != inEntry.entryLength)
return -1;
err = FSpSetFInfo(mOutSpec, &info.ioFlFndrInfo);
if (err!=noErr && err!=fnfErr)
return err;
Str31 name;
nsAppleSingleDecoder::PLstrncpy(name, mOutSpec->name, mOutSpec->name[0]);
name[0] = mOutSpec->name[0];
CInfoPBRec pb;
pb.hFileInfo.ioNamePtr = name;
pb.hFileInfo.ioVRefNum = mOutSpec->vRefNum;
pb.hFileInfo.ioDirID = mOutSpec->parID;
pb.hFileInfo.ioFDirIndex = 0; /* use ioNamePtr and ioDirID */
MAC_ERR_CHECK(PBGetCatInfoSync(&pb));
pb.hFileInfo.ioNamePtr = name;
pb.hFileInfo.ioVRefNum = mOutSpec->vRefNum;
pb.hFileInfo.ioDirID = mOutSpec->parID;
pb.hFileInfo.ioFDirIndex = 0; /* use ioNamePtr and ioDirID */
pb.hFileInfo.ioFlXFndrInfo = info.ioFlXFndrInfo;
err = PBSetCatInfo(&pb, false);
// save the creator and file type so we can set it on the renamed spec
mCreator = info.ioFlFndrInfo.fdCreator;
mType = info.ioFlFndrInfo.fdType;
return err;
}
OSErr
nsAppleSingleDecoder::ProcessMacInfo(ASEntry inEntry)
{
OSErr err = noErr;
ASMacInfo info;
long bytesRead;
if (inEntry.entryLength != sizeof( ASMacInfo ))
return -1;
MAC_ERR_CHECK(SetFPos(mInRefNum, fsFromStart, inEntry.entryOffset));
bytesRead = sizeof(info);
MAC_ERR_CHECK(FSRead(mInRefNum, &bytesRead, &info));
if (bytesRead != inEntry.entryLength)
return -1;
Str31 name;
nsAppleSingleDecoder::PLstrncpy(name, mOutSpec->name, mOutSpec->name[0]);
name[0] = mOutSpec->name[0];
CInfoPBRec pb;
pb.hFileInfo.ioNamePtr = name;
pb.hFileInfo.ioVRefNum = mOutSpec->vRefNum;
pb.hFileInfo.ioDirID = mOutSpec->parID;
pb.hFileInfo.ioFDirIndex = 0; /* use ioNamePtr and ioDirID */
MAC_ERR_CHECK(PBGetCatInfoSync(&pb));
pb.hFileInfo.ioNamePtr = name;
pb.hFileInfo.ioVRefNum = mOutSpec->vRefNum;
pb.hFileInfo.ioDirID = mOutSpec->parID;
pb.hFileInfo.ioFlAttrib = info.ioFlAttrib;
err = PBSetCatInfo(&pb, false);
return err;
}
OSErr
nsAppleSingleDecoder::EntryToMacFile(ASEntry inEntry, UInt16 inTargetSpecRefNum)
{
#define BUFFER_SIZE 8192
OSErr err = noErr;
char buffer[BUFFER_SIZE];
long totalRead = 0, bytesRead, bytesToWrite;
MAC_ERR_CHECK(SetFPos( mInRefNum, fsFromStart, inEntry.entryOffset ));
while ( totalRead < inEntry.entryLength )
{
// Should we yield in here?
bytesRead = BUFFER_SIZE;
err = FSRead( mInRefNum, &bytesRead, buffer );
if (err!=noErr && err!=eofErr)
return err;
if ( bytesRead <= 0 )
return -1;
bytesToWrite = totalRead + bytesRead > inEntry.entryLength ?
inEntry.entryLength - totalRead :
bytesRead;
totalRead += bytesRead;
MAC_ERR_CHECK(FSWrite(inTargetSpecRefNum, &bytesToWrite, buffer));
}
return err;
}
#pragma mark -
OSErr
nsAppleSingleDecoder::DTSetAPPL(Str255 volName, short vRefNum, OSType creator,
long applParID, Str255 applName)
{
OSErr err;
DTPBRec *pb = NULL;
short dtRefNum;
short realVRefNum;
Boolean newDTDatabase;
/* get the real vRefnum */
err = DetermineVRefNum(volName, vRefNum, &realVRefNum);
if (err == noErr)
{
err = DTOpen(volName, vRefNum, &dtRefNum, &newDTDatabase);
if (err == noErr && !newDTDatabase)
{
pb = (DTPBRec*) NewPtrClear( sizeof(DTPBRec) );
if (pb==NULL) return -1;
pb->ioNamePtr = applName;
pb->ioDTRefNum = dtRefNum;
pb->ioDirID = applParID;
pb->ioFileCreator = creator;
err = PBDTAddAPPLSync(pb);
if (pb) DisposePtr((Ptr)pb);
}
}
return err;
}
OSErr
nsAppleSingleDecoder::FSMakeUnique(FSSpec *ioSpec)
{
OSErr err = noErr;
Boolean bUnique = false;
FSSpec tmp;
long uniqueID = 0;
Str255 name;
short i, j;
unsigned char puniqueID[16];
char *cuniqueIDPtr;
// grab suggested name from in-out FSSpec
nsAppleSingleDecoder::PLstrncpy(name, ioSpec->name, ioSpec->name[0]);
name[0] = ioSpec->name[0];
for (i=0; i<65536; i++) // prevent infinite loop
{
if (!bUnique)
{
err = FSMakeFSSpec( ioSpec->vRefNum, ioSpec->parID, name, &tmp );
if (err == fnfErr)
{
bUnique = true;
break;
}
else if (err == noErr) // file already exists
{
// grab suggested name from in-out FSSpec
nsAppleSingleDecoder::PLstrncpy(name, ioSpec->name, ioSpec->name[0]);
name[0] = ioSpec->name[0];
// attempt to create a new unique file name
nsAppleSingleDecoder::PLstrncat( name, "\p-", 1 );
// tack on digit(s)
cuniqueIDPtr = nsAppleSingleDecoder::ltoa(uniqueID++);
puniqueID[0] = strlen(cuniqueIDPtr);
for (j=0; j<strlen(cuniqueIDPtr); j++)
{
puniqueID[j+1] = cuniqueIDPtr[j];
}
nsAppleSingleDecoder::PLstrncat( name, puniqueID, puniqueID[0] );
DisposePtr((Ptr)cuniqueIDPtr);
}
else
return err;
}
}
// put back unique name into in-out FSSpec
nsAppleSingleDecoder::PLstrncpy(ioSpec->name, name, name[0]);
ioSpec->name[0] = name[0];
return noErr;
}
/*----------------------------------------------------------------------*
* Utilities
*----------------------------------------------------------------------*/
char *
nsAppleSingleDecoder::ltoa(long n)
{
char *s;
int i, j, sign, tmp;
/* check sign and convert to positive to stringify numbers */
if ( (sign = n) < 0)
n = -n;
i = 0;
s = (char*) malloc(sizeof(char));
/* grow string as needed to add numbers from powers of 10 down till none left */
do
{
s = (char*) realloc(s, (i+1)*sizeof(char));
s[i++] = n % 10 + '0'; /* '0' or 30 is where ASCII numbers start from */
s[i] = '\0';
}
while( (n /= 10) > 0);
/* tack on minus sign if we found earlier that this was negative */
if (sign < 0)
{
s = (char*) realloc(s, (i+1)*sizeof(char));
s[i++] = '-';
}
s[i] = '\0';
/* pop numbers (and sign) off of string to push back into right direction */
for (i = 0, j = strlen(s) - 1; i < j; i++, j--)
{
tmp = s[i];
s[i] = s[j];
s[j] = tmp;
}
return s;
}
StringPtr
nsAppleSingleDecoder::PLstrncpy(StringPtr dst, ConstStr255Param src, short max)
{
int srcLen = src[0];
if (srcLen > max)
srcLen = max;
dst[0] = srcLen;
memcpy(&dst[1], &src[1], srcLen);
return dst;
}
StringPtr
nsAppleSingleDecoder::PLstrncat(StringPtr dst, ConstStr255Param src, short max)
{
int srcLen = src[0], dstLen = dst[0];
if (srcLen > max)
srcLen = max;
dst[0] += srcLen;
memcpy(&dst[dstLen+1], &src[1], srcLen);
return dst;
}
Boolean
nsAppleSingleDecoder::PLstrcmp(StringPtr str1, StringPtr str2)
{
Boolean bEqual = true;
// check for same length
if (str1[0] == str2[0])
{
// verify mem blocks match
if (0 != memcmp(&str1[1], &str2[1], str1[0]))
bEqual = false;
}
else
bEqual = false;
return bEqual;
}