blob: c4a667907c72877358e6d44bb91a438a0b7531a4 [file] [log] [blame]
// Copyright 2015 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.
#include <stddef.h>
#include <map>
#include <memory>
#include <string>
#include <unordered_set>
#include "base/macros.h"
#include "base/time/time.h"
#include "base/values.h"
#include "components/domain_reliability/beacon.h"
#include "components/domain_reliability/config.h"
#include "components/domain_reliability/context.h"
#include "components/domain_reliability/domain_reliability_export.h"
#include "url/gurl.h"
namespace domain_reliability {
class DOMAIN_RELIABILITY_EXPORT DomainReliabilityContextManager {
DomainReliabilityContext::Factory* context_factory);
// If |url| maps to a context added to this manager, calls |OnBeacon| on
// that context with |beacon|. Otherwise, does nothing.
void RouteBeacon(std::unique_ptr<DomainReliabilityBeacon> beacon);
void SetConfig(const GURL& origin,
std::unique_ptr<DomainReliabilityConfig> config,
base::TimeDelta max_age);
void ClearConfig(const GURL& origin);
// Calls |ClearBeacons| on all contexts matched by |origin_filter| added
// to this manager, but leaves the contexts themselves intact. A null
// |origin_filter| is interpreted as an always-true filter, indicating
// complete deletion.
void ClearBeacons(const base::Callback<bool(const GURL&)>& origin_filter);
// TODO(juliatuttle): Once unit tests test ContextManager directly, they can
// use a custom Context::Factory to get the created Context, and this can be
// void.
DomainReliabilityContext* AddContextForConfig(
std::unique_ptr<const DomainReliabilityConfig> config);
// Removes all contexts matched by |origin_filter| from this manager
// (discarding all queued beacons in the process). A null |origin_filter|
// is interpreted as an always-true filter, indicating complete deletion.
void RemoveContexts(const base::Callback<bool(const GURL&)>& origin_filter);
std::unique_ptr<base::Value> GetWebUIData() const;
size_t contexts_size_for_testing() const { return contexts_.size(); }
typedef std::map<std::string, DomainReliabilityContext*> ContextMap;
DomainReliabilityContext* GetContextForHost(const std::string& host);
DomainReliabilityContext::Factory* context_factory_;
// Owns DomainReliabilityContexts.
ContextMap contexts_;
// Currently, Domain Reliability only allows header-based configuration by
// origins that already have baked-in configs. This is the set of origins
// that have removed their context (by sending "NEL: max-age=0"), so the
// context manager knows they are allowed to set a config again later.
std::unordered_set<std::string> removed_contexts_;
} // namespace domain_reliability