| // Copyright 2020 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef CHROME_BROWSER_SAFE_BROWSING_CHROME_ENTERPRISE_URL_LOOKUP_SERVICE_H_ |
| #define CHROME_BROWSER_SAFE_BROWSING_CHROME_ENTERPRISE_URL_LOOKUP_SERVICE_H_ |
| |
| #include <memory> |
| #include <string> |
| |
| #include "chrome/browser/enterprise/connectors/connectors_service.h" |
| #include "components/safe_browsing/core/browser/realtime/url_lookup_service_base.h" |
| #include "components/safe_browsing/core/common/proto/csd.pb.h" |
| #include "url/gurl.h" |
| |
| namespace net { |
| struct NetworkTrafficAnnotationTag; |
| } |
| |
| namespace network { |
| class SharedURLLoaderFactory; |
| } // namespace network |
| |
| class Profile; |
| |
| namespace safe_browsing { |
| |
| class ReferrerChainProvider; |
| |
| // This class implements the real time lookup feature for a given user/profile. |
| // It is separated from the base class for logic that is related to enterprise |
| // users.(See: go/chrome-protego-enterprise-dd) |
| class ChromeEnterpriseRealTimeUrlLookupService |
| : public RealTimeUrlLookupServiceBase { |
| public: |
| ChromeEnterpriseRealTimeUrlLookupService( |
| scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory, |
| VerdictCacheManager* cache_manager, |
| Profile* profile, |
| base::RepeatingCallback<ChromeUserPopulation()> |
| get_user_population_callback, |
| enterprise_connectors::ConnectorsService* connectors_service, |
| ReferrerChainProvider* referrer_chain_provider); |
| |
| ChromeEnterpriseRealTimeUrlLookupService( |
| const ChromeEnterpriseRealTimeUrlLookupService&) = delete; |
| ChromeEnterpriseRealTimeUrlLookupService& operator=( |
| const ChromeEnterpriseRealTimeUrlLookupService&) = delete; |
| |
| ~ChromeEnterpriseRealTimeUrlLookupService() override; |
| |
| // RealTimeUrlLookupServiceBase: |
| bool CanPerformFullURLLookup() const override; |
| bool CanCheckSubresourceURL() const override; |
| bool CanCheckSafeBrowsingDb() const override; |
| |
| private: |
| // RealTimeUrlLookupServiceBase: |
| GURL GetRealTimeLookupUrl() const override; |
| net::NetworkTrafficAnnotationTag GetTrafficAnnotationTag() const override; |
| bool CanPerformFullURLLookupWithToken() const override; |
| bool CanAttachReferrerChain() const override; |
| int GetReferrerUserGestureLimit() const override; |
| bool CanSendPageLoadToken() const override; |
| void GetAccessToken( |
| const GURL& url, |
| const GURL& last_committed_url, |
| bool is_mainframe, |
| RTLookupRequestCallback request_callback, |
| RTLookupResponseCallback response_callback, |
| scoped_refptr<base::SequencedTaskRunner> callback_task_runner) override; |
| absl::optional<std::string> GetDMTokenString() const override; |
| std::string GetMetricSuffix() const override; |
| bool ShouldIncludeCredentials() const override; |
| double GetMinAllowedTimestampForReferrerChains() const override; |
| |
| // Unowned object used for checking profile based settings. |
| Profile* profile_; |
| |
| // Unowned pointer to ConnectorsService, used to get a DM token. |
| enterprise_connectors::ConnectorsService* connectors_service_; |
| |
| friend class ChromeEnterpriseRealTimeUrlLookupServiceTest; |
| |
| base::WeakPtrFactory<ChromeEnterpriseRealTimeUrlLookupService> weak_factory_{ |
| this}; |
| |
| }; // class ChromeEnterpriseRealTimeUrlLookupService |
| |
| } // namespace safe_browsing |
| |
| #endif // CHROME_BROWSER_SAFE_BROWSING_CHROME_ENTERPRISE_URL_LOOKUP_SERVICE_H_ |