Unreviewed followup to:
[WebKit-https.git] / Tools / TestWebKitAPI / Tests / WebKit / ResponsivenessTimerCrash.mm
1 /*
2  * Copyright (C) 2018 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 #import "config.h"
27
28 #if WK_API_ENABLED
29 #if WK_HAVE_C_SPI
30
31 #import "PlatformUtilities.h"
32 #import "TestWKWebView.h"
33 #import <WebKit/WKPagePrivateMac.h>
34 #import <WebKit/WKWebViewPrivate.h>
35 #import <wtf/RetainPtr.h>
36 #import <wtf/Vector.h>
37
38 static bool didBecomeUnresponsive;
39 static RetainPtr<TestWKWebView> webView;
40 static Vector<RetainPtr<id>> observableStates;
41 static bool webViewSeen;
42
43 @interface ResponsivenessTimerObserver : NSObject
44 @end
45
46 @implementation ResponsivenessTimerObserver
47
48 -(void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context
49 {
50     if (object == webView.get())
51         webViewSeen = true;
52
53     if (!observableStates.isEmpty())
54         observableStates.removeLast();
55     
56     if (webViewSeen && observableStates.isEmpty())
57         didBecomeUnresponsive = true;
58 }
59
60 @end
61
62 namespace TestWebKitAPI {
63
64 TEST(WebKit, ResponsivenessTimerCrash)
65 {
66     RetainPtr<ResponsivenessTimerObserver> observer = adoptNS([[ResponsivenessTimerObserver alloc] init]);
67     @autoreleasepool {
68         webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);
69         
70         [webView addObserver:observer.get() forKeyPath:@"_webProcessIsResponsive" options:0 context:nullptr];
71
72         auto pageRef = [webView _pageRefForTransitionToWKWebView];
73         
74         for (size_t i = 0; i < 10; ++i) {
75             RetainPtr<id> observableState = adoptNS(WKPageCreateObservableState(pageRef));
76             [observableState.get() addObserver:observer.get() forKeyPath:@"_webProcessIsResponsive" options:0 context:nullptr];
77             observableStates.append(WTFMove(observableState));
78         }
79
80         [webView synchronouslyLoadHTMLString:@"<script>document.addEventListener('keydown', function(){while(1){}});</script>"];
81         [webView typeCharacter:'a'];
82     }
83     
84     Util::run(&didBecomeUnresponsive);
85 }
86
87 } // namespace TestWebKitAPI
88
89 #endif // WK_HAVE_C_SPI
90 #endif // WK_API_ENABLED