gecko-dev/uriloader/exthandler/mac/nsInternetConfigService.cpp

628 строки
20 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):
*/
#include "nsInternetConfigService.h"
#include "nsCOMPtr.h"
#include "nsIMIMEInfo.h"
#include "nsIFactory.h"
#include "nsIComponentManager.h"
#include "nsLocalFileMac.h"
#include "nsIURL.h"
#include "nsXPIDLString.h"
#include "nsMimeTypes.h"
#include <TextUtils.h>
#include "nsColor.h"
#include <Processes.h>
// helper converter function.....
static void ConvertCharStringToStr255( char* inString, Str255& outString )
{
if ( inString == NULL )
return;
PRInt32 len = nsCRT::strlen(inString);
NS_ASSERTION( len<= 255 , " String is too big");
if ( len> 255 )
{
len = 255;
}
memcpy(&outString[1], inString, len);
outString[0] = len;
}
/* Define Class IDs */
static NS_DEFINE_CID(kICServiceCID, NS_INTERNETCONFIGSERVICE_CID);
nsInternetConfigService::nsInternetConfigService()
{
NS_INIT_REFCNT();
}
nsInternetConfigService::~nsInternetConfigService()
{
}
/*
* Implement the nsISupports methods...
*/
NS_IMPL_THREADSAFE_ADDREF(nsInternetConfigService)
NS_IMPL_THREADSAFE_RELEASE(nsInternetConfigService)
NS_INTERFACE_MAP_BEGIN(nsInternetConfigService)
NS_INTERFACE_MAP_ENTRY(nsIInternetConfigService)
NS_INTERFACE_MAP_ENTRY_AMBIGUOUS(nsISupports, nsIInternetConfigService)
NS_INTERFACE_MAP_END_THREADSAFE
// void LaunchURL (in string url);
// given a url string, call ICLaunchURL using url
NS_IMETHODIMP nsInternetConfigService::LaunchURL(const char *url)
{
nsresult result;
size_t len = strlen(url);
long selStart = 0, selEnd = len;
ICInstance inst = nsInternetConfig::GetInstance();
if (inst)
{
result = ::ICLaunchURL(inst, "\p", (Ptr)url, (long)len, &selStart, &selEnd);
if (result == noErr)
return NS_OK;
else
return NS_ERROR_FAILURE;
}
else
{
return NS_ERROR_FAILURE;
}
}
// boolean HasMappingForMIMEType (in string mimetype);
// given a mime type, search Internet Config database for a mapping for that mime type
NS_IMETHODIMP nsInternetConfigService::HasMappingForMIMEType(const char *mimetype, PRBool *_retval)
{
ICMapEntry entry;
nsresult rv = GetMappingForMIMEType(mimetype, nsnull, &entry);
if (rv == noErr)
*_retval = PR_TRUE;
else
*_retval = PR_FALSE;
return rv;
}
/* boolean hasProtocalHandler (in string protocol); */
// returns NS_ERROR_NOT_AVAILABLE if the current application is registered for as the
// protocol handler for protocol
NS_IMETHODIMP nsInternetConfigService::HasProtocolHandler(const char *protocol, PRBool *_retval)
{
*_retval = PR_FALSE;
// look for IC pref "\pHelper<65><protocol>"
Str255 pref = kICHelper;
if (nsCRT::strlen(protocol) > 248)
return NS_ERROR_OUT_OF_MEMORY;
nsCRT::memcpy(pref + pref[0] + 1, protocol, nsCRT::strlen(protocol));
pref[0] = pref[0] + nsCRT::strlen(protocol);
ICInstance instance = nsInternetConfig::GetInstance();
if ( !instance )
return NS_ERROR_FAILURE;
OSStatus err;
ICAttr junk;
ICAppSpec spec;
long ioSize = sizeof(ICAppSpec);
err = ::ICGetPref( instance, pref, &junk, (void *)&spec, &ioSize );
if ( err != noErr )
return NS_ERROR_FAILURE;
// check if registered protocol helper is us
// if so, return PR_FALSE because we'll go into infinite recursion
// continually launching back into ourselves
ProcessSerialNumber psn;
OSErr oserr = ::GetCurrentProcess(&psn);
if (oserr == noErr)
{
ProcessInfoRec info;
info.processInfoLength = sizeof(ProcessInfoRec);
info.processName = nsnull;
info.processAppSpec = nsnull;
err = ::GetProcessInformation(&psn, &info);
if (err == noErr)
{
if (info.processSignature != spec.fCreator)
*_retval = PR_TRUE;
else
return NS_ERROR_NOT_AVAILABLE;
}
else
return NS_ERROR_FAILURE;
}
else
return NS_ERROR_FAILURE;
return NS_OK;
}
// This method does the dirty work of traipsing through IC mappings database
// looking for a mapping for mimetype
nsresult nsInternetConfigService::GetMappingForMIMEType(const char *mimetype, const char *fileextension, ICMapEntry *entry)
{
ICInstance inst = nsInternetConfig::GetInstance();
OSStatus err = noErr;
ICAttr attr;
Handle prefH;
PRBool domimecheck = PR_TRUE, gotmatch = PR_FALSE;
ICMapEntry ent;
// if mime type is "unknown" or "octet stream" *AND* we have a file extension,
// then disable match on mime type
if (((nsCRT::strcasecmp(mimetype, UNKNOWN_CONTENT_TYPE) == 0) ||
(nsCRT::strcasecmp(mimetype, APPLICATION_OCTET_STREAM) == 0)) &&
fileextension)
domimecheck = PR_FALSE;
entry->totalLength = 0;
if (inst)
{
err = ::ICBegin(inst, icReadOnlyPerm);
if (err == noErr)
{
prefH = ::NewHandle(2048); // picked 2048 out of thin air
if (prefH)
{
err = ::ICFindPrefHandle(inst, kICMapping, &attr, prefH);
if (err == noErr)
{
long count;
err = ::ICCountMapEntries(inst, prefH, &count);
if (err == noErr)
{
long pos;
for(long i = 1; i <= count; i++)
{
err = ::ICGetIndMapEntry(inst, prefH, i, &pos, &ent);
if (err == noErr)
{
// first, do mime type check
if (domimecheck)
{
nsCAutoString temp((char *)&ent.MIMEType[1], (int)ent.MIMEType[0]);
if (!temp.EqualsIgnoreCase(mimetype))
{
// we need to do mime check, and check failed
// nothing here to see, move along
continue;
}
}
if (fileextension)
{
// if fileextension was passed in, compare that also
if (ent.extension[0]) // check for non-empty pascal string
{
nsCAutoString temp((char *)&ent.extension[1], (int)ent.extension[0]);
if (temp.EqualsIgnoreCase(fileextension))
{
// mime type and file extension match, we're outta here
gotmatch = PR_TRUE;
// copy over ICMapEntry
*entry = ent;
break;
}
}
}
else if(domimecheck)
{
// at this point, we've got our match because
// domimecheck is true, the mime strings match, and fileextension isn't passed in
// bad thing is we'll stop on first match, but what can you do?
gotmatch = PR_TRUE;
// copy over ICMapEntry
*entry = ent;
break;
}
}
}
}
}
::DisposeHandle(prefH);
}
else
{
err = memFullErr;
}
err = ::ICEnd(inst);
if (err == noErr && gotmatch == PR_FALSE)
{
err = fnfErr; // return SOME kind of error
}
}
}
if (err != noErr)
return NS_ERROR_FAILURE;
else
return NS_OK;
}
nsresult nsInternetConfigService::FillMIMEInfoForICEntry(ICMapEntry& entry, nsIMIMEInfo ** mimeinfo)
{
// create a mime info object and we'll fill it in based on the values from IC mapping entry
nsresult rv = NS_OK;
nsCOMPtr<nsIMIMEInfo> info (do_CreateInstance(NS_MIMEINFO_CONTRACTID));
if (info)
{
nsCAutoString mimetype ((char *)&entry.MIMEType[1], entry.MIMEType[0]);
// check if entry.MIMEType is empty, if so, set mime type to APPLICATION_OCTET_STREAM
if (entry.MIMEType[0])
info->SetMIMEType(mimetype.get());
else
info->SetMIMEType(APPLICATION_OCTET_STREAM);
// convert entry.extension which is a Str255
// don't forget to remove the '.' in front of the file extension....
nsCAutoString temp((char *)&entry.extension[2], (int)entry.extension[0]-1);
info->AppendExtension(temp);
info->SetMacType(entry.fileType);
info->SetMacCreator(entry.fileCreator);
temp.Assign((char *) &entry.entryName[1], entry.entryName[0]);
info->SetDescription(NS_ConvertASCIItoUCS2(temp.get()).get());
temp.Assign((char *) &entry.postAppName[1], entry.postAppName[0]);
info->SetApplicationDescription(NS_ConvertASCIItoUCS2(temp.get()).get());
if (entry.flags & kICMapPostMask)
{
// there is a post processor app
info->SetPreferredAction(nsIMIMEInfo::useSystemDefault);
nsCID cid = NS_LOCAL_FILE_CID;
nsCOMPtr<nsILocalFileMac> file (do_CreateInstance(cid));
if (file)
{
rv = file->InitFindingAppByCreatorCode(entry.postCreator);
if (rv == NS_OK)
{
//info->SetAlwaysAskBeforeHandling(PR_FALSE);
nsCOMPtr<nsIFile> nsfile = do_QueryInterface(file, &rv);
if (rv == NS_OK)
info->SetPreferredApplicationHandler(nsfile);
}
}
}
else
{
// there isn't a post processor app so set the preferred action to be save to disk.
info->SetPreferredAction(nsIMIMEInfo::saveToDisk);
}
*mimeinfo = info;
NS_IF_ADDREF(*mimeinfo);
}
else // we failed to allocate the info object...
rv = NS_ERROR_FAILURE;
return rv;
}
/* void FillInMIMEInfo (in string mimetype, in string fileExtension, out nsIMIMEInfo mimeinfo); */
NS_IMETHODIMP nsInternetConfigService::FillInMIMEInfo(const char *mimetype, const char * aFileExtension, nsIMIMEInfo **mimeinfo)
{
nsresult rv;
ICMapEntry entry;
NS_ENSURE_ARG_POINTER(mimeinfo);
*mimeinfo = nsnull;
if (aFileExtension)
{
nsCAutoString fileExtension;
fileExtension.Assign(".");
fileExtension.Append(aFileExtension);
rv = GetMappingForMIMEType(mimetype, fileExtension, &entry);
}
else
{
rv = GetMappingForMIMEType(mimetype, nsnull, &entry);
}
if (rv == NS_OK)
{
rv = FillMIMEInfoForICEntry(entry, mimeinfo);
}
else
{
rv = NS_ERROR_FAILURE;
}
return rv;
}
NS_IMETHODIMP nsInternetConfigService::GetMIMEInfoFromExtension(const char *aFileExt, nsIMIMEInfo **_retval)
{
nsresult rv = NS_ERROR_FAILURE;
ICInstance instance = nsInternetConfig::GetInstance();
if ( instance )
{
nsCAutoString filename("foobar.");
filename+=aFileExt;
Str255 pFileName;
ConvertCharStringToStr255( filename, pFileName );
ICMapEntry entry;
OSStatus err = ::ICMapFilename( instance, pFileName, &entry );
if( err == noErr )
{
rv = FillMIMEInfoForICEntry(entry, _retval);
}
}
return rv;
}
NS_IMETHODIMP nsInternetConfigService::GetMIMEInfoFromTypeCreator(PRUint32 aType, PRUint32 aCreator, const char *aFileExt, nsIMIMEInfo **_retval)
{
nsresult rv = NS_ERROR_FAILURE;
ICInstance instance = nsInternetConfig::GetInstance();
if ( instance )
{
nsCAutoString filename("foobar.");
filename+=aFileExt;
Str255 pFileName;
ConvertCharStringToStr255( filename, pFileName );
ICMapEntry entry;
OSStatus err = ::ICMapTypeCreator( instance, aType, aCreator, pFileName, &entry );
if( err == noErr )
rv = FillMIMEInfoForICEntry(entry,_retval);
}
return rv;
}
NS_IMETHODIMP nsInternetConfigService::GetFileMappingFlags(FSSpec* fsspec, PRBool lookupByExtensionFirst, PRInt32 *_retval)
{
nsresult rv = NS_ERROR_FAILURE;
OSStatus err = noErr;
NS_ENSURE_ARG(_retval);
*_retval = -1;
ICInstance instance = nsInternetConfig::GetInstance();
if ( instance )
{
ICMapEntry entry;
if (lookupByExtensionFirst)
err = ::ICMapFilename( instance, fsspec->name, &entry );
if (!lookupByExtensionFirst || err != noErr)
{
FInfo info;
err = FSpGetFInfo (fsspec, &info);
if (err == noErr)
err = ::ICMapTypeCreator( instance, info.fdType, info.fdCreator, fsspec->name, &entry );
}
if (err == noErr)
*_retval = entry.flags;
rv = NS_OK;
}
return rv;
}
/* void GetDownloadFolder (out FSSpec fsspec); */
NS_IMETHODIMP nsInternetConfigService::GetDownloadFolder(FSSpec *fsspec)
{
ICInstance inst = nsInternetConfig::GetInstance();
OSStatus err;
Handle prefH;
nsresult rv = NS_ERROR_FAILURE;
NS_ENSURE_ARG_POINTER(fsspec);
if (inst)
{
err = ::ICBegin(inst, icReadOnlyPerm);
if (err == noErr)
{
prefH = ::NewHandle(256); // ICFileSpec ~= 112 bytes + variable, 256 bytes hopefully is sufficient
if (prefH)
{
ICAttr attr;
err = ::ICFindPrefHandle(inst, kICDownloadFolder, &attr, prefH);
if (err == noErr)
{
err = ::Munger(prefH, 0, NULL, kICFileSpecHeaderSize, (Ptr)-1, 0);
if (err == noErr)
{
Boolean wasChanged;
err = ::ResolveAlias(NULL, (AliasHandle)prefH, fsspec, &wasChanged);
if (err == noErr)
{
rv = NS_OK;
}
}
}
}
err = ::ICEnd(inst);
}
}
return rv;
}
nsresult nsInternetConfigService::GetICKeyPascalString(PRUint32 inIndex, unsigned char **outICKey)
{
nsresult result = NS_OK;
switch (inIndex)
{
case eICColor_WebBackgroundColour: *outICKey = kICWebBackgroundColour; break;
case eICColor_WebReadColor: *outICKey = kICWebReadColor; break;
case eICColor_WebTextColor: *outICKey = kICWebTextColor; break;
case eICColor_WebUnreadColor: *outICKey = kICWebUnreadColor; break;
case eICBoolean_WebUnderlineLinks: *outICKey = kICWebUnderlineLinks; break;
case eICBoolean_UseFTPProxy: *outICKey = kICUseFTPProxy; break;
case eICBoolean_UsePassiveFTP: *outICKey = kICUsePassiveFTP; break;
case eICBoolean_UseHTTPProxy: *outICKey = kICUseHTTPProxy; break;
case eICBoolean_NewMailDialog: *outICKey = kICNewMailDialog; break;
case eICBoolean_NewMailFlashIcon: *outICKey = kICNewMailFlashIcon; break;
case eICBoolean_NewMailPlaySound: *outICKey = kICNewMailPlaySound; break;
case eICBoolean_UseGopherProxy: *outICKey = kICUseGopherProxy; break;
case eICBoolean_UseSocks: *outICKey = kICUseSocks; break;
case eICString_WWWHomePage: *outICKey = kICWWWHomePage; break;
case eICString_WebSearchPagePrefs: *outICKey = kICWebSearchPagePrefs; break;
case eICString_MacSearchHost: *outICKey = kICMacSearchHost; break;
case eICString_FTPHost: *outICKey = kICFTPHost; break;
case eICString_FTPProxyUser: *outICKey = kICFTPProxyUser; break;
case eICString_FTPProxyAccount: *outICKey = kICFTPProxyAccount; break;
case eICString_FTPProxyHost: *outICKey = kICFTPProxyHost; break;
case eICString_FTPProxyPassword: *outICKey = kICWWWHomePage; break;
case eICString_HTTPProxyHost: *outICKey = kICHTTPProxyHost; break;
case eICString_LDAPSearchbase: *outICKey = kICLDAPSearchbase; break;
case eICString_LDAPServer: *outICKey = kICLDAPServer; break;
case eICString_SMTPHost: *outICKey = kICSMTPHost; break;
case eICString_Email: *outICKey = kICEmail; break;
case eICString_MailAccount: *outICKey = kICMailAccount; break;
case eICString_MailPassword: *outICKey = kICMailPassword; break;
case eICString_NewMailSoundName: *outICKey = kICNewMailSoundName; break;
case eICString_NNTPHost: *outICKey = kICNNTPHost; break;
case eICString_NewsAuthUsername: *outICKey = kICNewsAuthUsername; break;
case eICString_NewsAuthPassword: *outICKey = kICNewsAuthPassword; break;
case eICString_InfoMacPreferred: *outICKey = kICInfoMacPreferred; break;
case eICString_Organization: *outICKey = kICOrganization; break;
case eICString_QuotingString: *outICKey = kICQuotingString; break;
case eICString_RealName: *outICKey = kICRealName; break;
case eICString_FingerHost: *outICKey = kICFingerHost; break;
case eICString_GopherHost: *outICKey = kICGopherHost; break;
case eICString_GopherProxy: *outICKey = kICGopherProxy; break;
case eICString_SocksHost: *outICKey = kICSocksHost; break;
case eICString_TelnetHost: *outICKey = kICTelnetHost; break;
case eICString_IRCHost: *outICKey = kICIRCHost; break;
case eICString_UMichPreferred: *outICKey = kICUMichPreferred; break;
case eICString_WAISGateway: *outICKey = kICWAISGateway; break;
case eICString_WhoisHost: *outICKey = kICWhoisHost; break;
case eICString_PhHost: *outICKey = kICPhHost; break;
case eICString_NTPHost: *outICKey = kICNTPHost; break;
case eICString_ArchiePreferred: *outICKey = kICArchiePreferred; break;
case eICText_MailHeaders: *outICKey = kICMailHeaders; break;
case eICText_Signature: *outICKey = kICSignature; break;
case eICText_NewsHeaders: *outICKey = kICNewsHeaders; break;
case eICText_SnailMailAddress: *outICKey = kICSnailMailAddress; break;
case eICText_Plan: *outICKey = kICPlan; break;
default:
result = NS_ERROR_INVALID_ARG;
}
return result;
}
nsresult nsInternetConfigService::GetICPreference(PRUint32 inKey,
void *outData, long *ioSize)
{
unsigned char *icKey;
nsresult result = GetICKeyPascalString( inKey, &icKey );
if (NS_FAILED(result))
return result;
ICInstance instance = nsInternetConfig::GetInstance();
if ( !instance )
return NS_ERROR_FAILURE;
OSStatus err;
ICAttr junk;
err = ::ICGetPref( instance, icKey, &junk, outData, ioSize );
if ( err )
return NS_ERROR_UNEXPECTED;
return NS_OK;
}
NS_IMETHODIMP nsInternetConfigService::GetString(PRUint32 inKey, char **value)
{
long size = 256;
char buffer[256];
nsresult result = GetICPreference( inKey, (void *)&buffer, &size );
if ( result == NS_OK ) {
if ( size == 0 ) {
*value = nsnull;
return NS_ERROR_UNEXPECTED;
}
// Buffer is a Pascal string; convert it to a c-string
nsCString temp( &buffer[1], buffer[0] );
*value = temp.ToNewCString();
}
return result;
}
NS_IMETHODIMP nsInternetConfigService::GetColor(PRUint32 inKey, PRUint32 *outColor)
{
RGBColor buffer;
long size = sizeof(RGBColor);
nsresult result = GetICPreference( inKey, &buffer, &size );
if ( result == NS_OK ) {
if ( size != sizeof(RGBColor) ) {
*outColor = NS_RGB(0xff, 0xff, 0xff); // default to white if we didn't get the right size
return NS_OK;
}
// convert to a web color
*outColor = NS_RGB( buffer.red>>8, buffer.green>>8, buffer.blue>>8 );
}
return result;
}
NS_IMETHODIMP nsInternetConfigService::GetBoolean(PRUint32 inKey, PRBool *outFlag)
{
Boolean buffer;
long size = sizeof(Boolean);
nsresult result = GetICPreference( inKey, (void *)&buffer, &size );
if ( result == NS_OK ) {
if ( size < sizeof(Boolean) ) {
*outFlag = PR_FALSE; // default to false if we didn't get the right amount of data
return NS_OK;
}
*outFlag = buffer;
result = NS_OK;
}
return result;
}
NS_IMETHODIMP nsInternetConfigService::GetText(PRUint32 inKey, PRUint32 *ioLength, PRUnichar **outText)
{
return NS_ERROR_NOT_IMPLEMENTED;
}