/* -*- 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 #include "nsColor.h" #include // 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¥" 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 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 file (do_CreateInstance(cid)); if (file) { rv = file->InitFindingAppByCreatorCode(entry.postCreator); if (rv == NS_OK) { //info->SetAlwaysAskBeforeHandling(PR_FALSE); nsCOMPtr 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; }