1 | /* |
2 | * Copyright (C) 2011 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 "config.h" |
27 | |
28 | #if WK_HAVE_C_SPI |
29 | |
30 | #include "PlatformUtilities.h" |
31 | #include "PlatformWebView.h" |
32 | #include <WebKit/WKDownload.h> |
33 | |
34 | namespace TestWebKitAPI { |
35 | |
36 | static bool didDecideDestination; |
37 | |
38 | static void decidePolicyForNavigationAction(WKPageRef, WKNavigationActionRef, WKFramePolicyListenerRef listener, WKTypeRef, const void*) |
39 | { |
40 | WKFramePolicyListenerDownload(listener); |
41 | } |
42 | |
43 | static WKStringRef decideDestinationWithSuggestedFilename(WKContextRef, WKDownloadRef download, WKStringRef, bool*, const void*) |
44 | { |
45 | didDecideDestination = true; |
46 | WKDownloadCancel(download); |
47 | return Util::toWK("/tmp/WebKitAPITest/DownloadDecideDestinationCrash" ).leakRef(); |
48 | } |
49 | |
50 | static void setContextDownloadClient(WKContextRef context) |
51 | { |
52 | WKContextDownloadClientV0 client; |
53 | memset(&client, 0, sizeof(client)); |
54 | |
55 | client.base.version = 0; |
56 | client.decideDestinationWithSuggestedFilename = decideDestinationWithSuggestedFilename; |
57 | |
58 | WKContextSetDownloadClient(context, &client.base); |
59 | } |
60 | |
61 | static void setPagePolicyClient(WKPageRef page) |
62 | { |
63 | WKPageNavigationClientV0 navigationClient; |
64 | memset(&navigationClient, 0, sizeof(navigationClient)); |
65 | |
66 | navigationClient.base.version = 0; |
67 | navigationClient.decidePolicyForNavigationAction = decidePolicyForNavigationAction; |
68 | |
69 | WKPageSetPageNavigationClient(page, &navigationClient.base); |
70 | } |
71 | |
72 | TEST(WebKit, DownloadDecideDestinationCrash) |
73 | { |
74 | WKRetainPtr<WKContextRef> context = adoptWK(WKContextCreateWithConfiguration(nullptr)); |
75 | setContextDownloadClient(context.get()); |
76 | |
77 | PlatformWebView webView(context.get()); |
78 | setPagePolicyClient(webView.page()); |
79 | |
80 | // The length of this filename was specially chosen to trigger the crash conditions in |
81 | // <http://webkit.org/b/61142>. Specifically, it causes ArgumentDecoder::m_bufferPos and m_bufferEnd |
82 | // to be equal after the DecideDestinationWithSuggestedFilename message has been handled. |
83 | WKPageLoadURL(webView.page(), adoptWK(Util::createURLForResource("18-characters" , "html" )).get()); |
84 | |
85 | Util::run(&didDecideDestination); |
86 | } |
87 | |
88 | } // namespace TestWebKitAPI |
89 | |
90 | #endif |
91 | |