root/trunk/llmozlib2/llembeddedbrowser.cpp

/* ***** BEGIN LICENSE BLOCK *****
 * Version: MPL 1.1/GPL 2.0/LGPL 2.1
 *
 * The contents of this file are subject to the Mozilla 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/MPL/
 *
 * 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 Linden Lab Inc. (http://lindenlab.com) code.
 *
 * The Initial Developer of the Original Code is:
 *   Callum Prentice (callum@ubrowser.com)
 *
 * Portions created by the Initial Developer are Copyright (C) 2006
 * the Initial Developer. All Rights Reserved.
 *
 * Contributor(s):
 *  Callum Prentice (callum@ubrowser.com)
 *
 * Alternatively, the contents of this file may be used under the terms of
 * either the GNU General Public License Version 2 or later (the "GPL"), or
 * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
 * in which case the provisions of the GPL or the LGPL are applicable instead
 * of those above. If you wish to allow use of your version of this file only
 * under the terms of either the GPL or the LGPL, and not to allow others to
 * use your version of this file under the terms of the MPL, indicate your
 * decision by deleting the provisions above and replace them with the notice
 * and other provisions required by the GPL or the LGPL. If you do not delete
 * the provisions above, a recipient may use your version of this file under
 * the terms of any one of the MPL, the GPL or the LGPL.
 *
 * ***** END LICENSE BLOCK ***** */

// Windows specific switches
#ifdef WIN32
        // appears to be required by LibXUL/Mozilla code to avoid crashes in debug versions of their code (undef'd at end of this file)
        #ifdef _DEBUG
                #define DEBUG 1
        #endif
#endif  // WIN32

#include <iostream>

#include "llembeddedbrowser.h"
#include "llembeddedbrowserwindow.h"

#ifdef WIN32
        #pragma warning( disable : 4265 )       // "class has virtual functions, but destructor is not virtual"
        #pragma warning( disable : 4291 )       // (no matching operator delete found; memory will not be freed if initialization throws an exception)
#endif  // WIN32

#include "nsBuildID.h"
#include "nsICacheService.h"
#include "nsICookieManager.h"
#include "nsIPref.h"
#include "nsNetCID.h"
#include "nsProfileDirServiceProvider.h"
#include "nsXULAppAPI.h"
#include "nsIAppShell.h"
#include "nsIPromptService.h"
#include "time.h"
#include "nsWidgetsCID.h"
#include "nsNetCID.h"

static nsIAppShell *sAppShell = nsnull;

// singleton pattern - initialization
LLEmbeddedBrowser* LLEmbeddedBrowser::sInstance = 0;

////////////////////////////////////////////////////////////////////////////////
//
LLEmbeddedBrowser::LLEmbeddedBrowser() :
        mErrorNum( 0 ),
        mNativeWindowHandle( 0 )
{
}

////////////////////////////////////////////////////////////////////////////////
//
LLEmbeddedBrowser::~LLEmbeddedBrowser()
{
}

////////////////////////////////////////////////////////////////////////////////
//
LLEmbeddedBrowser* LLEmbeddedBrowser::getInstance()
{
    if ( ! sInstance )
    {
        sInstance = new LLEmbeddedBrowser;
    };

        return sInstance;
};

////////////////////////////////////////////////////////////////////////////////
//
void LLEmbeddedBrowser::setLastError( int errorNumIn )
{
        mErrorNum = errorNumIn;
}

////////////////////////////////////////////////////////////////////////////////
//
void LLEmbeddedBrowser::clearLastError()
{
        mErrorNum = 0x0000;
}

////////////////////////////////////////////////////////////////////////////////
//
int LLEmbeddedBrowser::getLastError()
{
        return mErrorNum;
}

////////////////////////////////////////////////////////////////////////////////
//
std::string LLEmbeddedBrowser::getGREVersion()
{
        // take the string directly from Mozilla
        return std::string( GRE_BUILD_ID );
}

////////////////////////////////////////////////////////////////////////////////
//
bool LLEmbeddedBrowser::init( std::string applicationDir,
                                                                std::string componentDir,
                                                                        std::string profileDir,
                                                                                void* nativeWindowHandleIn )
{
        mNativeWindowHandle = nativeWindowHandleIn;

        nsCOMPtr< nsILocalFile > applicationDirNative;
        nsresult result = NS_NewNativeLocalFile( nsCString( applicationDir.c_str() ), PR_FALSE, getter_AddRefs( applicationDirNative ) );
        if ( NS_FAILED( result ) )
        {
                setLastError( 0x1000 );
                return false;
        };

        nsCOMPtr< nsILocalFile > componentDirNative;
        result = NS_NewNativeLocalFile( nsCString( componentDir.c_str() ), PR_FALSE, getter_AddRefs( componentDirNative ) );
        if ( NS_FAILED( result ) )
        {
                setLastError( 0x1001 );
                return false;
        };

        result = XRE_InitEmbedding( componentDirNative, applicationDirNative, nsnull, nsnull, 0 );
        if ( NS_FAILED( result ) )
        {
                setLastError( 0x1002 );
                return false;
        };

        nsCOMPtr< nsILocalFile > profileDirNative;
        result = NS_NewNativeLocalFile( nsCString( profileDir.c_str() ), PR_TRUE, getter_AddRefs( profileDirNative ) );
        if ( NS_FAILED( result ) )
        {
                setLastError( 0x1007 );
                return false;
        };
        nsCOMPtr< nsProfileDirServiceProvider > locProvider;
        NS_NewProfileDirServiceProvider( PR_TRUE, getter_AddRefs( locProvider ) );
        if ( ! locProvider )
        {
                setLastError( 0x1003 );
                XRE_TermEmbedding();
                return PR_FALSE;
        };

        result = locProvider->Register();
        if ( NS_FAILED( result ) )
        {
                setLastError( 0x1004 );
                XRE_TermEmbedding();
                return PR_FALSE;
        };

        result = locProvider->SetProfileDir( profileDirNative );
        if ( NS_FAILED( result ) )
        {
                setLastError( 0x1005 );
                XRE_TermEmbedding();
                return PR_FALSE;
        };

        nsCOMPtr<nsIPref> pref = do_CreateInstance( NS_PREF_CONTRACTID );
        if ( pref )
        {
                pref->SetBoolPref( "security.warn_entering_secure", PR_FALSE );
                pref->SetBoolPref( "security.warn_entering_weak", PR_FALSE );
                pref->SetBoolPref( "security.warn_leaving_secure", PR_FALSE );
                pref->SetBoolPref( "security.warn_submit_insecure", PR_FALSE );
                pref->SetBoolPref( "network.protocol-handler.warn-external-default", PR_FALSE );
        }
        else
        {
                setLastError( 0x1006 );
        };

        // disable proxy by default
        enableProxy( false, "", 0 );

    // Originally from Linux version but seems to help other platforms too
        nsresult rv;
        nsCOMPtr<nsIAppShell> appShell;
        NS_DEFINE_CID(kAppShellCID, NS_APPSHELL_CID);
        appShell = do_CreateInstance(kAppShellCID, &rv);
        if (!appShell)
        {
                setLastError( 0x1008 );
                return PR_FALSE;
        }
        sAppShell = appShell.get();
        NS_ADDREF(sAppShell);
        sAppShell->Create(0, nsnull);
        sAppShell->Spinup();

        clearLastError();

        return true;
}

////////////////////////////////////////////////////////////////////////////////
//
bool LLEmbeddedBrowser::reset()
{
        XRE_TermEmbedding();

        return true;
}

////////////////////////////////////////////////////////////////////////////////
//
bool LLEmbeddedBrowser::clearCache()
{
        nsCOMPtr< nsICacheService > cacheService = do_GetService( NS_CACHESERVICE_CONTRACTID );
        if (! cacheService)
                return false;

        cacheService->EvictEntries( nsICache::STORE_ANYWHERE );

        return true;
}

////////////////////////////////////////////////////////////////////////////////
//
bool LLEmbeddedBrowser::enableProxy( bool proxyEnabledIn, std::string proxyHostNameIn, int proxyPortIn )
{
        nsCOMPtr<nsIPref> pref = do_CreateInstance( NS_PREF_CONTRACTID );
        if ( pref )
        {
                if ( proxyEnabledIn )
                        pref->SetIntPref( "network.proxy.type", 1 );
                else
                        pref->SetIntPref( "network.proxy.type", 0 );

                pref->SetCharPref( "network.proxy.ssl", proxyHostNameIn.c_str() );
                pref->SetIntPref( "network.proxy.ssl_port", proxyPortIn );

                pref->SetCharPref( "network.proxy.ftp", proxyHostNameIn.c_str() );
                pref->SetIntPref( "network.proxy.ftp_port", proxyPortIn );

                pref->SetCharPref( "network.proxy.gopher", proxyHostNameIn.c_str() );
                pref->SetIntPref( "network.proxy.gopher_port", proxyPortIn );

                pref->SetCharPref( "network.proxy.http", proxyHostNameIn.c_str() );
                pref->SetIntPref( "network.proxy.http_port", proxyPortIn );

                pref->SetBoolPref( "network.proxy.share_proxy_settings", true );

                return true;
        };

        return false;
}

////////////////////////////////////////////////////////////////////////////////
//
bool LLEmbeddedBrowser::enableCookies( bool enabledIn )
{
        nsCOMPtr<nsIPref> pref = do_CreateInstance( NS_PREF_CONTRACTID );
        if ( pref )
        {
                if ( enabledIn )
                        pref->SetIntPref( "network.cookie.cookieBehavior", 0 );
                else
                        pref->SetIntPref( "network.cookie.cookieBehavior", 2 );

                return true;
        }

        return false;
}

////////////////////////////////////////////////////////////////////////////////
//
bool LLEmbeddedBrowser::clearAllCookies()
{
        nsCOMPtr< nsICookieManager > cookieManager = do_GetService( NS_COOKIEMANAGER_CONTRACTID );
        if ( ! cookieManager )
                return false;

        cookieManager->RemoveAll();

        return true;
}

////////////////////////////////////////////////////////////////////////////////
//
bool LLEmbeddedBrowser::enablePlugins( bool enabledIn )
{
        nsCOMPtr<nsIPref> pref = do_CreateInstance( NS_PREF_CONTRACTID );
        if ( pref )
        {
                if ( enabledIn )
                {
                        pref->SetBoolPref( "plugin.scan.plid.all", PR_TRUE );
                        pref->SetBoolPref( "xpinstall-enabled", PR_TRUE );
                }
                else
                {
                        pref->SetBoolPref( "plugin.scan.plid.all", PR_FALSE );
                        pref->SetBoolPref( "xpinstall-enabled", PR_FALSE );
                        pref->SetBoolPref( "plugin.scan.4xPluginFolder", PR_FALSE );
                        pref->SetCharPref( "plugin.scan.Quicktime", "20.0" );
                        pref->SetCharPref( "plugin.scan.Acrobat", "99.0" );
                        pref->SetCharPref( "plugin.scan.SunJRE", "99.0" );
                        pref->SetCharPref( "plugin.scan.WindowsMediaPlayer", "99.0" );
                };

                return true;
        };

        return false;
}

////////////////////////////////////////////////////////////////////////////////
//
void LLEmbeddedBrowser::setBrowserAgentId( std::string idIn )
{
        nsCOMPtr<nsIPref> pref = do_CreateInstance( NS_PREF_CONTRACTID );
        if ( pref )
        {
                pref->SetCharPref( "general.useragent.extra.* ", idIn.c_str() );
        };
}

////////////////////////////////////////////////////////////////////////////////
//
LLEmbeddedBrowserWindow* LLEmbeddedBrowser::createBrowserWindow( int browserWidthIn, int browserHeightIn )
{
    nsCOMPtr< nsIWebBrowserChrome > chrome;

        LLEmbeddedBrowserWindow* newWin = new LLEmbeddedBrowserWindow();
        if ( ! newWin )
        {
                return 0;
        };

        nsIWebBrowserChrome** aNewWindow = getter_AddRefs( chrome );

        CallQueryInterface( NS_STATIC_CAST( nsIWebBrowserChrome*, newWin ), aNewWindow );

        NS_ADDREF( *aNewWindow );

        newWin->SetChromeFlags( nsIWebBrowserChrome::CHROME_ALL );

        nsCOMPtr< nsIWebBrowser > newBrowser;

        newWin->createBrowser( mNativeWindowHandle, browserWidthIn, browserHeightIn, getter_AddRefs( newBrowser ) );
        if ( ! newBrowser )
        {
                return 0;
        };

    if ( newWin && chrome )
    {
                newWin->setParent( this );
        nsCOMPtr< nsIWebBrowser > newBrowser;
        chrome->GetWebBrowser( getter_AddRefs( newBrowser ) );
        nsCOMPtr< nsIWebNavigation > webNav( do_QueryInterface ( newBrowser ) );
                webNav->LoadURI( NS_ConvertUTF8toUTF16( "about:blank" ).get(), nsIWebNavigation::LOAD_FLAGS_NONE, nsnull, nsnull, nsnull );

                clearLastError();

                return newWin;
    };

        setLastError( 0x2001 );
        return 0;
};


////////////////////////////////////////////////////////////////////////////////
//
bool LLEmbeddedBrowser::destroyBrowserWindow( LLEmbeddedBrowserWindow* browserWindowIn )
{
        nsCOMPtr< nsIWebBrowser > webBrowser;
        nsCOMPtr< nsIWebNavigation > webNavigation;

        browserWindowIn->GetWebBrowser( getter_AddRefs( webBrowser ) );
        webNavigation = do_QueryInterface( webBrowser );
        if ( webNavigation )
        {
                webNavigation->Stop( nsIWebNavigation::STOP_ALL );
        };

        nsCOMPtr< nsIWebBrowser > browser = nsnull;
        browserWindowIn->GetWebBrowser( getter_AddRefs( browser ) );
        nsCOMPtr< nsIBaseWindow > browserAsWin = do_QueryInterface( browser );
        if ( browserAsWin )
        {
                browserAsWin->Destroy();
        };


        browserWindowIn->SetWebBrowser( nsnull );

        NS_RELEASE( browserWindowIn );

        delete browserWindowIn;

        clearLastError();

        return true;
}

// Windows specific switches
#ifdef WIN32
        #pragma warning( 3 : 4291 ) // (no matching operator delete found; memory will not be freed if initialization throws an exception)
        #pragma warning( 3 : 4265 )     // "class has virtual functions, but destructor is not virtual"

        // #define required by this file for LibXUL/Mozilla code to avoid crashes in their debug code
        #ifdef _DEBUG
                #undef DEBUG
        #endif

#endif  // WIN32