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#pragma once
27
28#include <wtf/URL.h>
29#include <wtf/RefCounted.h>
30#include <wtf/RetainPtr.h>
31#include <wtf/text/WTFString.h>
32
33OBJC_CLASS NSAttributedString;
34OBJC_CLASS NSURL;
35OBJC_CLASS SSBServiceLookupResult;
36
37namespace WebKit {
38
39class SafeBrowsingWarning : public RefCounted<SafeBrowsingWarning> {
40public:
41#if HAVE(SAFE_BROWSING)
42 static Ref<SafeBrowsingWarning> create(const URL& url, bool forMainFrameNavigation, SSBServiceLookupResult *result)
43 {
44 return adoptRef(*new SafeBrowsingWarning(url, forMainFrameNavigation, result));
45 }
46#endif
47#if PLATFORM(COCOA)
48 static Ref<SafeBrowsingWarning> create(URL&& url, String&& title, String&& warning, RetainPtr<NSAttributedString>&& details)
49 {
50 return adoptRef(*new SafeBrowsingWarning(WTFMove(url), WTFMove(title), WTFMove(warning), WTFMove(details)));
51 }
52#endif
53
54 const URL& url() const { return m_url; }
55 const String& title() const { return m_title; }
56 const String& warning() const { return m_warning; }
57 bool forMainFrameNavigation() const { return m_forMainFrameNavigation; }
58#if PLATFORM(COCOA)
59 RetainPtr<NSAttributedString> details() const { return m_details; }
60#endif
61
62 static NSURL *visitUnsafeWebsiteSentinel();
63 static NSURL *confirmMalwareSentinel();
64
65private:
66#if HAVE(SAFE_BROWSING)
67 SafeBrowsingWarning(const URL&, bool, SSBServiceLookupResult *);
68#endif
69#if PLATFORM(COCOA)
70 SafeBrowsingWarning(URL&&, String&&, String&&, RetainPtr<NSAttributedString>&&);
71#endif
72
73 URL m_url;
74 String m_title;
75 String m_warning;
76 bool m_forMainFrameNavigation { false };
77#if PLATFORM(COCOA)
78 RetainPtr<NSAttributedString> m_details;
79#endif
80};
81
82} // namespace WebKit
83