Home | History | Annotate | Download | only in WebKit2
      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 "Test.h"
     27 
     28 #include "PlatformUtilities.h"
     29 #include "PlatformWebView.h"
     30 #include <JavaScriptCore/JavaScriptCore.h>
     31 #include <WebKit2/WKRetainPtr.h>
     32 #include <WebKit2/WebKit2.h>
     33 #include <WebKit2/WKSerializedScriptValue.h>
     34 
     35 namespace TestWebKitAPI {
     36 
     37 static bool testDone;
     38 
     39 static void didRunJavaScript(WKSerializedScriptValueRef resultSerializedScriptValue, WKErrorRef error, void* context)
     40 {
     41     TEST_ASSERT(context == reinterpret_cast<void*>(0x1234578));
     42     TEST_ASSERT(resultSerializedScriptValue);
     43 
     44     JSGlobalContextRef scriptContext = JSGlobalContextCreate(0);
     45     JSValueRef scriptValue = WKSerializedScriptValueDeserialize(resultSerializedScriptValue, scriptContext, 0);
     46     TEST_ASSERT(JSValueIsString(scriptContext, scriptValue));
     47 
     48     // Make sure that the result of navigator.userAgent isn't empty, even if we set the custom
     49     // user agent to the empty string.
     50     JSStringRef scriptString = JSValueToStringCopy(scriptContext, scriptValue, 0);
     51     TEST_ASSERT(JSStringGetLength(scriptString));
     52 
     53     JSStringRelease(scriptString);
     54     JSGlobalContextRelease(scriptContext);
     55 
     56     testDone = true;
     57 }
     58 
     59 TEST(WebKit2, PreventEmptyUserAgent)
     60 {
     61     WKRetainPtr<WKContextRef> context(AdoptWK, WKContextCreate());
     62     PlatformWebView webView(context.get());
     63 
     64     WKPageSetCustomUserAgent(webView.page(), WKStringCreateWithUTF8CString(""));
     65     WKRetainPtr<WKStringRef> javaScriptString(AdoptWK, WKStringCreateWithUTF8CString("navigator.userAgent"));
     66     WKPageRunJavaScriptInMainFrame(webView.page(), javaScriptString.get(), reinterpret_cast<void*>(0x1234578), didRunJavaScript);
     67 
     68     Util::run(&testDone);
     69 }
     70 
     71 } // namespace TestWebKitAPI
     72