1 /*

     2  * Copyright (C) 2010 Apple Inc. All rights reserved.

     3  *

     4  * Redistribution and use in source and binary forms, with or without

     5  * modification, are permitted provided that the following conditions

     6  * are met:

     7  * 1. Redistributions of source code must retain the above copyright

     8  *    notice, this list of conditions and the following disclaimer.

     9  * 2. Redistributions in binary form must reproduce the above copyright

    10  *    notice, this list of conditions and the following disclaimer in the

    11  *    documentation and/or other materials provided with the distribution.

    12  *

    13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''

    14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,

    15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

    16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS

    17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR

    18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF

    19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS

    20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN

    21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

    22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF

    23  * THE POSSIBILITY OF SUCH DAMAGE.

    24  */

    25 

    26 #include "InjectedBundle.h"

    27 

    28 #include "WKBundleAPICast.h"

    29 #include "WKBundleInitialize.h"

    30 

    31 #include <windows.h>

    32 #include <winbase.h>

    33 #include <shlobj.h>

    34 #include <shlwapi.h>

    35 

    36 using namespace WebCore;

    37 

    38 namespace WebKit {

    39 

    40 // FIXME: This should try and use <WebCore/FileSystem.h>.

    41 

    42 static String pathGetFileName(const String& path)

    43 {

    44     return String(::PathFindFileName(String(path).charactersWithNullTermination()));

    45 }

    46 

    47 static String directoryName(const String& path)

    48 {

    49     String fileName = pathGetFileName(path);

    50     String dirName = String(path);

    51     dirName.truncate(dirName.length() - pathGetFileName(path).length());

    52     return dirName;

    53 }

    54 

    55 bool InjectedBundle::load()

    56 {

    57     WCHAR currentPath[MAX_PATH];

    58     if (!::GetCurrentDirectoryW(MAX_PATH, currentPath))

    59         return false;

    60 

    61     String directorBundleResidesIn = directoryName(m_path);

    62     if (!::SetCurrentDirectoryW(directorBundleResidesIn.charactersWithNullTermination()))

    63         return false;

    64 

    65     m_platformBundle = ::LoadLibraryExW(m_path.charactersWithNullTermination(), 0, LOAD_WITH_ALTERED_SEARCH_PATH);

    66     if (!m_platformBundle)

    67         return false;

    68 

    69     // Reset the current directory.

    70     if (!::SetCurrentDirectoryW(currentPath)) {

    71         return false;

    72     }

    73 

    74     WKBundleInitializeFunctionPtr initializeFunction = reinterpret_cast<WKBundleInitializeFunctionPtr>(::GetProcAddress(m_platformBundle, "WKBundleInitialize"));

    75     if (!initializeFunction)

    76         return false;

    77 

    78     initializeFunction(toRef(this));

    79     return true;

    80 }

    81 

    82 } // namespace WebKit