1/*
2 * Copyright (C) 2008 Apple Inc. All Rights Reserved.
3 * Copyright (C) 2010 Google Inc. All Rights Reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
15 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR
18 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
19 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
20 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
21 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
22 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
24 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 */
26
27#pragma once
28
29#include "CSSPreloadScanner.h"
30#include "HTMLTokenizer.h"
31#include "SegmentedString.h"
32
33namespace WebCore {
34
35class TokenPreloadScanner {
36 WTF_MAKE_NONCOPYABLE(TokenPreloadScanner);
37public:
38 explicit TokenPreloadScanner(const URL& documentURL, float deviceScaleFactor = 1.0);
39
40 void scan(const HTMLToken&, PreloadRequestStream&, Document&);
41
42 void setPredictedBaseElementURL(const URL& url) { m_predictedBaseElementURL = url; }
43
44 bool inPicture() { return !m_pictureSourceState.isEmpty(); }
45
46private:
47 enum class TagId {
48 // These tags are scanned by the StartTagScanner.
49 Img,
50 Input,
51 Link,
52 Script,
53 Meta,
54 Source,
55
56 // These tags are not scanned by the StartTagScanner.
57 Unknown,
58 Style,
59 Base,
60 Template,
61 Picture
62 };
63
64 class StartTagScanner;
65
66 static TagId tagIdFor(const HTMLToken::DataVector&);
67
68 static String initiatorFor(TagId);
69
70 void updatePredictedBaseURL(const HTMLToken&);
71
72 CSSPreloadScanner m_cssScanner;
73 const URL m_documentURL;
74 const float m_deviceScaleFactor { 1 };
75
76 URL m_predictedBaseElementURL;
77 bool m_inStyle { false };
78
79 Vector<bool> m_pictureSourceState;
80
81 unsigned m_templateCount { 0 };
82};
83
84class HTMLPreloadScanner {
85 WTF_MAKE_FAST_ALLOCATED;
86public:
87 HTMLPreloadScanner(const HTMLParserOptions&, const URL& documentURL, float deviceScaleFactor = 1.0);
88
89 void appendToEnd(const SegmentedString&);
90 void scan(HTMLResourcePreloader&, Document&);
91
92private:
93 TokenPreloadScanner m_scanner;
94 SegmentedString m_source;
95 HTMLTokenizer m_tokenizer;
96};
97
98WEBCORE_EXPORT bool testPreloadScannerViewportSupport(Document*);
99
100} // namespace WebCore
101