зеркало из https://github.com/mozilla/pjs.git
429 строки
12 KiB
C++
429 строки
12 KiB
C++
|
/* -*- Mode: C++; tab-width: 2; 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.
|
||
|
*
|
||
|
* The Initial Developer of the Original Code is Netscape Communications
|
||
|
* Corporation. Portions created by Netscape are
|
||
|
* Copyright (C) 1998 Netscape Communications Corporation. All
|
||
|
* Rights Reserved.
|
||
|
*
|
||
|
* Contributor(s):
|
||
|
* Don Bragg <dbragg@netscape.com>
|
||
|
* Samir Gehani <sgehani@netscape.com>
|
||
|
*/
|
||
|
|
||
|
#include <stdio.h>
|
||
|
#include <string.h>
|
||
|
#include <unistd.h>
|
||
|
#include <MacTypes.h>
|
||
|
#include "MoreFiles.h"
|
||
|
#include "MoreFilesExtras.h"
|
||
|
#include "FullPath.h"
|
||
|
#include <AppleEvents.h>
|
||
|
#include <Gestalt.h>
|
||
|
#include <TextUtils.h>
|
||
|
#include <Folders.h>
|
||
|
#include <Processes.h>
|
||
|
#include <Resources.h>
|
||
|
#include <Aliases.h>
|
||
|
|
||
|
#include "InstallCleanup.h"
|
||
|
|
||
|
#define kSleepMax 900 // sleep 15 seconds
|
||
|
|
||
|
Boolean gAppleEventsFlag, gQuitFlag;
|
||
|
long gSleepVal;
|
||
|
|
||
|
|
||
|
int strcasecmp(const char *str1, const char *str2);
|
||
|
OSErr GetFSSpecFromPath(char *aPath, FSSpec *aSpec);
|
||
|
|
||
|
int strcasecmp(const char *str1, const char *str2)
|
||
|
{
|
||
|
char currentChar1, currentChar2;
|
||
|
|
||
|
while (1) {
|
||
|
|
||
|
currentChar1 = *str1;
|
||
|
currentChar2 = *str2;
|
||
|
|
||
|
if ((currentChar1 >= 'a') && (currentChar1 <= 'z'))
|
||
|
currentChar1 += ('A' - 'a');
|
||
|
|
||
|
if ((currentChar2 >= 'a') && (currentChar2 <= 'z'))
|
||
|
currentChar2 += ('A' - 'a');
|
||
|
|
||
|
if (currentChar1 == '\0')
|
||
|
break;
|
||
|
|
||
|
if (currentChar1 != currentChar2)
|
||
|
return currentChar1 - currentChar2;
|
||
|
|
||
|
str1++;
|
||
|
str2++;
|
||
|
|
||
|
}
|
||
|
|
||
|
return currentChar1 - currentChar2;
|
||
|
}
|
||
|
|
||
|
OSErr GetFSSpecFromPath(const char *aPath, FSSpec *aSpec)
|
||
|
{
|
||
|
if (!aPath || !aSpec)
|
||
|
return paramErr;
|
||
|
|
||
|
// 1> verify path is not an empty string
|
||
|
// 2> verify path has leaf
|
||
|
// 3> verify path does not start with leading ':'
|
||
|
|
||
|
if ((!*aPath) ||
|
||
|
(*(aPath + strlen(aPath) - 1) == ':') ||
|
||
|
(*aPath == ':'))
|
||
|
{
|
||
|
return paramErr;
|
||
|
}
|
||
|
|
||
|
// path is kosher: get FSSpec for it
|
||
|
return FSpLocationFromFullPath(strlen(aPath), (const void *) aPath, aSpec);
|
||
|
}
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// Native Mac file deletion function
|
||
|
//----------------------------------------------------------------------------
|
||
|
int NativeDeleteFile(const char* aFileToDelete)
|
||
|
{
|
||
|
OSErr err;
|
||
|
FSSpec delSpec;
|
||
|
|
||
|
if (!aFileToDelete)
|
||
|
return DONE;
|
||
|
|
||
|
// stat the file
|
||
|
err = GetFSSpecFromPath(aFileToDelete, &delSpec);
|
||
|
if (err != noErr)
|
||
|
{
|
||
|
// return fine if it doesn't exist
|
||
|
return DONE;
|
||
|
}
|
||
|
|
||
|
// else try to delete it
|
||
|
err = FSpDelete(&delSpec);
|
||
|
if (err != noErr)
|
||
|
{
|
||
|
// tell user to try again later if deletion failed
|
||
|
return TRY_LATER;
|
||
|
}
|
||
|
|
||
|
return DONE;
|
||
|
}
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// Native Mac file replacement function
|
||
|
//----------------------------------------------------------------------------
|
||
|
int NativeReplaceFile(const char* aReplacementFile, const char* aDoomedFile )
|
||
|
{
|
||
|
OSErr err;
|
||
|
FSSpec replSpec, doomSpec, tgtDirSpec;
|
||
|
long dirID;
|
||
|
Boolean isDir;
|
||
|
|
||
|
if (!aReplacementFile || !aDoomedFile)
|
||
|
return DONE;
|
||
|
|
||
|
err = GetFSSpecFromPath(aReplacementFile, &replSpec);
|
||
|
if (err != noErr)
|
||
|
return DONE;
|
||
|
|
||
|
// stat replacement file
|
||
|
err = FSpGetDirectoryID(&replSpec, &dirID, &isDir);
|
||
|
if (err != noErr || isDir)
|
||
|
{
|
||
|
// return fine if it doesn't exist
|
||
|
return DONE;
|
||
|
}
|
||
|
|
||
|
// check if the replacement file and doomed file are the same
|
||
|
if (strcasecmp(aReplacementFile, aDoomedFile) == 0)
|
||
|
{
|
||
|
// return fine if they are the same
|
||
|
return DONE;
|
||
|
}
|
||
|
|
||
|
// try and delete doomed file (NOTE: this call also stats)
|
||
|
err = GetFSSpecFromPath(aDoomedFile, &doomSpec);
|
||
|
if (err == noErr)
|
||
|
{
|
||
|
// (will even try to delete a dir with this name)
|
||
|
err = FSpDelete(&doomSpec);
|
||
|
|
||
|
// if deletion failed tell user to try again later
|
||
|
if (err != noErr)
|
||
|
return TRY_LATER;
|
||
|
}
|
||
|
|
||
|
// get the target dir spec (parent directory of doomed file)
|
||
|
err = FSMakeFSSpec(doomSpec.vRefNum, doomSpec.parID, "\p", &tgtDirSpec);
|
||
|
if (err == noErr)
|
||
|
{
|
||
|
// now try and move replacment file to path of doomed file
|
||
|
err = FSpMoveRename(&replSpec, &tgtDirSpec, doomSpec.name);
|
||
|
if (err != noErr)
|
||
|
{
|
||
|
// if move failed tell user to try agian later
|
||
|
return TRY_LATER;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return DONE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// Routines for recovery on reboot
|
||
|
//----------------------------------------------------------------------------
|
||
|
OSErr
|
||
|
GetProgramSpec(FSSpecPtr aProgSpec)
|
||
|
{
|
||
|
OSErr err = noErr;
|
||
|
ProcessSerialNumber psn;
|
||
|
ProcessInfoRec pInfo;
|
||
|
|
||
|
if (!aProgSpec)
|
||
|
return paramErr;
|
||
|
|
||
|
/* get cwd based on curr ps info */
|
||
|
if (!(err = GetCurrentProcess(&psn)))
|
||
|
{
|
||
|
pInfo.processName = nil;
|
||
|
pInfo.processAppSpec = aProgSpec;
|
||
|
pInfo.processInfoLength = (sizeof(ProcessInfoRec));
|
||
|
|
||
|
err = GetProcessInformation(&psn, &pInfo);
|
||
|
}
|
||
|
|
||
|
return err;
|
||
|
}
|
||
|
|
||
|
void
|
||
|
PutAliasInStartupItems(FSSpecPtr aAlias)
|
||
|
{
|
||
|
OSErr err;
|
||
|
FSSpec fsProg, fsAlias;
|
||
|
long strtDirID = 0;
|
||
|
short strtVRefNum = 0;
|
||
|
FInfo info;
|
||
|
AliasHandle aliasH;
|
||
|
|
||
|
if (!aAlias)
|
||
|
return;
|
||
|
|
||
|
// find cwd
|
||
|
err = GetProgramSpec(&fsProg);
|
||
|
if (err != noErr)
|
||
|
return; // fail silently
|
||
|
|
||
|
// get startup items folder
|
||
|
err = FindFolder(kOnSystemDisk, kStartupFolderType, kCreateFolder,
|
||
|
&strtVRefNum, &strtDirID);
|
||
|
if (err != noErr)
|
||
|
return;
|
||
|
|
||
|
// check that an alias to ourselves doesn't already
|
||
|
// exist in the Startup Items folder
|
||
|
err = FSMakeFSSpec(strtVRefNum, strtDirID, fsProg.name, &fsAlias);
|
||
|
if (err == noErr)
|
||
|
{
|
||
|
// one's already there; not sure it's us so delete and recreate
|
||
|
// (being super paranoid; but hey it's a mac)
|
||
|
err = FSpDelete(&fsAlias);
|
||
|
if (err != noErr)
|
||
|
return; // fail silently
|
||
|
}
|
||
|
|
||
|
// create the alias file
|
||
|
err = NewAliasMinimal(&fsProg, &aliasH);
|
||
|
if (err != noErr)
|
||
|
return;
|
||
|
|
||
|
FSpGetFInfo(&fsProg, &info);
|
||
|
FSpCreateResFile(&fsAlias, info.fdCreator, info.fdType, smRoman);
|
||
|
short refNum = FSpOpenResFile(&fsAlias, fsRdWrPerm);
|
||
|
if (refNum != -1)
|
||
|
{
|
||
|
UseResFile(refNum);
|
||
|
AddResource((Handle)aliasH, rAliasType, 0, fsAlias.name);
|
||
|
ReleaseResource((Handle)aliasH);
|
||
|
UpdateResFile(refNum);
|
||
|
CloseResFile(refNum);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
ReleaseResource((Handle)aliasH);
|
||
|
FSpDelete(&fsAlias);
|
||
|
return; // non-fatal error
|
||
|
}
|
||
|
|
||
|
// mark newly created file as an alias file
|
||
|
FSpGetFInfo(&fsAlias, &info);
|
||
|
info.fdFlags |= kIsAlias;
|
||
|
FSpSetFInfo(&fsAlias, &info);
|
||
|
|
||
|
*aAlias = fsAlias;
|
||
|
}
|
||
|
|
||
|
void
|
||
|
RemoveAliasFromStartupItems(FSSpecPtr aAlias)
|
||
|
{
|
||
|
// try to delete the alias
|
||
|
FSpDelete(aAlias);
|
||
|
}
|
||
|
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// Apple event handlers to be installed
|
||
|
//----------------------------------------------------------------------------
|
||
|
|
||
|
static pascal OSErr DoAEOpenApplication(AppleEvent * theAppleEvent, AppleEvent * replyAppleEvent, long refCon)
|
||
|
{
|
||
|
#pragma unused (theAppleEvent, replyAppleEvent, refCon)
|
||
|
return noErr;
|
||
|
}
|
||
|
|
||
|
static pascal OSErr DoAEOpenDocuments(AppleEvent * theAppleEvent, AppleEvent * replyAppleEvent, long refCon)
|
||
|
{
|
||
|
#pragma unused (theAppleEvent, replyAppleEvent, refCon)
|
||
|
return errAEEventNotHandled;
|
||
|
}
|
||
|
|
||
|
static pascal OSErr DoAEPrintDocuments(AppleEvent * theAppleEvent, AppleEvent * replyAppleEvent, long refCon)
|
||
|
{
|
||
|
#pragma unused (theAppleEvent, replyAppleEvent, refCon)
|
||
|
return errAEEventNotHandled;
|
||
|
}
|
||
|
|
||
|
static pascal OSErr DoAEQuitApplication(AppleEvent * theAppleEvent, AppleEvent * replyAppleEvent, long refCon)
|
||
|
{
|
||
|
#pragma unused (theAppleEvent, replyAppleEvent, refCon)
|
||
|
gQuitFlag = true;
|
||
|
return noErr;
|
||
|
}
|
||
|
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// install Apple event handlers
|
||
|
//----------------------------------------------------------------------------
|
||
|
|
||
|
static void InitAppleEventsStuff(void)
|
||
|
{
|
||
|
OSErr retCode;
|
||
|
|
||
|
if (gAppleEventsFlag) {
|
||
|
|
||
|
retCode = AEInstallEventHandler(kCoreEventClass, kAEOpenApplication,
|
||
|
NewAEEventHandlerProc(DoAEOpenApplication), 0, false);
|
||
|
|
||
|
if (retCode == noErr)
|
||
|
retCode = AEInstallEventHandler(kCoreEventClass, kAEOpenDocuments,
|
||
|
NewAEEventHandlerProc(DoAEOpenDocuments), 0, false);
|
||
|
|
||
|
if (retCode == noErr)
|
||
|
retCode = AEInstallEventHandler(kCoreEventClass, kAEPrintDocuments,
|
||
|
NewAEEventHandlerProc(DoAEPrintDocuments), 0, false);
|
||
|
if (retCode == noErr)
|
||
|
retCode = AEInstallEventHandler(kCoreEventClass, kAEQuitApplication,
|
||
|
NewAEEventHandlerProc(DoAEQuitApplication), 0, false);
|
||
|
|
||
|
if (retCode != noErr) DebugStr("\pInstall event handler failed");
|
||
|
// a better way to indicate an error is to post a notification
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// high-level event dispatching
|
||
|
//----------------------------------------------------------------------------
|
||
|
|
||
|
static void DoHighLevelEvent(EventRecord * theEventRecPtr)
|
||
|
{
|
||
|
(void) AEProcessAppleEvent(theEventRecPtr);
|
||
|
}
|
||
|
|
||
|
void main(void)
|
||
|
{
|
||
|
OSErr retCode;
|
||
|
long gestResponse;
|
||
|
FSSpec aliasToSelf;
|
||
|
|
||
|
EventRecord mainEventRec;
|
||
|
Boolean eventFlag, bDone = false;;
|
||
|
|
||
|
HREG reg;
|
||
|
int rv = DONE;
|
||
|
|
||
|
// initialize QuickDraw globals
|
||
|
|
||
|
InitGraf(&qd.thePort);
|
||
|
|
||
|
// initialize application globals
|
||
|
|
||
|
gQuitFlag = false;
|
||
|
gSleepVal = kSleepMax;
|
||
|
|
||
|
// is the Apple Event Manager available?
|
||
|
retCode = Gestalt(gestaltAppleEventsAttr, &gestResponse);
|
||
|
if (retCode == noErr &&
|
||
|
(gestResponse & (1 << gestaltAppleEventsPresent)) != 0)
|
||
|
gAppleEventsFlag = true;
|
||
|
else gAppleEventsFlag = false;
|
||
|
|
||
|
// install Apple event handlers
|
||
|
InitAppleEventsStuff();
|
||
|
|
||
|
// put an alias to ourselves in the Startup Items folder
|
||
|
// so that if we are shutdown before finishing we do our
|
||
|
// tasks at startup
|
||
|
FSMakeFSSpec(0, 0, "\p", &aliasToSelf); // initialize
|
||
|
PutAliasInStartupItems(&aliasToSelf);
|
||
|
|
||
|
if ( REGERR_OK == NR_StartupRegistry())
|
||
|
{
|
||
|
if ( REGERR_OK == NR_RegOpen("", ®) )
|
||
|
{
|
||
|
// main event loop
|
||
|
|
||
|
while (!gQuitFlag)
|
||
|
{
|
||
|
eventFlag = WaitNextEvent(everyEvent, &mainEventRec, gSleepVal, nil);
|
||
|
|
||
|
if (mainEventRec.what == kHighLevelEvent)
|
||
|
DoHighLevelEvent(&mainEventRec);
|
||
|
|
||
|
rv = PerformScheduledTasks(reg);
|
||
|
if (rv == DONE)
|
||
|
{
|
||
|
bDone = true;
|
||
|
gQuitFlag = true;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
NR_ShutdownRegistry();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// clean up the alias to ouselves since we have
|
||
|
// completed our tasks successfully
|
||
|
if (bDone)
|
||
|
RemoveAliasFromStartupItems(&aliasToSelf);
|
||
|
}
|
||
|
|